THEN REPLACE -> THEN SUB <pattern> <replacement>

Author: elifarley

grammar.js

@@ -10,7 +10,7 @@ const SELECT_OTHER_TARGETS = choice(
 
 /*
 CASE
-    WHEN REGEX r"print\((.*?)\)" THEN REPLACE r"logger.debug(\1)"
+    WHEN REGEX r"print\((.*?)\)" THEN SUB r"logger.debug(\1)"
     WHEN REGEX r"raise" THEN DELETE
     WHEN PREFIX "try" THEN RELINDENT 1
     WHEN SUFFIX ";" THEN CONTENT '''xxx'''    ELSE CONTENT '''No match'''
@@ -33,7 +33,7 @@ WHEN MATCHES_FILE "*.py"          -- File pattern match
 WHEN LENGTH > <n>                 -- Line length condition
 
 -- Actions (THEN clause):
-THEN REPLACE r"pattern"           -- Replace with regex capture groups
+THEN SUB r"pattern" r"replacement"           -- Replace with regex capture groups
 THEN DELETE                       -- Remove the line
 THEN CONTENT '''text'''          -- Replace with specific content
 THEN CONTINUE                        -- Skip this line
@@ -403,7 +403,7 @@ module.exports = grammar({
     case_action: $ => choice(
       $.loop_control,
       seq(field('remove', 'REMOVE'), optional($.loop_control)), // Remove line
-      seq('REPLACE', field('replace', $.string), optional($.loop_control)), // Replace with regex capture groups
+      seq('SUB', field('pattern', $.string), field('repl', $.string), optional($.loop_control)), // Replace with regex capture groups
       seq('INDENT', field('indent', $.number), optional($.loop_control)), // Change indentation
       seq(choice($.content_literal, $.content_from_segment), optional($.loop_control)) // Replace with specific content
     ),

test/corpus/update.txt

@@ -991,10 +991,10 @@ REPLACE SEGMENT STARTING AT LINE 1 ENDING BEFORE LINE REGEX "ASDF"
 WITH CASE
   WHEN EMPTY THEN REMOVE
   WHEN LINE NUMBER 4 THEN CONTINUE
-  WHEN REGEX r"asdf" THEN REPLACE r"asdf" BREAK
+  WHEN REGEX r"asdf" THEN SUB r"pattern" r"replacement" BREAK
   WHEN REGEX r"asdf" THEN INDENT 4
   WHEN REGEX r"asdf" THEN BREAK
-  ELSE REPLACE "af"
+  ELSE SUB r"pattern" r'replacement'
 END;
 
 ---
@@ -1031,6 +1031,8 @@ END;
         (string
           (raw_string)))
       (case_action
+        (string
+          (raw_string))
         (string
           (raw_string))
         (loop_control
@@ -1048,7 +1050,9 @@ END;
           (loop_break)))
       (case_action
         (string
-          (single_quoted_string)))))
+          (raw_string))
+        (string
+          (raw_string)))))
   (command_separator))
 
 ==================