Merge pull request #4 from Brahma-fi/ft-automation-scripts

automation scripts

Author: Pradeep-selva

kernel-workflow/package.json

@@ -6,7 +6,9 @@
   "scripts": {
     "deploy-account": "ts-node src/deploy-automation-account.ts",
     "register-executor": "ts-node src/register-executor.ts",
-    "agent-workflow": "ts-node src/agent-workflow.ts"
+    "agent-workflow": "ts-node src/agent-workflow.ts",
+    "update-automation": "ts-node src/update-automation.ts",
+    "cancel-automation": "ts-node src/cancel-automation.ts"
   },
   "keywords": [],
   "author": "",

kernel-workflow/src/cancel-automation.ts

@@ -0,0 +1,134 @@
+import Safe from "@safe-global/protocol-kit";
+import {
+  Address,
+  ConsoleKit,
+  UpdateAutomationParams,
+  VendorCancelAutomationParams
+} from "brahma-console-kit";
+import { ethers, Wallet } from "ethers";
+import { encodeMulti } from "ethers-multisend";
+import { encodePacked } from "viem";
+import { SafeABI } from "./entity/abi";
+
+const OwnerEoaPK = process.env.USER_EOA_PRIVATE_KEY!; /// This user must have enough tokens to execute automation update
+const JsonRpcUrl = process.env.JSON_RPC_URL!;
+const ConsoleApiKey = process.env.CONSOLE_API_KEY!;
+const ConsoleBaseUrl = process.env.CONSOLE_BASE_URL!;
+
+const SUBACCOUNT_ADDRESS: Address = "0x"; // your automation subaccount address
+const CONSOLE_ADDRESS: Address = "0x"; // your owner console address
+
+const AutomationCancelData = {
+  execViaSubAcc: [] /**
+        array of arbitrary txns to execute via subAccount of type-
+        {
+          "to": "target Address",
+          "value": "0",
+          "data": "0xcallData"
+        }[]
+    */,
+  sweepTokens: [] // array of tokens to transfer out of subAccount
+};
+
+const cancelAutomation = async (
+  _consoleKit: ConsoleKit,
+  _rpcUrl: string,
+  _automationCancelParams: VendorCancelAutomationParams,
+  _ownerEoa: Wallet,
+  _consoleAddress: Address
+) => {
+  try {
+    const {
+      data: { transactions }
+    } = await _consoleKit.automationContext.cancelAutomation(
+      _automationCancelParams
+    );
+    console.log("[cancel-txns]", { transactions });
+
+    const safe = await Safe.init({
+      provider: _rpcUrl,
+      safeAddress: _consoleAddress
+    });
+    const cancelTx = await safe.createTransaction({
+      transactions: transactions,
+      onlyCalls: false
+    }); // safe sdk can also alternatively be used to create & propose tx: https://docs.safe.global/reference-sdk-protocol-kit/transactions/createtransaction
+    console.log("[safe-cancel-tx]", { cancelTx });
+
+    const {
+      baseGas,
+      data,
+      gasPrice,
+      gasToken,
+      operation,
+      refundReceiver,
+      safeTxGas,
+      to,
+      value
+    } = cancelTx.data;
+    const signature = encodePacked(
+      ["bytes12", "address", "bytes32", "bytes1"],
+      [
+        "0x000000000000000000000000",
+        _ownerEoa.address as Address,
+        "0x0000000000000000000000000000000000000000000000000000000000000000",
+        "0x01"
+      ]
+    );
+
+    const safeContract = new ethers.Contract(
+      _consoleAddress,
+      SafeABI,
+      _ownerEoa
+    );
+    const tx = await _ownerEoa.sendTransaction({
+      to: await safeContract.getAddress(),
+      value: 0,
+      data: safeContract.interface.encodeFunctionData("execTransaction", [
+        to,
+        value,
+        data,
+        operation,
+        safeTxGas,
+        baseGas,
+        gasPrice,
+        gasToken,
+        refundReceiver,
+        signature
+      ])
+    });
+    await tx.wait(2);
+
+    console.log("[automation-cancelled]", tx.hash);
+  } catch (e) {
+    console.log("an error occured", e);
+  }
+};
+
+(async () => {
+  const consoleKit = new ConsoleKit(ConsoleApiKey, ConsoleBaseUrl);
+
+  const provider = new ethers.JsonRpcProvider(JsonRpcUrl);
+  const ownerWallet = new ethers.Wallet(OwnerEoaPK, provider);
+
+  const { chainId: chainIdBig } = await provider.getNetwork();
+  const chainId = parseInt(chainIdBig.toString(), 10);
+
+  const automationCancelData: VendorCancelAutomationParams = {
+    chainId,
+    data: {
+      execViaSubAcc: AutomationCancelData.execViaSubAcc,
+      sweepTokens: AutomationCancelData.sweepTokens,
+      ownerConsole: CONSOLE_ADDRESS
+    },
+    subAccountAddress: SUBACCOUNT_ADDRESS
+  };
+
+  await cancelAutomation(
+    consoleKit,
+    JsonRpcUrl,
+    automationCancelData,
+    ownerWallet,
+    CONSOLE_ADDRESS
+  );
+})();

kernel-workflow/src/entity/abi/index.ts

@@ -0,0 +1 @@
+export * from "./safe";