chore: tidy up

Author: Ansonhkg

packages/wrapped-keys/src/lib/api/tria-batch-generate-private-keys.ts

@@ -21,10 +21,20 @@ import { computeAddress } from 'ethers/lib/utils';
 import { SessionSigsMap } from '@lit-protocol/types';
 
 /**
- *  TODO: Document batch behaviour
- * @param { BatchGeneratePrivateKeysParams } params Parameters to use for generating keys and optionally signing messages
- *
- * @returns { Promise<BatchGeneratePrivateKeysResult> } - The generated keys and, optionally, signed messages
+ * Generates a batch of private keys and optionally signs messages.
+ * 
+ * @param {Object} params - Parameters for generating keys and signing messages.
+ * @param {string} params.pkpPublicKey - The public key for the PKP (e.g., "0x123...").
+ * @param {string} params.ipfsId - The IPFS ID (e.g., "Qm...").
+ * @param {Object} params.authMethod - The authentication method.
+ * @param {string} params.authMethod.authMethodType - The type of authentication method (e.g., "0x...").
+ * @param {string} params.authMethod.accessToken - The access token (e.g., "eyJ...").
+ * @param {Object} params.litNodeClient - The Lit Node client instance.
+ * @param {Array} params.actions - The actions to perform.
+ * 
+ * @returns {Promise<BatchGeneratePrivateKeysResult>} - The generated keys and optionally signed messages.
+ * 
+ * @throws {Error} - Throws an error if required parameters are missing or if the Lit Action Session Sigs cannot be retrieved.
  */
 export async function triaBatchGeneratePrivateKeys(
   params: Omit<BatchGeneratePrivateKeysParams, 'pkpSessionSigs'> & {
@@ -46,11 +56,7 @@ export async function triaBatchGeneratePrivateKeys(
     throw new Error(`Error: ipfsId is required`);
   }
 
-  let pkpPubKey = params.pkpPublicKey;
-
-  if (pkpPubKey.startsWith('0x')) {
-    pkpPubKey = pkpPubKey.slice(2);
-  }
+  let pkpPubKey = params.pkpPublicKey.startsWith('0x') ? params.pkpPublicKey.slice(2) : params.pkpPublicKey;
 
   const pkpPubkeyBuffer = Buffer.from(pkpPubKey, 'hex');
   const pkpEthAddress = computeAddress(pkpPubkeyBuffer);
@@ -89,18 +95,14 @@ export async function triaBatchGeneratePrivateKeys(
       handleAllResponses: true,
       strategy: 'leastCommon',
     });
-  } catch (e) {
-    throw new Error(`Error getting Lit Action Session Sigs: ${e}`);
+  } catch (e: any) {
+    throw new Error(`Error getting Lit Action Session Sigs: ${e.message}`);
   }
 
   const firstSessionSig = Object.entries(litActionSessionSigs)[0][1];
-
   const firstSignedMessage = JSON.parse(firstSessionSig.signedMessage);
-
   const firstCapabilities = firstSignedMessage.capabilities[0];
-
   const theCustomAuthResources = firstCapabilities.customAuthResources;
-
   const pkpAddress = getPkpAddressFromSessionSig(firstSessionSig);
 
   const keyParamsBatch = theCustomAuthResources.map((keyData: any) => {