improve lighting, fog, clouds, force fields

scripts/screenshot.ts

@@ -0,0 +1,114 @@
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import puppeteer, { type KeyInput } from "puppeteer";
+import { parseArgs } from "node:util";
+
+function sleep(ms: number) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+const { values, positionals } = parseArgs({
+  options: {
+    camera: {
+      type: "string",
+      default: "1",
+    },
+    debug: {
+      type: "boolean",
+      default: false,
+    },
+    "no-fog": {
+      type: "boolean",
+      default: false,
+    },
+  },
+  allowPositionals: true,
+});
+
+const missionName = positionals[0];
+const cameraNumber = parseInt(values.camera, 10);
+const debugMode = values.debug;
+const fogEnabled = !values["no-fog"];
+
+if (!missionName) {
+  console.error(
+    "Usage: npx tsx scripts/screenshot.ts <missionName> [cameraNumber]",
+  );
+  console.error("Example: npx tsx scripts/screenshot.ts TWL2_WoodyMyrk 1");
+  process.exit(1);
+}
+
+const cameraKey = String(cameraNumber) as KeyInput;
+const outputType = "png";
+const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), "t2-screenshot-"));
+const outputPath = path.join(
+  tempDir,
+  `${missionName}.${cameraNumber}.${debugMode ? "debug." : ""}${outputType}`,
+);
+
+const browser = await puppeteer.launch({ headless: true });
+const page = await browser.newPage();
+await page.setViewport({ width: 900, height: 600 });
+
+const baseUrl = `http://localhost:3000/t2-mapper/?mission=${encodeURIComponent(missionName)}`;
+
+await page.evaluateOnNewDocument(
+  (debugMode, fogEnabled) => {
+    localStorage.setItem(
+      "settings",
+      JSON.stringify({
+        fov: 80,
+        audioEnabled: false,
+        animationEnabled: false,
+        debugMode,
+        fogEnabled,
+      }),
+    );
+  },
+  debugMode,
+  fogEnabled,
+);
+
+console.log(`Loading: ${baseUrl}`);
+await page.goto(baseUrl, { waitUntil: "load" });
+await page.waitForNetworkIdle({ idleTime: 500 });
+
+const mapViewer = await page.waitForSelector("canvas");
+if (!mapViewer) {
+  console.error("Could not find canvas element");
+  process.exit(1);
+}
+await sleep(50);
+
+// Close the popover by pressing Escape
+await page.keyboard.press("Escape");
+await sleep(50);
+
+// Hide controls from screenshots while keeping them selectable
+await page.$eval("#controls", (el: HTMLElement) => {
+  el.style.visibility = "hidden";
+});
+
+// Wait for mission to load
+await page.waitForSelector("#loadingIndicator", { hidden: true });
+await sleep(500);
+
+// Select the camera
+console.log(`Selecting camera: ${cameraNumber}`);
+await mapViewer.press(cameraKey);
+await page.waitForNetworkIdle({ idleTime: 250 });
+await sleep(100);
+
+// Take screenshot
+await mapViewer.screenshot({
+  path: outputPath,
+  type: outputType,
+});
+
+console.log(`Screenshot saved to: ${outputPath}`);
+
+await Promise.race([
+  browser.close(),
+  sleep(3000).then(() => browser.process()?.kill("SIGKILL")),
+]);