import type {
  ChainType,
  Execution,
  LiFiStep,
  Route,
  Step,
  Token,
  TokenAmount,
} from '@lifi/types';
import type { WalletClient } from 'viem';
import type { ChainId } from '../types/base.js';

export interface SDKProvider {
  readonly type: ChainType;
  isAddress(address: string): boolean;
  resolveAddress(name: string): Promise<string | undefined>;
  getStepExecutor(options: StepExecutorOptions): Promise<StepExecutor>;
  getBalance(walletAddress: string, tokens: Token[]): Promise<TokenAmount[]>;
}

export interface StepExecutorOptions {
  routeId: string;
  executionOptions?: ExecutionOptions;
}

export interface InteractionSettings {
  allowInteraction?: boolean;
  allowUpdates?: boolean;
  allowExecution?: boolean;
}

export interface StepExecutor {
  allowUserInteraction: boolean;
  allowExecution: boolean;
  setInteraction(settings?: InteractionSettings): void;
  executeStep(step: LiFiStepExtended): Promise<LiFiStepExtended>;
}

export interface RouteExtended extends Omit<Route, 'steps'> {
  steps: LiFiStepExtended[];
}

export interface LiFiStepExtended extends LiFiStep {
  execution?: Execution;
}

export type StepExtended = Step & {
  execution?: Execution;
};

export type TransactionParameters = {
  chainId?: number;
  to?: string;
  from?: string;
  nonce?: number;
  data?: string;
  value?: bigint;
  gas?: bigint;
  gasPrice?: bigint;
  maxFeePerGas?: bigint;
  maxPriorityFeePerGas?: bigint;
};

export interface RouteExecutionData {
  route: Route;
  executors: StepExecutor[];
  executionOptions?: ExecutionOptions;
}

export type RouteExecutionDataDictionary = Partial<
  Record<string, RouteExecutionData>
>;

export type RouteExecutionDictionary = Partial<Record<string, Promise<Route>>>;

export type UpdateRouteHook = (updatedRoute: RouteExtended) => void;

export interface TransactionRequestParameters extends TransactionParameters {
  requestType: 'approve' | 'transaction';
}

export type TransactionRequestUpdateHook = (
  updatedTxRequest: TransactionRequestParameters,
) => Promise<TransactionParameters>;

export type SwitchChainHook = (
  chainId: number,
) => Promise<WalletClient | undefined>;

export interface AcceptSlippageUpdateHookParams {
  toToken: Token;
  oldToAmount: string;
  newToAmount: string;
  oldSlippage: number;
  newSlippage: number;
}

export type AcceptSlippageUpdateHook = (
  params: AcceptSlippageUpdateHookParams,
) => Promise<boolean | undefined>;

export interface ExchangeRateUpdateParams {
  toToken: Token;
  oldToAmount: string;
  newToAmount: string;
}

export type AcceptExchangeRateUpdateHook = (
  params: ExchangeRateUpdateParams,
) => Promise<boolean | undefined>;

export interface ExecutionOptions {
  commissionBps?: Partial<Record<ChainId, number>>;
  commissionBpsSDK?: Partial<Record<ChainId, number>>;
  acceptExchangeRateUpdateHook?: AcceptExchangeRateUpdateHook;
  switchChainHook?: SwitchChainHook;
  updateRouteHook?: UpdateRouteHook;
  updateTransactionRequestHook?: TransactionRequestUpdateHook;
  executeInBackground?: boolean;
  infiniteApproval?: boolean;
  slippage?: string;
}

export interface LiFiStepExtended extends LiFiStep {
  execution?: Execution;
}