core: add utility for signing account signature message as backend wa…

…llet
TreasureProject · Sep 17, 2024 · 947e37c · 947e37c
1 parent 56d054b
commit 947e37c
Show file tree

Hide file tree

Showing 5 changed files with 76 additions and 16 deletions.
diff --git a/.changeset/quick-clouds-fly.md b/.changeset/quick-clouds-fly.md
@@ -0,0 +1,5 @@
+---
+"@treasure-dev/tdk-core": patch
+---
+
+Added utility for signing account signature message as backend wallet
diff --git a/examples/connect-core/src/main.ts b/examples/connect-core/src/main.ts
@@ -101,7 +101,7 @@ document.querySelector<HTMLDivElement>("#app")!.innerHTML = `
     try {
       const result = await logInWithSocial({
         client,
-        network: "google",
+        method: "google",
         apiUri,
         chainId,
         sessionOptions,

diff --git a/packages/core/src/api.ts b/packages/core/src/api.ts
@@ -208,19 +208,37 @@ export class TDKAPI {
           "inputs"
         >;
       },
-      options?: { includeAbi?: boolean; skipWaitForCompletion?: boolean },
+      options?: {
+        includeAbi?: boolean;
+        skipWaitForCompletion?: boolean;
+        accountAddress?: string;
+        accountSignature?: string;
+      },
     ) => {
       const result = await this.post<
         CreateTransactionBody,
         CreateTransactionReply
-      >("/transactions", {
-        ...params,
-        // biome-ignore lint/suspicious/noExplicitAny: abitype and the API schema don't play well
-        ...(options?.includeAbi ? { abi: params.abi as any } : {}),
-        // biome-ignore lint/suspicious/noExplicitAny: abitype and the API schema don't play well
-        args: params.args as any,
-        backendWallet: params.backendWallet ?? this.backendWallet,
-      });
+      >(
+        "/transactions",
+        {
+          ...params,
+          // biome-ignore lint/suspicious/noExplicitAny: abitype and the API schema don't play well
+          ...(options?.includeAbi ? { abi: params.abi as any } : {}),
+          // biome-ignore lint/suspicious/noExplicitAny: abitype and the API schema don't play well
+          args: params.args as any,
+          backendWallet: params.backendWallet ?? this.backendWallet,
+        },
+        {
+          headers: {
+            ...(options?.accountAddress
+              ? { "x-account-address": options.accountAddress }
+              : {}),
+            ...(options?.accountSignature
+              ? { "x-account-signature": options.accountSignature }
+              : {}),
+          },
+        },
+      );
 
       return options?.skipWaitForCompletion
         ? result
@@ -230,16 +248,33 @@ export class TDKAPI {
       params: Omit<CreateRawTransactionBody, "value"> & {
         value?: bigint;
       },
-      options?: { skipWaitForCompletion?: boolean },
+      options?: {
+        skipWaitForCompletion?: boolean;
+        accountAddress?: string;
+        accountSignature?: string;
+      },
     ) => {
       const result = await this.post<
         CreateRawTransactionBody,
         CreateRawTransactionReply
-      >("/transactions/raw", {
-        ...params,
-        value: params.value ? toHex(params.value) : undefined,
-        backendWallet: params.backendWallet ?? this.backendWallet,
-      });
+      >(
+        "/transactions/raw",
+        {
+          ...params,
+          value: params.value ? toHex(params.value) : undefined,
+          backendWallet: params.backendWallet ?? this.backendWallet,
+        },
+        {
+          headers: {
+            ...(options?.accountAddress
+              ? { "x-account-address": options.accountAddress }
+              : {}),
+            ...(options?.accountSignature
+              ? { "x-account-signature": options.accountSignature }
+              : {}),
+          },
+        },
+      );
 
       return options?.skipWaitForCompletion
         ? result

diff --git a/packages/core/src/connect/auth.ts b/packages/core/src/connect/auth.ts
@@ -0,0 +1,19 @@
+import { privateKeyToAccount } from "thirdweb/wallets";
+
+import type { TreasureConnectClient } from "../types";
+
+export const generateAccountSignature = ({
+  client,
+  accountAddress,
+  backendWalletPrivateKey,
+}: {
+  client: TreasureConnectClient;
+  accountAddress: string;
+  backendWalletPrivateKey: string;
+}) =>
+  privateKeyToAccount({
+    client,
+    privateKey: backendWalletPrivateKey,
+  }).signMessage({
+    message: JSON.stringify({ accountAddress }),
+  });
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
@@ -32,6 +32,7 @@ export {
 
 // Connect
 export { getAllActiveSigners } from "./connect/accounts";
+export { generateAccountSignature } from "./connect/auth";
 export { decodeAuthToken } from "./connect/jwt";
 export {
   type SocialConnectMethod,