session-desktop/libtextsecure/test/in_memory_signal_protocol_store.js

function SignalProtocolStore() {
	this.store = {};
}

SignalProtocolStore.prototype = {
	getIdentityKeyPair: function() {
		return Promise.resolve(this.get('identityKey'));
	},
	getLocalRegistrationId: function() {
		return Promise.resolve(this.get('registrationId'));
	},
	put: function(key, value) {
		if (key === undefined || value === undefined || key === null || value === null)
			throw new Error("Tried to store undefined/null");
		this.store[key] = value;
	},
	get: function(key, defaultValue) {
		if (key === null || key === undefined)
			throw new Error("Tried to get value for undefined/null key");
		if (key in this.store) {
			return this.store[key];
		} else {
			return defaultValue;
		}
	},
	remove: function(key) {
		if (key === null || key === undefined)
			throw new Error("Tried to remove value for undefined/null key");
		delete this.store[key];
	},

	isTrustedIdentity: function(identifier, identityKey) {
		if (identifier === null || identifier === undefined) {
			throw new error("tried to check identity key for undefined/null key");
		}
		if (!(identityKey instanceof ArrayBuffer)) {
			throw new error("Expected identityKey to be an ArrayBuffer");
		}
		var trusted = this.get('identityKey' + identifier);
		if (trusted === undefined) {
			return Promise.resolve(true);
		}
		return Promise.resolve(identityKey === trusted);
	},
	loadIdentityKey: function(identifier) {
		if (identifier === null || identifier === undefined)
			throw new Error("Tried to get identity key for undefined/null key");
		return new Promise(function(resolve) {
			resolve(this.get('identityKey' + identifier));
		}.bind(this));
	},
	putIdentityKey: function(identifier, identityKey) {
		if (identifier === null || identifier === undefined)
			throw new Error("Tried to put identity key for undefined/null key");
		return new Promise(function(resolve) {
			resolve(this.put('identityKey' + identifier, identityKey));
		}.bind(this));
	},

	/* Returns a prekeypair object or undefined */
	loadPreKey: function(keyId) {
		return new Promise(function(resolve) {
			var res = this.get('25519KeypreKey' + keyId);
			resolve(res);
		}.bind(this));
	},
	storePreKey: function(keyId, keyPair) {
		return new Promise(function(resolve) {
			resolve(this.put('25519KeypreKey' + keyId, keyPair));
		}.bind(this));
	},
	removePreKey: function(keyId) {
		return new Promise(function(resolve) {
			resolve(this.remove('25519KeypreKey' + keyId));
		}.bind(this));
	},

	/* Returns a signed keypair object or undefined */
	loadSignedPreKey: function(keyId) {
		return new Promise(function(resolve) {
			var res = this.get('25519KeysignedKey' + keyId);
			resolve(res);
		}.bind(this));
	},
	storeSignedPreKey: function(keyId, keyPair) {
		return new Promise(function(resolve) {
			resolve(this.put('25519KeysignedKey' + keyId, keyPair));
		}.bind(this));
	},
	removeSignedPreKey: function(keyId) {
		return new Promise(function(resolve) {
			resolve(this.remove('25519KeysignedKey' + keyId));
		}.bind(this));
	},

	loadSession: function(identifier) {
		return new Promise(function(resolve) {
			resolve(this.get('session' + identifier));
		}.bind(this));
	},
	storeSession: function(identifier, record) {
		return new Promise(function(resolve) {
			resolve(this.put('session' + identifier, record));
		}.bind(this));
	},
  removeAllSessions: function(identifier) {
		return new Promise(function(resolve) {
      for (key in this.store) {
        if (key.match(RegExp('^session' + identifier.replace('\+','\\\+') + '.+'))) {
          delete this.store[key];
        }
      }
      resolve();
    }.bind(this));
  },
  getDeviceIds: function(identifier) {
		return new Promise(function(resolve) {
      var deviceIds = [];
      for (key in this.store) {
        if (key.match(RegExp('^session' + identifier.replace('\+','\\\+') + '.+'))) {
          deviceIds.push(parseInt(key.split('.')[1]));
        }
      }
      resolve(deviceIds);
    }.bind(this));
  }
};