import { Server } from '@modelcontextprotocol/sdk/server/index.js';
import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
import { WebSocketServer } from '../websocket-server.js';
import { CreateNoteSchema } from '../schemas/remnote-schemas.js';
import { SearchSchema } from '../schemas/remnote-schemas.js';
import { ReadNoteSchema } from '../schemas/remnote-schemas.js';
import { UpdateNoteSchema } from '../schemas/remnote-schemas.js';
import { AppendJournalSchema } from '../schemas/remnote-schemas.js';
export const CREATE_NOTE_TOOL = {
name: 'remnote_create_note',
description: 'Create a new note in RemNote with optional content, parent, and tags',
inputSchema: {
type: 'object' as const,
properties: {
title: { type: 'string', description: 'The title of the note' },
content: { type: 'string', description: 'Content as child bullets (newline-separated)' },
parentId: { type: 'string', description: 'Parent Rem ID' },
tags: { type: 'array', items: { type: 'string' }, description: 'Tags to apply' },
},
required: ['title'],
},
};
export const SEARCH_TOOL = {
name: 'remnote_search',
description: 'Search the RemNote knowledge base for notes matching a query',
inputSchema: {
type: 'object' as const,
properties: {
query: { type: 'string', description: 'Search query text' },
limit: { type: 'number', description: 'Maximum results (1-100, default: 20)' },
includeContent: { type: 'boolean', description: 'Include child content (default: false)' },
},
required: ['query'],
},
};
export const READ_NOTE_TOOL = {
name: 'remnote_read_note',
description: 'Read a specific note from RemNote by its Rem ID',
inputSchema: {
type: 'object' as const,
properties: {
remId: { type: 'string', description: 'The Rem ID to read' },
depth: { type: 'number', description: 'Depth of children to include (0-10, default: 3)' },
},
required: ['remId'],
},
};
export const UPDATE_NOTE_TOOL = {
name: 'remnote_update_note',
description: 'Update an existing note in RemNote (change title, append content, or modify tags)',
inputSchema: {
type: 'object' as const,
properties: {
remId: { type: 'string', description: 'The Rem ID to update' },
title: { type: 'string', description: 'New title' },
appendContent: { type: 'string', description: 'Content to append as children' },
addTags: { type: 'array', items: { type: 'string' }, description: 'Tags to add' },
removeTags: { type: 'array', items: { type: 'string' }, description: 'Tags to remove' },
},
required: ['remId'],
},
};
export const APPEND_JOURNAL_TOOL = {
name: 'remnote_append_journal',
description: "Append content to today's daily document in RemNote",
inputSchema: {
type: 'object' as const,
properties: {
content: { type: 'string', description: "Content to append to today's daily document" },
timestamp: { type: 'boolean', description: 'Include timestamp (default: true)' },
},
required: ['content'],
},
};
export const STATUS_TOOL = {
name: 'remnote_status',
description: 'Check the connection status and statistics of the RemNote MCP bridge',
inputSchema: {
type: 'object' as const,
properties: {},
},
};
export function registerAllTools(server: Server, wsServer: WebSocketServer) {
// Single CallTool handler for all tools
server.setRequestHandler(CallToolRequestSchema, async (request) => {
try {
switch (request.params.name) {
case 'remnote_create_note': {
const args = CreateNoteSchema.parse(request.params.arguments);
const result = await wsServer.sendRequest('create_note', args);
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
};
}
case 'remnote_search': {
const args = SearchSchema.parse(request.params.arguments);
const result = await wsServer.sendRequest('search', args);
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
};
}
case 'remnote_read_note': {
const args = ReadNoteSchema.parse(request.params.arguments);
const result = await wsServer.sendRequest('read_note', args);
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
};
}
case 'remnote_update_note': {
const args = UpdateNoteSchema.parse(request.params.arguments);
const result = await wsServer.sendRequest('update_note', args);
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
};
}
case 'remnote_append_journal': {
const args = AppendJournalSchema.parse(request.params.arguments);
const result = await wsServer.sendRequest('append_journal', args);
return {
content: [{ type: 'text', text: JSON.stringify(result, null, 2) }],
};
}
case 'remnote_status': {
const connected = wsServer.isConnected();
if (!connected) {
return {
content: [
{
type: 'text',
text: JSON.stringify(
{ connected: false, message: 'RemNote plugin not connected' },
null,
2
),
},
],
};
}
const result = await wsServer.sendRequest('status', {});
return {
content: [
{
type: 'text',
text: JSON.stringify(
{ connected: true, ...(typeof result === 'object' ? result : {}) },
null,
2
),
},
],
};
}
default:
throw new Error(`Unknown tool: ${request.params.name}`);
}
} catch (error) {
return {
content: [
{
type: 'text',
text: `Error: ${error instanceof Error ? error.message : String(error)}`,
},
],
isError: true,
};
}
});
// Register list_tools handler
server.setRequestHandler(ListToolsRequestSchema, async () => ({
tools: [
CREATE_NOTE_TOOL,
SEARCH_TOOL,
READ_NOTE_TOOL,
UPDATE_NOTE_TOOL,
APPEND_JOURNAL_TOOL,
STATUS_TOOL,
],
}));
}
