blitzkit/scripts/buildTankopedia.ts

import { Octokit } from '@octokit/rest';
import decompress from 'decompress';
import { writeFileSync } from 'fs';
import { cp, mkdir, readFile, readdir, rm, writeFile } from 'fs/promises';
import { load } from 'js-yaml';
import { argv, env } from 'process';
import { ElementCompact, xml2js } from 'xml-js';
import { parse } from 'yaml';
import { NATION_IDS } from '../src/constants/nations';
import { BlitzkriegTankopedia, Tier } from '../src/core/blitzkrieg/tankopedia';
import commitMultipleFiles, { FileChange } from './commitMultipleFiles';

const LANGUAGE = 'en';
const TEMP = 'temp/blitz';

const publish = argv.includes('--publish');
const apkFile = argv
  .find((argument) => argument.startsWith('--file'))
  ?.split('=')[1];
const noRewrite = argv.includes('--no-rewrite');

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

const directoriesOfInterest = {
  vehicleDefinitions: 'assets/Data/XML/item_defs/vehicles',
  strings: 'assets/Data/Strings',
  tankParameters: 'assets/Data/3d/Tanks/Parameters',
  smallIcons: 'assets/Data/Gfx/UI/BattleScreenHUD/SmallTankIcons',
  bigIcons: 'assets/Data/Gfx/UI/BigTankIcons',
};
const directoriesOfInterestArray = Object.values(directoriesOfInterest);

console.log('Removing old tankopedia files...');
await rm('dist/tankopedia', { recursive: true, force: true });
if (!noRewrite) {
  console.log('Removing old temp files...');
  await rm(TEMP, { recursive: true, force: true });

  console.log('Extracting Blitz files...');
  await decompress(apkFile, TEMP, {
    filter: (file) =>
      directoriesOfInterestArray.some((dir) => file.path.startsWith(dir)),
  });
}

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

const strings = await readFile(
  `${TEMP}/${directoriesOfInterest.strings}/${LANGUAGE}.yaml`,
  'utf-8',
).then((data) => load(data) as Record<string, string>);
const tankopedia: BlitzkriegTankopedia = {};
const tankIds: number[] = [];
const images: Record<number, { big: string; small: string }> = {};
const nations = await readdir(
  `${TEMP}/${directoriesOfInterest.vehicleDefinitions}`,
);

writeFileSync('test.json', JSON.stringify(strings, null, 2));

Promise.all(
  nations
    .filter((nation) => nation !== 'common')
    .map(async (nation) => {
      const listXML = await readFile(
        `${TEMP}/${directoriesOfInterest.vehicleDefinitions}/${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;
            const name = strings[vehicleData.userString._text];
            const name_short = strings[vehicleData.shortUserString?._text];
            const isPremium = 'gold' in vehicleData.price;
            const isCollector = vehicleData.sellPrice
              ? 'gold' in vehicleData.sellPrice
              : false;
            const tier = parseInt(vehicleData.level._text) as Tier;
            const tags = vehicleData.tags._text.split(' ');
            const type = tags[0];

            if ((name ?? name_short) === undefined) {
              console.log(vehicleData.userString._text);
            }

            tankIds.push(id);
            tankopedia[id] = {
              id,
              name,
              name_short,
              nation,
              tree_type: isCollector
                ? 'collector'
                : isPremium
                ? 'premium'
                : 'tech-tree',
              tier,
              type,
            };

            // Tank images
            const parameterPath = `${TEMP}/${directoriesOfInterest.tankParameters}/${nation}/${vehicle}.yaml`;
            const parameterString = await readFile(parameterPath, 'utf-8');
            const { smallIconPath, bigIconPath } =
              parse(parameterString).resourcesPath;
            images[id] = {
              small: `${TEMP}/assets/Data/${(smallIconPath as string)
                .replace(/~res:\//, '')
                .replace(/\..+/, '')}.packed.webp`,
              big: `${TEMP}/assets/Data/${(bigIconPath as string)
                .replace(/~res:\//, '')
                .replace(/\..+/, '')}.packed.webp`,
            };

            if (!publish) {
              cp(images[id].small, `dist/tankopedia/icons/small/${id}.webp`);
              cp(images[id].big, `dist/tankopedia/icons/big/${id}.webp`);
            }
          },
        ),
      );
    }),
).then(async () => {
  if (publish) {
    const octokit = new Octokit({ auth: env.GH_TOKEN });
    const tankopediaChange: FileChange = {
      content: JSON.stringify(tankopedia),
      path: 'tankopedia.json',
      encoding: 'utf-8',
    };
    const iconsChange = (
      await Promise.all(
        tankIds.map(async (id) => {
          try {
            const [contentSmall, contentBig] = await Promise.all([
              readFile(images[id].small, { encoding: 'base64' }),
              readFile(images[id].big, { encoding: 'base64' }),
            ]);

            return [
              {
                content: contentBig,
                path: `icons/big/${id}.webp`,
                encoding: 'base64',
              },
              {
                content: contentSmall,
                path: `icons/small/${id}.webp`,
                encoding: 'base64',
              },
            ];
          } catch (error) {
            console.log(error);
            return undefined;
          }
        }),
      )
    ).flat() as FileChange[];

    console.log('Writing files...');
    await commitMultipleFiles(
      octokit,
      'tresabhi',
      'blitzkrieg-assets',
      'main',
      new Date().toString(),
      [tankopediaChange, ...iconsChange],
      true,
    );
  } else {
    writeFile(
      'dist/tankopedia/tankopedia.json',
      JSON.stringify(tankopedia, null, 2),
    );
  }
});