Merge pull request #50 from benber86/feat/hexstrings

Add support for hex strings
benber86 · Dec 27, 2024 · a20ac8f · a20ac8f
2 parents 746541e + eccc20f
commit a20ac8f
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/mamushi/parsing/grammar.lark b/src/mamushi/parsing/grammar.lark
@@ -324,7 +324,7 @@ NAME: /[a-zA-Z_]\w*/
 COMMENT.3: /#[^\n\r]*/
 _NEWLINE: ( /\r?\n[\t ]*/ | COMMENT )+
 
-STRING: /b?("(?!"").*?(?<!\\)(\\\\)*?"|'(?!'').*?(?<!\\)(\\\\)*?')/i
+STRING: /b?(x?"(?!"").*?(?<!\\)(\\\\)*?"|x?'(?!'').*?(?<!\\)(\\\\)*?')/i
 DOCSTRING: /(""".*?(?<!\\)(\\\\)*?"""|'''.*?(?<!\\)(\\\\)*?''')/is
 ?strings: STRING+
 

diff --git a/tests/data/declarations/bytes_string_spacing.vy b/tests/data/declarations/bytes_string_spacing.vy
@@ -0,0 +1,23 @@
+@internal
+def a():
+    b: String[10] = '1234567890'
+
+
+
+
+    c: Bytes[100] = b"\x01"
+
+
+    d: Bytes[100] =    x"01afef"
+
+    d: Bytes[100] =    x'01afef'
+# output
+@internal
+def a():
+    b: String[10] = "1234567890"
+
+    c: Bytes[100] = b"\x01"
+
+    d: Bytes[100] = x"01afef"
+
+    d: Bytes[100] = x"01afef"