3 files changed, 60 insertions, 27 deletions

diff --git a/js/controllers/code-editor-controller.js b/js/controllers/code-editor-controller.js
index d0b1f179..d292f838 100644
--- a/js/controllers/code-editor-controller.js
+++ b/js/controllers/code-editor-controller.js
@@ -32,6 +32,7 @@ exports.CodeEditorController = Montage.create(Component, {
             if(!value) {
 
             } else if(this._currentDocument.currentView === "code") {
+                this._currentDocument.model.views.code.editor.focus();
                 this.applySettings();
             }
         }
diff --git a/js/controllers/document-controller.js b/js/controllers/document-controller.js
index e059e7e1..f85e2f97 100755
--- a/js/controllers/document-controller.js
+++ b/js/controllers/document-controller.js
@@ -54,6 +54,7 @@ exports.DocumentController = Montage.create(Component, {
                 document.getElementById("codeViewContainer").style.display = "block";
             } else if(this._currentDocument.currentView === "design") {
                 this._currentDocument.model.currentView.show();
+                this._currentDocument.model.views.design._liveNodeList = this._currentDocument.model.documentRoot.getElementsByTagName('*');
             } else {
                 document.getElementById("iframeContainer").style.display = "none";
                 this._currentDocument.model.currentView.show();
@@ -149,10 +150,10 @@ exports.DocumentController = Montage.create(Component, {
     handleExecuteSave: {
         value: function(event) {
                 //
-                if((typeof this.activeDocument !== "undefined") && this.application.ninja.coreIoApi.cloudAvailable()){
+                if((typeof this.currentDocument !== "undefined") && this.application.ninja.coreIoApi.cloudAvailable()){
                         //Currently we don't need a callback handler
                         //this.activeDocument.model.save(this.saveExecuted.bind(this));
-                        this.activeDocument.model.save();
+                        this.currentDocument.model.save();
                 } else {
                         //Error: cloud not available and/or no active document
                 }
@@ -193,16 +194,13 @@ exports.DocumentController = Montage.create(Component, {
     ////////////////////////////////////////////////////////////////////
     handleExecuteFileClose:{
         value: function(event) {
-                if (this.activeDocument) {
-                this.closeFile(this.activeDocument);
-                }
+            this.closeFile(this.currentDocument);
         }
     },
     ////////////////////////////////////////////////////////////////////
     //TODO: Is this used, should be cleaned up
     handleExecuteFileCloseAll:{
                 value: function(event) {
-                        var i=0;//TODO: who is using this??
                         if(this.activeDocument && this.application.ninja.coreIoApi.cloudAvailable()){
                                 while(this._documents.length > 0){
                                         this.closeDocument(this._documents[this._documents.length -1].uuid);
@@ -355,14 +353,8 @@ exports.DocumentController = Montage.create(Component, {
             if(currentDocument) {
                 if(currentDocument.currentView === "design") {
                     currentDocument.serializeDocument();
-                    this.application.ninja._currentSelectedContainer = this.application.ninja.selectionController._selectionContainer = null;
-                    currentDocument.model.views.design.propertiesPanel.clear();
                 }
 
-                currentDocument.model.isActive = false;
-
-                currentDocument.model.currentView.hide();
-
                 if(currentDocument.model.parentContainer !== newDocument.model.parentContainer) {
                     currentDocument.model.parentContainer.style["display"] = "none";
                     newDocument.model.parentContainer.style["display"] = "block";
@@ -383,13 +375,9 @@ exports.DocumentController = Montage.create(Component, {
                 this.application.ninja.stage.hideRulers();
             }
 
-//            this.application.ninja.stage.clearAllCanvas();
-
             if(didCreate) {
-//                newDocument.model.currentView.show();
-
                 if(newDocument.currentView === "design") {
-                    NJevent("onOpenDocument", newDocument);
+
                 } else {
                     newDocument.model.parentContainer.style["display"] = "block";
                 }
diff --git a/js/controllers/selection-controller.js b/js/controllers/selection-controller.js
index 848355da..a219c597 100755
--- a/js/controllers/selection-controller.js
+++ b/js/controllers/selection-controller.js
@@ -20,6 +20,60 @@ exports.SelectionController = Montage.create(Component, {
         }
     },
 
+    _currentDocument: {
+            value : null
+    },
+
+    currentDocument : {
+        get : function() {
+            return this._currentDocument;
+        },
+        set : function(value) {
+            if (value === this._currentDocument) {
+                return;
+            }
+
+            if(this._currentDocument) {
+                this._currentDocument.model.selection = this.application.ninja.selectedElements;
+                this._currentDocument.model.selectionContainer = this.application.ninja._currentSelectedContainer;
+            }
+
+            this._currentDocument = value;
+
+            if(!value) {
+            } else if(this._currentDocument.currentView === "design") {
+            } else {
+            }
+
+        }
+    },
+
+    _selectedElements: {
+        value: null
+    },
+
+    selectedElements: {
+        get: function() {
+            return this._selectedElements;
+        },
+        set: function(value) {
+            if(value) {
+                this._selectedElements = value;
+
+                this.application.ninja.selectedElements = this._selectedElements;
+                this.application.ninja._currentSelectedContainer = this._selectionContainer = this.application.ninja.currentDocument.model.documentRoot;
+
+                if(this._selectedElements.length === 0) {
+                    this.executeSelectElement();
+                } else {
+                    this.executeSelectElement(this._selectedElements);
+                }
+
+
+            }
+        }
+    },
+
     // Bound property to the ninja currentSelectedContainer
     _selectionContainer: {
         value: null
@@ -44,16 +98,6 @@ exports.SelectionController = Montage.create(Component, {
             this.eventManager.addEventListener("elementsRemoved", this, false);
             this.eventManager.addEventListener("elementReplaced", this, false);
             this.eventManager.addEventListener("selectAll", this, false);
-            this.eventManager.addEventListener("switchDocument", this, false);
-        }
-    },
-
-    handleSwitchDocument: {
-        value: function() {
-//            if(this.application.ninja.documentController.activeDocument.currentView === "design"){
-                this._isDocument = this.application.ninja.selectedElements.length === 0;
-                NJevent("selectionChange", {"elements": this.application.ninja.selectedElements, "isDocument": this._isDocument} );
-//            }
         }
     },