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

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

@@ -1,5 +1,6 @@
 // 导入 Zod 工具集
 import { getComponentDependencies } from '@blocklet/pages-kit/utils/property';
+import { z } from 'zod';
 import { BASIC_COMPONENT_SECTION_TYPES, BASIC_COMPONENT_SECTION_JSON_SCHEMAS } from './block-props-utils';
 import { propertiesToZodSchema, zodSchemaToTypeString, zodSchemaToJsonSchema } from './zod-utils';
 export async function generateWrapperCode({ project, state, version }) {
@@ -461,4 +462,147 @@ 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 的序列化描述
+ */
+export function generatePageSectionsDataZod(state, pageId) {
+    const page = state.pages[pageId];
+    if (!page) {
+        return serializeZodSchema(z.object({}).describe('Empty page data'));
+    }
+    const sectionSchemas = {};
+    // 收集页面所有section的配置信息
+    page.sectionIds.forEach((sectionId) => {
+        const section = page.sections[sectionId];
+        if (!section || !section.isTemplateSection)
+            return;
+        const sectionName = section.name || section.id;
+        if (section.component === 'custom-component') {
+            const componentId = section.config?.componentId;
+            if (!componentId)
+                return;
+            const component = state.components[componentId]?.data || state.resources.components?.[componentId]?.component;
+            if (!component)
+                return;
+            // 使用propertiesToZodSchema生成Zod Schema，并包含llmConfig配置
+            const zodSchema = propertiesToZodSchema(component.properties || {}, {
+                addZodDescribe: false,
+                llmConfig: section.llmConfig?.properties || {},
+            });
+            // 添加section描述
+            sectionSchemas[sectionName] = zodSchema.describe(section.templateDescription || '');
+        }
+        else if (BASIC_COMPONENT_SECTION_TYPES[section.component]) {
+            // 处理基本组件类型
+            const componentType = section.component;
+            const schemaProperties = BASIC_COMPONENT_SECTION_JSON_SCHEMAS[componentType];
+            // 将JSON Schema转换为Zod Schema
+            const zodProperties = Object.entries(schemaProperties || {}).reduce((acc, [key, value]) => {
+                // 简单处理常见类型
+                if (value.type === 'string') {
+                    acc[key] = z.string().optional();
+                }
+                else if (value.type === 'number') {
+                    acc[key] = z.number().optional();
+                }
+                else if (value.type === 'boolean') {
+                    acc[key] = z.boolean().optional();
+                }
+                else if (value.type === 'array') {
+                    acc[key] = z.array(z.any()).optional();
+                }
+                else if (value.type === 'object') {
+                    acc[key] = z.record(z.string(), z.any()).optional();
+                }
+                else {
+                    acc[key] = z.any().optional();
+                }
+                return acc;
+            }, {});
+            sectionSchemas[sectionName] = z.object(zodProperties).describe(section.templateDescription || '');
+        }
+        else {
+            // 其他类型的section，使用默认对象
+            sectionSchemas[sectionName] = z.object({}).describe(section.templateDescription || '');
+        }
+    });
+    // FIXME: 这里需要优化，AIGNE Framework 校验修改后，不需要序列化，直接使用
+    // 创建包含所有section的schema并序列化返回
+    return serializeZodSchema(z.object(sectionSchemas).describe(`Page ${page.title || pageId} sections data`));
+}
 export default generateWrapperCode;