index.ts

import { Address } from "viem";
import axios, { AxiosInstance } from "axios";

import {
  AutomationSubscriptionLimits,
  PreComputedAddressData,
  TaskStatusData,
  TransferCalldataResponse,
  PrecomputeResponse,
} from "./types";

const routes = {
  // Public Deployer Routes
  fetchPreComputeData: "/deployer/public-strategy/precompute",
  fetchDeployerSignature: "/deployer/public-strategy/signature",
  deployPublicStrategy: "/deployer/public-strategy/deploy",
  fetchTaskStatus: "/relayer/tasks/status",
};

export class PublicDeployer {
  private readonly axiosInstance: AxiosInstance;

  constructor(apiKey: string, baseURL: string) {
    this.axiosInstance = axios.create({
      baseURL,
      headers: {
        "x-api-key": apiKey,
      },
    });
  }

  /**
   * Fetches the precomputed address data for a given owner, chain ID, and fee token.
   * @param {Address} owner - The address of the owner.
   * @param {number} chainId - Chain ID for connected network.
   * @param {Address} feeToken - The address of the fee token.
   * @returns {Promise<PreComputedAddressData | null>} The precomputed address data or null if an error occurs.
   */
  async fetchPreComputeData(
    owner: Address,
    chainId: number,
    feeToken: Address
  ): Promise<PreComputedAddressData | null> {
    const payload = { owner, chainID: chainId, feeToken };

    try {
      const response = await this.axiosInstance.post(
        routes.fetchPreComputeData,
        payload
      );
      return response.data.data;
    } catch (err) {
      console.error("Error fetching precompute address:", err);
      return null;
    }
  }

  /**
   * Generates an automation sub-account for a given set of parameters.
   * @param {Address} owner - The address of the owner.
   * @param {Address} precomputedConsoleAddress - The precomputed account address (from /pre-compute response)
   * @param {number} chainID - Chain ID for connected network.
   * @param {string} registryID - The registry ID. (Hardcoded)
   * @param {Address} feeToken - The address of the fee token. (from /pre-compute response)
   * @param {string} feeEstimate - The estimated fee.
   * @param {Address[]} tokens - The list of token addresses.
   * @param {string[]} amounts - The list of token amounts.
   * @param {AutomationSubscriptionLimits} automationSubscriptionLimits - The subscription limits.
   * @param {Record<string, unknown>} metadata - Additional metadata.
   * @returns {Promise<TransferCalldataResponse | null>} The transfer calldata response or null if an error occurs.
   */
  async generateAutomationSubAccount(
    owner: Address,
    precomputedConsoleAddress: Address,
    chainID: number,
    registryID: string,
    feeToken: Address,
    feeEstimate: string,
    tokens: Address[],
    amounts: string[],
    automationSubscriptionLimits: AutomationSubscriptionLimits,
    metadata: Record<string, unknown>
  ): Promise<TransferCalldataResponse | null> {
    const payload = {
      owner,
      precomputedConsoleAddress,
      automationSubscriptionLimits,
      chainID,
      registryID,
      feeToken,
      feeEstimate,
      tokens,
      amounts,
      metadata,
    };
    try {
      const response = await this.axiosInstance.post(
        routes.fetchDeployerSignature,
        payload
      );
      return response.data.data;
    } catch (err) {
      console.error("Error generating automation sub-account:", err);
      return null;
    }
  }

