SenMeshNet/electron/bridges/mcpServerBridge.attachments.test.cjs

const test = require("node:test");
const assert = require("node:assert/strict");
const fs = require("node:fs");
const os = require("node:os");
const path = require("node:path");

const bridge = require("./mcpServerBridge.cjs");

test("registered chat attachments can be listed and read by path or filename", async (t) => {
  t.after(() => bridge.cleanup());

  bridge.updateAttachmentMetadata([
    {
      filename: "note.txt",
      mediaType: "text/plain",
      filePath: "/tmp/netcatty-note.txt",
      base64Data: Buffer.from("hello attachment").toString("base64"),
    },
  ], "chat-a");

  const listed = bridge.handleListAttachments({ chatSessionId: "chat-a" });
  assert.equal(listed.ok, true);
  assert.deepEqual(listed.attachments, [{
    filename: "note.txt",
    mediaType: "text/plain",
    filePath: "/tmp/netcatty-note.txt",
    sizeBytes: 16,
  }]);

  const byPath = bridge.handleReadAttachment({
    chatSessionId: "chat-a",
    filePath: "/tmp/netcatty-note.txt",
  });
  assert.equal(byPath.ok, true);
  assert.equal(byPath.text, "hello attachment");

  const byName = bridge.handleReadAttachment({
    chatSessionId: "chat-a",
    filename: "note.txt",
  });
  assert.equal(byName.ok, true);
  assert.equal(byName.base64Data, Buffer.from("hello attachment").toString("base64"));
});

test("attachment reads reject unregistered local paths", async (t) => {
  const dir = fs.mkdtempSync(path.join(os.tmpdir(), "netcatty-attachment-test-"));
  const secretPath = path.join(dir, "secret.txt");
  fs.writeFileSync(secretPath, "secret");
  t.after(() => {
    bridge.cleanup();
    fs.rmSync(dir, { recursive: true, force: true });
  });

  bridge.updateAttachmentMetadata([
    {
      filename: "allowed.txt",
      mediaType: "text/plain",
      filePath: path.join(dir, "allowed.txt"),
      base64Data: Buffer.from("allowed").toString("base64"),
    },
  ], "chat-a");

  const result = bridge.handleReadAttachment({
    chatSessionId: "chat-a",
    filePath: secretPath,
  });

  assert.equal(result.ok, false);
  assert.match(result.error, /not registered/i);
});