bodyshop/server.js

const express = require("express");
const cors = require("cors");
const bodyParser = require("body-parser");
const path = require("path");
const compression = require("compression");
const cookieParser = require("cookie-parser");
const http = require("http");
const https = require("https");
const fs = require("fs");
const { Server } = require("socket.io");
const { createClient } = require("redis");
const { createAdapter } = require("@socket.io/redis-adapter");
const logger = require("./server/utils/logger");

// This file offers the following exports
// redisClient: Redis client for external use
// io: Socket.IO server for external use

// Load environment variables
require("dotenv").config({
  path: path.resolve(process.cwd(), `.env.${process.env.NODE_ENV || "development"}`)
});

/**
 * CORS Origin for Socket.IO
 * @type {string[][]}
 */
const SOCKETIO_CORS_ORIGIN = [
  "https://test.imex.online",
  "https://www.test.imex.online",
  "http://localhost:3000",
  "https://imex.online",
  "https://www.imex.online",
  "https://romeonline.io",
  "https://www.romeonline.io",
  "https://beta.test.romeonline.io",
  "https://www.beta.test.romeonline.io",
  "https://beta.romeonline.io",
  "https://www.beta.romeonline.io",
  "https://beta.test.imex.online",
  "https://www.beta.test.imex.online",
  "https://beta.imex.online",
  "https://www.beta.imex.online",
  "https://www.test.promanager.web-est.com",
  "https://test.promanager.web-est.com",
  "https://www.promanager.web-est.com",
  "https://www.promanager.web-est.com"
];

/**
 * Middleware for Express app
 * @param app
 */
const applyMiddleware = (app) => {
  app.use(compression());
  app.use(cookieParser());
  app.use(bodyParser.json({ limit: "50mb" }));
  app.use(bodyParser.urlencoded({ limit: "50mb", extended: true }));
  app.use(cors({ credentials: true, exposedHeaders: ["set-cookie"] }));

  // Helper middleware
  app.use((req, res, next) => {
    req.logger = logger;
    next();
  });
};

/**
 * Route groupings for Express app
 * @param app
 */
const applyRoutes = (app) => {
  app.use("/", require("./server/routes/miscellaneousRoutes"));
  app.use("/notifications", require("./server/routes/notificationsRoutes"));
  app.use("/render", require("./server/routes/renderRoutes"));
  app.use("/mixdata", require("./server/routes/mixDataRoutes"));
  app.use("/accounting", require("./server/routes/accountingRoutes"));
  app.use("/qbo", require("./server/routes/qboRoutes"));
  app.use("/media", require("./server/routes/mediaRoutes"));
  app.use("/sms", require("./server/routes/smsRoutes"));
  app.use("/job", require("./server/routes/jobRoutes"));
  app.use("/scheduling", require("./server/routes/schedulingRoutes"));
  app.use("/utils", require("./server/routes/utilRoutes"));
  app.use("/data", require("./server/routes/dataRoutes"));
  app.use("/adm", require("./server/routes/adminRoutes"));
  app.use("/tech", require("./server/routes/techRoutes"));
  app.use("/intellipay", require("./server/routes/intellipayRoutes"));
  app.use("/cdk", require("./server/routes/cdkRoutes"));
  app.use("/csi", require("./server/routes/csiRoutes"));
  app.use("/payroll", require("./server/routes/payrollRoutes"));

  // Default route for forbidden access
  app.get("/", (req, res) => {
    res.status(200).send("Access Forbidden.");
  });
};

/**
 * Main function to start the server
 * @returns {Promise<void>}
 */
const main = async () => {
  const app = express();
  const port = process.env.PORT || 5000;

  let server;

  // In production or default, use HTTP
  server = http.createServer(app);
  logger.log(`[${process.env.NODE_ENV}] Attempting to start process on port ${port}`, "INFO", "api");

  // Redis client setup for Pub/Sub and Key-Value Store
  const pubClient = createClient({ url: process.env.REDIS_URL || "redis://localhost:6379" });
  const subClient = pubClient.duplicate();

  pubClient.on("error", (err) => logger.log(`Redis pubClient error: ${err}`, "ERROR", "redis"));
  subClient.on("error", (err) => logger.log(`Redis subClient error: ${err}`, "ERROR", "redis"));

  try {
    await Promise.all([pubClient.connect(), subClient.connect()]);
    logger.log(`[${process.env.NODE_ENV}] Connected to Redis`, "INFO", "redis", "api");
  } catch (redisError) {
    logger.log("Failed to connect to Redis", "ERROR", "redis", redisError);
  }

  process.on("SIGINT", async () => {
    await Promise.all([pubClient.disconnect(), subClient.disconnect()]);
    process.exit(0);
  });

  exports.io = new Server(server, {
    path: "/ws",
    adapter: createAdapter(pubClient, subClient),
    cors: {
      origin: SOCKETIO_CORS_ORIGIN,
      methods: ["GET", "POST"],
      credentials: true,
      exposedHeaders: ["set-cookie"]
    }
  });

  exports.redisClient = pubClient;

  // Store session data in Redis
  exports.setSessionData = async (socketId, key, value) => {
    await pubClient.hSet(`socket:${socketId}`, key, JSON.stringify(value)); // Use Redis pubClient
  };

  // Retrieve session data from Redis
  exports.getSessionData = async (socketId, key) => {
    const data = await pubClient.hGet(`socket:${socketId}`, key);
    return data ? JSON.parse(data) : null;
  };

  // Clear session data from Redis
  exports.clearSessionData = async (socketId) => {
    await pubClient.del(`socket:${socketId}`);
  };

  // TODO: Remove, just a demo of hGet and hSet
  // async function demoSessionData() {
  //   // Store session data using setSessionData
  //   await exports.setSessionData("testSocketId", "field1", "Hello, Redis Hash!");
  //
  //   // Retrieve session data using getSessionData
  //   const value = await exports.getSessionData("testSocketId", "field1");
  //
  //   // Output the retrieved value
  //   console.log(value);
  // }
  // demoSessionData().catch(console.error);
  // TODO: End of demo

  require("./server/web-sockets/web-socket");

  applyMiddleware(app);
  applyRoutes(app);

  try {
    await server.listen(port);
    logger.log(`[${process.env.NODE_ENV}] Server started on port ${port}`, "INFO", "api");
  } catch (error) {
    logger.log(`[${process.env.NODE_ENV}] Server failed to start on port ${port}`, "ERROR", "api", error);
  }
};

// Start server
main();