From abe76917301d07e41316992b93742555303cc89f Mon Sep 17 00:00:00 2001
From: Bruno Alves de Faria <brunoalvesdefaria@gmail.com>
Date: Fri, 26 Jan 2018 07:47:45 -0200
Subject: [PATCH] Adding hooks to unsavedChanges mechanism

---
 .../dialog/unsavedChangesDialog.html          |  4 +-
 .../ui/unsavedChanges/unsavedChanges.js       | 65 +++++++++++++++++++
 2 files changed, 67 insertions(+), 2 deletions(-)

diff --git a/Packages/ohif-core/client/components/bootstrap/dialog/unsavedChangesDialog.html b/Packages/ohif-core/client/components/bootstrap/dialog/unsavedChangesDialog.html
index 92531d7c4..c6f39cfd9 100644
--- a/Packages/ohif-core/client/components/bootstrap/dialog/unsavedChangesDialog.html
+++ b/Packages/ohif-core/client/components/bootstrap/dialog/unsavedChangesDialog.html
@@ -1,12 +1,12 @@
 <template name="unsavedChangesDialog">
-    <div class="modal unsavedChangesDialog fade" tabindex="-1" role="dialog">
+    <div class="modal unsavedChangesDialog fade {{this.class}}" tabindex="-1" role="dialog">
         <div class="modal-dialog" role="document">
             <div class="modal-content">
                 <div class="modal-header">
                     <button class="close" aria-label="Close" data-choice="abort-action" data-dismiss="modal">
                         <span aria-hidden="true">&times;</span>
                     </button>
-                    <h4 class="modal-title">You have unsaved changes!</h4>
+                    <h4 class="modal-title">{{choose this.message 'You have unsaved changes!'}}</h4>
                 </div>
                 <div class="modal-footer">
 
diff --git a/Packages/ohif-core/client/ui/unsavedChanges/unsavedChanges.js b/Packages/ohif-core/client/ui/unsavedChanges/unsavedChanges.js
index f7bbb4269..237e7babe 100644
--- a/Packages/ohif-core/client/ui/unsavedChanges/unsavedChanges.js
+++ b/Packages/ohif-core/client/ui/unsavedChanges/unsavedChanges.js
@@ -318,6 +318,8 @@ export const unsavedChanges = {
 
     observer: new Tracker.Dependency(),
 
+    hooks: new Map(),
+
     /**
      * Register a reactive dependency on every change any path suffers
      */
@@ -483,6 +485,69 @@ export const unsavedChanges = {
             callback.call(null, hasChanges, null);
         }
 
+    },
+
+    addHook(saveCallback, options={}) {
+        _.defaults(options, {
+            path: '*',
+            message: 'There are unsaved changes'
+        });
+
+        this.hooks.set(saveCallback, options);
+    },
+
+    removeHook(saveCallback) {
+        this.hooks.delete(saveCallback);
+    },
+
+    confirmNavigation(navigateCallback, event) {
+        let dialogPresented = false;
+        Array.from(this.hooks.keys()).every(saveCallback => {
+            const options = this.hooks.get(saveCallback);
+            const probe = this.probe(options.path, true);
+            if (!probe) return true;
+
+            const dialogOptions = Object.assign({ class: 'themed' }, options);
+            if (event) {
+                dialogOptions.position = {
+                    x: event.clientX + 15,
+                    y: event.clientY + 15
+                };
+            }
+
+            OHIF.ui.unsavedChanges.presentProactiveDialog(options.path, (hasChanges, userChoice) => {
+                if (!hasChanges) return;
+
+                const clear = () => this.clear(options.path, true);
+                switch (userChoice) {
+                    case 'abort-action':
+                        return;
+                    case 'save-changes':
+                        const result = saveCallback();
+                        if (result instanceof Promise) {
+                            return result.then(() => {
+                                clear();
+                                this.confirmNavigation(navigateCallback, event);
+                            });
+                        }
+
+                        clear();
+                        return this.confirmNavigation(navigateCallback, event);
+                    case 'abandon-changes':
+                        clear();
+                        break;
+                }
+
+                navigateCallback();
+            }, dialogOptions);
+
+            dialogPresented = true;
+            return false;
+        });
+
+        if (!dialogPresented) {
+            navigateCallback();
+        }
     }
 
 };