import { cp, mkdir, readFile, readdir, rm, writeFile } from 'fs/promises';
import { load } from 'js-yaml';
import { argv } from 'process';
import { ElementCompact, xml2js } from 'xml-js';
import { parse } from 'yaml';
import { NATION_IDS } from '../src/constants/nations';

const VEHICLE_DEFS = 'assets/Data/XML/item_defs/vehicles';
const APK_FILE = argv
  .find((argument) => argument.startsWith('--file'))
  ?.split('=')[1];
const STRINGS = 'assets/Data/Strings';
const LANGUAGE = 'en';
const TANK_PARAMETERS = 'assets/Data/3d/Tanks/Parameters';
const SMALL_ICONS = 'assets/Data/Gfx/UI/BattleScreenHUD/SmallTankIcons';
const BIG_ICONS = 'assets/Data/Gfx/UI/BigTankIcons';
const DIRECTORIES_OF_INTEREST = [
  VEHICLE_DEFS,
  STRINGS,
  TANK_PARAMETERS,
  SMALL_ICONS,
  BIG_ICONS,
];

if (!APK_FILE) throw new Error('No file specified');

// console.log('Removing old Blitz files...');
// await rm('temp/blitz', { recursive: true, force: true });
await rm('dist/tankopedia', { recursive: true, force: true });

// console.log('Extracting Blitz files...');
// await decompress(APK_FILE, 'temp/blitz', {
//   filter: (file) =>
//     DIRECTORIES_OF_INTEREST.some((dir) => file.path.startsWith(dir)),
// });

console.log('Creating new tankopedia folder');
await mkdir('dist/tankopedia');

interface BlitzkriegTankopedia {
  id: number;
  nation: string;
  name: string;
  name_short?: string;
  is_premium: boolean;
  is_collector: boolean;
  tier: number;
  type: string;
}

const strings = await readFile(
  `temp/blitz/${STRINGS}/${LANGUAGE}.yaml`,
  'utf-8',
).then((data) => load(data) as Record<string, string>);
const tankopedia: Record<number, BlitzkriegTankopedia> = {};
const nations = await readdir(`temp/blitz/${VEHICLE_DEFS}`);

Promise.all(
  nations
    .filter((nation) => nation !== 'common')
    .map(async (nation) => {
      const listXML = await readFile(
        `temp/blitz/${VEHICLE_DEFS}/${nation}/list.xml`,
        'utf-8',
      );
      const list = xml2js(listXML, { compact: true }) as ElementCompact;

      await Promise.all(
        Object.entries<ElementCompact>(list.root).map(
          async ([vehicle, vehicleData]) => {
            // Tankopedia
            const nationVehicleId = parseInt(list.root[vehicle].id._text);
            const id = (nationVehicleId << 8) + (NATION_IDS[nation] << 4) + 1;
            // TODO: use userString property instead
            const name = strings[`#${nation}_vehicles:${vehicle}`];
            const name_short = strings[`#${nation}_vehicles:${vehicle}_short`];
            const is_premium = 'gold' in vehicleData.price;
            const is_collector = vehicleData.sellPrice
              ? 'gold' in vehicleData.sellPrice
              : false;
            const tier = parseInt(vehicleData.level._text);
            const tags = vehicleData.tags._text.split(' ');
            const type = tags[0];

            tankopedia[id] = {
              id,
              name,
              name_short,
              nation,
              is_premium,
              is_collector,
              tier,
              type,
            };

            // Tank images
            const parameterPath = `temp/blitz/${TANK_PARAMETERS}/${nation}/${vehicle}.yaml`;
            const parameterString = await readFile(parameterPath, 'utf-8');
            const { smallIconPath, bigIconPath } =
              parse(parameterString).resourcesPath;

            cp(
              `temp/blitz/assets/Data/${(smallIconPath as string)
                .replace(/~res:\//, '')
                .replace(/\..+/, '')}.packed.webp`,
              `dist/tankopedia/icons/small/${id}.webp`,
            );
            cp(
              `temp/blitz/assets/Data/${(bigIconPath as string)
                .replace(/~res:\//, '')
                .replace(/\..+/, '')}.packed.webp`,
              `dist/tankopedia/icons/big/${id}.webp`,
            );
          },
        ),
      );
    }),
).then(() =>
  writeFile(
    'dist/tankopedia/tankopedia.json',
    JSON.stringify(tankopedia, null, 2),
  ),
);