feat: update hooks script (#117)

* feat: update hooks script

* feat: new deployment

* chore: new deployment

scripts/configs/pragma_devnet.json

@@ -100,7 +100,7 @@
             "constructor": {
                 "local_domain": {
                     "type": "u32",
-                    "value": "23448591"
+                    "value": "6363709"
                 },
                 "owner": {
                     "type": "ContractAddress",

scripts/deploy.ts

@@ -1,23 +1,20 @@
 import {
   Account,
-  Contract,
   json,
-  Provider,
   CallData,
-  RpcProvider,
   ContractFactory,
   ContractFactoryParams
 } from "starknet";
 import fs from "fs";
 import path from "path";
 import dotenv from "dotenv";
+import { buildAccount } from "./utils";
 
 dotenv.config();
 
 const BUILD_PATH = "../cairo/target/dev/contracts";
 const MOCK_BUILD_PATH = "../cairo/target/dev/mocks";
 const ACCOUNT_ADDRESS = process.env.ACCOUNT_ADDRESS;
-const PRIVATE_KEY = process.env.PRIVATE_KEY;
 const CONFIGS_DIR = 'configs';
 const DEPLOYMENTS_DIR = 'deployments';
 const NETWORK = process.env.NETWORK;
@@ -84,18 +81,7 @@ function findContractFile(name: string, suffix: string): string {
   throw new Error(`Contract file not found for ${name} with suffix ${suffix} in either ${BUILD_PATH} or ${MOCK_BUILD_PATH}`);
 }
 
-async function buildAccount(): Promise<Account> {
-  const provider = new RpcProvider({ nodeUrl: process.env.STARKNET_RPC_URL });
 
-  if (!PRIVATE_KEY || !ACCOUNT_ADDRESS) {
-    throw new Error("Private key or account address not set in .env file");
-  }
-  if (!NETWORK) {
-    throw new Error('NETWORK environment variable is not set');
-  }
-
-  return new Account(provider, ACCOUNT_ADDRESS, PRIVATE_KEY);
-}
 
 function getCompiledContract(name: string): any {
   const contractPath = findContractFile(name, '.contract_class.json');

scripts/deployments.ts

@@ -0,0 +1,54 @@
+import fs from "fs";
+import path from "path";
+
+interface ContractAddresses {
+    [key: string]: string | object;
+}
+
+export function getDeployedAddress(
+    chainName: string,
+    contractName: string,
+): string {
+    try {
+        const filePath = path.join(
+            __dirname,
+            "deployments",
+            chainName,
+            "deployments.json",
+        );
+        const fileContents = fs.readFileSync(filePath, "utf8");
+        const config: ContractAddresses = JSON.parse(fileContents);
+
+        let address: string | undefined;
+
+        // Handle nested structures
+        const findAddress = (obj: any, key: string): string | undefined => {
+            if (obj[key] && typeof obj[key] === "string") {
+                return obj[key] as string;
+            }
+            for (const k in obj) {
+                if (typeof obj[k] === "object") {
+                    const found = findAddress(obj[k], key);
+                    if (found) return found;
+                }
+            }
+            return undefined;
+        };
+
+        address = findAddress(config, contractName);
+
+        if (!address) {
+            throw new Error(
+                `Invalid or missing address for contract ${contractName} in deployments.json for chain ${chainName}`,
+            );
+        }
+
+        return address;
+    } catch (error) {
+        console.error(
+            `Error reading configuration file for chain ${chainName}:`,
+            error,
+        );
+        throw error;
+    }
+}

scripts/deployments/PRAGMA_DEVNET/deployments.json

@@ -1,15 +1,15 @@
 {
-  "merkleroot_multisig_ism": "0xecdcfc130ef39d1165338615a875176873d4ef2100c7bbf52de2cd5a2b4081",
-  "messageid_multisig_ism": "0x54668196b83eeaf476b6623a8c7c8bd4646704cccbf09210f89d47541fe0b92",
-  "domain_routing_ism": "0x14ea7c5eae72dd0f39ff2263ed29d8a01bb48d64e75603f4ded7cc1b6d8dd20",
-  "noop_ism": "0x769f77987315e20eac6bc8bd426fb6242095e73039e49bead77c88c8c406b2c",
-  "pausable_ism": "0x59b29192815b814ccd1397e0868cce6874e2b44f096b30732bc1fca6870c3e6",
-  "aggregation": "0x45caa0f1bab9b1d34f46eaf280fad1970616d90d7acf27b9630596f8f0d6c15",
-  "protocol_fee": "0xffa8a03e994b7b35a52db403c389c987e11a6b34eda453ca58c3da314e6d50",
-  "hook": "0x7809aea51191017f541bb09885ef4c84fe6b11725edbd81c5fd8af6a0943dce",
-  "mailbox": "0x41c20175af14a0bfebfc9ae2f3bda29230a0bceb551844197d9f46faf76d6da",
-  "merkle_tree_hook": "0xb9a75496355e223652c40fe50d45b5f39b86d3cc5c4f7aed44be6c7f6a8b4c",
-  "default_fallback_routing_ism": "0x6b632654d261241234331a29f8b24a7838833035ea5142a26a0c7687a71730f",
-  "trusted_relayer_ism": "0x77559266dd71d2ed75a82608a3400d4d9da71e1b1029013b47f641c431ed5d3",
-  "validator_announce": "0x2305250e0adab1bafabe58bc905cf94440a55e0fc7664ff095250fe2f8684b8"
+  "merkleroot_multisig_ism": "0xb2d90f230c653fe4cd70c33acceab59b5a3a02484c51d2eff413a243d2a86b",
+  "messageid_multisig_ism": "0x53e2587a4c87e1a64a74a081633702db92318aa61ccb81fd9cf14daecc6d492",
+  "domain_routing_ism": "0x62c2d617f3d18fe457722ddcc87234d8bcbaca903022a35be1c7b1555f2f22e",
+  "noop_ism": "0x68f4f40e70ca22c57092a0e5119173ad2770594213ef6ea890967c3695d505d",
+  "pausable_ism": "0x661aad1ff8ef411a41f7adda6cced589d3599037294fe16f80cd7f89f778f66",
+  "aggregation": "0x6746609d234b008e156ce5068163c15b09cd07ace4d6539353fb64e7a87158",
+  "protocol_fee": "0x2bd2961a06f1acf8731c8605dfa3482162499f6bf55262ff35fa7428c547b9b",
+  "hook": "0x5356a792d7d0cfa68981e628d8d4d1923a8a73dd989d30b969d3368854468c8",
+  "mailbox": "0x5bfb1a565a1fa2eb33c5d8e587a7aeb5e6846d3aadf9fecb529ace1e3457096",
+  "merkle_tree_hook": "0x1520c48d7aced426c41e8b71587add7fb64c9945115d3ea677a49f45ddf81e3",
+  "default_fallback_routing_ism": "0x79f951180eaf2c8997bb4749b37c37cb898fb0cba0f398d6a8fe26a6e3ffb4e",
+  "trusted_relayer_ism": "0x3f90f55d0f865527e6eb2c47d6e429d1ac17fbea597bcc36e73e56874cd61c1",
+  "validator_announce": "0x22245997c5f4f5e6eb13764be91de00b4299147ce7f516dbad925c7aeb69d3"
 }

scripts/package.json

@@ -4,7 +4,8 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "deploy": "ts-node deploy.ts"
+    "deploy": "ts-node deploy.ts",
+    "update-hooks": "ts-node update-hooks.ts"
   },
   "keywords": [],
   "author": "",

scripts/update-hooks.ts

@@ -0,0 +1,52 @@
+import { Contract } from "starknet";
+import { buildAccount } from "./utils";
+import { getDeployedAddress } from "./deployments";
+import dotenv from "dotenv";
+
+dotenv.config();
+
+const NETWORK = process.env.NETWORK;
+const DEFAULT_HOOK = undefined;
+const REQUIRED_HOOK = "0x1520c48d7aced426c41e8b71587add7fb64c9945115d3ea677a49f45ddf81e3";
+
+async function updateHooks(): Promise<void> {
+    try {
+        if (!NETWORK) {
+            throw new Error('NETWORK environment variable is not set');
+        }
+
+        const account = await buildAccount();
+
+        const mailboxAddress = getDeployedAddress(NETWORK, 'mailbox');
+        const { abi } = await account.getClassAt(mailboxAddress);
+        const mailboxContract = new Contract(abi, mailboxAddress, account);
+
+        if (DEFAULT_HOOK != undefined) {
+            console.log(`🧩 Updating default hook ${DEFAULT_HOOK}..`);
+            const invoke = await mailboxContract.invoke("set_default_hook", [DEFAULT_HOOK])
+            await account.waitForTransaction(invoke.transaction_hash);
+
+            console.log(`⚡️ Transaction hash: ${invoke.transaction_hash}`);
+        }
+
+        if (REQUIRED_HOOK != undefined) {
+            console.log(`🧩 Updating required hook ${REQUIRED_HOOK}..`);
+            const invoke = await mailboxContract.invoke("set_required_hook", [REQUIRED_HOOK])
+            await account.waitForTransaction(invoke.transaction_hash);
+
+            console.log(`⚡️ Transaction hash: ${invoke.transaction_hash}`);
+        }
+
+        console.log(`🧩 Hooks updated successfully with ${DEFAULT_HOOK} & ${REQUIRED_HOOK}`);
+    } catch (error) {
+        console.error(`⛔ Error updating hooks ${DEFAULT_HOOK} & ${REQUIRED_HOOK}:`, error);
+    }
+}
+
+updateHooks().then(() => {
+    console.log('Hooks updated successfully');
+}
+).catch((error) => {
+    console.error('Error updating hooks:', error);
+}
+);

scripts/utils.ts

@@ -0,0 +1,22 @@
+import {
+    Account,
+    RpcProvider,
+} from "starknet";
+
+import dotenv from "dotenv";
+
+dotenv.config();
+
+
+const ACCOUNT_ADDRESS = process.env.ACCOUNT_ADDRESS;
+const PRIVATE_KEY = process.env.PRIVATE_KEY;
+
+export async function buildAccount(): Promise<Account> {
+    const provider = new RpcProvider({ nodeUrl: process.env.STARKNET_RPC_URL });
+
+    if (!PRIVATE_KEY || !ACCOUNT_ADDRESS) {
+        throw new Error("Private key or account address not set in .env file");
+    }
+
+    return new Account(provider, ACCOUNT_ADDRESS, PRIVATE_KEY);
+}