1 files changed, 43 insertions, 105 deletions

diff --git a/js/controllers/selection-controller.js b/js/controllers/selection-controller.js
index c713b6e5..5665b09c 100755
--- a/js/controllers/selection-controller.js
+++ b/js/controllers/selection-controller.js
@@ -44,13 +44,11 @@ exports.SelectionController = Montage.create(Component, {
         value: function() {
             this.eventManager.addEventListener("openDocument", this, false);
             this.eventManager.addEventListener("elementAdded", this, false);
-            this.eventManager.addEventListener("elementDeleted", this, false);
+            this.eventManager.addEventListener("elementsRemoved", this, false);
+            this.eventManager.addEventListener("elementReplaced", this, false);
             this.eventManager.addEventListener("selectAll", this, false);
-            this.eventManager.addEventListener("deleteSelection", this, false);
             this.eventManager.addEventListener("switchDocument", this, false);
             this.eventManager.addEventListener("closeDocument", this, false);
-//            defaultEventManager.addEventListener( "undo", this, false);
-//            defaultEventManager.addEventListener( "redo", this, false);
         }
     },
 
@@ -66,39 +64,25 @@ exports.SelectionController = Montage.create(Component, {
     
     initWithDocument: {
         value: function(currentSelectionArray) {
-            this._selectedItems = [];
             this._isDocument = true;
 
             if(currentSelectionArray) {
                 if(currentSelectionArray.length >= 1) {
-                    this._selectedItems = currentSelectionArray;
                     this._isDocument = false;
 
-
-
                     this.application.ninja.selectedElements = currentSelectionArray;
                     NJevent("selectionChange", {"elements": this.application.ninja.selectedElements, "isDocument": this._isDocument});
-
-
-
                 }
             }
 
-            //
             this._selectionContainer = this.application.ninja.currentSelectedContainer;
-
         }
     },
 
-     handleSwitchDocument: {
+    handleSwitchDocument: {
         value: function() {
             if(this.application.ninja.documentController.activeDocument.currentView === "design"){
-                this._selectedItems = this.application.ninja.selectedElements.slice(0);
-                if(this._selectedItems.length === 0 ){
-                    this._isDocument = true;
-                }else{
-                    this._isDocument = false;
-                }
+                this._isDocument = this.application.ninja.selectedElements.length === 0;
                 NJevent("selectionChange", {"elements": this.application.ninja.selectedElements, "isDocument": this._isDocument} );
             }
         }
@@ -110,16 +94,22 @@ exports.SelectionController = Montage.create(Component, {
         }
     },
 
-    handleElementDeleted: {
+    handleElementsRemoved: {
         value: function(event) {
             if(!this._isDocument) {
-                if(this.findSelectedElement(event.detail) !== -1) {
-                    this.executeSelectElement();
-                }
+                this.application.ninja.selectedElements = [];
+                this._isDocument = true;
+                NJevent("selectionChange", {"elements": this.application.ninja.selectedElements, "isDocument": this._isDocument});
             }
         }
     },
 
+    handleElementReplaced: {
+        value: function(event) {
+            this.application.ninja.selectedElements[this.application.ninja.selectedElements.indexOf(event.detail.data.oldChild)] = event.detail.data.newChild;
+        }
+    },
+
     handleSelectAll: {
         value: function(event) {
             var selected = [], childNodes = [];
@@ -136,23 +126,19 @@ exports.SelectionController = Montage.create(Component, {
         }
     },
 
-    handleDeleteSelection: {
-        value: function(event) {
-            this.application.ninja.selectedElements = [];
-            this._isDocument = true;
-            NJevent("selectionChange", {"elements": this.application.ninja.selectedElements, "isDocument": this._isDocument});
-        }
-    },
-
     /**
      * Select Element. This function will not check that element, it will simply add it to the selection array.
      */
     executeSelectElement: {
-        value: function(item) {
+        value: function(element) {
             this.application.ninja.selectedElements = [];
 
-            if(item) {
-                this.application.ninja.selectedElements.push({_element: item, uuid: item.uuid});
+            if(element) {
+                if(Array.isArray(element)) {
+                    this.application.ninja.selectedElements = Array.prototype.slice.call(element, 0);
+                } else {
+                    this.application.ninja.selectedElements.push(element);
+                }
                 this._isDocument = false;
             } else {
                 this._isDocument = true;
@@ -165,23 +151,20 @@ exports.SelectionController = Montage.create(Component, {
     },
 
     selectElement: {
-        value: function(item) {
+        value: function(element) {
 
-            if(this.findSelectedElement(item) === -1) {
+            if(this.findSelectedElement(element) === -1) {
 
-                if(this.application.ninja.currentDocument.inExclusion(item) !== -1){
+                if(this.application.ninja.currentDocument.inExclusion(element) !== -1){
                     if(this.isDocument) return;     // If the stage is already selected do nothing.
-                    this.executeSelectElement();    // Else execute selection with no item
+                    this.executeSelectElement();    // Else execute selection with no element
                 } else {
-
-//                    if(item.parentNode.id === "UserContent") {
-                    if(item.parentNode.uuid === this.selectionContainer.uuid) {
-                        this.executeSelectElement(item);
+                    if(element.parentNode.uuid === this.selectionContainer.uuid) {
+                        this.executeSelectElement(element);
                     } else {
-                        var outerElement = item.parentNode;
+                        var outerElement = element.parentNode;
 
                         while(outerElement.parentNode && outerElement.parentNode.uuid !== this.selectionContainer.uuid) {
-                        //while(outerElement.parentNode && outerElement.parentNode.id !== "UserContent") {
                             // If element is higher up than current container then return
                             if(outerElement.id === "UserContent") return;
                             // else keep going up the chain
@@ -196,13 +179,13 @@ exports.SelectionController = Montage.create(Component, {
     },
 
     selectElements: {
-        value: function(items) {
-            if(items && items.length > 0) {
+        value: function(elements) {
+            if(elements && elements.length > 0) {
                 var that = this;
                 this.application.ninja.selectedElements = [];
 
-                items.forEach(function(item) {
-                    that.application.ninja.selectedElements.push({_element: item, uuid: item.uuid});
+                elements.forEach(function(element) {
+                    that.application.ninja.selectedElements.push(element);
                     that._isDocument = false;
                 });
 
@@ -212,22 +195,22 @@ exports.SelectionController = Montage.create(Component, {
     },
 
     shiftSelectElement: {
-        value: function(item) {
-            if(this.application.ninja.currentDocument.inExclusion(item) !== -1) return;
+        value: function(element) {
+            if(this.application.ninja.currentDocument.inExclusion(element) !== -1) return;
 
-            (this.findSelectedElement(item) !== -1 ) ? this.removeElement(item) : this.insertElement(item);
+            (this.findSelectedElement(element) !== -1 ) ? this.removeElement(element) : this.insertElement(element);
         }
     },
 
     insertElement: {
-        value: function(item) {
-            if(item) {
+        value: function(element) {
+            if(element) {
                 if(this._isDocument) {