clay-server 2.13.0-beta.4 → 2.13.0-beta.6

package/lib/daemon.js

@@ -936,16 +936,40 @@ for (var i = 0; i < projects.length; i++) {
   }
 }
 
+// Migrate legacy mates storage before registration
+mates.migrateLegacyMates();
+
 // Register existing mates as projects
-var allMates = mates.getAllMates();
-for (var mi = 0; mi < allMates.length; mi++) {
-  var m = allMates[mi];
-  var mateDir = path.join(mates.MATES_DIR, m.id);
-  var mateSlug = "mate-" + m.id;
-  var mateName = (m.profile && m.profile.displayName) || m.name || "New Mate";
-  if (fs.existsSync(mateDir)) {
-    console.log("[daemon] Adding mate project:", mateSlug);
-    relay.addProject(mateDir, mateSlug, mateName, null, m.createdBy, null, { isMate: true });
+if (usersModule.isMultiUser()) {
+  // Multi-user mode: iterate over all users and load each user's mates
+  var allUsers = usersModule.getAllUsers();
+  for (var ui = 0; ui < allUsers.length; ui++) {
+    var mateCtx = mates.buildMateCtx(allUsers[ui].id);
+    var userMates = mates.getAllMates(mateCtx);
+    for (var mi = 0; mi < userMates.length; mi++) {
+      var m = userMates[mi];
+      var mateDir = mates.getMateDir(mateCtx, m.id);
+      var mateSlug = "mate-" + m.id;
+      var mateName = (m.profile && m.profile.displayName) || m.name || "New Mate";
+      if (fs.existsSync(mateDir)) {
+        console.log("[daemon] Adding mate project:", mateSlug);
+        relay.addProject(mateDir, mateSlug, mateName, null, m.createdBy, null, { isMate: true });
+      }
+    }
+  }
+} else {
+  // Single-user mode: use null ctx
+  var mateCtx = mates.buildMateCtx(null);
+  var allMates = mates.getAllMates(mateCtx);
+  for (var mi = 0; mi < allMates.length; mi++) {
+    var m = allMates[mi];
+    var mateDir = mates.getMateDir(mateCtx, m.id);
+    var mateSlug = "mate-" + m.id;
+    var mateName = (m.profile && m.profile.displayName) || m.name || "New Mate";
+    if (fs.existsSync(mateDir)) {
+      console.log("[daemon] Adding mate project:", mateSlug);
+      relay.addProject(mateDir, mateSlug, mateName, null, m.createdBy, null, { isMate: true });
+    }
   }
 }
 

package/lib/dm.js

@@ -118,10 +118,10 @@ function getDmList(userId) {
   return dms;
 }
 
-// Check if a user is a mate (AI persona)
+// Check if a user ID looks like a mate (AI persona) by format only
 function isMate(userId) {
   var mates = require("./mates");
-  return mates.isMate(userId);
+  return mates.isMateIdFormat(userId);
 }
 
 module.exports = {

package/lib/mates.js

@@ -3,8 +3,40 @@ var path = require("path");
 var crypto = require("crypto");
 var config = require("./config");
 
-var MATES_FILE = path.join(config.CONFIG_DIR, "mates.json");
-var MATES_DIR = path.join(process.cwd(), ".claude", "mates");
+// --- Path resolution ---
+
+function resolveMatesRoot(ctx) {
+  // OS-users mode: per-linuxUser home directory
+  if (ctx && ctx.linuxUser) {
+    return path.join("/home", ctx.linuxUser, ".clay", "mates");
+  }
+  // Multi-user mode: per-userId subdirectory
+  if (ctx && ctx.multiUser && ctx.userId) {
+    return path.join(config.CONFIG_DIR, "mates", ctx.userId);
+  }
+  // Single-user mode: flat directory
+  return path.join(config.CONFIG_DIR, "mates");
+}
+
+function buildMateCtx(userId) {
+  if (!userId) return { userId: null, multiUser: false, linuxUser: null };
+  // Lazy require to avoid circular dependency
+  var users = require("./users");
+  var multiUser = users.isMultiUser();
+  var linuxUser = null;
+  if (multiUser && userId) {
+    var user = users.findUserById(userId);
+    if (user && user.linuxUser) {
+      linuxUser = user.linuxUser;
+    }
+  }
+  return { userId: userId, multiUser: multiUser, linuxUser: linuxUser };
+}
+
+function isMateIdFormat(id) {
+  if (!id) return false;
+  return typeof id === "string" && id.indexOf("mate_") === 0;
+}
 
 // --- Default data ---
 
@@ -14,9 +46,13 @@ function defaultData() {
 
 // --- Load / Save ---
 
-function loadMates() {
+function matesFilePath(ctx) {
+  return path.join(resolveMatesRoot(ctx), "mates.json");
+}
+
+function loadMates(ctx) {
   try {
-    var raw = fs.readFileSync(MATES_FILE, "utf8");
+    var raw = fs.readFileSync(matesFilePath(ctx), "utf8");
     var data = JSON.parse(raw);
     if (!data.mates) data.mates = [];
     return data;
@@ -25,11 +61,12 @@ function loadMates() {
   }
 }
 
-function saveMates(data) {
-  fs.mkdirSync(path.dirname(MATES_FILE), { recursive: true });
-  var tmpPath = MATES_FILE + ".tmp";
+function saveMates(ctx, data) {
+  var filePath = matesFilePath(ctx);
+  fs.mkdirSync(path.dirname(filePath), { recursive: true });
+  var tmpPath = filePath + ".tmp";
   fs.writeFileSync(tmpPath, JSON.stringify(data, null, 2));
-  fs.renameSync(tmpPath, MATES_FILE);
+  fs.renameSync(tmpPath, filePath);
 }
 
 // --- CRUD ---
@@ -38,9 +75,10 @@ function generateMateId() {
   return "mate_" + crypto.randomUUID();
 }
 
-function createMate(seedData, userId) {
-  var data = loadMates();
+function createMate(ctx, seedData) {
+  var data = loadMates(ctx);
   var id = generateMateId();
+  var userId = ctx ? ctx.userId : null;
 
   // Pick a random avatar color from a pleasant palette
   var colors = ["#6c5ce7", "#00b894", "#e17055", "#0984e3", "#fdcb6e", "#e84393", "#00cec9", "#ff7675"];
@@ -63,10 +101,10 @@ function createMate(seedData, userId) {
   };
 
   data.mates.push(mate);
-  saveMates(data);
+  saveMates(ctx, data);
 
   // Create the mate's identity directory
-  var mateDir = path.join(MATES_DIR, id);
+  var mateDir = getMateDir(ctx, id);
   fs.mkdirSync(mateDir, { recursive: true });
 
   // Write initial mate.yaml
@@ -98,40 +136,40 @@ function createMate(seedData, userId) {
   return mate;
 }
 
-function getMate(id) {
-  var data = loadMates();
+function getMate(ctx, id) {
+  var data = loadMates(ctx);
   for (var i = 0; i < data.mates.length; i++) {
     if (data.mates[i].id === id) return data.mates[i];
   }
   return null;
 }
 
-function updateMate(id, updates) {
-  var data = loadMates();
+function updateMate(ctx, id, updates) {
+  var data = loadMates(ctx);
   for (var i = 0; i < data.mates.length; i++) {
     if (data.mates[i].id === id) {
       var keys = Object.keys(updates);
       for (var j = 0; j < keys.length; j++) {
         data.mates[i][keys[j]] = updates[keys[j]];
       }
-      saveMates(data);
+      saveMates(ctx, data);
       return data.mates[i];
     }
   }
   return null;
 }
 
-function deleteMate(id) {
-  var data = loadMates();
+function deleteMate(ctx, id) {
+  var data = loadMates(ctx);
   var before = data.mates.length;
   data.mates = data.mates.filter(function (m) {
     return m.id !== id;
   });
   if (data.mates.length === before) return { error: "Mate not found" };
-  saveMates(data);
+  saveMates(ctx, data);
 
   // Remove mate directory
-  var mateDir = path.join(MATES_DIR, id);
+  var mateDir = getMateDir(ctx, id);
   try {
     fs.rmSync(mateDir, { recursive: true, force: true });
   } catch (e) {
@@ -141,29 +179,86 @@ function deleteMate(id) {
   return { ok: true };
 }
 
-function getAllMates() {
-  var data = loadMates();
+function getAllMates(ctx) {
+  var data = loadMates(ctx);
   return data.mates;
 }
 
-function getMatesByUser(userId) {
-  var data = loadMates();
-  return data.mates.filter(function (m) {
-    return m.createdBy === userId;
-  });
-}
-
-function isMate(id) {
+function isMate(ctx, id) {
   if (!id) return false;
   if (typeof id === "string" && id.indexOf("mate_") === 0) {
     // Double check it exists in registry
-    return !!getMate(id);
+    return !!getMate(ctx, id);
   }
   return false;
 }
 
-function getMateDir(id) {
-  return path.join(MATES_DIR, id);
+function getMateDir(ctx, id) {
+  return path.join(resolveMatesRoot(ctx), id);
+}
+
+// --- Migration ---
+
+function migrateLegacyMates() {
+  var legacyFile = path.join(config.CONFIG_DIR, "mates.json");
+  if (!fs.existsSync(legacyFile)) return;
+
+  // Check if already migrated
+  var migratedMarker = legacyFile + ".migrated";
+  if (fs.existsSync(migratedMarker)) return;
+
+  try {
+    var raw = fs.readFileSync(legacyFile, "utf8");
+    var data = JSON.parse(raw);
+    if (!data.mates || data.mates.length === 0) {
+      // Nothing to migrate, just mark as done
+      fs.renameSync(legacyFile, migratedMarker);
+      return;
+    }
+
+    // Group mates by createdBy
+    var byUser = {};
+    for (var i = 0; i < data.mates.length; i++) {
+      var m = data.mates[i];
+      var key = m.createdBy || "__null__";
+      if (!byUser[key]) byUser[key] = [];
+      byUser[key].push(m);
+    }
+
+    // Write each user's mates to their own storage path
+    var keys = Object.keys(byUser);
+    for (var k = 0; k < keys.length; k++) {
+      var userId = keys[k] === "__null__" ? null : keys[k];
+      var ctx = buildMateCtx(userId);
+      var userData = { mates: byUser[keys[k]] };
+      saveMates(ctx, userData);
+
+      // Move mate identity directories to new location
+      var legacyMatesDir = path.join(config.CONFIG_DIR, "mates");
+      var newRoot = resolveMatesRoot(ctx);
+      for (var mi = 0; mi < byUser[keys[k]].length; mi++) {
+        var mateId = byUser[keys[k]][mi].id;
+        var oldDir = path.join(legacyMatesDir, mateId);
+        var newDir = path.join(newRoot, mateId);
+        if (fs.existsSync(oldDir) && oldDir !== newDir) {
+          fs.mkdirSync(path.dirname(newDir), { recursive: true });
+          try {
+            fs.renameSync(oldDir, newDir);
+          } catch (e) {
+            // Cross-device or other issue, copy instead
+            fs.cpSync(oldDir, newDir, { recursive: true });
+            fs.rmSync(oldDir, { recursive: true, force: true });
+          }
+        }
+      }
+    }
+
+    // Mark legacy file as migrated
+    fs.renameSync(legacyFile, migratedMarker);
+    console.log("[mates] Migrated legacy mates.json to per-user storage");
+  } catch (e) {
+    console.error("[mates] Legacy migration failed:", e.message);
+  }
 }
 
 // Format seed data as a human-readable context string
@@ -206,8 +301,9 @@ function formatSeedContext(seedData) {
 }
 
 module.exports = {
-  MATES_FILE: MATES_FILE,
-  MATES_DIR: MATES_DIR,
+  resolveMatesRoot: resolveMatesRoot,
+  buildMateCtx: buildMateCtx,
+  isMateIdFormat: isMateIdFormat,
   loadMates: loadMates,
   saveMates: saveMates,
   createMate: createMate,
@@ -215,8 +311,8 @@ module.exports = {
   updateMate: updateMate,
   deleteMate: deleteMate,
   getAllMates: getAllMates,
-  getMatesByUser: getMatesByUser,
   isMate: isMate,
   getMateDir: getMateDir,
+  migrateLegacyMates: migrateLegacyMates,
   formatSeedContext: formatSeedContext,
 };

package/lib/project.js

@@ -626,13 +626,12 @@ function createProjectContext(opts) {
         console.log("[loop-registry] Schedule triggered: " + record.name + " → linked task " + loopId);
       }
 
-      // Verify the loop directory and files exist
+      // Verify the loop directory and PROMPT.md exist
       var recDir = path.join(cwd, ".claude", "loops", loopId);
       try {
         fs.accessSync(path.join(recDir, "PROMPT.md"));
-        fs.accessSync(path.join(recDir, "JUDGE.md"));
       } catch (e) {
-        console.error("[loop-registry] Loop files missing for " + loopId);
+        console.error("[loop-registry] PROMPT.md missing for " + loopId);
         return;
       }
       // Set the loopId and start
@@ -675,8 +674,7 @@ function createProjectContext(opts) {
     try {
       judgeText = fs.readFileSync(judgePath, "utf8");
     } catch (e) {
-      send({ type: "loop_error", text: "Missing JUDGE.md in " + dir });
-      return;
+      judgeText = null;
     }
 
     var baseCommit;
@@ -700,7 +698,7 @@ function createProjectContext(opts) {
     loopState.promptText = promptText;
     loopState.judgeText = judgeText;
     loopState.iteration = 0;
-    loopState.maxIterations = loopConfig.maxIterations || loopOpts.maxIterations || 20;
+    loopState.maxIterations = judgeText ? (loopConfig.maxIterations || loopOpts.maxIterations || 20) : 1;
     loopState.baseCommit = baseCommit;
     loopState.currentSessionId = null;
     loopState.judgeSessionId = null;
@@ -778,7 +776,11 @@ function createProjectContext(opts) {
         setTimeout(function() { runNextIteration(); }, 2000);
         return;
       }
-      runJudge();
+      if (loopState.judgeText) {
+        runJudge();
+      } else {
+        finishLoop("pass");
+      }
     };
 
     // Watchdog: if onQueryComplete hasn't fired after 10 minutes, force error and retry
@@ -2975,6 +2977,11 @@ function createProjectContext(opts) {
           var tmpJudge = judgePath + ".tmp";
           fs.writeFileSync(tmpJudge, wData.judgeText);
           fs.renameSync(tmpJudge, judgePath);
+        } else {
+          // No judge: force single iteration
+          var singleJson = loopJsonPath + ".tmp2";
+          fs.writeFileSync(singleJson, JSON.stringify({ maxIterations: 1 }, null, 2));
+          fs.renameSync(singleJson, loopJsonPath);
         }
 
         // Go straight to approval (no crafting needed)
@@ -3190,9 +3197,8 @@ function createProjectContext(opts) {
       var rerunDir = path.join(cwd, ".claude", "loops", rerunRec.id);
       try {
         fs.accessSync(path.join(rerunDir, "PROMPT.md"));
-        fs.accessSync(path.join(rerunDir, "JUDGE.md"));
       } catch (e) {
-        sendTo(ws, { type: "loop_registry_error", text: "Loop files missing for " + rerunRec.id });
+        sendTo(ws, { type: "loop_registry_error", text: "PROMPT.md missing for " + rerunRec.id });
         return;
       }
       loopState.loopId = rerunRec.id;

package/lib/public/app.js

@@ -859,6 +859,10 @@ import { initCommandPalette, handlePaletteSessionSwitch, setPaletteVersion } fro
       mateWs = null;
     }
     mateProjectSlug = null;
+    // If main WS was disconnected while in mate DM, reconnect now
+    if (ws && ws.readyState !== 1) {
+      connect();
+    }
   }
 
   function appendDmMessage(msg) {
@@ -3122,9 +3126,9 @@ import { initCommandPalette, handlePaletteSessionSwitch, setPaletteVersion } fro
     ws = new WebSocket(protocol + "//" + location.host + wsPath);
 
 
-    // If not connected within 3s, force retry
+    // If not connected within 3s, force retry (skip if stashed during mate DM)
     connectTimeoutId = setTimeout(function () {
-      if (!connected) {
+      if (!connected && !savedMainWs) {
         ws.onclose = null;
         ws.onerror = null;
         ws.close();
@@ -3183,6 +3187,9 @@ import { initCommandPalette, handlePaletteSessionSwitch, setPaletteVersion } fro
     };
 
     ws.onclose = function (e) {
+      // If this WS is stashed while in mate DM, ignore close events
+      if (savedMainWs === this) return;
+
       if (connectTimeoutId) { clearTimeout(connectTimeoutId); connectTimeoutId = null; }
       setStatus("disconnected");
       processing = false;
@@ -3208,9 +3215,14 @@ import { initCommandPalette, handlePaletteSessionSwitch, setPaletteVersion } fro
     };
 
     ws.onerror = function () {
+      // If this WS is stashed while in mate DM, ignore error events
+      if (savedMainWs === this) return;
     };
 
     ws.onmessage = function (event) {
+      // If this WS is stashed while in mate DM, ignore its messages
+      if (savedMainWs === this) return;
+
       // Backup: if we're receiving messages, we're connected
       if (!connected) {
         setStatus("connected");
@@ -4372,6 +4384,8 @@ import { initCommandPalette, handlePaletteSessionSwitch, setPaletteVersion } fro
 
   function scheduleReconnect() {
     if (reconnectTimer) return;
+    // Don't reconnect main WS while in mate DM
+    if (savedMainWs) return;
     reconnectTimer = setTimeout(function () {
       reconnectTimer = null;
       // Check if auth is still valid before reconnecting

package/lib/public/modules/tools.js

@@ -242,9 +242,6 @@ export function renderAskUserQuestion(toolId, input) {
           answers[qIdx] = opt.label;
           var otherInput = qDiv.querySelector(".ask-user-other input");
           if (otherInput) otherInput.value = "";
-          if (questions.length === 1) {
-            submitAskUserAnswer(container, toolId, questions, answers, multiSelections);
-          }
         }
       });
 
@@ -280,17 +277,14 @@ export function renderAskUserQuestion(toolId, input) {
     container.appendChild(qDiv);
   });
 
-  // Submit button: show for multi-question or any multi-select question
-  var hasMultiSelect = questions.some(function (q) { return q.multiSelect; });
-  if (questions.length > 1 || hasMultiSelect) {
-    var submitBtn = document.createElement("button");
-    submitBtn.className = "ask-user-submit";
-    submitBtn.textContent = "Submit";
-    submitBtn.addEventListener("click", function () {
-      submitAskUserAnswer(container, toolId, questions, answers, multiSelections);
-    });
-    container.appendChild(submitBtn);
-  }
+  // Submit button: always show
+  var submitBtn = document.createElement("button");
+  submitBtn.className = "ask-user-submit";
+  submitBtn.textContent = "Submit";
+  submitBtn.addEventListener("click", function () {
+    submitAskUserAnswer(container, toolId, questions, answers, multiSelections);
+  });
+  container.appendChild(submitBtn);
 
   // Skip button
   var skipBtn = document.createElement("button");

package/lib/server.js

@@ -2614,7 +2614,8 @@ function createServer(opts) {
         }
       }
       // Include mates in the list
-      var mateList = mates.getAllMates();
+      var mateCtx = mates.buildMateCtx(userId);
+      var mateList = mates.getAllMates(mateCtx);
       ws.send(JSON.stringify({ type: "dm_list", dms: dmList, mates: mateList }));
       return;
     }
@@ -2623,16 +2624,25 @@ function createServer(opts) {
       if (!msg.targetUserId) return;
 
       // Check if target is a mate
-      if (mates.isMate(msg.targetUserId)) {
-        var mate = mates.getMate(msg.targetUserId);
+      var mateCtx2 = mates.buildMateCtx(userId);
+      if (mates.isMate(mateCtx2, msg.targetUserId)) {
+        var mate = mates.getMate(mateCtx2, msg.targetUserId);
         if (!mate) return;
+        // Ensure mate project is registered (survives server restarts)
+        var mateSlug2 = "mate-" + mate.id;
+        if (!projects.has(mateSlug2)) {
+          var mateDir2 = mates.getMateDir(mateCtx2, mate.id);
+          fs.mkdirSync(mateDir2, { recursive: true });
+          var mateName2 = (mate.profile && mate.profile.displayName) || mate.name || "New Mate";
+          addProject(mateDir2, mateSlug2, mateName2, null, mate.createdBy || userId, null, { isMate: true });
+        }
         var mp = mate.profile || {};
         ws.send(JSON.stringify({
           type: "dm_history",
           dmKey: "mate:" + mate.id,
           messages: dm.loadHistory("mate:" + mate.id),
           isMate: true,
-          projectSlug: "mate-" + mate.id,
+          projectSlug: mateSlug2,
           targetUser: {
             id: mate.id,
             displayName: mp.displayName || mate.name || "New Mate",
@@ -2690,8 +2700,9 @@ function createServer(opts) {
       var parts = msg.dmKey.split(":");
 
       // Handle mate DM: dmKey is "mate:mate_xxx"
-      if (parts[0] === "mate" && mates.isMate(parts[1])) {
-        var mate = mates.getMate(parts[1]);
+      var mateCtx3 = mates.buildMateCtx(userId);
+      if (parts[0] === "mate" && mates.isMate(mateCtx3, parts[1])) {
+        var mate = mates.getMate(mateCtx3, parts[1]);
         if (!mate) return;
         // Verify sender is the mate's creator
         if (mate.createdBy !== userId) return;
@@ -2758,9 +2769,10 @@ function createServer(opts) {
     if (msg.type === "mate_create") {
       if (!msg.seedData) return;
       try {
-        var mate = mates.createMate(msg.seedData, userId);
+        var mateCtx4 = mates.buildMateCtx(userId);
+        var mate = mates.createMate(mateCtx4, msg.seedData);
         // Register mate as a project
-        var mateDir = path.join(mates.MATES_DIR, mate.id);
+        var mateDir = mates.getMateDir(mateCtx4, mate.id);
         var mateSlug = "mate-" + mate.id;
         var mateName = (mate.profile && mate.profile.displayName) || mate.name || "New Mate";
         addProject(mateDir, mateSlug, mateName, null, mate.createdBy, null, { isMate: true });
@@ -2772,14 +2784,27 @@ function createServer(opts) {
     }
 
     if (msg.type === "mate_list") {
-      var mateList = mates.getAllMates();
+      var mateCtx5 = mates.buildMateCtx(userId);
+      var mateList = mates.getAllMates(mateCtx5);
+      // Ensure all mate projects are registered (survives server restarts)
+      for (var mi = 0; mi < mateList.length; mi++) {
+        var m = mateList[mi];
+        var mSlug = "mate-" + m.id;
+        if (!projects.has(mSlug)) {
+          var mDir = mates.getMateDir(mateCtx5, m.id);
+          fs.mkdirSync(mDir, { recursive: true });
+          var mName = (m.profile && m.profile.displayName) || m.name || "New Mate";
+          addProject(mDir, mSlug, mName, null, m.createdBy || userId, null, { isMate: true });
+        }
+      }
       ws.send(JSON.stringify({ type: "mate_list", mates: mateList }));
       return;
     }
 
     if (msg.type === "mate_delete") {
       if (!msg.mateId) return;
-      var result = mates.deleteMate(msg.mateId);
+      var mateCtx6 = mates.buildMateCtx(userId);
+      var result = mates.deleteMate(mateCtx6, msg.mateId);
       if (result.error) {
         ws.send(JSON.stringify({ type: "mate_error", error: result.error }));
       } else {
@@ -2799,7 +2824,8 @@ function createServer(opts) {
 
     if (msg.type === "mate_update") {
       if (!msg.mateId || !msg.updates) return;
-      var updated = mates.updateMate(msg.mateId, msg.updates);
+      var mateCtx7 = mates.buildMateCtx(userId);
+      var updated = mates.updateMate(mateCtx7, msg.mateId, msg.updates);
       if (updated) {
         ws.send(JSON.stringify({ type: "mate_updated", mate: updated }));
         // Broadcast update

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "clay-server",
-  "version": "2.13.0-beta.4",
+  "version": "2.13.0-beta.6",
   "description": "Web UI for Claude Code. Any device. Push notifications.",
   "bin": {
     "clay-server": "./bin/cli.js",