feat: add support for passing in your own storage to session key sign…

…er (alchemyplatform#682)
OMGWINNING · May 29, 2024 · 67474c0 · 67474c0
1 parent e275f5f
commit 67474c0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/packages/accounts/src/msca/plugins/session-key/signer.ts b/packages/accounts/src/msca/plugins/session-key/signer.ts
@@ -12,6 +12,7 @@ import { z } from "zod";
 export const SessionKeySignerSchema = z.object({
   storageType: z
     .union([z.literal("local-storage"), z.literal("session-storage")])
+    .or(z.custom<Storage>())
     .default("local-storage"),
   storageKey: z.string().default("session-key-signer:session-key"),
 });
@@ -30,7 +31,7 @@ export class SessionKeySigner
 {
   signerType: string;
   inner: LocalAccountSigner<PrivateKeyAccount>;
-  private storageType: "local-storage" | "session-storage";
+  private storageType: "local-storage" | "session-storage" | Storage;
   private storageKey: string;
 
   constructor(config_: SessionKeySignerConfig = {}) {
@@ -41,7 +42,9 @@ export class SessionKeySigner
 
     const sessionKey = (() => {
       const storage =
-        config.storageType === "session-storage"
+        typeof this.storageType !== "string"
+          ? this.storageType
+          : this.storageType === "session-storage"
           ? sessionStorage
           : localStorage;
       const key = storage.getItem(this.storageKey);