pi-web 0.12.0 → 0.12.1

package/build/server/server.js

@@ -1,4 +1,5 @@
 import { createServer } from 'node:http';
+import { randomUUID } from 'node:crypto';
 import { createReadStream, existsSync, readFileSync, statSync, unlinkSync } from 'node:fs';
 import { readdir } from 'node:fs/promises';
 import { basename, dirname, extname, isAbsolute, join, normalize, relative, resolve } from 'node:path';
@@ -210,8 +211,19 @@ const server = createServer((req, res) => {
 const wss = new WebSocketServer({ server });
 const rpcSessions = new Map();
 const socketBindings = new Map();
-function buildSessionKey(cwd, sessionFile) {
-    return `${cwd}::${sessionFile ? basename(sessionFile) : '__new__'}`;
+const socketClientIds = new WeakMap();
+function getSocketClientId(ws) {
+    const existing = socketClientIds.get(ws);
+    if (existing)
+        return existing;
+    const next = randomUUID();
+    socketClientIds.set(ws, next);
+    return next;
+}
+function buildSessionKey(cwd, sessionFile, scope) {
+    if (sessionFile)
+        return `${cwd}::${basename(sessionFile)}`;
+    return `${cwd}::__new__::${scope ?? 'shared'}`;
 }
 function getSessionRuntimeStatus(cwd, sessionFile) {
     const key = buildSessionKey(resolve(cwd), basename(sessionFile));
@@ -330,7 +342,7 @@ function updateSessionModelCapability(managed, event) {
         }
     }
 }
-function createManagedSession(cwd, sessionFile) {
+function createManagedSession(cwd, sessionFile, initialKey) {
     const sessionPath = sessionFile ? getSessionFilePath(cwd, sessionFile, AGENT) : undefined;
     let managed = null;
     const rpc = new RpcSession({
@@ -383,7 +395,7 @@ function createManagedSession(cwd, sessionFile) {
         idleCleanupTimer: null,
         isClosing: false,
     };
-    registerManagedSessionKey(managed, buildSessionKey(cwd, sessionFile));
+    registerManagedSessionKey(managed, initialKey);
     return managed;
 }
 wss.on('connection', (ws) => {
@@ -402,7 +414,7 @@ wss.on('connection', (ws) => {
             const sessionFile = typeof msg.sessionFile === 'string' && msg.sessionFile.length > 0
                 ? basename(msg.sessionFile)
                 : null;
-            const key = buildSessionKey(cwd, sessionFile);
+            const key = buildSessionKey(cwd, sessionFile, sessionFile ? undefined : getSocketClientId(ws));
             const currentlyBound = socketBindings.get(ws);
             if (currentlyBound?.keys.has(key)) {
                 clearIdleCleanupTimer(currentlyBound);
@@ -411,7 +423,7 @@ wss.on('connection', (ws) => {
             detachSocket(ws);
             let managed = rpcSessions.get(key);
             if (!managed || managed.isClosing)
-                managed = createManagedSession(cwd, sessionFile);
+                managed = createManagedSession(cwd, sessionFile, key);
             managed.clients.add(ws);
             clearIdleCleanupTimer(managed);
             socketBindings.set(ws, managed);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-web",
-  "version": "0.12.0",
+  "version": "0.12.1",
   "type": "module",
   "engines": {
     "node": "^24"