feat: add session key plugin abi and plugingen

alchemyplatform · Dec 15, 2023 · ec3de5b · ec3de5b
1 parent abd4558
commit ec3de5b
Show file tree

Hide file tree

Showing 9 changed files with 1,301 additions and 6 deletions.
diff --git a/examples/aa-simple-dapp/src/hooks/useAlchemyProvider.ts b/examples/aa-simple-dapp/src/hooks/useAlchemyProvider.ts
@@ -1,16 +1,25 @@
 import { chain, gasManagerPolicyId } from "@/config/client";
 import { getRpcUrl } from "@/config/rpc";
 import {
+  MSCA,
+  MultiOwnerPlugin,
+  SessionKeyPlugin,
   createMultiOwnerMSCA,
-  getDefaultLightAccountFactoryAddress,
+  encodeFunctionReference,
+  getDefaultMultiOwnerMSCAFactoryAddress,
+  installPlugin,
 } from "@alchemy/aa-accounts";
 import { AlchemyProvider } from "@alchemy/aa-alchemy";
 import {
   SmartAccountSigner,
   getDefaultEntryPointAddress,
 } from "@alchemy/aa-core";
 import { useCallback, useState } from "react";
-import { Address } from "viem";
+import { Address, encodeAbiParameters, parseAbiParameters } from "viem";
+
+export enum PluginType {
+  SESSION_KEY,
+}
 
 export const useAlchemyProvider = () => {
   const [provider, setProvider] = useState<AlchemyProvider>(
@@ -29,7 +38,7 @@ export const useAlchemyProvider = () => {
             owner: signer,
             chain,
             entryPointAddress: getDefaultEntryPointAddress(chain),
-            factoryAddress: getDefaultLightAccountFactoryAddress(chain),
+            factoryAddress: getDefaultMultiOwnerMSCAFactoryAddress(chain),
             accountAddress: account,
           });
         })
@@ -50,5 +59,42 @@ export const useAlchemyProvider = () => {
     return disconnectedProvider;
   }, [provider]);
 
-  return { provider, connectProviderToAccount, disconnectProviderFromAccount };
+  const pluginInstall = useCallback(
+    async (type: PluginType) => {
+      if (!provider.isConnected<MSCA>()) {
+        return;
+      }
+
+      switch (type) {
+        case PluginType.SESSION_KEY:
+          return installPlugin(provider, {
+            pluginAddress: SessionKeyPlugin.meta.address[chain.id],
+            pluginInitData: encodeAbiParameters(
+              parseAbiParameters("address[]"),
+              [[]]
+            ),
+            dependencies: [
+              encodeFunctionReference(
+                MultiOwnerPlugin.meta.address[chain.id],
+                "0x0"
+              ),
+              encodeFunctionReference(
+                MultiOwnerPlugin.meta.address[chain.id],
+                "0x1"
+              ),
+            ],
+          });
+        default:
+          throw new Error("Unexpected plugin type", type);
+      }
+    },
+    [provider]
+  );
+
+  return {
+    provider,
+    connectProviderToAccount,
+    disconnectProviderFromAccount,
+    pluginInstall,
+  };
 };
diff --git a/packages/accounts/plugindefs/index.ts b/packages/accounts/plugindefs/index.ts
@@ -1,2 +1,3 @@
 export { MultiOwnerPluginGenConfig } from "./multi-owner/config.js";
+export { SessionKeyPluginGenConfig } from "./session-key/config.js";
 export { TokenReceiverPluginGenConfig } from "./token-receiver/config.js";