feat: undo redo interceptor and change helper

Author: abose

src/editor/EditorCommandHandlers.js

@@ -1190,11 +1190,45 @@ define(function (require, exports, module) {
         return result.promise();
     }
 
+    let _undoInterceptor = null;
+    let _redoInterceptor = null;
+
+    /**
+     * Use require("editor/EditorHelper/ChangeHelper") to intercept. this for internal use only
+     * @private
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
+     */
+    function _setUndoInterceptor(interceptor) {
+        _undoInterceptor = interceptor;
+    }
+
+    /**
+     * Use require("editor/EditorHelper/ChangeHelper") to intercept. this for internal use only
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
+     */
+    function _setRedoInterceptor(interceptor) {
+        _redoInterceptor = interceptor;
+    }
+
     function handleUndo() {
+        if(_undoInterceptor){
+            const focusedEditor = EditorManager.getFocusedEditor();
+            const codeMirror = focusedEditor && focusedEditor._codeMirror;
+            if(_undoInterceptor(focusedEditor, codeMirror, null)){
+                return;
+            }
+        }
         return handleUndoRedo("undo");
     }
 
     function handleRedo() {
+        if(_redoInterceptor){
+            const focusedEditor = EditorManager.getFocusedEditor();
+            const codeMirror = focusedEditor && focusedEditor._codeMirror;
+            if(_redoInterceptor(focusedEditor, codeMirror, null)){
+                return;
+            }
+        }
         return handleUndoRedo("redo");
     }
 
@@ -1273,4 +1307,7 @@ define(function (require, exports, module) {
     CommandManager.register(Strings.CMD_COPY, Commands.EDIT_COPY, _execCommandCopy);
     CommandManager.register(Strings.CMD_PASTE, Commands.EDIT_PASTE, _execCommandPaste);
     CommandManager.register(Strings.CMD_SELECT_ALL, Commands.EDIT_SELECT_ALL, _handleSelectAll);
+
+    exports._setUndoInterceptor = _setUndoInterceptor;
+    exports._setRedoInterceptor = _setRedoInterceptor;
 });

src/editor/EditorHelper/ChangeHelper.js

@@ -31,7 +31,8 @@ define(function (require, exports, module) {
     let _keyEventInterceptor = null;
 
     const CodeMirror = require("thirdparty/CodeMirror/lib/codemirror"),
-        Menus = require("command/Menus");
+        Menus = require("command/Menus"),
+        EditorCommandHandlers = require("editor/EditorCommandHandlers");
 
     function _applyChanges(changeList) {
         // eslint-disable-next-line no-invalid-this
@@ -314,43 +315,57 @@ define(function (require, exports, module) {
         Editor.prototype._dontDismissPopupOnScroll = _dontDismissPopupOnScroll;
     }
 
+    /**
+     * Sets the undo interceptor function in before it goes to codemirror
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
+     */
+    function setUndoInterceptor(interceptor) {
+        EditorCommandHandlers._setUndoInterceptor(interceptor);
+    }
+
+    /**
+     * Sets the redo interceptor function in before it goes to codemirror
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
+     */
+    function setRedoInterceptor(interceptor) {
+        EditorCommandHandlers._setRedoInterceptor(interceptor);
+    }
+
     /**
      * Sets the cut interceptor function in codemirror
-     * @param {Function} interceptor - Function(editor, cm, event) that returns true to
-     preventDefault
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
      */
     function setCutInterceptor(interceptor) {
         _cutInterceptor = interceptor;
     }
 
     /**
      * Sets the copy interceptor function in codemirror
-     * @param {Function} interceptor - Function(editor, cm, event) that returns true to
-     preventDefault
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
      */
     function setCopyInterceptor(interceptor) {
         _copyInterceptor = interceptor;
     }
 
     /**
      * Sets the paste interceptor function in codemirror
-     * @param {Function} interceptor - Function(editor, cm, event) that returns true to
-     preventDefault
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
      */
     function setPasteInterceptor(interceptor) {
         _pasteInterceptor = interceptor;
     }
 
     /**
      * Sets the key down/up/press interceptor function in codemirror
-     * @param {Function} interceptor - Function(editor, cm, event) that returns true to
-     preventDefault
+     * @param {Function} interceptor - Function(editor, cm, event) that returns true to preventDefault
      */
     function setKeyEventInterceptor(interceptor) {
         _keyEventInterceptor = interceptor;
     }
 
     exports.addHelpers =addHelpers;
+    exports.setUndoInterceptor = setUndoInterceptor;
+    exports.setRedoInterceptor = setRedoInterceptor;
     exports.setCutInterceptor = setCutInterceptor;
     exports.setCopyInterceptor = setCopyInterceptor;
     exports.setPasteInterceptor = setPasteInterceptor;