支持 W3X

test/indexer.test.mjs

@@ -2,8 +2,12 @@ import { test } from "node:test";
 import assert from "node:assert/strict";
 import { fileURLToPath } from "node:url";
 import { dirname, join } from "node:path";
+import fs from "node:fs";
+import os from "node:os";
 import { ModIndexer } from "../out/indexer/indexer.js";
 import { CachedDirectoryWalker } from "../out/indexer/fileScanner.js";
+import { DocumentCache, ShallowScanCache } from "../out/indexer/caches.js";
+import { resolveReferenceTargetsForType } from "../out/indexer/refs.js";
 
 const root = dirname(dirname(fileURLToPath(import.meta.url)));
 const project = join(root, "test", "fixtures", "minimod");
@@ -70,3 +74,120 @@ test("provides include source candidates", async () => {
   assert.ok(xml.some((c) => c.source === "Includes/Units.xml"));
   assert.ok(xml.some((c) => c.source === "DATA:static.xml"));
 });
+
+test("indexes art-asset XML (.w3x / sniffed .w3d) via shallow scan and skips binary", async () => {
+  const idx = await buildIndex();
+
+  // The .w3x hub chain: Mod.xml -> VehicleArt.xml -> Models/Tank_SKN.w3x.
+  const skn = idx.assetsById.get("tank_skn");
+  assert.ok(skn?.some((d) => d.type === "W3DContainer"), "W3DContainer from .w3x indexed");
+  assert.ok(
+    idx.assetsById.get("tank_skl")?.some((d) => d.type === "W3DHierarchy"),
+    "W3DHierarchy from .w3x indexed",
+  );
+  const container = skn.find((d) => d.type === "W3DContainer");
+  assert.match(container.file, /Tank_SKN\.w3x$/);
+  assert.ok(container.line > 0, "definition line recorded from shallow scan");
+
+  // Unknown extension with XML content is sniffed and indexed.
+  assert.ok(
+    idx.assetsById.get("tank_fp")?.some((d) => d.type === "W3DMesh"),
+    "unknown-extension XML (.w3d) sniffed and indexed",
+  );
+
+  // Binary content is registered as a file but never parsed.
+  assert.equal(idx.assetsById.get("tank_damaged"), undefined, "binary asset never indexed");
+  assert.ok(
+    idx.files.has(
+      join(project, "Data", "Includes", "Models", "Tank_Damaged.dds").toLowerCase(),
+    ),
+    "binary include target registered as a file",
+  );
+
+  // The reported Harbinger scenario: <Model Name="Tank_SKN"/> (refType
+  // BaseRenderAssetType) must resolve to the W3DContainer defined in the w3x.
+  const targets = resolveReferenceTargetsForType(
+    idx,
+    "ScriptedModelDrawModel",
+    "Name",
+    "Tank_SKN",
+  );
+  assert.equal(targets.length, 1);
+  assert.equal(targets[0].def.type, "W3DContainer");
+  assert.match(targets[0].def.file, /Tank_SKN\.w3x$/);
+});
+
+test("w3x files appear in Include source completion candidates", async () => {
+  const idx = await buildIndex();
+  assert.ok(
+    idx.sourceCandidates.some((c) => c.source === "Includes/Models/Tank_SKN.w3x"),
+    "project-relative w3x candidate",
+  );
+  assert.ok(
+    idx.sourceCandidates.some((c) => c.source === "DATA:Includes/Models/Tank_SKN.w3x"),
+    "DATA: w3x candidate",
+  );
+});
+
+test("shallow scans and full parses are cached across rebuilds", async () => {
+  const documentCache = new DocumentCache();
+  const shallowCache = new ShallowScanCache();
+  const makeIndexer = () =>
+    new ModIndexer({
+      projectDir: project,
+      sdkDir: sdk,
+      builtmodsDirs: [join(sdk, "builtmods")],
+      indexSageXml: true,
+      additionalDataSearchPaths: [],
+      walker: new CachedDirectoryWalker(),
+      documentCache,
+      shallowCache,
+    });
+
+  const first = await makeIndexer().build();
+  const second = await makeIndexer().build();
+
+  assert.equal(first.stats.shallowScannedFiles, 2, "w3x + sniffed w3d scanned on first build");
+  assert.equal(second.stats.shallowScannedFiles, 0, "unchanged art assets are not re-scanned");
+  assert.equal(second.stats.shallowCacheHits, 2);
+  assert.equal(second.stats.assetCount, first.stats.assetCount);
+  assert.equal(second.stats.indexedFiles, first.stats.indexedFiles);
+});
+
+test("w3x with a UTF-8 BOM is indexed with correct offsets", async (t) => {
+  const tmp = fs.mkdtempSync(join(os.tmpdir(), "ra3modxml-bom-"));
+  t.after(() => fs.rmSync(tmp, { recursive: true, force: true }));
+  const projectDir = join(tmp, "project");
+  fs.mkdirSync(join(projectDir, "Data", "Models"), { recursive: true });
+  fs.writeFileSync(
+    join(projectDir, "Data", "Mod.xml"),
+    `<?xml version="1.0" encoding="utf-8"?>
+<AssetDeclaration>
+  <Includes>
+    <Include type="all" source="Models/Tank_BOM.w3x"/>
+  </Includes>
+</AssetDeclaration>`,
+    "utf8",
+  );
+  fs.writeFileSync(
+    join(projectDir, "Data", "Models", "Tank_BOM.w3x"),
+    "\uFEFF<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" +
+      "<AssetDeclaration>\n" +
+      "  <W3DContainer id=\"Tank_BOM\" />\n" +
+      "</AssetDeclaration>\n",
+    "utf8",
+  );
+
+  const idx = await new ModIndexer({
+    projectDir,
+    sdkDir: sdk,
+    builtmodsDirs: [join(sdk, "builtmods")],
+    indexSageXml: false,
+    additionalDataSearchPaths: [],
+    walker: new CachedDirectoryWalker(),
+  }).build();
+
+  const def = idx.assetsById.get("tank_bom")?.find((d) => d.type === "W3DContainer");
+  assert.ok(def, "BOM-prefixed w3x asset indexed");
+  assert.equal(def.line, 3, "id line is correct despite the BOM");
+});