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

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,74 @@ export function generatePropertiesTypesWithZod(properties) {
     const typeString = zodSchemaToTypeString(zodSchema, 'Props');
     return typeString;
 }
+/**
+ * 生成特定页面所有 section 数据的 Zod 类型定义
+ * @param state 应用状态
+ * @param pageId 目标页面ID
+ * @returns 包含所有 section 数据类型的 Zod schema
+ */
+export function generatePageSectionsDataZod(state, pageId) {
+    const page = state.pages[pageId];
+    if (!page) {
+        return 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 || '');
+        }
+    });
+    // 创建包含所有section的schema
+    return z.object(sectionSchemas).describe(`Page ${page.title || pageId} sections data`);
+}
 export default generateWrapperCode;