tempo.ts 0.1.5 → 0.2.0

package/src/viem/index.ts

@@ -1,16 +1,19 @@
+export * as Tick from '../ox/Tick.js'
 export * as Abis from './Abis.js'
+export * as Account from './Account.js'
 export * as Actions from './Actions/index.js'
 export * as Addresses from './Addresses.js'
 export * as Chain from './Chain.js'
-export * as Client from './Client.js'
-export { createTempoClient } from './Client.js'
 export {
   type Decorator as TempoActions,
   decorator as tempoActions,
 } from './Decorator.js'
 export * as Formatters from './Formatters.js'
-export * as Tick from './Tick.js'
+export * as P256 from './P256.js'
+export * as Secp256k1 from './Secp256k1.js'
 export * as TokenIds from './TokenIds.js'
 export * as Transaction from './Transaction.js'
 export * as Transport from './Transport.js'
 export { withFeePayer } from './Transport.js'
+export * as WebAuthnP256 from './WebAuthnP256.js'
+export * as WebCryptoP256 from './WebCryptoP256.js'

package/src/viem/internal/account.ts

@@ -0,0 +1,107 @@
+import * as Hex from 'ox/Hex'
+import * as PublicKey from 'ox/PublicKey'
+import type { LocalAccount } from 'viem'
+import {
+  hashAuthorization,
+  hashMessage,
+  hashTypedData,
+  keccak256,
+} from 'viem/utils'
+import type { RequiredBy } from '../../internal/types.js'
+import * as SignatureEnvelope from '../../ox/SignatureEnvelope.js'
+import * as Transaction from '../Transaction.js'
+
+export type Account = RequiredBy<
+  Omit<LocalAccount, 'signTransaction'>,
+  'sign' | 'signAuthorization'
+> & {
+  signTransaction: (
+    transaction: Transaction.TransactionSerializable,
+  ) => Promise<Hex.Hex>
+}
+
+// TODO: this function will be redundant when Viem migrates to Ox.
+/** @internal */
+export function toPrivateKeyAccount(
+  parameters: toPrivateKeyAccount.Parameters,
+): toPrivateKeyAccount.ReturnValue {
+  const { address, sign, source = 'privateKey' } = parameters
+  const publicKey = PublicKey.toHex(parameters.publicKey, {
+    includePrefix: false,
+  })
+  return {
+    address,
+    sign,
+    async signAuthorization(parameters) {
+      const { chainId, nonce } = parameters
+      const address = parameters.contractAddress ?? parameters.address
+      const signature = await sign({
+        hash: hashAuthorization({ address, chainId, nonce }),
+      })
+      const envelope = SignatureEnvelope.from(signature)
+      if (envelope.type !== 'secp256k1')
+        throw new Error(
+          'Unsupported signature type. Expected `secp256k1` but got `' +
+            envelope.type +
+            '`.',
+        )
+      const { r, s, yParity } = envelope.signature
+      return {
+        address,
+        chainId,
+        nonce,
+        r: Hex.fromNumber(r, { size: 32 }),
+        s: Hex.fromNumber(s, { size: 32 }),
+        yParity,
+      }
+    },
+    async signMessage(parameters) {
+      const { message } = parameters
+      const signature = await sign({ hash: hashMessage(message) })
+      const envelope = SignatureEnvelope.from(signature)
+      if (envelope.type !== 'secp256k1')
+        throw new Error(
+          'Unsupported signature type. Expected `secp256k1` but got `' +
+            envelope.type +
+            '`.',
+        )
+      return SignatureEnvelope.serialize(envelope)
+    },
+    async signTransaction(transaction) {
+      const signature = await sign({
+        hash: keccak256(await Transaction.serialize(transaction)),
+      })
+      const envelope = SignatureEnvelope.from(signature)
+      return await Transaction.serialize(transaction, envelope)
+    },
+    async signTypedData(typedData) {
+      const signature = await sign({ hash: hashTypedData(typedData) })
+      const envelope = SignatureEnvelope.from(signature)
+      if (envelope.type !== 'secp256k1')
+        throw new Error(
+          'Unsupported signature type. Expected `secp256k1` but got `' +
+            envelope.type +
+            '`.',
+        )
+      return SignatureEnvelope.serialize(envelope)
+    },
+    publicKey,
+    source,
+    type: 'local',
+  }
+}
+
+export declare namespace toPrivateKeyAccount {
+  export type Parameters = {
+    /** Address. */
+    address: Hex.Hex
+    /** Public key. */
+    publicKey: PublicKey.PublicKey
+    /** Sign function. */
+    sign: NonNullable<LocalAccount['sign']>
+    /** Source. */
+    source?: string | undefined
+  }
+
+  export type ReturnValue = Account
+}

