feat(c): add support for parsing #include directives

This commit adds support for parsing #include directives in C code. The `CAnalyser` class now uses a regular expression to extract the included file paths and stores them in the `includesDirective` property. This allows for better analysis of code dependencies. Additionally, a test case in `CFullIdentListenerTest` has been updated to reflect the new behavior.
phodal · Feb 3, 2024 · e1cda08 · e1cda08
1 parent 037695d
commit e1cda08
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/chapi-ast-c/src/main/kotlin/chapi/ast/cast/CAnalyser.kt b/chapi-ast-c/src/main/kotlin/chapi/ast/cast/CAnalyser.kt
@@ -30,7 +30,13 @@ open class CAnalyser : Analyser {
         pp.addInput(LexerSource(InputStreamReader(code.byteInputStream()), true))
     }
 
+    private val importRegex = Regex("""#include\s+(<[^>]+>|\"[^\"]+\")""")
+
     override fun analysis(code: String, filePath: String): CodeContainer {
+        includesDirective = importRegex.findAll(code).map {
+            it.groupValues[1]
+        }.toMutableList()
+
         val output = preProcessing(code)
 
         val context = this.parse(output).compilationUnit()

diff --git a/chapi-ast-c/src/test/kotlin/chapi/ast/cast/CFullIdentListenerTest.kt b/chapi-ast-c/src/test/kotlin/chapi/ast/cast/CFullIdentListenerTest.kt
@@ -61,8 +61,8 @@ static RedisModuleType *MemAllocType;
 """
         val codeFile = CAnalyser().analysis(code, "helloworld.c")
 
-        assertEquals(codeFile.Imports.size, 0)
-//        assertEquals(codeFile.Imports[0].Source, "stdio.h")
+        assertEquals(codeFile.Imports.size, 1)
+        assertEquals(codeFile.Imports[0].Source, "stdio.h")
     }
 
     @Test
@@ -300,7 +300,7 @@ typedef struct {
             """.trimIndent()
 
         val codeFile = CAnalyser().analysis(code, "helloworld.c")
-        assertEquals(codeFile.Imports.size, 0)
+        assertEquals(codeFile.Imports.size, 6)
     }
 
     @Test