beech-api 3.5.12 → 3.7.0

package/packages/cli/core/index.js

@@ -1,5 +1,5 @@
-const fs = require("fs");
 const appRoot = require("app-root-path");
+const { performance } = require("perf_hooks");
 const moduleAlias = require("module-alias");
 moduleAlias.addAlias("@", appRoot + "/src");
 const _express_ = require("express");
@@ -8,6 +8,7 @@ const cors = require("cors");
 global.endpoint = _express_.Router();
 const cookieParser = require("cookie-parser");
 const bodyParser = require("body-parser");
+const methodOverride = require("method-override");
 const expressSession = require("express-session");
 const expressValidator = require("express-validator");
 const globalVariable = require(appRoot + "/global.config.js");
@@ -16,33 +17,46 @@ globalVariable.init();
 global._config_ = require(appRoot + "/app.config");
 const mySqlDbConnect = require("./databases/mysql");
 const SequelizeDbConnect = require("./databases/sequelize");
+// database test
+const {
+  testConnectInProcess,
+  filterDbIsTrue,
+  disConnectTestDB,
+} = require("./databases/test");
 // create global sequelize object
 const { QueryTypes, DataTypes, Op } = require("sequelize");
 global.QueryTypes = QueryTypes;
 global.DataTypes = DataTypes;
 global.Op = Op;
-// engine import
-const httpExpress = require("./services/http.express");
-const fileWalk = require("./file-walk/file-walk");
+// allow whitelist cors
+_app_.use(cors({ origin: true, credentials: true }));
+const { whitelist, sign } = require("./origin/index");
 // View engine
 _app_.use(bodyParser.json());
 _app_.use(bodyParser.urlencoded({ extended: true }));
+_app_.use(methodOverride());
 _app_.use(cookieParser());
-_app_.use(expressSession({
-  secret: 'keyboard cat',
-  resave: true,
-  saveUninitialized: true
-}));
+_app_.use(
+  expressSession({
+    secret: "surprise you mother f*cker",
+    resave: true,
+    saveUninitialized: true,
+  })
+);
 _app_.use(expressValidator());
-_app_.use(cors({ origin: true, credentials: true }));
-// Allow Origin
-_app_.all("/", (req, res, next) => {
-  res.header("Access-Control-Allow-Origin", "*");
-  res.header("Access-Control-Allow-Headers", "X-Requested-With");
-  res.header("Access-Control-Allow-Methods", "GET, PUT, PATCH, POST, DELETE, OPTIONS");
-  res.header("Content-Type", "application/json; charset=utf-8");
+// Dev. activity
+_app_.use((req, res, next) => {
+  console.log("Request URL:", req.method, req.originalUrl);
+  var t0 = performance.now();
+  res.on("finish", () => {
+    var t1 = performance.now();
+    console.log(`Responded with status : ${res.statusCode} (${(t1 - t0).toFixed(2)}ms)`);
+  });
   next();
 });
+// engine import
+const httpExpress = require("./services/http.express");
+const fileWalk = require("./file-walk/file-walk");
 // passport initialization
 const authPassport = require("./auth/Passport");
 const passport = require("passport");
@@ -58,12 +72,6 @@ passport.deserializeUser((user, done) => {
 const walk = require("walk");
 let jsfiles = [];
 let walker = walk.walk(appRoot + "/src/endpoints", { followLinks: false });
-// check add-on file exists ?
-if (fs.existsSync(appRoot + "/src/Add-on.js")) {
-  if (_config_.addOn) {
-    jsfiles.push(appRoot + "/src/Add-on.js");
-  }
-}
 // Walk file on push
 walker.on("file", (root, stat, next) => {
   jsfiles.push(root + "/" + stat.name);
@@ -76,14 +84,74 @@ walker.on("end", () => {
 // Initialize the application
 init = async (jsfiles) => {
   try {
-    await ((pool_base == "basic") ? new Promise((resolve) => resolve(mySqlDbConnect.connect())) : new Promise((resolve) => resolve(SequelizeDbConnect.connect())));
-    await new Promise((resolve) => resolve(authPassport.init()));
-    await new Promise((resolve) => resolve(fileWalk.fileWalk(jsfiles)));
-    await new Promise((resolve) => resolve(httpExpress.expressStart()));
+    const testConnectToDB = new Promise((resolve) => {
+      filterDbIsTrue(_config_.database_config, (err, dbTruthy) => {
+        if (err) {
+          throw ("Config file crash.", err);
+        }
+        // leave data to disconnect database
+        let leaveDataForDisconnect = dbTruthy.slice(0);
+        // check db connect truthy length ?
+        if (dbTruthy.length > 0) {
+          testConnectInProcess(dbTruthy, dbTruthy.length, (err, result, dbs) => {
+            if (err) {
+              throw (" Failed  Database connect failed.", err);
+            }
+            if (result) {
+              // Disconnect database
+              disConnectTestDB(leaveDataForDisconnect, dbs, (err, disResult) => {
+                if (err) {
+                  throw (" Failed  Testing Database connect failed.", err);
+                }
+                if (disResult) {
+                  // Disconnect and Next to real
+                  resolve(true);
+                } else {
+                  throw err;
+                }
+              });
+            }
+          });
+        } else {
+          // Not ON connect, Next to real
+          resolve(true);
+        }
+      });
+    });
+    Promise.all([testConnectToDB]).then(async (x) => {
+      if (x[0]) {
+        await (pool_base == "basic"
+          ? new Promise((resolve) => resolve(mySqlDbConnect.connect()))
+          : new Promise((resolve) => resolve(SequelizeDbConnect.connect())));
+        await whitelist(async (lists, originSensitive) => {
+          await _app_.use((req, res, next) => {
+            sign(req, res, lists, originSensitive, (err) => {
+              if (!err) {
+                next();
+              } else {
+                throw err;
+              }
+            });
+          });
+          await authPassport.init().then(async (x) => {
+            if (x[0]) {
+              throw x[0];
+            } else {
+              await new Promise((resolve) => resolve(fileWalk.fileWalk(jsfiles)));
+              await new Promise((resolve) => {
+                httpExpress.expressStart().then((expss) => {
+                  resolve(expss);
+                });
+              });
+            }
+          });
+        });
+      }
+    });
   } catch (error) {
     console.log(" Compile failed ", error);
     throw error;
   }
-}
+};
 // use router
-_app_.use(endpoint);
+_app_.use(endpoint);

package/packages/cli/core/origin/index.js

@@ -0,0 +1,2 @@
+const { whitelist, sign } = require("./whitelist/cors");
+module.exports = { whitelist, sign };

package/packages/cli/core/origin/whitelist/cors.js

@@ -0,0 +1,91 @@
+const fs = require("fs");
+const appRoot = require("app-root-path");
+function whitelist(cb) {
+  var whitelists = [];
+  const getWhitelists = new Promise((resolve) => {
+    if (fs.existsSync(appRoot + "/beech.config.js")) {
+      fs.readFile(appRoot + "/beech.config.js", "utf8", (err, e) => {
+        if (!err) {
+          let defineConfig = eval(e).defineConfig;
+          let origin = defineConfig.server.origin;
+          let originSensitive = defineConfig.server.originSensitive;
+          let allAllow = origin.filter((allow) => {
+            return allow == "*";
+          });
+          if (allAllow.length) {
+            resolve([whitelists, originSensitive]);
+          } else {
+            resolve([origin, originSensitive]);
+          }
+        } else {
+          // error resolve default whitelist
+          resolve([whitelists, originSensitive]);
+        }
+      });
+    } else {
+      resolve([whitelists, originSensitive]);
+    }
+  });
+  // promise all
+  Promise.all([getWhitelists]).then((final) => {
+    cb(final[0][0], final[0][1]);
+  });
+}
+
+function sign(req, res, whitelist, originSensitive, cb) {
+  try {
+    const origin = req.headers.origin;
+    let doYouSignSomeOrigin = false;
+    //var host = req.get("host");
+    console.log("Request from origin:", origin || "localhost");
+
+    if (whitelist.length > 0) {
+      whitelist.forEach((val, k) => {
+        if (origin) {
+          if (origin.indexOf(val) > -1) {
+            doYouSignSomeOrigin = true;
+            if (originSensitive) {
+              res.setHeader("Access-Control-Allow-Origin", val);
+            } else {
+              res.setHeader("Access-Control-Allow-Origin", origin);
+            }
+          }
+        }
+        if (whitelist.length == k + 1) {
+          if (!doYouSignSomeOrigin) {
+            res.setHeader(
+              "Access-Control-Allow-Origin",
+              "http://localhost:" + _config_.main_config.app_port
+            );
+          }
+        }
+      });
+    } else {
+      res.setHeader("Access-Control-Allow-Origin", origin);
+    }
+    // Request methods you wish to allow
+    res.setHeader(
+      "Access-Control-Allow-Methods",
+      "GET, POST, OPTIONS, PUT, PATCH, DELETE"
+    );
+    // Request headers you wish to allow
+    res.setHeader(
+      "Access-Control-Allow-Headers",
+      "X-Requested-With",
+      "Content-Type",
+      "Accept",
+      "Authorization",
+      "Origin",
+      "application/json; charset=utf-8"
+    );
+    // Set to true if you need the website to include cookies in the requests sent
+    // to the API (e.g. in case you use sessions)
+    res.setHeader("Access-Control-Allow-Credentials", true);
+    // callback
+    cb(null);
+  } catch (error) {
+    cb(error);
+  }
+}
+
+module.exports = { whitelist, sign };