import {
  PerpMeta,
  PerpMetaAndAssetCtxs,
  ClearinghouseState,
  UserFunding,
  UserNonFundingLedgerUpdates,
  FundingHistory,
  PredictedFundings,
  PerpsAtOpenInterestCap,
} from '../../types/index.js';
import { HttpApi } from '../../utils/httpApi.js';
import { InfoType } from '../../types/constants.js';
import { SymbolConversion } from '../../utils/symbolConversion.js';
import { Hyperliquid } from '../../index.js';

export class PerpetualsInfoAPI {
  private httpApi: HttpApi;
  private symbolConversion: SymbolConversion;
  private parent: Hyperliquid;

  constructor(httpApi: HttpApi, symbolConversion: SymbolConversion, parent: Hyperliquid) {
    this.httpApi = httpApi;
    this.symbolConversion = symbolConversion;
    this.parent = parent;
  }

  async getMeta(rawResponse: boolean = false): Promise<PerpMeta> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>({ type: InfoType.META });
    return rawResponse
      ? response
      : await this.symbolConversion.convertResponse(response, ['name', 'coin', 'symbol'], 'PERP');
  }

  async getMetaAndAssetCtxs(rawResponse: boolean = false): Promise<PerpMetaAndAssetCtxs> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>({ type: InfoType.PERPS_META_AND_ASSET_CTXS });
    return rawResponse
      ? response
      : await this.symbolConversion.convertResponse(response, ['name', 'coin', 'symbol'], 'PERP');
  }

  async getClearinghouseState(user: string, rawResponse: boolean = false): Promise<ClearinghouseState> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>({ type: InfoType.PERPS_CLEARINGHOUSE_STATE, user: user });
    return rawResponse ? response : await this.symbolConversion.convertResponse(response);
  }

  async getUserFunding(
    user: string,
    startTime: number,
    endTime?: number,
    rawResponse: boolean = false,
  ): Promise<UserFunding> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>(
      {
        type: InfoType.USER_FUNDING,
        user: user,
        startTime: startTime,
        endTime: endTime,
      },
      20,
    );
    return rawResponse ? response : await this.symbolConversion.convertResponse(response);
  }

  async getUserNonFundingLedgerUpdates(
    user: string,
    startTime: number,
    endTime?: number,
    rawResponse: boolean = false,
  ): Promise<UserNonFundingLedgerUpdates> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>(
      {
        type: InfoType.USER_NON_FUNDING_LEDGER_UPDATES,
        user: user,
        startTime: startTime,
        endTime: endTime,
      },
      20,
    );
    return rawResponse ? response : await this.symbolConversion.convertResponse(response);
  }

  async getFundingHistory(
    coin: string,
    startTime: number,
    endTime?: number,
    rawResponse: boolean = false,
  ): Promise<FundingHistory> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>(
      {
        type: InfoType.FUNDING_HISTORY,
        coin: await this.symbolConversion.convertSymbol(coin, 'reverse'),
        startTime: startTime,
        endTime: endTime,
      },
      20,
    );
    return rawResponse ? response : await this.symbolConversion.convertResponse(response);
  }

  async getPredictedFundings(rawResponse: boolean = false): Promise<PredictedFundings> {
    await this.parent.ensureInitialized();
    const response = await this.httpApi.makeRequest<any>(
      {
        type: InfoType.PREDICTED_FUNDINGS,
      },
      20,
    );

    return rawResponse ? response : await this.symbolConversion.convertResponse(response);
  }

  async getPerpsAtOpenInterestCap(rawResponse: boolean = false): Promise<PerpsAtOpenInterestCap> {
    await this.parent.ensureInitialized();
    const response = (await this.httpApi.makeRequest<any>({
      type: InfoType.PERPS_AT_OPEN_INTEREST_CAP,
    })) as string[];

    if (rawResponse) {
      return response;
    }

    // Convert each symbol in the array
    const convertedResponse = await Promise.all(
      response.map((symbol: string) => this.symbolConversion.convertSymbol(symbol, '', 'PERP')),
    );

    return convertedResponse;
  }
}