feat: migration 4

packages/scripts/src/applyPatches.ts

@@ -0,0 +1,78 @@
+import { mkdir, writeFile } from 'fs/promises';
+import { parse as parsePath } from 'path';
+import ProgressBar from 'progress';
+import { parse as parseYaml, stringify as stringifyYaml } from 'yaml';
+import { readStringDVPL } from '../src/core/blitz/readStringDVPL';
+import { readYAMLDVPL } from '../src/core/blitz/readYAMLDVPL';
+import { writeDVPL } from '../src/core/blitz/writeDVPL';
+import { assertSecret } from '../src/core/blitzkit/secret';
+import { dvp } from '../submodules/blitzkit-closed/src/dvp';
+import { DATA } from './buildAssets/constants';
+
+const versionTextFile = await readStringDVPL(`${DATA}/version.txt.dvpl`);
+const currentVersion = versionTextFile
+  .split(' ')[0]
+  .split('.')
+  .slice(0, 3)
+  .join('.');
+
+console.log(`Installing patches for ${currentVersion}...`);
+
+let patchIndex = 1;
+while (true) {
+  const response = await fetch(
+    `${assertSecret(process.env.WOTB_DLC_CDN)}/dlc/s${currentVersion}_${patchIndex}.yaml`,
+  );
+
+  if (response.status === 200) {
+    console.log(`Applying patch ${patchIndex}...`);
+
+    const data = parseYaml(await response.text());
+    const dvpm = await fetch(
+      `${assertSecret(process.env.WOTB_DLC_CDN)}/dlc/${data.dx11}`,
+    ).then((response) => response.arrayBuffer());
+    const dvpd = await fetch(
+      `${assertSecret(process.env.WOTB_DLC_CDN)}/dlc/${data.dx11.replace('.dvpm', '.dvpd')}`,
+    ).then((response) => response.arrayBuffer());
+    const files = await dvp(dvpm, dvpd);
+    const bar = new ProgressBar(
+      `Patching ${files.length} files :bar`,
+      files.length,
+    );
+
+    await Promise.all(
+      files.map(async ({ path, data }) => {
+        const { dir } = parsePath(path);
+
+        await mkdir(`${DATA}/${dir}`, { recursive: true });
+        await writeFile(
+          `${DATA}/${path}.dvpl`,
+          new Uint8Array(writeDVPL(Buffer.from(data))),
+        );
+
+        bar.tick();
+      }),
+    );
+
+    if ('dynamicContentLocalizationsDir' in data) {
+      console.log('Found dynamic content localizations; patching...');
+
+      const localizationsResponse = await fetch(
+        `${assertSecret(process.env.WOTB_DLC_CDN)}/dlc/${data.dynamicContentLocalizationsDir}/en.yaml`,
+      );
+      const newStrings = parseYaml(await localizationsResponse.text());
+      const oldStrings = await readYAMLDVPL<Record<string, string>>(
+        `${DATA}/Strings/en.yaml.dvpl`,
+      );
+      const patchedStrings = { ...oldStrings, ...newStrings };
+      const patchedContent = stringifyYaml(patchedStrings);
+
+      await writeFile(
+        `${DATA}/Strings/en.yaml.dvpl`,
+        new Uint8Array(writeDVPL(Buffer.from(patchedContent))),
+      );
+    }
+
+    patchIndex++;
+  } else break;
+}