@riil-frontend/component-topology 3.5.0-a.9 → 3.5.2

package/es/core/editor/components/titlebar/TopoEditorTitleBar.js

@@ -1,15 +1,59 @@
 import _Box from "@alifd/next/es/box";
-import _asyncToGenerator from "@babel/runtime/helpers/asyncToGenerator";
 import _Dialog from "@alifd/next/es/dialog";
 import _Button from "@alifd/next/es/button";
+import _asyncToGenerator from "@babel/runtime/helpers/asyncToGenerator";
 import _regeneratorRuntime from "@babel/runtime/regenerator";
 import React from "react";
 import classnames from "classnames";
 import TitleWidget from "../../../components/titlebar/widgets/TitleWidget";
 import styles from "../../../components/titlebar/TitleBar.module.scss";
+import { useState } from "react";
+import { useEffect } from "react";
 export default function TopoEditorTitleBar(props) {
   var topo = props.topo,
       topoEditApi = props.topoEditApi;
+  var store = topo.store;
+
+  var _store$useModel = store.useModel("topoEdit"),
+      topoEditState = _store$useModel[0],
+      topoEditDispatchers = _store$useModel[1];
+
+  var saving = topoEditState.saving;
+
+  var _useState = useState(false),
+      exitFlag = _useState[0],
+      setExitFlag = _useState[1];
+
+  var saveAndExit = /*#__PURE__*/function () {
+    var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
+      return _regeneratorRuntime.wrap(function _callee$(_context) {
+        while (1) {
+          switch (_context.prev = _context.next) {
+            case 0:
+              _context.next = 2;
+              return topo.save();
+
+            case 2:
+              setExitFlag(true);
+
+            case 3:
+            case "end":
+              return _context.stop();
+          }
+        }
+      }, _callee);
+    }));
+
+    return function saveAndExit() {
+      return _ref.apply(this, arguments);
+    };
+  }();
+
+  useEffect(function () {
+    if (!saving && exitFlag) {
+      topo.exitEditMode();
+    }
+  }, [exitFlag, saving]);
   return /*#__PURE__*/React.createElement("div", {
     className: classnames(styles.titlebar, styles.titlebarEditMode)
   }, /*#__PURE__*/React.createElement("div", {
@@ -30,6 +74,7 @@ export default function TopoEditorTitleBar(props) {
   }, "\u6DFB\u52A0\u8D44\u6E90"), /*#__PURE__*/React.createElement(_Button, {
     className: styles.button,
     onClick: function onClick() {
+      setExitFlag(false);
       topo.save();
     }
   }, "\u4FDD\u5B58"), /*#__PURE__*/React.createElement(_Button, {
@@ -39,19 +84,19 @@ export default function TopoEditorTitleBar(props) {
         title: "提示",
         content: "若直接退出，图上修改的内容将不会保存！",
         onOk: function () {
-          var _onOk = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee() {
-            return _regeneratorRuntime.wrap(function _callee$(_context) {
+          var _onOk = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2() {
+            return _regeneratorRuntime.wrap(function _callee2$(_context2) {
               while (1) {
-                switch (_context.prev = _context.next) {
+                switch (_context2.prev = _context2.next) {
                   case 0:
-                    topo.exitEditMode();
+                    saveAndExit();
 
                   case 1:
                   case "end":
-                    return _context.stop();
+                    return _context2.stop();
                 }
               }
-            }, _callee);
+            }, _callee2);
           }));
 
           function onOk() {
@@ -61,23 +106,19 @@ export default function TopoEditorTitleBar(props) {
           return onOk;
         }(),
         onCancel: function () {
-          var _onCancel = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee2() {
-            return _regeneratorRuntime.wrap(function _callee2$(_context2) {
+          var _onCancel = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime.mark(function _callee3() {
+            return _regeneratorRuntime.wrap(function _callee3$(_context3) {
               while (1) {
-                switch (_context2.prev = _context2.next) {
+                switch (_context3.prev = _context3.next) {
                   case 0:
-                    _context2.next = 2;
-                    return topo.save();
-
-                  case 2:
                     topo.exitEditMode();
 
-                  case 3:
+                  case 1:
                   case "end":
-                    return _context2.stop();
+                    return _context3.stop();
                 }
               }
-            }, _callee2);
+            }, _callee3);
           }));
 
           function onCancel() {
@@ -87,7 +128,10 @@ export default function TopoEditorTitleBar(props) {
           return onCancel;
         }(),
         cancelProps: {
-          children: '保存'
+          children: "退出"
+        },
+        okProps: {
+          children: "保存并退出"
         }
       });
     }