import { isSignature } from "@solana/kit";
import type { Chain, Client, Hex, Transport } from "viem";
import { isHash } from "viem";
import type { SmartAccount } from "viem/account-abstraction";
import type { CombinedIntentRpcSchema } from "../client/intentClient.js";
import { IntentVersionToAddressesMap } from "../config/constants.js";
import type { INTENT_VERSION_TYPE } from "../types/intent.js";

export type UserIntentStatus = "NOT_FOUND" | "PENDING" | "OPENED" | "EXECUTED";

export type Transaction = {
  chainId: number;
  txHash: Hex;
};

export type GetUserIntentStatusParameters = {
  uiHash: Hex;
};

export type GetUserIntentStatusResult = {
  status: UserIntentStatus;
  openTransaction?: Transaction;
  executionTransaction?: Transaction;
};

/**
 * Gets the status of a user intent.
 *
 * @param client - Client to use
 * @param parameters - {@link GetUserIntentStatusParameters}
 * @returns The status of the user intent. {@link GetUserIntentStatusResult}
 *
 * @example
 * import { createIntentClient, http } from '@zerodev/intent'
 * import { mainnet } from 'viem/chains'
 *
 * const client = createIntentClient({
 *   chain: mainnet,
 *   transport: http(),
 * })
 *
 * const status = await client.getUserIntentStatus({ uiHash: "0x..." })
 */
export async function getUserIntentStatus<
  transport extends Transport = Transport,
  chain extends Chain | undefined = Chain | undefined,
  account extends SmartAccount | undefined = SmartAccount | undefined,
>(
  client: Client<transport, chain, account, CombinedIntentRpcSchema>,
  parameters: GetUserIntentStatusParameters,
  version: INTENT_VERSION_TYPE,
): Promise<GetUserIntentStatusResult> {
  // return solana signature if uiHash is a signature
  if (!isHash(parameters.uiHash) && isSignature(parameters.uiHash)) {
    return {
      status: "EXECUTED",
      executionTransaction: {
        chainId: 792703809,
        txHash: parameters.uiHash,
      },
      openTransaction: {
        chainId: 792703809,
        txHash: parameters.uiHash,
      },
    };
  }

  const result = await client.request({
    method: "rl_getUserIntentStatus",
    params: [
      parameters.uiHash,
      IntentVersionToAddressesMap[version].intentExecutorAddress,
    ],
  });

  return result as GetUserIntentStatusResult;
}