demo: use Turnkey server wallets

Author: tremarkley

packages/demo/backend/package.json

@@ -41,6 +41,7 @@
     "@eth-optimism/viem": "^0.4.13",
     "@hono/node-server": "^1.14.0",
     "@privy-io/server-auth": "^1.31.1",
+    "@turnkey/sdk-server": "^4.9.1",
     "commander": "^13.1.0",
     "dotenv": "^16.4.5",
     "envalid": "^8.1.0",

packages/demo/backend/src/config/env.ts

@@ -58,4 +58,7 @@ export const env = cleanEnv(process.env, {
   UNICHAIN_BUNDLER_URL: str({ devDefault: 'dummy' }),
   UNICHAIN_BUNDLER_SPONSORSHIP_POLICY: str({ devDefault: 'dummy' }),
   SESSION_SIGNER_PK: str(),
+  TURNKEY_ORGANIZATION_ID: str(),
+  TURNKEY_API_KEY: str(),
+  TURNKEY_API_SECRET: str(),
 })

packages/demo/backend/src/config/verbs.ts

@@ -4,21 +4,23 @@ import {
   type VerbsConfig,
 } from '@eth-optimism/verbs-sdk/node'
 import { PrivyClient } from '@privy-io/server-auth'
+import { Turnkey } from '@turnkey/sdk-server'
 import { baseSepolia, unichain } from 'viem/chains'
 
 import { env } from './env.js'
 import { GauntletUSDC, MetaMorphoUSDC, USDCDemoVault } from './markets.js'
 
-let verbsInstance: Verbs<'privy'>
+let verbsInstance: Verbs<'turnkey'>
 
