2 files changed, 15 insertions, 15 deletions

diff --git a/js/controllers/document-controller.js b/js/controllers/document-controller.js
index a90375af..a3ebac24 100755
--- a/js/controllers/document-controller.js
+++ b/js/controllers/document-controller.js
@@ -174,9 +174,9 @@ exports.DocumentController = Montage.create(Component, {
     handleExecuteSaveAll: {
         value: function(event) {
            //
-                if((typeof this.activeDocument !== "undefined") && this.application.ninja.coreIoApi.cloudAvailable()){
+                if((typeof this.currentDocument !== "undefined") && this.application.ninja.coreIoApi.cloudAvailable()){
                         //
-                        this.activeDocument.model.saveAll();
+                        this.currentDocument.model.saveAll();
                 } else {
                         //TODO: Add error handling
                 }
@@ -186,9 +186,9 @@ exports.DocumentController = Montage.create(Component, {
     handleExecuteSaveAs: {
         value: function(event) {
             var saveAsSettings = event._event.settings || {};
-            if((typeof this.activeDocument !== "undefined") && this.application.ninja.coreIoApi.cloudAvailable()){
-                saveAsSettings.fileName = this.activeDocument.model.file.name;
-                saveAsSettings.folderUri = this.activeDocument.model.file.uri.substring(0, this.activeDocument.model.file.uri.lastIndexOf("/"));
+            if((typeof this.currentDocument !== "undefined") && this.application.ninja.coreIoApi.cloudAvailable()){
+                saveAsSettings.fileName = this.currentDocument.model.file.name;
+                saveAsSettings.folderUri = this.currentDocument.model.file.uri.substring(0, this.currentDocument.model.file.uri.lastIndexOf("/"));
                 saveAsSettings.callback = this.saveAsCallback.bind(this);
                 this.application.ninja.newFileController.showSaveAsDialog(saveAsSettings);
             }
@@ -204,9 +204,9 @@ exports.DocumentController = Montage.create(Component, {
     //TODO: Is this used, should be cleaned up
     handleExecuteFileCloseAll:{
                 value: function(event) {
-                        if(this.activeDocument && this.application.ninja.coreIoApi.cloudAvailable()){
-                                while(this._documents.length > 0){
-                                        this.closeDocument(this._documents[this._documents.length -1].uuid);
+                        if(this.currentDocument && this.application.ninja.coreIoApi.cloudAvailable()){
+                                while(this.currentDocument.length > 0){
+                                        this.closeDocument(this.currentDocument[this.currentDocument.length -1].uuid);
                                 }
                         }
                 }
@@ -291,19 +291,19 @@ exports.DocumentController = Montage.create(Component, {
         value:function(saveAsDetails){
             var fileUri = null, filename = saveAsDetails.filename, destination = saveAsDetails.destination;
             //update document metadata
-            this.activeDocument.name = ""+filename;
+            this.currentDocument.model.file.name = ""+filename;
             //prepare new file uri
             if(destination && (destination.charAt(destination.length -1) !== "/")){
                 destination = destination + "/";
             }
             fileUri = destination+filename;
 
-            this.activeDocument.uri = fileUri;
+            this.currentDocument.model.file.uri = fileUri;
             //save a new file
             //use the ioMediator.fileSaveAll when implemented
-            this.activeDocument.model.file.name = filename;
-            this.activeDocument.model.file.uri = fileUri;
-            this.activeDocument.model.save();
+            this.currentDocument.model.file.name = filename;
+            this.currentDocument.model.file.uri = fileUri;
+            this.currentDocument.model.save();
         }
     },
 
diff --git a/js/controllers/elements/component-controller.js b/js/controllers/elements/component-controller.js
index 5b0aaeac..dd0766df 100755
--- a/js/controllers/elements/component-controller.js
+++ b/js/controllers/elements/component-controller.js
@@ -11,7 +11,7 @@ exports.ComponentController = Montage.create(ElementController, {
 
     getProperty: {
         value: function(el, prop) {
-            var component = el.controller || this.application.ninja.currentDocument.model.getComponentFromElement(el);
+            var component = el.controller;
 
             switch(prop) {
                 case "id":
@@ -34,7 +34,7 @@ exports.ComponentController = Montage.create(ElementController, {
 
     setProperty: {
         value: function(el, p, value) {
-            var component = el.controller || this.application.ninja.currentDocument.model.getComponentFromElement(el);
+            var component = el.controller;
 
             switch(p) {
                 case "id":