1 files changed, 22 insertions, 0 deletions

diff --git a/js/controllers/selection-controller.js b/js/controllers/selection-controller.js
index d69b53e0..c4623d3f 100755
--- a/js/controllers/selection-controller.js
+++ b/js/controllers/selection-controller.js
@@ -47,6 +47,7 @@ exports.SelectionController = Montage.create(Component, {
             this.eventManager.addEventListener("elementDeleted", this, false);
             this.eventManager.addEventListener("selectAll", this, false);
             this.eventManager.addEventListener("deleteSelection", this, false);
+            this.eventManager.addEventListener("switchDocument", this, false);
 //            defaultEventManager.addEventListener( "undo", this, false);
 //            defaultEventManager.addEventListener( "redo", this, false);
         }
@@ -58,6 +59,7 @@ exports.SelectionController = Montage.create(Component, {
     handleOpenDocument: {
         value: function() {
             // Handle initializing the selection array here.
+            this.initWithDocument([]);
         }
     },
     
@@ -70,6 +72,14 @@ exports.SelectionController = Montage.create(Component, {
                 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} );
+
+
+
                 }
             }
 
@@ -79,6 +89,18 @@ exports.SelectionController = Montage.create(Component, {
         }
     },
 
+    handleSwitchDocument: {
+        value: function() {
+            this._selectedItems = this.application.ninja.selectedElements.slice(0);
+            if(this._selectedItems.length === 0 ){
+                this._isDocument = true;
+            }else{
+                this._isDocument = false;
+            }
+            NJevent("selectionChange", {"elements": this.application.ninja.selectedElements, "isDocument": this._isDocument} );
+        }
+    },
+
     handleElementAdded: {
         value: function(event) {
             this.executeSelectElement(event.detail);