Merge pull request #135 from beNative/codex/fix-auto-update-checksum-mismatch

Fix release metadata hashes and strengthen auto-update checks

Author: beNative

package-lock.json

@@ -63,7 +63,8 @@
     "remark-math": "^6.0.0",
     "shiki": "^1.22.0",
     "tailwindcss": "^3.4.14",
-    "typescript": "^5.5.3"
+    "typescript": "^5.5.3",
+    "yaml": "^2.8.1"
   },
   "build": {
     "publish": [

package.json

@@ -1,7 +1,10 @@
 #!/usr/bin/env node
+import { createReadStream } from 'fs';
 import { promises as fs } from 'fs';
 import path from 'path';
 import process from 'process';
+import crypto from 'crypto';
+import YAML from 'yaml';
 
 function parseArgs(argv) {
   const args = {};
@@ -162,6 +165,21 @@ async function normaliseReleaseAsset(filePath) {
   return { filePath: targetPath, fileName: normalisedName };
 }
 
+async function computeSha512(filePath) {
+  return new Promise((resolve, reject) => {
+    const hash = crypto.createHash('sha512');
+    const stream = createReadStream(filePath);
+    stream.on('data', (chunk) => hash.update(chunk));
+    stream.on('error', reject);
+    stream.on('end', () => resolve(hash.digest('base64')));
+  });
+}
+
+async function getFileSize(filePath) {
+  const stats = await fs.stat(filePath);
+  return stats.size;
+}
+
 async function collectAssets(artifactRoot) {
   const releaseAssets = [];
   const updateSupportFiles = [];
@@ -191,12 +209,20 @@ async function collectAssets(artifactRoot) {
       if (!isReleaseAsset(fileName)) {
         continue;
       }
+
+      const [sha512, size] = await Promise.all([
+        computeSha512(filePath),
+        getFileSize(filePath),
+      ]);
+
       releaseAssets.push({
         platform,
         arch,
         fileName,
         filePath,
         format: detectFormat(fileName),
+        sha512,
+        size,
       });
     }
   }
@@ -218,6 +244,117 @@ async function collectAssets(artifactRoot) {
   return { releaseAssets, updateSupportFiles };
 }
 
+async function updateMetadataFiles(metadataFiles, releaseAssets) {
+  if (metadataFiles.length === 0) {
+    return;
+  }
+
+  const assetByName = new Map(releaseAssets.map((asset) => [asset.fileName, asset]));
+
+  const ensureEntryMatchesAsset = (entry) => {
+    if (!entry) {
+      return false;
+    }
+    const key = entry.url || entry.path;
+    if (!key) {
+      return false;
+    }
+    const asset = assetByName.get(key);
+    if (!asset) {
+      return false;
+    }
+    let changed = false;
+    if (entry.url && entry.url !== asset.fileName) {
+      entry.url = asset.fileName;
+      changed = true;
+    }
+    if (entry.path && entry.path !== asset.fileName) {
+      entry.path = asset.fileName;
+      changed = true;
+    }
+    if (typeof asset.size === 'number' && entry.size !== asset.size) {
+      entry.size = asset.size;
+      changed = true;
+    }
+    if (asset.sha512 && entry.sha512 !== asset.sha512) {
+      entry.sha512 = asset.sha512;
+      changed = true;
+    }
+    return changed;
+  };
+
+  for (const metadataPath of metadataFiles) {
+    let parsed;
+    try {
+      const source = await fs.readFile(metadataPath, 'utf8');
+      parsed = YAML.parse(source);
+    } catch (error) {
+      console.warn(`Failed to parse auto-update metadata at ${metadataPath}:`, error);
+      continue;
+    }
+
+    if (!parsed || typeof parsed !== 'object') {
+      continue;
+    }
+
+    let changed = false;
+
+    if (Array.isArray(parsed.files)) {
+      for (const entry of parsed.files) {
+        if (ensureEntryMatchesAsset(entry)) {
+          changed = true;
+        }
+      }
+    }
+
+    const primaryKey =
+      (typeof parsed.path === 'string' && parsed.path) ||
+      (Array.isArray(parsed.files) && parsed.files[0] && (parsed.files[0].path || parsed.files[0].url));
+
+    if (primaryKey) {
+      const asset = assetByName.get(primaryKey);
+      if (asset) {
+        if (parsed.path !== asset.fileName) {
+          parsed.path = asset.fileName;
+          changed = true;
+        }
+        if (asset.sha512 && parsed.sha512 !== asset.sha512) {
+          parsed.sha512 = asset.sha512;
+          changed = true;
+        }
+        if (typeof asset.size === 'number' && parsed.size !== undefined && parsed.size !== asset.size) {
+          parsed.size = asset.size;
+          changed = true;
+        }
+      }
+    }
+
+    if (!parsed.sha512 && parsed.path && assetByName.has(parsed.path)) {
+      parsed.sha512 = assetByName.get(parsed.path).sha512;
+      changed = true;
+    }
+
+    if (!Array.isArray(parsed.files) && parsed.path && assetByName.has(parsed.path)) {
+      parsed.files = [
+        {
+          url: parsed.path,
+          sha512: parsed.sha512,
+          size: assetByName.get(parsed.path).size,
+        },
+      ];
+      changed = true;
+    }
+
+    if (!changed) {
+      continue;
+    }
+
+    const serialised = YAML.stringify(parsed, { lineWidth: 0 }).trimEnd();
+    await fs.writeFile(metadataPath, `${serialised}\n`, 'utf8');
+    console.log(`Updated auto-update metadata checksums in ${metadataPath}`);
+  }
+}
+
 function buildDownloadTable(entries, repo, tag) {
   const header = ['| Platform | Architecture | Format | Download |', '| --- | --- | --- | --- |'];
   const rows = entries.map((entry) => {
@@ -244,6 +381,7 @@ async function main() {
   const body = entryLines.join('\n').trim();
 
   const { releaseAssets, updateSupportFiles } = await collectAssets(artifactRoot);
+  await updateMetadataFiles(updateSupportFiles, releaseAssets);
   const table = buildDownloadTable(releaseAssets, repository, tag);
 
   const sections = [`# DocForge v${version}`];