feat: add WalletClientSigner (#90)

* feat: add `JsonRpcAccountSigner` * fix: plain string message signing in `JsonRpcAccountSigner` Co-authored-by: Michael Moldoveanu <moldy530@gmail.com> * fix(core): rename `JsonRpcAccountSigner` to `WalletClientSigner` Co-authored-by: Michael Moldoveanu <moldy530@gmail.com> * chore: `json-rpc-account` -> `wallet-client` * fix: lint error in `wallet-client.ts` Co-authored-by: Michael Moldoveanu <moldy530@gmail.com> * fix: lint error in `wallet-client.ts` --------- Co-authored-by: Michael Moldoveanu <moldy530@gmail.com>
alchemyplatform · Sep 8, 2023 · b1ede6c · b1ede6c
1 parent ed7d220
commit b1ede6c
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
@@ -11,6 +11,7 @@ export type { BaseSmartAccountParams } from "./account/base.js";
 export { SimpleSmartContractAccount } from "./account/simple.js";
 export type { SimpleSmartAccountParams } from "./account/simple.js";
 export type * from "./account/types.js";
+export { WalletClientSigner } from "./signer/wallet-client.js";
 export { HdAccountSigner } from "./signer/hd-account.js";
 export { LocalAccountSigner } from "./signer/local-account.js";
 export { PrivateKeySigner } from "./signer/private-key.js";

diff --git a/packages/core/src/signer/wallet-client.ts b/packages/core/src/signer/wallet-client.ts
@@ -0,0 +1,46 @@
+import {
+  getAddress,
+  type ByteArray,
+  type Hex,
+  type WalletClient,
+  isHex,
+} from "viem";
+import type { SmartAccountSigner } from "./types";
+import type { SignTypedDataParams } from "../account/types";
+
+export class WalletClientSigner implements SmartAccountSigner {
+  private client: WalletClient;
+
+  constructor(client: WalletClient) {
+    this.client = client;
+  }
+
+  getAddress: () => Promise<`0x${string}`> = async () => {
+    let addresses = await this.client.getAddresses();
+    return getAddress(addresses[0]);
+  };
+
+  readonly signMessage: (
+    message: string | Hex | ByteArray
+  ) => Promise<`0x${string}`> = async (message) => {
+    if (typeof message === "string" && !isHex(message)) {
+      return this.client.signMessage({
+        account: await this.getAddress(),
+        message,
+      });
+    } else {
+      return this.client.signMessage({
+        account: await this.getAddress(),
+        message: { raw: message },
+      });
+    }
+  };
+
+  signTypedData: (params: SignTypedDataParams) => Promise<`0x${string}`> =
+    async (params) => {
+      return this.client.signTypedData({
+        account: await this.getAddress(),
+        ...params,
+      });
+    };
+}