package/src/viem/internal/types.ts

@@ -10,6 +10,7 @@ import type {
   MaybeRequired,
   UnionPick,
 } from '../../internal/types.js'
+import type * as TokenId from '../../ox/TokenId.js'
 import type { TransactionRequestAA } from '../Transaction.js'
 
 export type GetAccountParameter<
@@ -33,6 +34,14 @@ export type GetAccountParameter<
     : false
 >
 
+export type GetFeeTokenParameter<
+  chain extends Chain | undefined = Chain | undefined,
+> = chain extends { feeToken: infer feeToken }
+  ? IsUndefined<feeToken> extends true
+    ? { feeToken: TokenId.TokenIdOrAddress | null }
+    : { feeToken?: TokenId.TokenIdOrAddress | null | undefined }
+  : { feeToken: TokenId.TokenIdOrAddress | null }
+
 export type ReadParameters = Pick<
   viem_ReadContractParameters<never, never, never>,
   'account' | 'blockNumber' | 'blockOverrides' | 'blockTag' | 'stateOverride'

package/src/wagmi/Actions/__snapshots__/dex.test.ts.snap

@@ -0,0 +1,310 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`getOrders > default 1`] = `
+{
+  "nextCursor": "0x9",
+  "orders": [
+    {
+      "amount": 230000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 39n,
+      "orderId": 19n,
+      "prev": 0n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 230000000n,
+      "tick": 800,
+    },
+    {
+      "amount": 430000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 0n,
+      "orderId": 39n,
+      "prev": 19n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 430000000n,
+      "tick": 800,
+    },
+    {
+      "amount": 210000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 37n,
+      "orderId": 17n,
+      "prev": 0n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 210000000n,
+      "tick": 600,
+    },
+    {
+      "amount": 410000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 0n,
+      "orderId": 37n,
+      "prev": 17n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 410000000n,
+      "tick": 600,
+    },
+    {
+      "amount": 190000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 35n,
+      "orderId": 15n,
+      "prev": 0n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 190000000n,
+      "tick": 400,
+    },
+    {
+      "amount": 390000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 0n,
+      "orderId": 35n,
+      "prev": 15n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 390000000n,
+      "tick": 400,
+    },
+    {
+      "amount": 170000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 33n,
+      "orderId": 13n,
+      "prev": 0n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 170000000n,
+      "tick": 200,
+    },
+    {
+      "amount": 370000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 0n,
+      "orderId": 33n,
+      "prev": 13n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 370000000n,
+      "tick": 200,
+    },
+    {
+      "amount": 150000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 31n,
+      "orderId": 11n,
+      "prev": 0n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 150000000n,
+      "tick": 0,
+    },
+    {
+      "amount": 350000000n,
+      "baseToken": "0x20c0000000000000000000000000000000000005",
+      "flipTick": 0,
+      "isBid": true,
+      "isFlip": false,
+      "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+      "next": 0n,
+      "orderId": 31n,
+      "prev": 11n,
+      "quoteToken": "0x20c0000000000000000000000000000000000004",
+      "remaining": 350000000n,
+      "tick": 0,
+    },
+  ],
+}
+`;
+
+exports[`getOrders > infiniteQueryOptions > default 1`] = `
+{
+  "pageParams": [
+    undefined,
+    "0x23",
+  ],
+  "pages": [
+    {
+      "nextCursor": "0x23",
+      "orders": [
+        {
+          "amount": 230000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 39n,
+          "orderId": 19n,
+          "prev": 0n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 230000000n,
+          "tick": 800,
+        },
+        {
+          "amount": 430000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 0n,
+          "orderId": 39n,
+          "prev": 19n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 430000000n,
+          "tick": 800,
+        },
+        {
+          "amount": 210000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 37n,
+          "orderId": 17n,
+          "prev": 0n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 210000000n,
+          "tick": 600,
+        },
+        {
+          "amount": 410000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 0n,
+          "orderId": 37n,
+          "prev": 17n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 410000000n,
+          "tick": 600,
+        },
+        {
+          "amount": 190000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 35n,
+          "orderId": 15n,
+          "prev": 0n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 190000000n,
+          "tick": 400,
+        },
+      ],
+    },
+    {
+      "nextCursor": "0x9",
+      "orders": [
+        {
+          "amount": 390000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 0n,
+          "orderId": 35n,
+          "prev": 15n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 390000000n,
+          "tick": 400,
+        },
+        {
+          "amount": 170000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 33n,
+          "orderId": 13n,
+          "prev": 0n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 170000000n,
+          "tick": 200,
+        },
+        {
+          "amount": 370000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 0n,
+          "orderId": 33n,
+          "prev": 13n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 370000000n,
+          "tick": 200,
+        },
+        {
+          "amount": 150000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 31n,
+          "orderId": 11n,
+          "prev": 0n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 150000000n,
+          "tick": 0,
+        },
+        {
+          "amount": 350000000n,
+          "baseToken": "0x20c0000000000000000000000000000000000005",
+          "flipTick": 0,
+          "isBid": true,
+          "isFlip": false,
+          "maker": "0xf39fd6e51aad88f6f4ce6ab8827279cfffb92266",
+          "next": 0n,
+          "orderId": 31n,
+          "prev": 11n,
+          "quoteToken": "0x20c0000000000000000000000000000000000004",
+          "remaining": 350000000n,
+          "tick": 0,
+        },
+      ],
+    },
+  ],
+}
+`;

