clay-server 2.26.1-beta.1 → 2.27.0-beta.2

package/lib/project-image.js

@@ -0,0 +1,94 @@
+var fs = require("fs");
+var path = require("path");
+var crypto = require("crypto");
+
+/**
+ * Attach image handling to a project context.
+ *
+ * ctx fields:
+ *   cwd, slug
+ */
+function attachImage(ctx) {
+  var cwd = ctx.cwd;
+  var slug = ctx.slug;
+
+  // --- Chat image storage ---
+  var _imgConfig = require("./config");
+  var _imgUtils = require("./utils");
+  var _imagesBaseDir = path.join(_imgConfig.CONFIG_DIR, "images");
+  var _imagesEncodedCwd = _imgUtils.encodeCwd(cwd);
+  var imagesDir = path.join(_imagesBaseDir, _imagesEncodedCwd);
+
+  // Convert imageRefs in history entries to images with URLs for the client
+  function hydrateImageRefs(entry) {
+    if (!entry) return entry;
+    // Hydrate context_preview: convert screenshotFile to screenshotUrl
+    if (entry.type === "context_preview" && entry.tab && entry.tab.screenshotFile) {
+      var hydrated = {};
+      for (var k in entry) hydrated[k] = entry[k];
+      hydrated.tab = {};
+      for (var tk in entry.tab) hydrated.tab[tk] = entry.tab[tk];
+      hydrated.tab.screenshotUrl = "/p/" + slug + "/images/" + entry.tab.screenshotFile;
+      delete hydrated.tab.screenshotFile;
+      return hydrated;
+    }
+    if (!entry.imageRefs) return entry;
+    if (entry.type !== "user_message" && entry.type !== "mention_user") return entry;
+    var images = [];
+    for (var ri = 0; ri < entry.imageRefs.length; ri++) {
+      var ref = entry.imageRefs[ri];
+      images.push({ mediaType: ref.mediaType, url: "/p/" + slug + "/images/" + ref.file });
+    }
+    var hydrated = {};
+    for (var k2 in entry) {
+      if (k2 !== "imageRefs") hydrated[k2] = entry[k2];
+    }
+    hydrated.images = images;
+    return hydrated;
+  }
+
+  function saveImageFile(mediaType, base64data, ownerLinuxUser) {
+    try { fs.mkdirSync(imagesDir, { recursive: true }); } catch (e) {}
+    var ext = mediaType === "image/png" ? ".png" : mediaType === "image/gif" ? ".gif" : mediaType === "image/webp" ? ".webp" : ".jpg";
+    var hash = crypto.createHash("sha256").update(base64data).digest("hex").substring(0, 16);
+    var fileName = Date.now() + "-" + hash + ext;
+    var filePath = path.join(imagesDir, fileName);
+    try {
+      fs.writeFileSync(filePath, Buffer.from(base64data, "base64"));
+      if (process.platform !== "win32") {
+        // 644 so all local users can read (needed for git, copy, etc.)
+        try { fs.chmodSync(filePath, 0o644); } catch (e) {}
+        // In OS-user mode the daemon runs as root, so chown the file
+        // (and parent dirs) to the session owner to avoid permission issues.
+        if (ownerLinuxUser) {
+          try {
+            var osUsersMod = require("./os-users");
+            var uid = osUsersMod.getLinuxUserUid(ownerLinuxUser);
+            if (uid != null) {
+              require("child_process").execSync("chown " + uid + " " + JSON.stringify(filePath));
+              // Also fix parent dirs if root-owned
+              try {
+                var dirStat = fs.statSync(imagesDir);
+                if (dirStat.uid !== uid) {
+                  require("child_process").execSync("chown " + uid + " " + JSON.stringify(imagesDir));
+                }
+              } catch (e2) {}
+            }
+          } catch (e) {}
+        }
+      }
+      return fileName;
+    } catch (e) {
+      console.error("[images] Failed to save image:", e.message);
+      return null;
+    }
+  }
+
+  return {
+    imagesDir: imagesDir,
+    hydrateImageRefs: hydrateImageRefs,
+    saveImageFile: saveImageFile,
+  };
+}
+
+module.exports = { attachImage: attachImage };

package/lib/project-knowledge.js

