@zerodev/wallet-core 0.0.1-alpha.4 → 0.0.1-alpha.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zerodev/wallet-core",
-  "version": "0.0.1-alpha.4",
+  "version": "0.0.1-alpha.5",
   "description": "ZeroDev Wallet SDK built on Turnkey",
   "main": "./dist/_cjs/index.js",
   "module": "./dist/_esm/index.js",

package/src/actions/wallet/getUserWallet.ts

@@ -6,11 +6,13 @@ export type GetUserWalletParameters = {
   organizationId: string
   /** The project ID for the request */
   projectId: string
+  /** The token for the request */
+  token: string
 }
 
 export type GetUserWalletReturnType = {
   /** The wallet address */
-  walletAddress: Hex
+  walletAddresses: Hex[]
   /** The user ID */
   userId?: string
 }
@@ -28,19 +30,24 @@ export type GetUserWalletReturnType = {
  *   organizationId: 'org_123',
  *   projectId: 'proj_456'
  * });
- * console.log(wallet.walletAddress); // '0x...'
+ * console.log(wallet.walletAddresses); // ['0x...', '0x...']
  * ```
  */
 export async function getUserWallet(
   client: Client,
   params: GetUserWalletParameters,
 ): Promise<GetUserWalletReturnType> {
-  const { organizationId, projectId } = params
+  const { organizationId, projectId, token } = params
 
   return await client.request({
     path: `${projectId}/user-wallet`,
     body: {
       organizationId,
     },
+    headers: {
+      Authorization: `Bearer ${token}`,
+    },
+    stamp: true,
+    stampPostion: 'headers',
   })
 }

package/src/adapters/viem.ts

@@ -24,12 +24,13 @@ export interface ToViemAccountParams {
   client: ZeroDevWalletClient
   organizationId: string
   projectId: string
+  token: string
 }
 
 export async function toViemAccount(
   params: ToViemAccountParams,
 ): Promise<LocalAccount> {
-  const { client, organizationId, projectId } = params
+  const { client, organizationId, projectId, token } = params
 
   let address: Hex = zeroAddress
 
@@ -37,8 +38,9 @@ export async function toViemAccount(
     const walletResponse = await client.getUserWallet({
       organizationId,
       projectId,
+      token,
     })
-    address = walletResponse.walletAddress
+    address = walletResponse.walletAddresses[0]
   } catch {
     address = zeroAddress
   }

package/src/client/transports/rest.ts

@@ -8,6 +8,7 @@ export type RestRequestArgs = {
   headers?: Record<string, string>
   stamp?: boolean
   stampWith?: 'indexedDb' | 'webAuthn'
+  stampPostion?: 'body' | 'headers'
 }
 
 export type RestRequestFn = <T = any>(args: RestRequestArgs) => Promise<T>
@@ -44,7 +45,7 @@ export function rest(url: string, cfg: RestTransportConfig): RestTransport {
 
     try {
       let requestBody = args.body
-      const requestHeaders = {
+      let requestHeaders = {
         'content-type': 'application/json',
         ...(args.headers ?? {}),
         ...(cfg.fetchOptions?.headers ?? {}),
@@ -65,7 +66,12 @@ export function rest(url: string, cfg: RestTransportConfig): RestTransport {
         const stamp = await stamper.stamp(bodyString)
 
         // Restructure request body to match backend expectation
-        if (body) {
+        if (args.stampPostion === 'headers') {
+          requestHeaders = {
+            ...requestHeaders,
+            [stamp.stampHeaderName]: stamp.stampHeaderValue,
+          }
+        } else if (body) {
           requestBody = {
             body: bodyString,
             stamp: {

package/src/core/createZeroDevWallet.ts

@@ -422,6 +422,7 @@ export async function createZeroDevWallet(
         client,
         organizationId: session.organizationId,
         projectId,
+        token: session.token ?? '',
       })
     },
   }