clay-server 3.8.1 → 4.0.0-beta.2

package/lib/public/modules/tool-runtime.js

@@ -0,0 +1,309 @@
+// tool-runtime.js - Web Worker host for user-installed tool logic.
+
+import { getWs } from './ws-ref.js';
+
+var ACTION_TIMEOUT_MS = 10000;
+var LLM_ACTION_TIMEOUT_MS = 90000;
+var LLM_TIMEOUT_MS = 60000;
+var WS_WAIT_MS = 5000;
+var WS_RETRY_MS = 250;
+
+function workerProgram(logicSource, allowLlm) {
+  return [
+    "self.fetch = undefined; self.XMLHttpRequest = undefined; self.WebSocket = undefined; self.EventSource = undefined; self.importScripts = undefined;",
+    "var tool; var currentState = {}; var storageSeq = 0; var storagePending = Object.create(null); var llmSeq = 0; var llmPending = Object.create(null); var actionQueue = Promise.resolve();",
+    "function storageCall(op, args) {",
+    "  return new Promise(function (resolve, reject) {",
+    "    storageSeq++; storagePending[storageSeq] = { resolve: resolve, reject: reject };",
+    "    self.postMessage({ type: 'storage', op: op, args: args || {}, seq: storageSeq });",
+    "  });",
+    "}",
+    "function llmCall(args, callerId) {",
+    allowLlm ? "  return new Promise(function (resolve, reject) { llmSeq++; llmPending[llmSeq] = { resolve: resolve, reject: reject }; self.postMessage({ type: 'llm', seq: llmSeq, args: args || {}, callerId: callerId || 'user' }); });" : "  return Promise.reject(new Error('This capsule does not have the llm permission.'));",
+    "}",
+    "var api = { storage: {",
+    "  list: function () { return storageCall('list', {}); },",
+    "  get: function (id) { return storageCall('get', { id: id }); },",
+    "  put: function (doc) { return storageCall('put', { doc: doc }); },",
+    "  delete: function (id) { return storageCall('delete', { id: id }); },",
+    "  query: function (query) { return storageCall('query', { query: query }); }",
+    "}, llm: { complete: function () { return Promise.reject(new Error('api.llm.complete is only available inside an action.')); } } };",
+    logicSource,
+    "if (!tool || !tool.actions || typeof tool.actions !== 'object') throw new Error('logic.js must define var tool with an actions object.');",
+    "self.onmessage = function (event) {",
+    "  var msg = event.data || {};",
+    "  if (msg.type === 'init') {",
+    "    currentState = msg.state || tool.initialState || {};",
+    "    self.postMessage({ type: 'state', newState: currentState }); return;",
+    "  }",
+    "  if (msg.type === 'storage_result') {",
+    "    var pending = storagePending[msg.seq]; if (!pending) return; delete storagePending[msg.seq];",
+    "    if (msg.error) pending.reject(new Error(msg.error)); else pending.resolve(msg.data); return;",
+    "  }",
+    "  if (msg.type === 'llm_result') {",
+    "    var llmRequest = llmPending[msg.seq]; if (!llmRequest) return; delete llmPending[msg.seq];",
+    "    if (msg.error) llmRequest.reject(new Error(msg.error)); else llmRequest.resolve(msg.data); return;",
+    "  }",
+    "  if (msg.type === 'action') {",
+    "    var action = tool.actions[msg.name];",
+    "    if (typeof action !== 'function') { self.postMessage({ type: 'error', message: 'Unknown action: ' + msg.name, actionSeq: msg.actionSeq }); return; }",
+    "    var actionStartState;",
+    "    actionQueue = actionQueue.then(function () {",
+    "      actionStartState = currentState;",
+    "      var actionApi = Object.create(api); actionApi.callerId = msg.callerId || 'user'; actionApi.llm = { complete: function (args) { return llmCall(args, actionApi.callerId); } };",
+    "      actionApi.setState = function (nextState) { if (!nextState || typeof nextState !== 'object' || Array.isArray(nextState)) throw new Error('api.setState requires a state object.'); currentState = nextState; self.postMessage({ type: 'state', newState: currentState, actionSeq: msg.actionSeq, intermediate: true }); };",
+    "      return action(currentState, msg.args || {}, actionApi);",
+    "    }).then(function (nextState) {",
+    "      if (nextState !== undefined) currentState = nextState;",
+    "      self.postMessage({ type: 'state', newState: currentState, actionSeq: msg.actionSeq });",
+    "    }).catch(function (error) { if (actionStartState !== undefined) { currentState = actionStartState; self.postMessage({ type: 'state', newState: currentState, actionSeq: msg.actionSeq, intermediate: true, rollback: true }); } self.postMessage({ type: 'error', message: error && error.message ? error.message : String(error), actionSeq: msg.actionSeq }); });",
+    "  }",
+    "};",
+  ].join("\n");
+}
+
+export function createToolRuntime(config) {
+  var worker = null;
+  var state = config.state || null;
+  var actionSeq = 0;
+  var actionTimeouts = Object.create(null);
+  var stopped = false;
+  var restartCount = 0;
+  var initialActionSent = false;
+  var workerGeneration = 0;
+  var pendingStorage = Object.create(null);
+  var pendingLlm = Object.create(null);
+  var pendingActions = Object.create(null);
+  var readyWaiters = [];
+  var isReady = false;
+
+  function clearActionTimeout() {
+    var keys = Object.keys(actionTimeouts);
+    for (var i = 0; i < keys.length; i++) clearTimeout(actionTimeouts[keys[i]]);
+    actionTimeouts = Object.create(null);
+  }
+
+  function clearOneActionTimeout(seq) {
+    if (seq === undefined || !actionTimeouts[seq]) return;
+    clearTimeout(actionTimeouts[seq]);
+    delete actionTimeouts[seq];
+  }
+
+  function rejectActions(message) {
+    var keys = Object.keys(pendingActions);
+    for (var i = 0; i < keys.length; i++) pendingActions[keys[i]].reject(new Error(message));
+    pendingActions = Object.create(null);
+  }
+
+  function reportError(message) {
+    if (typeof config.onError === "function") config.onError(message);
+  }
+
+  function markReady() {
+    isReady = true;
+    if (readyWaiters.length === 0) return;
+    var waiters = readyWaiters;
+    readyWaiters = [];
+    for (var i = 0; i < waiters.length; i++) waiters[i].resolve(state);
+  }
+
+  function terminateWorker() {
+    clearActionTimeout();
+    if (worker) worker.terminate();
+    worker = null;
+    pendingStorage = Object.create(null);
+    var llmKeys = Object.keys(pendingLlm);
+    for (var li = 0; li < llmKeys.length; li++) clearTimeout(pendingLlm[llmKeys[li]].timer);
+    pendingLlm = Object.create(null);
+  }
+
+  function restartWorker(message) {
+    rejectActions(message);
+    terminateWorker();
+    reportError(message);
+    if (stopped) return;
+    restartCount++;
+    if (restartCount > 3) {
+      reportError("Tool worker stopped after repeated failures.");
+      var waiters = readyWaiters;
+      readyWaiters = [];
+      for (var i = 0; i < waiters.length; i++) waiters[i].reject(new Error(message));
+      return;
+    }
+    initialActionSent = false;
+    isReady = false;
+    setTimeout(function () { if (!stopped) spawnWorker(); }, 100);
+  }
+
+  function withReadySocket(startedAt, generation, onReady, onTimeout) {
+    if (stopped || generation !== workerGeneration || !worker) return;
+    var ws = getWs();
+    if (ws && ws.readyState === 1) {
+      try {
+        onReady(ws);
+        return;
+      } catch (error) {}
+    }
+    var waitMs = config.wsWaitMs === undefined ? WS_WAIT_MS : config.wsWaitMs;
+    if (Date.now() - startedAt >= waitMs) {
+      onTimeout();
+      return;
+    }
+    var retryMs = config.wsRetryMs === undefined ? WS_RETRY_MS : config.wsRetryMs;
+    setTimeout(function () { withReadySocket(startedAt, generation, onReady, onTimeout); }, retryMs);
+  }
+
+  function sendStorageOperation(msg) {
+    var generation = workerGeneration;
+    withReadySocket(Date.now(), generation, function (ws) {
+      var externalSeq = config.toolId + ":" + generation + ":" + msg.seq;
+      pendingStorage[externalSeq] = msg.seq;
+      try {
+        ws.send(JSON.stringify({ type: "tool_storage_op", toolId: config.toolId, op: msg.op, args: msg.args, seq: externalSeq }));
+      } catch (error) {
+        delete pendingStorage[externalSeq];
+        throw error;
+      }
+    }, function () {
+      if (worker && generation === workerGeneration) {
+        worker.postMessage({ type: "storage_result", seq: msg.seq, error: "Tool storage is unavailable after waiting 5 seconds for the connection." });
+      }
+    });
+  }
+
+  function sendLlmOperation(msg) {
+    if (!config.allowLlm) {
+      worker.postMessage({ type: "llm_result", seq: msg.seq, error: "This capsule does not have the llm permission." });
+      return;
+    }
+    if (typeof config.onLlmRequest === "function") config.onLlmRequest(msg.args && msg.args.model ? msg.args.model : "fast");
+    var generation = workerGeneration;
+    withReadySocket(Date.now(), generation, function (ws) {
+      var requestId = config.toolId + ":llm:" + generation + ":" + msg.seq;
+      pendingLlm[requestId] = {
+        seq: msg.seq,
+        timer: setTimeout(function () {
+          if (!pendingLlm[requestId] || !worker) return;
+          delete pendingLlm[requestId];
+          worker.postMessage({ type: "llm_result", seq: msg.seq, error: "LLM request timed out after 60 seconds." });
+        }, LLM_TIMEOUT_MS),
+      };
+      try {
+        ws.send(JSON.stringify({ type: "tool_llm_op", toolId: config.toolId, requestId: requestId, args: msg.args, callerId: msg.callerId || "user" }));
+      } catch (error) {
+        clearTimeout(pendingLlm[requestId].timer);
+        delete pendingLlm[requestId];
+        throw error;
+      }
+    }, function () {
+      if (worker && generation === workerGeneration) {
+        worker.postMessage({ type: "llm_result", seq: msg.seq, error: "Tool LLM service is unavailable after waiting 5 seconds for the connection." });
+      }
+    });
+  }
+
+  function handleWorkerMessage(event) {
+    var msg = event.data || {};
+    if (msg.type === "state") {
+      restartCount = 0;
+      state = msg.newState || {};
+      if (typeof config.onState === "function") config.onState(state);
+      if (msg.intermediate) return;
+      clearOneActionTimeout(msg.actionSeq);
+      if (msg.actionSeq !== undefined && pendingActions[msg.actionSeq]) {
+        pendingActions[msg.actionSeq].resolve(state);
+        delete pendingActions[msg.actionSeq];
+      }
+      var shouldInitialize = !initialActionSent && config.initialAction;
+      if (shouldInitialize) {
+        initialActionSent = true;
+        dispatchAction(config.initialAction, {}, "user").catch(function () { markReady(); });
+      } else {
+        markReady();
+      }
+    } else if (msg.type === "storage") {
+      sendStorageOperation(msg);
+    } else if (msg.type === "llm") {
+      sendLlmOperation(msg);
+    } else if (msg.type === "error") {
+      clearOneActionTimeout(msg.actionSeq);
+      if (msg.actionSeq !== undefined) {
+        if (pendingActions[msg.actionSeq]) {
+          pendingActions[msg.actionSeq].reject(new Error(msg.message || "Tool action failed."));
+          delete pendingActions[msg.actionSeq];
+        }
+        reportError(msg.message || "Tool action failed.");
+        return;
+      }
+      restartWorker(msg.message || "Tool worker failed.");
+    }
+  }
+
+  function spawnWorker() {
+    terminateWorker();
+    workerGeneration++;
+    var blob = new Blob([workerProgram(config.logicSource, !!config.allowLlm)], { type: "text/javascript" });
+    var url = URL.createObjectURL(blob);
+    worker = new Worker(url);
+    URL.revokeObjectURL(url);
+    worker.onmessage = handleWorkerMessage;
+    worker.onerror = function (event) {
+      if (event.preventDefault) event.preventDefault();
+      restartWorker(event.message || "Tool worker crashed.");
+    };
+    worker.postMessage({ type: "init", state: state });
+  }
+
+  function dispatchAction(name, args, callerId) {
+    if (!worker || stopped) return Promise.reject(new Error("Tool worker is not running."));
+    actionSeq++;
+    var currentActionSeq = actionSeq;
+    return new Promise(function (resolve, reject) {
+      pendingActions[currentActionSeq] = { resolve: resolve, reject: reject };
+      var timeoutMs = config.allowLlm ? LLM_ACTION_TIMEOUT_MS : ACTION_TIMEOUT_MS;
+      actionTimeouts[currentActionSeq] = setTimeout(function () {
+        restartWorker("Action '" + name + "' timed out after " + Math.round(timeoutMs / 1000) + " seconds.");
+      }, timeoutMs);
+      worker.postMessage({ type: "action", name: name, args: args || {}, callerId: callerId || "user", actionSeq: currentActionSeq });
+    });
+  }
+
+  function handleStorageResult(msg) {
+    if (!worker) return;
+    var localSeq = pendingStorage[msg.seq];
+    if (localSeq === undefined) return;
+    delete pendingStorage[msg.seq];
+    worker.postMessage({ type: "storage_result", seq: localSeq, data: msg.data, error: msg.error });
+  }
+
+  function handleLlmResult(msg) {
+    if (!worker) return;
+    var pending = pendingLlm[msg.requestId];
+    if (!pending) return;
+    clearTimeout(pending.timer);
+    delete pendingLlm[msg.requestId];
+    worker.postMessage({ type: "llm_result", seq: pending.seq, data: msg.data, error: msg.error });
+  }
+
+  function start() {
+    stopped = false;
+    isReady = false;
+    spawnWorker();
+  }
+
+  function stop() {
+    stopped = true;
+    rejectActions("Tool worker stopped.");
+    var waiters = readyWaiters;
+    readyWaiters = [];
+    for (var i = 0; i < waiters.length; i++) waiters[i].reject(new Error("Tool worker stopped."));
+    terminateWorker();
+  }
+
+  function ready() {
+    if (isReady) return Promise.resolve(state);
+    return new Promise(function (resolve, reject) { readyWaiters.push({ resolve: resolve, reject: reject }); });
+  }
+
+  return { start: start, ready: ready, action: dispatchAction, handleStorageResult: handleStorageResult, handleLlmResult: handleLlmResult, stop: stop, getState: function () { return state; } };
+}

