@nka212bg/backend-utils 0.1.31 → 0.1.33

package/misc.js

@@ -451,7 +451,7 @@ exports.crypto = ({ salt, embedSalt } = {}) => {
 };
 
 exports.secToTime = (sec) => {
-    sec = parseInt(sec);
+    sec = Math.abs(parseInt(sec));
     if (!sec) return;
 
     const SEC_IN_MIN = 60;

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@nka212bg/backend-utils",
     "author": "nka212bg",
-    "version": "0.1.31",
+    "version": "0.1.33",
     "main": "index.js",
     "dependencies": {
         "@maxmind/geoip2-node": "^5.0.0",

package/session.js

@@ -1,28 +1,138 @@
-const { existsSync, readFileSync, writeFileSync, mkdirSync } = require("fs");
+const { existsSync, readFileSync, mkdirSync } = require("fs");
+const { writeFile } = require("fs/promises");
 const { createToken } = require("./misc");
 
 !existsSync(`${__basedir}/state`) && mkdirSync(`${__basedir}/state`, { recursive: true });
-const sessionStateFile = `${__basedir}/state/session.json`;
 
+/* -------------------- ban list -------------------- */
+const banListFile = `${__basedir}/state/banList.json`;
+const BAN_LIST = existsSync(banListFile) ? JSON.parse(readFileSync(banListFile, "utf8") || "{}") : {};
+
+/* -------------------- session --------------------- */
+const sessionStateFile = `${__basedir}/state/session.json`;
 const SESSION = existsSync(sessionStateFile) ? JSON.parse(readFileSync(sessionStateFile, "utf8") || "{}") : {};
 
+/* --------------------  state  --------------------- */
+const serverStateFile = `${__basedir}/state/state.json`;
+const STATE = existsSync(serverStateFile) ? JSON.parse(readFileSync(serverStateFile, "utf8") || "{}") : {};
+
 setInterval(() => {
-    // delete expired sessions
     const now = Date.now();
+
+    // delete expired sessions
     Object.keys(SESSION).forEach((e) => {
         SESSION[e].expireTime < now && delete SESSION[e];
     });
+    // save the sessions in file (non-blocking write, JSON.stringify itself still runs sync)
+    writeFile(sessionStateFile, JSON.stringify(SESSION)).catch((error) => console.error("session save failed:", error));
 
-    // save the sessions in file
-    writeFileSync(sessionStateFile, JSON.stringify(SESSION));
+    // delete expired entries
+    Object.keys(STATE).forEach((e) => {
+        STATE[e].expireTime < now && delete STATE[e];
+    });
+    // save the state in file (non-blocking write, JSON.stringify itself still runs sync)
+    writeFile(serverStateFile, JSON.stringify(STATE)).catch((error) => console.error("state save failed:", error));
 }, 60000);
 
+// document.cookie-style: reading serverState[token] gets it, assigning
+// serverState[token] = {...} sets/updates it, assigning a falsy value
+// deletes it (the same "expire it out" trick browsers use for cookies).
+exports.serverState = new Proxy(
+    {},
+    {
+        // GET
+        // const s = serverState["abc123"]; // => {name: 'alice', ..., expireIn: 604859999}
+        // const s = serverState[req.headers.authorization];
+        // const all = serverState.getAll; // => whole STATE map
+        // const alices = serverState["name=alice"]; // => all entries where name == "alice", each with its token
+        get(_, token) {
+            if (!token) return false;
+
+            if (token === "getAll") return STATE;
+
+            if (token.includes("=")) {
+                const [field, value] = token.trim().split(/\s*=\s*/);
+                return Object.entries(STATE)
+                    .filter(([, s]) => s[field] == value)
+                    .map(([token, s]) => ({ ...s, token }));
+            }
+
+            if (!STATE[token]) return false;
+            return { ...STATE[token], expireIn: STATE[token].expireTime - Date.now() + 60000 };
+        },
+
+        // DELETE
+        // delete serverState["abc123"];
+        // delete serverState["name=alice"]; // bulk delete, removes every entry where name == "alice"
+        deleteProperty(_, token) {
+            if (!token) return false;
+
+            if (token.includes("=")) {
+                const [field, value] = token.trim().split(/\s*=\s*/);
+                let itemsCount = 0;
+                Object.entries(STATE).forEach(([t, s]) => {
+                    if (s[field] == value) {
+                        delete STATE[t];
+                        itemsCount += 1;
+                    }
+                });
+                return !itemsCount ? false : String(itemsCount);
+            }
+
+            delete STATE[token];
+            return true;
+        },
+
+        // SET / UPDATE / DELETE
+        // serverState["abc123"] = {name: 'alice'}; // create (stamps expirePeriod/expireTime/uuid)
+        // serverState["abc123"] = {name: 'alice2'}; // update (merges into existing)
+        // serverState["abc123"] = null; // delete, cookie-expiry trick
+        // serverState["name=alice"] = {name: 'john'}; // bulk update, merges into every entry where name == "alice"
+        set(_, token, fields) {
+            if (!token) return false;
+
+            if (!fields) {
+                delete STATE[token];
+                return true;
+            }
+
+            if (token.includes("=")) {
+                const [field, value] = token.trim().split(/\s*=\s*/);
+                let itemsCount = 0;
+                Object.entries(STATE).forEach(([t, s]) => {
+                    if (s[field] == value) {
+                        STATE[t] = { ...s, ...fields };
+                        itemsCount += 1;
+                    }
+                });
+                return !itemsCount ? false : String(itemsCount);
+            }
+
+            const now = Date.now();
+
+            if (STATE[token]) {
+                if (fields.expirePeriod) fields.expireTime = now + fields.expirePeriod;
+                STATE[token] = { ...STATE[token], ...fields };
+                return { ...STATE[token] };
+            }
+
+            fields.expirePeriod ||= 60000 * 60 * 24 * 7; // 7 days
+            fields.expireTime = now + fields.expirePeriod;
+            fields.uuid = createToken({ type: "date" });
+
+            STATE[token] = fields;
+            return { ...fields };
+        },
+    },
+);
+
 // session C.R.U.D.
 exports.session = {
     set(payload = {}) {
         // payload.set => {token: 23456546, whatever: 'someValue', whatever_2: 'someValue'}
         if (!payload.token) return { error: "missing_token" };
 
+        payload = { ...payload };
         const token = payload.token;
         delete payload.token;
 
@@ -64,6 +174,7 @@ exports.session = {
 
     update(payload = {}) {
         // payload => {id: 23456546, whatever: 'someValue', whatever_2: 'someValue'}
+        payload = { ...payload };
         const _key = Object.keys(payload)[0];
         const _value = Object.values(payload)[0];
         delete payload[_key];
@@ -95,9 +206,6 @@ exports.session = {
     },
 };
 
-// -------- check or get ban users --------
-const banListFile = `${__basedir}/state/banList.json`;
-const BAN_LIST = existsSync(banListFile) ? JSON.parse(readFileSync(banListFile, "utf8") || "{}") : {};
 exports.banList = {
     isBanned(payload) {
         // payload = {ip: 123.67.432.43}
@@ -106,7 +214,7 @@ exports.banList = {
         const _key = Object.keys(payload)[0];
         const _value = Object.values(payload)[0];
 
-        BAN_LIST[_key]?.includes(_value);
+        return BAN_LIST[_key]?.includes(_value);
     },
     banUser(payload) {
         // payload = {whatever: 'someValue'}
@@ -117,17 +225,17 @@ exports.banList = {
         if (!BAN_LIST[_key]) BAN_LIST[_key] = [];
 
         BAN_LIST[_key].push(_value);
-        writeFileSync(banListFile, JSON.stringify(BAN_LIST));
+        writeFile(banListFile, JSON.stringify(BAN_LIST)).catch((error) => console.error("banList save failed:", error));
     },
     unBanUser(payload) {
         // payload = {whatever: 'someValue'}
         const _key = Object.keys(payload)[0];
         const _value = Object.values(payload)[0];
 
-        const index = BAN_LIST[_key]?.indexOf(_value) || -1;
+        const index = BAN_LIST[_key]?.indexOf(_value) ?? -1;
         if (index > -1) {
             BAN_LIST[_key].splice(index, 1);
-            writeFileSync(banListFile, JSON.stringify(BAN_LIST));
+            writeFile(banListFile, JSON.stringify(BAN_LIST)).catch((error) => console.error("banList save failed:", error));
         }
     },
 };