import { Logger } from "../common";

import { LpTokenInfo, SplTokenInfo } from "./type";

const logger = Logger.from("token/util");

/**
 * Token list
 */
export class TokenList {
  constructor(private tokenList: (SplTokenInfo | LpTokenInfo)[]) {}

  /**
   * Filter token by mint of token list.
   *
   * @param mint - Token's mint address
   */
  filterByMint = (mint: string) => {
    return this.tokenList.filter((token) => token.mint === mint);
  };

  /**
   * Filter unique token by mint of token list, must and can only have one result.
   */
  filterUniqueByMint = <T extends "all" | "spl" | "lp">(mint: string, tokenType: T | "all" | "spl" | "lp" = "all") => {
    const result = this.tokenList.filter((token) => token.mint === mint);

    if (result.length === 0) {
      return logger.throwArgumentError(`No token found`, "mint", mint);
    } else if (result.length > 1) {
      return logger.throwArgumentError(`Multiple tokens found: ${result.length}`, "mint", mint);
    }

    const token = result[0];

    if (tokenType === "spl" && "version" in token) {
      return logger.throwArgumentError("invalid SPL token mint", "mint", mint);
    } else if (tokenType === "lp" && !("version" in token)) {
      return logger.throwArgumentError("invalid LP token mint", "mint", mint);
    }

    return token as T extends "all" ? SplTokenInfo | LpTokenInfo : T extends "spl" ? SplTokenInfo : LpTokenInfo;
  };

  /**
   * Get list of token list
   */
  getList = () => {
    return this.tokenList;
  };
}