@mseep/affine-mcp-server 2.3.0

package/dist/tools/history.js

@@ -0,0 +1,22 @@
+import { text } from "../util/mcp.js";
+import { z } from "zod";
+export function registerHistoryTools(server, gql, defaults) {
+    const listHistoriesHandler = async (parsed) => {
+        const workspaceId = parsed.workspaceId || defaults.workspaceId || parsed.workspaceId;
+        if (!workspaceId)
+            throw new Error("workspaceId required (or set AFFINE_WORKSPACE_ID)");
+        const query = `query Histories($workspaceId:String!,$guid:String!,$take:Int,$before:DateTime){ workspace(id:$workspaceId){ histories(guid:$guid, take:$take, before:$before){ id timestamp workspaceId } } }`;
+        const data = await gql.request(query, { workspaceId, guid: parsed.guid, take: parsed.take, before: parsed.before });
+        return text(data.workspace.histories);
+    };
+    server.registerTool("list_histories", {
+        title: "List Histories",
+        description: "List doc histories (timestamps) for a doc.",
+        inputSchema: {
+            workspaceId: z.string().optional(),
+            guid: z.string(),
+            take: z.number().optional(),
+            before: z.string().optional()
+        }
+    }, listHistoriesHandler);
+}

package/dist/tools/icons.js

@@ -0,0 +1,125 @@
+import { z } from "zod";
+import { receipt } from "../util/mcp.js";
+import { connectWorkspaceSocket, joinWorkspace, wsUrlFromGraphQLEndpoint, } from "../ws.js";
+import { docIconKey, folderIconKey, getExplorerIcon, normalizeIconInput, setExplorerIcon, } from "../util/explorerIcon.js";
+/**
+ * Zod shape for the `icon` parameter shared by both setters. Accepts an emoji
+ * shorthand string, a full `{type:"emoji",unicode}` / `{type:"icon",name}`
+ * object, or `null` to clear the icon.
+ */
+const iconSchema = z
+    .union([
+    z.string(),
+    z.object({ type: z.literal("emoji"), unicode: z.string() }),
+    z.object({ type: z.literal("icon"), name: z.string() }),
+    z.null(),
+])
+    .describe('Emoji shorthand ("🧪"), a full object ({type:"emoji",unicode:"🧪"} or ' +
+    '{type:"icon",name:"check"}), or null to remove the icon.');
+/**
+ * Registers the explorer-icon tools: set/clear and read the Notion-style
+ * sidebar icon on a document or an organize folder. All four share the
+ * `explorerIcon` sub-doc helper so the storage model lives in one place.
+ */
+export function registerIconTools(server, gql, defaults) {
+    function resolveWorkspaceId(workspaceId) {
+        const resolved = workspaceId || defaults.workspaceId;
+        if (!resolved) {
+            throw new Error("workspaceId is required. Provide it as a parameter or set AFFINE_WORKSPACE_ID.");
+        }
+        return resolved;
+    }
+    async function withSocket(workspaceId, fn) {
+        const wsUrl = wsUrlFromGraphQLEndpoint(gql.endpoint);
+        const socket = await connectWorkspaceSocket(wsUrl, gql.cookie, gql.bearer);
+        try {
+            await joinWorkspace(socket, workspaceId);
+            return await fn(socket);
+        }
+        finally {
+            socket.disconnect();
+        }
+    }
+    // ─── update_doc_icon ────────────────────────────────────────────────────────
+    const updateDocIconHandler = async (parsed) => {
+        const workspaceId = resolveWorkspaceId(parsed.workspaceId);
+        const icon = normalizeIconInput(parsed.icon);
+        const result = await withSocket(workspaceId, (socket) => setExplorerIcon(socket, workspaceId, docIconKey(parsed.docId), icon));
+        return receipt("doc.update_icon", {
+            workspaceId,
+            docId: parsed.docId,
+            icon: result.icon,
+            cleared: result.icon === null,
+        });
+    };
+    server.registerTool("update_doc_icon", {
+        title: "Update Document Icon",
+        description: "Set or clear the sidebar icon (the Notion-style emoji slot) on a document. " +
+            "Pass an emoji string, a full icon object, or null to remove it.",
+        inputSchema: {
+            workspaceId: z.string().optional(),
+            docId: z.string().describe("The document whose icon to update."),
+            icon: iconSchema,
+        },
+    }, updateDocIconHandler);
+    // ─── update_folder_icon ─────────────────────────────────────────────────────
+    const updateFolderIconHandler = async (parsed) => {
+        const workspaceId = resolveWorkspaceId(parsed.workspaceId);
+        const icon = normalizeIconInput(parsed.icon);
+        const result = await withSocket(workspaceId, (socket) => setExplorerIcon(socket, workspaceId, folderIconKey(parsed.folderId), icon));
+        return receipt("folder.update_icon", {
+            workspaceId,
+            folderId: parsed.folderId,
+            icon: result.icon,
+            cleared: result.icon === null,
+        });
+    };
+    server.registerTool("update_folder_icon", {
+        title: "Update Folder Icon",
+        description: "Set or clear the sidebar icon on an organize folder. " +
+            "Pass an emoji string, a full icon object, or null to remove it. Experimental.",
+        inputSchema: {
+            workspaceId: z.string().optional(),
+            folderId: z.string().describe("The organize folder whose icon to update."),
+            icon: iconSchema,
+        },
+    }, updateFolderIconHandler);
+    // ─── get_doc_icon ───────────────────────────────────────────────────────────
+    const getDocIconHandler = async (parsed) => {
+        const workspaceId = resolveWorkspaceId(parsed.workspaceId);
+        const result = await withSocket(workspaceId, (socket) => getExplorerIcon(socket, workspaceId, docIconKey(parsed.docId)));
+        return receipt("doc.get_icon", {
+            workspaceId,
+            docId: parsed.docId,
+            icon: result.icon,
+            hasIcon: result.icon !== null,
+        });
+    };
+    server.registerTool("get_doc_icon", {
+        title: "Get Document Icon",
+        description: "Read the current sidebar icon of a document. Returns null when none is set.",
+        inputSchema: {
+            workspaceId: z.string().optional(),
+            docId: z.string().describe("The document whose icon to read."),
+        },
+    }, getDocIconHandler);
+    // ─── get_folder_icon ────────────────────────────────────────────────────────
+    const getFolderIconHandler = async (parsed) => {
+        const workspaceId = resolveWorkspaceId(parsed.workspaceId);
+        const result = await withSocket(workspaceId, (socket) => getExplorerIcon(socket, workspaceId, folderIconKey(parsed.folderId)));
+        return receipt("folder.get_icon", {
+            workspaceId,
+            folderId: parsed.folderId,
+            icon: result.icon,
+            hasIcon: result.icon !== null,
+        });
+    };
+    server.registerTool("get_folder_icon", {
+        title: "Get Folder Icon",
+        description: "Read the current sidebar icon of an organize folder. Returns null when none is set. Experimental.",
+        inputSchema: {
+            workspaceId: z.string().optional(),
+            folderId: z.string().describe("The organize folder whose icon to read."),
+        },
+    }, getFolderIconHandler);
+}

