Handle MPQ maps without HM3W headers

Author: Frotty

de.peeeq.wurstscript/src/main/java/de/peeeq/wurstio/languageserver/ProjectConfigBuilder.java

@@ -358,18 +358,37 @@ private static void setVolatilePlayerConfig(WurstProjectBuildPlayer wplayer, W3I
 
     private static void applyMapHeader(WurstProjectConfigData projectConfig, File targetMap) throws IOException {
         boolean shouldWrite = false;
-        MapHeader mapHeader = MapHeader.ofFile(targetMap);
-        if (!projectConfig.buildMapData().players().isEmpty()) {
-            mapHeader.setMaxPlayersCount(projectConfig.buildMapData().players().size());
+        WurstProjectBuildMapData buildMapData = projectConfig.buildMapData();
+        if (buildMapData.players().isEmpty() && StringUtils.isBlank(buildMapData.name())) {
+            return;
+        }
+
+        // A Warcraft III map may omit the optional 512-byte HM3W prefix and start
+        // directly with its MPQ archive. MapHeader.ofFile only reads the prefix,
+        // so use a new header in that case; writeToMapFile will insert it before
+        // the archive.
+        MapHeader mapHeader = startsWithMpqArchive(targetMap)
+            ? new MapHeader()
+            : MapHeader.ofFile(targetMap);
+        if (!buildMapData.players().isEmpty()) {
+            mapHeader.setMaxPlayersCount(buildMapData.players().size());
             shouldWrite = true;
         }
-        if (StringUtils.isNotBlank(projectConfig.buildMapData().name())) {
-            mapHeader.setMapName(projectConfig.buildMapData().name());
+        if (StringUtils.isNotBlank(buildMapData.name())) {
+            mapHeader.setMapName(buildMapData.name());
             shouldWrite = true;
         }
         if (shouldWrite) {
             WLogger.info("Applying map header");
             mapHeader.writeToMapFile(targetMap);
         }
     }
+
+    private static boolean startsWithMpqArchive(File targetMap) throws IOException {
+        try (InputStream input = new FileInputStream(targetMap)) {
+            byte[] startToken = input.readNBytes(4);
+            return startToken.length == 4
+                && new String(startToken, StandardCharsets.US_ASCII).startsWith("MPQ");
+        }
+    }
 }

de.peeeq.wurstscript/src/test/java/tests/wurstscript/tests/WurstBuildConfigTests.java

@@ -1,11 +1,13 @@
 package tests.wurstscript.tests;
 
 import org.wurstscript.projectconfig.WurstProjectConfigData;
+import org.wurstscript.projectconfig.WurstProjectBuildMapData;
 import de.peeeq.wurstio.languageserver.WFile;
 import de.peeeq.wurstio.languageserver.ProjectConfigBuilder;
 import de.peeeq.wurstio.languageserver.WurstBuildConfig;
 import de.peeeq.wurstio.languageserver.WurstCommands;
 import de.peeeq.wurstio.utils.W3InstallationData;
+import net.moonlightflower.wc3libs.bin.app.MapHeader;
 import net.moonlightflower.wc3libs.port.GameVersion;
 import org.testng.annotations.Test;
 
@@ -223,6 +225,57 @@ public void configInjectionPrefersPinnedPatchOverDetectedInstallVersion() throws
         );
     }
 
+    @Test
+    public void mapHeaderConfigAcceptsAnArchiveWithoutAnHm3wPrefix() throws Exception {
+        Path mapWithoutHeader = Files.createTempFile("wurst-map-without-header", ".w3x");
+        byte[] original = new byte[1024];
+        original[0] = 'M';
+        original[1] = 'P';
+        original[2] = 'Q';
+        original[3] = 0x1a;
+        Files.write(mapWithoutHeader, original);
+
+        Method applyMapHeader = ProjectConfigBuilder.class.getDeclaredMethod(
+            "applyMapHeader", WurstProjectConfigData.class, File.class
+        );
+        applyMapHeader.setAccessible(true);
+
+        applyMapHeader.invoke(
+            null,
+            new WurstProjectConfigData(
+                "Test",
+                List.of(),
+                new WurstProjectBuildMapData("Configured map", null, null, null, null, List.of(), List.of()),
+                null,
+                null
+            ),
+            mapWithoutHeader.toFile()
+        );
+
+        assertEquals(Files.readAllBytes(mapWithoutHeader)[0], (byte) 'H');
+        assertEquals(MapHeader.ofFile(mapWithoutHeader.toFile()).getMaxPlayersCount(), 0);
+        assertEquals(Files.size(mapWithoutHeader), original.length + 512);
+    }
+
+    @Test
+    public void mapHeaderConfigDoesNotReadAnArchiveWhenNothingNeedsChanging() throws Exception {
+        Path mapWithoutHeader = Files.createTempFile("wurst-map-without-header-noop", ".w3x");
+        byte[] original = new byte[1024];
+        original[0] = 'M';
+        original[1] = 'P';
+        original[2] = 'Q';
+        original[3] = 0x1a;
+        Files.write(mapWithoutHeader, original);
+
+        Method applyMapHeader = ProjectConfigBuilder.class.getDeclaredMethod(
+            "applyMapHeader", WurstProjectConfigData.class, File.class
+        );
+        applyMapHeader.setAccessible(true);
+        applyMapHeader.invoke(null, WurstProjectConfigData.empty(), mapWithoutHeader.toFile());
+
+        assertEquals(Files.readAllBytes(mapWithoutHeader), original);
+    }
+
     private static String calculateProjectConfigHash(File buildDir) throws Exception {
         Method method = ProjectConfigBuilder.class.getDeclaredMethod(
             "calculateProjectConfigHash",