fix: dataSource pane export schema

twinkle77 · twinkle77 · commit a0033a0b0f17 · 2022-05-11T09:46:30.000+08:00
diff --git a/packages/plugin-datasource-pane/src/classname-setter/index.tsx b/packages/plugin-datasource-pane/src/classname-setter/index.tsx
@@ -1,6 +1,7 @@
 import React, { PureComponent } from 'react';
 import PropTypes from 'prop-types';
 import { Select } from '@alifd/next';
+import { common } from '@alilc/lowcode-engine';
 
 interface Color {
   rgb: any;
@@ -27,7 +28,7 @@ export default class ClassNameView extends PureComponent<PluginProps> {
 
   getClassNameList = () => {
     const { project } = this.context;
-    const schema = project.exportSchema();
+    const schema = project.exportSchema(common.designerCabin.TransformStage.Save);
     const { css } = schema.componentsTree[0];
     const classNameList = [];
     const re = /\.?\w+[^{]+\{[^}]*\}/g;
diff --git a/packages/plugin-datasource-pane/src/components/ClassNameView/ClassNameView.tsx b/packages/plugin-datasource-pane/src/components/ClassNameView/ClassNameView.tsx
@@ -1,7 +1,7 @@
 import React, { PureComponent } from 'react';
 import PropTypes from 'prop-types';
 import { Select } from '@alifd/next';
-
+import { common } from '@alilc/lowcode-engine';
 
 export interface PluginProps {
     value: string;
@@ -30,7 +30,7 @@ export default class ClassNameView extends PureComponent<PluginProps, ClassNameV
 
     getClassNameList = () => {
         const { project } = this.context;
-        const schema = project.exportSchema();
+        const schema = project.exportSchema(common.designerCabin.TransformStage.Save);
         const { css } = schema.componentsTree[0];
         const classNameList: string[] = [];
         const re = /\.?\w+[^{]+\{[^}]*\}/g;
diff --git a/packages/plugin-datasource-pane/src/pane/index.tsx b/packages/plugin-datasource-pane/src/pane/index.tsx
@@ -24,6 +24,7 @@ import { isSchemaValid, correctSchema } from '../utils/schema';
 import { createStateService } from '../utils/stateMachine';
 import { DataSourcePaneContext } from '../utils/panel-context';
 import { mergeTwoObjectListByKey } from '../utils/misc';
+import { common } from '@alilc/lowcode-engine';
 
 import './index.scss';
 
@@ -119,7 +120,7 @@ export default class DataSourcePanePlugin extends PureComponent<
   handleSchemaChange = (schema: DataSource) => {
     const { project, onSchemaChange } = this.props;
     if (project) {
-      const docSchema = project.exportSchema();
+      const docSchema = project.exportSchema(common.designerCabin.TransformStage.Save);
       if (!_isEmpty(docSchema)) {
         _set(docSchema, 'componentsTree[0].dataSource', schema);
         project.importSchema(docSchema);
@@ -148,7 +149,7 @@ export default class DataSourcePanePlugin extends PureComponent<
 
     if (!active) return null;
 
-    const projectSchema = project.exportSchema() ?? {};
+    const projectSchema = project.exportSchema(common.designerCabin.TransformStage.Save) ?? {};
     let schema = defaultSchema;
     if (_isFunction(defaultSchema)) {
       schema = defaultSchema();
