hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-17 09:13:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Kotlin: Get extractor working in a Java context

Browse Source
This commit is contained in:
Ian Lynagh
2021-07-27 18:14:57 +01:00
parent e3ecf4c52d
commit f15c6dede1
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
java/kotlin-extractor/src/main/kotlin/KotlinExtractorExtension.kt
View File

@@ -19,9 +19,9 @@ import org.jetbrains.kotlin.ir.IrFileEntry
class KotlinExtractorExtension(private val tests: List<String>) : IrGenerationExtension {
override fun generate(moduleFragment: IrModuleFragment, pluginContext: IrPluginContext) {
val trapDir = File(System.getenv("CODEQL_EXTRACTOR_KOTLIN_TRAP_DIR").takeUnless { it.isNullOrEmpty() } ?: "kotlin-extractor/trap")
val trapDir = File(System.getenv("CODEQL_EXTRACTOR_JAVA_TRAP_DIR").takeUnless { it.isNullOrEmpty() } ?: "kotlin-extractor/trap")
trapDir.mkdirs()
val srcDir = File(System.getenv("CODEQL_EXTRACTOR_KOTLIN_SOURCE_ARCHIVE_DIR").takeUnless { it.isNullOrEmpty() } ?: "kotlin-extractor/src")
val srcDir = File(System.getenv("CODEQL_EXTRACTOR_JAVA_SOURCE_ARCHIVE_DIR").takeUnless { it.isNullOrEmpty() } ?: "kotlin-extractor/src")
srcDir.mkdirs()
moduleFragment.accept(KotlinExtractorVisitor(trapDir, srcDir), RootTrapWriter())
// We don't want the compiler to continue and generate class
@@ -105,11 +105,14 @@ class KotlinExtractorVisitor(val trapDir: File, val srcDir: File) : IrElementVis
}
override fun visitClass(declaration: IrClass, data: TrapWriter) {
val id = data.getFreshId()
val pkgId = data.getFreshId()
val locId = data.getLocation(declaration.startOffset, declaration.endOffset)
val pkg = declaration.packageFqName?.asString() ?: ""
val cls = declaration.name.asString()
data.writeTrap("#$pkgId = @\"pkg;$pkg\"\n")
data.writeTrap("packages(#$pkgId, \"$pkg\")\n")
data.writeTrap("#$id = @\"class;$pkg.$cls\"\n")
data.writeTrap("classes(#$id, \"$cls\")\n")
data.writeTrap("classes(#$id, \"$cls\", #$pkgId, #$id)\n")
data.writeTrap("hasLocation(#$id, #$locId)\n")
declaration.acceptChildren(this, data)
}