From ccd7eec56ed00d81e000633141e18a9f1df3645d Mon Sep 17 00:00:00 2001
From: Thomas Weber <weber_t@mailbox.org>
Date: Thu, 8 Jul 2021 02:05:17 -0500
Subject: [PATCH] Append build ID to scaffolding JS bundles

---
 src/build/add-build-id-to-output-plugin.js | 23 ++++++++++++++++++++++
 webpack.config.js                          |  7 +++++--
 2 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 src/build/add-build-id-to-output-plugin.js

diff --git a/src/build/add-build-id-to-output-plugin.js b/src/build/add-build-id-to-output-plugin.js
new file mode 100644
index 0000000..e7450a7
--- /dev/null
+++ b/src/build/add-build-id-to-output-plugin.js
@@ -0,0 +1,23 @@
+const RawSource = require('webpack-sources').RawSource;
+
+const PLUGIN_NAME = 'AddBuildIDToOutputPlugin';
+
+class AddBuildIDToOutputPlugin {
+  constructor (id) {
+    this.magic = `\n// ${id} =^..^=`;
+  }
+  apply(compiler) {
+    compiler.hooks.emit.tap(PLUGIN_NAME, (compilation) => {
+      for (const assetName of Object.keys(compilation.assets)) {
+        if (assetName.endsWith('.js')) {
+          const asset = compilation.getAsset(assetName);
+          const source = asset.source.source().toString();
+          const newSource = `${source}${this.magic}`;
+          compilation.updateAsset(assetName, new RawSource(newSource));
+        }
+      }
+    });
+  }
+}
+
+module.exports = AddBuildIDToOutputPlugin;
diff --git a/webpack.config.js b/webpack.config.js
index 2a0b0b7..1a3d9e5 100644
--- a/webpack.config.js
+++ b/webpack.config.js
@@ -3,6 +3,7 @@ const webpack = require('webpack');
 const HtmlWebpackPlugin = require('html-webpack-plugin');
 const BundleAnalyzerPlugin = require('webpack-bundle-analyzer').BundleAnalyzerPlugin;
 const CopyWebpackPlugin = require('copy-webpack-plugin');
+const AddBuildIDToOutputPlugin = require('./src/build/add-build-id-to-output-plugin');
 
 const base = {
   mode: process.env.NODE_ENV || 'development',
@@ -10,6 +11,8 @@ const base = {
 };
 const dist = path.resolve(__dirname, 'dist');
 
+const buildId = process.env.NODE_ENV === 'production' ? require('./src/build/generate-scaffolding-build-id') : null;
+
 const makeScaffolding = ({full}) => ({
   ...base,
   output: {
@@ -76,6 +79,7 @@ const makeScaffolding = ({full}) => ({
         }
       ]
     }),
+    ...(buildId ? [new AddBuildIDToOutputPlugin(buildId)] : []),
     ...(process.env.BUNDLE_ANALYZER === (full ? '1' : '2') ? [new BundleAnalyzerPlugin()] : [])
   ]
 });
@@ -126,8 +130,7 @@ const makeWebsite = () => ({
     }),
     new webpack.DefinePlugin({
       'process.env.LARGE_ASSET_BASE': JSON.stringify(process.env.LARGE_ASSET_BASE || 'https://packagerdata.turbowarp.org/'),
-      // In production mode, generate a unique hash. In development mode, use a random ID every reload as those files are coming from localhost anyways.
-      'process.env.SCAFFOLDING_BUILD_ID': process.env.NODE_ENV === 'production' ? JSON.stringify(require('./src/build/generate-scaffolding-build-id')) : 'Math.random().toString()',
+      'process.env.SCAFFOLDING_BUILD_ID': buildId ? JSON.stringify(buildId) : 'Math.random().toString()',
       'process.env.NODE_ENV': JSON.stringify(process.env.NODE_ENV || 'development'),
       'process.env.PLAUSIBLE_API': JSON.stringify(process.env.PLAUSIBLE_API),
       'process.env.PLAUSIBLE_DOMAIN': JSON.stringify(process.env.PLAUSIBLE_DOMAIN),