Add a key parameter to the new initial state API

Signed-off-by: Christoph Wurst <christoph@winzerhof-wurst.at>

2 files changed, 16 insertions, 18 deletions

diff --git a/core/src/OCP/index.js b/core/src/OCP/index.js
index 6ec38668e72..aa3650b0614 100644
--- a/core/src/OCP/index.js
+++ b/core/src/OCP/index.js
@@ -3,7 +3,7 @@
  */
 import * as AppConfig from './appconfig'
 import * as Comments from './comments'
-import initialState from './initialstate'
+import * as InitialState from './initialstate'
 import Loader from './loader'
 import * as WhatsNew from './whatsnew'
 
@@ -11,7 +11,7 @@ import * as WhatsNew from './whatsnew'
 export default {
 	AppConfig,
 	Comments,
-	InitialState: initialState,
+	InitialState,
 	Loader,
 	WhatsNew,
 };
diff --git a/core/src/OCP/initialstate.js b/core/src/OCP/initialstate.js
index 39e913157a7..591a7e868cc 100644
--- a/core/src/OCP/initialstate.js
+++ b/core/src/OCP/initialstate.js
@@ -21,22 +21,20 @@
  */
 
 /**
- * @namespace OCP
- * @class InitialState
+ * @namespace OCP.InitialState
  */
-export default {
-	loadState: function(app) {
-		const elem = document.querySelector('#initial-state-' + app);
-		if (elem === null) {
-			console.error('Could not find initial state of ' + app);
-			throw new Error('Could not find initial state of ' + app);
-		}
 
-		try {
-			return JSON.parse(atob(elem.value));
-		} catch (e) {
-			console.error('Could not parse initial state of ' + app);
-			throw new Error('Could not parse initial state of ' + app);
-		}
-	},
+export function loadState (app, key) {
+	const elem = document.querySelector(`#initial-state-${app}-${key}`);
+	if (elem === null) {
+		console.error('Could not find initial state of ' + app);
+		throw new Error('Could not find initial state of ' + app);
+	}
+
+	try {
+		return JSON.parse(atob(elem.value));
+	} catch (e) {
+		console.error('Could not parse initial state of ' + app);
+		throw new Error('Could not parse initial state of ' + app);
+	}
 }