jsir 2.5.1 → 2.5.3

package/cmd/oaa.js

@@ -645,7 +645,7 @@ async function _wrapperInput(str) {
                 if (strs[1] === '-') {
                     await offServer(uniqueName)
                 } else {
-                    await joinServer(uniqueName)
+                    await joinServer(uniqueName, strs[1] === '+')
                 }
             } else {
                 console.log(String(fs.readFileSync(path)));
@@ -656,7 +656,7 @@ async function _wrapperInput(str) {
     }
 }
 
-async function joinServer(uniqueName) {
+async function joinServer(uniqueName, isApi) {
     let exportLib = await _requireSource(setting.defaultSpace, uniqueName, true);
     let asyncFnSize = 0
     for (let key of Object.keys(exportLib)) {
@@ -675,12 +675,12 @@ async function joinServer(uniqueName) {
         let val = exportLib[key];
         Server.setRoute('get', `/${pair[0]}/${trimJsirFileName(pair[1])}/${key}`,async (req, res) => {
             res.result = await val(...req);
-        })
+        }, isApi)
         Server.setRoute('post', `/${pair[0]}/${trimJsirFileName(pair[1])}/${key}`,async (req, res) => {
             res.result = await val(...req);
-        })
+        }, isApi)
     }
-    console.msg(uniqueName, 'has become a service.');
+    console.msg(uniqueName, `has become a ${isApi ? "api ":""}service.`);
 }
 
 async function offServer(uniqueName) {

package/deps/room.js

@@ -54,7 +54,6 @@ async function localConfigs() {
 function onRoom(wrapperInput) {
     if (!setting.roomTid[0]) {
         try {
-            setRoute("post", "/", (req, res) => setting.selfRoom)
             setRoute("get", "/", (req, res) => setting.selfRoom)
             setRoute("post", "/enter", async (req, res) => {
                 if (!getConfig("enableRemote", true)) {
@@ -114,44 +113,45 @@ function offRoom() {
 function getSelfIP(nodes) {
     let ips = getLocalIPs().ipv4;
     ips = ips.filter(i => nodes.indexOf(i) !== -1);
-    return ips[0]
+    return ips[0] || '127.0.0.1'
 }
 
 async function getTailscaleNodes() {
     let resp = await e(`${tailscalePath} status`);
     let nodes = {}
     for (let line of resp.split("\n").map(trim)) {
-        if (line.indexOf('offline') !== -1) {
-            continue
-        }
+        let offline = line.indexOf('offline') !== -1
         let ss = line.split(/\s+/);
         if (/^\d+\.\d+\.\d+\.\d+$/.test(ss[0])) {
-            nodes[ss[0]] = ss[1]
+            nodes[ss[0]] = {
+                name: ss[1],
+                offline
+            }
         }
     }
     return nodes;
 }
 
 async function updateRoomInfo() {
-    let nodeMap = await getTailscaleNodes();
-    let nodes = Object.keys(nodeMap)
+    setting.nodeMap = await getTailscaleNodes();
+    let nodes = Object.keys(setting.nodeMap)
     let ip = getSelfIP(nodes)
     await fileJson(roomDataFile, async room => {
+        room = {
+            jsirs: room.jsirs
+        }
         // 设置roomName
-        let name = nodeMap[ip] || os.hostname();
+        let name = setting.nodeMap[ip]?.name || os.hostname();
         if (!vl(room.name) || room.name !== name) {
             room.name = name;
             debug("set roomName", name)
         }
         room.selfNode = ip;
-        room.nodes = nodes;
         room.lastUpdateTime = Date.now()
         debug("init room", room.name)
 
-        // 提前设置一下
-        setting.selfRoom = Object.assign(setting.selfRoom || {}, room);
-
         await cleanRoom(room)
+        return room;
     })
 }
 
@@ -161,10 +161,13 @@ async function syncRooms() {
 
     let syncRooms = []
     let fns = []
-    for (let node of room.nodes) {
+    for (let node of Object.keys(setting.nodeMap)) {
         if (node === room.selfNode) {
             continue
         }
+        if (setting.nodeMap[node]?.offline) {
+            continue
+        }
         fns.push(async () => {
             let respBody = ''
             try {
@@ -196,9 +199,8 @@ async function syncRooms() {
 
     // 清理 roomsDir
     let files = await fp.readdir(roomsDir)
-    let nodes = room.nodes;
     for (let file of files) {
-        if (nodes.indexOf(file) === -1) {
+        if (!setting.nodeMap[file]) {
             try {
                 await fp.unlink(roomsDir + '/' + file)
             } catch (e) {
@@ -219,7 +221,7 @@ async function _syncSetting(room) {
     let roomDir = getRoomsDir();
     let rooms = []
     let files = await fp.readdir(roomDir);
-    for (let node of setting.selfRoom.nodes) {
+    for (let node of Object.keys(setting.nodeMap)) {
         if (setting.selfRoom.selfNode === node) {
             continue
         }
@@ -247,6 +249,9 @@ async function _syncSetting(room) {
     setting.services = {}
     setting.serviceFns = {}
     for (const room of setting.rooms) {
+        if (setting.nodeMap[room.selfNode]?.offline) {
+            continue
+        }
         for (const pid of Object.keys(room.jsirs || {})) {
             let jsir = room.jsirs[pid];
             for (const service of Object.keys(jsir.services || {})) {
@@ -281,7 +286,7 @@ async function initRoom() {
     if (roomUpdateTouchTime) {
         await fileLock(updateRoomInfoLockKey, updateRoomInfo, false)
         fileLock(syncRoomsLockKey, syncRooms, false);
-        if (!configMain && setting.serviceFns.hasOwnProperty(setting.configMainFnKey)) {
+        if (!configMain && setting.serviceFns[setting.configMainFnKey]) {
             fileLock(syncConfigsLockKey, syncConfigs, false);
         }
     }
@@ -382,7 +387,7 @@ async function initRoomJsir(room) {
         busy: await getEventLoopDelay() || (lastJsir ? lastJsir.busy:0),
         back: isRunningInBackground(),
         lastUpdateTime: Date.now(),
-        port: setting.server ? setting.server.address().port:null,
+        port: setting.server ? setting.server.address()?.port:null,
         services: services,
         newError: global.$newError,
         version: packageJson.version,

package/deps/server.js

@@ -7,7 +7,6 @@ const preferredPort = setting.defaultPort;
 // 路由存储
 const routes = setting.routes;
 let invokeStart = false;
-const apiReg = /[^a-zA-Z]api[^a-zA-Z]|[^a-zA-Z]api$/i;
 
 function debug(...args) {
     if (global.$DEBUG) {
@@ -17,7 +16,7 @@ function debug(...args) {
 
 function verifySign(sign) {
     try {
-        let key = md5([...setting.selfRoom.nodes].sort().join(":")).substring(0, 16);
+        let key = md5(Object.keys(setting.nodeMap).sort().join(":")).substring(0, 16);
         return parseInt(aesDecipher(sign, key), 10);  // 假设sign是base64编码的毫秒数
     } catch (e) {
         return null;  // 解码失败返回null
@@ -26,7 +25,7 @@ function verifySign(sign) {
 
 function createSign() {
     try {
-        let key = md5([...setting.selfRoom.nodes].sort().join(":")).substring(0, 16);
+        let key = md5(Object.keys(setting.nodeMap).sort().join(":")).substring(0, 16);
         return aesCipher(String(Date.now()), key)
     } catch (e) {
         return null;  // 解码失败返回null
@@ -34,8 +33,9 @@ function createSign() {
 }
 
 async function process(routeKey, req, res, params) {
-    if (routes[routeKey]) {
-        if (!apiReg.test(routeKey)) {
+    let router = routes[routeKey]
+    if (router) {
+        if ("get /" !== routeKey && !router.isApi) {
             // 如果不是api接口，则校验签名
             // 取header里的sign字段
             const sign = req.headers['sign'];
@@ -70,7 +70,7 @@ async function process(routeKey, req, res, params) {
                     resolve()
                 })
             })
-            await routes[routeKey](body || params, res);
+            await router.fn(body || params, res);
         } catch (e) {
             res.writeHead(500, {'Content-Type': 'text/plain'});
             res.end(String(isError(e) ? e.message : e));
@@ -140,16 +140,22 @@ async function startServer() {
     method, url, fn(req, res, err), null
     err: {code, msg}
  */
-function setRoute(method, url, fn, handler = jsonHandle) {
+function setRoute(method, url, fn, handler = jsonHandle, isApi) {
     startServer();
 
     const routeKey = `${method.toLowerCase()} ${url}`;
     if (handler) {
-        routes[routeKey] = async (req, res) => {
-            await handler(req, res, fn)
+        routes[routeKey] = {
+            fn: async (req, res) => {
+                await handler(req, res, fn)
+            },
+            isApi
         };
     } else {
-        routes[routeKey] = fn;
+        routes[routeKey] = {
+            fn,
+            isApi
+        };
     }
     debug(`setRoute: ${method.toUpperCase()} ${url} ${handler ? handler.name:''}`);
     return setting.server;

package/deps/setting.js

@@ -22,6 +22,7 @@ module.exports = {
     workspaceMap: {},
     enableNextLine: true,
     roomTid: [],
+    nodeMap: {},
     selfRoom: {},
     rooms: [],
     server: null,

package/deps/util.js

@@ -935,21 +935,25 @@ function validStr(str, name) {
     }
 }
 
-function aesCipher(str, key){
+function aesCipher(str, key, useIv = false){
     validStr(str, "str");
     validStr(key, "key");
 
     key = trim(key);
     if (!key || key.length > 16) {
-        throw "aesCipher key length should between 1 and 16"
+        throw "aesCipher key length should be between 1 and 16"
     }
-    try {
-        const cipher = crypto.createCipheriv('aes-128-ecb', pad(key, 16, '0'), null);
-        return cipher.update(str, 'utf8', 'hex') + cipher.final('hex');
-    } catch(err) {
-        $log(err)
-        throw 'aesCipher failed: ' + String(err)
+    if (useIv) {
+        // Generate random IV
+        const iv = crypto.randomBytes(16);
+        const cipher = crypto.createCipheriv('aes-128-cbc', pad(key, 16, '0'), iv);
+
+        // Return IV and ciphertext, as IV is needed for decryption
+        const ciphertext = cipher.update(str, 'utf8', 'hex') + cipher.final('hex');
+        return iv.toString('hex') + ':' + ciphertext; // Combine IV and ciphertext
     }
+    const cipher = crypto.createCipheriv('aes-128-ecb', pad(key, 16, '0'), null);
+    return cipher.update(str, 'utf8', 'hex') + cipher.final('hex');
 }
 
 function aesDecipher(str, key){
@@ -958,15 +962,21 @@ function aesDecipher(str, key){
 
     key = trim(key);
     if (!key || key.length > 16) {
-        throw "aesCipher key length should between 1 and 16"
+        throw "aesCipher key length should be between 1 and 16";
     }
-    try {
-        const decipher = crypto.createDecipheriv('aes-128-ecb', pad(key, 16, '0'), null);
-        return decipher.update(str, 'hex', 'utf8') + decipher.final('utf8');
-    } catch(err) {
-        $log(err)
-        throw 'aesDecipher failed: ' + String(err)
+
+    // Extract IV and ciphertext
+    if(str.indexOf(":") !== -1) {
+        const [ivHex, ciphertext] = str.split(':');
+        if (!ivHex || !ciphertext) {
+            throw "Invalid encrypted string format";
+        }
+        const iv = Buffer.from(ivHex, 'hex');
+        const decipher = crypto.createDecipheriv('aes-128-cbc', pad(key, 16, '0'), iv);
+        return decipher.update(ciphertext, 'hex', 'utf8') + decipher.final('utf8');
     }
+    const decipher = crypto.createDecipheriv('aes-128-ecb', pad(key, 16, '0'), null);
+    return decipher.update(str, 'hex', 'utf8') + decipher.final('utf8');
 }
 
 function getConfig(key, defaultVal) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jsir",
-  "version": "2.5.1",
+  "version": "2.5.3",
   "description": "JavaScript Script Management Tool",
   "main": "index.js",
   "scripts": {