Merge pull request #11520 from ethereum/local-streams-in-command-line-parser

Local output streams in CommandLineParser

Author: chriseth

solc/CommandLineInterface.cpp

@@ -70,7 +70,6 @@
 	#include <unistd.h>
 #endif
 
-#include <iostream>
 #include <fstream>
 
 #if !defined(STDERR_FILENO)
@@ -86,27 +85,21 @@ using namespace solidity::langutil;
 namespace solidity::frontend
 {
 
-namespace
+ostream& CommandLineInterface::sout(bool _markAsUsed)
 {
-
-static bool g_hasOutput = false;
-
-std::ostream& sout(bool _used = true)
-{
-	if (_used)
-		g_hasOutput = true;
-	return cout;
+	if (_markAsUsed)
+		m_hasOutput = true;
+	return m_sout;
 }
 
-std::ostream& serr(bool _used = true)
+ostream& CommandLineInterface::serr(bool _markAsUsed)
 {
-	if (_used)
-		g_hasOutput = true;
-	return cerr;
-}
-
+	if (_markAsUsed)
+		m_hasOutput = true;
+	return m_serr;
 }
 
+#define cin
 #define cout
 #define cerr
 
@@ -423,7 +416,7 @@ bool CommandLineInterface::readInputFilesAndConfigureFileReader()
 	}
 
 	if (m_options.input.addStdin)
