devops-mcp-server-extension 1.0.0

package/dist/common/types.js

@@ -0,0 +1,60 @@
+// Organization types
+export { UserInfoSchema, CurrentUserSchema, GetOrganizationMembersSchema, OrganizationRoleSchema, OrganizationRole, ListOrganizationRolesSchema, GetOrganizationRoleSchema, GetOrganizationDepartmentAncestorsSchema, GetOrganizationDepartmentInfoSchema, GetOrganizationDepartmentsSchema, CurrentOrganizationInfoSchema, OrganizationInfoSchema, UserOrganizationsInfoSchema, DepartmentInfoSchema, OrganizationDepartmentsSchema, MemberInfoSchema, OrganizationMembersSchema, GetOrganizationMemberInfoSchema, GetOrganizationMemberByUserIdInfoSchema, SearchOrganizationMembersSchema, SearchOrganizationMembersResultSchema } from "../operations/organization/types.js";
+// Codeup types
+export { 
+// Branch schemas
+CreateBranchSchema, GetBranchSchema, DeleteBranchSchema, ListBranchesSchema, 
+// File schemas
+GetFileBlobsSchema, CreateFileSchema, UpdateFileSchema, DeleteFileSchema, ListFilesSchema, 
+// Repository schemas
+GetRepositorySchema, ListRepositoriesSchema, 
+// Compare schemas
+GetCompareSchema, 
+// Change request schemas
+GetChangeRequestSchema, ListChangeRequestsSchema, CreateChangeRequestSchema, ListChangeRequestPatchSetsSchema, 
+// Change request comment schemas
+CreateChangeRequestCommentSchema, ListChangeRequestCommentsSchema, UpdateChangeRequestCommentSchema, 
+// Commit schemas
+ListCommitsRequestSchema, GetCommitRequestSchema, CreateCommitCommentRequestSchema, DevopsCommitVOSchema, DevopsCommitStatVOSchema, CreateCommitCommentVOSchema } from "../operations/codeup/types.js";
+// Projex types
+export { 
+// Project schemas
+GetProjectSchema, SearchProjectsSchema, ListProjectMembersSchema, 
+// Sprint schemas
+GetSprintSchema, ListSprintsSchema, CreateSprintSchema, UpdateSprintSchema, 
+// Work item schemas
+GetWorkItemSchema, CreateWorkItemSchema, SearchWorkitemsSchema, UpdateWorkItemSchema, 
+// Work item type schemas
+ListAllWorkItemTypesSchema, ListWorkItemTypesSchema, GetWorkItemTypeSchema, ListWorkItemRelationWorkItemTypesSchema, GetWorkItemTypeFieldConfigSchema, GetWorkItemWorkflowSchema, 
+// Work item comment schemas
+ListWorkItemCommentsSchema, CreateWorkItemCommentSchema, 
+// Work item activity schemas
+ListWorkitemActivitiesSchema, 
+// Work item attachment schemas
+ListWorkitemAttachmentsSchema, 
+// Work item file schemas
+GetWorkitemFileSchema, 
+// Work item relation record schemas
+ListWorkitemRelationRecordsSchema, CreateWorkitemRelationRecordSchema, DeleteWorkitemRelationRecordSchema, 
+// Effort schemas
+ListCurrentUserEffortRecordsSchema, ListEffortRecordsSchema, CreateEffortRecordSchema, ListEstimatedEffortsSchema, CreateEstimatedEffortSchema, UpdateEffortRecordSchema, UpdateEstimatedEffortSchema } from "../operations/projex/types.js";
+// Flow types
+export { 
+// Pipeline schemas
+GetPipelineSchema, ListPipelinesSchema, CreatePipelineFromDescriptionSchema, CreatePipelineRunSchema, GetLatestPipelineRunSchema, GetPipelineRunSchema, ListPipelineRunsSchema, UpdatePipelineSchema, 
+// Pipeline job schemas
+ListPipelineJobsByCategorySchema, ListPipelineJobHistorysSchema, ExecutePipelineJobRunSchema, GetPipelineJobRunLogSchema, 
+// Service connection schemas
+ListServiceConnectionsSchema, 
+// Resource member schemas
+DeleteResourceMemberSchema, UpdateResourceMemberSchema, CreateResourceMemberSchema, UpdateResourceOwnerSchema, ResourceMemberSchema, ResourceMemberBaseSchema, 
+// Tag schemas
+CreateTagSchema, CreateTagGroupSchema, DeleteTagGroupSchema, UpdateTagGroupSchema, DeleteTagSchema, UpdateTagSchema, GetTagGroupSchema, TagGroupSchema, TagSchema, TagGroupWithTagsSchema, BaseTagSchema, 
+// VM Deploy Order schemas
+StopVMDeployOrderSchema, SkipVMDeployMachineSchema, RetryVMDeployMachineSchema, ResumeVMDeployOrderSchema, GetVMDeployOrderSchema, GetVMDeployMachineLogSchema, DeployOrderSchema, DeployOrderLogSchema } from "../operations/flow/types.js";
+// Packages types
+export { 
+// Package repository schemas
+ListPackageRepositoriesSchema, 
+// Artifact schemas
+ListArtifactsSchema, GetArtifactSchema } from "../operations/packages/types.js";

