@opendatalabs/vana-sdk 0.1.0-alpha.7ee7635 → 0.1.0-alpha.80df35f

package/dist/controllers/permissions.cjs

@@ -41,9 +41,10 @@ var import_grantValidation = require("../utils/grantValidation");
 var import_signatureCache = require("../utils/signatureCache");
 var import_signatureFormatter = require("../utils/signatureFormatter");
 var import_typedDataConverter = require("../utils/typedDataConverter");
-class PermissionsController {
+var import_base = require("./base");
+class PermissionsController extends import_base.BaseController {
   constructor(context) {
-    this.context = context;
+    super(context);
   }
   /**
    * Grants permission for an application to access user data with gasless transactions.
@@ -80,6 +81,7 @@ class PermissionsController {
    * ```
    */
   async grant(params) {
+    this.assertWallet();
     const { typedData, signature } = await this.createAndSign(params);
     const result = await this.submitSignedGrantWithEvents(typedData, signature);
     return result;
@@ -110,6 +112,7 @@ class PermissionsController {
    * ```
    */
   async submitPermissionGrant(params) {
+    this.assertWallet();
     const { typedData, signature } = await this.createAndSign(params);
     return await this.submitSignedGrant(typedData, signature);
   }
@@ -139,6 +142,7 @@ class PermissionsController {
    * ```
    */
   async prepareGrant(params) {
+    this.assertWallet();
     try {
       const grantFile = (0, import_grantFiles.createGrantFile)(params);
       (0, import_grantValidation.validateGrant)(grantFile);
@@ -179,7 +183,7 @@ class PermissionsController {
    */
   async confirmGrantInternal(params, grantFile) {
     try {
-      let grantUrl = params.grantUrl;
+      let { grantUrl } = params;
       console.debug("\u{1F50D} Debug - Grant URL from params:", grantUrl);
       if (!grantUrl) {
         if (!this.context.relayerCallbacks?.storeGrantFile && !this.context.storageManager) {
@@ -269,10 +273,11 @@ class PermissionsController {
    * ```
    */
   async createAndSign(params) {
+    this.assertWallet();
     try {
       const grantFile = (0, import_grantFiles.createGrantFile)(params);
       (0, import_grantValidation.validateGrant)(grantFile);
-      let grantUrl = params.grantUrl;
+      let { grantUrl } = params;
       console.debug("\u{1F50D} Debug - Grant URL from params:", grantUrl);
       if (!grantUrl) {
         if (!this.context.relayerCallbacks?.storeGrantFile && !this.context.storageManager) {
@@ -360,7 +365,7 @@ class PermissionsController {
         "\u{1F50D} Debug - submitSignedGrant called with typed data:",
         JSON.stringify(
           typedData,
-          (key, value) => typeof value === "bigint" ? value.toString() : value,
+          (_key, value) => typeof value === "bigint" ? value.toString() : value,
           2
         )
       );
@@ -369,7 +374,7 @@ class PermissionsController {
           typedData,
           signature
         );
-        const account = this.context.walletClient.account || await this.getUserAddress();
+        const account = this.context.walletClient?.account ?? this.context.userAddress;
         const { tx } = await import("../utils/transactionHelpers");
         return tx({
           hash,
@@ -421,11 +426,11 @@ class PermissionsController {
         trustServerInput,
         signature
       );
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
-        from: typeof account === "string" ? account : account.address,
+        from: account,
         contract: "DataPortabilityServers",
         fn: "trustServerWithSignature"
       });
@@ -486,7 +491,7 @@ class PermissionsController {
         addAndTrustServerInput,
         signature
       );
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -506,7 +511,7 @@ class PermissionsController {
   }
   /**
    * Internal method to submit a signed grant and wait for events.
-   * 
+   *
    * @internal
    * @param typedData - The EIP-712 typed data for the permission grant
    * @param signature - The user's signature authorizing the transaction
@@ -520,7 +525,9 @@ class PermissionsController {
     const result = await this.context.waitForTransactionEvents(txResult);
     const event = result.expectedEvents.PermissionAdded;
     if (!event) {
-      throw new import_errors.BlockchainError("PermissionAdded event not found in transaction");
+      throw new import_errors.BlockchainError(
+        "PermissionAdded event not found in transaction"
+      );
     }
     const receipt = await this.context.publicClient.getTransactionReceipt({
       hash: result.hash
@@ -537,7 +544,7 @@ class PermissionsController {
   }
   /**
    * Internal method for confirm grant with events.
-   * 
+   *
    * @internal
    * @param params - The permission grant parameters
    * @param grantFile - The pre-created grant file object
@@ -551,7 +558,9 @@ class PermissionsController {
     const result = await this.context.waitForTransactionEvents(txResult);
     const event = result.expectedEvents.PermissionAdded;
     if (!event) {
-      throw new import_errors.BlockchainError("PermissionAdded event not found in transaction");
+      throw new import_errors.BlockchainError(
+        "PermissionAdded event not found in transaction"
+      );
     }
     const receipt = await this.context.publicClient.getTransactionReceipt({
       hash: result.hash
@@ -566,34 +575,6 @@ class PermissionsController {
       fileIds: event.fileIds
     };
   }
-  /**
-   * Internal method to submit a signed revoke and wait for events.
-   * 
-   * @internal
-   * @param typedData - The EIP-712 typed data for the permission revoke
-   * @param signature - The user's signature authorizing the transaction  
-   * @returns Promise resolving to PermissionRevokeResult with parsed events
-   */
-  async submitSignedRevokeWithEvents(typedData, signature) {
-    const txResult = await this.submitSignedRevoke(typedData, signature);
-    if (!this.context.waitForTransactionEvents) {
-      throw new import_errors.BlockchainError("waitForTransactionEvents not configured");
-    }
-    const result = await this.context.waitForTransactionEvents(txResult);
-    const event = result.expectedEvents.PermissionRevoked;
-    if (!event) {
-      throw new import_errors.BlockchainError("PermissionRevoked event not found in transaction");
-    }
-    const receipt = await this.context.publicClient.getTransactionReceipt({
-      hash: result.hash
-    });
-    return {
-      transactionHash: result.hash,
-      blockNumber: receipt.blockNumber,
-      gasUsed: receipt.gasUsed,
-      permissionId: event.permissionId
-    };
-  }
   /**
    * Submits an already-signed permission revoke transaction to the blockchain.
    *
@@ -624,9 +605,12 @@ class PermissionsController {
           signature
         );
       } else {
-        hash = await this.submitDirectRevokeTransaction(typedData, signature);
+        hash = await this.submitDirectRevokeTransaction(
+          typedData,
+          signature
+        );
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -674,9 +658,12 @@ class PermissionsController {
           signature
         );
       } else {
-        hash = await this.submitSignedUntrustTransaction(typedData, signature);
+        hash = await this.submitSignedUntrustTransaction(
+          typedData,
+          signature
+        );
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -707,7 +694,8 @@ class PermissionsController {
    * @throws {BlockchainError} When contract submission fails
    */
   async submitDirectTransaction(typedData, signature) {
-    const chainId = await this.context.walletClient.getChainId();
+    this.assertWallet();
+    const chainId = await this.context.publicClient.getChainId();
     const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
       chainId,
       "DataPortabilityPermissions"
@@ -727,17 +715,17 @@ class PermissionsController {
     console.debug("\u{1F50D} Debug - Grant field value:", typedData.message.grant);
     console.debug(
       "\u{1F50D} Debug - Grant field length:",
-      typedData.message.grant?.length || 0
+      typedData.message.grant?.length ?? 0
     );
     const formattedSignature = (0, import_signatureFormatter.formatSignatureForContract)(signature);
-    const account = this.context.walletClient.account || await this.getUserAddress();
+    const account = this.context.walletClient?.account ?? this.context.userAddress;
     const txHash = await this.context.walletClient.writeContract({
       address: DataPortabilityPermissionsAddress,
       abi: DataPortabilityPermissionsAbi,
       functionName: "addPermission",
       args: [permissionInput, formattedSignature],
       account,
-      chain: this.context.walletClient.chain || null
+      chain: this.context.walletClient?.chain ?? null
     });
     const { tx } = await import("../utils/transactionHelpers");
     return tx({
@@ -772,6 +760,7 @@ class PermissionsController {
    * ```
    */
   async revoke(params) {
+    this.assertWallet();
     const txResult = await this.submitPermissionRevoke(params);
     if (!this.context.waitForTransactionEvents) {
       throw new import_errors.BlockchainError("waitForTransactionEvents not configured");
@@ -779,7 +768,9 @@ class PermissionsController {
     const result = await this.context.waitForTransactionEvents(txResult);
     const event = result.expectedEvents.PermissionRevoked;
     if (!event) {
-      throw new import_errors.BlockchainError("PermissionRevoked event not found in transaction");
+      throw new import_errors.BlockchainError(
+        "PermissionRevoked event not found in transaction"
+      );
     }
     const receipt = await this.context.publicClient.getTransactionReceipt({
       hash: result.hash
@@ -811,11 +802,12 @@ class PermissionsController {
    * ```
    */
   async submitPermissionRevoke(params) {
+    this.assertWallet();
     try {
-      if (!this.context.walletClient.chain?.id) {
+      if (!this.context.walletClient?.chain?.id) {
         throw new import_errors.BlockchainError("Chain ID not available");
       }
-      const chainId = await this.context.walletClient.getChainId();
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityPermissions"
@@ -823,14 +815,14 @@ class PermissionsController {
       const DataPortabilityPermissionsAbi = (0, import_abi.getAbi)(
         "DataPortabilityPermissions"
       );
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const txHash = await this.context.walletClient.writeContract({
         address: DataPortabilityPermissionsAddress,
         abi: DataPortabilityPermissionsAbi,
         functionName: "revokePermission",
         args: [params.permissionId],
         account,
-        chain: this.context.walletClient.chain || null
+        chain: this.context.walletClient?.chain ?? null
       });
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
@@ -878,8 +870,9 @@ class PermissionsController {
    * ```
    */
   async submitRevokeWithSignature(params) {
+    this.assertWallet();
     try {
-      if (!this.context.walletClient.chain?.id) {
+      if (!this.context.walletClient?.chain?.id) {
         throw new import_errors.BlockchainError("Chain ID not available");
       }
       const nonce = await this.getPermissionsUserNonce();
@@ -906,9 +899,12 @@ class PermissionsController {
           signature
         );
       } else {
-        hash = await this.submitDirectRevokeTransaction(typedData, signature);
+        hash = await this.submitDirectRevokeTransaction(
+          typedData,
+          signature
+        );
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -923,64 +919,6 @@ class PermissionsController {
       );
     }
   }
-  /**
-   * @deprecated Use getPermissionsUserNonce() for permission operations or getServersUserNonce() for server operations
-   *
-   * Retrieves the user's current nonce from the DataPortabilityServers contract.
-   * This method is deprecated in favor of more specific nonce methods.
-   *
-   * The nonce is used to prevent replay attacks in signed transactions and must
-   * be incremented with each transaction to maintain security.
-   *
-   * @returns Promise resolving to the user's current nonce value as a bigint
-   * @throws {Error} When wallet account is not available
-   * @throws {Error} When chain ID is not available
-   * @throws {NonceError} When reading nonce from contract fails
-   * @private
-   * @example
-   * ```typescript
-   * // Deprecated - use specific methods instead
-   * const nonce = await this.getUserNonce();
-   *
-   * // Use these instead:
-   * const permissionsNonce = await this.getPermissionsUserNonce();
-   * const serversNonce = await this.getServersUserNonce();
-   * ```
-   */
-  /**
-   * @deprecated Use getPermissionsUserNonce() for permission operations or getServersUserNonce() for server operations
-   *
-   * Retrieves the user's current nonce from the DataPortabilityServers contract.
-   *
-   * @remarks
-   * This method is deprecated in favor of more specific nonce methods that target
-   * the appropriate contract for the operation being performed.
-   *
-   * @returns Promise resolving to the user's current nonce as a bigint
-   * @throws {NonceError} When retrieving the nonce fails
-   */
-  async getUserNonce() {
-    try {
-      const userAddress = await this.getUserAddress();
-      const chainId = await this.context.walletClient.getChainId();
-      const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
-        chainId,
-        "DataPortabilityServers"
-      );
-      const DataPortabilityServersAbi = (0, import_abi.getAbi)("DataPortabilityServers");
-      const nonce = await this.context.publicClient.readContract({
-        address: DataPortabilityServersAddress,
-        abi: DataPortabilityServersAbi,
-        functionName: "userNonce",
-        args: [userAddress]
-      });
-      return nonce;
-    } catch (error) {
-      throw new import_errors.NonceError(
-        `Failed to retrieve user nonce: ${error instanceof Error ? error.message : "Unknown error"}`
-      );
-    }
-  }
   /**
    * Retrieves the user's current nonce from the DataPortabilityServers contract.
    * This nonce is used for server-related operations (AddAndTrustServer, TrustServer, UntrustServer).
@@ -1007,8 +945,8 @@ class PermissionsController {
    */
   async getServersUserNonce() {
     try {
-      const userAddress = await this.getUserAddress();
-      const chainId = await this.context.walletClient.getChainId();
+      const userAddress = this.context.userAddress;
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityServers"
@@ -1053,8 +991,8 @@ class PermissionsController {
    */
   async getPermissionsUserNonce() {
     try {
-      const userAddress = await this.getUserAddress();
-      const chainId = await this.context.walletClient.getChainId();
+      const userAddress = this.context.userAddress;
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityPermissions"
@@ -1093,7 +1031,7 @@ class PermissionsController {
       "\u{1F50D} Debug - Composing permission message with grantUrl:",
       params.grantUrl
     );
-    const chainId = await this.context.walletClient.getChainId();
+    const chainId = await this.context.publicClient.getChainId();
     const DataPortabilityGranteesAddress = (0, import_addresses.getContractAddress)(
       chainId,
       "DataPortabilityGrantees"
@@ -1202,7 +1140,7 @@ class PermissionsController {
    * @returns Promise resolving to the EIP-712 domain configuration
    */
   async getPermissionDomain() {
-    const chainId = await this.context.walletClient.getChainId();
+    const chainId = await this.context.publicClient.getChainId();
     const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
       chainId,
       "DataPortabilityPermissions"
@@ -1221,8 +1159,9 @@ class PermissionsController {
    * @returns Promise resolving to the cryptographic signature
    */
   async signTypedData(typedData) {
+    this.assertWallet();
     try {
-      const walletAddress = this.context.walletClient.account?.address || await this.getUserAddress();
+      const walletAddress = this.context.walletClient.account?.address ?? this.context.userAddress;
       return await (0, import_signatureCache.withSignatureCache)(
         this.context.platform.cache,
         walletAddress,
@@ -1231,9 +1170,9 @@ class PermissionsController {
           const viemCompatibleTypedData = (0, import_typedDataConverter.toViemTypedDataDefinition)(typedData);
           return await this.context.walletClient.signTypedData({
             ...viemCompatibleTypedData,
-            // Non-null assertion is safe here because getUserAddress() above ensures account exists
-            // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-            account: this.context.walletClient.account
+            // Use the account if available, otherwise use the wallet address
+            // This follows the same pattern used throughout this file
+            account: this.context.walletClient.account ?? walletAddress
           });
         }
       );
@@ -1248,17 +1187,10 @@ class PermissionsController {
     }
   }
   /**
-   * Gets the user's address from the wallet client.
+   * Gets the user's address from the context.
    *
-   * @returns Promise resolving to the user's wallet address
+   * @returns The user's address
    */
-  async getUserAddress() {
-    const addresses = await this.context.walletClient.getAddresses();
-    if (addresses.length === 0) {
-      throw new import_errors.BlockchainError("No addresses available in wallet client");
-    }
-    return addresses[0];
-  }
   /**
    * Gets on-chain permission grant data without expensive off-chain resolution.
    *
@@ -1297,8 +1229,8 @@ class PermissionsController {
   async getUserPermissionGrantsOnChain(options = {}) {
     const { limit = 50, subgraphUrl } = options;
     try {
-      const userAddress = await this.getUserAddress();
-      const graphqlEndpoint = subgraphUrl || this.context.subgraphUrl;
+      const userAddress = this.context.userAddress;
+      const graphqlEndpoint = subgraphUrl ?? this.context.subgraphUrl;
       if (!graphqlEndpoint) {
         throw new import_errors.BlockchainError(
           "subgraphUrl is required. Please provide a valid subgraph endpoint or configure it in Vana constructor."
@@ -1350,7 +1282,7 @@ class PermissionsController {
         );
       }
       const userData = result.data?.user;
-      if (!userData || !userData.permissions?.length) {
+      if (!userData?.permissions?.length) {
         return [];
       }
       const onChainGrants = userData.permissions.slice(0, limit).map(
@@ -1361,8 +1293,8 @@ class PermissionsController {
           nonce: BigInt(permission.nonce),
           startBlock: BigInt(permission.startBlock),
           addedAtBlock: BigInt(permission.addedAtBlock),
-          addedAtTimestamp: BigInt(permission.addedAtTimestamp || "0"),
-          transactionHash: permission.transactionHash || "",
+          addedAtTimestamp: BigInt(permission.addedAtTimestamp ?? "0"),
+          transactionHash: permission.transactionHash ?? "",
           grantor: userAddress,
           grantee: permission.grantee,
           active: !permission.endBlock || BigInt(permission.endBlock) === 0n
@@ -1383,26 +1315,6 @@ class PermissionsController {
       );
     }
   }
-  /**
-   * Normalizes grant ID to hex format.
-   * Handles conversion from permission ID (bigint/number/string) to proper hex hash format.
-   *
-   * @param grantId - Permission ID or grant hash in various formats
-   * @returns Normalized hex hash
-   */
-  normalizeGrantId(grantId) {
-    if (typeof grantId === "string" && grantId.startsWith("0x") && grantId.length === 66) {
-      return grantId;
-    }
-    try {
-      const bigIntId = BigInt(grantId);
-      return `0x${bigIntId.toString(16).padStart(64, "0")}`;
-    } catch {
-      throw new Error(
-        `Invalid grant ID format: ${grantId}. Must be a permission ID (number/bigint/string) or a 32-byte hex hash.`
-      );
-    }
-  }
   /**
    * Registers a new server and immediately trusts it in the DataPortabilityServers contract.
    *
@@ -1411,7 +1323,6 @@ class PermissionsController {
    * Trusted servers can handle data export and portability requests from the user.
    *
    * @param params - Parameters for adding and trusting the server
-   * @param params.owner - Ethereum address that will own this server registration
    * @param params.serverAddress - Ethereum address of the server
    * @param params.serverUrl - HTTPS URL where the server can be reached
    * @param params.publicKey - Server's public key for encryption (hex string)
@@ -1438,6 +1349,7 @@ class PermissionsController {
    * ```
    */
   async addAndTrustServer(params) {
+    this.assertWallet();
     try {
       const chainId = await this.context.walletClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
@@ -1445,7 +1357,7 @@ class PermissionsController {
         "DataPortabilityServers"
       );
       const DataPortabilityServersAbi = (0, import_abi.getAbi)("DataPortabilityServers");
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const userAddress = typeof account === "string" ? account : account.address;
       const normalizedUserAddress = (0, import_viem.getAddress)(userAddress);
       const normalizedServerAddress = (0, import_viem.getAddress)(params.serverAddress);
@@ -1462,7 +1374,7 @@ class PermissionsController {
           }
         ],
         account,
-        chain: this.context.walletClient.chain || null
+        chain: this.context.walletClient?.chain ?? null
       });
       const { tx } = await import("../utils/transactionHelpers");
       const txResult = tx({
@@ -1477,9 +1389,13 @@ class PermissionsController {
       const result = await this.context.waitForTransactionEvents(txResult);
       const event = result.expectedEvents.ServerTrusted;
       if (!event) {
-        throw new import_errors.BlockchainError("ServerTrusted event not found in transaction");
+        throw new import_errors.BlockchainError(
+          "ServerTrusted event not found in transaction"
+        );
       }
-      const receipt = await this.context.publicClient.getTransactionReceipt({ hash: txHash });
+      const receipt = await this.context.publicClient.getTransactionReceipt({
+        hash: txHash
+      });
       return {
         transactionHash: txHash,
         blockNumber: receipt.blockNumber,
@@ -1510,6 +1426,7 @@ class PermissionsController {
    * @deprecated Use addAndTrustServer instead
    */
   async submitTrustServer(params) {
+    this.assertWallet();
     try {
       const chainId = await this.context.walletClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
@@ -1517,14 +1434,14 @@ class PermissionsController {
         "DataPortabilityServers"
       );
       const DataPortabilityServersAbi = (0, import_abi.getAbi)("DataPortabilityServers");
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const txHash = await this.context.walletClient.writeContract({
         address: DataPortabilityServersAddress,
         abi: DataPortabilityServersAbi,
         functionName: "trustServer",
         args: [BigInt(params.serverId)],
         account,
-        chain: this.context.walletClient.chain || null
+        chain: this.context.walletClient?.chain ?? null
       });
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
@@ -1550,6 +1467,7 @@ class PermissionsController {
    * @returns Promise resolving to TransactionResult with ServerTrustResult event data
    */
   async submitAddAndTrustServerWithSignature(params) {
+    this.assertWallet();
     try {
       const nonce = await this.getServersUserNonce();
       const serverAddress = (0, import_viem.getAddress)(params.serverAddress);
@@ -1584,7 +1502,7 @@ class PermissionsController {
           signature
         );
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -1621,6 +1539,7 @@ class PermissionsController {
    * @throws {BlockchainError} When trust operation fails for any other reason
    */
   async submitTrustServerWithSignature(params) {
+    this.assertWallet();
     try {
       const nonce = await this.getServersUserNonce();
       const trustServerInput = {
@@ -1641,7 +1560,7 @@ class PermissionsController {
           signature
         );
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -1680,6 +1599,7 @@ class PermissionsController {
    * @throws {BlockchainError} When contract submission fails
    */
   async submitDirectUntrustTransaction(params) {
+    this.assertWallet();
     try {
       const chainId = await this.context.walletClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
@@ -1687,14 +1607,14 @@ class PermissionsController {
         "DataPortabilityServers"
       );
       const DataPortabilityServersAbi = (0, import_abi.getAbi)("DataPortabilityServers");
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const txHash = await this.context.walletClient.writeContract({
         address: DataPortabilityServersAddress,
         abi: DataPortabilityServersAbi,
         functionName: "untrustServer",
         args: [BigInt(params.serverId)],
         account,
-        chain: this.context.walletClient.chain || null
+        chain: this.context.walletClient?.chain ?? null
       });
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
@@ -1742,6 +1662,7 @@ class PermissionsController {
    * ```
    */
   async submitUntrustServer(params) {
+    this.assertWallet();
     const nonce = await this.getServersUserNonce();
     const untrustServerInput = {
       nonce,
@@ -1762,6 +1683,7 @@ class PermissionsController {
    * @throws {BlockchainError} When untrust transaction fails
    */
   async submitUntrustServerWithSignature(params) {
+    this.assertWallet();
     try {
       const nonce = await this.getServersUserNonce();
       const untrustServerInput = {
@@ -1779,7 +1701,7 @@ class PermissionsController {
       } else {
         hash = await this.submitSignedUntrustTransaction(typedData, signature);
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -1824,8 +1746,8 @@ class PermissionsController {
    */
   async getTrustedServers(userAddress) {
     try {
-      const user = userAddress || await this.getUserAddress();
-      const chainId = await this.context.walletClient.getChainId();
+      const user = userAddress ?? this.context.userAddress;
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityServers"
@@ -1854,8 +1776,8 @@ class PermissionsController {
    */
   async getTrustedServersCount(userAddress) {
     try {
-      const user = userAddress || await this.getUserAddress();
-      const chainId = await this.context.walletClient.getChainId();
+      const user = userAddress ?? this.context.userAddress;
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityServers"
@@ -1884,10 +1806,10 @@ class PermissionsController {
    */
   async getTrustedServersPaginated(options = {}) {
     try {
-      const user = options.userAddress || await this.getUserAddress();
-      const limit = options.limit || 50;
-      const offset = options.offset || 0;
-      const chainId = await this.context.walletClient.getChainId();
+      const user = options.userAddress ?? this.context.userAddress;
+      const limit = options.limit ?? 50;
+      const offset = options.offset ?? 0;
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityServers"
@@ -2033,7 +1955,7 @@ class PermissionsController {
       };
     }
     try {
-      const chainId = await this.context.walletClient.getChainId();
+      const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
         "DataPortabilityServers"
@@ -2108,7 +2030,7 @@ class PermissionsController {
    */
   async checkServerTrustStatus(serverId, userAddress) {
     try {
-      const user = userAddress || await this.getUserAddress();
+      const user = userAddress ?? this.context.userAddress;
       const trustedServers = await this.getTrustedServers(user);
       const trustIndex = trustedServers.findIndex(
         (server) => server === serverId
@@ -2198,7 +2120,7 @@ class PermissionsController {
    * @returns Promise resolving to the EIP-712 domain configuration
    */
   async getServersDomain() {
-    const chainId = await this.context.walletClient.getChainId();
+    const chainId = await this.context.publicClient.getChainId();
     const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
       chainId,
       "DataPortabilityServers"
@@ -2218,6 +2140,7 @@ class PermissionsController {
    * @returns Promise resolving to the transaction hash
    */
   async submitAddAndTrustServerTransaction(addAndTrustServerInput, signature) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -2249,8 +2172,8 @@ class PermissionsController {
         },
         formattedSignature
       ],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      account: this.context.walletClient?.account ?? this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return txHash;
   }
@@ -2262,6 +2185,7 @@ class PermissionsController {
    * @returns Promise resolving to the transaction hash
    */
   async submitTrustServerTransaction(trustServerInput, signature) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -2280,8 +2204,8 @@ class PermissionsController {
         },
         formattedSignature
       ],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      account: this.context.walletClient?.account ?? this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return txHash;
   }
@@ -2293,6 +2217,7 @@ class PermissionsController {
    * @returns Promise resolving to the transaction hash
    */
   async submitDirectRevokeTransaction(typedData, signature) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -2304,10 +2229,9 @@ class PermissionsController {
       address: DataPortabilityPermissionsAddress,
       abi: DataPortabilityPermissionsAbi,
       functionName: "revokePermissionWithSignature",
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
       args: [typedData.message, formattedSignature],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      account: this.context.walletClient?.account ?? this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return txHash;
   }
@@ -2319,6 +2243,7 @@ class PermissionsController {
    * @returns Promise resolving to the transaction hash
    */
   async submitSignedUntrustTransaction(typedData, signature) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -2326,14 +2251,17 @@ class PermissionsController {
     );
     const DataPortabilityServersAbi = (0, import_abi.getAbi)("DataPortabilityServers");
     const formattedSignature = (0, import_signatureFormatter.formatSignatureForContract)(signature);
+    const contractMessage = {
+      nonce: typedData.message.nonce,
+      serverId: BigInt(typedData.message.serverId)
+    };
     const txHash = await this.context.walletClient.writeContract({
       address: DataPortabilityServersAddress,
       abi: DataPortabilityServersAbi,
       functionName: "untrustServerWithSignature",
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      args: [typedData.message, formattedSignature],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      args: [contractMessage, formattedSignature],
+      account: this.context.walletClient?.account ?? this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return txHash;
   }
@@ -2366,6 +2294,7 @@ class PermissionsController {
    * ```
    */
   async submitRegisterGrantee(params) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityGranteesAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -2374,14 +2303,14 @@ class PermissionsController {
     const DataPortabilityGranteesAbi = (0, import_abi.getAbi)("DataPortabilityGrantees");
     const ownerAddress = (0, import_viem.getAddress)(params.owner);
     const granteeAddress = (0, import_viem.getAddress)(params.granteeAddress);
-    const account = this.context.walletClient.account || await this.getUserAddress();
+    const account = this.context.walletClient?.account ?? this.context.userAddress;
     const txHash = await this.context.walletClient.writeContract({
       address: DataPortabilityGranteesAddress,
       abi: DataPortabilityGranteesAbi,
       functionName: "registerGrantee",
       args: [ownerAddress, granteeAddress, params.publicKey],
       account,
-      chain: this.context.walletClient.chain || null
+      chain: this.context.walletClient?.chain ?? null
     });
     const { tx } = await import("../utils/transactionHelpers");
     return tx({
@@ -2407,6 +2336,7 @@ class PermissionsController {
    * ```
    */
   async submitRegisterGranteeWithSignature(params) {
+    this.assertWallet();
     const nonce = await this.getServersUserNonce();
     const owner = (0, import_viem.getAddress)(params.owner);
     const granteeAddress = (0, import_viem.getAddress)(params.granteeAddress);
@@ -2422,7 +2352,7 @@ class PermissionsController {
       typedData,
       signature
     );
-    const account = this.context.walletClient.account || await this.getUserAddress();
+    const account = this.context.walletClient?.account ?? this.context.userAddress;
     const { tx } = await import("../utils/transactionHelpers");
     return tx({
       hash,
@@ -2448,7 +2378,7 @@ class PermissionsController {
       typedData,
       signature
     );
-    const account = this.context.walletClient.account || await this.getUserAddress();
+    const account = this.context.walletClient?.account ?? this.context.userAddress;
     const { tx } = await import("../utils/transactionHelpers");
     return tx({
       hash,
@@ -2502,8 +2432,8 @@ class PermissionsController {
       functionName: "granteesCount"
     });
     const total = Number(totalCount);
-    const limit = options.limit || 50;
-    const offset = options.offset || 0;
+    const limit = options.limit ?? 50;
+    const offset = options.offset ?? 0;
     const grantees = [];
     const endIndex = Math.min(offset + limit, total);
     for (let i = offset; i < endIndex; i++) {
@@ -2651,7 +2581,7 @@ class PermissionsController {
    * @private
    */
   async buildRegisterGranteeTypedData(input) {
-    const chainId = await this.context.walletClient.getChainId();
+    const chainId = await this.context.publicClient.getChainId();
     const verifyingContract = (0, import_addresses.getContractAddress)(
       chainId,
       "DataPortabilityGrantees"
@@ -2684,6 +2614,7 @@ class PermissionsController {
    * @private
    */
   async submitSignedRegisterGranteeTransaction(typedData, _signature) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityGranteesAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -2699,8 +2630,8 @@ class PermissionsController {
         typedData.message.granteeAddress,
         typedData.message.publicKey
       ],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      account: this.context.walletClient?.account ?? this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return txHash;
   }
@@ -2715,7 +2646,7 @@ class PermissionsController {
    */
   async getUserServerIds(userAddress) {
     try {
-      const targetAddress = userAddress || await this.getUserAddress();
+      const targetAddress = userAddress ?? this.context.userAddress;
       const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
@@ -2773,7 +2704,7 @@ class PermissionsController {
    */
   async getUserServerCount(userAddress) {
     try {
-      const targetAddress = userAddress || await this.getUserAddress();
+      const targetAddress = userAddress ?? this.context.userAddress;
       const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
@@ -2802,7 +2733,7 @@ class PermissionsController {
    */
   async getUserTrustedServers(userAddress) {
     try {
-      const targetAddress = userAddress || await this.getUserAddress();
+      const targetAddress = userAddress ?? this.context.userAddress;
       const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
         chainId,
@@ -2919,7 +2850,7 @@ class PermissionsController {
    */
   async getUserPermissionIds(userAddress) {
     try {
-      const targetAddress = userAddress || await this.getUserAddress();
+      const targetAddress = userAddress ?? this.context.userAddress;
       const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
         chainId,
@@ -2981,7 +2912,7 @@ class PermissionsController {
    */
   async getUserPermissionCount(userAddress) {
     try {
-      const targetAddress = userAddress || await this.getUserAddress();
+      const targetAddress = userAddress ?? this.context.userAddress;
       const chainId = await this.context.publicClient.getChainId();
       const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
         chainId,
@@ -3426,6 +3357,7 @@ class PermissionsController {
    * @returns Promise resolving to transaction hash
    */
   async submitUpdateServer(serverId, url) {
+    this.assertWallet();
     try {
       const chainId = await this.context.walletClient.getChainId();
       const DataPortabilityServersAddress = (0, import_addresses.getContractAddress)(
@@ -3433,13 +3365,13 @@ class PermissionsController {
         "DataPortabilityServers"
       );
       const DataPortabilityServersAbi = (0, import_abi.getAbi)("DataPortabilityServers");
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const hash = await this.context.walletClient.writeContract({
         address: DataPortabilityServersAddress,
         abi: DataPortabilityServersAbi,
         functionName: "updateServer",
         args: [serverId, url],
-        chain: this.context.walletClient.chain,
+        chain: this.context.walletClient?.chain,
         account
       });
       const { tx } = await import("../utils/transactionHelpers");
@@ -3598,6 +3530,7 @@ class PermissionsController {
    * @throws {NetworkError} When network communication fails
    */
   async submitAddPermission(params) {
+    this.assertWallet();
     try {
       const nonce = await this.getPermissionsUserNonce();
       const addPermissionInput = {
@@ -3636,6 +3569,7 @@ class PermissionsController {
    * @throws {NetworkError} When network communication fails
    */
   async submitSignedAddPermission(typedData, signature) {
+    this.assertWallet();
     try {
       let hash;
       if (this.context.relayerCallbacks?.submitAddPermission) {
@@ -3649,7 +3583,7 @@ class PermissionsController {
           signature
         );
       }
-      const account = this.context.walletClient.account || await this.getUserAddress();
+      const account = this.context.walletClient?.account ?? this.context.userAddress;
       const { tx } = await import("../utils/transactionHelpers");
       return tx({
         hash,
@@ -3685,7 +3619,7 @@ class PermissionsController {
    * @param params.serverAddress - Server wallet address for decryption permissions
    * @param params.serverUrl - Server endpoint URL
    * @param params.serverPublicKey - Server's public key for encryption.
-   *   Obtain via `vana.server.getIdentity(userAddress).public_key`.
+   *   Obtain via `vana.server.getIdentity(userAddress).publicKey`.
    * @param params.filePermissions - Nested array of permissions for each file
    * @returns TransactionResult with immediate hash access and optional event data
    * @throws {Error} When schemaIds array length doesn't match fileUrls array length
@@ -3707,7 +3641,7 @@ class PermissionsController {
    *   schemaIds: [123], // LinkedIn profile schema ID
    *   serverAddress: "0x742d35Cc6634C0532925a3b844Bc9e7595f0b0Bb",
    *   serverUrl: "https://server.example.com",
-   *   serverPublicKey: serverInfo.public_key,
+   *   serverPublicKey: serverInfo.publicKey,
    *   filePermissions: [[{
    *     account: "0x742d35Cc6634C0532925a3b844Bc9e7595f0b0Bb",
    *     key: encryptedKey
@@ -3718,6 +3652,7 @@ class PermissionsController {
    * ```
    */
   async submitAddServerFilesAndPermissions(params) {
+    this.assertWallet();
     try {
       if (params.schemaIds.length !== params.fileUrls.length) {
         throw new Error(
@@ -3782,6 +3717,7 @@ class PermissionsController {
    * ```
    */
   async submitSignedAddServerFilesAndPermissions(typedData, signature) {
+    this.assertWallet();
     try {
       console.debug("\u{1F50D} submitSignedAddServerFilesAndPermissions Debug Info:", {
         hasRelayerCallbacks: !!this.context.relayerCallbacks,
@@ -3796,7 +3732,7 @@ class PermissionsController {
           typedData,
           signature
         );
-        const account = this.context.walletClient.account || await this.getUserAddress();
+        const account = this.context.walletClient?.account ?? this.context.userAddress;
         const { tx } = await import("../utils/transactionHelpers");
         return tx({
           hash,
@@ -3812,11 +3748,11 @@ class PermissionsController {
           typedData,
           signature
         );
-        const account = this.context.walletClient.account || await this.getUserAddress();
+        const account = this.context.userAddress;
         const { tx } = await import("../utils/transactionHelpers");
         return tx({
           hash,
-          from: typeof account === "string" ? account : account.address,
+          from: account,
           contract: "DataPortabilityPermissions",
           fn: "addServerFilesAndPermissions"
         });
@@ -3838,6 +3774,7 @@ class PermissionsController {
    * @returns Promise resolving to transaction hash
    */
   async submitRevokePermission(permissionId) {
+    this.assertWallet();
     try {
       const chainId = await this.context.walletClient.getChainId();
       const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
@@ -3847,7 +3784,7 @@ class PermissionsController {
       const DataPortabilityPermissionsAbi = (0, import_abi.getAbi)(
         "DataPortabilityPermissions"
       );
-      const account = this.context.walletClient.account;
+      const { account } = this.context.walletClient;
       if (!account) {
         throw new Error("No wallet account connected");
       }
@@ -3856,7 +3793,7 @@ class PermissionsController {
         abi: DataPortabilityPermissionsAbi,
         functionName: "revokePermission",
         args: [permissionId],
-        chain: this.context.walletClient.chain,
+        chain: this.context.walletClient?.chain,
         account
       });
       const { tx } = await import("../utils/transactionHelpers");
@@ -3881,6 +3818,7 @@ class PermissionsController {
    * @returns Promise resolving to the transaction hash
    */
   async submitDirectAddPermissionTransaction(typedData, signature) {
+    this.assertWallet();
     const chainId = await this.context.walletClient.getChainId();
     const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
       chainId,
@@ -3891,7 +3829,7 @@ class PermissionsController {
       nonce: typedData.message.nonce,
       granteeId: typedData.message.granteeId,
       grant: typedData.message.grant,
-      fileIds: typedData.message.fileIds || []
+      fileIds: typedData.message.fileIds ?? []
     };
     const formattedSignature = (0, import_signatureFormatter.formatSignatureForContract)(signature);
     const hash = await this.context.walletClient.writeContract({
@@ -3899,8 +3837,8 @@ class PermissionsController {
       abi: DataPortabilityPermissionsAbi,
       functionName: "addPermission",
       args: [permissionInput, formattedSignature],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      account: this.context.walletClient?.account ?? this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return hash;
   }
@@ -3912,7 +3850,8 @@ class PermissionsController {
    * @returns Promise resolving to the transaction hash
    */
   async submitDirectAddServerFilesAndPermissionsTransaction(typedData, signature) {
-    const chainId = await this.context.walletClient.getChainId();
+    this.assertWallet();
+    const chainId = await this.context.publicClient.getChainId();
     const DataPortabilityPermissionsAddress = (0, import_addresses.getContractAddress)(
       chainId,
       "DataPortabilityPermissions"
@@ -3936,8 +3875,8 @@ class PermissionsController {
       functionName: "addServerFilesAndPermissions",
       // @ts-expect-error - Viem's type inference for nested Permission[][] arrays is incompatible with our Permission type
       args: [serverFilesAndPermissionInput, formattedSignature],
-      account: this.context.walletClient.account || await this.getUserAddress(),
-      chain: this.context.walletClient.chain || null
+      account: this.context.userAddress,
+      chain: this.context.walletClient?.chain ?? null
     });
     return hash;
   }