Populating codeOf of ASTModule (#1634)

* Populating `codeOf` of `ASTModule` This error message was annoying me, so here is a fix. I am not sure if it is easy to find the "location" of the module. The region should be able to be computed out of the lengt of `fileContents` and its actual content. I actually thought we had a function like that somewhere @konradweiss? * make sonarcube happy --------- Co-authored-by: Maximilian Kaul <[email protected]>
Fraunhofer-AISEC · Jul 26, 2024 · 2076f44 · 2076f44
1 parent fdc642b
commit 2076f44
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/...python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonLanguageFrontend.kt b/...python/src/main/kotlin/de/fraunhofer/aisec/cpg/frontends/python/PythonLanguageFrontend.kt
@@ -181,15 +181,20 @@ class PythonLanguageFrontend(language: Language<PythonLanguageFrontend>, ctx: Tr
      * 3) Remove trailing whitespaces / tabs
      */
     override fun codeOf(astNode: Python.AST): String? {
-        val location = locationOf(astNode)
-        if (location != null) {
-            var lines = getRelevantLines(location)
-            lines = removeExtraAtEnd(location, lines)
-            lines = fixStartColumn(location, lines)
+        return if (astNode is Python.ASTModule) {
+            fileContent
+        } else {
+            val location = locationOf(astNode)
+            if (location != null) {
+                var lines = getRelevantLines(location)
+                lines = removeExtraAtEnd(location, lines)
+                lines = fixStartColumn(location, lines)
 
-            return lines.joinToString(separator = lineSeparator.toString())
+                lines.joinToString(separator = lineSeparator.toString())
+            } else {
+                null
+            }
         }
-        return null
     }
 
     private fun getRelevantLines(location: PhysicalLocation): MutableList<String> {