feat：添加AI返回JSON处理

Author: lichunn

packages/canvas/container/src/composables/useAIChat.ts

@@ -12,6 +12,7 @@
 
 import { utils } from '@opentiny/tiny-engine-utils'
 import * as jsonpatch from 'fast-json-patch'
+import { jsonrepair } from 'jsonrepair'
 import {
   useCanvas,
   useMaterial,
@@ -373,6 +374,86 @@ const shouldShowNodeAILoading = (nodeId: string): boolean => {
   return status?.state === 'loading'
 }
 
+/**
+ * 逐个解析 JSON Patch 数组中的顶级 patch 对象
+ * 当整个数组因括号不匹配等原因无法解析时，通过顶层 `{"op":` 模式分割后分别解析
+ * 对每个单独的对象尝试 JSON.parse → jsonrepair → 修复多余括号
+ */
+const parseIndividualPatches = (jsonStr: string): any[] => {
+  const patches: any[] = []
+
+  // 移除外层数组的 [ ] 包裹
+  let str = jsonStr.trim()
+  if (str.startsWith('[')) str = str.slice(1)
+  if (str.endsWith(']')) str = str.slice(0, -1)
+
+  // 通过顶层 {"op": 模式分割各个 patch 对象
+  const segments: string[] = []
+  let depth = 0
+  let inString = false
+  let escape = false
+  let segStart = -1
+
+  for (let i = 0; i < str.length; i++) {
+    const ch = str[i]
+
+    if (escape) {
+      escape = false
+      continue
+    }
+    if (ch === '\\' && inString) {
+      escape = true
+      continue
+    }
+    if (ch === '"') {
+      inString = !inString
+      continue
+    }
+    if (inString) continue
+
+    if (ch === '{') {
+      if (depth === 0) segStart = i
+      depth++
+    } else if (ch === '}') {
+      depth--
+      if (depth === 0 && segStart >= 0) {
+        segments.push(str.slice(segStart, i + 1))
+        segStart = -1
+      }
+    }
+  }
+
+  // 对每个 segment 尝试解析，逐步增强容错
+  for (const seg of segments) {
+    let patch = null
+    try {
+      patch = JSON.parse(seg)
+    } catch {
+      try {
+        patch = JSON.parse(jsonrepair(seg))
+      } catch {
+        // 尝试修复多余的尾部 }
+        // 找到最后一个合法的 } 位置（通过从后往前逐步移除多余的 }）
+        let fixed = seg
+        for (let trim = 0; trim < 5; trim++) {
+          fixed = fixed.replace(/\}(\}*)$/, '$1') // 移除最末尾的一个 }
+          try {
+            patch = JSON.parse(fixed)
+            break
+          } catch {
+            // 继续尝试
+          }
+        }
+      }
+    }
+    if (patch && patch.op && patch.path) {
+      patches.push(patch)
+    }
+  }
+
+  return patches
+}
+
 const findJsonPatchPath = (node, targetId, path = []) => {
   if (!node || typeof node !== 'object') return null
 
@@ -401,6 +482,169 @@ const findJsonPatchPath = (node, targetId, path = []) => {
   return null
 }
 
+/**
+ * 解析 AI 返回的 JSON Patch 字符串，逐步增强容错：
+ * 1. 剥离 Markdown 代码块包裹
+ * 2. 直接 JSON.parse
+ * 3. 修复未转义换行符后再 parse
+ * 4. jsonrepair 修复
+ * 5. 逐个 patch 解析（跳过损坏的单个 patch）
+ */
+const parseJsonPatches = (content: string): any[] | null => {
+  let jsonStr = content
+  // 剥离 Markdown 代码块
+  const codeBlockMatch = jsonStr.match(/```(?:json|schema)?([\s\S]*?)```/)
+  if (codeBlockMatch) {
+    jsonStr = codeBlockMatch[1].trim()
+  }
+
+  let patches: any[] = []
+  try {
+    // 策略1：直接解析
+    try {
+      patches = JSON.parse(jsonStr)
+    } catch {
+      // 策略2：修复 JSON 字符串值中未转义的换行符
+      try {
+        const fixedStr = jsonStr.replace(
+          /"((?:[^"\\]|\\.)*)"/g,
+          (match, inner) => '"' + inner.replace(/\n/g, '\\n').replace(/\r/g, '\\r') + '"'
+        )
+        patches = JSON.parse(fixedStr)
+      } catch {
+        // 策略3：使用 jsonrepair 修复
+        try {
+          patches = JSON.parse(jsonrepair(jsonStr))
+        } catch {
+          // 策略4：逐个 patch 解析 — AI 常在深层嵌套的 children 中产生多余的括号，
+          // 导致整个数组解析失败。逐个提取顶级 patch 对象分别解析，跳过损坏的。
+          patches = parseIndividualPatches(jsonStr)
+        }
+      }
+    }
+  } catch (error) {
+    return null
+  }
+
+  if (!Array.isArray(patches)) {
+    patches = [patches]
+  }
+  return patches
+}
+
+/**
+ * 辅助函数：根据路径段从对象中安全取值
+ */
+const getValueBySegments = (obj: any, segments: string[]): any => {
+  return segments.reduce((o, key) => (o !== null ? o[key] : undefined), obj)
+}
+
+/**
+ * 辅助函数：将路径段拼接为 JSON Pointer 格式（以 / 开头）
+ */
+const toPointer = (segments: string[]): string => '/' + segments.filter(Boolean).join('/')
+
+/**
+ * 将 JSON Patch 数组应用到页面 schema 上
+ * 按操作类型分步应用：先 replace/remove，再 add，最后其他
+ * add 操作会自动处理：目标数组不存在时初始化、索引越界时追加到末尾
+ * @param patches JSON Patch 数组
+ * @param pageSchema 当前页面 schema
+ * @param parentPath 当前节点在 schema 中的路径
+ * @returns 应用后的新 schema
+ */
+const applyPatchesToSchema = (patches: any[], pageSchema: object, parentPath: string): object => {
+  // 分离操作类型
+  const replacePatches = patches.filter((p) => p.op === 'replace' || p.op === 'remove')
+  const addPatches = patches.filter((p) => p.op === 'add')
+  const otherPatches = patches.filter((p) => p.op !== 'replace' && p.op !== 'remove' && p.op !== 'add')
+
+  // 先应用 replace/remove 操作
+  let newSchema = replacePatches.reduce((acc, patch) => {
+    try {
+      const fullPatch = {
+        ...patch,
+        path: parentPath + patch.path
+      }
+      return jsonpatch.applyPatch(acc, [fullPatch], false, false).newDocument
+    } catch (error) {
+      return acc
+    }
+  }, pageSchema)
+
+  // 再应用 add 操作
+  // 需要处理的情况：
+  // 1. 目标数组不存在 → 先初始化空数组，再追加
+  // 2. 索引超出数组长度 → 降级为追加到末尾（/-）
+  // 3. 最后一段不是数字索引且目标是数组 → 追加到末尾（/-）
+  newSchema = addPatches.reduce((acc, patch) => {
+    try {
+      const pathSegments = patch.path.split('/').filter(Boolean)
+      const lastSegment = pathSegments[pathSegments.length - 1]
+
+      if (!lastSegment) {
+        return acc
+      }
+
+      const parentSegments = pathSegments.slice(0, -1)
+      const fullParentSegments = (parentPath + '/' + parentSegments.join('/')).split('/').filter(Boolean)
+      let parentValue = getValueBySegments(acc, fullParentSegments)
+      let fixedPath = patch.path
+
+      // 父路径对应的值不是数组但路径暗示要往数组中插入（如 /children/0），
+      // 需要先初始化空数组
+      if (!Array.isArray(parentValue) && /^\d+$/.test(lastSegment)) {
+        const arrayPath = toPointer(parentPath.split('/').filter(Boolean).concat(parentSegments))
+        try {
+          const patched = jsonpatch.applyPatch(
+            acc,
+            [{ op: 'add', path: arrayPath, value: [] }],
+            false,
+            false
+          ).newDocument
+          acc = patched
+          parentValue = getValueBySegments(acc, fullParentSegments)
+        } catch {
+          // 路径已存在或初始化失败，尝试继续
+        }
+      }
+
+      // 根据目标数组的状态修正路径
+      if (Array.isArray(parentValue)) {
+        const index = Number(lastSegment)
+        if (Number.isNaN(index)) {
+          fixedPath = toPointer(pathSegments) + '/-'
+        } else if (index >= parentValue.length) {
+          fixedPath = toPointer(parentSegments) + '/-'
+        }
+      }
+
+      const fullPatch = {
+        ...patch,
+        path: parentPath + fixedPath
+      }
+      return jsonpatch.applyPatch(acc, [fullPatch], false, false).newDocument
+    } catch (error) {
+      return acc
+    }
+  }, newSchema)
+
+  // 最后应用其他操作（move, copy, test 等）
+  newSchema = otherPatches.reduce((acc, patch) => {
+    try {
+      const fullPatch = {
+        ...patch,
+        path: parentPath + patch.path
+      }
+      return jsonpatch.applyPatch(acc, [fullPatch], false, false).newDocument
+    } catch (error) {
+      return acc
+    }
+  }, newSchema)
+
+  return newSchema
+}
+
 /**
  * 应用AI返回的JSON Patch到页面schema，完成画布更新
  * 逻辑：设置chatContent、设置aiModifiedNodeData为AI修改后的节点schema、修改画布节点schema为AI的schema
@@ -418,20 +662,13 @@ const applyAIPatches = (nodeId: string, chatResponse: any, chatContent?: string)
   const { getPageSchema, getNode, updatePageSchema, setSaved } = useCanvas()
 
   const content = chatResponse.choices[0].message.content
-  const validJsonPatches = JSON.parse(content)
-  const parentPath = findJsonPatchPath(getPageSchema(), nodeId)
+  const validJsonPatches = parseJsonPatches(content)
+  if (!validJsonPatches) {
+    return false
+  }
 
-  const newSchema = validJsonPatches.reduce((acc, patch) => {
-    try {
-      const fullPatch = {
-        ...patch,
-        path: parentPath + patch.path
-      }
-      return jsonpatch.applyPatch(acc, [fullPatch], false, false).newDocument
-    } catch (error) {
-      return acc
-    }
-  }, getPageSchema())
+  const parentPath = findJsonPatchPath(getPageSchema(), nodeId)
+  const newSchema = applyPatchesToSchema(validJsonPatches, getPageSchema(), parentPath)
 
   fixMethods(newSchema.methods)
   schemaAutoFix(newSchema.children)
@@ -441,7 +678,7 @@ const applyAIPatches = (nodeId: string, chatResponse: any, chatContent?: string)
 
   // 设置 AI 状态：chatContent、aiModifiedNodeData
   const modifiedNode = getNode(nodeId)
-  const modifiedNodeData = modifiedNode ? deepClone(modifiedNode) : validJsonPatches[0].value
+  const modifiedNodeData = modifiedNode ? deepClone(modifiedNode) : validJsonPatches[0]?.value
 
   updateNodeAIStatus(nodeId, {
     state: 'confirm',