refactor(protobuf): extract field construction to function and add ex…

…tendDef spike TODO #31
phodal · Oct 25, 2024 · beadbcd · beadbcd
1 parent f856656
commit beadbcd
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/chapi-ast-protobuf/src/main/kotlin/chapi/ast/protobuf/ProtobufFullIdentListener.kt b/chapi-ast-protobuf/src/main/kotlin/chapi/ast/protobuf/ProtobufFullIdentListener.kt
@@ -26,6 +26,10 @@ class ProtobufFullIdentListener(var fileName: String) : Protobuf3BaseListener()
         codeContainer.DataStructures += enumDs
     }
 
+    override fun enterExtendDef(ctx: Protobuf3Parser.ExtendDefContext?) {
+        /// todo spike for scene
+    }
+
     private fun constructMessageDef(ctx: Protobuf3Parser.MessageDefContext): CodeDataStruct {
         val messageName = ctx.messageName().text
         val codeDataStruct = CodeDataStruct(
@@ -38,11 +42,7 @@ class ProtobufFullIdentListener(var fileName: String) : Protobuf3BaseListener()
         ctx.messageBody().messageElement().map { context ->
             when (val child = context.getChild(0)) {
                 is Protobuf3Parser.FieldContext -> {
-                    codeDataStruct.Fields += CodeField(
-                        TypeType = child.type_().text,
-                        TypeKey = child.fieldName().text,
-                        TypeValue = child.fieldNumber().text
-                    )
+                    codeDataStruct.Fields += constructField(child)
                 }
 
                 is Protobuf3Parser.EnumDefContext -> {
@@ -56,7 +56,7 @@ class ProtobufFullIdentListener(var fileName: String) : Protobuf3BaseListener()
                 }
 
                 is Protobuf3Parser.ExtendDefContext -> {
-
+                    // skip
                 }
 
                 is Protobuf3Parser.OptionStatementContext -> {
@@ -83,6 +83,12 @@ class ProtobufFullIdentListener(var fileName: String) : Protobuf3BaseListener()
         return codeDataStruct
     }
 
+    private fun constructField(child: Protobuf3Parser.FieldContext) = CodeField(
+        TypeType = child.type_().text,
+        TypeKey = child.fieldName().text,
+        TypeValue = child.fieldNumber().text
+    )
+
     private fun constructEnum(child: Protobuf3Parser.EnumDefContext): CodeDataStruct {
         val name = child.enumName().text
         val enumDs = CodeDataStruct(