@steerprotocol/sdk 1.31.0 → 2.0.0

package/src/base/FeeManagerClient.ts

@@ -1,711 +0,0 @@
-import {
-  type Address,
-  type Hash,
-  type PublicClient,
-  type WalletClient,
-  isAddressEqual,
-} from 'viem';
-import type {
-  DefaultFeeManagerConfigParams,
-  Fee,
-  FeeManagerConfigParams,
-  FeeWithdrawalParams,
-  MultipleFeeWithdrawalParams,
-  MultiVaultFeeWithdrawalParams,
-  SteerResponse,
-} from '../types.js';
-import { FEE_MANAGER_ABI } from '../const/abis/feeManager.js';
-import { VAULT_FEES_ABI } from '../const/abis/vaultFees.js';
-import { SubgraphClient } from './SubgraphClient.js';
-
-export class FeeManagerClient extends SubgraphClient {
-  protected readonly publicClient: PublicClient;
-  protected readonly walletClient: WalletClient;
-  protected feeManagerAddress: Address;
-
-  public constructor(
-    feeManagerAddress: Address,
-    publicClient: PublicClient,
-    walletClient: WalletClient
-  ) {
-    super(publicClient as PublicClient & WalletClient);
-    this.publicClient = publicClient;
-    this.walletClient = walletClient;
-    this.feeManagerAddress = feeManagerAddress;
-  }
-
-  /**
-   * Overrides the FeeManager contract address.
-   * Useful when deployment addresses are not yet hardcoded or need custom targeting.
-   * @param address The new FeeManager contract address
-   */
-  public setFeeManagerAddress(address: Address): void {
-    this.feeManagerAddress = address;
-  }
-
-  public async getFees(vault: Address): Promise<SteerResponse<Fee[]>> {
-    try {
-      const fees = await this.publicClient.readContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'getFees',
-        args: [vault],
-      });
-
-      return {
-        data: fees as Fee[],
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async getTotalFees(vault: Address): Promise<SteerResponse<bigint>> {
-    try {
-      const totalFees = await this.publicClient.readContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'vaultTotalFees',
-        args: [vault],
-      });
-
-      return {
-        data: totalFees as bigint,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async getWithdrawer(
-    vault: Address,
-    feeIdentifier: string
-  ): Promise<SteerResponse<Address>> {
-    try {
-      const withdrawer = await this.publicClient.readContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'withdrawalPermissions',
-        args: [vault, feeIdentifier],
-      });
-
-      return {
-        data: withdrawer as Address,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async validateAdminPermission(): Promise<SteerResponse<boolean>> {
-    try {
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const owner = await this.publicClient.readContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'owner',
-      });
-
-      return {
-        data: isAddressEqual(owner as Address, accounts[0]),
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async validateWithdrawerPermission(
-    vault: Address,
-    feeIdentifier: string
-  ): Promise<SteerResponse<boolean>> {
-    try {
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const withdrawer = await this.publicClient.readContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'withdrawalPermissions',
-        args: [vault, feeIdentifier],
-      });
-
-      return {
-        data: isAddressEqual(withdrawer as Address, accounts[0]),
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async setFeeAndWithdrawalPermission(
-    params: FeeManagerConfigParams
-  ): Promise<SteerResponse<Hash>> {
-    try {
-      if (
-        params.feeIdentifiers.length !== params.feeValues.length ||
-        params.feeIdentifiers.length !== params.withdrawers.length
-      ) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'feeIdentifiers, feeValues, and withdrawers must have the same length',
-        };
-      }
-
-      const isAdmin = await this.validateAdminPermission();
-      if (!isAdmin.success) {
-        throw new Error(isAdmin.error ?? 'Unknown error occurred');
-      }
-      if (!isAdmin.data) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'Caller is not FeeManager owner',
-        };
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'setFeeAndWithdrawalPermission',
-        args: [params.vault, params.feeIdentifiers, params.feeValues, params.withdrawers],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async setDefaultFeeAndWithdrawalPermission(
-    params: DefaultFeeManagerConfigParams
-  ): Promise<SteerResponse<Hash>> {
-    try {
-      if (
-        params.feeIdentifiers.length !== params.feeValues.length ||
-        params.feeIdentifiers.length !== params.withdrawers.length
-      ) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'feeIdentifiers, feeValues, and withdrawers must have the same length',
-        };
-      }
-
-      const isAdmin = await this.validateAdminPermission();
-      if (!isAdmin.success) {
-        throw new Error(isAdmin.error ?? 'Unknown error occurred');
-      }
-      if (!isAdmin.data) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'Caller is not FeeManager owner',
-        };
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'setDefaultFeeAndWithdrawalPermission',
-        args: [
-          params.vault,
-          params.totalVaultFees,
-          params.feeIdentifiers,
-          params.feeValues,
-          params.withdrawers,
-        ],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async setMigratedVaultFeeAndWithdrawalPermission(): Promise<SteerResponse<Hash>> {
-    try {
-      const isAdmin = await this.validateAdminPermission();
-      if (!isAdmin.success) {
-        throw new Error(isAdmin.error ?? 'Unknown error occurred');
-      }
-      if (!isAdmin.data) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'Caller is not FeeManager owner',
-        };
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'setMigratedVaultFeeAndWithdrawalPermission',
-        args: [],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async setTotalFees(vault: Address, newTotalFees: bigint): Promise<SteerResponse<Hash>> {
-    try {
-      const isAdmin = await this.validateAdminPermission();
-      if (!isAdmin.success) {
-        throw new Error(isAdmin.error ?? 'Unknown error occurred');
-      }
-      if (!isAdmin.data) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'Caller is not FeeManager owner',
-        };
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'setTotalFees',
-        args: [vault, newTotalFees],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async changeFeeWithdrawer(
-    vault: Address,
-    feeIdentifier: string,
-    newWithdrawer: Address
-  ): Promise<SteerResponse<Hash>> {
-    try {
-      const isAdmin = await this.validateAdminPermission();
-      if (!isAdmin.success) {
-        throw new Error(isAdmin.error ?? 'Unknown error occurred');
-      }
-      if (!isAdmin.data) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'Caller is not FeeManager owner',
-        };
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'changeFeeWithdrawer',
-        args: [vault, feeIdentifier, newWithdrawer],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async withdrawFee(params: FeeWithdrawalParams): Promise<SteerResponse<Hash>> {
-    try {
-      const canWithdraw = await this.validateWithdrawerPermission(
-        params.vault,
-        params.feeIdentifier
-      );
-      if (!canWithdraw.success) {
-        throw new Error(canWithdraw.error ?? 'Unknown error occurred');
-      }
-      if (!canWithdraw.data) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'Caller is not authorized withdrawer',
-        };
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'withdrawFee',
-        args: [params.vault, params.feeIdentifier, params.amount0, params.amount1],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async withdrawMultipleFees(
-    params: MultipleFeeWithdrawalParams
-  ): Promise<SteerResponse<Hash>> {
-    try {
-      if (
-        params.feeIdentifiers.length !== params.amounts0.length ||
-        params.feeIdentifiers.length !== params.amounts1.length
-      ) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'feeIdentifiers, amounts0, and amounts1 must have the same length',
-        };
-      }
-
-      for (const feeIdentifier of params.feeIdentifiers) {
-        const canWithdraw = await this.validateWithdrawerPermission(params.vault, feeIdentifier);
-        if (!canWithdraw.success) {
-          throw new Error(canWithdraw.error ?? 'Unknown error occurred');
-        }
-        if (!canWithdraw.data) {
-          return {
-            data: null,
-            status: 400,
-            success: false,
-            error: `Caller is not authorized withdrawer for ${feeIdentifier}`,
-          };
-        }
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'withdrawMultipleFees',
-        args: [params.vault, params.feeIdentifiers, params.amounts0, params.amounts1],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  public async withdrawFeesFromMultipleVaults(
-    params: MultiVaultFeeWithdrawalParams
-  ): Promise<SteerResponse<Hash>> {
-    try {
-      if (
-        params.vaults.length !== params.feeIdentifiers.length ||
-        params.vaults.length !== params.amounts0.length ||
-        params.vaults.length !== params.amounts1.length
-      ) {
-        return {
-          data: null,
-          status: 400,
-          success: false,
-          error: 'vaults, feeIdentifiers, amounts0, and amounts1 must have the same length',
-        };
-      }
-
-      for (let i = 0; i < params.vaults.length; i++) {
-        const vault = params.vaults[i];
-        const identifiers = params.feeIdentifiers[i];
-        const amount0 = params.amounts0[i];
-        const amount1 = params.amounts1[i];
-
-        if (identifiers.length !== amount0.length || identifiers.length !== amount1.length) {
-          return {
-            data: null,
-            status: 400,
-            success: false,
-            error: 'feeIdentifiers[i], amounts0[i], and amounts1[i] must have the same length',
-          };
-        }
-
-        for (const feeIdentifier of identifiers) {
-          const canWithdraw = await this.validateWithdrawerPermission(vault, feeIdentifier);
-          if (!canWithdraw.success) {
-            throw new Error(canWithdraw.error ?? 'Unknown error occurred');
-          }
-          if (!canWithdraw.data) {
-            return {
-              data: null,
-              status: 400,
-              success: false,
-              error: `Caller is not authorized withdrawer for ${feeIdentifier}`,
-            };
-          }
-        }
-      }
-
-      const accounts = await this.walletClient.getAddresses();
-      if (!accounts[0]) {
-        throw new Error('No account found');
-      }
-
-      const { request } = await this.publicClient.simulateContract({
-        address: this.feeManagerAddress,
-        abi: FEE_MANAGER_ABI,
-        functionName: 'withdrawFeesFromMultipleVaults',
-        args: [params.vaults, params.feeIdentifiers, params.amounts0, params.amounts1],
-        account: accounts[0],
-      });
-
-      const hash = await this.walletClient.writeContract(request);
-
-      return {
-        data: hash,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  /**
-   * Gets the accrued revenue for a specific fee identifier in a vault
-   * @param vault The vault address
-   * @param feeIdentifier The fee identifier (e.g., 'STEER_FEES', 'STRATEGIST_FEES')
-   * @returns Promise resolving to the accrued amounts for token0 and token1
-   */
-  public async getAccruedRevenue(
-    vault: Address,
-    feeIdentifier: string
-  ): Promise<SteerResponse<{ amount0: bigint; amount1: bigint }>> {
-    try {
-      const [amount0, amount1] = await Promise.all([
-        this.publicClient.readContract({
-          address: vault,
-          abi: VAULT_FEES_ABI,
-          functionName: 'accruedFees0',
-          args: [feeIdentifier],
-        }),
-        this.publicClient.readContract({
-          address: vault,
-          abi: VAULT_FEES_ABI,
-          functionName: 'accruedFees1',
-          args: [feeIdentifier],
-        }),
-      ]);
-
-      return {
-        data: { amount0, amount1 },
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-
-  /**
-   * Gets a summary of accrued revenue for all configured fee identifiers in a vault
-   * @param vault The vault address
-   * @returns Promise resolving to a list of fee identifiers and their accrued amounts
-   */
-  public async getVaultRevenueSummary(
-    vault: Address
-  ): Promise<SteerResponse<{ feeIdentifier: string; amount0: bigint; amount1: bigint }[]>> {
-    try {
-      const feesResponse = await this.getFees(vault);
-      if (!feesResponse.success || !feesResponse.data) {
-        throw new Error(feesResponse.error || 'Failed to fetch vault fees configuration');
-      }
-
-      const summary = await Promise.all(
-        feesResponse.data.map(async (fee) => {
-          const revenue = await this.getAccruedRevenue(vault, fee.feeIdentifier);
-          return {
-            feeIdentifier: fee.feeIdentifier,
-            amount0: revenue.data?.amount0 ?? 0n,
-            amount1: revenue.data?.amount1 ?? 0n,
-          };
-        })
-      );
-
-      return {
-        data: summary,
-        status: 200,
-        success: true,
-      };
-    } catch (error) {
-      return {
-        data: null,
-        status: 500,
-        success: false,
-        error: error instanceof Error ? error.message : 'Unknown error occurred',
-      };
-    }
-  }
-}