package/lib/public/modules/tool-ui-evaluator.js

@@ -0,0 +1,97 @@
+// Pure bounded evaluation for safe declarative Capsule UI.
+
+var MAX_COLLECTION_ITEMS = 500;
+
+export function normalizeOptions(value) {
+  if (!Array.isArray(value)) return [];
+  return value.slice(0, 100).reduce(function (result, option) {
+    if (typeof option === "string" || typeof option === "number") result.push({ value: option, label: String(option), disabled: false });
+    else if (option && typeof option === "object" && (typeof option.value === "string" || typeof option.value === "number") && typeof option.label === "string") result.push({ value: option.value, label: option.label.slice(0, 500), disabled: option.disabled === true });
+    return result;
+  }, []);
+}
+
+export function normalizeColumns(value) {
+  if (!Array.isArray(value)) return [];
+  return value.slice(0, 20).filter(function (column) {
+    return column && typeof column === "object" && typeof column.key === "string" && /^(?:\$(?:state|item)\.)?[A-Za-z0-9_-]+(?:\.[A-Za-z0-9_-]+)*$/.test(column.key) && !/(?:__proto__|constructor|prototype)/.test(column.key);
+  }).map(function (column) { return { key: column.key, label: typeof column.label === "string" ? column.label.slice(0, 500) : column.key }; });
+}
+
+export function valueAtPath(state, path, item) {
+  if (!path) return undefined;
+  var source = state;
+  var parts = path.split(".");
+  if (parts[0] === "$item") { source = item; parts.shift(); }
+  else if (parts[0] === "$state") parts.shift();
+  for (var i = 0; i < parts.length; i++) {
+    if (source === null || source === undefined) return undefined;
+    source = source[parts[i]];
+  }
+  return source;
+}
+
+export function resolveValue(value, state, item) {
+  if (value && typeof value === "object" && !Array.isArray(value) && value.$bind) {
+    var bound = valueAtPath(state, value.$bind, item);
+    if (Array.isArray(value.$enum)) return value.$enum.indexOf(bound) !== -1 ? bound : value.fallback !== undefined ? value.fallback : value.$enum[0];
+    return bound === undefined ? value.fallback : bound;
+  }
+  if (typeof value === "string" && (value.indexOf("$item") === 0 || value.indexOf("$state") === 0)) return valueAtPath(state, value, item);
+  if (Array.isArray(value)) return value.map(function (entry) { return resolveValue(entry, state, item); });
+  if (value && typeof value === "object") {
+    var result = {};
+    Object.keys(value).forEach(function (key) { result[key] = resolveValue(value[key], state, item); });
+    return result;
+  }
+  return value;
+}
+
+export function evaluateCondition(condition, state, item) {
+  if (typeof condition === "string") return !!valueAtPath(state, condition, item);
+  if (!condition || typeof condition !== "object") return false;
+  if (condition.all) return condition.all.every(function (entry) { return evaluateCondition(entry, state, item); });
+  if (condition.any) return condition.any.some(function (entry) { return evaluateCondition(entry, state, item); });
+  if (condition.not) return !evaluateCondition(condition.not, state, item);
+  var operators = ["equals", "notEquals", "in", "gt", "gte", "lt", "lte"];
+  for (var i = 0; i < operators.length; i++) {
+    var operator = operators[i];
+    if (!condition[operator]) continue;
+    var rule = condition[operator];
+    var actual = valueAtPath(state, rule.path, item);
+    if (operator === "equals") return actual === rule.value;
+    if (operator === "notEquals") return actual !== rule.value;
+    if (operator === "in") return rule.values.indexOf(actual) !== -1;
+    if (typeof actual !== "number") return false;
+    if (operator === "gt") return actual > rule.value;
+    if (operator === "gte") return actual >= rule.value;
+    if (operator === "lt") return actual < rule.value;
+    if (operator === "lte") return actual <= rule.value;
+  }
+  return false;
+}
+
+export function resolveProps(props, state, item) {
+  var result = {};
+  Object.keys(props || {}).forEach(function (key) { result[key] = resolveValue(props[key], state, item); });
+  return result;
+}
+
+export function collectionView(value, props) {
+  var items = Array.isArray(value) ? value.slice(0, MAX_COLLECTION_ITEMS) : [];
+  var filter = String(props.filter || "").trim().toLocaleLowerCase();
+  if (filter) items = items.filter(function (item) {
+    var candidate = props.filterKey ? valueAtPath(item, props.filterKey, null) : item;
+    return String(candidate === undefined ? "" : candidate).toLocaleLowerCase().indexOf(filter) !== -1;
+  });
+  if (props.sortKey) items.sort(function (left, right) {
+    var a = valueAtPath(left, props.sortKey, null);
+    var b = valueAtPath(right, props.sortKey, null);
+    if (a === b) return 0;
+    var result = a === undefined || a === null ? 1 : b === undefined || b === null ? -1 : a < b ? -1 : 1;
+    return props.sortDirection === "desc" ? -result : result;
+  });
+  var pageSize = Math.max(1, Math.min(100, Number(props.pageSize) || 100));
+  var page = Math.max(1, Number(props.page) || 1);
+  return items.slice((page - 1) * pageSize, page * pageSize);
+}

