blitzkit/packages/scripts/src/buildAssets/tankModels.ts

import { toUniqueId } from '@blitzkit/core';
import { NodeIO } from '@gltf-transform/core';
import { ALL_EXTENSIONS } from '@gltf-transform/extensions';
import { readdir, writeFile } from 'fs/promises';
import { extractModel } from '../core/blitz/extractModel';
import { readXMLDVPL } from '../core/blitz/readXMLDVPL';
import { readYAMLDVPL } from '../core/blitz/readYAMLDVPL';
import { commitAssets } from '../core/github/commitAssets';
import { FileChange } from '../core/github/commitMultipleFiles';
import { DATA } from './constants';
import { VehicleDefinitionList } from './definitions';
import { TankParameters } from './tankIcons';

export async function tankModels() {
  console.log('Building tank models...');

  const nodeIO = new NodeIO().registerExtensions(ALL_EXTENSIONS);
  const nations = await readdir(`${DATA}/XML/item_defs/vehicles`).then(
    (nations) => nations.filter((nation) => nation !== 'common'),
  );

  for (const nationIndex in nations) {
    const nation = nations[nationIndex];
    const changes: FileChange[] = [];
    const tanks = await readXMLDVPL<{ root: VehicleDefinitionList }>(
      `${DATA}/XML/item_defs/vehicles/${nation}/list.xml`,
    );

    console.log(`Building models for ${nation}`);

    for (const [tankKey, tank] of Object.entries(tanks.root)) {
      if (tankKey.includes('tutorial_bot')) continue;

      const id = toUniqueId(nation, tank.id);
      const parameters = await readYAMLDVPL<TankParameters>(
        `${DATA}/3d/Tanks/Parameters/${nation}/${tankKey}.yaml`,
      );
      const model = await extractModel(
        DATA,
        parameters.resourcesPath.blitzModelPath.replace(/\.sc2$/, ''),
      );

      changes.push({
        path: `3d/tanks/models/${id}.glb`,
        content: Buffer.from(await nodeIO.writeBinary(model)),
      });
    }

    await commitAssets(`tank models ${nation}`, changes);
  }
}