tune-sdk 0.3.4 → 0.3.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tune-sdk",
-  "version": "0.3.4",
+  "version": "0.3.5",
   "description": "chat with llm in a text file. core package",
   "main": "dist/tune.js",
   "module": "dist/tune.mjs",

package/src/contextws.js

@@ -72,7 +72,6 @@ function ContextWebsocket(url, { debug } = {}) {
   assert(typeof url !== 'undefined', 'url is not set');
 
   socket = new WebSocket(url);
-  console.log(socket)
 
   this.socket = socket;
   this.callbacks = Object.create(null);
@@ -105,7 +104,7 @@ ContextWebsocket.prototype._onClose = function () {
 
 ContextWebsocket.prototype._send = async function (payload) {
   const data = JSON.stringify(payload);
-  this.debug('==>', data);
+  this.debug('tune ==>', data);
   await this.ready;
   this.socket.send(data);
 };
@@ -135,7 +134,7 @@ ContextWebsocket.prototype._onMessage = function (event) {
     return;
   }
 
-  this.debug('<==', msg);
+  this.debug('tune <==', msg);
   if (!msg || !msg.id) return;
 
   const cb = this.callbacks[msg.id];
@@ -176,6 +175,7 @@ ContextWebsocket.prototype.read = function (name, binary) {
   return this._call('read', [name, binary], false);
 };
 
+
 ContextWebsocket.prototype.write = function (name, content) {
   return this._call('write', [name, content], false);
 };
@@ -197,5 +197,12 @@ if (typeof module !== 'undefined' && module.exports) {
   module.exports = ContextWebsocket;
 } else {
   window.ContextWebsocket = ContextWebsocket;
-  window.ctx = new ContextWebsocket(window.location.origin.replace(/^http/, 'ws'));
+  window.ctx = new ContextWebsocket(window.location.origin.replace(/^http/, 'ws'), { debug: console.debug });
+  window.addEventListener('error', async (event) => {
+    window.ctx._call('log', [event.error?.stack || event.message ], false);
+  })
+  window.addEventListener('unhandledrejection', async (event) => {
+    const err = event.reason instanceof Error ? event.reason : new Error(String(event.reason));
+    window.ctx._call('log', [err.stack || err.message], false);
+  });
 }

package/src/ws.js

@@ -1,9 +1,27 @@
 const http = require('http');
 const fs = require('fs');
 const path = require('path');
+const os = require('os');
 const WebSocket = require('ws');
 const mime = require('mime-types');
 
+// skips loopback, docker (172.x, 192.168.x docker ranges), and common vpn (tun/tap) interfaces
+function getPublicIPv4() {
+  const ifaces = os.networkInterfaces();
+  const skipNames = /^(docker|br-|veth|tun|tap|vmnet|vbox)/i;
+  const skipRanges = /^(172\.(1[6-9]|2\d|3[01])\.|10\.)/; // docker bridge ranges
+
+  for (const [name, addrs] of Object.entries(ifaces)) {
+    if (skipNames.test(name)) continue;
+    for (const addr of addrs) {
+      if (addr.family === 'IPv4' && !addr.internal && !skipRanges.test(addr.address)) {
+        return addr.address;
+      }
+    }
+  }
+  return 'localhost';
+}
+
 function makeServer({ port, static, root, ctx }) {
   root = path.join(root || process.cwd());
 
@@ -46,6 +64,7 @@ function makeServer({ port, static, root, ctx }) {
       let lctx = ctx.clone()
       let { id, method, args, ...rest } = JSON.parse(msg.toString());
       let result = { id }
+      console.log("[context]", method, ...args)
       if (method === "read") {
         const [ name, binary ] = args
         let content = await lctx.read(name, binary)
@@ -104,6 +123,9 @@ function makeServer({ port, static, root, ctx }) {
             }
           })()
         }
+      } else if (method === "log") {
+        console.log("[browser]:", ...args)
+        result.done = true
       } else {
         result.error = { message: `method not found ${method}`}
       }
@@ -115,7 +137,8 @@ function makeServer({ port, static, root, ctx }) {
     if (Number.isNaN(parseInt(port))) {
       console.log(`listening ${port}`);
     } else {
-      console.log(`listening http://localhost:${port}`);
+      const host = getPublicIPv4();
+      console.log(`listening http://${host}:${port}`);
     }
   });
 }