blitzkit/packages/bot/src/core/streams/wotbreplay.ts

import { ReadStream } from '@blitzkit/core';
import AdmZip from 'adm-zip';
import { readFile, writeFile } from 'fs/promises';
import { Parser } from 'pickleparser';
import protobuf from 'protobufjs';

interface WotbReplayMetaJson {
  version: string;
  title: string;
  dbid: string;
  playerName: string;
  battleStartTime: string;
  playerVehicleName: string;
  mapName: string;
  arenaUniqueId: string;
  battleDuration: number;
  vehicleCompDescriptor: number;
  camouflageId: number;
  mapId: number;
  arenaBonusType: number;
}

export class WotbReplayReadStream extends ReadStream {
  async wotbReplay() {
    const pickleParser = new Parser();
    const zip = new AdmZip(Buffer.from(this.buffer));
    const metaJsonBuffer = zip.getEntry('meta.json')?.getData();
    const battleResultsDat = zip.getEntry('battle_results.dat')?.getData();

    if (!metaJsonBuffer || !battleResultsDat)
      throw new SyntaxError('meta.json not found');

    const metaJson = JSON.parse(
      metaJsonBuffer.toString(),
    ) as WotbReplayMetaJson;
    const unpickledBattleResults = pickleParser.parse(battleResultsDat);
    const [, battleResultsString] = unpickledBattleResults as [number, string];
    const root = await protobuf.load(
      `${__dirname}/../../protos/BattleResultsClient.proto`,
    );
    const BattleResults = root.lookupType(
      'BattleResultsGenerated.BattleResultsClient',
    );

    writeFile('test.dat', Buffer.from(battleResultsString));

    const reader = protobuf.Reader.create(Buffer.from(battleResultsString));

    // const decodedBattleResults = BattleResults.decodeDelimited(
    //   Buffer.from(battleResultsString),
    // );
    // console.log(decodedBattleResults.toJSON());
  }
}

const replay = await readFile('temp/test3.wotbreplay');
const stream = new WotbReplayReadStream(replay.buffer as ArrayBuffer);

stream.wotbReplay();