diff options
-rw-r--r-- | CommandLineInterface.cpp | 41 | ||||
-rw-r--r-- | CommandLineInterface.h | 6 |
2 files changed, 25 insertions, 22 deletions
diff --git a/CommandLineInterface.cpp b/CommandLineInterface.cpp index 87edb24b..aa651eb4 100644 --- a/CommandLineInterface.cpp +++ b/CommandLineInterface.cpp @@ -61,6 +61,7 @@ static string const g_argBinaryStr = "binary"; static string const g_argOpcodesStr = "opcodes"; static string const g_argNatspecDevStr = "natspec-dev"; static string const g_argNatspecUserStr = "natspec-user"; +static string const g_argAddStandard = "add-std"; static void version() { @@ -116,13 +117,13 @@ void CommandLineInterface::handleBinary(string const& _contract) if (outputToStdout(choice)) { cout << "Binary: " << endl; - cout << toHex(m_compiler.getBytecode(_contract)) << endl; + cout << toHex(m_compiler->getBytecode(_contract)) << endl; } if (outputToFile(choice)) { ofstream outFile(_contract + ".binary"); - outFile << toHex(m_compiler.getBytecode(_contract)); + outFile << toHex(m_compiler->getBytecode(_contract)); outFile.close(); } } @@ -133,14 +134,14 @@ void CommandLineInterface::handleOpcode(string const& _contract) if (outputToStdout(choice)) { cout << "Opcodes: " << endl; - cout << eth::disassemble(m_compiler.getBytecode(_contract)); + cout << eth::disassemble(m_compiler->getBytecode(_contract)); cout << endl; } if (outputToFile(choice)) { ofstream outFile(_contract + ".opcode"); - outFile << eth::disassemble(m_compiler.getBytecode(_contract)); + outFile << eth::disassemble(m_compiler->getBytecode(_contract)); outFile.close(); } } @@ -191,13 +192,13 @@ void CommandLineInterface::handleMeta(DocumentationType _type, string const& _co if (outputToStdout(choice)) { cout << title << endl; - cout << m_compiler.getMetadata(_contract, _type) << endl; + cout << m_compiler->getMetadata(_contract, _type) << endl; } if (outputToFile(choice)) { ofstream outFile(_contract + suffix); - outFile << m_compiler.getMetadata(_contract, _type); + outFile << m_compiler->getMetadata(_contract, _type); outFile.close(); } } @@ -211,6 +212,7 @@ bool CommandLineInterface::parseArguments(int argc, char** argv) ("help", "Show help message and exit") ("version", "Show version and exit") ("optimize", po::value<bool>()->default_value(false), "Optimize bytecode for size") + ("add-std", po::value<bool>()->default_value(false), "Add standard contracts") ("input-file", po::value<vector<string>>(), "input file") (g_argAstStr.c_str(), po::value<OutputType>()->value_name("stdout|file|both"), "Request to output the AST of the contract.") @@ -292,31 +294,32 @@ bool CommandLineInterface::processInput() m_sourceCodes[infile] = asString(dev::contents(infile)); } + m_compiler.reset(new CompilerStack(m_args["add-std"].as<bool>())); try { for (auto const& sourceCode: m_sourceCodes) - m_compiler.addSource(sourceCode.first, sourceCode.second); + m_compiler->addSource(sourceCode.first, sourceCode.second); // TODO: Perhaps we should not compile unless requested - m_compiler.compile(m_args["optimize"].as<bool>()); + m_compiler->compile(m_args["optimize"].as<bool>()); } catch (ParserError const& _exception) { - SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Parser error", m_compiler); + SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Parser error", *m_compiler); return false; } catch (DeclarationError const& _exception) { - SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Declaration error", m_compiler); + SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Declaration error", *m_compiler); return false; } catch (TypeError const& _exception) { - SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Type error", m_compiler); + SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Type error", *m_compiler); return false; } catch (CompilerError const& _exception) { - SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Compiler error", m_compiler); + SourceReferenceFormatter::printExceptionInformation(cerr, _exception, "Compiler error", *m_compiler); return false; } catch (InternalCompilerError const& _exception) @@ -369,12 +372,12 @@ void CommandLineInterface::handleAst(string const& _argStr) cout << endl << "======= " << sourceCode.first << " =======" << endl; if (_argStr == g_argAstStr) { - ASTPrinter printer(m_compiler.getAST(sourceCode.first), sourceCode.second); + ASTPrinter printer(m_compiler->getAST(sourceCode.first), sourceCode.second); printer.print(cout); } else { - ASTJsonConverter converter(m_compiler.getAST(sourceCode.first)); + ASTJsonConverter converter(m_compiler->getAST(sourceCode.first)); converter.print(cout); } } @@ -388,12 +391,12 @@ void CommandLineInterface::handleAst(string const& _argStr) ofstream outFile(p.stem().string() + ".ast"); if (_argStr == g_argAstStr) { - ASTPrinter printer(m_compiler.getAST(sourceCode.first), sourceCode.second); + ASTPrinter printer(m_compiler->getAST(sourceCode.first), sourceCode.second); printer.print(outFile); } else { - ASTJsonConverter converter(m_compiler.getAST(sourceCode.first)); + ASTJsonConverter converter(m_compiler->getAST(sourceCode.first)); converter.print(outFile); } outFile.close(); @@ -408,7 +411,7 @@ void CommandLineInterface::actOnInput() handleAst(g_argAstStr); handleAst(g_argAstJson); - vector<string> contracts = m_compiler.getContractNames(); + vector<string> contracts = m_compiler->getContractNames(); for (string const& contract: contracts) { if (needStdout(m_args)) @@ -421,13 +424,13 @@ void CommandLineInterface::actOnInput() if (outputToStdout(choice)) { cout << "EVM assembly:" << endl; - m_compiler.streamAssembly(cout, contract); + m_compiler->streamAssembly(cout, contract); } if (outputToFile(choice)) { ofstream outFile(contract + ".evm"); - m_compiler.streamAssembly(outFile, contract); + m_compiler->streamAssembly(outFile, contract); outFile.close(); } } diff --git a/CommandLineInterface.h b/CommandLineInterface.h index 2862773b..79029f9d 100644 --- a/CommandLineInterface.h +++ b/CommandLineInterface.h @@ -21,9 +21,9 @@ */ #pragma once -#include <boost/program_options.hpp> - #include <libsolidity/CompilerStack.h> +#include <memory> +#include <boost/program_options.hpp> namespace dev { @@ -65,7 +65,7 @@ private: /// map of input files to source code strings std::map<std::string, std::string> m_sourceCodes; /// Solidity compiler stack - dev::solidity::CompilerStack m_compiler; + std::unique_ptr<dev::solidity::CompilerStack> m_compiler; }; } |