#!/usr/bin/env node
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
import {
CallToolRequestSchema,
ListToolsRequestSchema,
} from '@modelcontextprotocol/sdk/types.js';
import { fetchUrl } from './fetcher.js';
import { optimizeContent } from './llm.js';
import { encode } from '@toon-format/toon';
const server = new Server(
{
name: 'toon-fetch',
version: '1.0.0',
},
{
capabilities: {
tools: {},
},
}
);
server.setRequestHandler(ListToolsRequestSchema, async () => {
return {
tools: [
{
name: 'fetch_url',
description: 'Fetches content from a URL, converts it to Markdown, and optionally optimizes it using a local LLM. Returns data in TOON format.',
inputSchema: {
type: 'object',
properties: {
url: {
type: 'string',
description: 'The URL to fetch',
},
prompt: {
type: 'string',
description: 'Optional prompt for local LLM optimization (e.g., "Summarize this", "Extract dates"). If omitted, raw content is returned.',
},
},
required: ['url'],
},
},
],
};
});
server.setRequestHandler(CallToolRequestSchema, async (request) => {
if (request.params.name === 'fetch_url') {
const url = String(request.params.arguments?.url);
const prompt = request.params.arguments?.prompt ? String(request.params.arguments.prompt) : '';
try {
const result = await fetchUrl(url);
let content = result.content;
if (prompt) {
content = await optimizeContent(content, prompt);
}
const responseData = {
title: result.title,
content: content,
metadata: {
url: result.url,
excerpt: result.excerpt,
siteName: result.siteName,
byline: result.byline,
optimized: !!prompt
}
};
// Serialize to TOON
const toonOutput = encode(responseData);
return {
content: [
{
type: 'text',
text: toonOutput,
mimeType: 'text/toon',
},
],
};
} catch (error) {
return {
content: [
{
type: 'text',
text: `Error fetching URL: ${error instanceof Error ? error.message : String(error)}`,
},
],
isError: true,
};
}
}
throw new Error(`Tool not found: ${request.params.name}`);
});
async function main() {
const transport = new StdioServerTransport();
await server.connect(transport);
console.error('Toto Fetch MCP Server running on stdio');
}
main().catch((error) => {
console.error('Fatal error in main():', error);
process.exit(1);
});
