@blocklet/pages-kit-block-studio 0.4.91 → 0.4.92

package/lib/esm/utils/generate-wrapper-code.js

@@ -462,16 +462,88 @@ export function generatePropertiesTypesWithZod(properties) {
     const typeString = zodSchemaToTypeString(zodSchema, 'Props');
     return typeString;
 }
+/**
+ * 将 Zod schema 序列化为 JSON 描述结构
+ */
+export function serializeZodSchema(schema) {
+    // 处理不同类型的 schema
+    if (schema instanceof z.ZodObject) {
+        const shape = schema._def.shape();
+        const properties = {};
+        Object.entries(shape).forEach(([key, value]) => {
+            properties[key] = serializeZodSchema(value);
+        });
+        return {
+            type: 'object',
+            properties,
+            description: schema.description,
+        };
+    }
+    if (schema instanceof z.ZodString) {
+        return {
+            type: 'string',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof z.ZodNumber) {
+        return {
+            type: 'number',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof z.ZodBoolean) {
+        return {
+            type: 'boolean',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof z.ZodArray) {
+        return {
+            type: 'array',
+            items: serializeZodSchema(schema._def.type),
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof z.ZodRecord) {
+        return {
+            type: 'record',
+            valueType: serializeZodSchema(schema._def.valueType),
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof z.ZodOptional) {
+        const innerType = serializeZodSchema(schema._def.innerType);
+        innerType.optional = true;
+        return innerType;
+    }
+    if (schema instanceof z.ZodAny) {
+        return {
+            type: 'any',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    // 默认情况
+    return {
+        type: 'unknown',
+        description: schema.description,
+    };
+}
 /**
  * 生成特定页面所有 section 数据的 Zod 类型定义
  * @param state 应用状态
  * @param pageId 目标页面ID
- * @returns 包含所有 section 数据类型的 Zod schema
+ * @returns 包含所有 section 数据类型的 Zod schema 的序列化描述
  */
 export function generatePageSectionsDataZod(state, pageId) {
     const page = state.pages[pageId];
     if (!page) {
-        return z.object({}).describe('Empty page data');
+        return serializeZodSchema(z.object({}).describe('Empty page data'));
     }
     const sectionSchemas = {};
     // 收集页面所有section的配置信息
@@ -529,7 +601,8 @@ export function generatePageSectionsDataZod(state, pageId) {
             sectionSchemas[sectionName] = z.object({}).describe(section.templateDescription || '');
         }
     });
-    // 创建包含所有section的schema
-    return z.object(sectionSchemas).describe(`Page ${page.title || pageId} sections data`);
+    // FIXME: 这里需要优化，AIGNE Framework 校验修改后，不需要序列化，直接使用
+    // 创建包含所有section的schema并序列化返回
+    return serializeZodSchema(z.object(sectionSchemas).describe(`Page ${page.title || pageId} sections data`));
 }
 export default generateWrapperCode;