Cleaning up mediator

1 files changed, 13 insertions, 3 deletions

diff --git a/js/controllers/document-controller.js b/js/controllers/document-controller.js
index 84b994ce..7d38c600 100755
--- a/js/controllers/document-controller.js
+++ b/js/controllers/document-controller.js
@@ -104,16 +104,26 @@ DocumentController = exports.DocumentController = Montage.create(Component, {
     handleExecuteSave: {
         value: function(event) {
                 //Text and HTML document classes should return the same save object for fileSave
-                this.application.ninja.ioMediator.fileSave(this.activeDocument.save(), this.clearDocumentDirtyFlag.bind(this));
+                this.application.ninja.ioMediator.fileSave(this.activeDocument.save(), this.fileSaveResult.bind(this));
                 }
     },
-
+    ////////////////////////////////////////////////////////////////////
+    //
+    fileSaveResult: {
+        value: function (result) {
+                if(result.status === 204){
+                this.clearDocumentDirtyFlag();
+            }
+        }
+    },
+    ////////////////////////////////////////////////////////////////////
+        
+        
     clearDocumentDirtyFlag:{
         value: function(){
             this.activeDocument.dirtyFlag = false;
         }
     },
-        ////////////////////////////////////////////////////////////////////
         
         
     createNewFile:{