index.js•13 kB
#!/usr/bin/env node
import { z } from 'zod';
import { OpenAlService } from './services/openalService.js';
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
const flightService = new OpenAlService();
// 创建服务器
const server = new McpServer({
name: "tripmatch-mcp",
version: "0.0.1",
});
// 注册工具: 通过出发地和目的地查询航班
server.tool("searchFlightsByDepArr", "Search for flights between airports or cities by date. For cities with multiple airports, use depcity and arrcity parameters; otherwise use dep and arr parameters. Date must be in YYYY-MM-DD format. For today's date, use the getTodayDate tool. All airport/city codes must be valid IATA 3-letter codes (e.g.BJS for city of Beijing, PEK for Beijing Capital Airport).", {
dep: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Departure airport IATA 3-letter code (e.g. PEK for Beijing, CAN for Guangzhou)")
.optional(),
depcity: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Departure city IATA 3-letter code (e.g. BJS for Beijing, CAN for Guangzhou)")
.optional(),
arr: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Arrival airport IATA 3-letter code (e.g. SHA for Shanghai, HFE for Hefei)")
.optional(),
arrcity: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Arrival city IATA 3-letter code (e.g. SHA for Shanghai, BJS for Beijing)")
.optional(),
date: z.string()
.regex(/^\d{4}-\d{2}-\d{2}$/)
.describe("Flight date in YYYY-MM-DD format. IMPORTANT: If user input only cotains month and date, you should use getTodayDate tool to get the year. For today's date, use getTodayDate tool instead of hardcoding")
}, async ({ dep, depcity, arr, arrcity, date }) => {
try {
const flights = await flightService.getFlightsByDepArr(dep, depcity, arr, arrcity, date);
return {
content: [
{
type: "text",
text: JSON.stringify(flights, null, 2)
}
]
};
}
catch (error) {
console.error('Error searching flights by dep/arr:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 注册工具: 通过航班号查询航班
server.tool("searchFlightsByNumber", "Search flights by flight number and date. Flight number should include airline code (e.g. MU2157, CZ3969). dep and arr are optional, keep empty if you don't know them. Date format: YYYY-MM-DD. IMPORTANT: For today's date, you MUST use getTodayDate tool instead of hardcoding any date. Airport codes (optional) should be IATA 3-letter codes. ", {
fnum: z.string()
.regex(/^[A-Z0-9]{2,3}[0-9]{1,4}$/)
.describe("Flight number including airline code (e.g. MU2157, CZ3969)"),
date: z.string()
.regex(/^\d{4}-\d{2}-\d{2}$/)
.describe("Flight date in YYYY-MM-DD format. IMPORTANT: If user input only cotains month and date, you should use getTodayDate tool to get the year. For today's date, use getTodayDate tool instead of hardcoding"),
dep: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Departure airport IATA 3-letter code (e.g. HFE for Hefei)")
.optional(),
arr: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Arrival airport IATA 3-letter code (e.g. CAN for Guangzhou)")
.optional(),
}, async ({ fnum, date, dep, arr }) => {
try {
const flights = await flightService.getFlightByNumber(fnum, date, dep, arr);
return {
content: [
{
type: "text",
text: JSON.stringify(flights, null, 2)
}
]
};
}
catch (error) {
console.error('Error searching flights by number:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
server.tool("getFlightAndTrainTransferInfo", "Get flight and train transfer info by departure city and arrival city and departure date. Date format: YYYY-MM-DD. IMPORTANT: For today's date, you MUST use getTodayDate tool instead of hardcoding any date. Airport codes should be IATA 3-letter codes. ", {
depdate: z.string()
.regex(/^\d{4}-\d{2}-\d{2}$/)
.describe("Flight date in YYYY-MM-DD format. IMPORTANT: If user input only cotains month and date, you should use getTodayDate tool to get the year. For today's date, use getTodayDate tool instead of hardcoding"),
depcity: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Departure airport IATA 3-letter code (e.g. BJS for Beijing, CAN for Guangzhou)"),
arrcity: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Arrival airport IATA 3-letter code (e.g. SHA for Shanghai, LAX for Los Angeles)"),
}, async ({ depcity, arrcity, depdate }) => {
try {
const flights = await flightService.getTransferInfo(depcity, arrcity, depdate);
return {
content: [
{
type: "text",
text: JSON.stringify(flights, null, 2)
}
]
};
}
catch (error) {
console.error('Error searching flights by number:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 注册工具: 获取航班舒适度指数
server.tool("flightHappinessIndex", "using this tool when you need information related to following topics: Detailed flight comparisons (punctuality, amenities, cabin specs),Health safety protocols for air travel,Baggage allowance verification,Environmental impact assessments,Aircraft configuration visualization,Comfort-focused trip planning (seat dimensions, entertainment, food). etc.", {
fnum: z.string()
.regex(/^[A-Z0-9]{2,3}[0-9]{1,4}$/)
.describe("Flight number including airline code (e.g. MU2157, CZ3969)"),
date: z.string()
.regex(/^\d{4}-\d{2}-\d{2}$/)
.describe("Flight date in YYYY-MM-DD format. IMPORTANT: If user input only cotains month and date, you should use getTodayDate tool to get the year. For today's date, use getTodayDate tool instead of hardcoding"),
dep: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Departure airport IATA 3-letter code (e.g. HFE for Hefei)")
.optional(),
arr: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Arrival airport IATA 3-letter code (e.g. CAN for Guangzhou)")
.optional(),
}, async ({ fnum, date, dep, arr }) => {
try {
const flights = await flightService.getFlightHappinessIndex(fnum, date, dep, arr);
return {
content: [
{
type: "text",
text: JSON.stringify(flights, null, 2)
}
]
};
}
catch (error) {
console.error('Error searching flights by number:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 注册工具: 获取今天的日期
server.tool("getTodayDate", "Get today's date in local timezone (YYYY-MM-DD format). Use this tool whenever you need today's date - NEVER hardcode dates.", {
random_string: z.string()
.optional()
.describe("Dummy parameter for no-parameter tools")
}, async () => {
const today = new Date();
const year = today.getFullYear();
const month = String(today.getMonth() + 1).padStart(2, '0');
const day = String(today.getDate()).padStart(2, '0');
const todayDate = `${year}-${month}-${day}`;
return {
content: [
{
type: "text",
text: todayDate
}
]
};
});
// 注册工具：获取机场天气
server.tool('getFutureWeatherByAirport', 'Get airport future weather for 3days (today、tomorrow、the day after tomorrow) by airport IATA 3-letter code. Airport codes should be IATA 3-letter codes (e.g. PEK for Beijing, SHA for Shanghai, CAN for Guangzhou, HFE for Hefei).', {
airport: z.string()
.regex(/^[A-Z]{3}$/)
.describe("Airport IATA 3-letter code (e.g. PEK for Beijing, SHA for Shanghai, CAN for Guangzhou, HFE for Hefei)")
}, async ({ airport }) => {
try {
const weather = await flightService.getAirportWeather(airport);
return {
content: [
{
type: "text",
text: JSON.stringify(weather, null, 2)
}
]
};
}
catch (error) {
console.error('Error getting airport weather:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 注册工具: 查询火车票信息
server.tool("searchTrainTickets", "Search for train tickets between two cities on a specific date. Date must be in YYYY-MM-DD format.", {
from: z.string().describe("Departure city name (e.g. 合肥)"),
to: z.string().describe("Arrival city name (e.g. 北京)"),
date: z.string()
.regex(/^\d{4}-\d{2}-\d{2}$/)
.describe("Travel date in YYYY-MM-DD format. IMPORTANT: If user input only cotains month and date, you should use getTodayDate tool to get the year. For today's date, use getTodayDate tool instead of hardcoding"),
}, async ({ from, to, date }) => {
try {
const trainTickets = await flightService.getTrainTickets(from, to, date);
return {
content: [
{
type: "text",
text: JSON.stringify(trainTickets, null, 2)
}
]
};
}
catch (error) {
console.error('Error searching train tickets:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 注册工具: 获取航班价格信息
server.tool("getFlightPriceByCities", "Get flight price information by departure city, arrival city, and departure date. All city codes must be valid IATA 3-letter codes (e.g. HFE for Hefei, CAN for Guangzhou). Date must be in YYYY-MM-DD format.", {
dep_city: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Departure city IATA 3-letter code (e.g. HFE for Hefei)"),
arr_city: z.string()
.length(3)
.regex(/^[A-Z]{3}$/)
.describe("Arrival city IATA 3-letter code (e.g. CAN for Guangzhou)"),
dep_date: z.string()
.regex(/^\d{4}-\d{2}-\d{2}$/)
.describe("Departure date in YYYY-MM-DD format. IMPORTANT: If user input only cotains month and date, you should use getTodayDate tool to get the year. For today's date, use getTodayDate tool instead of hardcoding"),
}, async ({ dep_city, arr_city, dep_date }) => {
try {
const flightPrices = await flightService.getFlightPriceByCities(dep_city, arr_city, dep_date);
return {
content: [
{
type: "text",
text: JSON.stringify(flightPrices, null, 2)
}
]
};
}
catch (error) {
console.error('Error getting flight prices by cities:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 注册工具: 查询火车站信息
server.tool("searchTrainStations", "Search for train stations by keyword.", {
query: z.string().describe("Keyword to search for train stations (e.g. 北京西)"),
}, async ({ query }) => {
try {
const trainStations = await flightService.searchTrainStations(query);
return {
content: [
{
type: "text",
text: JSON.stringify(trainStations, null, 2)
}
]
};
}
catch (error) {
console.error('Error searching train stations:', error);
return {
content: [{ type: "text", text: `Error: ${error.message}` }],
isError: true
};
}
});
// 连接传输并启动服务器
const transport = new StdioServerTransport();
// 启动服务器并处理错误
async function startServer() {
try {
await server.connect(transport);
}
catch (error) {
process.exit(1);
}
}
startServer().catch((error) => {
console.error('Unhandled error:', error);
process.exit(1);
});