Duck Duck MCP
by qwang07
#!/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 { search, SafeSearchType } from 'duck-duck-scrape';
import { z } from "zod";
import { zodToJsonSchema } from "zod-to-json-schema";
// 服务器配置
const SERVER_CONFIG = {
name: "search-server",
version: "1.0.0",
};
const SearchArgsSchema = z.object({
query: z.string(),
options: z.object({
region: z.string().default('zh-cn'),
safeSearch: z.enum(['OFF', 'MODERATE', 'STRICT']).default('MODERATE'),
numResults: z.number().default(50)
}).optional()
});
function detectContentType(result) {
const url = result.url.toLowerCase();
if (url.includes('docs.') || url.includes('/docs/') || url.includes('/documentation/')) {
return 'documentation';
}
if (url.includes('github.com') || url.includes('stackoverflow.com')) {
return 'documentation';
}
if (url.includes('twitter.com') || url.includes('facebook.com') || url.includes('linkedin.com')) {
return 'social';
}
return 'article';
}
function detectLanguage(query) {
return /[\u4e00-\u9fa5]/.test(query) ? 'zh-cn' : 'en';
}
function detectTopics(results) {
const topics = new Set();
results.forEach(result => {
if (result.title.toLowerCase().includes('github')) topics.add('technology');
if (result.title.toLowerCase().includes('docs')) topics.add('documentation');
});
return Array.from(topics);
}
function processSearchResults(results, query, options) {
return {
type: 'search_results',
data: results.map(result => ({
title: result.title.replace(/'/g, "'").replace(/"/g, '"'),
url: result.url,
description: result.description.trim(),
metadata: {
type: detectContentType(result),
source: new URL(result.url).hostname
}
})),
metadata: {
query,
timestamp: new Date().toISOString(),
resultCount: results.length,
searchContext: {
region: options.region || 'zh-cn',
safeSearch: options.safeSearch?.toString() || 'MODERATE'
},
queryAnalysis: {
language: detectLanguage(query),
topics: detectTopics(results)
}
}
};
}
const server = new Server(
SERVER_CONFIG,
{
capabilities: {
tools: {},
},
}
);
server.setRequestHandler(ListToolsRequestSchema, async () => {
return {
tools: [
{
name: "search",
description: "Search the web using DuckDuckGo",
inputSchema: zodToJsonSchema(SearchArgsSchema),
}
],
};
});
server.setRequestHandler(CallToolRequestSchema, async (request) => {
try {
const { name, arguments: args } = request.params;
if (name !== "search") {
throw Object.assign(
new Error(`Unknown tool: ${name}`),
{ errorType: 'UNKNOWN_TOOL', name }
);
}
const parsed = SearchArgsSchema.safeParse(args);
if (!parsed.success) {
throw Object.assign(
new Error(`Invalid arguments: ${parsed.error}`),
{ errorType: 'INVALID_ARGS', details: parsed.error }
);
}
const searchResults = await search(parsed.data.query, {
region: parsed.data.options?.region || 'zh-cn',
safeSearch: parsed.data.options?.safeSearch ? SafeSearchType[parsed.data.options.safeSearch] : SafeSearchType.MODERATE,
maxResults: parsed.data.options?.numResults || 50
});
const response = processSearchResults(
searchResults.results,
parsed.data.query,
{
region: parsed.data.options?.region,
safeSearch: parsed.data.options?.safeSearch ? SafeSearchType[parsed.data.options.safeSearch] : SafeSearchType.MODERATE
}
);
return {
content: [{
type: "text",
text: JSON.stringify(response, null, 2)
}]
};
} catch (error) {
const errorResponse = {
type: 'search_error',
message: error instanceof Error ? error.message : String(error),
suggestion: '你可以尝试：1. 修改搜索关键词 2. 减少结果数量 3. 更换地区',
context: {
query: request.params.arguments?.query,
options: request.params.arguments?.options
}
};
return {
content: [{
type: "text",
text: JSON.stringify(errorResponse, null, 2)
}],
isError: true
};
}
});
async function runServer() {
const transport = new StdioServerTransport();
await server.connect(transport);
console.error("MCP Search Server running on stdio");
}
runServer().catch((error) => {
console.error("Fatal error running server:", error);
process.exit(1);
});