Add support for SavePos statements

README.md

@@ -38,7 +38,7 @@ The following QuickBMS commands are currently supported by the langauge server.
     * [ ] Support all operators
 * [ ] `XMath VAR INSTR`
 * [ ] `Open FOLDER NAME [FILENUM] [EXISTS]`
-* [ ] `SavePos VAR [FILENUM]`
+* [x] `SavePos VAR [FILENUM]`
 * [x] `Set VAR [TYPE] VAR`
 * [ ] `Do`
 * [ ] `While VAR COND VAR`

src/server/keyword_docs/savepos.txt

@@ -0,0 +1,10 @@
+SavePos VAR [FILENUM]
+
+    Current position of the file, like ftell in C.
+
+    Arguments:
+      VAR       Variable which will contain the offset
+      FILENUM   Number of the file associated to the archive (0)
+
+    Examples:
+      SavePos OFFSET

src/server/state.rs

@@ -484,6 +484,10 @@ pub fn get_keyword_docs() -> HashMap<String, String> {
             "padding".to_string(),
             include_str!("keyword_docs/padding.txt").to_string(),
         ),
+        (
+            "savepos".to_string(),
+            include_str!("keyword_docs/savepos.txt").to_string(),
+        ),
     ]
     .iter()
     .cloned()

tree-sitter-quickbms/grammar.js

@@ -48,6 +48,7 @@ module.exports = grammar({
       $.comtype_statement,
       $.clog_statement,
       $.padding_statement,
+      $.savepos_statement,
     ),
     set_statement: $ => seq(
       $.set,
@@ -275,6 +276,11 @@ module.exports = grammar({
         ),
       )),
     ),
+    savepos_statement: $ => seq(
+      $.savepos,
+      field("variable", $._expression),
+      field("file_number", optional($._expression)),
+    ),
     comparison: $ => choice(
       "<",
       ">",
@@ -401,6 +407,7 @@ module.exports = grammar({
     comtype: $ => /[Cc][Oo][Mm][Tt][Yy][Pp][Ee]/,
     clog: $ => /[Cc][Ll][Oo][Gg]/,
     padding: $ => /[Pp][Aa][Dd][Dd][Ii][Nn][Gg]/,
+    savepos: $ => /[Ss][Aa][Vv][Ee][Pp][Oo][Ss]/,
     question_mark: $ => /\?/,
 
     identifier: $ => /[a-zA-Z_\\\.]+[a-zA-Z0-9_\-\\\.]*/,

tree-sitter-quickbms/test/corpus/savepos.bms

@@ -0,0 +1,12 @@
+================================================================================
+SavePos statements
+================================================================================
+
+savepos OFFSET
+
+--------------------------------------------------------------------------------
+
+(source_file
+  (savepos_statement
+    (savepos)
+    (identifier)))