CompilerStack: Expose contracts per source unit

Author: erak

Diff for: libsolidity/interface/CompilerStack.cpp

@@ -1256,6 +1256,18 @@ ContractDefinition const& CompilerStack::contractDefinition(std::string const& _
 	return *contract(_contractName).contract;
 }
 
+std::vector<ContractDefinition const*> CompilerStack::contractDefinitions(
+	std::string const& _sourceName
+) const
+{
+	std::vector<ContractDefinition const*> contracts;
+	if (auto source = m_sources.find(_sourceName); source != m_sources.end())
+		for (auto const* contract: ASTNode::filteredNodes<ContractDefinition>(source->second.ast->nodes()))
+			contracts.emplace_back(contract);
+
+	return contracts;
+}
+
 size_t CompilerStack::functionEntryPoint(
 	std::string const& _contractName,
 	FunctionDefinition const& _function

Diff for: libsolidity/interface/CompilerStack.h

@@ -307,6 +307,9 @@ class CompilerStack: public langutil::CharStreamProvider, public evmasm::Abstrac
 	/// does not exist.
 	ContractDefinition const& contractDefinition(std::string const& _contractName) const;
 
+	/// @returns a list of parsed contracts per source unit with the supplied name.
+	std::vector<ContractDefinition const*> contractDefinitions(std::string const& _sourceName) const;
+
 	/// @returns a list of unhandled queries to the SMT solver (has to be supplied in a second run
 	/// by calling @a addSMTLib2Response).
 	std::vector<std::string> const& unhandledSMTLib2Queries() const { return m_unhandledSMTLib2Queries; }