aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--libsolidity/ast/AST.cpp10
-rw-r--r--libsolidity/ast/AST.h3
2 files changed, 13 insertions, 0 deletions
diff --git a/libsolidity/ast/AST.cpp b/libsolidity/ast/AST.cpp
index 1e1e4839..c5fb35f6 100644
--- a/libsolidity/ast/AST.cpp
+++ b/libsolidity/ast/AST.cpp
@@ -84,6 +84,16 @@ SourceUnitAnnotation& SourceUnit::annotation() const
return dynamic_cast<SourceUnitAnnotation&>(*m_annotation);
}
+vector<SourceUnit const*> SourceUnit::referencedSourceUnits() const
+{
+ vector<SourceUnit const*> sourceUnits;
+ for (ImportDirective const* importDirective: filteredNodes<ImportDirective>(nodes()))
+ {
+ sourceUnits.push_back(importDirective->annotation().sourceUnit);
+ }
+ return sourceUnits;
+}
+
SourceUnit const& Declaration::sourceUnit() const
{
solAssert(!!m_scope, "");
diff --git a/libsolidity/ast/AST.h b/libsolidity/ast/AST.h
index 41785780..e2117613 100644
--- a/libsolidity/ast/AST.h
+++ b/libsolidity/ast/AST.h
@@ -136,6 +136,9 @@ public:
std::vector<ASTPointer<ASTNode>> nodes() const { return m_nodes; }
+ /// @returns a vector of referenced SourceUnits.
+ std::vector<SourceUnit const*> referencedSourceUnits() const;
+
private:
std::vector<ASTPointer<ASTNode>> m_nodes;
};