barehttp 0.4.2 → 0.6.0

package/lib/server.js

@@ -5,19 +5,22 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BareHttp = exports.BareServer = void 0;
 const find_my_way_1 = __importDefault(require("find-my-way"));
+const ajv_1 = __importDefault(require("ajv"));
 const request_1 = require("./request");
 const logger_1 = require("./logger");
 const context_1 = require("./context");
-const report_1 = require("./report");
 const utils_1 = require("./utils");
 const cors_1 = require("./middlewares/cors/cors");
 const websocket_1 = require("./websocket");
+const generator_1 = require("./schemas/generator");
 const dns_1 = __importDefault(require("dns"));
 const http_1 = require("http");
 class BareServer {
     bareOptions;
     server;
     ws;
+    ajv;
+    route = {};
     #middlewares = [];
     #routes = new Map();
     #routesLib = new Map();
@@ -26,26 +29,37 @@ class BareServer {
     #corsInstance;
     #port = 3000;
     #host = '0.0.0.0';
-    #runMiddlewaresSequence = (_) => _;
+    #globalMiddlewaresRun = (_) => _;
+    #routeMiddlewaresStore = new Map();
+    #routeRuntimeSchemas = new Map();
     constructor(bareOptions = {}) {
         this.bareOptions = bareOptions;
         // init
         this.server = (0, http_1.createServer)(this.#listener.bind(this));
         this.attachGracefulHandlers();
         this.attachRoutesDeclarator();
-        this.mainOptionsSetter();
+        this.applyLaunchOptions();
+        this.loadRoutesSchemas();
         return this;
     }
     #listener = (request, response) => {
-        const { requestTimeFormat, logging } = this.bareOptions;
-        const flow = new request_1.BareRequest(request, response, { logging, requestTimeFormat });
+        const flow = new request_1.BareRequest(request, response, this.bareOptions);
         // init and attach request uuid to the context
         if (this.bareOptions.context) {
             (0, context_1.newContext)('request');
             context_1.context.current?.store.set('id', flow.ID.code);
         }
-        // attach a flow to the flow memory storage
-        this.applyMiddlewares(flow).catch((e) => this.#errorHandler(e, flow, 400));
+        // execute global middlewares on the request
+        this.applyMiddlewares(flow)
+            .catch((e) => {
+            this.#errorHandler(e, flow, 400);
+        })
+            .then(() => {
+            // if middlewares sent the response back, stop here
+            if (flow.sent)
+                return;
+            this.#router.lookup(flow._originalRequest, flow._originalResponse);
+        });
     };
     /**
      * This function generates previously defined middlewares for the sequential execution
@@ -63,11 +77,16 @@ class BareServer {
             }
         }
         const text = lines.join('\n');
-        this.#runMiddlewaresSequence = new AsyncFunction('flow', text);
+        this.#globalMiddlewaresRun = new AsyncFunction('flow', text);
     };
-    mainOptionsSetter = () => {
+    applyLaunchOptions = () => {
         const { bareOptions: bo } = this;
-        this.#port = +(bo.serverPort || process.env.PORT || 3000);
+        if (bo.setRandomPort) {
+            this.#port = undefined;
+        }
+        else {
+            this.#port = +(bo.serverPort || process.env.PORT || 3000);
+        }
         this.#host = typeof bo.serverAddress === 'string' ? bo.serverAddress : '0.0.0.0';
         // context setting
         if (bo.context)
@@ -85,18 +104,15 @@ class BareServer {
             this.#corsInstance = new cors_1.Cors(corsOpts);
         }
         this.#middlewares.push(...(bo.middlewares || []));
-        if (bo.statisticsReport)
-            this.registerReport();
     };
     async applyMiddlewares(flow) {
-        if (!flow) {
-            throw new Error(`No flow been found to apply middlewares for, theres a sync mistake in the server.`); // should NEVER happen
-        }
         if (this.bareOptions.cors) {
             this.resolveMiddleware(flow, 0, this.#corsInstance?.corsMiddleware.bind(this.#corsInstance));
         }
-        // invoke body stream consumption
-        await flow['readBody']();
+        if (this.bareOptions.doNotParseBody !== true) {
+            // invoke body stream consumption
+            await flow['readBody']();
+        }
         // attach cookies middleware
         if (this.bareOptions.cookies) {
             flow['attachCookieManager'](this.bareOptions.cookiesOptions);
@@ -109,11 +125,7 @@ class BareServer {
             flow['setRemoteClient'](remoteClient[0]);
         }
         if (this.#middlewares.length)
-            await this.#runMiddlewaresSequence(flow);
-        // now route the request if middlewares did not send the response back
-        if (!flow.sent) {
-            this.#router.lookup(flow._originalRequest, flow._originalResponse);
-        }
+            await this.#globalMiddlewaresRun(flow);
     }
     /**
      * This handler is used in async generated middlewares runtime function
@@ -129,15 +141,13 @@ class BareServer {
             this.#errorHandler(e, flow);
         }
     }
-    setRoute(method, route, runtime, handler, opts) {
+    setRoute(method, route, isRuntime, handler, opts) {
         const encode = this.encodeRoute(method, route);
-        this.#routes.set(encode, { hits: 0, fails: 0, success: 0 });
         const handleFn = (req, _, routeParams) => {
-            this.#routes.get(encode).hits++;
-            this.handleRoute(req, checkParams(routeParams), handler, encode, opts);
+            this.handleRoute(req, checkParams(routeParams), handler, opts);
         };
         this.#routesLib.set(encode, handleFn);
-        if (runtime) {
+        if (isRuntime) {
             this.#router.reset();
             this.#routesLib.forEach((handlerFn, route) => {
                 const [m, r] = this.explodeRoute(route);
@@ -148,53 +158,63 @@ class BareServer {
             this.#router.on(method, route, handleFn);
         }
     }
-    registerReport() {
-        this.setRoute('GET', '/_report', false, (flow) => {
-            flow.setHeader('Content-Type', 'text/html');
-            flow.send((0, report_1.generateReport)(this.#routes));
-        });
-    }
-    handleRoute(req, routeParams, handle, encodedRoute, opts) {
+    handleRoute(req, routeParams, handle, routeOpts) {
         const flow = req.flow;
         if (!flow) {
             throw new Error(`No flow been found to route this request, theres a sync mistake in the server.`); // should NEVER happen
         }
-        // apply possible route options
-        if (opts) {
-            if (opts.disableCache)
-                flow.disableCache();
-            if (opts.cache)
-                flow.setCache(opts.cache);
-            if (opts.timeout)
-                flow['attachTimeout'](opts.timeout);
-        }
         // populate with route params
         if (routeParams)
             flow['setParams'](routeParams);
-        // attach a general statistic reports counter
-        if (this.bareOptions.statisticsReport) {
-            flow._originalRequest.on('close', () => {
-                if (flow.statusToSend < 300 && flow.statusToSend >= 200) {
-                    this.#routes.get(encodedRoute).success++;
-                }
-                else {
-                    this.#routes.get(encodedRoute).fails++;
-                }
-            });
+        // apply possible route options
+        if (routeOpts) {
+            if (routeOpts.disableCache)
+                flow.disableCache();
+            if (routeOpts.cache)
+                flow.setCache(routeOpts.cache);
+            if (routeOpts.timeout)
+                flow['attachTimeout'](routeOpts.timeout);
         }
+        // TODO: implement per route middlewares!
         try {
-            const routeReturn = handle.bind(undefined)(flow);
+            const routeReturn = handle(flow);
+            if (flow.sent)
+                return;
             if (routeReturn instanceof Promise) {
-                routeReturn.then((result) => flow.send(result)).catch((e) => this.#errorHandler(e, flow));
-            }
-            else {
-                flow.send(routeReturn);
+                routeReturn
+                    .then((result) => this.resolveResponse(flow, result, req.url, req.method?.toLowerCase(), routeOpts?.builtInRuntime?.output))
+                    .catch((e) => {
+                    this.#errorHandler(e, flow);
+                });
+                return;
             }
+            this.resolveResponse(flow, routeReturn, req.url, req.method?.toLowerCase(), routeOpts?.builtInRuntime?.output);
         }
         catch (e) {
             this.#errorHandler(e, flow);
         }
     }
+    resolveResponse(flow, response, url, method, builtInRuntime) {
+        if (!builtInRuntime || !method || !url) {
+            flow.send(response);
+            return;
+        }
+        const schema = this.#routeRuntimeSchemas.get(`${method}-${url}`);
+        const check = schema?.compiled(response);
+        if ((schema && check) || !schema)
+            flow.send(response);
+        else {
+            logger_1.logMe.error('Response schema error!', {
+                method,
+                url,
+                errors: schema?.compiled.errors,
+                received: response,
+            });
+            flow
+                .status(500)
+                .send({ message: `Response schema error, please communicate to server administrator.` });
+        }
+    }
     encodeRoute(method, route) {
         if (route.endsWith('/'))
             route = route.slice(0, -1);
@@ -232,7 +252,7 @@ class BareServer {
     }
     attachRoutesDeclarator() {
         for (const method of [...Object.keys(utils_1.HttpMethods), 'declare']) {
-            this[method] = (routeSetUp) => {
+            this.route[method] = (routeSetUp) => {
                 checkRouteSetUp(routeSetUp, method);
                 if (method === 'declare') {
                     for (const m of new Set(routeSetUp.methods))
@@ -298,8 +318,10 @@ class BareServer {
         // }
         if (!this.ws)
             await this.stopWs();
+        if (!this.server?.listening)
+            return;
         await new Promise((res, rej) => {
-            this.server?.close((e) => {
+            this.server.close((e) => {
                 if (e) {
                     rej(e);
                     cb?.(e);
@@ -311,6 +333,23 @@ class BareServer {
             });
         });
     }
+    loadRoutesSchemas() {
+        if (!this.bareOptions.enableSchemaValidation) {
+            return;
+        }
+        if (this.bareOptions.declaredRoutesPaths?.length) {
+            this.ajv = new ajv_1.default({ strict: true });
+            for (const path of this.bareOptions.declaredRoutesPaths) {
+                const schemas = (0, generator_1.generateRouteSchema)(path);
+                for (const schema of schemas) {
+                    this.#routeRuntimeSchemas.set(`${schema.methodName}-${schema.route}`, {
+                        raw: schema.jsonSchema,
+                        compiled: this.ajv.compile(schema.jsonSchema),
+                    });
+                }
+            }
+        }
+    }
     use(middleware) {
         this.#middlewares.push(middleware);
         return this;
@@ -321,11 +360,15 @@ class BareServer {
     setCustomErrorHandler(eh) {
         this.#errorHandler = eh;
     }
+    getServerPort() {
+        return this.server.address().port;
+    }
     getRoutes() {
         return [...this.#routes.keys()];
     }
 }
 exports.BareServer = BareServer;
+exports.BareHttp = BareServer;
 function checkRouteSetUp(routeSetUp, key) {
     if (typeof routeSetUp.route !== 'string') {
         throw new TypeError(`A route path for the method ${key} is not a a string`);
@@ -358,5 +401,3 @@ function checkParams(params) {
     }
     return params;
 }
-const BareHttp = BareServer;
-exports.BareHttp = BareHttp;

package/lib/websocket.js

@@ -49,7 +49,9 @@ class WebSocketServer {
                 }
             }
             catch (e) {
-                this.rejectUpgrade(request, socket, e?.message, e);
+                if (e instanceof Error) {
+                    this.rejectUpgrade(request, socket, e?.message, e);
+                }
             }
         });
         this.customUpgradeDone = true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "barehttp",
-  "version": "0.4.2",
+  "version": "0.6.0",
   "description": "Lightweight and fast Node.js web server",
   "main": "lib/index.js",
   "directories": {
@@ -11,9 +11,9 @@
     "lib"
   ],
   "scripts": {
-    "build": "rm -rf ./lib && tsc",
+    "build": "rm -rf ./lib && tsc -p tsconfig.build.json",
     "build:dev": "rm -rf ./dev-lib && tsc -p tsconfig.dev.json",
-    "test": "jest --runInBand --coverage",
+    "test": "jest --coverage",
     "lint": "eslint ./src --fix",
     "release": "semantic-release -e ./.releaserc.json"
   },
@@ -33,14 +33,17 @@
   },
   "license": "MIT",
   "dependencies": {
+    "ajv": "^8.11.0",
     "callsites": "^3.1.0",
-    "cookie": "^0.4.1",
-    "cookie-signature": "^1.1.0",
-    "find-my-way": "^5.0.0",
+    "cookie": "^0.5.0",
+    "cookie-signature": "^1.2.0",
+    "find-my-way": "^5.6.0",
     "hyperid": "^2.3.1",
-    "pino": "^7.5.1",
-    "pino-pretty": "^7.2.0",
-    "ws": "^8.3.0"
+    "lodash": "^4.17.21",
+    "pino": "^7.11.0",
+    "pino-pretty": "^7.6.1",
+    "ts-morph": "^14.0.0",
+    "ws": "^8.6.0"
   },
   "devDependencies": {
     "@semantic-release/git": "^10.0.1",
@@ -61,11 +64,12 @@
     "express": "^4.17.1",
     "fastify": "^3.24.1",
     "jest": "^27.4.3",
+    "prettier": "^2.5.1",
     "semantic-release": "^18.0.1",
     "supertest": "^6.1.3",
     "ts-jest": "^27.0.7",
     "ts-node-dev": "^1.1.6",
-    "typescript": "^4.5.2"
+    "typescript": "^4.6.4"
   },
   "optionalDependencies": {
     "bufferutil": "^4.0.3",

package/lib/report.d.ts

@@ -1,2 +0,0 @@
-import { RouteReport } from './server';
-export declare function generateReport(routes: Map<string, RouteReport>): string;

package/lib/report.js

@@ -1,20 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateReport = void 0;
-function generateReport(routes) {
-    const lines = [];
-    lines.push('<!DOCTYPE html><html><head><title>Routes usage</title><meta charset="utf-8"></head><body><table style="border: 2px;border-style: ridge;border-radius: 5px;padding: 10px;"><tr><th>Route</th><th>Hits</th><th>Successes</th><th>Fails</th></tr>');
-    const sorted = [...routes].sort(([a], [b]) => {
-        if (a > b)
-            return 1;
-        else if (b > a)
-            return -1;
-        return 0;
-    });
-    sorted.forEach(([route, stats]) => {
-        lines.push(`<tr><td>${route}</td><td>${stats.hits}</td><td>${stats.success}</td><td>${stats.fails}</td></tr>`);
-    });
-    lines.push('</table></body></html>');
-    return lines.join('');
-}
-exports.generateReport = generateReport;