package/dist/common/utils.js

@@ -0,0 +1,381 @@
+import { getUserAgent } from "universal-user-agent";
+import { createYunxiaoError } from "./errors.js";
+import { VERSION } from "./version.js";
+const DEFAULT_YUNXIAO_API_BASE_URL = "https://openapi-rdc.aliyuncs.com";
+/**
+ * Get the Yunxiao API base URL from environment variables or use the default
+ * @returns The Yunxiao API base URL
+ */
+export function getYunxiaoApiBaseUrl() {
+    return process.env.YUNXIAO_API_BASE_URL || DEFAULT_YUNXIAO_API_BASE_URL;
+}
+/**
+ * 根据 YUNXIAO_API_BASE_URL 自动判断是否为 region 站：
+ * - 域名包含 openapi-rdc.aliyuncs.com -> 中心站
+ * - 其他域名 -> 认为是 region 站
+ */
+export function isRegionEdition() {
+    const baseUrl = getYunxiaoApiBaseUrl();
+    return !baseUrl.includes("openapi-rdc.aliyuncs.com");
+}
+/**
+ * region 站下使用的"默认 organizationId"（占位值）
+ * - 默认使用 "default"
+ * - 如有需要，可以通过环境变量覆盖
+ */
+export function getRegionDefaultOrganizationId() {
+    return process.env.YUNXIAO_REGION_DEFAULT_ORG_ID || "default";
+}
+export function debug(message, data) {
+    if (data !== undefined) {
+        console.error(`[DEBUG] ${message}`, typeof data === 'object' ? JSON.stringify(data, null, 2) : data);
+    }
+    else {
+        console.error(`[DEBUG] ${message}`);
+    }
+}
+async function parseResponseBody(response) {
+    const contentType = response.headers.get("content-type");
+    if (contentType?.includes("application/json")) {
+        return response.json();
+    }
+    return response.text();
+}
+export function buildUrl(baseUrl, params) {
+    // Handle baseUrl that doesn't have protocol
+    const isAbsolute = baseUrl.startsWith("http://") || baseUrl.startsWith("https://");
+    const fullBaseUrl = isAbsolute ? baseUrl : `${getYunxiaoApiBaseUrl()}${baseUrl.startsWith('/') ? baseUrl : `/${baseUrl}`}`;
+    try {
+        const url = new URL(fullBaseUrl);
+        Object.entries(params).forEach(([key, value]) => {
+            if (value !== undefined) {
+                url.searchParams.append(key, value.toString());
+            }
+        });
+        const result = url.toString();
+        console.error(`[DEBUG] Final URL: ${result}`);
+        // If we started with a relative URL, return just the path portion
+        if (!baseUrl.startsWith('http')) {
+            // Extract the path and query string from the full URL
+            const urlObj = new URL(result);
+            return urlObj.pathname + urlObj.search;
+        }
+        return result;
+    }
+    catch (error) {
+        console.error(`[ERROR] Failed to build URL: ${error}`);
+        // Fallback: manually append query parameters
+        let urlWithParams = baseUrl;
+        const queryParts = [];
+        Object.entries(params).forEach(([key, value]) => {
+            if (value !== undefined) {
+                queryParts.push(`${encodeURIComponent(key)}=${encodeURIComponent(value.toString())}`);
+            }
+        });
+        if (queryParts.length > 0) {
+            urlWithParams += (urlWithParams.includes('?') ? '&' : '?') + queryParts.join('&');
+        }
+        console.error(`[DEBUG] Fallback URL: ${urlWithParams}`);
+        return urlWithParams;
+    }
+}
+const USER_AGENT = `modelcontextprotocol/servers/alibabacloud-devops-mcp-server/v${VERSION} ${getUserAgent()}`;
+let currentSessionToken = undefined;
+/**
+ * Set the token for the current session (used in SSE mode)
+ * @param yunxiao_access_token The token to use for the current session
+ */
+export function setCurrentSessionToken(yunxiao_access_token) {
+    currentSessionToken = yunxiao_access_token;
+}
+/**
+ * Get the token for the current session
+ * @returns The token for the current session, or the default token from environment
+ */
+export function getCurrentSessionToken() {
+    return currentSessionToken || process.env.YUNXIAO_ACCESS_TOKEN;
+}
+export async function yunxiaoRequest(urlPath, options = {}) {
+    // Check if the URL is already a full URL or a path
+    const isAbsolute = urlPath.startsWith("http://") || urlPath.startsWith("https://");
+    let url = isAbsolute ? urlPath : `${getYunxiaoApiBaseUrl()}${urlPath.startsWith("/") ? urlPath : `/${urlPath}`}`;
+    const requestHeaders = {
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+        "User-Agent": USER_AGENT,
+        ...options.headers,
+    };
+    // Use session-specific token if available, otherwise use default token
+    const token = getCurrentSessionToken();
+    if (token) {
+        requestHeaders["x-yunxiao-token"] = token;
+    }
+    debug(`Request: ${options.method} ${url}`);
+    debug(`Headers:`, requestHeaders);
+    debug(`Body:`, options.body);
+    const response = await fetch(url, {
+        method: options.method || "GET",
+        headers: requestHeaders,
+        body: options.body ? JSON.stringify(options.body) : undefined,
+    });
+    const responseBody = await parseResponseBody(response);
+    debug(`Response Body:`, responseBody);
+    debug(`Response status:`, response.ok);
+    if (!response.ok) {
+        throw createYunxiaoError(response.status, responseBody, url, options.method || "GET", requestHeaders, options.body);
+    }
+    return responseBody;
+}
+export function pathEscape(filePath) {
+    // 先使用encodeURIComponent进行编码
+    let encoded = encodeURIComponent(filePath);
+    // 将编码后的%2F（/的编码）替换回/
+    encoded = encoded.replace(/%2F/gi, "/");
+    return encoded;
+}
+/**
+ * Handle repository ID encoding
+ * @param repositoryId Repository ID which may contain unencoded slash
+ * @returns Properly encoded repository ID
+ */
+export function handleRepositoryIdEncoding(repositoryId) {
+    let encodedRepoId = repositoryId;
+    // Automatically handle unencoded slashes in repositoryId
+    if (repositoryId.includes("/")) {
+        // Found unencoded slash, automatically URL encode it
+        const parts = repositoryId.split("/", 2);
+        if (parts.length === 2) {
+            const encodedRepoName = encodeURIComponent(parts[1]);
+            // Remove + signs from encoding (spaces are encoded as +, but we need %20)
+            const formattedEncodedName = encodedRepoName.replace(/\+/g, "%20");
+            encodedRepoId = `${parts[0]}%2F${formattedEncodedName}`;
+        }
+    }
+    return encodedRepoId;
+}
+/**
+ * Converts a floating point number to an integer string (removes decimal point and decimal part)
+ * Used primarily for handling numeric IDs that might come as floats from JSON parsing
+ * @param value Value to convert
+ * @returns Integer string representation
+ */
+export function floatToIntString(value) {
+    // 如果传入的是字符串，先尝试转为浮点数
+    if (typeof value === 'string') {
+        const floatValue = parseFloat(value);
+        if (!isNaN(floatValue)) {
+            value = floatValue;
+        }
+        else {
+            return value; // 如果转换失败，返回原字符串
+        }
+    }
+    // 处理浮点数
+    if (typeof value === 'number') {
+        const intValue = Math.floor(value + 0.5); // 四舍五入转整数
+        return intValue.toString();
+    }
+    // 处理其他情况，直接转字符串
+    return String(value);
+}
+/**
+ * 将各种时间格式转换为毫秒时间戳
+ * 支持：
+ * - 已有时间戳（number）直接返回
+ * - Date对象转换为时间戳
+ * - ISO格式日期字符串 (如: '2023-01-01T00:00:00Z')
+ * - 日期字符串 (如: '2023-01-01')
+ *
+ * @param time 时间输入
+ * @returns 毫秒时间戳
+ */
+export function convertToTimestamp(time) {
+    if (typeof time === 'number') {
+        // 如果已经是数字，假设已是时间戳
+        return time;
+    }
+    else if (time instanceof Date) {
+        // 如果是Date对象，转换为时间戳
+        return time.getTime();
+    }
+    else if (typeof time === 'string') {
+        // 尝试解析日期字符串
+        const date = new Date(time);
+        if (!isNaN(date.getTime())) {
+            return date.getTime();
+        }
+    }
+    // 无法转换时返回原值（如果是数字）或当前时间戳
+    return typeof time === 'number' ? time : Date.now();
+}
+/**
+ * Get start of today timestamp
+ * @returns Timestamp for start of the current day (00:00:00)
+ */
+export function getStartOfTodayTimestamp() {
+    const now = new Date();
+    // Reset time to start of day (00:00:00.000)
+    now.setHours(0, 0, 0, 0);
+    return now.getTime();
+}
+/**
+ * Get end of today timestamp
+ * @returns Timestamp for end of the current day (23:59:59.999)
+ */
+export function getEndOfTodayTimestamp() {
+    const now = new Date();
+    // Set time to end of day (23:59:59.999)
+    now.setHours(23, 59, 59, 999);
+    return now.getTime();
+}
+/**
+ * Get timestamp for start of a specific day
+ * @param date Date object or date string
+ * @returns Timestamp for start of the specified day
+ */
+export function getStartOfDayTimestamp(date) {
+    const targetDate = typeof date === 'string' ? new Date(date) : new Date(date);
+    targetDate.setHours(0, 0, 0, 0);
+    return targetDate.getTime();
+}
+/**
+ * Get timestamp for end of a specific day
+ * @param date Date object or date string
+ * @returns Timestamp for end of the specified day
+ */
+export function getEndOfDayTimestamp(date) {
+    const targetDate = typeof date === 'string' ? new Date(date) : new Date(date);
+    targetDate.setHours(23, 59, 59, 999);
+    return targetDate.getTime();
+}
+/**
+ * Get timestamp for start of current week
+ * @param startOnMonday Whether week should start on Monday (true) or Sunday (false)
+ * @returns Timestamp for start of the current week
+ */
+export function getStartOfWeekTimestamp(startOnMonday = true) {
+    const now = new Date();
+    const dayOfWeek = now.getDay(); // 0 is Sunday, 1 is Monday, etc.
+    const diff = startOnMonday ?
+        (dayOfWeek === 0 ? 6 : dayOfWeek - 1) : // If startOnMonday, set Sunday as day 7
+        dayOfWeek;
+    // Set to beginning of the week
+    now.setDate(now.getDate() - diff);
+    now.setHours(0, 0, 0, 0);
+    return now.getTime();
+}
+/**
+ * Get timestamp for end of current week
+ * @param startOnMonday Whether week should start on Monday (true) or Sunday (false)
+ * @returns Timestamp for end of the current week
+ */
+export function getEndOfWeekTimestamp(startOnMonday = true) {
+    const now = new Date();
+    const dayOfWeek = now.getDay(); // 0 is Sunday, 1 is Monday, etc.
+    const diff = startOnMonday ?
+        (dayOfWeek === 0 ? 0 : 7 - dayOfWeek) : // If startOnMonday, set Sunday as day 7
+        (6 - dayOfWeek);
+    // Set to end of the week
+    now.setDate(now.getDate() + diff);
+    now.setHours(23, 59, 59, 999);
+    return now.getTime();
+}
+/**
+ * Get timestamp for start of current month
+ * @returns Timestamp for start of the current month
+ */
+export function getStartOfMonthTimestamp() {
+    const now = new Date();
+    now.setDate(1); // First day of current month
+    now.setHours(0, 0, 0, 0);
+    return now.getTime();
+}
+/**
+ * Get timestamp for end of current month
+ * @returns Timestamp for end of the current month
+ */
+export function getEndOfMonthTimestamp() {
+    const now = new Date();
+    now.setMonth(now.getMonth() + 1); // Move to next month
+    now.setDate(0); // Last day of previous month (i.e., current month)
+    now.setHours(23, 59, 59, 999);
+    return now.getTime();
+}
+/**
+ * Analyzes natural language date reference and returns corresponding timestamp range
+ * @param dateReference Natural language date reference (e.g., "today", "this week", "last month")
+ * @returns Object containing start and end timestamps
+ */
+export function parseDateReference(dateReference) {
+    if (!dateReference) {
+        // Default to all time
+        return {
+            startTime: 0,
+            endTime: Date.now()
+        };
+    }
+    const normalizedRef = dateReference.trim().toLowerCase();
+    // Today/yesterday
+    if (normalizedRef === 'today' || normalizedRef === '今天') {
+        return {
+            startTime: getStartOfTodayTimestamp(),
+            endTime: getEndOfTodayTimestamp()
+        };
+    }
+    if (normalizedRef === 'yesterday' || normalizedRef === '昨天') {
+        const yesterday = new Date();
+        yesterday.setDate(yesterday.getDate() - 1);
+        return {
+            startTime: getStartOfDayTimestamp(yesterday),
+            endTime: getEndOfDayTimestamp(yesterday)
+        };
+    }
+    // This week/last week
+    if (normalizedRef === 'this week' || normalizedRef === '本周' ||
+        normalizedRef === 'current week' || normalizedRef === '这周' ||
+        normalizedRef === '这个星期') {
+        return {
+            startTime: getStartOfWeekTimestamp(),
+            endTime: getEndOfWeekTimestamp()
+        };
+    }
+    if (normalizedRef === 'last week' || normalizedRef === '上周' ||
+        normalizedRef === '上個星期' || normalizedRef === '上个星期') {
+        const lastWeekStart = new Date(getStartOfWeekTimestamp());
+        lastWeekStart.setDate(lastWeekStart.getDate() - 7);
+        const lastWeekEnd = new Date(getEndOfWeekTimestamp());
+        lastWeekEnd.setDate(lastWeekEnd.getDate() - 7);
+        return {
+            startTime: lastWeekStart.getTime(),
+            endTime: lastWeekEnd.getTime()
+        };
+    }
+    // This month/last month
+    if (normalizedRef === 'this month' || normalizedRef === '本月' ||
+        normalizedRef === 'current month' || normalizedRef === '这个月') {
+        return {
+            startTime: getStartOfMonthTimestamp(),
+            endTime: getEndOfMonthTimestamp()
+        };
+    }
+    if (normalizedRef === 'last month' || normalizedRef === '上月' ||
+        normalizedRef === '上个月') {
+        const now = new Date();
+        const lastMonth = new Date(now.getFullYear(), now.getMonth() - 1);
+        // Start of last month
+        const startOfLastMonth = new Date(lastMonth.getFullYear(), lastMonth.getMonth(), 1);
+        startOfLastMonth.setHours(0, 0, 0, 0);
+        // End of last month
+        const endOfLastMonth = new Date(now.getFullYear(), now.getMonth(), 0);
+        endOfLastMonth.setHours(23, 59, 59, 999);
+        return {
+            startTime: startOfLastMonth.getTime(),
+            endTime: endOfLastMonth.getTime()
+        };
+    }
+    // Default to all time
+    return {
+        startTime: 0,
+        endTime: Date.now()
+    };
+}

