From 51d6a64c4594d4ec311d51c8f4c4e4c915dbecfb Mon Sep 17 00:00:00 2001
From: Evren Ozkan <evren217@gmail.com>
Date: Wed, 13 Dec 2017 19:41:47 -0500
Subject: [PATCH] Fix the issues with customized store and retrieve functions
 for hot keys and window/level presets

---
 Packages/ohif-hotkeys/client/classes/HotkeysManager.js |  2 +-
 Packages/ohif-viewerbase/client/lib/WLPresets.js       | 10 +++++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/Packages/ohif-hotkeys/client/classes/HotkeysManager.js b/Packages/ohif-hotkeys/client/classes/HotkeysManager.js
index 3383c1235..ffeb4ffca 100644
--- a/Packages/ohif-hotkeys/client/classes/HotkeysManager.js
+++ b/Packages/ohif-hotkeys/client/classes/HotkeysManager.js
@@ -33,7 +33,7 @@ export class HotkeysManager {
         const storageKey = `hotkeysDefinitions.${contextName}`;
         return new Promise((resolve, reject) => {
             if (this.storeFunction) {
-                this.storeFunction(contextName, definitions).then(resolve).catch(reject);
+                this.storeFunction.call(this, storageKey, definitions).then(resolve).catch(reject);
             } else if (Meteor.userId()) {
                 OHIF.user.setData(storageKey, definitions).then(resolve).catch(reject);
             } else {
diff --git a/Packages/ohif-viewerbase/client/lib/WLPresets.js b/Packages/ohif-viewerbase/client/lib/WLPresets.js
index 01fefce3a..05702655f 100644
--- a/Packages/ohif-viewerbase/client/lib/WLPresets.js
+++ b/Packages/ohif-viewerbase/client/lib/WLPresets.js
@@ -50,6 +50,14 @@ class WindowLevelPresetsManager {
         this.changeObserver = new Tracker.Dependency();
     }
 
+    setRetrieveFunction(retrieveFunction) {
+        this.retrieveFunction = retrieveFunction;
+    }
+
+    setStoreFunction(storeFunction) {
+        this.storeFunction = storeFunction;
+    }
+
     updateElementWLPresetData(element) {
         const wlPresetData = cornerstone.getElementData(element, 'wlPreset');
         const enabledElement = cornerstone.getEnabledElement(element);
@@ -123,7 +131,7 @@ class WindowLevelPresetsManager {
     store(wlPresets) {
         return new Promise((resolve, reject) => {
             if (this.storeFunction) {
-                this.storeFunction.call(this, wlPresets).then(resolve).catch(reject);
+                this.storeFunction.call(this, WL_STORAGE_KEY, wlPresets).then(resolve).catch(reject);
             } else if (Meteor.userId()) {
                 OHIF.user.setData(WL_STORAGE_KEY, wlPresets).then(resolve).catch(reject);
             } else {