/*

  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/>.

*/

// SPDX-License-Identifier: GPL-3.0

#include <libevmasm/EVMAssemblyStack.h>

#include <libsolutil/JSON.h>

#include <liblangutil/Exceptions.h>

#include <libsolidity/codegen/CompilerContext.h>

#include <range/v3/view/enumerate.hpp>

#include <range/v3/view/transform.hpp>

#include <tuple>

using namespace solidity::util;

using namespace solidity::langutil;

using namespace solidity::frontend;

namespace solidity::evmasm

{

void EVMAssemblyStack::parseAndAnalyze(std::string const& _sourceName, std::string const& _source)

{

  Json assemblyJson;

  solRequire(jsonParseStrict(_source, assemblyJson), AssemblyImportException, "Could not parse JSON file.");

  analyze(_sourceName, assemblyJson);

}

void EVMAssemblyStack::analyze(std::string const& _sourceName, Json const& _assemblyJson)

{

  solAssert(!m_evmAssembly);

  m_name = _sourceName;

  std::tie(m_evmAssembly, m_sourceList) = evmasm::Assembly::fromJSON(_assemblyJson, {}, 0, m_eofVersion);

  solRequire(m_evmAssembly != nullptr, AssemblyImportException, "Could not create evm assembly object.");

}

void EVMAssemblyStack::assemble()

{

  solAssert(m_evmAssembly);

  solAssert(m_evmAssembly->isCreation());

  solAssert(!m_evmRuntimeAssembly);

  m_evmAssembly->optimise(m_optimiserSettings);

  m_object = m_evmAssembly->assemble();

  // TODO: Check for EOF

  solAssert(m_evmAssembly->codeSections().size() == 1);

  m_sourceMapping = AssemblyItem::computeSourceMapping(m_evmAssembly->codeSections().front().items, sourceIndices());

  if (m_evmAssembly->numSubs() > 0)

  {

    m_evmRuntimeAssembly = std::make_shared<evmasm::Assembly>(m_evmAssembly->sub(0));

    solAssert(m_evmRuntimeAssembly && !m_evmRuntimeAssembly->isCreation());

    // TODO: Check for EOF

    solAssert(m_evmRuntimeAssembly->codeSections().size() == 1);

    m_runtimeSourceMapping = AssemblyItem::computeSourceMapping(m_evmRuntimeAssembly->codeSections().front().items, sourceIndices());

    m_runtimeObject = m_evmRuntimeAssembly->assemble();

  }

}

LinkerObject const& EVMAssemblyStack::object(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  return object();

}

LinkerObject const& EVMAssemblyStack::runtimeObject(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  return runtimeObject();

}

std::map<std::string, unsigned> EVMAssemblyStack::sourceIndices() const

{

  solAssert(m_evmAssembly);

  return m_sourceList

    | ranges::views::enumerate

    | ranges::views::transform([](auto const& _source) { return std::make_pair(_source.second, _source.first); })

    | ranges::to<std::map<std::string, unsigned>>;

}

std::string const* EVMAssemblyStack::sourceMapping(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  return &sourceMapping();

}

std::string const* EVMAssemblyStack::runtimeSourceMapping(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  return &runtimeSourceMapping();

}

Json EVMAssemblyStack::ethdebug(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  solAssert(m_ethdebug != nullptr);

  return *m_ethdebug;

}

Json EVMAssemblyStack::ethdebugRuntime(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  solAssert(m_ethdebugRuntime != nullptr);

  return *m_ethdebugRuntime;

}

Json EVMAssemblyStack::ethdebug() const

{

  return {};

}

Json EVMAssemblyStack::assemblyJSON() const

{

  solAssert(m_evmAssembly);

  return m_evmAssembly->assemblyJSON(sourceIndices());

}

Json EVMAssemblyStack::assemblyJSON(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  return assemblyJSON();

}

std::string EVMAssemblyStack::assemblyString(StringMap const& _sourceCodes) const

{

  solAssert(m_evmAssembly);

  return m_evmAssembly->assemblyString(m_debugInfoSelection, _sourceCodes);

}

std::string EVMAssemblyStack::assemblyString(std::string const& _contractName, StringMap const& _sourceCodes) const

{

  solAssert(_contractName == m_name);

  return assemblyString(_sourceCodes);

}

std::string const EVMAssemblyStack::filesystemFriendlyName(std::string const& _contractName) const

{

  solAssert(_contractName == m_name);

  return m_name;

}

std::vector<std::string> EVMAssemblyStack::sourceNames() const

{

  return m_sourceList;

}

} // namespace solidity::evmasm