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

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

@@ -38,6 +38,7 @@ exports.generatePropertiesJsonSchemaWithZod = generatePropertiesJsonSchemaWithZo
 exports.generatePageDataJsonSchemasWithZod = generatePageDataJsonSchemasWithZod;
 exports.generatePageDataTypesWithZod = generatePageDataTypesWithZod;
 exports.generatePropertiesTypesWithZod = generatePropertiesTypesWithZod;
+exports.serializeZodSchema = serializeZodSchema;
 exports.generatePageSectionsDataZod = generatePageSectionsDataZod;
 // 导入 Zod 工具集
 const property_1 = require("@blocklet/pages-kit/utils/property");
@@ -503,16 +504,88 @@ function generatePropertiesTypesWithZod(properties) {
     const typeString = (0, zod_utils_1.zodSchemaToTypeString)(zodSchema, 'Props');
     return typeString;
 }
+/**
+ * 将 Zod schema 序列化为 JSON 描述结构
+ */
+function serializeZodSchema(schema) {
+    // 处理不同类型的 schema
+    if (schema instanceof zod_1.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 zod_1.z.ZodString) {
+        return {
+            type: 'string',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof zod_1.z.ZodNumber) {
+        return {
+            type: 'number',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof zod_1.z.ZodBoolean) {
+        return {
+            type: 'boolean',
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof zod_1.z.ZodArray) {
+        return {
+            type: 'array',
+            items: serializeZodSchema(schema._def.type),
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof zod_1.z.ZodRecord) {
+        return {
+            type: 'record',
+            valueType: serializeZodSchema(schema._def.valueType),
+            description: schema.description,
+            optional: schema.isOptional(),
+        };
+    }
+    if (schema instanceof zod_1.z.ZodOptional) {
+        const innerType = serializeZodSchema(schema._def.innerType);
+        innerType.optional = true;
+        return innerType;
+    }
+    if (schema instanceof zod_1.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 的序列化描述
  */
 function generatePageSectionsDataZod(state, pageId) {
     const page = state.pages[pageId];
     if (!page) {
-        return zod_1.z.object({}).describe('Empty page data');
+        return serializeZodSchema(zod_1.z.object({}).describe('Empty page data'));
     }
     const sectionSchemas = {};
     // 收集页面所有section的配置信息
@@ -570,7 +643,8 @@ function generatePageSectionsDataZod(state, pageId) {
             sectionSchemas[sectionName] = zod_1.z.object({}).describe(section.templateDescription || '');
         }
     });
-    // 创建包含所有section的schema
-    return zod_1.z.object(sectionSchemas).describe(`Page ${page.title || pageId} sections data`);
+    // FIXME: 这里需要优化，AIGNE Framework 校验修改后，不需要序列化，直接使用
+    // 创建包含所有section的schema并序列化返回
+    return serializeZodSchema(zod_1.z.object(sectionSchemas).describe(`Page ${page.title || pageId} sections data`));
 }
 exports.default = generateWrapperCode;