feat: Add SUB command to CaseAction

Author: elifarley

src/cedarscript_ast_parser/__init__.py

@@ -5,7 +5,7 @@
     CreateCommand, RmFileCommand, MvFileCommand, UpdateCommand,
     SelectCommand, IdentifierFromFile, SingleFileClause, Segment, Marker, BodyOrWhole, MarkerType, RelativeMarker,
     RelativePositionType, MoveClause, DeleteClause, InsertClause, ReplaceClause, EditingAction, Region,
-    WhereClause, RegionClause, EdScript
+    WhereClause, RegionClause, EdScript, CaseStatement, CaseWhen, CaseAction, LoopControl
 )
 
 __all__ = [
@@ -14,6 +14,7 @@
     "CreateCommand", "RmFileCommand", "MvFileCommand", "UpdateCommand",
     "SelectCommand", "IdentifierFromFile", "SingleFileClause", "Segment", "Marker", "BodyOrWhole", "MarkerType",
     "RelativeMarker", "RelativePositionType", "MoveClause", "DeleteClause", "InsertClause", "ReplaceClause",
-    "EditingAction", "Region", "WhereClause", "RegionClause", "EdScript"
+    "EditingAction", "Region", "WhereClause", "RegionClause", "EdScript", "CaseStatement", "CaseWhen", "CaseAction",
+    "LoopControl"
 ]
 

src/cedarscript_ast_parser/cedarscript_ast_parser.py

@@ -255,7 +255,8 @@ class CaseAction:
     """Represents a THEN action in a CASE statement"""
     loop_control: LoopControl | None = None
     remove: bool = False
-    replace: str | None = None
+    sub_pattern: str | None = None  # For SUB command pattern
+    sub_repl: str | None = None    # For SUB command replacement
     indent: int | None = None
     content: Optional[str | tuple[Region, int | None]] = None
 
@@ -716,8 +717,10 @@ def parse_case_action(self, node) -> CaseAction:
         # Parse other action types
         if self.find_first_by_field_name(node, 'remove'):
             action.remove = True
-        elif replace := self.find_first_by_field_name(node, 'replace'):
-            action.replace = self.parse_string(replace)
+        elif pattern := self.find_first_by_field_name(node, 'pattern'):
+            action.sub_pattern = re.compile(self.parse_string(pattern))
+            if repl := self.find_first_by_field_name(node, 'repl'):
+                action.sub_repl = self.parse_string(repl)
         elif indent := self.find_first_by_field_name(node, 'indent'):
             action.indent = int(indent.text)
         else: