feat(wrapped-keys): add userMaxPrice to the params

Author: Ansonhkg

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

@@ -24,7 +24,7 @@ import { GeneratePrivateKeyParams, GeneratePrivateKeyResult } from '../types';
 export async function generatePrivateKey(
   params: GeneratePrivateKeyParams
 ): Promise<GeneratePrivateKeyResult> {
-  const { pkpSessionSigs, network, litClient, memo } = params;
+  const { pkpSessionSigs, network, litClient, memo, userMaxPrice } = params;
 
   const firstSessionSig = getFirstSessionSig(pkpSessionSigs);
   const pkpAddress = getPkpAddressFromSessionSig(firstSessionSig);
@@ -43,6 +43,7 @@ export async function generatePrivateKey(
       litActionIpfsCid: litActionCode ? undefined : litActionIpfsCid,
       litActionCode: litActionCode ? litActionCode : undefined,
       accessControlConditions: [allowPkpAddressToDecrypt],
+      userMaxPrice,
     });
 
   const { id } = await storePrivateKey({

packages/wrapped-keys/src/lib/lit-actions-client/batch-generate-keys.ts

@@ -8,6 +8,7 @@ interface BatchGeneratePrivateKeysWithLitActionParams
   accessControlConditions: AccessControlConditions;
   litActionIpfsCid?: string;
   litActionCode?: string;
+  userMaxPrice?: bigint;
 }
 
 interface GeneratePrivateKeyLitActionResult {
@@ -33,13 +34,15 @@ export async function batchGenerateKeysWithLitAction(
     pkpSessionSigs,
     litActionIpfsCid,
     litActionCode,
+    userMaxPrice,
   } = args;
 
   const result = await litClient.executeJs({
     useSingleNode: true,
     sessionSigs: pkpSessionSigs,
     ipfsId: litActionIpfsCid,
     code: litActionCode,
+    userMaxPrice,
     jsParams: {
       actions,
       accessControlConditions,

packages/wrapped-keys/src/lib/lit-actions-client/export-private-key.ts

@@ -8,6 +8,7 @@ interface SignMessageWithLitActionParams extends ExportPrivateKeyParams {
   storedKeyMetadata: StoredKeyData;
   litActionIpfsCid?: string;
   litActionCode?: string;
+  userMaxPrice?: bigint;
 }
 
 export async function exportPrivateKeyWithLitAction(
@@ -20,6 +21,7 @@ export async function exportPrivateKeyWithLitAction(
     litActionCode,
     litActionIpfsCid,
     storedKeyMetadata,
+    userMaxPrice,
   } = args;
 
   const {
@@ -32,6 +34,7 @@ export async function exportPrivateKeyWithLitAction(
     sessionSigs: pkpSessionSigs,
     code: litActionCode,
     ipfsId: litActionIpfsCid,
+    userMaxPrice,
     jsParams: {
       pkpAddress,
       ciphertext,

packages/wrapped-keys/src/lib/lit-actions-client/generate-key.ts

@@ -8,6 +8,7 @@ interface GeneratePrivateKeyLitActionParams extends GeneratePrivateKeyParams {
   accessControlConditions: AccessControlConditions;
   litActionIpfsCid?: string;
   litActionCode?: string;
+  userMaxPrice?: bigint;
 }
 
 interface GeneratePrivateKeyLitActionResult {
@@ -23,12 +24,14 @@ export async function generateKeyWithLitAction({
   litActionCode,
   accessControlConditions,
   pkpAddress,
+  userMaxPrice,
 }: GeneratePrivateKeyLitActionParams): Promise<GeneratePrivateKeyLitActionResult> {
   const result = await litClient.executeJs({
     useSingleNode: true,
     sessionSigs: pkpSessionSigs,
     ipfsId: litActionIpfsCid,
     code: litActionCode,
+    userMaxPrice,
     jsParams: {
       pkpAddress,
       accessControlConditions,

packages/wrapped-keys/src/lib/lit-actions-client/sign-message.ts

@@ -9,6 +9,7 @@ interface SignMessageWithLitActionParams
   storedKeyMetadata: StoredKeyData;
   litActionIpfsCid?: string;
   litActionCode?: string;
+  userMaxPrice?: bigint;
 }
 
 export async function signMessageWithLitAction(
@@ -22,13 +23,15 @@ export async function signMessageWithLitAction(
     litActionIpfsCid,
     litActionCode,
     storedKeyMetadata,
+    userMaxPrice,
   } = args;
 
   const { pkpAddress, ciphertext, dataToEncryptHash } = storedKeyMetadata;
   const result = await litClient.executeJs({
     sessionSigs: pkpSessionSigs,
     ipfsId: litActionIpfsCid,
     code: litActionCode,
+    userMaxPrice,
     jsParams: {
       pkpAddress,
       ciphertext,

packages/wrapped-keys/src/lib/lit-actions-client/sign-transaction.ts

@@ -18,6 +18,7 @@ interface SignTransactionWithLitActionParams {
   accessControlConditions: AccessControlConditions;
   broadcast: boolean;
   versionedTransaction?: boolean;
+  userMaxPrice?: bigint;
 }
 
 export async function signTransactionWithLitAction({
@@ -30,11 +31,13 @@ export async function signTransactionWithLitAction({
   storedKeyMetadata: { ciphertext, dataToEncryptHash, pkpAddress },
   unsignedTransaction,
   versionedTransaction,
+  userMaxPrice,
 }: SignTransactionWithLitActionParams): Promise<string> {
   const result = await litClient.executeJs({
     sessionSigs: pkpSessionSigs,
     ipfsId: litActionIpfsCid,
     code: litActionCode,
+    userMaxPrice,
     jsParams: {
       pkpAddress,
       ciphertext,

packages/wrapped-keys/src/lib/types.ts

@@ -22,6 +22,7 @@ export type LitClient = Awaited<ReturnType<typeof createLitClient>> & {
 export interface BaseApiParams {
   pkpSessionSigs: SessionSigsMap;
   litClient: LitClient;
+  userMaxPrice?: bigint;
 }
 
 export interface ApiParamsSupportedNetworks {