Alternate implementation of localStorage for embedded veresion.

Author: baffles

codepulse/src/main/resources/toserve/common/CodePulse.js

@@ -54,6 +54,11 @@
 		return CodePulse.pathPrefix + '/api/' + endpoint
 	}
 
+	if (CodePulse.isEmbedded)
+		CodePulse.storage = process.mainModule.exports.storage;
+	else
+		CodePulse.storage = window.localStorage;
+
 	// Handle external links via the "external-href" attribute
 	$(document).ready(function(){
 

codepulse/src/main/resources/toserve/common/UpdateController.js

@@ -73,10 +73,10 @@
 
 		/** returns true if the user has ignored this version */
 		function isUpdateIgnored(version) {
-			return localStorage.getItem('updater.ignoredVersion') == version
+			return CodePulse.storage.getItem('updater.ignoredVersion') == version
 		}
 
-		this.ignoreCurrentRelease = function() { localStorage.setItem('updater.ignoredVersion', self.releaseVersion) }
+		this.ignoreCurrentRelease = function() { CodePulse.storage.setItem('updater.ignoredVersion', self.releaseVersion) }
 
 		/** checks `release`, firing off the appropriate events as necessary */
 		function checkRelease(release) {
@@ -97,15 +97,15 @@
 			getLatestRelease(function(latestRelease) {
 				checkRelease(latestRelease)
 
-				localStorage.setItem('updater.latestReleaseCache', JSON.stringify(latestRelease))
-				localStorage.setItem('updater.lastCheck', Date.now())
+				CodePulse.storage.setItem('updater.latestReleaseCache', JSON.stringify(latestRelease))
+				CodePulse.storage.setItem('updater.lastCheck', Date.now())
 				setTimeout(runUpdateCheck, CheckInterval)
 			})
 		}
 
 		/** schedule an update check (or run one immediately), based on the last check time */
 		function scheduleUpdateCheck() {
-			var lastCheck = parseInt(localStorage.getItem('updater.lastCheck'))
+			var lastCheck = parseInt(CodePulse.storage.getItem('updater.lastCheck'))
 			var curTime = Date.now()
 
 			if (!lastCheck || curTime > (lastCheck + CheckInterval))
@@ -115,7 +115,7 @@
 				setTimeout(runUpdateCheck, lastCheck + CheckInterval - curTime)
 
 				// for now, check what's cached, if we can
-				var cachedRelease = JSON.parse(localStorage.getItem('updater.latestReleaseCache'))
+				var cachedRelease = JSON.parse(CodePulse.storage.getItem('updater.latestReleaseCache'))
 				if (cachedRelease)
 					checkRelease(cachedRelease)
 			}

codepulse/src/main/resources/toserve/pages/projects/trace-recording-controls.js

@@ -98,17 +98,17 @@
 
 		this.getOrElseUpdate = function(rawKey, defaultValue){
 			var key = storageKey(rawKey)
-				entry = localStorage.getItem(key)
+				entry = CodePulse.storage.getItem(key)
 			if(!entry) {
 				entry = defaultValue
-				localStorage.setItem(key, entry)
+				CodePulse.storage.setItem(key, entry)
 			}
 			return entry
 		}
 
 		this.update = function(rawKey, newValue){
 			var key = storageKey(rawKey)
-			localStorage.setItem(key, newValue)
+			CodePulse.storage.setItem(key, newValue)
 		}
 	}
 	var localStorageColors = new LocalStorageColors()

distrib/common/app/main.js

@@ -221,11 +221,14 @@ function checkWindowClose(window, cb) {
 	} else cb();
 }
 
-var gotMainWindow = false;
+var mainWindow = false;
 
 exports.registerMainWindow = function(window) {
-	if (!gotMainWindow) {
+	if (!mainWindow) {
 		writeLog('Registering main window...\n');
+		mainWindow = window;
+
+		initStorage();
 
 		window.gui.on('close', function() {
 			checkWindowClose(window.window, function() {
@@ -246,4 +249,59 @@ exports.registerMainWindow = function(window) {
 	} else {
 		writeLog('Attempt to re-register main window ignored...\n');
 	}
+}
+
+// an alternate implementation of localStorage for the embedded version
+// (since port number changes, so we're on a different origin every time, and therefore receive a
+// different localStorage)
+function initStorage() {
+	var gui = global.window.nwDispatcher.requireNwGui(),
+		path = require('path'), fs = require('fs');
+
+	var storage = {},
+		storeFile = path.join(gui.App.dataPath, 'storage.json');
+
+	function loadStorage() {
+		try {
+			if (fs.existsSync(storeFile))
+				storage = JSON.parse(fs.readFileSync(storeFile, 'utf8'));
+		} catch (e) {
+			writeLog('Error loading storage.json: ' + e + '\n');
+		}
+	}
+
+	function writeStorage() {
+		fs.writeFileSync(storeFile, JSON.stringify(storage));
+	}
+
+	function getItem(key) {
+		return storage[key];
+	}
+
+	function setItem(key, value) {
+		if (typeof value !== 'string')
+			value = value.toString();
+
+		storage[key] = value;
+		writeStorage();
+	}
+
+	function removeItem(key) {
+		delete storage[key];
+		writeStorage();
+	}
+
+	function clear() {
+		storage = {};
+		writeStorage();
+	}
+
+	loadStorage();
+
+	exports.storage = {
+		getItem: function(key) { return getItem(key); },
+		setItem: function(key, value) { setItem(key, value); },
+		removeItem: function(key) { removeItem(key); },
+		clear: function() { clear(); }
+	};
 }