Webpack - Generate Netlify HTTP2 Server Push _headers File when using the HtmlWebpackPlugin

NetlifyServerPushPlugin.js

/**
 * Generate a Netlify HTTP2 Server Push configuration.
 *
 * Options:
 * - headersFile {string} path to the _headers file that should be generated (relative to your output dir)
 */
function NetlifyServerPushPlugin(options) {
  this.options = options;
}

NetlifyServerPushPlugin.prototype.generateAssetHeaders = function (assets) {
  // Turn script files into script tags
  const scriptHeaders = assets.js.map(path => `  Link: <${path}>; rel=preload; as=script`);
  const styleHeaders = assets.css.map(path => `  Link: <${path}>; rel=preload; as=style`);
  return `/
${scriptHeaders.concat(styleHeaders).join('\n')}
`;
};

NetlifyServerPushPlugin.prototype.apply = function (compiler) {
  compiler.plugin('compilation', (compilation) => {
    compilation.plugin('html-webpack-plugin-before-html-generation', (htmlPluginData, callback) => {
      const assets = htmlPluginData.assets;
      const source = this.generateAssetHeaders(assets);
      compilation.assets[`${this.options.headersFile}`] = {
        source: () => source,
        size: () => source.length
      };
      callback(null, htmlPluginData);
    });
  });
};

module.exports = NetlifyServerPushPlugin;

webpack.config.js

const NetlifyServerPushPlugin = require('./NetlifyServerPushPlugin');

module.exports = {
  plugins: [
    new HtmlWebpackPlugin({
      template: 'src/index.html',
    }),
    new NetlifyServerPushPlugin({
      headersFile: '_headers'
    })
  ]
};