needware-cli 1.2.8 → 1.3.0

package/dist/utils/prompt-parser.js

@@ -0,0 +1,228 @@
+/**
+ * Prompt 解析工具
+ */
+import * as fs from 'fs';
+import * as path from 'path';
+/**
+ * 解析 Prompt 输入（自动判断类型）
+ *
+ * 支持两种格式：
+ * 1. 字符串 - 普通文本或文件路径（自动识别）
+ * 2. SDKUserMessage JSON - 符合 SDK 标准的消息对象
+ */
+export async function parsePrompt(input) {
+    // 如果是异步迭代器（交互式模式），直接返回
+    if (isAsyncIterable(input)) {
+        return {
+            prompt: input,
+            isInteractive: true,
+        };
+    }
+    // 如果是字符串，进一步判断
+    if (typeof input === 'string') {
+        // 检查是否是文件路径
+        if (input.endsWith('.txt') || input.endsWith('.json') || input.endsWith('.md')) {
+            if (fs.existsSync(path.resolve(input))) {
+                return await parsePromptFromFile(input);
+            }
+        }
+        // 尝试解析为 SDKUserMessage JSON
+        if (input.trim().startsWith('{')) {
+            try {
+                let data;
+                try {
+                    // 首先尝试直接解析
+                    data = JSON.parse(input);
+                }
+                catch (firstError) {
+                    console.log('First parse failed, trying to unescape...');
+                    // 如果失败，尝试反转义字符串（处理命令行转义问题）
+                    // 将 \" 替换为 "，\\ 替换为 \
+                    const unescaped = input.replace(/\\"/g, '"').replace(/\\\\/g, '\\');
+                    console.log('Unescaped input:', unescaped);
+                    data = JSON.parse(unescaped);
+                }
+                // 检查是否双重编码（data 是字符串而非对象）
+                if (typeof data === 'string') {
+                    console.log('Detected double-encoded JSON, parsing again...');
+                    const actualData = JSON.parse(data);
+                    if (isSDKUserMessage(actualData)) {
+                        return parseSDKUserMessage(actualData);
+                    }
+                    if (isSimplifiedUserMessage(actualData)) {
+                        return parseSimplifiedUserMessage(actualData);
+                    }
+                }
+                if (isSDKUserMessage(data)) {
+                    return parseSDKUserMessage(data);
+                }
+                // 检查是否为简化的用户消息格式
+                if (isSimplifiedUserMessage(data)) {
+                    return parseSimplifiedUserMessage(data);
+                }
+            }
+            catch (error) {
+                console.error('JSON parse error:', error);
+                // 不是有效的 JSON，继续作为普通字符串处理
+            }
+        }
+        // 作为普通字符串处理
+        return {
+            prompt: input.trim(),
+            isInteractive: false,
+        };
+    }
+    // 如果是 SDKUserMessage 对象
+    if (isSDKUserMessage(input)) {
+        return parseSDKUserMessage(input);
+    }
+    throw new Error(`Unsupported prompt input type: ${typeof input}`);
+}
+/**
+ * 从文件读取 Prompt
+ */
+async function parsePromptFromFile(filePath) {
+    const resolvedPath = path.resolve(filePath);
+    if (!fs.existsSync(resolvedPath)) {
+        throw new Error(`Prompt file not found: ${resolvedPath}`);
+    }
+    const content = fs.readFileSync(resolvedPath, 'utf-8').trim();
+    const ext = path.extname(filePath).toLowerCase();
+    // 如果是 JSON 文件，尝试解析为 SDKUserMessage
+    if (ext === '.json') {
+        try {
+            const data = JSON.parse(content);
+            if (isSDKUserMessage(data)) {
+                return parseSDKUserMessage(data);
+            }
+            throw new Error('JSON file must contain a valid SDKUserMessage object');
+        }
+        catch (error) {
+            throw new Error(`Invalid JSON in prompt file: ${error instanceof Error ? error.message : 'Unknown error'}`);
+        }
+    }
+    // 否则作为纯文本处理
+    return {
+        prompt: content,
+        isInteractive: false,
+    };
+}
+/**
+ * 解析 SDKUserMessage
+ */
+function parseSDKUserMessage(data) {
+    // 如果 content 包含非文本内容（如图片），使用交互式模式
+    const message = data.message;
+    const hasMultimodalContent = Array.isArray(message.content) &&
+        message.content.some((item) => item.type !== 'text');
+    if (hasMultimodalContent) {
+        // 创建消息流来支持多模态内容
+        const messageStream = createMessageStreamFromSDKMessage(data);
+        return {
+            prompt: messageStream,
+            sessionId: data.session_id,
+            isInteractive: true,
+        };
+    }
+    // 提取文本内容
+    let promptText = '';
+    if (typeof message.content === 'string') {
+        promptText = message.content;
+    }
+    else if (Array.isArray(message.content)) {
+        // 如果 content 是数组，提取文本部分
+        promptText = message.content
+            .filter((item) => item.type === 'text')
+            .map((item) => item.text || '')
+            .join('\n');
+    }
+    return {
+        prompt: promptText.trim(),
+        sessionId: data.session_id,
+        isInteractive: false,
+    };
+}
+/**
+ * 解析简化的用户消息格式
+ */
+function parseSimplifiedUserMessage(data) {
+    // 将简化格式转换为标准的 SDKUserMessage
+    const sessionId = data.session_id || `session-${Date.now()}`;
+    const sdkMessage = {
+        type: 'user',
+        session_id: sessionId,
+        message: {
+            role: 'user',
+            content: data.user.content,
+        },
+        parent_tool_use_id: null,
+    };
+    // 使用 parseSDKUserMessage 处理转换后的消息
+    return parseSDKUserMessage(sdkMessage);
+}
+/**
+ * 从 SDKUserMessage 创建消息流
+ */
+async function* createMessageStreamFromSDKMessage(message) {
+    yield message;
+}
+/**
+ * 将字符串转换为消息流（用于交互式模式）
+ */
+export async function* createMessageStream(sessionId, initialPrompt) {
+    if (initialPrompt) {
+        yield {
+            session_id: sessionId,
+            type: 'user',
+            message: { role: 'user', content: initialPrompt.trim() },
+            parent_tool_use_id: null,
+        };
+    }
+}
+/**
+ * 类型守卫：检查是否为 AsyncIterable
+ */
+function isAsyncIterable(obj) {
+    return (obj !== null &&
+        obj !== undefined &&
+        typeof obj[Symbol.asyncIterator] === 'function');
+}
+/**
+ * 类型守卫：检查是否为 SDKUserMessage
+ */
+function isSDKUserMessage(obj) {
+    return (obj !== null &&
+        obj !== undefined &&
+        typeof obj === 'object' &&
+        'type' in obj &&
+        obj.type === 'user' &&
+        'message' in obj &&
+        typeof obj.message === 'object' &&
+        'session_id' in obj &&
+        typeof obj.session_id === 'string');
+}
+/**
+ * 类型守卫：检查是否为简化的用户消息格式（只有 user 字段）
+ */
+function isSimplifiedUserMessage(obj) {
+    return (obj !== null &&
+        obj !== undefined &&
+        typeof obj === 'object' &&
+        'user' in obj &&
+        obj.user !== null &&
+        obj.user !== undefined &&
+        typeof obj.user === 'object' &&
+        'content' in obj.user);
+}
+/**
+ * 从命令行参数解析 Prompt（自动判断类型）
+ */
+export async function parsePromptFromArgs(args) {
+    if (args.length === 0) {
+        throw new Error('No prompt provided');
+    }
+    const firstArg = args[0];
+    // 使用统一的 parsePrompt 函数，它会自动判断类型
+    return parsePrompt(firstArg);
+}
+//# sourceMappingURL=prompt-parser.js.map

package/dist/utils/prompt-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-parser.js","sourceRoot":"","sources":["../../src/utils/prompt-parser.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAE7B;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAkB;IAClD,uBAAuB;IACvB,IAAI,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO;YACL,MAAM,EAAE,KAAK;YACb,aAAa,EAAE,IAAI;SACpB,CAAC;IACJ,CAAC;IAED,eAAe;IACf,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,YAAY;QACZ,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/E,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACvC,OAAO,MAAM,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QACD,4BAA4B;QAC5B,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,IAAI,IAAI,CAAC;gBACT,IAAI,CAAC;oBACH,WAAW;oBACX,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC3B,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBACpB,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;oBACzD,2BAA2B;oBAC3B,sBAAsB;oBACtB,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;oBACpE,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;oBAC3C,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC/B,CAAC;gBAED,0BAA0B;gBAC1B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC7B,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;oBAC9D,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACpC,IAAI,gBAAgB,CAAC,UAAU,CAAC,EAAE,CAAC;wBACjC,OAAO,mBAAmB,CAAC,UAAU,CAAC,CAAC;oBACzC,CAAC;oBACD,IAAI,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC;wBACxC,OAAO,0BAA0B,CAAC,UAAU,CAAC,CAAC;oBAChD,CAAC;gBACH,CAAC;gBAED,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC3B,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBACnC,CAAC;gBACD,iBAAiB;gBACjB,IAAI,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;oBAClC,OAAO,0BAA0B,CAAC,IAAI,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;gBAC1C,yBAAyB;YAC3B,CAAC;QACH,CAAC;QAED,YAAY;QACZ,OAAO;YACL,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE;YACpB,aAAa,EAAE,KAAK;SACrB,CAAC;IACJ,CAAC;IAED,wBAAwB;IACxB,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,kCAAkC,OAAO,KAAK,EAAE,CAAC,CAAC;AACpE,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,mBAAmB,CAAC,QAAgB;IACjD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAE5C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,0BAA0B,YAAY,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IAC9D,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IAEjD,mCAAmC;IACnC,IAAI,GAAG,KAAK,OAAO,EAAE,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACjC,IAAI,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC3B,OAAO,mBAAmB,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QAC1E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,gCAAgC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAC3F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,YAAY;IACZ,OAAO;QACL,MAAM,EAAE,OAAO;QACf,aAAa,EAAE,KAAK;KACrB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,IAAoB;IAC/C,kCAAkC;IAClC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;IAC7B,MAAM,oBAAoB,GACxB,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;QAC9B,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAuB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IAE1E,IAAI,oBAAoB,EAAE,CAAC;QACzB,gBAAgB;QAChB,MAAM,aAAa,GAAG,iCAAiC,CAAC,IAAI,CAAC,CAAC;QAC9D,OAAO;YACL,MAAM,EAAE,aAAa;YACrB,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,aAAa,EAAE,IAAI;SACpB,CAAC;IACJ,CAAC;IAED,SAAS;IACT,IAAI,UAAU,GAAG,EAAE,CAAC;IACpB,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACxC,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC;IAC/B,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1C,wBAAwB;QACxB,UAAU,GAAG,OAAO,CAAC,OAAO;aACzB,MAAM,CAAC,CAAC,IAAuB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;aACzD,GAAG,CAAC,CAAC,IAAuB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;aACjD,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,OAAO;QACL,MAAM,EAAE,UAAU,CAAC,IAAI,EAAE;QACzB,SAAS,EAAE,IAAI,CAAC,UAAU;QAC1B,aAAa,EAAE,KAAK;KACrB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,0BAA0B,CAAC,IAGnC;IACC,6BAA6B;IAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,IAAI,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;IAC7D,MAAM,UAAU,GAAmB;QACjC,IAAI,EAAE,MAAM;QACZ,UAAU,EAAE,SAAS;QACrB,OAAO,EAAE;YACP,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,OAAO;SAC3B;QACD,kBAAkB,EAAE,IAAI;KACzB,CAAC;IAEF,kCAAkC;IAClC,OAAO,mBAAmB,CAAC,UAAU,CAAC,CAAC;AACzC,CAAC;AAED;;GAEG;AACH,KAAK,SAAS,CAAC,CAAC,iCAAiC,CAC/C,OAAuB;IAEvB,MAAM,OAAO,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,mBAAmB,CACxC,SAAiB,EACjB,aAAsB;IAEtB,IAAI,aAAa,EAAE,CAAC;QAClB,MAAM;YACJ,UAAU,EAAE,SAAS;YACrB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,CAAC,IAAI,EAAE,EAAE;YACxD,kBAAkB,EAAE,IAAI;SACzB,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,GAAY;IACnC,OAAO,CACL,GAAG,KAAK,IAAI;QACZ,GAAG,KAAK,SAAS;QACjB,OAAQ,GAA+B,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAC7E,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,GAAY;IACpC,OAAO,CACL,GAAG,KAAK,IAAI;QACZ,GAAG,KAAK,SAAS;QACjB,OAAO,GAAG,KAAK,QAAQ;QACvB,MAAM,IAAI,GAAG;QACZ,GAAyB,CAAC,IAAI,KAAK,MAAM;QAC1C,SAAS,IAAI,GAAG;QAChB,OAAQ,GAA4B,CAAC,OAAO,KAAK,QAAQ;QACzD,YAAY,IAAI,GAAG;QACnB,OAAQ,GAA+B,CAAC,UAAU,KAAK,QAAQ,CAChE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,GAAY;IAC3C,OAAO,CACL,GAAG,KAAK,IAAI;QACZ,GAAG,KAAK,SAAS;QACjB,OAAO,GAAG,KAAK,QAAQ;QACvB,MAAM,IAAI,GAAG;QACZ,GAAyB,CAAC,IAAI,KAAK,IAAI;QACvC,GAAyB,CAAC,IAAI,KAAK,SAAS;QAC7C,OAAQ,GAAyB,CAAC,IAAI,KAAK,QAAQ;QACnD,SAAS,IAAK,GAAwB,CAAC,IAAI,CAC5C,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,IAAc;IACtD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IAEzB,gCAAgC;IAChC,OAAO,WAAW,CAAC,QAAQ,CAAC,CAAC;AAC/B,CAAC"}