Merge pull request #206 from rabbitholegg/mmackz-utils-balancer

refactor(balancer): use built-in utils

Author: mmackz

.changeset/five-jobs-pump.md

@@ -0,0 +1,5 @@
+---
+"@rabbitholegg/questdk-plugin-balancer": minor
+---
+
+use utils package

packages/balancer/src/Balancer.test.ts

@@ -2,7 +2,7 @@ import { apply, GreaterThanOrEqual } from '@rabbitholegg/questdk/filter'
 import { describe, expect, test } from 'vitest'
 import { zeroAddress, parseEther, parseUnits } from 'viem'
 import { swap, getSupportedTokenAddresses } from './Balancer'
-import { Chains } from './utils'
+import { Chains } from '@rabbitholegg/questdk-plugin-utils'
 import { BALANCER_ABI } from './abi'
 import { VAULT_CONTRACT, CHAIN_ID_ARRAY } from './constants'
 import { failingTestCases, passingTestCases } from './test-transactions'

packages/balancer/src/constants.ts

@@ -1,4 +1,4 @@
-import { Chains } from './utils'
+import { Chains } from '@rabbitholegg/questdk-plugin-utils'
 
 export const VAULT_CONTRACT = '0xba12222222228d8ba445958a75a0704d566bf2c8'
 export const PATCH_CONTRACT = '0x5ff137d4b0fdcd49dca30c7cf57e578a026d2789'

packages/balancer/src/test-transactions.ts

@@ -3,7 +3,11 @@ import {
   GreaterThanOrEqual,
   type SwapActionParams,
 } from '@rabbitholegg/questdk'
-import { type TestParams, createTestCase, Chains } from './utils'
+import {
+  type TestParams,
+  createTestCase,
+  Chains,
+} from '@rabbitholegg/questdk-plugin-utils'
 import { VAULT_CONTRACT } from './constants'
 
 export const POLYGON_SWAP: TestParams<SwapActionParams> = {

packages/balancer/src/token-addresses.ts

@@ -1,5 +1,5 @@
 import { type Address, zeroAddress as ETH_ADDRESS } from 'viem'
-import { Chains } from './utils'
+import { Chains } from '@rabbitholegg/questdk-plugin-utils'
 
 const ethereumTokenAddresses: Address[] = [
   ETH_ADDRESS, // ETH

packages/balancer/src/utils.ts

@@ -1,69 +1,7 @@
-import type { ActionParams, FilterOperator } from '@rabbitholegg/questdk'
-import type { Address, Hash } from 'viem'
+import type { FilterOperator } from '@rabbitholegg/questdk'
+import type { Address } from 'viem'
 import { getAddress } from 'viem'
 
-export enum Chains {
-  ETHEREUM = 1,
-  OPTIMISM = 10,
-  BINANCE_SMART_CHAIN = 56,
-  GNOSIS = 100,
-  POLYGON_POS = 137,
-  ZK_SYNC_ERA = 324,
-  POLYGON_ZK = 1101,
-  MANTLE = 5000,
-  BASE = 8453,
-  ARBITRUM_ONE = 42161,
-  CELO = 42220,
-  AVALANCHE = 43114,
-  LINEA = 59144,
-  SCROLL = 534352,
-}
-
-interface Transaction {
-  chainId: number
-  from: Address
-  hash?: Hash
-  input: string
-  to: Address
-  value: string
-}
-
-export interface TestCase<T extends ActionParams> {
-  transaction: Transaction
-  params: T
-  description: string
-}
-
-export type TestParams<T extends ActionParams> = {
-  transaction: Transaction
-  params: T
-}
-
-/**
- * Creates a test case object for a given action and transaction.
- *
- * This function takes a `TestParams` object that includes both a `Transaction` and
- * `ActionParams`, a description of the test case, and an optional set of overrides
- * for the action parameters. It returns a `TestCase` object that contains the transaction,
- * the combined action parameters with any overrides applied, and the description.
- *
- * @param {TestParams<T>} testParams - An object containing the transaction and action parameters.
- * @param {string} description - A brief description of the test case.
- * @param {Partial<T>} [overrides] - Optional overrides for the action parameters.
- * @returns {TestCase<T>} A test case object with the transaction, params, and description.
- */
-export function createTestCase<T extends ActionParams>(
-  testParams: TestParams<T>,
-  description: string,
-  overrides: Partial<T> = {},
-): TestCase<T> {
-  return {
-    transaction: testParams.transaction,
-    params: { ...testParams.params, ...overrides },
-    description,
-  }
-}
-
 export const buildPathQuery = (tokenIn?: Address, tokenOut?: Address) => {
   const conditions: FilterOperator[] = []