feat(chapi-ast-cpp): add support for importing C++ headers

This commit adds support for importing C++ headers in the CPPBasicIdentListener class. The includes are extracted from the code using a regular expression and added to the CodeContainer as CodeImport objects. This allows for better analysis of C++ code that relies on external headers.

Author: phodal

chapi-ast-c/src/test/kotlin/chapi/ast/cast/CFullIdentListenerTest.kt

@@ -304,7 +304,7 @@ typedef struct {
     }
 
     @Test
-    fun readlWorld() {
+    fun shouldSuccessHandleDs() {
         val code = """
             // https://stackoverflow.com/questions/12642830/can-i-define-a-function-inside-a-c-structure
             #include <stdio.h>

chapi-ast-cpp/src/main/kotlin/chapi/ast/cppast/CPPAnalyser.kt

@@ -45,11 +45,16 @@ open class CPPAnalyser: Analyser  {
         return output
     }
 
+    private val importRegex = Regex("""#include\s+(<[^>]+>|\"[^\"]+\")""")
 
     override fun analysis(code: String, filePath: String): CodeContainer {
         val processedCode = preProcessing(code)
+        val includesDirective = importRegex.findAll(code).map {
+            it.groupValues[1]
+        }.toMutableList()
+
         val context = this.parse(processedCode).translationUnit()
-        val listener = CPPBasicIdentListener(fileName = filePath)
+        val listener = CPPBasicIdentListener(fileName = filePath, includesDirective)
 
         ParseTreeWalker().walk(listener, context)
 

chapi-ast-cpp/src/main/kotlin/chapi/ast/cppast/CPPBasicIdentListener.kt

@@ -4,14 +4,30 @@ import chapi.ast.antlr.CPP14Parser
 import chapi.ast.antlr.CPP14ParserBaseListener
 import chapi.domain.core.*
 
-class CPPBasicIdentListener(fileName: String) : CPP14ParserBaseListener() {
+class CPPBasicIdentListener(fileName: String, includes: MutableList<String>) : CPP14ParserBaseListener() {
     private var codeContainer: CodeContainer = CodeContainer(FullName = fileName)
     /// for example, Friend function, global function (`main`), etc.
     private var defaultNode = CodeDataStruct()
     private var currentFunction: CodeFunction? = null
     private var classes = mutableListOf<CodeDataStruct>()
     private var currentNode: CodeDataStruct? = null
 
+    init {
+        includes.forEach {
+            val value = it
+                .removeSurrounding("\"", "\"")
+                .removeSurrounding("<", ">")
+
+            val imp = CodeImport(
+                Source = value,
+                AsName = value
+            )
+
+            codeContainer.Imports += imp
+        }
+
+    }
+
     override fun enterNamespaceDefinition(ctx: CPP14Parser.NamespaceDefinitionContext?) {
         ctx?.Identifier()?.let {
             codeContainer.PackageName = it.text

chapi-ast-cpp/src/test/kotlin/chapi/ast/cppast/CPPBasicIdentListenerTest.kt

@@ -31,6 +31,10 @@ int main(){}
 """
         val container = CPPAnalyser().analysis(code, "helloworld.cpp")
         assertEquals(container.DataStructures[0].Functions[0].ReturnType, "int")
+
+        // imports
+        assertEquals(container.Imports.size, 1)
+        assertEquals(container.Imports[0].Source, "iostream")
     }
 
     @Test