package/lib/public/modules/tool-ui-tree.js

@@ -0,0 +1,21 @@
+// Pure helpers for inspecting declarative Capsule UI trees.
+
+export function hasToolUiNodeType(node, type) {
+  if (!node || typeof node !== "object") return false;
+  if (node.type === type) return true;
+  var children = Array.isArray(node.children) ? node.children : [];
+  for (var i = 0; i < children.length; i++) {
+    if (hasToolUiNodeType(children[i], type)) return true;
+  }
+  return false;
+}
+
+export function isToolModelAlias(value) {
+  return value === "fast" || value === "standard" || value === "deep";
+}
+
+export function shouldInjectToolLlmStatus(definition) {
+  var permissions = definition && definition.manifest && Array.isArray(definition.manifest.permissions)
+    ? definition.manifest.permissions : [];
+  return permissions.indexOf("llm") !== -1 && !hasToolUiNodeType(definition.uiTree, "model-select");
+}

package/lib/public/modules/tui-attention.js

@@ -29,6 +29,10 @@ var modalWs = null;
 var modalResizeObserver = null;
 var modalKeyHandler = null;
 var modalResizeDebounce = null;
+// Optional per-open callback fired once the modal is torn down, whatever
+// dismissed it (Esc, backdrop click, term_closed). The login flow uses it to
+// cancel a login the user walked away from.
+var modalCloseHook = null;
 // Debounced fit+redraw for the modal xterm. Same rationale as
 // session-tui-view.js: collapses rapid resize events into a single
 // SIGWINCH so claude can redraw cleanly without mid-resize corruption.
