Add onFnCall trigger interfaces

Author: makemake-kbo

src/Assertion.sol

@@ -71,6 +71,26 @@ abstract contract Assertion is Credible, StateChanges {
         triggerRecorder.registerBalanceChangeTrigger(fnSelector);
     }
 
+    /// @notice Registers an onFnCall trigger for a specific adopter selector.
+    /// @param fnSelector The function selector of the assertion function.
+    /// @param triggerSelector The adopter function selector to watch.
+    function registerFnCallTrigger(bytes4 fnSelector, bytes4 triggerSelector) internal view {
+        triggerRecorder.registerFnCallTrigger(fnSelector, triggerSelector);
+    }
+
+    /// @notice Registers a trigger that fires once after the transaction completes.
+    /// @param fnSelector The function selector of the assertion function.
+    function registerTxEndTrigger(bytes4 fnSelector) internal view {
+        triggerRecorder.registerTxEndTrigger(fnSelector);
+    }
+
+    /// @notice Registers a trigger that fires when the given token's balances change.
+    /// @param fnSelector The function selector of the assertion function.
+    /// @param token The ERC20 token address to watch.
+    function registerErc20ChangeTrigger(bytes4 fnSelector, address token) internal view {
+        triggerRecorder.registerErc20ChangeTrigger(fnSelector, token);
+    }
+
     /// @notice Registers the desired assertion spec. Must be called within the constructor.
     /// The assertion spec defines what subset of precompiles are available.
     /// Can only be called once. For an assertion to be valid, it needs a defined spec.

src/PhEvm.sol

@@ -93,6 +93,17 @@ interface PhEvm {
         uint256 callIndex;
     }
 
+    /// @notice Context for an onFnCall-triggered assertion invocation.
+    /// @dev Only valid inside an assertion function triggered by registerFnCallTrigger.
+    struct TriggerContext {
+        /// @notice The adopter selector that triggered this assertion invocation.
+        bytes4 selector;
+        /// @notice Call index for constructing a PreCall fork.
+        uint256 callStart;
+        /// @notice Call index for constructing a PostCall fork.
+        uint256 callEnd;
+    }
+
     /// @notice Fork to the state before the assertion-triggering transaction
     /// @dev Allows inspection of pre-transaction state for comparison
     function forkPreTx() external;
@@ -259,6 +270,10 @@ interface PhEvm {
     /// @return txObject The transaction data struct
     function getTxObject() external view returns (TxObject memory txObject);
 
+    /// @notice Returns the current onFnCall trigger context.
+    /// @dev Reverts when called outside an onFnCall-triggered assertion execution.
+    function context() external view returns (TriggerContext memory);
+
     /// @notice Returns canonical Solidity key encodings h(key) for keys
     ///         whose mapping entry at baseSlot was written during the tx.
     /// @dev Best-effort heuristic: traces KECCAK256 -> SSTORE provenance in the

src/TriggerRecorder.sol

@@ -31,4 +31,18 @@ interface TriggerRecorder {
     /// if the assertion adopter is called.
     /// @param fnSelector The function selector of the assertion function.
     function registerCallTrigger(bytes4 fnSelector) external view;
+
+    /// @notice Registers a trigger that fires once per matching adopter call.
+    /// @param fnSelector The function selector of the assertion function.
+    /// @param triggerSelector The adopter function selector to watch.
+    function registerFnCallTrigger(bytes4 fnSelector, bytes4 triggerSelector) external view;
+
+    /// @notice Registers a trigger that fires once after the transaction completes.
+    /// @param fnSelector The function selector of the assertion function.
+    function registerTxEndTrigger(bytes4 fnSelector) external view;
+
+    /// @notice Registers a trigger that fires when the given token's balances change.
+    /// @param fnSelector The function selector of the assertion function.
+    /// @param token The ERC20 token address to watch.
+    function registerErc20ChangeTrigger(bytes4 fnSelector, address token) external view;
 }