6 files changed, 189 insertions, 30 deletions

diff --git a/js/components/prompt.reel/prompt.js b/js/components/prompt.reel/prompt.js
new file mode 100644
index 00000000..652d7b7e
--- /dev/null
+++ b/js/components/prompt.reel/prompt.js
@@ -0,0 +1,73 @@
+/* <copyright>
+This file contains proprietary software owned by Motorola Mobility, Inc.<br/>
+No rights, expressed or implied, whatsoever to this software are provided by Motorola Mobility, Inc. hereunder.<br/>
+(c) Copyright 2011 Motorola Mobility, Inc.  All Rights Reserved.
+</copyright> */
+
+////////////////////////////////////////////////////////////////////////
+//
+var Montage =   require("montage/core/core").Montage,
+        Component = require("montage/ui/component").Component;
+////////////////////////////////////////////////////////////////////////
+//      
+exports.NinjaPrompt = Montage.create(Component, {
+        ////////////////////////////////////////////////////////////////////
+        //TODO: This should have an UI template eventually
+        hasTemplate: {
+        value: false
+    },
+    ////////////////////////////////////////////////////////////////////
+        //Type of prompt window (should be confirm, prompt, alert, or input)
+        _type: {
+                value: null
+        },
+        ////////////////////////////////////////////////////////////////////
+        //
+        _params: {
+                value: null
+        },
+        ////////////////////////////////////////////////////////////////////
+        //
+        _callback: {
+                value: null
+        },
+    ////////////////////////////////////////////////////////////////////
+        //
+        initialize: {
+                value: function (type, params, callback) {
+                        //
+                        this._type = type.toLowerCase();
+                        this._params = params;
+                        this._callback = callback;
+                }
+        },
+        ////////////////////////////////////////////////////////////////////
+        //
+        show: {
+                value: function () {
+                        //
+                        var input;
+                        //
+                        switch (this._type) {
+                                case 'confirm':
+                                        input = confirm(this._params.message);
+                                        if (this._callback) this._callback(input);
+                                        break;
+                                default:
+                                        //TODO: Add support for other standard box types
+                                        break;
+                        }
+                }
+        },
+        ////////////////////////////////////////////////////////////////////
+        //This is for later, need to hide if need (overwrite)
+        hide: {
+                value: function () {
+                        //TODO: Add support as real UI component
+                }
+        }
+        ////////////////////////////////////////////////////////////////////
+        ////////////////////////////////////////////////////////////////////
+});
+////////////////////////////////////////////////////////////////////////
+////////////////////////////////////////////////////////////////////////
\ No newline at end of file
diff --git a/js/document/document-html.js b/js/document/document-html.js
index 15f88d09..aded9241 100755
--- a/js/document/document-html.js
+++ b/js/document/document-html.js
@@ -121,10 +121,8 @@ exports.HtmlDocument = Montage.create(Component, {
         //
         closeDocument: {
                 value: function (context, callback) {
-                        //Closing document and getting outcome
-                        var closed = this.model.close(null);
-                        //Making callback if specified
-                        if (callback) callback.call(context, this);
+                        //Closing document (sending null to close all views)
+                        this.model.close(null, function () {if (callback) callback.call(context, this);}.bind(this));
                 }
         },
     ////////////////////////////////////////////////////////////////////
diff --git a/js/document/models/base.js b/js/document/models/base.js
index 1307e0c0..8ff52132 100755
--- a/js/document/models/base.js
+++ b/js/document/models/base.js
@@ -6,8 +6,9 @@ No rights, expressed or implied, whatsoever to this software are provided by Mot
 
 ////////////////////////////////////////////////////////////////////////
 //
-var Montage =   require("montage/core/core").Montage,
-        Component = require("montage/ui/component").Component;
+var Montage =           require("montage/core/core").Montage,
+        Component =     require("montage/ui/component").Component,
+        NinjaPrompt =   require("js/components/prompt.reel").NinjaPrompt;
 ////////////////////////////////////////////////////////////////////////
 //      
 exports.BaseDocumentModel = Montage.create(Component, {
@@ -267,30 +268,59 @@ exports.BaseDocumentModel = Montage.create(Component, {
                         if (this.callback) this.callback(result);
                 }
         },
+        ////////////////////////////////////////////////////////////////////
+        //
+        handleSavePrompt: {
+                value: function (continueToClose, callback) {
+                        //TODO: Perhaps add logic to save the file is the user wants
+                        if (continueToClose) {
+                                if (callback) callback();
+                        } else {
+                                //User canceled
+                                //this.saveAll(null, callback);
+                        }
+                }
+        },
     ////////////////////////////////////////////////////////////////////
         //TODO: Implement better logic to include different views on single document
         close: {
         value: function (view, callback) {
-                //Outcome of close (pending on save logic)
-                var success;
-                //
+                //Checking if files needs to be saved to avoid losing data
                 if (this.needsSave) {
-                        //TODO: Prompt user to save or lose data
+                        //Creating prompt to ask user to save the file
+                        var prompt = NinjaPrompt.create();
+                        prompt.initialize('confirm', {message: 'Do you want to save the changes you made in the document '+this.file.name+'?\n\nYour changes will be lost if you do not save them.'}, function (result){this.handleSavePrompt(result, callback);}.bind(this));
+                        //Showing the prompt, it will make callback with user input
+                        prompt.show();
                 } else {
-                        //Close file
-                        success = true;
-                }
-                //Checking for view mode to close
-                if (this.views.design && (!view || view === 'design')) {
-                        //TODO: Create a destroy method, this is messy
-                        this.views.design.pauseAndStopVideos();
-                        this.parentContainer.removeChild(this.views.design.iframe);
-                        this.views.design = null;
+                        //TODO: Add support for other views
+                        if (!view || view === 'design') {
+                                this.closeView('design');
+                        }
+                        //Making callback
+                        if (callback) callback();
                 }
-                //Returning result of operation
-                return success;
+                
         }
-    }
+    },
+    ////////////////////////////////////////////////////////////////////
+        //
+        closeView: {
+                value: function (view) {
+                        //Checking for view mode to close
+                switch (view.toLowerCase()) {
+                        case 'design':
+                                //TODO: Make into clean method in the design view
+                                this.views.design.pauseAndStopVideos();
+                                this.parentContainer.removeChild(this.views.design.iframe);
+                                this.views.design = null;
+                                break;
+                        default:
+                                //TODO: Error?
+                                break;
+                }
+                }
+        }
         ////////////////////////////////////////////////////////////////////
         ////////////////////////////////////////////////////////////////////
 });
diff --git a/js/models/element-model.js b/js/models/element-model.js
index 76f15b21..e5a0f33a 100755
--- a/js/models/element-model.js
+++ b/js/models/element-model.js
@@ -91,6 +91,8 @@ var elementNameToController = function(name) {
         return "block";
     } else if(name === "img") {
         return "image";
+    } else if(name === "embed") {
+        return "image";
     } else {
         return name;
     }
@@ -174,6 +176,8 @@ var elmo = exports.ElementModel = Montage.create(Montage, {
                 return "block";
             } else if(name === "img") {
                 return "image";
+            } else if(name === "embed") {
+                return "image";
             } else {
                 return name;
             }
diff --git a/js/ninja.reel/ninja.js b/js/ninja.reel/ninja.js
index b57aecd2..2d43af7a 100755
--- a/js/ninja.reel/ninja.js
+++ b/js/ninja.reel/ninja.js
@@ -169,6 +169,10 @@ exports.Ninja = Montage.create(Component, {
             this.setupGlobalHelpers();
 
             window.addEventListener("resize", this, false);
+            //Prompting the user to make sure data was saved before closing Ninja
+            window.onbeforeunload = function () {
+                return 'Are you sure you want to close Ninja? Any unsaved data will be lost.';
+            };
 
             this.eventManager.addEventListener("selectTool", this, false);