Update custom step function signatures

Author: tothtamas28

src/kontrol/foundry.py

@@ -158,7 +158,13 @@ def _console_log_pattern(self) -> KSequence:
     def _ffi_pattern(self) -> KSequence:
         return KSequence([KApply('ffi_shell', KVariable('###CMD')), KVariable('###CONTINUATION')])
 
-    def _exec_rename_custom_step(self, subst: Subst, cterm: CTerm, _c: CTermSymbolic) -> KCFGExtendResult | None:
+    def _exec_rename_custom_step(
+        self,
+        subst: Subst,
+        cterm: CTerm,
+        _c: CTermSymbolic,
+        _node_id: int,
+    ) -> KCFGExtendResult | None:
         # Extract the target var and new name from the substitution
         target_var = subst['###RENAME_TARGET']
         name_token = subst['###NEW_NAME']
@@ -182,12 +188,17 @@ def _exec_rename_custom_step(self, subst: Subst, cterm: CTerm, _c: CTermSymbolic
         return Step(CTerm(new_cterm.config, constraints), 1, (), ['foundry_rename'], cut=True)
 
     def _exec_forget_custom_step(
-        self, subst: Subst, cterm: CTerm, cterm_symbolic: CTermSymbolic
+        self,
+        subst: Subst,
+        cterm: CTerm,
+        cterm_symbolic: CTermSymbolic,
+        _node_id: int,
     ) -> KCFGExtendResult | None:
         """Remove the constraint at the top of K_CELL of a given CTerm from its path constraints,
            as part of the 'FOUNDRY-ACCOUNTS.forget' cut-rule.
         :param cterm: CTerm representing a proof node
         :param cterm_symbolic: CTermSymbolic instance
+        :param _node_id: Current node id (unused)
         :return: A Step of depth 1 carrying a new configuration in which the constraint is consumed from the top
                  of the K cell and is removed from the initial path constraints if it existed, together with
                  information that the `cheatcode_forget` rule has been applied.
@@ -293,7 +304,13 @@ def _filter_constraints_by_simplification(
         new_cterm = CTerm.from_kast(set_cell(cterm.kast, 'K_CELL', KSequence(subst['###CONTINUATION'])))
         return Step(CTerm(new_cterm.config, new_constraints), 1, (), ['cheatcode_forget'], cut=True)
 
-    def _exec_console_log_custom_step(self, subst: Subst, cterm: CTerm, _c: CTermSymbolic) -> KCFGExtendResult | None:
+    def _exec_console_log_custom_step(
+        self,
+        subst: Subst,
+        cterm: CTerm,
+        _c: CTermSymbolic,
+        _node_id: int,
+    ) -> KCFGExtendResult | None:
         selector_token = subst['###SELECTOR']
         data = subst['###DATA']
         assert type(selector_token) is KToken
@@ -314,7 +331,13 @@ def _exec_console_log_custom_step(self, subst: Subst, cterm: CTerm, _c: CTermSym
         new_cterm = CTerm.from_kast(set_cell(cterm.kast, 'K_CELL', KSequence(subst['###CONTINUATION'])))
         return Step(CTerm(new_cterm.config, cterm.constraints), 1, (), ['console.log'], cut=True)
 
-    def _exec_ffi_custom_step(self, subst: Subst, cterm: CTerm, _c: CTermSymbolic) -> KCFGExtendResult | None:
+    def _exec_ffi_custom_step(
+        self,
+        subst: Subst,
+        cterm: CTerm,
+        _c: CTermSymbolic,
+        _node_id: int,
+    ) -> KCFGExtendResult | None:
         """Execute vm.ffi() cheatcode by running external commands and encoding their output as ABI bytes.
 
         This function decodes the command from the ABI-encoded string array, executes it as a subprocess, and processes
@@ -325,6 +348,7 @@ def _exec_ffi_custom_step(self, subst: Subst, cterm: CTerm, _c: CTermSymbolic) -
         :param subst: Substitution containing the values obtained by matching the `self._ffi_pattern`.
         :param cterm: Current configuration term representing the EVM state.
         :param _c: Symbolic configuration term (unused).
+        :param _node_id: Current node id (unused).
         :return: None if FFI is disabled. Otherwise, Step with OUTPUT_CELL set to ABI-encoded result and updated K_CELL
                 continuation, tagged with 'kontrol.ffi.success'.
         """