import type { Chain, Transport } from "viem";
import type { SmartAccount } from "viem/account-abstraction";
import {
  type EnableIntentResult,
  enableIntent,
} from "../../actions/enableIntent.js";
import {
  type EstimateUserIntentFeesResult,
  estimateUserIntentFees,
} from "../../actions/estimateUserIntentFees.js";
import {
  type GetCABParameters,
  type GetCABResult,
  getCAB,
} from "../../actions/getCAB.js";
import {
  type GetIntentParameters,
  type GetIntentReturnType,
  getIntent,
} from "../../actions/getIntent.js";
import { getUserIntentExecutionReceipt } from "../../actions/getUserIntentExecutionReceipt.js";
import { getUserIntentOpenReceipt } from "../../actions/getUserIntentOpenReceipt.js";
import {
  type GetUserIntentStatusParameters,
  type GetUserIntentStatusResult,
  getUserIntentStatus,
} from "../../actions/getUserIntentStatus.js";
import {
  type PrepareUserIntentParameters,
  type PrepareUserIntentResult,
  prepareUserIntent,
} from "../../actions/prepareUserIntent.js";
import {
  type SendUserIntentParameters,
  type SendUserIntentResult,
  sendUserIntent,
} from "../../actions/sendUserIntent.js";
import type {
  GetUserIntentExecutionReceiptResult,
  GetUserIntentOpenReceiptResult,
  GetUserIntentReceiptParameters,
} from "../../actions/types.js";
import {
  type WaitForUserIntentExecutionReceiptParameters,
  type WaitForUserIntentExecutionReceiptReturnType,
  waitForUserIntentExecutionReceipt,
} from "../../actions/waitForUserIntentExecutionReceipt.js";
import {
  type WaitForUserIntentOpenReceiptParameters,
  type WaitForUserIntentOpenReceiptReturnType,
  waitForUserIntentOpenReceipt,
} from "../../actions/waitForUserIntentOpenReceipt.js";
import type { INTENT_VERSION_TYPE } from "../../types/intent.js";
import type {
  BaseIntentClient,
  CombinedIntentRpcSchema,
} from "../intentClient.js";

export type IntentClientActions<
  chain extends Chain | undefined = Chain | undefined,
  account extends SmartAccount | undefined = SmartAccount | undefined,
> = {
  getIntent: (parameters: GetIntentParameters) => Promise<GetIntentReturnType>;
  getCAB: (parameters: GetCABParameters) => Promise<GetCABResult>;
  prepareUserIntent: <
    accountOverride extends SmartAccount | undefined = undefined,
    calls extends readonly unknown[] = readonly unknown[],
  >(
    parameters: PrepareUserIntentParameters<account, accountOverride, calls>,
  ) => Promise<PrepareUserIntentResult>;
  sendUserIntent: <
    accountOverride extends SmartAccount | undefined = undefined,
    calls extends readonly unknown[] = readonly unknown[],
  >(
    parameters: SendUserIntentParameters<account, accountOverride, calls>,
  ) => Promise<SendUserIntentResult>;
  estimateUserIntentFees: <
    accountOverride extends SmartAccount | undefined = undefined,
    calls extends readonly unknown[] = readonly unknown[],
  >(
    parameters: Parameters<
      typeof estimateUserIntentFees<account, chain, accountOverride, calls>
    >[1],
  ) => Promise<EstimateUserIntentFeesResult>;
  getUserIntentStatus: (
    parameters: GetUserIntentStatusParameters,
  ) => Promise<GetUserIntentStatusResult>;
  getUserIntentOpenReceipt: (
    parameters: GetUserIntentReceiptParameters,
  ) => Promise<GetUserIntentOpenReceiptResult>;
  getUserIntentExecutionReceipt: (
    parameters: GetUserIntentReceiptParameters,
  ) => Promise<GetUserIntentExecutionReceiptResult>;
  waitForUserIntentExecutionReceipt: (
    parameters: WaitForUserIntentExecutionReceiptParameters,
  ) => Promise<WaitForUserIntentExecutionReceiptReturnType>;
  waitForUserIntentOpenReceipt: (
    parameters: WaitForUserIntentOpenReceiptParameters,
  ) => Promise<WaitForUserIntentOpenReceiptReturnType>;
  enableIntent: () => Promise<EnableIntentResult>;
};

export function intentClientActions(
  version: INTENT_VERSION_TYPE,
): <
  transport extends Transport = Transport,
  chain extends Chain | undefined = Chain | undefined,
  account extends SmartAccount | undefined = SmartAccount | undefined,
>(
  client: BaseIntentClient<transport, chain, account, CombinedIntentRpcSchema>,
) => IntentClientActions<chain, account> {
  return (client) => ({
    getIntent: (parameters) => getIntent(client, parameters, version),
    getCAB: (parameters) => getCAB(client, parameters),
    prepareUserIntent: (parameters) =>
      prepareUserIntent(client, parameters, version),
    sendUserIntent: (parameters) => sendUserIntent(client, parameters, version),
    estimateUserIntentFees: (parameters) =>
      estimateUserIntentFees(client, parameters, version),
    getUserIntentStatus: (parameters) =>
      getUserIntentStatus(client, parameters, version),
    getUserIntentOpenReceipt: (parameters) =>
      getUserIntentOpenReceipt(client, parameters, version),
    getUserIntentExecutionReceipt: (parameters) =>
      getUserIntentExecutionReceipt(client, parameters, version),
    waitForUserIntentExecutionReceipt: (parameters) =>
      waitForUserIntentExecutionReceipt(client, parameters, version),
    waitForUserIntentOpenReceipt: (parameters) =>
      waitForUserIntentOpenReceipt(client, parameters, version),
    enableIntent: () => enableIntent(client, version),
  });
}