clay-server 2.47.0-beta.2 → 2.47.0-beta.4

package/lib/project-mate-datastore.js

@@ -1,232 +0,0 @@
-var z = require("zod");
-var datastore = require("./mate-datastore");
-var FEATURE_ENABLED = false;
-
-function attachMateDatastore(ctx) {
-  var cwd = ctx.cwd;
-  var isMate = ctx.isMate;
-  var send = ctx.send;
-  var sendTo = ctx.sendTo;
-
-  function isFeatureAvailable() {
-    return FEATURE_ENABLED && datastore.isMateDatastoreAvailable();
-  }
-
-  function ensureProjectDatastore() {
-    if (!FEATURE_ENABLED) {
-      return { ok: false, code: "MATE_DATASTORE_NOT_ALLOWED", message: "Mate datastore is not enabled." };
-    }
-    if (!isMate) {
-      return { ok: false, code: "MATE_DATASTORE_NOT_ALLOWED", message: "Mate datastore is only available in Mate sessions." };
-    }
-    try {
-      return datastore.ensureMateDatastore({ mateDir: cwd });
-    } catch (e) {
-      return { ok: false, code: "MATE_DATASTORE_UNAVAILABLE", message: e.message || "Mate datastore is unavailable." };
-    }
-  }
-
-  function getDatastoreWarning(handle) {
-    if (!handle || !handle.warning) return null;
-    return handle.warning;
-  }
-
-  function normalizeParams(input) {
-    if (!input) return [];
-    if (Array.isArray(input.params)) return input.params;
-    return [];
-  }
-
-  function normalizeToolResult(result, requestId) {
-    var payload = result || { ok: false, code: "MATE_DATASTORE_UNAVAILABLE", message: "Mate datastore is unavailable." };
-    if (typeof requestId !== "undefined") payload.requestId = requestId;
-    return payload;
-  }
-
-  function callToolInternal(toolName, input) {
-    var handle = ensureProjectDatastore();
-    if (!handle || handle.ok === false) {
-      return normalizeToolResult(handle, input && input.requestId);
-    }
-
-    var warning = getDatastoreWarning(handle);
-    var params = normalizeParams(input);
-
-    if (toolName === "clay_db_query") {
-      var queryResult = datastore.runQuery(handle, input.sql, params, { maxRows: 200, maxBytes: 1024 * 1024, includeSizeInfo: true });
-      if (warning && queryResult.ok && !queryResult.warning) queryResult.warning = warning;
-      return normalizeToolResult(queryResult, input && input.requestId);
-    }
-
-    if (toolName === "clay_db_exec") {
-      var execResult = datastore.runExec(handle, input.sql, params, { maxBytes: 1024 * 1024, includeSizeInfo: true });
-      if (warning && execResult.ok && !execResult.warning) execResult.warning = warning;
-      if (execResult.ok) {
-        broadcastDbChange({ tool: toolName, sql: input.sql, changes: execResult.changes || 0 });
-      }
-      return normalizeToolResult(execResult, input && input.requestId);
-    }
-
-    if (toolName === "clay_db_tables") {
-      var tablesResult = datastore.listSchemaObjects(handle);
-      if (warning && tablesResult.ok && !tablesResult.warning) tablesResult.warning = warning;
-      return normalizeToolResult(tablesResult, input && input.requestId);
-    }
-
-    if (toolName === "clay_db_describe") {
-      var describeResult = datastore.describeTable(handle, input.table);
-      if (warning && describeResult.ok && !describeResult.warning) describeResult.warning = warning;
-      return normalizeToolResult(describeResult, input && input.requestId);
-    }
-
-    return normalizeToolResult({ ok: false, code: "MATE_DATASTORE_NOT_ALLOWED", message: "Unknown datastore tool: " + toolName }, input && input.requestId);
-  }
-
-  function broadcastDbChange(details) {
-    var payload = {
-      type: "mate_db_change",
-      details: details || {},
-    };
-    send(payload);
-  }
-
-  function handleMateDatastoreMessage(ws, msg) {
-    if (msg.type !== "mate_db_tables" && msg.type !== "mate_db_describe" && msg.type !== "mate_db_query" && msg.type !== "mate_db_exec") {
-      return false;
-    }
-
-    if (!FEATURE_ENABLED) {
-      return false;
-    }
-
-    if (!isMate) {
-      sendTo(ws, {
-        type: "mate_db_error",
-        requestId: msg.requestId || null,
-        code: "MATE_DATASTORE_NOT_ALLOWED",
-        message: "Mate datastore is only available in Mate sessions.",
-      });
-      return true;
-    }
-
-    if (msg.type === "mate_db_tables") {
-      sendResult(ws, "mate_db_tables_result", callToolInternal("clay_db_tables", msg));
-      return true;
-    }
-
-    if (msg.type === "mate_db_describe") {
-      sendResult(ws, "mate_db_describe_result", callToolInternal("clay_db_describe", msg));
-      return true;
-    }
-
-    if (msg.type === "mate_db_query") {
-      sendResult(ws, "mate_db_query_result", callToolInternal("clay_db_query", msg));
-      return true;
-    }
-
-    if (msg.type === "mate_db_exec") {
-      sendResult(ws, "mate_db_exec_result", callToolInternal("clay_db_exec", msg));
-      return true;
-    }
-
-    return true;
-  }
-
-  function sendResult(ws, type, result) {
-    var payload = result || { ok: false, code: "MATE_DATASTORE_UNAVAILABLE", message: "Mate datastore is unavailable." };
-    payload.type = type;
-    sendTo(ws, payload);
-  }
-
-  function getToolDefinitions() {
-    if (!isMate || !isFeatureAvailable()) return [];
-    return [
-      {
-        name: "clay_db_query",
-        description: "Execute read-only SQL against the current Mate datastore.",
-        inputSchema: {
-          sql: z.string().min(1).describe("SQL query"),
-          params: z.array(z.any()).optional().describe("Positional parameters"),
-        },
-        handler: function (input) {
-          return callToolInternal("clay_db_query", input || {});
-        },
-      },
-      {
-        name: "clay_db_exec",
-        description: "Execute schema or write SQL against the current Mate datastore.",
-        inputSchema: {
-          sql: z.string().min(1).describe("SQL statement"),
-          params: z.array(z.any()).optional().describe("Positional parameters"),
-        },
-        handler: function (input) {
-          return callToolInternal("clay_db_exec", input || {});
-        },
-      },
-      {
-        name: "clay_db_tables",
-        description: "List schema objects in the current Mate datastore.",
-        inputSchema: undefined,
-        handler: function (input) {
-          return callToolInternal("clay_db_tables", input || {});
-        },
-      },
-      {
-        name: "clay_db_describe",
-        description: "Describe a table or view in the current Mate datastore.",
-        inputSchema: {
-          table: z.string().min(1).describe("Table name"),
-        },
-        handler: function (input) {
-          return callToolInternal("clay_db_describe", input || {});
-        },
-      },
-    ];
-  }
-
-  function createMcpServer() {
-    var defs = getToolDefinitions();
-    if (!defs.length) return null;
-    var registered = {};
-    for (var i = 0; i < defs.length; i++) {
-      registered[defs[i].name] = {
-        description: defs[i].description,
-        inputSchema: defs[i].inputSchema,
-        handler: defs[i].handler,
-      };
-    }
-    return {
-      name: "clay-datastore",
-      version: "1.0.0",
-      instance: {
-        _registeredTools: registered,
-      },
-    };
-  }
-
-  function getSessionToolDefinitions() {
-    if (!isMate || !isFeatureAvailable()) return null;
-    return getToolDefinitions();
-  }
-
-  function callMateTool(session, toolName, input) {
-    return callToolInternal(toolName, input || {});
-  }
-
-  function closeAllDatastores() {
-    datastore.closeAllMateDatastores();
-  }
-
-  return {
-    isFeatureAvailable: isFeatureAvailable,
-    handleMateDatastoreMessage: handleMateDatastoreMessage,
-    getToolDefinitions: getToolDefinitions,
-    createMcpServer: createMcpServer,
-    getSessionToolDefinitions: getSessionToolDefinitions,
-    callMateTool: callMateTool,
-    ensureProjectDatastore: ensureProjectDatastore,
-    closeAllDatastores: closeAllDatastores,
-  };
-}
-
-module.exports = { attachMateDatastore: attachMateDatastore };

