Javascript

2025-11-08 07:03:30 -06:00
parent e17efce338
commit 6f9ed49b2e
23 changed files with 445 additions and 891 deletions
--- a/server/index.js
+++ b/server/index.js
@@ -0,0 +1,149 @@
+import express from 'express';
+import cors from 'cors'
+import cookieParser from 'cookie-parser'
+import http from 'http'
+import chalk from 'chalk'
+import moment from 'moment'
+import path from 'path';
+import { initWebSocket } from './ws.js'
+
+import Database from "./db/db.js"
+import AuthHandler from './auth.js';
+import handlers from "./handlers.js";
+
+// Get __dirname in ES6 environment
+import { fileURLToPath } from 'url';
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+class Server {
+    db;
+    auth;
+    UIPath = path.join(__dirname, '../ui')
+
+    registerRoutes(router) {
+        // router.post('/api/location', handlers.updateLocation)
+        router.post('/login', this.auth.login)
+        router.get('/signout', this.auth.logout)
+        router.get('/*', this.get)
+        return router
+    }
+
+    authMiddleware = (req, res, next) => {
+        const authHeader = req.headers.authorization;
+        if (!authHeader) {
+            return res.status(401).json({ error: 'Authorization token required.' });
+        }
+
+        const [scheme, token] = authHeader.split(' ');
+        if (scheme !== 'Bearer' || !token) {
+            return res.status(401).json({ error: 'Malformed authorization header.' })
+        }
+
+        try {
+            const payload = this.auth.verify(token);
+            req.user = payload;
+            return next();
+        } catch (err) {
+            return res.status(403).json({ error: 'Invalid or expired token.' });
+        }
+    }
+
+    get = async (req, res) => {
+        if(!this.auth.isLoggedInUser(req, res)) {
+            console.log("Not logged in")
+            let url = req.url
+            if(url === "/") {
+                url = "/index.html"
+            } else if(!url.includes(".")) { // TODO: Make public app single-page
+                url = path.join("/pages", url) + ".html"
+            }
+
+            let filePath;
+            if(url.startsWith("/_")) {
+                filePath = path.join(this.UIPath, url);
+            } else {
+                filePath = path.join(this.UIPath, "public", url);
+            }
+
+            res.sendFile(filePath);
+        } else {
+            let url = req.url
+
+            let filePath;
+            if(url.startsWith("/_")) {
+                filePath = path.join(this.UIPath, url);
+            } else if(url.includes("75820185")) {
+                filePath = path.join(this.UIPath, "site", url.split("75820185")[1]);
+            } else {
+                filePath = path.join(this.UIPath, "site", "index.html");
+            }
+
+            res.sendFile(filePath);
+        }
+    }
+
+    logRequest(req, res, next) {
+        const formattedDate = moment().format('M.D');
+        const formattedTime = moment().format('h:mma');
+        if(req.url.includes("/api/")) {
+            console.log(chalk.blue(`       ${req.method} ${req.url} | ${formattedDate} ${formattedTime}`));
+        } else {
+            if(req.url === "/")
+                console.log(chalk.gray(`       ${req.method} ${req.url} | ${formattedDate} ${formattedTime}`));
+        }
+        next(); 
+    }
+
+    logResponse(req, res, next) {
+        const originalSend = res.send;
+        res.send = function (body) {
+            if(res.statusCode >= 400) {
+                console.log(chalk.blue( `<-${chalk.red(res.statusCode)}- ${req.method} ${req.url} | ${chalk.red(body)}`));
+            } else {
+                console.log(chalk.blue(`<-${res.statusCode}- ${req.method} ${req.url}`));
+            }
+            originalSend.call(this, body);
+        };
+        next();
+    }
+
+    constructor() {
+        this.db = new Database()
+        global.db = this.db
+        this.auth = new AuthHandler()
+        const app = express();
+        app.use(cors({ origin: '*' }));
+        app.use(express.json());
+        app.use(express.urlencoded({ extended: true }));
+        app.use(cookieParser());
+
+        app.use(this.logRequest);
+        app.use(this.logResponse);
+
+        let router = express.Router();
+        this.registerRoutes(router)
+        app.use('/', router);
+
+        const server = http.createServer(app);
+        initWebSocket(server);
+        const PORT = 3003;
+        server.listen(PORT, () => {
+            console.log("\n")
+            console.log(chalk.yellow("**************America****************"))
+            console.log(chalk.yellowBright(`Server is running on port ${PORT}: http://localhost`));
+            console.log(chalk.yellow("***************************************"))
+            console.log("\n")
+        });
+
+        process.on('SIGINT', async () => {
+            console.log(chalk.red('Closing server...'));
+            console.log(chalk.green('Database connection closed.'));
+            process.exit(0);
+        });
+
+        Object.preventExtensions(this);
+    }
+}
+
+const server = new Server()