@@ -139,6 +143,7 @@ export function openTuiModal(terminalId, sourceSlug, info) {
   modalTerminalId = terminalId;
   modalSourceSlug = sourceSlug;
   var infoObj = info || {};
+  modalCloseHook = typeof infoObj.onClose === "function" ? infoObj.onClose : null;
   setModalBreadcrumb({
     projectIcon: infoObj.projectIcon || null,
     projectName: infoObj.projectName || sourceSlug,
@@ -232,6 +237,8 @@ export function openTuiModal(terminalId, sourceSlug, info) {
 
 export function closeTuiModal() {
   if (!modalEl) return;
+  var closeHook = modalCloseHook;
+  modalCloseHook = null;
   if (modalTerminalId != null && modalWs && modalWs.readyState === 1) {
     try { modalWs.send(JSON.stringify({ type: "term_detach", id: modalTerminalId })); } catch (e) {}
   }
@@ -247,6 +254,9 @@ export function closeTuiModal() {
     document.removeEventListener("keydown", modalKeyHandler);
     modalKeyHandler = null;
   }
+  if (closeHook) {
+    try { closeHook(); } catch (e) {}
+  }
 }
 
 export function isTuiModalOpen() {

package/lib/public/modules/user-settings.js

@@ -211,22 +211,20 @@ export function initUserSettings(appCtx) {
     });
   }
 
-  // Mates UI toggle. Default-on, so flipping off hides every Mates
-  // surface in the app (sidebar avatars, DM picker entry, home-hub strip)
-  // via the body.mates-disabled CSS gate. Flipping back on restores the
-  // surface; existing Mate data is never deleted.
+  // This legacy preference now controls only project-local Mate DM entry
+  // points. Home, mentions, creation, debates, and Mate data stay available.
   var matesEnabledToggle = document.getElementById('us-mates-enabled');
   if (matesEnabledToggle) {
     matesEnabledToggle.addEventListener('change', function () {
       var want = !!this.checked;
-      applyMatesEnabledClass(want);
-      store.set({ matesEnabled: want });
+      applyProjectMateDmsClass(want);
+      store.set({ projectMateDmsEnabled: want });
       fetch('/api/user/mates-enabled', {
         method: 'PUT',
         headers: { 'Content-Type': 'application/json' },
         body: JSON.stringify({ enabled: want }),
       }).then(function (r) { return r.json(); }).then(function (data) {
-        if (data.ok) showToast(data.matesEnabled ? 'Mates on' : 'Mates off');
+        if (data.ok) showToast(data.matesEnabled ? 'Mate DMs shown in projects' : 'Mate DMs hidden in projects');
       }).catch(function () {});
     });
   }
@@ -317,22 +315,17 @@ function stopProp(e) {
   e.stopPropagation();
 }
 
-// Mates UI gate. When the toggle is off we add `mates-disabled` to the
-// body so CSS can hide every Mates surface (sidebar avatars, DM picker
-// "Create a Mate" entry, home-hub mates strip) without touching the
-// underlying data. JS code paths that gate on `store.get('matesEnabled')`
-// also short-circuit so we don't, for example, open the mate wizard if
-// someone fires it programmatically.
-function applyMatesEnabledClass(enabled) {
+// Project-only Mate DM gate. The persisted server field retains its legacy
+// name for migration compatibility, but it no longer disables Mate features.
+function applyProjectMateDmsClass(enabled) {
   if (!document.body) return;
-  document.body.classList.toggle('mates-disabled', !enabled);
+  document.body.classList.toggle('project-mate-dms-disabled', !enabled);
 }
 
-export function isMatesEnabled() {
-  // Default to true if the store hasn't received profile yet — populating
-  // the false state happens after /api/profile resolves, so an undefined
-  // value at boot should not hide Mates UI.
-  return store.get('matesEnabled') !== false;
+export function isProjectMateDmsEnabled() {
+  // Default to true until the profile arrives so boot does not briefly hide
+  // a project shortcut that the persisted preference enables.
+  return store.get('projectMateDmsEnabled') !== false;
 }
 
 function showPinForm() {
@@ -467,12 +460,12 @@ function populateAccount() {
     if (data.mateOnboardingShown) {
       try { localStorage.setItem("clay-mate-onboarding-shown", "1"); } catch (e) {}
     }
-    // Mates UI toggle: default OFF (opt-in); enable only when explicitly true
+    // Project Mate DM shortcuts remain opt-in for legacy profiles.
     var matesOn = data.matesEnabled === true;
     var matesToggle = document.getElementById('us-mates-enabled');
     if (matesToggle) matesToggle.checked = matesOn;
-    store.set({ matesEnabled: matesOn });
-    applyMatesEnabledClass(matesOn);
+    store.set({ projectMateDmsEnabled: matesOn });
+    applyProjectMateDmsClass(matesOn);
     if (data.chatLayout) {
       setChatLayout(data.chatLayout); // update local cache + CSS
     }