tempo.ts 0.0.6 → 0.1.1

package/src/viem/{actions → Actions}/amm.ts

@@ -23,10 +23,10 @@ import {
 } from 'viem/actions'
 import type { Compute, UnionOmit } from '../../internal/types.js'
 import * as TokenId from '../../ox/TokenId.js'
-import { feeAmmAbi } from '../abis.js'
-import { feeManagerAddress } from '../addresses.js'
-import type { ReadParameters, WriteParameters } from '../types.js'
-import { defineCall } from '../utils.js'
+import * as Abis from '../Abis.js'
+import * as Addresses from '../Addresses.js'
+import type { ReadParameters, WriteParameters } from '../internal/types.js'
+import { defineCall } from '../internal/utils.js'
 
 /**
  * Gets the pool ID for a token pair.
@@ -56,9 +56,10 @@ export async function getPoolId<chain extends Chain | undefined>(
   client: Client<Transport, chain>,
   parameters: getPoolId.Parameters,
 ): Promise<getPoolId.ReturnValue> {
+  const { userToken, validatorToken, ...rest } = parameters
   return readContract(client, {
-    ...parameters,
-    ...getPoolId.call(parameters),
+    ...rest,
+    ...getPoolId.call({ userToken, validatorToken }),
   })
 }
 
@@ -73,7 +74,7 @@ export namespace getPoolId {
   }
 
   export type ReturnValue = ReadContractReturnType<
-    typeof feeAmmAbi,
+    typeof Abis.feeAmm,
     'getPoolId',
     never
   >
@@ -87,8 +88,8 @@ export namespace getPoolId {
   export function call(args: Args) {
     const { userToken, validatorToken } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       args: [TokenId.toAddress(userToken), TokenId.toAddress(validatorToken)],
       functionName: 'getPoolId',
     })
@@ -123,9 +124,10 @@ export async function getPool<chain extends Chain | undefined>(
   client: Client<Transport, chain>,
   parameters: getPool.Parameters,
 ): Promise<getPool.ReturnValue> {
+  const { userToken, validatorToken, ...rest } = parameters
   return readContract(client, {
-    ...parameters,
-    ...getPool.call(parameters),
+    ...rest,
+    ...getPool.call({ userToken, validatorToken }),
   })
 }
 
@@ -155,8 +157,8 @@ export namespace getPool {
   export function call(args: Args) {
     const { userToken, validatorToken } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       args: [TokenId.toAddress(userToken), TokenId.toAddress(validatorToken)],
       functionName: 'getPool',
     })
@@ -193,9 +195,10 @@ export async function getTotalSupply<chain extends Chain | undefined>(
   client: Client<Transport, chain>,
   parameters: getTotalSupply.Parameters,
 ): Promise<getTotalSupply.ReturnValue> {
+  const { poolId, ...rest } = parameters
   return readContract(client, {
-    ...parameters,
-    ...getTotalSupply.call(parameters),
+    ...rest,
+    ...getTotalSupply.call({ poolId }),
   })
 }
 
@@ -208,7 +211,7 @@ export namespace getTotalSupply {
   }
 
   export type ReturnValue = ReadContractReturnType<
-    typeof feeAmmAbi,
+    typeof Abis.feeAmm,
     'totalSupply',
     never
   >
@@ -222,8 +225,8 @@ export namespace getTotalSupply {
   export function call(args: Args) {
     const { poolId } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       args: [poolId],
       functionName: 'totalSupply',
     })
@@ -263,9 +266,10 @@ export async function getLiquidityBalance<chain extends Chain | undefined>(
   client: Client<Transport, chain>,
   parameters: getLiquidityBalance.Parameters,
 ): Promise<getLiquidityBalance.ReturnValue> {
+  const { address, poolId, ...rest } = parameters
   return readContract(client, {
-    ...parameters,
-    ...getLiquidityBalance.call(parameters),
+    ...rest,
+    ...getLiquidityBalance.call({ address, poolId }),
   })
 }
 
@@ -280,7 +284,7 @@ export namespace getLiquidityBalance {
   }
 
   export type ReturnValue = ReadContractReturnType<
-    typeof feeAmmAbi,
+    typeof Abis.feeAmm,
     'liquidityBalances',
     never
   >
@@ -294,8 +298,8 @@ export namespace getLiquidityBalance {
   export function call(args: Args) {
     const { address, poolId } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       args: [poolId, address],
       functionName: 'liquidityBalances',
     })
@@ -369,9 +373,15 @@ export namespace rebalanceSwap {
     client: Client<Transport, chain, account>,
     parameters: rebalanceSwap.Parameters<chain, account>,
   ): Promise<ReturnType<action>> {
-    const call = rebalanceSwap.call(parameters)
+    const { userToken, validatorToken, amountOut, to, ...rest } = parameters
+    const call = rebalanceSwap.call({
+      userToken,
+      validatorToken,
+      amountOut,
+      to,
+    })
     return (await action(client, {
-      ...parameters,
+      ...rest,
       ...call,
     } as never)) as never
   }
@@ -419,8 +429,8 @@ export namespace rebalanceSwap {
   export function call(args: Args) {
     const { userToken, validatorToken, amountOut, to } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       functionName: 'rebalanceSwap',
       args: [
         TokenId.toAddress(userToken),
@@ -439,7 +449,7 @@ export namespace rebalanceSwap {
    */
   export function extractEvent(logs: Log[]) {
     const [log] = parseEventLogs({
-      abi: feeAmmAbi,
+      abi: Abis.feeAmm,
       logs,
       eventName: 'RebalanceSwap',
       strict: true,
@@ -506,7 +516,7 @@ export namespace rebalanceSwapSync {
 
   export type ReturnValue = Compute<
     GetEventArgs<
-      typeof feeAmmAbi,
+      typeof Abis.feeAmm,
       'RebalanceSwap',
       { IndexedOnly: false; Required: true }
     > & {
@@ -596,9 +606,10 @@ export namespace mint {
     client: Client<Transport, chain, account>,
     parameters: mint.Parameters<chain, account>,
   ): Promise<ReturnType<action>> {
-    const call = mint.call(parameters)
+    const { to, userToken, validatorToken, ...rest } = parameters
+    const call = mint.call({ to, userToken, validatorToken })
     return (await action(client, {
-      ...parameters,
+      ...rest,
       ...call,
     } as never)) as never
   }
@@ -656,8 +667,8 @@ export namespace mint {
   export function call(args: Args) {
     const { to, userToken, validatorToken } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       functionName: 'mint',
       args: [
         TokenId.toAddress(userToken.address),
@@ -677,7 +688,7 @@ export namespace mint {
    */
   export function extractEvent(logs: Log[]) {
     const [log] = parseEventLogs({
-      abi: feeAmmAbi,
+      abi: Abis.feeAmm,
       logs,
       eventName: 'Mint',
       strict: true,
@@ -749,7 +760,7 @@ export namespace mintSync {
 
   export type ReturnValue = Compute<
     GetEventArgs<
-      typeof feeAmmAbi,
+      typeof Abis.feeAmm,
       'Mint',
       { IndexedOnly: false; Required: true }
     > & {
@@ -826,9 +837,10 @@ export namespace burn {
     client: Client<Transport, chain, account>,
     parameters: burn.Parameters<chain, account>,
   ): Promise<ReturnType<action>> {
-    const call = burn.call(parameters)
+    const { liquidity, to, userToken, validatorToken, ...rest } = parameters
+    const call = burn.call({ liquidity, to, userToken, validatorToken })
     return (await action(client, {
-      ...parameters,
+      ...rest,
       ...call,
     } as never)) as never
   }
@@ -876,8 +888,8 @@ export namespace burn {
   export function call(args: Args) {
     const { liquidity, to, userToken, validatorToken } = args
     return defineCall({
-      address: feeManagerAddress,
-      abi: feeAmmAbi,
+      address: Addresses.feeManager,
+      abi: Abis.feeAmm,
       functionName: 'burn',
       args: [
         TokenId.toAddress(userToken),
@@ -896,7 +908,7 @@ export namespace burn {
    */
   export function extractEvent(logs: Log[]) {
     const [log] = parseEventLogs({
-      abi: feeAmmAbi,
+      abi: Abis.feeAmm,
       logs,
       eventName: 'Burn',
       strict: true,
@@ -963,7 +975,7 @@ export namespace burnSync {
 
   export type ReturnValue = Compute<
     GetEventArgs<
-      typeof feeAmmAbi,
+      typeof Abis.feeAmm,
       'Burn',
       { IndexedOnly: false; Required: true }
     > & {
@@ -1008,8 +1020,8 @@ export function watchRebalanceSwap<
   const { onRebalanceSwap, userToken, validatorToken, ...rest } = parameters
   return watchContractEvent(client, {
     ...rest,
-    address: feeManagerAddress,
-    abi: feeAmmAbi,
+    address: Addresses.feeManager,
+    abi: Abis.feeAmm,
     eventName: 'RebalanceSwap',
     args:
       userToken !== undefined && validatorToken !== undefined
@@ -1027,7 +1039,7 @@ export function watchRebalanceSwap<
 
 export declare namespace watchRebalanceSwap {
   export type Args = GetEventArgs<
-    typeof feeAmmAbi,
+    typeof Abis.feeAmm,
     'RebalanceSwap',
     { IndexedOnly: false; Required: true }
   >
@@ -1036,12 +1048,12 @@ export declare namespace watchRebalanceSwap {
     bigint,
     number,
     false,
-    ExtractAbiItem<typeof feeAmmAbi, 'RebalanceSwap'>,
+    ExtractAbiItem<typeof Abis.feeAmm, 'RebalanceSwap'>,
     true
   >
 
   export type Parameters = UnionOmit<
-    WatchContractEventParameters<typeof feeAmmAbi, 'RebalanceSwap', true>,
+    WatchContractEventParameters<typeof Abis.feeAmm, 'RebalanceSwap', true>,
     'abi' | 'address' | 'batch' | 'eventName' | 'onLogs' | 'strict'
   > & {
     /** Callback to invoke when a rebalance swap occurs. */
@@ -1088,8 +1100,8 @@ export function watchFeeSwap<
   const { onFeeSwap, userToken, validatorToken, ...rest } = parameters
   return watchContractEvent(client, {
     ...rest,
-    address: feeManagerAddress,
-    abi: feeAmmAbi,
+    address: Addresses.feeManager,
+    abi: Abis.feeAmm,
     eventName: 'FeeSwap',
     args:
       userToken !== undefined && validatorToken !== undefined
@@ -1107,7 +1119,7 @@ export function watchFeeSwap<
 
 export declare namespace watchFeeSwap {
   export type Args = GetEventArgs<
-    typeof feeAmmAbi,
+    typeof Abis.feeAmm,
     'FeeSwap',
     { IndexedOnly: false; Required: true }
   >
@@ -1116,12 +1128,12 @@ export declare namespace watchFeeSwap {
     bigint,
     number,
     false,
-    ExtractAbiItem<typeof feeAmmAbi, 'FeeSwap'>,
+    ExtractAbiItem<typeof Abis.feeAmm, 'FeeSwap'>,
     true
   >
 
   export type Parameters = UnionOmit<
-    WatchContractEventParameters<typeof feeAmmAbi, 'FeeSwap', true>,
+    WatchContractEventParameters<typeof Abis.feeAmm, 'FeeSwap', true>,
     'abi' | 'address' | 'batch' | 'eventName' | 'onLogs' | 'strict'
   > & {
     /** Callback to invoke when a fee swap occurs. */
@@ -1165,8 +1177,8 @@ export function watchMint<
   const { onMint, sender, userToken, validatorToken, ...rest } = parameters
   return watchContractEvent(client, {
     ...rest,
-    address: feeManagerAddress,
-    abi: feeAmmAbi,
+    address: Addresses.feeManager,
+    abi: Abis.feeAmm,
     eventName: 'Mint',
     args: {
       ...(sender !== undefined && {
@@ -1219,12 +1231,12 @@ export declare namespace watchMint {
     bigint,
     number,
     false,
-    ExtractAbiItem<typeof feeAmmAbi, 'Mint'>,
+    ExtractAbiItem<typeof Abis.feeAmm, 'Mint'>,
     true
   >
 
   export type Parameters = UnionOmit<
-    WatchContractEventParameters<typeof feeAmmAbi, 'Mint', true>,
+    WatchContractEventParameters<typeof Abis.feeAmm, 'Mint', true>,
     'abi' | 'address' | 'batch' | 'eventName' | 'onLogs' | 'strict'
   > & {
     /** Callback to invoke when liquidity is added. */
@@ -1270,8 +1282,8 @@ export function watchBurn<
   const { onBurn, userToken, validatorToken, ...rest } = parameters
   return watchContractEvent(client, {
     ...rest,
-    address: feeManagerAddress,
-    abi: feeAmmAbi,
+    address: Addresses.feeManager,
+    abi: Abis.feeAmm,
     eventName: 'Burn',
     args:
       userToken !== undefined && validatorToken !== undefined
@@ -1289,7 +1301,7 @@ export function watchBurn<
 
 export declare namespace watchBurn {
   export type Args = GetEventArgs<
-    typeof feeAmmAbi,
+    typeof Abis.feeAmm,
     'Burn',
     { IndexedOnly: false; Required: true }
   >
@@ -1298,12 +1310,12 @@ export declare namespace watchBurn {
     bigint,
     number,
     false,
-    ExtractAbiItem<typeof feeAmmAbi, 'Burn'>,
+    ExtractAbiItem<typeof Abis.feeAmm, 'Burn'>,
     true
   >
 
   export type Parameters = UnionOmit<
-    WatchContractEventParameters<typeof feeAmmAbi, 'Burn', true>,
+    WatchContractEventParameters<typeof Abis.feeAmm, 'Burn', true>,
     'abi' | 'address' | 'batch' | 'eventName' | 'onLogs' | 'strict'
   > & {
     /** Callback to invoke when liquidity is removed. */