✨ : add HCL parsing for boolean and number variables

src/main/antlr4/io/codeka/gaia/hcl/antlr/hcl.g4

@@ -17,21 +17,37 @@ variableBlock
   ;
 
 type
-  : 'type' '=' STRING
+  : 'type' '=' TYPE
   ;
 
 description
   : 'description' '=' STRING
   ;
 
 r_default
-  : 'default' '=' STRING
+  : 'default' '=' defaultValue
+  ;
+
+defaultValue
+  : STRING
+  | NUMBER
+  | 'true'
+  | 'false'
   ;
 
 identifier
   : STRING
   ;
 
+TYPE
+  : 'string'
+  | '"string"'
+  | 'number'
+  | '"number"'
+  | 'bool'
+  | '"bool"'
+  ;
+
 /**
  * STRING Lexer Rule comes from the JSON grammar
  * https://github.com/antlr/grammars-v4/blob/master/json/JSON.g4
@@ -53,6 +69,10 @@ fragment SAFECODEPOINT
    : ~ ["\\\u0000-\u001F]
    ;
 
+NUMBER
+   : '0' | [1-9] [0-9]*
+   ;
+
 WS
   : [ \t\r\n]+ -> skip
   ; // skip spaces, tabs, newlines

src/main/java/io/codeka/gaia/hcl/HclTreeListener.kt

@@ -20,14 +20,14 @@ class HclTreeListener : hclBaseListener() {
     }
 
     override fun enterType(ctx: hclParser.TypeContext) {
-        this.current.type = ctx.STRING().text
+        this.current.type = ctx.TYPE().text
     }
 
     override fun enterDescription(ctx: hclParser.DescriptionContext) {
         this.current.description = ctx.STRING().text
     }
 
     override fun enterR_default(ctx: hclParser.R_defaultContext) {
-        this.current.default = ctx.STRING().text
+        this.current.default = ctx.defaultValue().text
     }
 }

src/test/java/io/codeka/gaia/hcl/HCLParserTest.java

@@ -35,14 +35,13 @@ void parsing_variable_shouldWork() throws IOException {
         walker.walk(hclTreeListener, parser.file());
 
         // then
-        assertThat(hclTreeListener.getVariables()).hasSize(1);
+        assertThat(hclTreeListener.getVariables()).hasSize(3);
 
-        Variable foo = new Variable();
-        foo.setName("\"foo\"");
-        foo.setDescription("\"bar\"");
-        foo.setDefault("\"baz\"");
+        Variable stringVar = new Variable("\"string_var\"", "\"string\"", "\"a test string var\"", "\"foo\"");
+        Variable numberVar = new Variable("\"number_var\"", "number", "\"a test number var\"", "42");
+        Variable boolVar = new Variable("\"bool_var\"", "", "", "false");
 
-        assertThat(hclTreeListener.getVariables()).contains(foo);
+        assertThat(hclTreeListener.getVariables()).contains(stringVar, numberVar, boolVar);
     }
 
 }

src/test/resources/hcl/variables.tf

@@ -1,4 +1,15 @@
-variable "foo" {
-    description = "bar"
-    default = "baz"
+variable "string_var" {
+    type = "string"
+    description = "a test string var"
+    default = "foo"
+}
+
+variable "number_var" {
+    type = number
+    description = "a test number var"
+    default = 42
+}
+
+variable "bool_var" {
+    default = false
 }