#!/usr/bin/env node
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { z } from "zod";
import dotenv from "dotenv";
import { GazeSource } from "./sources/Gaze.source";
import { ShenQiZheSource } from "./sources/ShenQiZhe.source";
import { ImtlinkSource } from "./sources/Imtlink.source";
import { GazeValidatorService } from "./core/gaze.validator";
import { ShenQiZheValidatorService } from "./core/shenqizhe.validator";
import { ImtlinkValidatorService } from "./core/imtlink.validator";
import { SearchQuery } from "./types/index";
dotenv.config();
/**
* 电影搜索工具 MCP Server (STDIO 版本)
* 使用 STDIO 传输和 SDK 顶层封装
*/
// 创建 MCP Server 实例的工厂函数
const createMovieSearchServer = () => {
const server = new McpServer(
{
name: "movie-search-tool-stdio",
version: "1.0.0",
},
{
capabilities: {
tools: {},
logging: {},
},
}
);
// 初始化服务
const gazeSource = new GazeSource();
const shenQiZheSource = new ShenQiZheSource();
const imtlinkSource = new ImtlinkSource();
const gazeValidator = new GazeValidatorService();
const shenQiZheValidator = new ShenQiZheValidatorService();
const imtlinkValidator = new ImtlinkValidatorService();
/**
* 根据URL选择合适的验证器
*/
const getValidatorForUrl = (url: string) => {
if (url.includes("gaze.run")) {
return gazeValidator;
} else if (url.includes("shenqizhe.com")) {
return shenQiZheValidator;
} else if (url.includes("imtlink.com")) {
return imtlinkValidator;
} else {
// 默认使用 gaze 验证器
return gazeValidator;
}
};
// 注册电影搜索工具
server.tool(
"search_movie",
"搜索电影或电视剧资源。返回未验证的搜索结果列表，包含标题、链接和质量信息。使用此工具获取候选资源后，请从结果中选择最匹配的链接，然后使用 validate_video_url 工具验证其可播放性。",
{
title: z.string().describe("电影或电视剧的标题"),
type: z
.enum(["movie", "tv"])
.describe("内容类型：movie（电影）或 tv（电视剧）"),
season: z.number().min(1).optional().describe("季数（仅限电视剧）"),
episode: z.number().min(1).optional().describe("集数（仅限电视剧）"),
},
async ({ title, type, season, episode }) => {
try {
// 在 STDIO 模式下不输出日志，避免干扰 MCP 协议通信
const query: SearchQuery = {
title,
type,
...(season && { season }),
...(episode && { episode }),
};
// 并行搜索所有源
const sources = [gazeSource, shenQiZheSource, imtlinkSource];
const searchPromises = sources.map((source) => source.find(query));
const resultsFromAllSources = await Promise.all(searchPromises);
// 合并并展平结果
const searchResults = resultsFromAllSources.flat();
if (searchResults.length === 0) {
return {
content: [
{
type: "text",
text: JSON.stringify(
{
success: false,
message: `未找到 "${title}" 的任何资源`,
results: [],
total: 0,
next_action: "请尝试使用不同的关键词重新搜索",
},
null,
2
),
},
],
};
}
// 返回结构化的搜索结果
const structuredResults = {
success: true,
title: title,
type: type,
...(season && { season }),
...(episode && { episode }),
total: searchResults.length,
results: searchResults.map((result, index) => ({
id: index + 1,
url: result.url,
title: result.title,
quality: result.quality,
source: result.source,
verified: false,
})),
next_action:
"请从上述结果中选择最合适的链接，然后使用 validate_video_url 工具验证其可播放性",
};
return {
content: [
{
type: "text",
text: JSON.stringify(structuredResults, null, 2),
},
],
};
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
// 错误不输出到 stderr，避免干扰 MCP 协议
return {
content: [
{
type: "text",
text: JSON.stringify(
{
success: false,
error: errorMessage,
results: [],
total: 0,
},
null,
2
),
},
],
isError: true,
};
}
}
);
// 注册视频验证工具
server.tool(
"validate_video_url",
"验证特定视频链接的可播放性。接收一个视频播放页面的 URL 或 URL 数组，返回该链接是否可以正常播放。支持批量验证多个链接。只有通过验证的链接才能确保用户可以观看。",
{
url: z
.union([
z.string().describe("要验证的视频播放页面 URL"),
z.array(z.string()).describe("要批量验证的视频播放页面 URL 数组"),
])
.describe("要验证的视频播放页面 URL（单个或数组）"),
},
async ({ url }) => {
try {
// 在 STDIO 模式下不输出日志，避免干扰 MCP 协议通信
// 处理单个 URL 或 URL 数组
const urls = Array.isArray(url) ? url : [url];
// 并行验证所有 URL
const validationPromises = urls.map(async (singleUrl, index) => {
try {
// 根据URL选择合适的验证器
const validator = getValidatorForUrl(singleUrl);
const validatorName = singleUrl.includes("gaze.run")
? "Gaze"
: singleUrl.includes("shenqizhe.com")
? "ShenQiZhe"
: singleUrl.includes("imtlink.com")
? "Imtlink"
: "Default";
// 验证过程不输出日志
const isValid = await validator.isValid(singleUrl);
const result = {
url: singleUrl,
valid: isValid,
validator: validatorName,
status: isValid ? "可播放" : "无法播放",
message: isValid
? "视频链接验证成功，可以正常播放"
: "视频链接验证失败，可能已失效或无法访问",
timestamp: new Date().toISOString(),
};
return result;
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
// 验证错误不输出日志
return {
url: singleUrl,
valid: false,
validator: "Unknown",
status: "验证失败",
message: `验证过程中发生错误: ${errorMessage}`,
error: errorMessage,
timestamp: new Date().toISOString(),
};
}
});
const validationResults = await Promise.all(validationPromises);
// 统计结果
const validCount = validationResults.filter(
(result) => result.valid
).length;
const totalCount = validationResults.length;
const result = {
success: true,
total: totalCount,
valid: validCount,
invalid: totalCount - validCount,
results: validationResults,
summary: `共验证 ${totalCount} 个链接，${validCount} 个可播放，${totalCount - validCount} 个无法播放`,
timestamp: new Date().toISOString(),
};
return {
content: [
{
type: "text",
text: JSON.stringify(result, null, 2),
},
],
};
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
// 验证错误不输出日志
return {
content: [
{
type: "text",
text: JSON.stringify(
{
success: false,
url: url,
valid: false,
error: errorMessage,
timestamp: new Date().toISOString(),
},
null,
2
),
},
],
isError: true,
};
}
}
);
return server;
};
// 启动服务器
async function main() {
const server = createMovieSearchServer();
const transport = new StdioServerTransport();
await server.connect(transport);
// STDIO 模式下不输出启动日志，避免干扰 MCP 协议通信
}
main().catch((error) => {
// 启动失败也不输出到 stderr，静默退出
process.exit(1);
});