package/src/wagmi/Actions/amm.test.ts

@@ -0,0 +1,198 @@
+import { connect, getConnectorClient } from '@wagmi/core'
+import { parseUnits } from 'viem'
+import { describe, expect, test } from 'vitest'
+import {
+  accounts,
+  addresses,
+  setupPoolWithLiquidity,
+} from '../../../test/viem/config.js'
+import { config, queryClient } from '../../../test/wagmi/config.js'
+import * as ammActions from './amm.js'
+import * as tokenActions from './token.js'
+
+const account = accounts[0]
+
+describe('getPool', () => {
+  test('default', async () => {
+    const pool = await ammActions.getPool(config, {
+      userToken: addresses.alphaUsd,
+      validatorToken: '0x20c0000000000000000000000000000000000001',
+    })
+    expect(pool).toMatchInlineSnapshot(`
+      {
+        "reserveUserToken": 0n,
+        "reserveValidatorToken": 0n,
+        "totalSupply": 0n,
+      }
+    `)
+  })
+
+  describe('queryOptions', () => {
+    test('default', async () => {
+      const options = ammActions.getPool.queryOptions(config, {
+        userToken: addresses.alphaUsd,
+        validatorToken: '0x20c0000000000000000000000000000000000001',
+      })
+      const pool = await queryClient.fetchQuery(options)
+      expect(pool).toMatchInlineSnapshot(`
+        {
+          "reserveUserToken": 0n,
+          "reserveValidatorToken": 0n,
+          "totalSupply": 0n,
+        }
+      `)
+    })
+  })
+})
+
+describe('getLiquidityBalance', () => {
+  test('default', async () => {
+    const balance = await ammActions.getLiquidityBalance(config, {
+      userToken: addresses.alphaUsd,
+      validatorToken: '0x20c0000000000000000000000000000000000001',
+      address: account.address,
+    })
+    expect(balance).toMatchInlineSnapshot(`0n`)
+  })
+
+  describe('queryOptions', () => {
+    test('default', async () => {
+      const options = ammActions.getLiquidityBalance.queryOptions(config, {
+        userToken: addresses.alphaUsd,
+        validatorToken: '0x20c0000000000000000000000000000000000001',
+        address: account.address,
+      })
+      const balance = await queryClient.fetchQuery(options)
+      expect(balance).toMatchInlineSnapshot(`0n`)
+    })
+  })
+})
+
+describe('mintSync', () => {
+  test('default', async () => {
+    await connect(config, {
+      connector: config.connectors[0]!,
+    })
+
+    // Create a new token for testing
+    const { token } = await tokenActions.createSync(config, {
+      name: 'Test Token',
+      symbol: 'TEST',
+      currency: 'USD',
+    })
+
+    // Grant issuer role to mint tokens
+    await tokenActions.grantRolesSync(config, {
+      token,
+      roles: ['issuer'],
+      to: account.address,
+    })
+
+    // Mint some tokens to account
+    await tokenActions.mintSync(config, {
+      to: account.address,
+      amount: parseUnits('1000', 6),
+      token,
+    })
+
+    // Add liquidity to pool
+    const { receipt: mintReceipt, ...mintResult } = await ammActions.mintSync(
+      config,
+      {
+        userToken: {
+          address: token,
+          amount: parseUnits('100', 6),
+        },
+        validatorToken: {
+          address: addresses.alphaUsd,
+          amount: parseUnits('100', 6),
+        },
+        to: account.address,
+      },
+    )
+    expect(mintReceipt).toBeDefined()
+    expect(mintResult).toMatchInlineSnapshot(`
+      {
+        "amountUserToken": 100000000n,
+        "amountValidatorToken": 100000000n,
+        "liquidity": 4999999999999000n,
+        "sender": "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
+        "userToken": "0x20C0000000000000000000000000000000000004",
+        "validatorToken": "0x20C0000000000000000000000000000000000001",
+      }
+    `)
+  })
+})
+
+describe('burnSync', () => {
+  test('default', async () => {
+    await connect(config, {
+      connector: config.connectors[0]!,
+    })
+
+    const client = await getConnectorClient(config)
+    const { tokenAddress } = await setupPoolWithLiquidity(client)
+
+    // Get LP balance before burn
+    const lpBalanceBefore = await ammActions.getLiquidityBalance(config, {
+      userToken: tokenAddress,
+      validatorToken: addresses.alphaUsd,
+      address: account.address,
+    })
+
+    // Burn half of LP tokens
+    const { receipt: burnReceipt, ...burnResult } = await ammActions.burnSync(
+      config,
+      {
+        userToken: tokenAddress,
+        validatorToken: addresses.alphaUsd,
+        liquidity: lpBalanceBefore / 2n,
+        to: account.address,
+      },
+    )
+    expect(burnReceipt).toBeDefined()
+    expect(burnResult).toMatchInlineSnapshot(`
+      {
+        "amountUserToken": 49999999n,
+        "amountValidatorToken": 49999999n,
+        "liquidity": 2499999999999500n,
+        "sender": "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
+        "to": "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
+        "userToken": "0x20c0000000000000000000000000000000000005",
+        "validatorToken": "0x20C0000000000000000000000000000000000001",
+      }
+    `)
+  })
+})
+
+describe('rebalanceSwapSync', () => {
+  test('default', async () => {
+    await connect(config, {
+      connector: config.connectors[0]!,
+    })
+
+    const client = await getConnectorClient(config)
+    const { tokenAddress } = await setupPoolWithLiquidity(client)
+
+    const account2 = accounts[1]
+
+    // Perform rebalance swap
+    const { receipt: swapReceipt, ...swapResult } =
+      await ammActions.rebalanceSwapSync(config, {
+        userToken: tokenAddress,
+        validatorToken: addresses.alphaUsd,
+        amountOut: parseUnits('10', 6),
+        to: account2.address,
+      })
+    expect(swapReceipt).toBeDefined()
+    expect(swapResult).toMatchInlineSnapshot(`
+      {
+        "amountIn": 9985001n,
+        "amountOut": 10000000n,
+        "swapper": "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
+        "userToken": "0x20C0000000000000000000000000000000000006",
+        "validatorToken": "0x20C0000000000000000000000000000000000001",
+      }
+    `)
+  })
+})