blitzkit/scripts/buildTankopedia.ts

import { Octokit } from '@octokit/rest';
import decompress from 'decompress';
import { config } from 'dotenv';
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';

config();

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',
  flags: 'assets/Data/Gfx/Lobby/flags',
};
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.warn(`Missing name for ${vehicle}`);
            }

            tankIds.push(id);
            tankopedia[id] = {
              id,
              name,
              name_short,
              nation,
              tree_type: isCollector
                ? 'collector'
                : isPremium
                  ? 'premium'
                  : 'researchable',
              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[];
    const flagChanges = await readdir(
      `${TEMP}/${directoriesOfInterest.flags}`,
    ).then((flags) =>
      Promise.all(
        flags
          .filter((flag) => flag.startsWith('flag_tutor-tank_'))
          .map(async (flag) => {
            const content = await readFile(
              `${TEMP}/${directoriesOfInterest.flags}/${flag}`,
              { encoding: 'base64' },
            );
            const name = flag.match(/flag_tutor-tank_(.+)\.packed\.webp/)![1];

            return {
              content,
              encoding: 'base64',
              path: `flags/scratched/${name}.webp`,
            } satisfies FileChange;
          }),
      ),
    );

    console.log(flagChanges);

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