index.js

const fs = require('fs');
const path = require('path');
const express = require('express');
const webpack = require('webpack');
const devMiddleware = require('webpack-dev-middleware');
const hotMiddleware = require('webpack-hot-middleware');
const webpackConfigBuilder = require('./webpack.config.js');
const { updateContent } = require('./middleware.js');

const ENV = 'development';

const hostname = process.argv[3] || '127.0.0.1';
const port = 3333;

const outputPath = path.join(__dirname, 'out');

const webpackConfig = webpackConfigBuilder(ENV, outputPath);
const compiler = webpack(webpackConfig);
const devMiddlewareOptions = {
  // writeToDisk: true,
};
const app = express();

app.use(devMiddleware(compiler, devMiddlewareOptions));
app.use(hotMiddleware(compiler));

// Middleware to inject script tag into HTML files
app.use(async (req, res, next) => {
  let filePath = path.join(outputPath, req.path === '/' ? 'index.html' : req.path);
  if (path.extname(filePath) === '' && fs.existsSync(`${filePath}.html`)) {
    filePath += '.html';
  }
  if (path.extname(filePath) !== '.html' && fs.existsSync(`${filePath}.html`)) {
    filePath += '.html';
  }
  if (!path.basename(filePath).startsWith("_") && path.extname(filePath) === '.html' && fs.existsSync(filePath)) {
    try {
      const html = await fs.promises.readFile(filePath, 'utf8');
      res.setHeader('Content-Type', 'text/html');
      res.send(updateContent(html, {
        relativePath: path.relative(outputPath, filePath),
        genSearchWidgetConfigId: process.env.GEN_SEARCH_WIDGET_ID || '',
        gtmId: process.env.GTM_ID || '',
        environment: ENV
      }));
    } catch (err) {
      next();
    }
  } else {
    next();
  }
});

// Serve static files from /out
app.use(express.static(outputPath));

// 404 error handler - must be placed after all other routes
app.use((req, res, next) => {
  const notFoundPath = path.join(__dirname, 'public', '404.html');
  if (fs.existsSync(notFoundPath)) {
    try {
      const html = fs.readFileSync(notFoundPath, 'utf8');
      res.status(404);
      res.setHeader('Content-Type', 'text/html');
      res.send(html);
    } catch (err) {
      res.status(404).send('404 - Page Not Found');
    }
  } else {
    res.status(404).send('404 - Page Not Found');
  }
});

app.listen(port, hostname, () => {
  process.stdout.write(`Server running at http://${hostname}:${port}/\n`);
});