/*
	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/>.
*/
/**
 * Simplified in-memory representation of a Wasm AST.
 */

#pragma once

#include <variant>
#include <string>
#include <vector>
#include <map>
#include <memory>

namespace yul
{
namespace wasm
{

struct Literal;
struct StringLiteral;
struct LocalVariable;
struct GlobalVariable;
struct FunctionCall;
struct BuiltinCall;
struct LocalAssignment;
struct GlobalAssignment;
struct Block;
struct If;
struct Loop;
struct Break;
struct BreakIf;
struct Return;
using Expression = std::variant<
	Literal, StringLiteral, LocalVariable, GlobalVariable,
	FunctionCall, BuiltinCall, LocalAssignment, GlobalAssignment,
	Block, If, Loop, Break, BreakIf, Return
>;

struct Literal { uint64_t value; };
struct StringLiteral { std::string value; };
struct LocalVariable { std::string name; };
struct GlobalVariable { std::string name; };
struct Label { std::string name; };
struct FunctionCall { std::string functionName; std::vector<Expression> arguments; };
struct BuiltinCall { std::string functionName; std::vector<Expression> arguments; };
struct LocalAssignment { std::string variableName; std::unique_ptr<Expression> value; };
struct GlobalAssignment { std::string variableName; std::unique_ptr<Expression> value; };
struct Block { std::string labelName; std::vector<Expression> statements; };
struct If {
	std::unique_ptr<Expression> condition;
	std::vector<Expression> statements;
	std::unique_ptr<std::vector<Expression>> elseStatements;
};
struct Loop { std::string labelName; std::vector<Expression> statements; };
struct Break { Label label; };
struct Return {};
struct BreakIf { Label label; std::unique_ptr<Expression> condition; };

struct VariableDeclaration { std::string variableName; };
struct GlobalVariableDeclaration { std::string variableName; };
struct FunctionImport {
	std::string module;
	std::string externalName;
	std::string internalName;
	std::vector<std::string> paramTypes;
	std::unique_ptr<std::string> returnType;
};

struct FunctionDefinition
{
	std::string name;
	std::vector<std::string> parameterNames;
	bool returns;
	std::vector<VariableDeclaration> locals;
	std::vector<Expression> body;
};

/**
 * Abstract representation of a wasm module.
 */
struct Module
{
	std::vector<GlobalVariableDeclaration> globals;
	std::vector<FunctionImport> imports;
	std::vector<FunctionDefinition> functions;
	std::map<std::string, Module> subModules;
};

}
}