fix(naga-local): add assertion and TS Typeguard

Author: Ansonhkg

packages/e2e/src/init.ts

@@ -4,6 +4,7 @@ import {
   ViemAccountAuthenticator,
 } from '@lit-protocol/auth';
 import { createLitClient, utils as litUtils } from '@lit-protocol/lit-client';
+import type { NagaLocalModule } from '@lit-protocol/networks';
 import { generatePrivateKey, privateKeyToAccount } from 'viem/accounts';
 import { z } from 'zod';
 import { fundAccount } from './helper/fundAccount';
@@ -130,17 +131,24 @@ export const init = async (
   if (_network === 'naga-local') {
     const localContextPath = process.env['NAGA_LOCAL_CONTEXT_PATH'];
     if (localContextPath) {
-      const withLocalContext = (_baseNetworkModule as any)?.withLocalContext;
-
-      if (typeof withLocalContext === 'function') {
+      const isNagaLocalModule = (
+        module: unknown
+      ): module is NagaLocalModule =>
+        !!module &&
+        typeof (module as { withLocalContext?: unknown }).withLocalContext ===
+          'function';
+
+      if (isNagaLocalModule(_baseNetworkModule)) {
         const localContextName = process.env['NAGA_LOCAL_CONTEXT_NAME'];
 
         console.log(
           '✅ Loading naga-local signatures from NAGA_LOCAL_CONTEXT_PATH:',
           localContextPath
         );
 
-        _baseNetworkModule = await withLocalContext({
+        const nagaLocalModule: NagaLocalModule = _baseNetworkModule;
+
+        _baseNetworkModule = await nagaLocalModule.withLocalContext({
           networkContextPath: localContextPath,
           networkName: localContextName,
         });

packages/networks/src/networks/vNaga/envs/naga-local/naga-local.module.ts

@@ -6,6 +6,7 @@ import {
   nagaLocalEnvironment,
   type NagaLocalSignatures,
 } from './naga-local.env';
+import { signatures as defaultSignatures } from './generated/naga-develop';
 import type { ExpectedAccountOrWalletClient } from '../../shared/managers/contract-manager/createContractsManager';
 
 type NagaLocalContextOptions = {
@@ -14,6 +15,32 @@ type NagaLocalContextOptions = {
   rpcUrlOverride?: string;
 };
 
+const REQUIRED_SIGNATURE_KEYS = Object.keys(
+  defaultSignatures
+) as (keyof NagaLocalSignatures)[];
+
+function assertIsNagaLocalSignatures(
+  value: unknown
+): asserts value is NagaLocalSignatures {
+  if (typeof value !== 'object' || value === null) {
+    throw new Error('Generated signatures is not an object');
+  }
+
+  for (const key of REQUIRED_SIGNATURE_KEYS) {
+    const contract = (value as Record<string, unknown>)[key];
+    if (
+      typeof contract !== 'object' ||
+      contract === null ||
+      typeof (contract as { address?: unknown }).address !== 'string' ||
+      typeof (contract as { methods?: unknown }).methods !== 'object'
+    ) {
+      throw new Error(
+        `Generated signatures missing required contract metadata for ${key as string}`
+      );
+    }
+  }
+}
+
 const createChainManager = (
   env: NagaLocalEnvironment,
   account: ExpectedAccountOrWalletClient
@@ -48,12 +75,12 @@ const buildModule = (env: NagaLocalEnvironment) => {
       networkName: networkName ?? 'naga-develop',
     });
 
-    const resolvedSignatures = signatures as unknown as NagaLocalSignatures;
+    assertIsNagaLocalSignatures(signatures);
 
     return createWithEnv(
       new NagaLocalEnvironment({
         rpcUrlOverride: rpcUrlOverride ?? env.getConfig().rpcUrl,
-        signatures: resolvedSignatures,
+        signatures,
       })
     );
   };