@@ -0,0 +1,161 @@
+var fs = require("fs");
+var path = require("path");
+
+/**
+ * Attach knowledge file management to a project context.
+ *
+ * ctx fields:
+ *   cwd, isMate, sendTo, matesModule, projectOwnerId
+ */
+function attachKnowledge(ctx) {
+  var cwd = ctx.cwd;
+  var isMate = ctx.isMate;
+  var sendTo = ctx.sendTo;
+  var matesModule = ctx.matesModule;
+  var getProjectOwnerId = ctx.getProjectOwnerId;
+
+  function listKnowledgeFiles() {
+    var knowledgeDir = path.join(cwd, "knowledge");
+    var files = [];
+    try {
+      var entries = fs.readdirSync(knowledgeDir);
+      for (var ki = 0; ki < entries.length; ki++) {
+        if (entries[ki] === "session-digests.jsonl") continue;
+        if (entries[ki] === "sticky-notes.md") continue;
+        if (entries[ki] === "memory-summary.md") continue;
+        if (entries[ki].endsWith(".md") || entries[ki].endsWith(".jsonl")) {
+          var stat = fs.statSync(path.join(knowledgeDir, entries[ki]));
+          files.push({ name: entries[ki], size: stat.size, mtime: stat.mtimeMs, common: false });
+        }
+      }
+    } catch (e) { /* dir may not exist */ }
+    files.sort(function (a, b) { return b.mtime - a.mtime; });
+
+    if (isMate) {
+      var mateCtx = matesModule.buildMateCtx(getProjectOwnerId());
+      var thisMateId = path.basename(cwd);
+      for (var pi = 0; pi < files.length; pi++) {
+        files[pi].promoted = matesModule.isPromoted(mateCtx, thisMateId, files[pi].name);
+      }
+      var commonFiles = matesModule.getCommonKnowledgeForMate(mateCtx, thisMateId);
+      for (var ci = 0; ci < commonFiles.length; ci++) {
+        if (commonFiles[ci].ownMateId !== thisMateId) {
+          files.push(commonFiles[ci]);
+        }
+      }
+    }
+    return files;
+  }
+
+  function listKnowledgeFilesBasic() {
+    var knowledgeDir = path.join(cwd, "knowledge");
+    var files = [];
+    try {
+      var entries = fs.readdirSync(knowledgeDir);
+      for (var ki = 0; ki < entries.length; ki++) {
+        if (entries[ki].endsWith(".md") || entries[ki].endsWith(".jsonl")) {
+          var stat = fs.statSync(path.join(knowledgeDir, entries[ki]));
+          files.push({ name: entries[ki], size: stat.size, mtime: stat.mtimeMs });
+        }
+      }
+    } catch (e) {}
+    files.sort(function (a, b) { return b.mtime - a.mtime; });
+    if (isMate) {
+      var mateCtx = matesModule.buildMateCtx(getProjectOwnerId());
+      var thisMateId = path.basename(cwd);
+      for (var pi = 0; pi < files.length; pi++) {
+        files[pi].common = false;
+        files[pi].promoted = matesModule.isPromoted(mateCtx, thisMateId, files[pi].name);
+      }
+      var commonFiles = matesModule.getCommonKnowledgeForMate(mateCtx, thisMateId);
+      for (var ci = 0; ci < commonFiles.length; ci++) {
+        if (commonFiles[ci].ownMateId !== thisMateId) files.push(commonFiles[ci]);
+      }
+    }
+    return files;
+  }
+
+  function handleKnowledgeMessage(ws, msg) {
+    if (msg.type === "knowledge_list") {
+      sendTo(ws, { type: "knowledge_list", files: listKnowledgeFiles() });
+      return true;
+    }
+
+    if (msg.type === "knowledge_read") {
+      if (!msg.name) return true;
+      var safeName = path.basename(msg.name);
+      if (msg.common && msg.ownMateId && isMate) {
+        var mateCtx = matesModule.buildMateCtx(getProjectOwnerId());
+        try {
+          var content = matesModule.readCommonKnowledgeFile(mateCtx, msg.ownMateId, safeName);
+          sendTo(ws, { type: "knowledge_content", name: safeName, content: content, common: true, ownMateId: msg.ownMateId });
+        } catch (e) {
+          sendTo(ws, { type: "knowledge_content", name: safeName, content: "", error: "File not found", common: true });
+        }
+      } else {
+        var filePath = path.join(cwd, "knowledge", safeName);
+        try {
+          var content = fs.readFileSync(filePath, "utf8");
+          sendTo(ws, { type: "knowledge_content", name: safeName, content: content });
+        } catch (e) {
+          sendTo(ws, { type: "knowledge_content", name: safeName, content: "", error: "File not found" });
+        }
+      }
+      return true;
+    }
+
+    if (msg.type === "knowledge_save") {
+      if (!msg.name || typeof msg.content !== "string") return true;
+      var safeName = path.basename(msg.name);
+      if (!safeName.endsWith(".md") && !safeName.endsWith(".jsonl")) safeName += ".md";
+      var knowledgeDir = path.join(cwd, "knowledge");
+      fs.mkdirSync(knowledgeDir, { recursive: true });
+      fs.writeFileSync(path.join(knowledgeDir, safeName), msg.content);
+      sendTo(ws, { type: "knowledge_saved", name: safeName });
+      sendTo(ws, { type: "knowledge_list", files: listKnowledgeFilesBasic() });
+      return true;
+    }
+
+    if (msg.type === "knowledge_delete") {
+      if (!msg.name) return true;
+      var safeName = path.basename(msg.name);
+      var filePath = path.join(cwd, "knowledge", safeName);
+      try { fs.unlinkSync(filePath); } catch (e) {}
+      sendTo(ws, { type: "knowledge_deleted", name: safeName });
+      sendTo(ws, { type: "knowledge_list", files: listKnowledgeFilesBasic() });
+      return true;
+    }
+
+    if (msg.type === "knowledge_promote") {
+      if (!isMate || !msg.name) return true;
+      var safeName = path.basename(msg.name);
+      var mateCtx = matesModule.buildMateCtx(getProjectOwnerId());
+      var thisMateId = path.basename(cwd);
+      var mate = matesModule.getMate(mateCtx, thisMateId);
+      var mateName = (mate && mate.name) || null;
+      matesModule.promoteKnowledge(mateCtx, thisMateId, mateName, safeName);
+      sendTo(ws, { type: "knowledge_promoted", name: safeName });
+      sendTo(ws, { type: "knowledge_list", files: listKnowledgeFiles() });
+      return true;
+    }
+
+    if (msg.type === "knowledge_depromote") {
+      if (!isMate || !msg.name) return true;
+      var safeName = path.basename(msg.name);
+      var mateCtx = matesModule.buildMateCtx(getProjectOwnerId());
+      var thisMateId = path.basename(cwd);
+      matesModule.depromoteKnowledge(mateCtx, thisMateId, safeName);
+      sendTo(ws, { type: "knowledge_depromoted", name: safeName });
+      sendTo(ws, { type: "knowledge_list", files: listKnowledgeFiles() });
+      return true;
+    }
+
+    return false;
+  }
+
+  return {
+    handleKnowledgeMessage: handleKnowledgeMessage,
+  };
+}
+
+module.exports = { attachKnowledge: attachKnowledge };