#59, parse "primitive" types

openapi-processor · Sep 24, 2023 · 08924d9 · 08924d9
1 parent 6b3bd25
commit 08924d9
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/openapi-processor-core/src/main/antlr/Mapping.g4 b/openapi-processor-core/src/main/antlr/Mapping.g4
@@ -27,13 +27,17 @@ annotate
     ;
 
 anyType
-    : plainType | targetType
+    : plainType | primitiveType | targetType
     ;
 
 plainType
     : Plain
     ;
 
+primitiveType
+    : Primitive
+    ;
+
 sourceType
     : sourceIdentifier (':' formatIdentifier)?
     ;
@@ -99,6 +103,7 @@ Arrow: '=>';
 Annotate: '@';
 
 Plain: 'plain';
+Primitive: 'byte' | 'short' | 'int' | 'long' | 'float' | 'double' | 'boolean' | 'char';
 Boolean: 'true' | 'false';
 Package: '{package-name}';
 

diff --git a/...ain/kotlin/io/openapiprocessor/core/processor/mapping/v2/parser/antlr/MappingExtractor.kt b/...ain/kotlin/io/openapiprocessor/core/processor/mapping/v2/parser/antlr/MappingExtractor.kt
@@ -44,6 +44,10 @@ class MappingExtractor: MappingBaseListener(), Mapping {
         targetType = ctx.text
     }
 
+    override fun enterPrimitiveType(ctx: MappingParser.PrimitiveTypeContext) {
+        targetType = ctx.text
+    }
+
     override fun enterSourceIdentifier(ctx: MappingParser.SourceIdentifierContext) {
         sourceType = ctx.text
     }

diff --git a/...sor-core/src/test/kotlin/io/openapiprocessor/core/processor/mapping/v2/AntlrParserSpec.kt b/...sor-core/src/test/kotlin/io/openapiprocessor/core/processor/mapping/v2/AntlrParserSpec.kt
@@ -259,4 +259,13 @@ class AntlrParserSpec: StringSpec({
         mapping.sourceType shouldBe "array"
         mapping.targetType shouldBe "plain"
     }
+
+    "primitive byte array" {
+        val type = "byte"
+
+        val mapping = parseMapping(type)
+        mapping.kind shouldBe Mapping.Kind.TYPE
+        mapping.sourceType.shouldBeNull()
+        mapping.targetType shouldBe "byte"
+    }
 })