fix: ellipsis in script

lang/semgrep-grammars/src/semgrep-move-on-aptos/grammar.js

@@ -19,6 +19,7 @@ module.exports = grammar(base_grammar, {
 
   precedences: $ => [
     [$._sequence_item, $.declaration],
+    [$._script_use_decl, $._script_constant_decl, $._script_func_decl, $._script_spec_block],
   ],
 
   /*
@@ -71,6 +72,13 @@ module.exports = grammar(base_grammar, {
       $.ellipsis,
     ),
 
+    // Script members
+    // We cannot mimic the `declaration` trick here, as the script members are strictly ordered.
+    _script_use_decl: ($, previous) => choice(previous, $.ellipsis),
+    _script_constant_decl: ($, previous) => choice(previous, $.ellipsis),
+    _script_func_decl: ($, previous) => choice(previous, $.ellipsis),
+    _script_spec_block: ($, previous) => choice(previous, $.ellipsis),
+
     // Spec block members
     _spec_block_member: ($, previous) => choice(
       previous,

lang/semgrep-grammars/src/semgrep-move-on-aptos/test/corpus/semgrep.txt

@@ -399,4 +399,47 @@ __SEMGREP_EXPRESSION {
       (ellipsis)
       (var
         (name_access_chain
-          (identifier))))))
+          (identifier))))))
+
+=======================
+Script
+=======================
+
+script {
+  ...
+  use std::vec::{...};
+  ...
+  const aa: u32 = ...;
+  ...
+  spec { ... }
+  ...
+}
+---
+
+(source_file
+  (script
+    (declaration
+      (ellipsis))
+    (declaration
+      (use_decl
+        (module_ident
+          (identifier)
+          (identifier))
+        (member
+          (ellipsis))))
+    (declaration
+      (ellipsis))
+    (declaration
+      (constant_decl
+        (identifier)
+        (type
+          (primitive_type
+            (number_type)))
+        (ellipsis)))
+    (declaration
+      (ellipsis))
+    (declaration
+      (spec_block
+        (ellipsis)))
+    (declaration
+      (ellipsis))))