feat(kotlin): add object class support

Author: phodal

chapi-ast-kotlin/src/main/kotlin/chapi/ast/kotlinast/KotlinBasicIdentListener.kt

@@ -109,6 +109,7 @@ open class KotlinBasicIdentListener(private val fileName: String) : KotlinAstLis
         currentNode = buildClass(ctx)
     }
 
+
     open fun buildClass(ctx: KotlinParser.ClassDeclarationContext): CodeDataStruct {
         val implements = ctx.delegationSpecifiers()?.text?.let(::getTypeFullName)?.let(::listOf) ?: emptyList()
         val annotations = ctx.modifiers().getAnnotations()
@@ -139,6 +140,29 @@ open class KotlinBasicIdentListener(private val fileName: String) : KotlinAstLis
         isEnteredClass.decrementAndGet()
     }
 
+    override fun enterObjectDeclaration(ctx: KotlinParser.ObjectDeclarationContext?) {
+        isEnteredClass.incrementAndGet()
+        currentNode = buildObject(ctx!!)
+    }
+
+    private fun buildObject(ctx: KotlinParser.ObjectDeclarationContext): CodeDataStruct {
+        val annotations = ctx.modifiers().getAnnotations()
+        return CodeDataStruct().apply {
+            NodeName = ctx.simpleIdentifier().Identifier().text
+            Type = DataStructType.OBJECT
+            Package = codeContainer.PackageName
+            FilePath = codeContainer.FullName
+            Annotations = annotations
+            Imports = imports
+            Position = buildPosition(ctx)
+        }
+    }
+
+    override fun exitObjectDeclaration(ctx: KotlinParser.ObjectDeclarationContext?) {
+        classes.add(currentNode)
+        isEnteredClass.decrementAndGet()
+    }
+
     override fun enterPrimaryConstructor(ctx: KotlinParser.PrimaryConstructorContext) {
         val parameters = ctx.classParameters().classParameter().map(::buildProperty)
         val fields = ctx.classParameters().classParameter().mapNotNull(::buildField)

chapi-ast-kotlin/src/test/kotlin/chapi/ast/kotlinast/KotlinFunctionCallTest.kt

@@ -24,6 +24,23 @@ class KotlinFunctionCallTest {
         assert(functionCall.NodeName == "A")
     }
 
+    @Test
+    fun should_handle_for_kotlin_object() {
+        val code = """
+            object A {
+                fun foo() {
+                    println("Hello, world!")
+                }
+            }
+        """.trimIndent()
+
+        val codeContainer = KotlinAnalyser().analysis(code, "Test.kt", ParseMode.Full)
+        val dataStructures = codeContainer.DataStructures.filter { it.NodeName == "A" }
+        val codeFunction = dataStructures[0].Functions[0]
+
+        assert(codeFunction.Name == "foo")
+    }
+
     @Test
     fun should_success_parse_test_usecase() {
         val code = """