-export function createVerbsConfig(): VerbsConfig<'privy'> {
+export function createVerbsConfig(): VerbsConfig<'turnkey'> {
   return {
     wallet: {
       hostedWalletConfig: {
         provider: {
-          type: 'privy',
+          type: 'turnkey',
           config: {
-            privyClient: getPrivyClient(),
+            client: getTurnkeyClient().apiClient(),
+            organizationId: env.TURNKEY_ORGANIZATION_ID,
           },
         },
       },
@@ -59,7 +61,7 @@ export function createVerbsConfig(): VerbsConfig<'privy'> {
   }
 }
 
-export function initializeVerbs(config?: VerbsConfig<'privy'>): void {
+export function initializeVerbs(config?: VerbsConfig<'turnkey'>): void {
   const verbsConfig = config || createVerbsConfig()
   verbsInstance = createVerbs(verbsConfig)
 }
@@ -78,3 +80,12 @@ export function getPrivyClient() {
   }
   return privy
 }
+
+export function getTurnkeyClient() {
+  return new Turnkey({
+    apiBaseUrl: 'https://api.turnkey.com',
+    apiPublicKey: env.TURNKEY_API_KEY,
+    apiPrivateKey: env.TURNKEY_API_SECRET,
+    defaultOrganizationId: env.TURNKEY_ORGANIZATION_ID,
+  })
+}

packages/demo/backend/src/controllers/wallet.ts

@@ -165,12 +165,7 @@ export class WalletController {
       // TODO (https://github.com/ethereum-optimism/verbs/issues/124): enforce auth and clean
       // up this route.
       if (auth && auth.userId) {
-        const wallet = await walletService.getUserWallet(auth.userId)
-        if (!wallet) {
-          throw new Error('Wallet not found')
-        }
-        const balance = await walletService.getWalletBalance(wallet)
-        return c.json({ balance: serializeBigInt(balance) })
+        throw new Error('Not implemented')
       }
 
       const balance = await walletService.getBalance(userId)
@@ -203,12 +198,7 @@ export class WalletController {
       // TODO (https://github.com/ethereum-optimism/verbs/issues/124): enforce auth and clean
       // up this route.
       if (auth && auth.userId) {
-        const wallet = await walletService.getUserWallet(auth.userId)
-        if (!wallet) {
-          throw new Error('Wallet not found')
-        }
-        const result = await walletService.fundWallet(wallet)
-        return c.json(result)
+        throw new Error('Not implemented')
       }
 
       const wallet = await walletService.getWallet(userId)

packages/demo/backend/src/services/lend.ts

@@ -9,7 +9,7 @@ import type { Address } from 'viem'
 import { baseSepolia, unichain } from 'viem/chains'
 
 import { getVerbs } from '../config/verbs.js'
-import { getUserWallet, getWallet } from './wallet.js'
+import { getWallet } from './wallet.js'
 
 interface MarketBalanceResult {
   balance: bigint
@@ -114,31 +114,12 @@ export async function formatMarketBalanceResponse(
 }
 
 export async function depositWithUserWallet(
-  userId: string,
-  amount: number,
-  tokenAddress: Address,
-  chainId: SupportedChainId,
+  _userId: string,
+  _amount: number,
+  _tokenAddress: Address,
+  _chainId: SupportedChainId,
 ): Promise<LendTransaction> {
-  const wallet = await getUserWallet(userId)
-
-  if (!wallet) {
-    throw new Error(`Wallet not found for user ID: ${userId}`)
-  }
-
-  const asset = SUPPORTED_TOKENS.find(
-    (token) => token.address[chainId] === tokenAddress,
-  )
-  if (!asset) {
-    throw new Error(`Asset not found for token address: ${tokenAddress}`)
-  }
-
-  if ('lendExecute' in wallet && typeof wallet.lendExecute === 'function') {
-    return await wallet.lendExecute(amount, asset, chainId)
-  } else {
-    throw new Error(
-      'Lend functionality not yet implemented for this wallet type.',
-    )
-  }
+  throw new Error('Not implemented')
 }
 
 export async function deposit(
@@ -170,35 +151,11 @@ export async function deposit(
 }
 
 export async function executeLendTransactionWithUserWallet(
-  userId: string,
-  lendTransaction: LendTransaction,
-  chainId: SupportedChainId,
+  _userId: string,
+  _lendTransaction: LendTransaction,
+  _chainId: SupportedChainId,
 ): Promise<LendTransaction & { blockExplorerUrl: string }> {
-  const wallet = await getUserWallet(userId)
-
-  if (!wallet) {
-    throw new Error(`Wallet not found for user ID: ${userId}`)
-  }
-
-  if (!lendTransaction.transactionData) {
-    throw new Error('No transaction data available for execution')
-  }
-
-  const depositHash = lendTransaction.transactionData.approval
-    ? await wallet.sendBatch(
-        [
-          lendTransaction.transactionData.approval,
-          lendTransaction.transactionData.deposit,
-        ],
-        chainId,
-      )
-    : await wallet.send(lendTransaction.transactionData.deposit, chainId)
-
-  return {
-    ...lendTransaction,
-    hash: depositHash,
-    blockExplorerUrl: await getBlockExplorerUrl(chainId),
-  }
+  throw new Error('Not implemented')
 }
 
 export async function executeLendTransaction(

packages/demo/backend/src/services/wallet.ts

@@ -11,8 +11,8 @@ import { baseSepolia } from 'viem/chains'
 
 import { mintableErc20Abi } from '@/abis/mintableErc20Abi.js'
 import { USDC } from '@/config/assets.js'
-import { getPrivyClient, getVerbs } from '@/config/verbs.js'
-
+import { env } from '@/config/env.js'
+import { getTurnkeyClient, getVerbs } from '@/config/verbs.js'
 /**
  * Options for getting all wallets
  * @description Parameters for filtering and paginating wallet results
@@ -24,18 +24,50 @@ export interface GetAllWalletsOptions {
   cursor?: string
 }
 
+async function turnkeyCreateWallet() {
+  const turnkeyClient = getTurnkeyClient()
+  const activityResponse = await turnkeyClient.apiClient().createWallet({
+    walletName: 'ETH Wallet',
+    accounts: [
+      {
+        curve: 'CURVE_SECP256K1',
+        pathFormat: 'PATH_FORMAT_BIP32',
+        path: "m/44'/60'/0'/0/0",
+        addressFormat: 'ADDRESS_FORMAT_ETHEREUM',
+      },
+    ],
+  })
+  console.log(activityResponse)
+  return activityResponse
+}
+
+async function turnkeyGetWallet(walletId: string) {
+  const turnkeyClient = getTurnkeyClient()
+  const activityResponse = await turnkeyClient.apiClient().getWalletAccount({
+    organizationId: env.TURNKEY_ORGANIZATION_ID!,
+    walletId: walletId,
+  })
+  console.log(activityResponse)
+  return activityResponse
+}
+
+async function turnkeyGetAllWallets() {
+  const turnkeyClient = getTurnkeyClient()
+  const activityResponse = await turnkeyClient.apiClient().getWallets({
+    organizationId: env.TURNKEY_ORGANIZATION_ID!,
+  })
+  console.log(activityResponse)
+  return activityResponse
+}
+
 export async function createWallet(): Promise<{
   privyAddress: string
   smartWalletAddress: string
 }> {
   const verbs = getVerbs()
-  const privyClient = getPrivyClient()
-  const privyWallet = await privyClient.walletApi.createWallet({
-    chainType: 'ethereum',
-  })
+  const turnkeyWallet = await turnkeyCreateWallet()
   const verbsPrivyWallet = await verbs.wallet.hostedWalletToVerbsWallet({
-    walletId: privyWallet.id,
-    address: privyWallet.address,
+    signWith: turnkeyWallet.addresses[0],
   })
   const wallet = await verbs.wallet.createSmartWallet({
     owners: [verbsPrivyWallet.address],
@@ -50,70 +82,65 @@ export async function createWallet(): Promise<{
 
 export async function getWallet(userId: string): Promise<SmartWallet | null> {
   const verbs = getVerbs()
-  const privyClient = getPrivyClient()
-  const privyWallet = await privyClient.walletApi
-    .getWallet({
-      id: userId,
-    })
-    .catch(() => null)
-  if (!privyWallet) {
-    return privyWallet
-  }
-  const verbsPrivyWallet = await verbs.wallet.hostedWalletToVerbsWallet({
-    walletId: privyWallet.id,
-    address: privyWallet.address,
-  })
-  const wallet = await verbs.wallet.getSmartWallet({
-    signer: verbsPrivyWallet.signer,
-    deploymentOwners: [getAddress(privyWallet.address)],
-  })
-  return wallet
-}
-
-export async function getUserWallet(
-  userId: string,
-): Promise<SmartWallet | null> {
-  const verbs = getVerbs()
-  const privyClient = getPrivyClient()
-  const privyUser = await privyClient.getUserById(userId).catch(() => null)
-  if (!privyUser) {
+  const turnkeyWallet = await turnkeyGetWallet(userId).catch(() => null)
+  if (!turnkeyWallet) {
     return null
   }
-  const privyWallet = privyUser.wallet
-  if (!privyWallet) {
-    return null
-  }
-  const verbsPrivyWallet = await verbs.wallet.hostedWalletToVerbsWallet({
-    walletId: privyWallet.id!,
-    address: privyWallet.address,
+  const verbsTurnkeyWallet = await verbs.wallet.hostedWalletToVerbsWallet({
+    signWith: turnkeyWallet.account.address,
   })
   const wallet = await verbs.wallet.getSmartWallet({
-    signer: verbsPrivyWallet.signer,
-    deploymentOwners: [getAddress(privyWallet.address)],
+    signer: verbsTurnkeyWallet.signer,
+    deploymentOwners: [getAddress(turnkeyWallet.account.address)],
   })
   return wallet
 }
 
+// export async function getUserWallet(
+//   userId: string,
+// ): Promise<SmartWallet | null> {
+//   const verbs = getVerbs()
+//   const privyClient = getPrivyClient()
+//   const privyUser = await privyClient.getUserById(userId).catch(() => null)
+//   if (!privyUser) {
+//     return null
+//   }
+//   const privyWallet = privyUser.wallet
+//   if (!privyWallet) {
+//     return null
+//   }
+//   const verbsPrivyWallet = await verbs.wallet.hostedWalletToVerbsWallet({
+//     walletId: privyWallet.id!,
+//     address: privyWallet.address,
+//   })
+//   const wallet = await verbs.wallet.getSmartWallet({
+//     signer: verbsPrivyWallet.signer,
+//     deploymentOwners: [getAddress(privyWallet.address)],
+//   })
+//   return wallet
+// }
+
 export async function getAllWallets(
-  options?: GetAllWalletsOptions,
+  _options?: GetAllWalletsOptions,
 ): Promise<Array<{ wallet: SmartWallet; id: string }>> {
   try {
     const verbs = getVerbs()
-    const privyClient = getPrivyClient()
-    const response = await privyClient.walletApi.getWallets(options)
+    const turnkeyWallets = await turnkeyGetAllWallets()
     return Promise.all(
-      response.data.map(async (privyWallet) => {
-        const verbsPrivyWallet = await verbs.wallet.hostedWalletToVerbsWallet({
-          walletId: privyWallet.id,
-          address: privyWallet.address,
-        })
+      turnkeyWallets.wallets.map(async ({ walletId }) => {
+        const turnkeyWallet = await turnkeyGetWallet(walletId)
+        const verbsTurnkeyWallet = await verbs.wallet.hostedWalletToVerbsWallet(
+          {
+            signWith: turnkeyWallet.account.address,
+          },
+        )
         const wallet = await verbs.wallet.getSmartWallet({
-          signer: verbsPrivyWallet.signer,
-          deploymentOwners: [getAddress(privyWallet.address)],
+          signer: verbsTurnkeyWallet.signer,
+          deploymentOwners: [getAddress(turnkeyWallet.account.address)],
         })
         return {
           wallet,
-          id: privyWallet.id,
+          id: walletId,
         }
       }),
     )