editor: plugins/core - support for undo/redo selection restore in most actions

[fnpeditor.git] / src / editor / modules / documentCanvas / documentCanvas.js

diff --git a/src/editor/modules/documentCanvas/documentCanvas.js b/src/editor/modules/documentCanvas/documentCanvas.js
index 98c09d2..8742aef 100644 (file)
--- a/src/editor/modules/documentCanvas/documentCanvas.js
+++ b/src/editor/modules/documentCanvas/documentCanvas.js
@@ -84,18 +84,9 @@ return function(sandbox) {
         onAfterActionExecuted: function(action, ret) {
             if(ret && ret.isValid()) {
                 logger.debug('The action returned a valid fragment');
-                if(ret instanceof canvas.wlxmlDocument.RangeFragment) {
-                    canvas.setCurrentElement(ret.endNode, {caretTo: ret.endOffset});
-                } else if(ret instanceof canvas.wlxmlDocument.NodeFragment) {
-                    var params = {
-                        caretTo: ret instanceof canvas.wlxmlDocument.CaretFragment ? ret.offset : 'start'
-                    };
-                    canvas.setCurrentElement(ret.node, params);
-                } else {
-                    logger.debug('Fragment not supported');
-                }
-                return;
+                return canvas.select(ret);
             }
+
             logger.debug('No valid fragment returned from the action');
 
             (actionHandlers[action.getPluginName()] || []).forEach(function(handler) {