Add support for basename

ExcaliburZero · Nov 5, 2023 · 882ca7b · 882ca7b
1 parent 7e1f6f9
commit 882ca7b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/tree-sitter-quickbms/grammar.js b/tree-sitter-quickbms/grammar.js
@@ -286,6 +286,7 @@ module.exports = grammar({
       $.binary,
       $.unicode,
       $.byte,
+      $.basename,
     ),
     _endian_type: $ => choice(
       $.little,
@@ -319,6 +320,7 @@ module.exports = grammar({
     binary: $ => /[Bb][Ii][Nn][Aa][Rr][Yy]/,
     unicode: $ => /[Uu][Nn][Ii][Cc][Oo][Dd][Ee]/,
     byte: $ => /[Bb][Yy][Tt][Ee]/,
+    basename: $ => /[Bb][Aa][Ss][Ee][Nn][Aa][Mm][Ee]/,
     endian: $ => /[Ee][Nn][Dd][Ii][Aa][Nn]/,
     little: $ => /[Ll][Ii][Tt][Tt][Ll][Ee]/,
     intel: $ => /[Ii][Nn][Tt][Ee][Ll]/,

diff --git a/tree-sitter-quickbms/test/corpus/get.bms b/tree-sitter-quickbms/test/corpus/get.bms
@@ -6,6 +6,9 @@ Get NAME string
 get NAME string 0
 get SIZE asize
 
+get SIZE asize MEMORY_FILE
+get NAME basename
+
 --------------------------------------------------------------------------------
 
 (source_file
@@ -29,4 +32,15 @@ get SIZE asize
     (get)
     (identifier)
     (type
-      (asize))))
+      (asize)))
+  (get_statement
+    (get)
+    (identifier)
+    (type
+      (asize))
+    (identifier))
+  (get_statement
+    (get)
+    (identifier)
+    (type
+      (basename))))