@qwe8652591/abap-recursive-query 1.1.0

package/dist/common/utils.js

@@ -0,0 +1,300 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ErrorCode = exports.McpError = void 0;
+exports.getConfig = getConfig;
+exports.return_response = return_response;
+exports.return_error = return_error;
+exports.createAxiosInstance = createAxiosInstance;
+exports.getBaseUrl = getBaseUrl;
+exports.getAuthHeaders = getAuthHeaders;
+exports.escapeXml = escapeXml;
+exports.sanitizeObjectName = sanitizeObjectName;
+exports.makeAdtRequest = makeAdtRequest;
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+Object.defineProperty(exports, "McpError", { enumerable: true, get: function () { return types_js_1.McpError; } });
+Object.defineProperty(exports, "ErrorCode", { enumerable: true, get: function () { return types_js_1.ErrorCode; } });
+const axios_1 = __importStar(require("axios"));
+const https_1 = require("https");
+/**
+ * 从环境变量获取 SAP 配置
+ */
+function getConfig() {
+    const url = process.env.SAP_URL;
+    const username = process.env.SAP_USERNAME;
+    const password = process.env.SAP_PASSWORD;
+    const client = process.env.SAP_CLIENT;
+    if (!url || !username || !password || !client) {
+        throw new Error(`缺少必需的环境变量。需要的变量:
+- SAP_URL
+- SAP_USERNAME
+- SAP_PASSWORD
+- SAP_CLIENT`);
+    }
+    return { url, username, password, client };
+}
+function return_response(response) {
+    return {
+        isError: false,
+        content: [{
+                type: 'text',
+                text: response.data
+            }]
+    };
+}
+function return_error(error) {
+    return {
+        isError: true,
+        content: [{
+                type: 'text',
+                text: `Error: ${error instanceof axios_1.AxiosError ? String(error.response?.data)
+                    : error instanceof Error ? error.message
+                        : String(error)}`
+            }]
+    };
+}
+let axiosInstance = null;
+function createAxiosInstance() {
+    if (!axiosInstance) {
+        axiosInstance = axios_1.default.create({
+            httpsAgent: new https_1.Agent({
+                rejectUnauthorized: false // Allow self-signed certificates
+            })
+        });
+    }
+    return axiosInstance;
+}
+let config;
+let csrfToken = null;
+let cookies = null; // Variable to store cookies
+async function getBaseUrl() {
+    if (!config) {
+        config = getConfig();
+    }
+    const { url } = config;
+    try {
+        const urlObj = new URL(url);
+        const baseUrl = Buffer.from(`${urlObj.origin}`);
+        return baseUrl;
+    }
+    catch (error) {
+        const errorMessage = `Invalid URL in configuration: ${error instanceof Error ? error.message : error}`;
+        throw new Error(errorMessage);
+    }
+}
+async function getAuthHeaders() {
+    if (!config) {
+        config = getConfig();
+    }
+    const { username, password, client } = config;
+    const auth = Buffer.from(`${username}:${password}`).toString('base64'); // Create Basic Auth string
+    return {
+        'Authorization': `Basic ${auth}`, // Basic Authentication header
+        'SAP-Client': client // SAP client header
+    };
+}
+async function fetchCsrfToken(url) {
+    try {
+        const response = await createAxiosInstance()({
+            method: 'GET',
+            url,
+            headers: {
+                ...(await getAuthHeaders()),
+                'x-csrf-token': 'fetch'
+            }
+        });
+        const token = response.headers['x-csrf-token'];
+        if (!token) {
+            throw new Error('No CSRF token in response headers');
+        }
+        // Extract and store cookies
+        if (response.headers['set-cookie']) {
+            cookies = response.headers['set-cookie'].join('; ');
+        }
+        return token;
+    }
+    catch (error) {
+        // Even if the request fails, try to get token from error response
+        if (error instanceof axios_1.AxiosError && error.response?.headers['x-csrf-token']) {
+            const token = error.response.headers['x-csrf-token'];
+            if (token) {
+                // Extract and store cookies from the error response as well
+                if (error.response.headers['set-cookie']) {
+                    cookies = error.response.headers['set-cookie'].join('; ');
+                }
+                return token;
+            }
+        }
+        // If we couldn't get token from error response either, throw the original error
+        throw new Error(`Failed to fetch CSRF token: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}
+/**
+ * XML转义函数 - 将特殊字符转换为XML实体
+ */
+function escapeXml(text) {
+    if (text == null) {
+        return '';
+    }
+    return String(text)
+        .replace(/&/g, '&amp;')
+        .replace(/</g, '&lt;')
+        .replace(/>/g, '&gt;')
+        .replace(/"/g, '&quot;')
+        .replace(/'/g, '&apos;');
+}
+/**
+ * 对象名称转义函数 - 将特殊字符转换为文件系统安全的字符
+ *
+ * 转义规则：
+ * - 斜杠 (/, \) -> 下划线 (_)
+ * - 尖括号 (<, >) -> 井号 (#)
+ * - 其他非字母数字字符 -> 下划线 (_)
+ *
+ * @param name - 原始对象名称
+ * @returns 转义后的安全名称
+ *
+ * @example
+ * sanitizeObjectName('SAPLMEGUI') // 'SAPLMEGUI'
+ * sanitizeObjectName('OBJECT/NAME') // 'OBJECT_NAME'
+ * sanitizeObjectName('OBJECT<TYPE>') // 'OBJECT#TYPE#'
+ */
+function sanitizeObjectName(name) {
+    if (!name) {
+        return '';
+    }
+    return String(name)
+        .replace(/[/\\]/g, '_')
+        .replace(/[<>]/g, '#')
+        .replace(/[^a-zA-Z0-9_#]/g, '_');
+}
+async function makeAdtRequest(url, method, timeout, data, params) {
+    // For POST/PUT requests, ensure we have a CSRF token
+    if ((method === 'POST' || method === 'PUT') && !csrfToken) {
+        try {
+            csrfToken = await fetchCsrfToken(url);
+        }
+        catch (error) {
+            throw new Error('CSRF token is required for POST/PUT requests but could not be fetched');
+        }
+    }
+    const requestHeaders = {
+        ...(await getAuthHeaders())
+    };
+    // Add CSRF token for POST/PUT requests
+    if ((method === 'POST' || method === 'PUT') && csrfToken) {
+        requestHeaders['x-csrf-token'] = csrfToken;
+    }
+    // Add cookies if available
+    if (cookies) {
+        requestHeaders['Cookie'] = cookies;
+    }
+    const config = {
+        method,
+        url,
+        headers: requestHeaders,
+        timeout,
+        params: params
+    };
+    // Include data in the request configuration if provided
+    if (data) {
+        config.data = data;
+    }
+    const requestStartTime = Date.now();
+    const urlPath = new URL(url).pathname + (config.params ? '?' + new URLSearchParams(config.params).toString() : '');
+    const paramsInfo = config.params ? ` | 参数: ${JSON.stringify(config.params)}` : '';
+    try {
+        console.error(`[HTTP请求] 开始请求: ${method} ${url} (超时: ${timeout}ms)${paramsInfo}`);
+        // 添加请求进度监控
+        const progressInterval = setInterval(() => {
+            const elapsed = Date.now() - requestStartTime;
+            const elapsedSeconds = Math.floor(elapsed / 1000);
+            if (elapsedSeconds > 10) { // 超过10秒才输出警告
+                console.warn(`[HTTP请求-等待中] ${method} ${url} - 已等待 ${elapsedSeconds}秒，可能网络较慢或服务器无响应${paramsInfo}`);
+            }
+        }, 10000); // 每10秒检查一次
+        let response;
+        try {
+            response = await createAxiosInstance()(config);
+        }
+        finally {
+            clearInterval(progressInterval);
+        }
+        const requestDuration = Date.now() - requestStartTime;
+        console.error(`[HTTP请求] 请求完成: ${method} ${url} (耗时: ${requestDuration}ms)${paramsInfo}`);
+        return response;
+    }
+    catch (error) {
+        const requestDuration = Date.now() - requestStartTime;
+        // If we get a 403 with "CSRF token validation failed", try to fetch a new token and retry
+        if (error instanceof axios_1.AxiosError && error.response?.status === 403 &&
+            error.response.data?.includes('CSRF')) {
+            console.error(`[HTTP请求] CSRF token 失效，重新获取token并重试: ${url}${paramsInfo}`);
+            csrfToken = await fetchCsrfToken(url);
+            config.headers['x-csrf-token'] = csrfToken;
+            console.error(`[HTTP请求] 重试请求: ${method} ${url}${paramsInfo}`);
+            const retryResponse = await createAxiosInstance()(config);
+            const retryDuration = Date.now() - requestStartTime;
+            console.error(`[HTTP请求] 重试完成: ${method} ${url} (总耗时: ${retryDuration}ms)${paramsInfo}`);
+            return retryResponse;
+        }
+        // 构造详细的错误信息并抛出
+        let errorMessage;
+        if (error instanceof axios_1.AxiosError) {
+            if (error.code === 'ECONNABORTED') {
+                errorMessage = `请求超时 - ${method} ${url} (耗时: ${requestDuration}ms, 超时设置: ${timeout}ms)${paramsInfo}`;
+                console.error(`[HTTP请求-超时] ${errorMessage}`);
+            }
+            else if (error.response) {
+                errorMessage = `服务器返回错误 - ${method} ${url} (状态: ${error.response.status}, 耗时: ${requestDuration}ms)${paramsInfo}`;
+                console.error(`[HTTP请求-错误] ${errorMessage}`);
+            }
+            else if (error.request) {
+                errorMessage = `无响应 - ${method} ${url} (耗时: ${requestDuration}ms)${paramsInfo}`;
+                console.error(`[HTTP请求-错误] ${errorMessage}`);
+            }
+            else {
+                errorMessage = `${error.message} - ${method} ${url} (耗时: ${requestDuration}ms)${paramsInfo}`;
+                console.error(`[HTTP请求-错误] ${errorMessage}`);
+            }
+        }
+        else {
+            errorMessage = `${error instanceof Error ? error.message : String(error)} - ${method} ${url} (耗时: ${requestDuration}ms)${paramsInfo}`;
+            console.error(`[HTTP请求-错误] ${errorMessage}`);
+        }
+        // 抛出包含详细信息的新错误
+        throw new Error(errorMessage);
+    }
+}

package/dist/consumer/FileDownloadConsumer.d.ts

@@ -0,0 +1,127 @@
+/**
+ * 文件下载消费者
+ * 负责从队列拉取任务并下载文件
+ */
+import { RedisQueueManager } from '../common';
+/**
+ * 下载结果
+ */
+interface DownloadResult {
+    objectName: string;
+    objectType: string;
+    filePath: string;
+    success: boolean;
+    error?: string;
+}
+/**
+ * 文件下载消费者
+ */
+export declare class FileDownloadConsumer {
+    private workerId;
+    private queueManager;
+    private isRunning;
+    private activeJobIds;
+    private downloadedFiles;
+    private downloadRootPath;
+    private organizeByProgram;
+    private abaplintRegistry;
+    constructor(queueManager: RedisQueueManager, downloadRootPath: string, jobIds?: string[], organizeByProgram?: boolean);
+    /**
+     * 启动消费者
+     */
+    start(jobIds?: string[]): Promise<void>;
+    /**
+     * 停止消费者
+     */
+    stop(): Promise<void>;
+    /**
+     * 获取下载的文件列表
+     */
+    getDownloadedFiles(): DownloadResult[];
+    /**
+     * 处理任务
+     */
+    private processTask;
+    /**
+     * 生成进度条
+     */
+    private generateProgressBar;
+    /**
+     * 下载对象源代码
+     */
+    private downloadObjectSource;
+    /**
+     * 处理附加数据
+     */
+    private processAdditionalData;
+    /**
+     * 处理对象元数据
+     */
+    private processObjectMetadata;
+    /**
+     * 处理函数的附加数据
+     */
+    private processFunctionData;
+    /**
+     * 处理程序的附加数据
+     */
+    private processProgramData;
+    /**
+     * 处理类的附加数据
+     */
+    private processClassData;
+    /**
+     * 下载 Include 文件并根据内容自动检测类型，添加符合 abaplint 规范的后缀
+     */
+    private downloadInclude;
+    /**
+     * ✅ 分析 include 文件内容，返回类型和真实名称
+     * 使用 abaplint 进行精确的语法分析
+     * - 只对 FUNCTION 类型提取真实函数名
+     * - 其他类型（类、表单等）保持使用原始 include 名称
+     * - 遵循零硬编码原则：所有信息从源代码中动态提取
+     */
+    private analyzeIncludeContent;
+    /**
+     * 根据 include 类型生成符合 abaplint 规范的文件路径
+     */
+    private generateIncludeFilePath;
+    /**
+     * 提取函数组名称
+     */
+    private extractFunctionGroup;
+    /**
+     * 生成文件路径
+     */
+    private generateFilePath;
+    /**
+     * 获取默认文件扩展名
+     */
+    private getDefaultExtension;
+    /**
+     * 获取对象目录
+     */
+    private getObjectDirectory;
+    /**
+     * 获取活跃的 Job 列表
+     */
+    private getActiveJobs;
+    /**
+     * 延迟函数
+     */
+    private sleep;
+    /**
+     * 处理结构体列表
+     */
+    private processStructures;
+    /**
+     * 处理单个结构体
+     */
+    private processSingleStructure;
+    /**
+     * 处理屏幕数据
+     */
+    private processScreens;
+}
+export {};
+//# sourceMappingURL=FileDownloadConsumer.d.ts.map

package/dist/consumer/FileDownloadConsumer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"FileDownloadConsumer.d.ts","sourceRoot":"","sources":["../../src/consumer/FileDownloadConsumer.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,iBAAiB,EAAyC,MAAM,WAAW,CAAC;AAyBrF;;GAEG;AACH,UAAU,cAAc;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,qBAAa,oBAAoB;IAC7B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,YAAY,CAAoB;IACxC,OAAO,CAAC,SAAS,CAAkB;IACnC,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,eAAe,CAAwB;IAC/C,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,iBAAiB,CAAU;IACnC,OAAO,CAAC,gBAAgB,CAAW;gBAG/B,YAAY,EAAE,iBAAiB,EAC/B,gBAAgB,EAAE,MAAM,EACxB,MAAM,CAAC,EAAE,MAAM,EAAE,EACjB,iBAAiB,GAAE,OAAc;IAiBrC;;OAEG;IACG,KAAK,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqD7C;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAK3B;;OAEG;IACH,kBAAkB,IAAI,cAAc,EAAE;IAItC;;OAEG;YACW,WAAW;IA+EzB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAM3B;;OAEG;YACW,oBAAoB;IAuGlC;;OAEG;YACW,qBAAqB;IAuBnC;;OAEG;YACW,qBAAqB;IA4BnC;;OAEG;YACW,mBAAmB;IA6IjC;;OAEG;YACW,kBAAkB;IAuIhC;;OAEG;YACW,gBAAgB;IA6D9B;;OAEG;YACW,eAAe;IA8D7B;;;;;;OAMG;IACH,OAAO,CAAC,qBAAqB;IAkH7B;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAmC/B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAQ5B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAKxB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAU3B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAU1B;;OAEG;YACW,aAAa;IAI3B;;OAEG;IACH,OAAO,CAAC,KAAK;IAIb;;OAEG;YACW,iBAAiB;IAc/B;;OAEG;YACW,sBAAsB;IA6DpC;;OAEG;YACW,cAAc;CA2G/B"}