allow wallet selection in directly in lucid instantiation

spacebudz · Jan 22, 2025 · 87c6db5 · 87c6db5
1 parent 5e13430
commit 87c6db5
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 5 deletions.
diff --git a/examples/emulate_something.ts b/examples/emulate_something.ts
@@ -12,10 +12,10 @@ const { payment } = Addresses.inspect(address);
 
 const emulator = new Emulator([{ address, assets: { lovelace: 3000000000n } }]);
 
-const lucid = new Lucid({ provider: emulator })
-  .selectWalletFromPrivateKey(
-    privateKey,
-  );
+const lucid = new Lucid({
+  provider: emulator,
+  wallet: { PrivateKey: privateKey },
+});
 
 const mintingPolicy = lucid.newScript({
   type: "All",

diff --git a/src/core/types.ts b/src/core/types.ts
@@ -90,6 +90,19 @@ export interface Wallet {
   submit(tx: string): Promise<string>;
 }
 
+export type WalletSelection =
+  | { PrivateKey: string }
+  | { Api: WalletApi }
+  | {
+    Seed: {
+      seed: string;
+      options?: { type?: "Base" | "Enterprise"; index?: number };
+    };
+  }
+  | {
+    ReadOnly: ReadOnlyWallet;
+  };
+
 // deno-lint-ignore no-explicit-any
 export type Json = any;
 

diff --git a/src/lucid/lucid.ts b/src/lucid/lucid.ts
@@ -28,6 +28,7 @@ import {
   Utxo,
   Wallet,
   WalletApi,
+  WalletSelection,
 } from "../mod.ts";
 import { signMessage, verifyMessage } from "../misc/sign_message.ts";
 import { resolveInstructions } from "./mod.ts";
@@ -52,9 +53,10 @@ export class Lucid {
   };
 
   constructor(
-    { provider, network }: {
+    { provider, network, wallet }: {
       provider?: Provider;
       network?: Network;
+      wallet?: WalletSelection;
     } = {},
   ) {
     if (provider) this.provider = provider;
@@ -113,6 +115,18 @@ export class Lucid {
           (slots - slotConfig.zeroSlot) * slotConfig.slotLength;
       },
     };
+
+    if (wallet) {
+      if ("PrivateKey" in wallet) {
+        this.selectWalletFromPrivateKey(wallet.PrivateKey);
+      } else if ("Seed" in wallet) {
+        this.selectWalletFromSeed(wallet.Seed.seed, wallet.Seed.options);
+      } else if ("Api" in wallet) {
+        this.selectWalletFromApi(wallet.Api);
+      } else if ("ReadOnly" in wallet) {
+        this.selectReadOnlyWallet(wallet.ReadOnly);
+      }
+    }
   }
 
   newScript<T extends unknown[] = Data[]>(