index.js

'use strict';

var express = require('express');
var path = require('path');
var dotenv = require('dotenv');
var exphbs = require('express-handlebars');
var fs = require('fs');
var React = require('react');
var ReactDOM = require('react-dom');
var ReactDOMServer = require('react-dom/server');
var handlebars = require('handlebars');

var logger = require('./logger');

function bootstrapServer(options) {
  // Load environment variables
  dotenv.config();

  // @TODO: Compose page object with options and route data
  var page = {
    title: 'Welcome to Hydra-JS!',
    description: 'This the welcome page for Hydra-JS application.',
  };

  var app = express();
  var port = options.port || process.env.PORT || 3000;

  app.set('port', port);

  // Set security headers
  app.use(function (req, res, next) {
    res.setHeader('X-XSS-Protection', '1; mode=block');
    res.setHeader('X-Frame-Options', 'DENY');
    res.setHeader('X-Content-Type-Options', 'nosniff');
    res.setHeader('Referrer-Policy', 'strict-origin-when-cross-origin');
    res.setHeader('Content-Security-Policy', "default-src 'self'");
    next();
  });

  app.disable('x-powered-by'); // Reduce fingerprinting

  // Set up Handlebars as the view engine
  app.engine(
    'html',
    exphbs.engine({
      defaultLayout: 'index',
      layoutsDir: path.join(process.cwd(), 'layouts'),
      extname: '.html',
    })
  );
  app.set('view engine', 'html');
  app.set('views', path.join(process.cwd(), 'routes'));

  // Allow the main app to initialize routes and middleware
  if (options.init) {
    options.init(app);
  }

  // Serve static files from the 'public' directory
  app.use(express.static(path.join(process.cwd(), 'public')));

  app.get('*', function (req, res) {
    // @TODO: Implement route handling properly
    return res.render(req.path.replace(/^\/+/g, ''), { page: page });
  });

  // Request handler
  app.use((err, req, res, next) => {
    if (!err.hasOwnProperty('view')) {
      logger.error(err);
      return res.render('404', { page: page });
    }

    var handlerPath = path.join(process.cwd(), 'dist', 'routes', err.view.name + '.js');
    if (!fs.existsSync(handlerPath)) {
      logger.error(err);
      return res.render('404', { page: page });
    }

    var method = req.method.toLowerCase();
    var handler = require(handlerPath)();

    if (handler.render && typeof handler.render === 'function' && method === 'get') {
      const reactElement = handler.render();
      const jsx = React.createElement(reactElement, {});
      const jsxToHtml = ReactDOMServer.renderToString(jsx);

      const layoutPath = path.join(process.cwd(), 'dist', 'layouts', 'index.html');
      const withTemplate = handlebars.compile(fs.readFileSync(layoutPath, 'utf8'));
      const html = withTemplate({ body: jsxToHtml });
      return res.send(html);
    }

    if (!handler[req.method.toLowerCase()]) return res.render('404', { page: page });

    return handler[req.method.toLowerCase()](req, res, next);
  });

  function startServer() {
    return app.listen(port, function () {
      logger.log('Server running on port ' + port);
    });
  }

  return {
    app: app,
    start: startServer,
  };
}

module.exports = { bootstrapServer, React, ReactDOM };