  /**
   * Computes the deployment addresses for account and sub-account setup.
   * @param {Address} owner - The address of the owner.
   * @param {number} chainID - Chain ID for connected network.
   * @param {string} registryID - The registry ID (Hardcoded).
   * @param {string} subscriptionDraftID - The subscription draft ID (from /signature response).
   * @param {string} subAccountPolicyCommit - The sub-account policy commit (from /signature response).
   * @param {Address} feeToken - The address of the fee token.
   * @param {Address[]} tokens - The list of token addresses.
   * @param {string[]} amounts - The list of token amounts.
   * @param {string} subAccountChainerSignature - The sub-account chainer signature (from /signature response).
   * @param {string} feeEstimateSignature - The fee estimate signature (from /pre-compute response).
   * @param {string} feeEstimate - The estimated fee.
   * @param {Record<string, unknown>} metadata - Additional metadata.
   * @returns {Promise<PrecomputeResponse | null>} The precomputed deployment addresses or null if an error occurs.
   */
  async computeDeploymentAddresses(
    owner: Address,
    chainID: number,
    registryID: string,
    subscriptionDraftID: string,
    subAccountPolicyCommit: string,
    feeToken: Address,
    tokens: Address[],
    amounts: string[],
    subAccountChainerSignature: string,
    feeEstimateSignature: string,
    feeEstimate: string,
    metadata: Record<string, unknown>
  ): Promise<PrecomputeResponse | null> {
    const payload = {
      owner,
      chainID,
      registryID,
      subscriptionDraftID,
      subAccountPolicyCommit,
      feeToken,
      tokens,
      amounts,
      subAccountChainerSignature,
      feeEstimateSignature,
      feeEstimate,
      metadata,
      preComputeAddresses: true,
    };

    try {
      const response = await this.axiosInstance.post(
        routes.deployPublicStrategy,
        payload
      );
      return response.data.data;
    } catch (err) {
      console.error("Error deploying account and sub-account:", err);
      return null;
    }
  }

  /**
   * Deploys a account and sub-account with the given parameters.
   * @param {Address} owner - The address of the owner.
   * @param {number} chainID - The ID of the blockchain network.
   * @param {string} registryID - The registry ID. (Hardcoded)
   * @param {string} subscriptionDraftID - The subscription draft ID. (from /signature response)
   * @param {string} subAccountPolicyCommit - The sub-account policy commit. (from /signature response)
   * @param {Address} feeToken - The address of the fee token.
   * @param {Address[]} tokens - The list of token addresses.
   * @param {string[]} amounts - The list of token amounts.
   * @param {string} subAccountChainerSignature - The sub-account chainer signature. (from /signature response)
   * @param {string} feeEstimateSignature - The fee estimate signature. (from /pre-compute response)
   * @param {string} feeEstimate - The estimated fee. (from /pre-compute response)
   * @param {Record<string, unknown>} metadata - Additional metadata.
   * @returns {Promise<{ taskId: string } | null>} The task ID or null if an error occurs.
   */
  async deployBrahmaAccount(
    owner: Address,
    chainID: number,
    registryID: string,
    subscriptionDraftID: string,
    subAccountPolicyCommit: string,
    feeToken: Address,
    tokens: Address[],
    amounts: string[],
    subAccountChainerSignature: string,
    feeEstimateSignature: string,
    feeEstimate: string,
    metadata: Record<string, unknown>
  ): Promise<{ taskId: string } | null> {
    const payload = {
      owner,
      chainID,
      registryID,
      subscriptionDraftID,
      subAccountPolicyCommit,
      feeToken,
      tokens,
      amounts,
      subAccountChainerSignature,
      feeEstimateSignature,
      feeEstimate,
      metadata,
    };

    try {
      const response = await this.axiosInstance.post(
        routes.deployPublicStrategy,
        payload
      );
      return response.data.data;
    } catch (err) {
      console.error("Error deploying account and sub-account:", err);
      return null;
    }
  }

  /**
   * Fetches the status of a task by its ID.
   * @param {string} taskId - The ID of the task. (from /deploy endpoint)
   * @returns {Promise<TaskStatusData>} The task status data.
   * @throws Will throw an error if the request fails.
   */
  async fetchDeploymentStatus(taskId: string): Promise<TaskStatusData> {
    try {
      const response = await this.axiosInstance.get(
        `${routes.fetchTaskStatus}/${taskId}`
      );
      return response.data.data;
    } catch (err) {
      console.error("Error fetching task status:", err);
      throw err;
    }
  }
}