Skip to content

Commit 57c39e9

Browse files
authored
Merge pull request #8574 from erik-krogh/qlForQlFixes
QL: two small improvements
2 parents 5efc19c + 77aff04 commit 57c39e9

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

ql/ql/src/codeql_ql/ast/Ast.qll

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,8 @@ class TopLevel extends TTopLevel, AstNode {
135135
pred = directMember("getAModule") and result = this.getAModule()
136136
or
137137
pred = directMember("getANewType") and result = this.getANewType()
138+
or
139+
pred = directMember("getQLDoc") and result = this.getQLDoc()
138140
}
139141

140142
QLDoc getQLDocFor(ModuleMember m) {
@@ -1127,6 +1129,21 @@ class Import extends TImport, ModuleMember, ModuleRef {
11271129
result = imp.getChild(0).(QL::ImportModuleExpr).getChild().getName(i).getValue()
11281130
}
11291131

1132+
/**
1133+
* Gets the full string specifying the module being imported.
1134+
*/
1135+
string getImportString() {
1136+
exists(string selec |
1137+
not exists(getQualifiedName(_)) and selec = ""
1138+
or
1139+
selec =
1140+
"::" + strictconcat(int i, string q | q = this.getSelectionName(i) | q, "::" order by i)
1141+
|
1142+
result =
1143+
strictconcat(int i, string q | q = this.getQualifiedName(i) | q, "." order by i) + selec
1144+
)
1145+
}
1146+
11301147
final override FileOrModule getResolvedModule() { resolve(this, result) }
11311148
}
11321149

0 commit comments

Comments
 (0)