sloth-d2c-mcp 1.0.4-beta100

package/dist/build/utils/extract.js

@@ -0,0 +1,168 @@
+/**
+ * 从LLM的文本输出中提取JSON代码块
+ * @param text LLM返回的文本内容
+ * @returns 提取到的JSON对象和解析状态
+ */
+export function extractJson(text) {
+    try {
+        // 移除文本前后的空白字符
+        const trimmedText = text.trim();
+        // 正则表达式匹配 ```json 代码块
+        const jsonBlockRegex = /```json\s*\n([\s\S]*?)\n\s*```/gi;
+        const matches = Array.from(trimmedText.matchAll(jsonBlockRegex));
+        if (matches.length > 0) {
+            // 取最后一个匹配的JSON代码块
+            const jsonContent = matches[matches.length - 1][1].trim();
+            try {
+                const parsed = JSON.parse(jsonContent);
+                return { state: 'successful-parse', value: parsed };
+            }
+            catch (parseError) {
+                return {
+                    state: 'failed-parse',
+                    error: `JSON解析失败: ${parseError instanceof Error ? parseError.message : String(parseError)}`
+                };
+            }
+        }
+        // 如果没有找到 ```json 代码块，尝试直接解析整个文本
+        // 寻找可能的JSON对象（以 { 开始，以 } 结束）
+        const jsonObjectRegex = /\{[\s\S]*\}/;
+        const jsonMatch = trimmedText.match(jsonObjectRegex);
+        if (jsonMatch) {
+            try {
+                const parsed = JSON.parse(jsonMatch[0]);
+                return { state: 'successful-parse', value: parsed };
+            }
+            catch (parseError) {
+                return {
+                    state: 'failed-parse',
+                    error: `JSON对象解析失败: ${parseError instanceof Error ? parseError.message : String(parseError)}`
+                };
+            }
+        }
+        // 尝试寻找数组格式的JSON（以 [ 开始，以 ] 结束）
+        const jsonArrayRegex = /\[[\s\S]*\]/;
+        const arrayMatch = trimmedText.match(jsonArrayRegex);
+        if (arrayMatch) {
+            try {
+                const parsed = JSON.parse(arrayMatch[0]);
+                return { state: 'successful-parse', value: parsed };
+            }
+            catch (parseError) {
+                return {
+                    state: 'failed-parse',
+                    error: `JSON数组解析失败: ${parseError instanceof Error ? parseError.message : String(parseError)}`
+                };
+            }
+        }
+        return {
+            state: 'failed-parse',
+            error: '未找到有效的JSON内容。请确保响应包含 ```json 代码块或有效的JSON对象/数组。'
+        };
+    }
+    catch (error) {
+        return {
+            state: 'failed-parse',
+            error: `提取JSON时发生错误: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+/**
+ * 从 markdown 文本中提取代码块
+ * 遵循 CommonMark 规范，使用三个反引号包围的代码块格式
+ * 支持从代码块 info string 中提取组件名，格式：```language:ComponentName
+ * @param text 包含 markdown 代码块的文本
+ * @returns 提取到的代码块数组
+ */
+export function extractCodeBlocks(text) {
+    const trimmedText = text.trim();
+    // 匹配带有组件名的代码块
+    // 格式: ```language:ComponentName\ncode\n```
+    // 组件名是可选的，通过冒号分隔
+    // 示例: ```tsx:UserCard 或 ```vue:ProductList
+    const codeBlockRegex = /```([\w-]+)(?::([A-Z][a-zA-Z0-9]*))?\s*\r?\n([\s\S]*?)\r?\n?\s*```/g;
+    const codeBlocks = [];
+    let match;
+    while ((match = codeBlockRegex.exec(trimmedText)) !== null) {
+        const language = match[1];
+        const declaredComponentName = match[2]; // 从 info string 中提取的组件名（可能为 undefined）
+        const code = match[3].trim();
+        if (code) {
+            // 优先使用声明的组件名，否则降级到正则提取
+            const componentName = declaredComponentName || extractComponentName(code, language);
+            codeBlocks.push({
+                language,
+                code,
+                componentName
+            });
+        }
+    }
+    return codeBlocks;
+}
+/**
+ * @deprecated: 为支持跨框架已废弃，改为从模型返回中提取
+ * 从代码中提取组件名
+ * 支持多种常见的组件定义模式
+ * @param code 代码内容
+ * @param language 编程语言
+ * @returns 组件名或 undefined
+ */
+export function extractComponentName(code, language) {
+    // 移除注释和空白字符
+    const cleanCode = code.trim();
+    // React 函数组件模式
+    const patterns = [
+        // export default function ComponentName / export default async function ComponentName
+        /export\s+default\s+(?:async\s+)?function\s+([A-Z][a-zA-Z0-9]*)/,
+        // export async function ComponentName / export function ComponentName
+        /export\s+(?:async\s+)?function\s+([A-Z][a-zA-Z0-9]*)/,
+        // function ComponentName
+        /^function\s+([A-Z][a-zA-Z0-9]*)/m,
+        // export default memo(function ComponentName ...) / React.memo
+        /export\s+default\s+(?:React\.)?memo\(\s*function\s+([A-Z][a-zA-Z0-9]*)/,
+        // export default forwardRef(..., function ComponentName ...) / React.forwardRef
+        /export\s+default\s+(?:React\.)?forwardRef\([^,]*,\s*function\s+([A-Z][a-zA-Z0-9]*)/,
+        // export default NamedComponent
+        /export\s+default\s+([A-Z][a-zA-Z0-9]*)/,
+        // export const ComponentName: Type = ...  (可选类型注解)
+        /export\s+const\s+([A-Z][a-zA-Z0-9]*)\s*(?::[^=]+)?\s*=/,
+        // const ComponentName: Type = ...  (可选类型注解)
+        /const\s+([A-Z][a-zA-Z0-9]*)\s*(?::[^=]+)?\s*=/,
+        // class ComponentName / export default class ComponentName
+        /export\s+default\s+class\s+([A-Z][a-zA-Z0-9]*)/,
+        /class\s+([A-Z][a-zA-Z0-9]*)/,
+        // export { ComponentName } / export { ComponentName as default }
+        /export\s*\{\s*([A-Z][a-zA-Z0-9]*)\s*(?:as\s*default)?\s*\}/,
+        // interface ComponentNameProps / type ComponentNameProps  (用于推断组件名)
+        /interface\s+([A-Z][a-zA-Z0-9]*)Props/,
+        /type\s+([A-Z][a-zA-Z0-9]*)Props/
+    ];
+    for (const pattern of patterns) {
+        const match = cleanCode.match(pattern);
+        if (match && match[1]) {
+            let componentName = match[1];
+            // 如果是 Props 接口，移除 Props 后缀
+            if (componentName.endsWith('Props')) {
+                componentName = componentName.slice(0, -5);
+            }
+            // 验证组件名符合 PascalCase 规范
+            if (/^[A-Z][a-zA-Z0-9]*$/.test(componentName)) {
+                return componentName;
+            }
+        }
+    }
+    return undefined;
+}
+/**
+ * 从文本中提取所有代码块并获取组件信息
+ * @param text 包含代码块的文本
+ * @returns 包含代码和组件名的信息数组
+ */
+export function extractCodeAndComponents(text) {
+    const codeBlocks = extractCodeBlocks(text);
+    return codeBlocks.map(block => ({
+        code: block.code,
+        componentName: block.componentName,
+        language: block.language
+    }));
+}