package/dist/common/version.js

@@ -0,0 +1 @@
+export const VERSION = "0.1.17";

package/dist/index.js

@@ -0,0 +1,225 @@
+#!/usr/bin/env node
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
+import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { z } from 'zod';
+import { isYunxiaoError, YunxiaoValidationError } from "./common/errors.js";
+import { VERSION } from "./common/version.js";
+import { config } from "dotenv";
+import { getAllTools, getEnabledTools } from "./tool-registry/index.js";
+import { handleToolRequest, handleEnabledToolRequest } from "./tool-handlers/index.js";
+import { Toolset } from "./common/toolsets.js";
+const server = new Server({
+    name: "alibabacloud-devops-mcp-server",
+    version: VERSION,
+}, {
+    capabilities: {
+        tools: {},
+    },
+});
+function formatYunxiaoError(error) {
+    let message = `Yunxiao API Error: ${error.message}`;
+    // 添加请求上下文信息
+    if (error.method || error.url) {
+        message += `\n Request: ${error.method || 'GET'} ${error.url || 'unknown'}`;
+    }
+    if (error.requestHeaders) {
+        message += `\n Request Headers: ${JSON.stringify(error.requestHeaders, null, 2)}`;
+    }
+    if (error.requestBody) {
+        message += `\n Request Body: ${JSON.stringify(error.requestBody, null, 2)}`;
+    }
+    if (error instanceof YunxiaoValidationError) {
+        message = `Parameter validation failed: ${error.message}`;
+        if (error.response) {
+            message += `\n Response: ${JSON.stringify(error.response, null, 2)}`;
+        }
+        // 添加常见参数错误的提示
+        if (error.message.includes('name')) {
+            message += `\n Suggestion: Please check whether the pipeline name meets the requirements.`;
+        }
+        if (error.message.includes('content') || error.message.includes('yaml')) {
+            message += `\n Suggestion: Please check whether the generated YAML format is correct.`;
+        }
+    }
+    else {
+        // 处理通用的Yunxiao错误
+        message = `Yunxiao API error (${error.status}): ${error.message}`;
+        if (error.response) {
+            const response = error.response;
+            if (response.errorCode) {
+                message += `\n errorCode: ${response.errorCode}`;
+            }
+            if (response.errorMessage && response.errorMessage !== error.message) {
+                message += `\n errorMessage: ${response.errorMessage}`;
+            }
+            if (response.data && typeof response.data === 'object') {
+                message += `\n data: ${JSON.stringify(response.data, null, 2)}`;
+            }
+            // 如果响应体中有更多详细信息，也一并显示
+            if (Object.keys(response).length > 0) {
+                message += `\n Full Response: ${JSON.stringify(response, null, 2)}`;
+            }
+        }
+        // 根据状态码提供通用建议
+        switch (error.status) {
+            case 400:
+                message += `\n Suggestion: Please check whether the request parameters are correct, especially whether all required parameters have been provided.`;
+                break;
+            case 500:
+                message += `\n Suggestion: Internal server error. Please try again later or contact technical support.`;
+                break;
+            case 502:
+            case 503:
+            case 504:
+                message += `\n Suggestion: The service is temporarily unavailable. Please try again later.`;
+                break;
+        }
+    }
+    return message;
+}
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+    let tools;
+    if (enabledToolsets.length > 0) {
+        // 获取基础工具（总是加载）
+        const baseTools = getEnabledTools([Toolset.BASE]);
+        // 获取启用的工具集工具
+        const enabledTools = getEnabledTools(enabledToolsets);
+        // 合并基础工具和启用的工具集工具
+        tools = [...baseTools, ...enabledTools];
+    }
+    else {
+        // 如果没有指定启用的工具集，则获取所有工具（已包含基础工具）
+        tools = getAllTools();
+    }
+    return {
+        tools,
+    };
+});
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    try {
+        if (!request.params.arguments) {
+            throw new Error("Arguments are required");
+        }
+        // Delegate to our modular tool handler with toolset support
+        const result = enabledToolsets.length > 0
+            ? await handleEnabledToolRequest(request, enabledToolsets)
+            : await handleToolRequest(request);
+        return result;
+    }
+    catch (error) {
+        if (error instanceof z.ZodError) {
+            throw new Error(`Invalid input: ${JSON.stringify(error.errors)}`);
+        }
+        if (isYunxiaoError(error)) {
+            throw new Error(formatYunxiaoError(error));
+        }
+        throw error;
+    }
+});
+config();
+// 解析启用的工具集
+const parseEnabledToolsets = (input) => {
+    if (!input)
+        return [];
+    return input.split(',').map(toolset => {
+        const trimmed = toolset.trim();
+        // 验证工具集名称是否有效
+        if (!Object.values(Toolset).includes(trimmed)) {
+            throw new Error(`Unknown toolset: ${trimmed}`);
+        }
+        return trimmed;
+    });
+};
+// 获取启用的工具集（从命令行参数或环境变量）
+const enabledToolsets = parseEnabledToolsets(process.argv.find(arg => arg.startsWith('--toolsets='))?.split('=')[1] ||
+    process.env.DEVOPS_TOOLSETS);
+// Check if we should run in SSE mode
+const useSSE = process.argv.includes('--sse') || process.env.MCP_TRANSPORT === 'sse';
+async function runServer() {
+    if (useSSE) {
+        // Import express only when needed for SSE mode
+        const { default: express } = await import('express');
+        const app = express();
+        const port = process.env.PORT || 3000;
+        // Store sessions with their tokens
+        const sessions = {};
+        // SSE endpoint - handles initial connection
+        app.get('/sse', async (req, res) => {
+            // In SSE mode, we can use console.log for debugging since it doesn't interfere with the protocol
+            console.log(`New SSE connection from ${req.ip}`);
+            // Get token from query parameters or headers
+            const yunxiao_access_token = req.query.yunxiao_access_token || req.headers['x-yunxiao-token'] || process.env.YUNXIAO_ACCESS_TOKEN;
+            // Create transport with endpoint for POST messages
+            const sseTransport = new SSEServerTransport('/messages', res);
+            const sessionId = sseTransport.sessionId;
+            if (sessionId) {
+                sessions[sessionId] = { transport: sseTransport, server, yunxiao_access_token };
+            }
+            try {
+                await server.connect(sseTransport);
+                // In SSE mode, console.error is acceptable for status messages
+                console.info(`Yunxiao MCP Server connected via SSE with session ${sessionId}`);
+                if (yunxiao_access_token) {
+                    console.error(`Session ${sessionId} using custom token`);
+                }
+                else {
+                    console.error(`Session ${sessionId} using default token from environment`);
+                }
+            }
+            catch (error) {
+                console.error("Failed to start SSE server:", error);
+                res.status(500).send("Server error");
+            }
+        });
+        // POST endpoint - handles incoming messages
+        app.use(express.json({ limit: '10mb' })); // Add JSON body parser
+        app.post('/messages', async (req, res) => {
+            const sessionId = req.query.sessionId;
+            const session = sessions[sessionId];
+            if (!session) {
+                res.status(404).send("Session not found");
+                return;
+            }
+            try {
+                // Set the session token before handling the message
+                const utils = await import('./common/utils.js');
+                utils.setCurrentSessionToken(session.yunxiao_access_token);
+                await session.transport.handlePostMessage(req, res, req.body);
+            }
+            catch (error) {
+                console.error("Error handling POST message:", error);
+                res.status(500).send("Server error");
+            }
+        });
+        // Start server
+        const serverInstance = app.listen(port, () => {
+            console.log(`Yunxiao MCP Server running in SSE mode on port ${port}`);
+            console.log(`Connect via SSE at http://localhost:${port}/sse`);
+            console.log(`Send messages to http://localhost:${port}/messages?sessionId=<session-id>`);
+        });
+        // Handle graceful shutdown
+        process.on('SIGINT', () => {
+            console.log('Shutting down SSE server...');
+            serverInstance.close(() => {
+                console.log('Server closed.');
+                process.exit(0);
+            });
+        });
+    }
+    else {
+        // Stdio mode (default)
+        // In stdio mode, we must avoid console.log/console.error as they interfere with the JSON-RPC protocol
+        const transport = new StdioServerTransport();
+        await server.connect(transport);
+        // Don't output anything to stdout/stderr in stdio mode - only JSON-RPC messages should go through the transport
+    }
+}
+runServer().catch((error) => {
+    // Only output error to stderr in SSE mode, not in stdio mode
+    if (useSSE) {
+        console.error("Fatal error in main():", error);
+    }
+    process.exit(1);
+});