package/dist/tools/notifications.js

@@ -0,0 +1,79 @@
+import { z } from "zod";
+import { text } from "../util/mcp.js";
+export function registerNotificationTools(server, gql) {
+    // LIST NOTIFICATIONS
+    const listNotificationsHandler = async ({ first = 20, offset, after, unreadOnly = false }) => {
+        try {
+            const query = `
+        query GetNotifications($pagination: PaginationInput!) {
+          currentUser {
+            notifications(pagination: $pagination) {
+              edges {
+                cursor
+                node {
+                  id
+                  type
+                  body
+                  read
+                  level
+                  createdAt
+                  updatedAt
+                }
+              }
+              totalCount
+              pageInfo {
+                hasNextPage
+                endCursor
+              }
+            }
+          }
+        }
+      `;
+            const data = await gql.request(query, {
+                pagination: {
+                    first,
+                    offset,
+                    after
+                }
+            });
+            let notifications = (data.currentUser?.notifications?.edges || []).map((edge) => edge.node);
+            if (unreadOnly) {
+                notifications = notifications.filter((n) => !n.read);
+            }
+            return text(notifications);
+        }
+        catch (error) {
+            return text({ error: error.message });
+        }
+    };
+    server.registerTool("list_notifications", {
+        title: "List Notifications",
+        description: "Get user notifications.",
+        inputSchema: {
+            first: z.number().optional().describe("Number of notifications to fetch"),
+            offset: z.number().optional().describe("Offset for pagination"),
+            after: z.string().optional().describe("Cursor for pagination"),
+            unreadOnly: z.boolean().optional().describe("Show only unread notifications")
+        }
+    }, listNotificationsHandler);
+    // MARK ALL NOTIFICATIONS READ
+    const readAllNotificationsHandler = async () => {
+        try {
+            const mutation = `
+        mutation ReadAllNotifications {
+          readAllNotifications
+        }
+      `;
+            const data = await gql.request(mutation);
+            return text({ success: data.readAllNotifications, message: "All notifications marked as read" });
+        }
+        catch (error) {
+            return text({ error: error.message });
+        }
+    };
+    server.registerTool("read_all_notifications", {
+        title: "Mark All Notifications Read",
+        description: "Mark all notifications as read.",
+        inputSchema: {}
+    }, readAllNotificationsHandler);
+}