From 33663f96031076e8eea5fcd4d132f53eb48c0833 Mon Sep 17 00:00:00 2001
From: liujuping <liujup@foxmail.com>
Date: Tue, 4 Jul 2023 09:51:17 +0800
Subject: [PATCH] feat: add document-model shell return types

---
 docs/docs/api/model/document-model.md            | 6 ++++--
 packages/shell/src/model/document-model.ts       | 2 +-
 packages/types/src/shell/model/document-model.ts | 2 +-
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/docs/docs/api/model/document-model.md b/docs/docs/api/model/document-model.md
index b609f0050..8d813e767 100644
--- a/docs/docs/api/model/document-model.md
+++ b/docs/docs/api/model/document-model.md
@@ -139,10 +139,12 @@ importSchema(schema: IPublicTypeRootSchema): void;
  * @param stage
  * @returns
  */
-exportSchema(stage: IPublicEnumTransformStage): any;
+exportSchema(stage: IPublicEnumTransformStage): IPublicTypeRootSchema | undefined;
 ```
 
-相关类型：[IPublicEnumTransformStage](https://github.com/alibaba/lowcode-engine/blob/main/packages/types/src/shell/enum/transform-stage.ts)
+相关类型：
+- [IPublicEnumTransformStage](https://github.com/alibaba/lowcode-engine/blob/main/packages/types/src/shell/enum/transform-stage.ts)
+- [IPublicTypeRootSchema](https://github.com/alibaba/lowcode-engine/blob/main/packages/types/src/shell/type/root-schema.ts)
 
 ### insertNode
 
diff --git a/packages/shell/src/model/document-model.ts b/packages/shell/src/model/document-model.ts
index 06f3cebcd..2c5c7b632 100644
--- a/packages/shell/src/model/document-model.ts
+++ b/packages/shell/src/model/document-model.ts
@@ -164,7 +164,7 @@ export class DocumentModel implements IPublicModelDocumentModel {
    * @param stage
    * @returns
    */
-  exportSchema(stage: IPublicEnumTransformStage = IPublicEnumTransformStage.Render): any {
+  exportSchema(stage: IPublicEnumTransformStage = IPublicEnumTransformStage.Render): IPublicTypeRootSchema | undefined {
     return this[documentSymbol].export(stage);
   }
 
diff --git a/packages/types/src/shell/model/document-model.ts b/packages/types/src/shell/model/document-model.ts
index 2ef0b532b..4c9344eb4 100644
--- a/packages/types/src/shell/model/document-model.ts
+++ b/packages/types/src/shell/model/document-model.ts
@@ -89,7 +89,7 @@ export interface IPublicModelDocumentModel<
    * @param stage
    * @returns
    */
-  exportSchema(stage: IPublicEnumTransformStage): any;
+  exportSchema(stage: IPublicEnumTransformStage): IPublicTypeRootSchema | undefined;
 
   /**
    * 插入节点