Merge pull request #265 from lungarella-raffaele/feat-regex-grammar

Add regex support

Author: nojaf

grammar.js

@@ -464,6 +464,7 @@ module.exports = grammar({
         $.extension_expression,
         $.lazy_expression,
         $._jsx_element,
+        $.regex,
       ),
 
     parenthesized_expression: ($) =>
@@ -752,6 +753,7 @@ module.exports = grammar({
         $.number,
         $.true,
         $.false,
+        $.regex,
       ),
 
     variant_pattern: ($) =>
@@ -1185,6 +1187,30 @@ module.exports = grammar({
 
     extension_identifier: ($) => /[a-zA-Z0-9_\.]+/,
 
+    regex: ($) =>
+      seq(
+        "/",
+        field("pattern", $.regex_pattern),
+        token.immediate(prec(1, "/")),
+        optional(field("flags", $.regex_flags)),
+      ),
+
+    regex_pattern: (_) =>
+      token.immediate(
+        prec(
+          -1,
+          repeat1(
+            choice(
+              seq("[", repeat(choice(seq("\\", /./), /[^\]\n\\]/)), "]"),
+              seq("\\", /./),
+              /[^/\\\[\n]/,
+            ),
+          ),
+        ),
+      ),
+
+    regex_flags: (_) => token.immediate(/[a-z]+/),
+
     number: ($) => {
       // OCaml: https://github.com/tree-sitter/tree-sitter-ocaml/blob/f1106bf834703f1f2f795da1a3b5f8f40174ffcc/ocaml/grammar.js#L26
       const hex_literal = seq(

queries/highlights.scm

@@ -73,6 +73,8 @@
 ] @string
 
 
+(regex) @string.special
+
 (character) @character
 (escape_sequence) @string.escape
 

queries/injections.scm

@@ -1,5 +1,9 @@
 ((comment) @injection.content (#set! injection.language "comment"))
 
+; regex literal
+(regex
+  (regex_pattern) @injection.content (#set! injection.language "regex"))
+
 ; %re
 (extension_expression
   (extension_identifier) @_name
@@ -26,4 +30,3 @@
   (#eq? @_name "relay")
   (expression_statement
     (_ (_) @injection.content (#set! injection.language "graphql") )))
-