Add support for constants

grammar.js

@@ -16,7 +16,8 @@ module.exports = grammar({
       $.import,
       $.type_alias,
       $.type_struct,
-      $.type_enum
+      $.type_enum,
+      $.constant,
     ),
 
     // Handles import definitions
@@ -74,6 +75,25 @@ module.exports = grammar({
       seq($.type_identifier, optional(seq("<", repeat_separated_by($.type_argument, ","), ">"))),
     type_argument: ($) => field("type_argument", choice($.identifier, $.type_definition)),
 
+    // Constants
+    constant: ($) => seq("const", $.identifier, optional(seq(":", $.type_definition)), "=", $.constant_value),
+    constant_value: ($) => choice(
+      $.int,
+      $.string,
+      $.bytes,
+      // $.curvepoint - Should this be here?
+    ),
+
+    base10: (_$) => token(/[0-9]+/),
+    base10_underscore: (_$) => token(/[0-9]+(_[0-9]+)+/),
+    base16: (_$) => token(/0x[0-9a-fA-F]+/),
+    int: ($) => choice($.base10, $.base10_underscore, $.base16),
+
+    string: ($) => seq("@", $.string_inner),
+    bytes: ($) => seq(optional("#"), $.string_inner),
+    string_inner: ($) => seq('"', repeat(choice(/[^\\"]/, $.escape)), '"'),
+    escape: ($) => token(/\\./),
+
     //  Comments
     module_comment: (_$) => token(seq("////", /.*/)),
     definition_comment: (_$) => token(seq("///", /.*/)),

test/corpus/constants.txt

@@ -0,0 +1,149 @@
+================================================================================
+Constant Int Base10 w/ Type
+================================================================================
+
+const name: Int = 100
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (type_definition (type_identifier))
+   (constant_value (int (base10)))))
+
+================================================================================
+Constant Int Base10 w/o Type
+================================================================================
+
+const name = 100
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (constant_value (int (base10)))))
+
+================================================================================
+Constant Int Base16 w/ Type
+================================================================================
+
+const name: Int = 0xaa
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (type_definition (type_identifier))
+   (constant_value (int (base16)))))
+
+================================================================================
+Constant Int Base16 w/o Type
+================================================================================
+
+const name = 0xaa
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (constant_value (int (base16)))))
+
+================================================================================
+Constant Int base10_underscore w/ Type
+================================================================================
+
+const name: Int = 1_000_000
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (type_definition (type_identifier))
+   (constant_value (int (base10_underscore)))))
+
+================================================================================
+Constant Int base10_underscore w/o Type
+================================================================================
+
+const name = 1_000_000
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (constant_value (int (base10_underscore)))))
+
+================================================================================
+Constant String w/ Type
+================================================================================
+
+const name: String = @"ilikecardano"
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (type_definition (type_identifier))
+   (constant_value (string (string_inner)))))
+
+================================================================================
+Constant String w/o Type
+================================================================================
+
+const name = @"ilikecardano"
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (constant_value (string (string_inner)))))
+
+================================================================================
+Constant Bytes w/ Type + #
+================================================================================
+
+const name: Bytes = #"deadbeef"
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (type_definition (type_identifier))
+   (constant_value (bytes (string_inner)))))
+
+================================================================================
+Constant Bytes w/o Type + #
+================================================================================
+
+const name = #"deadbeef"
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (constant_value (bytes (string_inner)))))
+
+================================================================================
+Constant Bytes w/ Type
+================================================================================
+
+const name: Bytes = "deadbeef"
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (type_definition (type_identifier))
+   (constant_value (bytes (string_inner)))))
+
+================================================================================
+Constant Bytes w/o Type
+================================================================================
+
+const name = "deadbeef"
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (constant (identifier)
+   (constant_value (bytes (string_inner)))))