Merge pull request #3021 from ruby/multi-splat

Multi-splat flag for arguments
ruby · Aug 28, 2024 · 0299f89 · 0299f89
2 parents 163a265 + 21cb9b7
commit 0299f89
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/config.yml b/config.yml
@@ -653,6 +653,8 @@ flags:
         comment: "if arguments contain keyword splat"
       - name: CONTAINS_SPLAT
         comment: "if arguments contain splat"
+      - name: CONTAINS_MULTIPLE_SPLATS
+        comment: "if arguments contain multiple splats"
     comment: Flags for arguments nodes.
   - name: ArrayNodeFlags
     values:

diff --git a/src/prism.c b/src/prism.c
@@ -2065,7 +2065,11 @@ pm_arguments_node_arguments_append(pm_arguments_node_t *node, pm_node_t *argumen
     pm_node_list_append(&node->arguments, argument);
 
     if (PM_NODE_TYPE_P(argument, PM_SPLAT_NODE)) {
-        pm_node_flag_set((pm_node_t *) node, PM_ARGUMENTS_NODE_FLAGS_CONTAINS_SPLAT);
+        if (PM_NODE_FLAG_P(node, PM_ARGUMENTS_NODE_FLAGS_CONTAINS_SPLAT)) {
+            pm_node_flag_set((pm_node_t *) node, PM_ARGUMENTS_NODE_FLAGS_CONTAINS_MULTIPLE_SPLATS);
+        } else {
+            pm_node_flag_set((pm_node_t *) node, PM_ARGUMENTS_NODE_FLAGS_CONTAINS_SPLAT);
+        }
     }
 }
 

diff --git a/test/prism/snapshots/unparser/corpus/literal/send.txt b/test/prism/snapshots/unparser/corpus/literal/send.txt