rootr-cli 0.2.1 → 0.4.0

package/lib/mcp-tools/attachments.js

@@ -0,0 +1,110 @@
+import { readFileSync } from 'node:fs';
+
+import { z } from 'zod';
+
+import { jsonResult, errorResult, makeClient, requireWorkspace } from './shared.js';
+import { resolveTarget, requireResolved } from '../resolve.js';
+import { RootrApiError } from '../client.js';
+
+/** Reads file bytes from a local path or a base64 string; exactly one must be given. */
+function readAttachmentBytes({ filePath, base64Data }) {
+  if (filePath) return readFileSync(filePath);
+  if (base64Data) return Buffer.from(base64Data, 'base64');
+  throw new RootrApiError('filePath 또는 base64Data 중 하나가 필요합니다.');
+}
+
+const uploadInputSchema = {
+  filename: z.string().describe('File name to store, e.g. "screenshot.png"'),
+  mimeType: z.string().describe('MIME type, e.g. "image/png"'),
+  filePath: z
+    .string()
+    .optional()
+    .describe('Local filesystem path to the file to upload (the MCP server reads this itself, e.g. "/tmp/screenshot.png")'),
+  base64Data: z
+    .string()
+    .optional()
+    .describe('Base64-encoded file bytes; use this when the caller has the file in memory instead of on disk'),
+  nodeId: z.string().optional().describe('Document/node id to associate the attachment with (permission-checked)'),
+  workspace: z.string().optional().describe('Workspace id; defaults to ROOTR_WORKSPACE/config'),
+};
+
+export function registerAttachmentTools(server) {
+  server.registerTool(
+    'rootr_upload_attachment',
+    {
+      title: 'Upload a file/image as a Rootr attachment',
+      description:
+        'Upload a local file (screenshot, photo, diagram export, ...) or base64 bytes to the Rootr (루터) ' +
+        'workspace as an attachment, so it can be referenced in a document, presentation slide, or elsewhere. ' +
+        'Give exactly one of filePath (read from the local filesystem) or base64Data. Returns ' +
+        '{ id, url, markdown, filename, mimeType, size } — `url` is the public, unauthenticated ' +
+        '/attachments/{id}/raw link (renders in <img> tags and on published /p/ documents without login), and ' +
+        '`markdown` is ready to paste into a document body (![alt](url)). For the common case of inserting an ' +
+        'image straight into a document, prefer rootr_attach_image_to_document instead.',
+      annotations: { readOnlyHint: false, destructiveHint: false },
+      inputSchema: uploadInputSchema,
+    },
+    async ({ filename, mimeType, filePath, base64Data, nodeId, workspace }) => {
+      try {
+        const client = makeClient();
+        const ws = requireWorkspace(workspace);
+        const data = readAttachmentBytes({ filePath, base64Data });
+        const { id } = await client.uploadAttachment(ws, { filename, mimeType, nodeId, data });
+        const url = client.attachmentRawUrl(id);
+        return jsonResult({
+          id,
+          url,
+          markdown: `![${filename}](${url})`,
+          filename,
+          mimeType,
+          size: data.length,
+        });
+      } catch (err) {
+        return errorResult(err);
+      }
+    },
+  );
+
+  server.registerTool(
+    'rootr_attach_image_to_document',
+    {
+      title: 'Upload an image and insert it into a document',
+      description:
+        'Upload a local image (or base64 bytes) and append it — as `![alt](url)` markdown — into a Rootr (루터) ' +
+        'document in one step. This is the tool to use when an AI is writing a report/spec/inspection log that ' +
+        'needs a screenshot or photo embedded, since the web editor lets you drag-and-drop images but MCP callers ' +
+        'otherwise cannot. Give exactly one of filePath or base64Data. Uses rootr_append semantics (conflict-free, ' +
+        'additive) — pass underHeading to place it under a specific section.',
+      annotations: { readOnlyHint: false, destructiveHint: false },
+      inputSchema: {
+        target: z.string().describe('Document path (starting with "/") or node id to insert the image into'),
+        filename: uploadInputSchema.filename,
+        mimeType: uploadInputSchema.mimeType,
+        filePath: uploadInputSchema.filePath,
+        base64Data: uploadInputSchema.base64Data,
+        alt: z.string().optional().describe('Alt text / caption for the image (defaults to filename)'),
+        underHeading: z
+          .string()
+          .optional()
+          .describe('Heading to insert the image under, e.g. "## Screenshots". If omitted, appends to the document end.'),
+        workspace: z.string().optional().describe('Workspace id; defaults to ROOTR_WORKSPACE/config'),
+      },
+    },
+    async ({ target, filename, mimeType, filePath, base64Data, alt, underHeading, workspace }) => {
+      try {
+        const client = makeClient();
+        const ws = requireWorkspace(workspace);
+        const resolved = await resolveTarget(client, target);
+        const nodeId = requireResolved(resolved);
+        const data = readAttachmentBytes({ filePath, base64Data });
+        const { id } = await client.uploadAttachment(ws, { filename, mimeType, nodeId, data });
+        const url = client.attachmentRawUrl(id);
+        const markdown = `![${alt || filename}](${url})`;
+        const result = await client.appendDocument(nodeId, { content: markdown, underHeading });
+        return jsonResult({ id, url, markdown, appended: true, etag: result.etag });
+      } catch (err) {
+        return errorResult(err);
+      }
+    },
+  );
+}