history.ts•3.03 kB
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { GraphQLClient } from "../graphqlClient.js";
import { text } from "../util/mcp.js";
import { z } from "zod";
export function registerHistoryTools(server: McpServer, gql: GraphQLClient, defaults: { workspaceId?: string }) {
const listHistoriesHandler = async (parsed: { workspaceId?: string; guid: string; take?: number; before?: string }) => {
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<{ workspace: any }>(query, { workspaceId, guid: parsed.guid, take: parsed.take, before: parsed.before });
return text(data.workspace.histories);
};
server.registerTool(
"affine_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 as any
);
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 as any
);
const recoverDocHandler = async (parsed: { workspaceId?: string; guid: string; timestamp: string }) => {
const workspaceId = parsed.workspaceId || defaults.workspaceId || parsed.workspaceId;
if (!workspaceId) throw new Error("workspaceId required (or set AFFINE_WORKSPACE_ID)");
const mutation = `mutation Recover($workspaceId:String!,$guid:String!,$timestamp:DateTime!){ recoverDoc(workspaceId:$workspaceId, guid:$guid, timestamp:$timestamp) }`;
const data = await gql.request<{ recoverDoc: string }>(mutation, { workspaceId, guid: parsed.guid, timestamp: parsed.timestamp });
return text({ recoveredAt: data.recoverDoc });
};
server.registerTool(
"affine_recover_doc",
{
title: "Recover Document",
description: "Recover a doc to a previous timestamp.",
inputSchema: {
workspaceId: z.string().optional(),
guid: z.string(),
timestamp: z.string()
}
},
recoverDocHandler as any
);
server.registerTool(
"recover_doc",
{
title: "Recover Document",
description: "Recover a doc to a previous timestamp.",
inputSchema: {
workspaceId: z.string().optional(),
guid: z.string(),
timestamp: z.string()
}
},
recoverDocHandler as any
);
}