-		m_fileReader.setSource(g_stdinFileName, readUntilEnd(cin));
+		m_fileReader.setSource(g_stdinFileName, readUntilEnd(m_sin));
 
 	if (m_fileReader.sourceCodes().size() == 0)
 	{
@@ -497,11 +490,11 @@ void CommandLineInterface::createJson(string const& _fileName, string const& _js
 
 bool CommandLineInterface::parseArguments(int _argc, char const* const* _argv)
 {
-	CommandLineParser parser;
+	CommandLineParser parser(sout(/* _markAsUsed */ false), serr(/* _markAsUsed */ false));
 	bool success = parser.parse(_argc, _argv, isatty(fileno(stdin)));
 	if (!success)
 		return false;
-	g_hasOutput = g_hasOutput || CommandLineParser::hasOutput();
+	m_hasOutput = m_hasOutput || parser.hasOutput();
 	m_options = parser.options();
 
 	return true;
@@ -521,7 +514,7 @@ bool CommandLineInterface::processInput()
 	{
 		string input;
 		if (m_options.input.standardJsonFile.empty())
-			input = readUntilEnd(cin);
+			input = readUntilEnd(m_sin);
 		else
 		{
 			try
@@ -626,7 +619,7 @@ bool CommandLineInterface::compile()
 
 		for (auto const& error: m_compiler->errors())
 		{
-			g_hasOutput = true;
+			m_hasOutput = true;
 			formatter.printErrorInformation(*error);
 		}
 
@@ -635,7 +628,7 @@ bool CommandLineInterface::compile()
 	}
 	catch (CompilerError const& _exception)
 	{
-		g_hasOutput = true;
+		m_hasOutput = true;
 		formatter.printExceptionInformation(_exception, "Compiler error");
 		return false;
 	}
@@ -669,7 +662,7 @@ bool CommandLineInterface::compile()
 			serr() << "Documentation parsing error: " << *boost::get_error_info<errinfo_comment>(_error) << endl;
 		else
 		{
-			g_hasOutput = true;
+			m_hasOutput = true;
 			formatter.printExceptionInformation(_error, _error.typeName());
 		}
 
@@ -985,7 +978,7 @@ bool CommandLineInterface::assemble(
 
 		for (auto const& error: stack.errors())
 		{
-			g_hasOutput = true;
+			m_hasOutput = true;
 			formatter.printErrorInformation(*error);
 		}
 		if (!Error::containsOnlyWarnings(stack.errors()))
@@ -1134,7 +1127,7 @@ void CommandLineInterface::outputCompilationResults()
 		handleNatspec(false, contract);
 	} // end of contracts iteration
 
-	if (!g_hasOutput)
+	if (!m_hasOutput)
 	{
 		if (!m_options.output.dir.empty())
 			sout() << "Compiler run successful. Artifact(s) can be found in directory " << m_options.output.dir << "." << endl;

solc/CommandLineInterface.h

@@ -29,6 +29,7 @@
 #include <libsolidity/interface/FileReader.h>
 #include <libyul/AssemblyStack.h>
 
+#include <iostream>
 #include <memory>
 #include <string>
 
@@ -38,7 +39,15 @@ namespace solidity::frontend
 class CommandLineInterface
 {
 public:
-	explicit CommandLineInterface(CommandLineOptions const& _options = CommandLineOptions{}):
+	explicit CommandLineInterface(
+		std::istream& _sin,
+		std::ostream& _sout,
+		std::ostream& _serr,
+		CommandLineOptions const& _options = CommandLineOptions{}
+	):
+		m_sin(_sin),
+		m_sout(_sout),
+		m_serr(_serr),
 		m_options(_options)
 	{}
 
@@ -106,6 +115,18 @@ class CommandLineInterface
 	/// @arg _json json string to be written
 	void createJson(std::string const& _fileName, std::string const& _json);
 
+	/// Returns the stream that should receive normal output. Sets m_hasOutput to true if the
+	/// stream has ever been used unless @arg _markAsUsed is set to false.
+	std::ostream& sout(bool _markAsUsed = true);
+
+	/// Returns the stream that should receive error output. Sets m_hasOutput to true if the
+	/// stream has ever been used unless @arg _markAsUsed is set to false.
+	std::ostream& serr(bool _markAsUsed = true);
+
+	std::istream& m_sin;
+	std::ostream& m_sout;
+	std::ostream& m_serr;
+	bool m_hasOutput = false;
 	bool m_error = false; ///< If true, some error occurred.
 	FileReader m_fileReader;
 	std::unique_ptr<frontend::CompilerStack> m_compiler;

solc/CommandLineParser.cpp

@@ -34,24 +34,16 @@ namespace po = boost::program_options;
 namespace solidity::frontend
 {
 
-static bool g_hasOutput = false;
-
-namespace
-{
-
-std::ostream& sout()
+ostream& CommandLineParser::sout()
 {
-	g_hasOutput = true;
-	return cout;
+	m_hasOutput = true;
+	return m_sout;
 }
 
-std::ostream& serr(bool _used = true)
+ostream& CommandLineParser::serr()
 {
-	if (_used)
-		g_hasOutput = true;
-	return cerr;
-}
-
+	m_hasOutput = true;
+	return m_serr;
 }
 
 #define cout
@@ -185,8 +177,7 @@ static set<string> const g_metadataHashArgs
 	g_strNone
 };
 
-[[noreturn]]
-static void printVersionAndExit()
+void CommandLineParser::printVersionAndExit()
 {
 	sout() <<
 		"solc, the solidity compiler commandline interface" <<
@@ -197,19 +188,16 @@ static void printVersionAndExit()
 	exit(EXIT_SUCCESS);
 }
 
-[[noreturn]]
-static void printLicenseAndExit()
+void CommandLineParser::printLicenseAndExit()
 {
 	sout() << otherLicenses << endl;
 	// This is a static variable generated by cmake from LICENSE.txt
 	sout() << licenseText << endl;
 	exit(EXIT_SUCCESS);
 }
 
-namespace
-{
 
-bool checkMutuallyExclusive(boost::program_options::variables_map const& args, std::string const& _optionA, std::string const& _optionB)
+bool CommandLineParser::checkMutuallyExclusive(boost::program_options::variables_map const& args, string const& _optionA, string const& _optionB)
 {
 	if (args.count(_optionA) && args.count(_optionB))
 	{
@@ -220,8 +208,6 @@ bool checkMutuallyExclusive(boost::program_options::variables_map const& args, s
 	return true;
 }
 
-}
-
 bool CompilerOutputs::operator==(CompilerOutputs const& _other) const noexcept
 {
 	static_assert(
@@ -441,7 +427,7 @@ bool CommandLineParser::parseLibraryOption(string const& _input)
 
 bool CommandLineParser::parse(int _argc, char const* const* _argv, bool interactiveTerminal)
 {
-	g_hasOutput = false;
+	m_hasOutput = false;
 
 	// Declare the supported options.
 	po::options_description desc((R"(solc, the Solidity commandline compiler.
@@ -1136,11 +1122,6 @@ General Information)").c_str(),
 	return true;
 }
 
-bool CommandLineParser::hasOutput()
-{
-	return g_hasOutput;
-}
-
 bool CommandLineParser::parseCombinedJsonOption()
 {
 	if (!m_args.count(g_strCombinedJson))
@@ -1192,4 +1173,5 @@ string CommandLineParser::joinOptionNames(vector<string> const& _optionNames, st
 		_separator
 	);
 }
+
 } // namespace solidity::frontend

solc/CommandLineParser.h

@@ -33,6 +33,7 @@
 #include <map>
 #include <memory>
 #include <optional>
+#include <ostream>
 #include <set>
 #include <string>
 #include <vector>
@@ -177,6 +178,11 @@ struct CommandLineOptions
 class CommandLineParser
 {
 public:
+	explicit CommandLineParser(std::ostream& _sout, std::ostream& _serr):
+		m_sout(_sout),
+		m_serr(_serr)
+	{}
+
 	/// Parses the command-line arguments and fills out the internal CommandLineOptions structure.
 	/// Performs validation and prints error messages. If requested, prints usage banner, version
 	/// or license.
@@ -191,9 +197,8 @@ class CommandLineParser
 
 	CommandLineOptions const& options() const { return m_options; }
 
-	/// Returns true if any parser instance has written anything to cout or cerr since the last
-	/// call to parse().
-	static bool hasOutput();
+	/// Returns true if the parser has written anything to any of its output streams.
+	bool hasOutput() const { return m_hasOutput; }
 
 private:
 	/// Parses the value supplied to --combined-json.
@@ -210,9 +215,28 @@ class CommandLineParser
 	/// @return false if there are any validation errors, true otherwise.
 	bool parseLibraryOption(std::string const& _input);
 
+	bool checkMutuallyExclusive(
+		boost::program_options::variables_map const& args,
+		std::string const& _optionA,
+		std::string const& _optionB
+	);
+	[[noreturn]] void printVersionAndExit();
+	[[noreturn]] void printLicenseAndExit();
 	size_t countEnabledOptions(std::vector<std::string> const& _optionNames) const;
 	static std::string joinOptionNames(std::vector<std::string> const& _optionNames, std::string _separator = ", ");
 
+	/// Returns the stream that should receive normal output. Sets m_hasOutput to true if the
+	/// stream has ever been used.
+	std::ostream& sout();
+
+	/// Returns the stream that should receive error output. Sets m_hasOutput to true if the
+	/// stream has ever been used.
+	std::ostream& serr();
+
+	std::ostream& m_sout;
+	std::ostream& m_serr;
+	bool m_hasOutput = false;
+
 	CommandLineOptions m_options;
 
 	/// Map of command-line arguments produced by boost::program_options.

solc/main.cpp

@@ -54,7 +54,7 @@ static void setDefaultOrCLocale()
 int main(int argc, char** argv)
 {
 	setDefaultOrCLocale();
-	solidity::frontend::CommandLineInterface cli;
+	solidity::frontend::CommandLineInterface cli(cin, cout, cerr);
 	if (!cli.parseArguments(argc, argv))
 		return 1;
 	if (!cli.processInput())