hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2026-05-22 07:07:09 +02:00
Code Issues Packages Projects Releases Wiki Activity

KE2: Handle multiple files

Browse Source
This commit is contained in:
Ian Lynagh
2024-08-22 18:08:54 +01:00
parent 1bd1789861
commit 1cfbc8e86d
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
java/kotlin-extractor2/src/main/kotlin/KotlinExtractor.kt
View File

@@ -45,14 +45,20 @@ fun main(args : Array<String>) {
}
}
val ktFile = session.modulesWithFiles.getValue(sourceModule).single() as KtFile
analyze(ktFile) {
val c = ktFile.getDeclarations()[0] as KtClass
for (d: KtDeclaration in c.getDeclarations()) {
val f = d as KtFunction
if (f.name == "f") {
dumpFunction(f)
val psiFiles = session.modulesWithFiles.getValue(sourceModule)
for (psiFile in psiFiles) {
if (psiFile is KtFile) {
analyze(psiFile) {
val c = psiFile.getDeclarations()[0] as KtClass
for (d: KtDeclaration in c.getDeclarations()) {
val f = d as KtFunction
if (f.name == "f") {
dumpFunction(f)
}
}
}
} else {
System.out.println("Warning: Not a KtFile")
}
}
}