@zerodev/sdk 4.0.9 → 4.0.10

package/src/kernel-zerodev/validator-provider/session-key-provider.ts

@@ -0,0 +1,151 @@
+import {
+  ValidatorProvider,
+  type ExtendedValidatorProviderParams,
+  type ValidatorProviderParams,
+} from "./base.js";
+import {
+  SessionKeyValidator,
+  type SessionKeyParams,
+  type SessionKeyValidatorParams,
+} from "../validator/session-key-validator.js";
+import {
+  getChain,
+  LocalAccountSigner,
+  type Hex,
+  type SmartAccountSigner,
+} from "@alchemy/aa-core";
+import { getChainId } from "../api/index.js";
+import { polygonMumbai } from "viem/chains";
+import { SESSION_KEY_VALIDATOR_ADDRESS } from "../constants.js";
+import { base64ToBytes, bytesToBase64 } from "../utils.js";
+import type { RequiredProps, WithRequired } from "../types.js";
+import type { KernelBaseValidatorParams } from "../validator/base.js";
+
+export type PrefillSessionData = {
+  sessionKeyParams: SessionKeyParams &
+    WithRequired<SessionKeyParams, "sessionPrivateKey">;
+};
+
+export type SessionKeyProviderParams = PrefillSessionData &
+  ValidatorProviderParams<SessionKeyValidatorParams> &
+  Partial<SessionKeyValidatorParams> &
+  RequiredProps<KernelBaseValidatorParams>;
+
+export class SessionKeyProvider extends ValidatorProvider<
+  SessionKeyValidator,
+  SessionKeyValidatorParams
+> {
+  constructor(
+    params: ExtendedValidatorProviderParams<SessionKeyValidatorParams>
+  ) {
+    const chain =
+      typeof params.opts?.providerConfig?.chain === "number"
+        ? getChain(params.opts.providerConfig.chain)
+        : params.opts?.providerConfig?.chain ?? polygonMumbai;
+    const validator = new SessionKeyValidator({
+      projectId: params.projectId,
+      sessionKey: params.sessionKey,
+      sessionKeyData: params.sessionKeyData,
+      chain,
+      validatorAddress:
+        params.opts?.validatorConfig?.validatorAddress ??
+        SESSION_KEY_VALIDATOR_ADDRESS,
+      ...params.opts?.validatorConfig,
+    });
+    super(
+      {
+        ...params,
+        opts: {
+          ...params.opts,
+          providerConfig: { ...params.opts?.providerConfig, chain },
+        },
+      },
+      validator
+    );
+  }
+
+  public static async init(
+    params: ExtendedValidatorProviderParams<SessionKeyValidatorParams>
+  ): Promise<SessionKeyProvider> {
+    const chainId = await getChainId(params.projectId);
+    if (!chainId) {
+      throw new Error("ChainId not found");
+    }
+    const chain = getChain(chainId);
+    const instance = new SessionKeyProvider({
+      ...params,
+      opts: {
+        ...params.opts,
+        providerConfig: {
+          chain,
+          ...params.opts?.providerConfig,
+        },
+      },
+    });
+    return instance;
+  }
+
+  public static async fromSessionKeyParams(
+    params: SessionKeyProviderParams,
+    signer?: SmartAccountSigner
+  ) {
+    let sessionKey = signer;
+
+    if (!sessionKey) {
+      sessionKey = LocalAccountSigner.privateKeyToAccountSigner(
+        params.sessionKeyParams.sessionPrivateKey
+      );
+    }
+    return await SessionKeyProvider.init({
+      ...params,
+      sessionKey,
+      sessionKeyData: params.sessionKeyParams.sessionKeyData,
+      opts: {
+        ...params.opts,
+        accountConfig: {
+          ...params.opts?.accountConfig,
+          accountAddress: params.sessionKeyParams.accountAddress,
+          initCode: params.sessionKeyParams.initCode,
+        },
+        validatorConfig: {
+          ...params.opts?.validatorConfig,
+          executor: params.sessionKeyParams.executor,
+          selector: params.sessionKeyParams.selector,
+          enableSignature: params.sessionKeyParams.enableSignature,
+        },
+      },
+    });
+  }
+
+  async serializeSessionKeyParams(sessionPrivateKey?: Hex): Promise<string> {
+    await this.getAccount().approvePlugin();
+    let sessionKeyParams = this.getValidator().getSessionData();
+    const initCode = await this.getAccount().getInitCode();
+    const accountAddress = await this.getAddress();
+    if (!initCode) {
+      throw Error("initCode not set");
+    }
+    sessionKeyParams = {
+      ...sessionKeyParams,
+      sessionPrivateKey,
+      initCode,
+      accountAddress,
+    };
+    const jsonString = JSON.stringify(sessionKeyParams);
+    const uint8Array = new TextEncoder().encode(jsonString);
+    const base64String = bytesToBase64(uint8Array);
+    return base64String;
+  }
+
+  public static deserializeSessionKeyParams(
+    sessionKeyParams: string
+  ): PrefillSessionData["sessionKeyParams"] {
+    const uint8Array = base64ToBytes(sessionKeyParams);
+    const jsonString = new TextDecoder().decode(uint8Array);
+    return JSON.parse(jsonString) as PrefillSessionData["sessionKeyParams"];
+  }
+
+  changeSessionKeyData = this.sendEnableUserOperation;
+
+  deleteSessionKey = this.sendDisableUserOperation;
+}

package/src/kernel-zerodev/validator-provider/types.ts

@@ -1,22 +1,30 @@
 import type { ECDSAValidatorParams } from "../validator/ecdsa-validator.js";
 import type { ERC165SessionKeyValidatorParams } from "../validator/erc165-session-key-validator.js";
 import type { KillSwitchValidatorParams } from "../validator/kill-switch-validator.js";
+import type { SessionKeyValidatorParams } from "../validator/session-key-validator.js";
 import type { SupportedValidators } from "../validator/types.js";
 import type { ExtendedValidatorProviderParams } from "./base.js";
 import type { ECDSAProvider } from "./ecdsa-provider.js";
 import type { ERC165SessionKeyProvider } from "./erc165-session-key-provider.js";
 import type { KillSwitchProvider } from "./kill-switch-provider.js";
+import type {
+  SessionKeyProvider,
+  PrefillSessionData,
+  SessionKeyProviderParams,
+} from "./session-key-provider.js";
 
 export type ValidatorProviderTypeMap = {
   ECDSA: ECDSAProvider;
   KILL_SWITCH: KillSwitchProvider;
   ERC165_SESSION_KEY: ERC165SessionKeyProvider;
+  SESSION_KEY: SessionKeyProvider;
 };
 
 export type ValidatorProviderParamsMap = {
   ECDSA: ExtendedValidatorProviderParams<ECDSAValidatorParams>;
   KILL_SWITCH: ExtendedValidatorProviderParams<KillSwitchValidatorParams>;
   ERC165_SESSION_KEY: ExtendedValidatorProviderParams<ERC165SessionKeyValidatorParams>;
+  SESSION_KEY: ExtendedValidatorProviderParams<SessionKeyValidatorParams>;
 };
 
 export type ValidatorProviderMap = {
@@ -24,3 +32,5 @@ export type ValidatorProviderMap = {
     params: ValidatorProviderParamsMap[V]
   ) => ValidatorProviderTypeMap[V];
 };
+
+export type { PrefillSessionData, SessionKeyProviderParams };