/*
	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 2014
 * Framework for executing Solidity contracts and testing them against C++ implementation.
 */

#pragma once

#include <functional>

#include <test/ExecutionFramework.h>

#include <libsolidity/interface/CompilerStack.h>

#include <libyul/AssemblyStack.h>

#include <liblangutil/Exceptions.h>
#include <liblangutil/SourceReferenceFormatter.h>

namespace dev
{
namespace solidity
{

namespace test
{

class SolidityExecutionFramework: public dev::test::ExecutionFramework
{

public:
	SolidityExecutionFramework();
	SolidityExecutionFramework(std::string const& _ipcPath, langutil::EVMVersion _evmVersion);

	virtual bytes const& compileAndRunWithoutCheck(
		std::string const& _sourceCode,
		u256 const& _value = 0,
		std::string const& _contractName = "",
		bytes const& _arguments = bytes(),
		std::map<std::string, dev::test::Address> const& _libraryAddresses = std::map<std::string, dev::test::Address>()
	) override
	{
		bytes bytecode = compileContract(_sourceCode, _contractName, _libraryAddresses);
		sendMessage(bytecode + _arguments, true, _value);
		return m_output;
	}

	bytes compileContract(
		std::string const& _sourceCode,
		std::string const& _contractName = "",
		std::map<std::string, dev::test::Address> const& _libraryAddresses = std::map<std::string, dev::test::Address>()
	)
	{
		// Silence compiler version warning
		std::string sourceCode = "pragma solidity >=0.0;\n";
		if (dev::test::Options::get().useABIEncoderV2 && _sourceCode.find("pragma experimental ABIEncoderV2;") == std::string::npos)
			sourceCode += "pragma experimental ABIEncoderV2;\n";
		sourceCode += _sourceCode;
		m_compiler.reset();
		m_compiler.setSources({{"", sourceCode}});
		m_compiler.setLibraries(_libraryAddresses);
		m_compiler.setEVMVersion(m_evmVersion);
		m_compiler.setOptimiserSettings(m_optimiserSettings);
		m_compiler.enableIRGeneration(m_compileViaYul);
		if (!m_compiler.compile())
		{
			langutil::SourceReferenceFormatter formatter(std::cerr);

			for (auto const& error: m_compiler.errors())
				formatter.printExceptionInformation(
					*error,
					(error->type() == langutil::Error::Type::Warning) ? "Warning" : "Error"
				);
			BOOST_ERROR("Compiling contract failed");
		}
		eth::LinkerObject obj;
		if (m_compileViaYul)
		{
			yul::AssemblyStack asmStack(
				m_evmVersion,
				yul::AssemblyStack::Language::StrictAssembly,
				m_optimiserSettings
			);
			if (!asmStack.parseAndAnalyze("", m_compiler.yulIROptimized(
				_contractName.empty() ? m_compiler.lastContractName() : _contractName
			)))
			{
				langutil::SourceReferenceFormatter formatter(std::cerr);

				for (auto const& error: m_compiler.errors())
					formatter.printExceptionInformation(
						*error,
						(error->type() == langutil::Error::Type::Warning) ? "Warning" : "Error"
					);
				BOOST_ERROR("Assembly contract failed. IR: " + m_compiler.yulIROptimized({}));
			}
			asmStack.optimize();
			obj = std::move(*asmStack.assemble(yul::AssemblyStack::Machine::EVM).bytecode);
		}
		else
			obj = m_compiler.object(_contractName.empty() ? m_compiler.lastContractName() : _contractName);
		BOOST_REQUIRE(obj.linkReferences.empty());
		return obj.bytecode;
	}

protected:
	dev::solidity::CompilerStack m_compiler;
	bool m_compileViaYul = false;
};

}
}
} // end namespaces