import { go } from 'fuzzysort';
import { UserError } from '../blitzkrieg/userError';
import { TANKS, tankopedia } from '../blitzstars/tankopedia';

export default async function resolveTankId(tank: string | number) {
  const number = typeof tank === 'string' ? parseInt(tank) : tank;

  if (Number.isNaN(number)) {
    const searchResult = go(`${tank}`, await TANKS, {
      keys: ['name'],
      limit: 1,
    });

    if (searchResult.length === 0) {
      throw new UserError('Tank not found', {
        cause: `Could not find tank by the name "${tank}".`,
      });
    } else {
      return searchResult[0].obj.tank_id;
    }
  } else {
    if ((await tankopedia)[number]) {
      return number;
    } else {
      throw new UserError('Tank not found', {
        cause: `Could not find tank by the ID "${number}".`,
      });
    }
  }
}