package/lib/public/modules/mate-datastore-ui.js

@@ -1,280 +0,0 @@
-import { escapeHtml } from './utils.js';
-import { refreshIcons } from './icons.js';
-import { getWs } from './ws-ref.js';
-
-var wsGetter = null;
-var panelEl = null;
-var tableListEl = null;
-var tableNameEl = null;
-var tableSchemaEl = null;
-var resultEl = null;
-var statusEl = null;
-var dataBtnEl = null;
-var mainColumnEl = null;
-var currentTables = [];
-var currentTable = null;
-var panelOpen = false;
-var routingToScheduler = false;
-
-function sendWs(msg) {
-  var ws = wsGetter ? wsGetter() : getWs();
-  if (ws && ws.readyState === 1) ws.send(JSON.stringify(msg));
-}
-
-function quoteIdentifier(name) {
-  return '"' + String(name || "").replace(/"/g, '""') + '"';
-}
-
-function ensurePanel() {
-  if (panelEl) return;
-  mainColumnEl = document.getElementById("main-column");
-  if (!mainColumnEl) return;
-
-  panelEl = document.createElement("div");
-  panelEl.id = "mate-datastore-panel";
-  panelEl.className = "hidden";
-
-  panelEl.innerHTML =
-    '<div class="scheduler-top-bar mate-datastore-top-bar">' +
-      '<span class="scheduler-top-title mate-datastore-top-title"><i data-lucide="database"></i>Data</span>' +
-      '<div class="mate-datastore-top-actions">' +
-        '<button id="mate-db-refresh-btn" class="scheduler-close-btn" type="button" title="Refresh tables"><i data-lucide="refresh-cw"></i></button>' +
-        '<button id="mate-db-back-btn" class="scheduler-close-btn" type="button" title="Close"><i data-lucide="x"></i></button>' +
-      '</div>' +
-    '</div>' +
-    '<div class="mate-db-status" id="mate-db-status"></div>' +
-    '<div class="mate-db-layout">' +
-      '<div class="mate-db-table-column">' +
-        '<div class="mate-db-section-title">Objects</div>' +
-        '<div id="mate-db-table-list" class="mate-db-table-list"></div>' +
-      '</div>' +
-      '<div class="mate-db-detail">' +
-        '<div class="mate-db-section-title" id="mate-db-table-name">No table selected</div>' +
-        '<pre id="mate-db-table-schema" class="mate-db-table-schema"></pre>' +
-        '<div class="mate-db-section-title">Rows</div>' +
-        '<pre id="mate-db-result" class="mate-db-result"></pre>' +
-      '</div>' +
-    '</div>';
-
-  mainColumnEl.appendChild(panelEl);
-  tableListEl = document.getElementById("mate-db-table-list");
-  tableNameEl = document.getElementById("mate-db-table-name");
-  tableSchemaEl = document.getElementById("mate-db-table-schema");
-  resultEl = document.getElementById("mate-db-result");
-  statusEl = document.getElementById("mate-db-status");
-
-  var refreshBtn = document.getElementById("mate-db-refresh-btn");
-  var backBtn = document.getElementById("mate-db-back-btn");
-
-  if (refreshBtn) {
-    refreshBtn.addEventListener("click", function () {
-      requestTables();
-    });
-  }
-
-  if (backBtn) {
-    backBtn.addEventListener("click", function () {
-      setSectionVisibility(false);
-    });
-  }
-
-  refreshIcons();
-}
-
-function setSectionVisibility(open) {
-  ensurePanel();
-  panelOpen = open;
-  if (panelEl) panelEl.classList.toggle("hidden", !open);
-  if (mainColumnEl) mainColumnEl.classList.toggle("mate-datastore-open", open);
-  if (dataBtnEl) dataBtnEl.classList.toggle("active", open);
-  refreshIcons();
-}
-
-function requestTables() {
-  sendWs({ type: "mate_db_tables" });
-}
-
-function renderStatus(text, kind) {
-  if (!statusEl) return;
-  statusEl.textContent = text || "";
-  statusEl.dataset.kind = kind || "";
-}
-
-function renderResult(payload) {
-  if (!resultEl) return;
-  resultEl.textContent = JSON.stringify(payload, null, 2);
-}
-
-function formatColumns(columns) {
-  var list = Array.isArray(columns) ? columns : [];
-  if (!list.length) return "No column information available.";
-  var lines = [];
-  for (var i = 0; i < list.length; i++) {
-    var col = list[i] || {};
-    var line = (col.name || "?") + " " + (col.type || "");
-    if (col.pk) line += " PRIMARY KEY";
-    if (col.notnull) line += " NOT NULL";
-    if (typeof col.dflt_value !== "undefined" && col.dflt_value !== null) line += " DEFAULT " + col.dflt_value;
-    lines.push(line.trim());
-  }
-  return lines.join("\n");
-}
-
-function renderTableList(objects) {
-  currentTables = objects || [];
-  if (!tableListEl) return;
-  tableListEl.innerHTML = "";
-
-  if (!currentTables.length) {
-    var empty = document.createElement("div");
-    empty.className = "mate-db-empty";
-    empty.textContent = "No tables or views found.";
-    tableListEl.appendChild(empty);
-    return;
-  }
-
-  for (var i = 0; i < currentTables.length; i++) {
-    (function (obj) {
-      var row = document.createElement("button");
-      row.type = "button";
-      row.className = "mate-db-table-item" + (currentTable === obj.name ? " active" : "");
-      var label = obj.type || "object";
-      row.innerHTML = '<span class="mate-db-table-item-name">' + escapeHtml(obj.name) + '</span>' +
-        '<span class="mate-db-table-item-type">' + escapeHtml(label) + '</span>';
-      row.addEventListener("click", function () {
-        selectTable(obj.name);
-      });
-      tableListEl.appendChild(row);
-    })(currentTables[i]);
-  }
-}
-
-function findFirstDescribableObject(objects) {
-  var list = objects || [];
-  for (var i = 0; i < list.length; i++) {
-    if (list[i] && (list[i].type === "table" || list[i].type === "view")) return list[i];
-  }
-  return null;
-}
-
-function findObjectByName(name) {
-  for (var i = 0; i < currentTables.length; i++) {
-    if (currentTables[i] && currentTables[i].name === name) return currentTables[i];
-  }
-  return null;
-}
-
-function selectTable(tableName) {
-  currentTable = tableName;
-  renderTableList(currentTables);
-  var obj = findObjectByName(tableName);
-  if (!obj) return;
-  if (obj.type !== "table" && obj.type !== "view") {
-    renderStatus("Selected " + (obj.type || "object") + " " + tableName + ".", "ok");
-    if (tableNameEl) tableNameEl.textContent = tableName;
-    if (tableSchemaEl) tableSchemaEl.textContent = (obj.type || "object") + ": " + tableName;
-    renderResult(obj);
-    return;
-  }
-  renderStatus("Loading " + tableName + "...", "info");
-  sendWs({ type: "mate_db_describe", table: tableName });
-  sendWs({ type: "mate_db_query", sql: "SELECT * FROM " + quoteIdentifier(tableName) + " LIMIT 100", params: [] });
-}
-
-export function initMateDatastoreUI(getWsFn) {
-  wsGetter = getWsFn;
-  dataBtnEl = document.getElementById("mate-data-btn");
-  if (!dataBtnEl) return;
-
-  dataBtnEl.addEventListener("click", function () {
-    if (panelOpen) {
-      setSectionVisibility(false);
-    } else {
-      setSectionVisibility(true);
-      requestTables();
-    }
-  });
-
-  document.addEventListener("click", function (e) {
-    var btn = e.target && e.target.closest ? e.target.closest("#scheduler-btn, #mate-scheduler-btn") : null;
-    if (!btn || !panelOpen || routingToScheduler) return;
-    e.preventDefault();
-    e.stopPropagation();
-    if (typeof e.stopImmediatePropagation === "function") e.stopImmediatePropagation();
-    routingToScheduler = true;
-    setSectionVisibility(false);
-    setTimeout(function () {
-      var schedulerBtn = document.getElementById("scheduler-btn");
-      if (schedulerBtn) schedulerBtn.click();
-      routingToScheduler = false;
-    }, 0);
-  }, true);
-
-  setSectionVisibility(false);
-}
-
-export function showMateDatastorePanel() {
-  setSectionVisibility(true);
-  requestTables();
-}
-
-export function hideMateDatastorePanel() {
-  setSectionVisibility(false);
-}
-
-export function handleMateDatastoreTablesResult(msg) {
-  if (msg.ok === false) {
-    renderStatus(msg.message || "Failed to load datastore tables.", "error");
-    renderResult(msg);
-    return;
-  }
-  renderStatus(msg.warning || "Datastore tables loaded.", msg.warning ? "warn" : "ok");
-  renderTableList(msg.objects || []);
-  if (msg.objects && msg.objects.length > 0) {
-    var found = false;
-    for (var i = 0; i < msg.objects.length; i++) {
-      if (msg.objects[i].name === currentTable) {
-        found = true;
-        break;
-      }
-    }
-    if (!found) {
-      var firstObject = findFirstDescribableObject(msg.objects);
-      if (firstObject) selectTable(firstObject.name);
-    }
-  }
-  renderResult(msg.objects || []);
-}
-
-export function handleMateDatastoreDescribeResult(msg) {
-  if (msg.ok === false) {
-    renderStatus(msg.message || "Failed to describe table.", "error");
-    renderResult(msg);
-    return;
-  }
-  renderStatus(msg.warning || ("Described " + (msg.table || "table") + "."), msg.warning ? "warn" : "ok");
-  if (tableNameEl) tableNameEl.textContent = msg.table || "Table";
-  if (tableSchemaEl) tableSchemaEl.textContent = formatColumns(msg.columns);
-}
-
-export function handleMateDatastoreQueryResult(msg) {
-  if (msg.ok === false) {
-    renderStatus(msg.message || "Query failed.", "error");
-    renderResult(msg);
-    return;
-  }
-  var status = "Showing " + (msg.rows ? msg.rows.length : 0) + " row(s).";
-  if (msg.truncated) status = status.replace(".", " (truncated).");
-  renderStatus(msg.warning || status, msg.warning ? "warn" : "ok");
-  renderResult(msg.rows || []);
-}
-
-export function handleMateDatastoreError(msg) {
-  renderStatus(msg.message || "Mate datastore error.", "error");
-  renderResult(msg);
-}
-
-export function handleMateDatastoreChange(msg) {
-  renderStatus("Datastore changed.", "info");
-  if (panelOpen) requestTables();
-}