/*
	This file is part of solidity.

	solidity is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	solidity is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with solidity.  If not, see <http://www.gnu.org/licenses/>.
*/
/**
 * @author Christian <c@ethdev.com>
 * @date 2016
 * Framework for executing Solidity contracts and testing them against C++ implementation.
 */

#include <cstdlib>
#include <iostream>
#include <boost/test/framework.hpp>
#include <test/libsolidity/SolidityExecutionFramework.h>

using namespace solidity;
using namespace solidity::test;
using namespace solidity::frontend;
using namespace solidity::frontend::test;
using namespace std;

bytes SolidityExecutionFramework::multiSourceCompileContract(
	map<string, string> const& _sourceCode,
	string const& _contractName,
	map<string, Address> const& _libraryAddresses
)
{
	map<string, string> sourcesWithPreamble = _sourceCode;
	for (auto& entry: sourcesWithPreamble)
		entry.second = addPreamble(entry.second);

	m_compiler.reset();
	m_compiler.setSources(sourcesWithPreamble);
	m_compiler.setLibraries(_libraryAddresses);
	m_compiler.setRevertStringBehaviour(m_revertStrings);
	m_compiler.setEVMVersion(m_evmVersion);
	m_compiler.setOptimiserSettings(m_optimiserSettings);
	m_compiler.enableIRGeneration(m_compileViaYul);
	m_compiler.setRevertStringBehaviour(m_revertStrings);
	if (!m_compiler.compile())
	{
		langutil::SourceReferenceFormatter formatter(std::cerr);

		for (auto const& error: m_compiler.errors())
			formatter.printErrorInformation(*error);
		BOOST_ERROR("Compiling contract failed");
	}
	std::string contractName(_contractName.empty() ? m_compiler.lastContractName() : _contractName);
	evmasm::LinkerObject obj;
	if (m_compileViaYul)
	{
		// Try compiling twice: If the first run fails due to stack errors, forcefully enable
		// the optimizer.
		for (bool forceEnableOptimizer: {false, true})
		{
			OptimiserSettings optimiserSettings = m_optimiserSettings;
			if (!forceEnableOptimizer && !optimiserSettings.runYulOptimiser)
			{
				// Enable some optimizations on the first run
				optimiserSettings.runYulOptimiser = true;
				optimiserSettings.yulOptimiserSteps = "uljmul jmul";
			}
			else if (forceEnableOptimizer)
				optimiserSettings = OptimiserSettings::full();

			yul::AssemblyStack asmStack(
				m_evmVersion,
				yul::AssemblyStack::Language::StrictAssembly,
				optimiserSettings
			);
			bool analysisSuccessful = asmStack.parseAndAnalyze("", m_compiler.yulIROptimized(contractName));
			solAssert(analysisSuccessful, "Code that passed analysis in CompilerStack can't have errors");

			try
			{
				asmStack.optimize();
				obj = std::move(*asmStack.assemble(yul::AssemblyStack::Machine::EVM).bytecode);
				break;
			}
			catch (...)
			{
				if (forceEnableOptimizer || optimiserSettings == OptimiserSettings::full())
					throw;
			}
		}
	}
	else
		obj = m_compiler.object(contractName);
	BOOST_REQUIRE(obj.linkReferences.empty());
	if (m_showMetadata)
		cout << "metadata: " << m_compiler.metadata(contractName) << endl;
	return obj.bytecode;
}

bytes SolidityExecutionFramework::compileContract(
	string const& _sourceCode,
	string const& _contractName,
	map<string, Address> const& _libraryAddresses
)
{
	return multiSourceCompileContract(
		{{"", _sourceCode}},
		_contractName,
		_libraryAddresses
	);
}

string SolidityExecutionFramework::addPreamble(string const& _sourceCode)
{
	// Silence compiler version warning
	string preamble = "pragma solidity >=0.0;\n";
	if (
		solidity::test::CommonOptions::get().useABIEncoderV2 &&
		_sourceCode.find("pragma experimental ABIEncoderV2;") == string::npos
	)
		preamble += "pragma experimental ABIEncoderV2;\n";
	return preamble + _sourceCode;
}