refactor(goast): improve type resolution in GoFullIdentListener.kt

Remove unnecessary checks and ensure correct type retrieval for struct references. This enhances the accuracy of the type declaration extraction by checking the existence of children before accessing them.
phodal · Nov 12, 2024 · c825273 · c825273
1 parent ec368f8
commit c825273
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/chapi-ast-go/src/main/kotlin/chapi/ast/goast/GoFullIdentListener.kt b/chapi-ast-go/src/main/kotlin/chapi/ast/goast/GoFullIdentListener.kt
@@ -456,13 +456,12 @@ class GoFullIdentListener(var fileName: String) : GoAstListener() {
                 else -> {
                     if (firstChild.text == "&") {
                         val second = it.getChild(1)
-                        /// new struct, like &http.Request{}
                         if (second is ExpressionContext) {
                             val expression = second.getChild(0)
                             if (expression is PrimaryExprContext) {
                                 val typeDecl = expression.children.first().getChild(0)?.getChild(0)?.getChild(0)
-                                if (typeDecl is GoParser.LiteralTypeContext) {
-                                    return typeDecl.getChild(0).text
+                                if (typeDecl is GoParser.LiteralTypeContext && typeDecl.childCount > 0) {
+                                    return typeDecl.getChild(0)!!.text
                                 }
                             }
                         }