ES6 SessionStore

src/components/structures/MatrixChat.js

@@ -40,7 +40,7 @@ var PageTypes = require('../../PageTypes');
 var createRoom = require("../../createRoom");
 import * as UDEHandler from '../../UnknownDeviceErrorHandler';
 
-import getSessionStore from '../../stores/SessionStore';
+import sessionStore from '../../stores/SessionStore';
 
 module.exports = React.createClass({
     displayName: 'MatrixChat',
@@ -251,7 +251,7 @@ module.exports = React.createClass({
             });
         }
 
-        this._sessionStore = getSessionStore();
+        this._sessionStore = sessionStore;
         this._sessionStore.on('update', this._setStateFromSessionStore);
         this._setStateFromSessionStore();
     },

src/components/views/settings/ChangePassword.js

@@ -22,7 +22,7 @@ var Modal = require("../../../Modal");
 var sdk = require("../../../index");
 import AccessibleButton from '../elements/AccessibleButton';
 
-import getSessionStore from '../../../stores/SessionStore';
+import sessionStore from '../../../stores/SessionStore';
 
 module.exports = React.createClass({
     displayName: 'ChangePassword',
@@ -68,7 +68,7 @@ module.exports = React.createClass({
     },
 
     componentWillMount: function() {
-        this.sessionStore = getSessionStore();
+        this.sessionStore = sessionStore;
         this.sessionStore.on('update', this.setStateFromSessionStore);
 
         this.setStateFromSessionStore();

src/stores/SessionStore.js

@@ -6,53 +6,54 @@ import EventEmitter from 'events';
  * store that listens for actions and updates its state accordingly, informing any
  * listeners (views) of state changes via the 'update' event.
  */
-function SessionStore() {
+class SessionStore extends EventEmitter {
-    // Initialise state
+    constructor() {
-    this._state = {
+        super();
-        cachedPassword: localStorage.getItem('mx_pass'),
-    };
 
-    dis.register(this._onAction.bind(this));
+        // Initialise state
+        this._state = {
+            cachedPassword: localStorage.getItem('mx_pass'),
+        };
+
+        dis.register(this._onAction.bind(this));
+    }
+
+    _update() {
+        // Persist state to localStorage
+        if (this._state.cachedPassword) {
+            localStorage.setItem('mx_pass', this._state.cachedPassword);
+        } else {
+            localStorage.removeItem('mx_pass', this._state.cachedPassword);
+        }
+
+        this.emit('update');
+    }
+
+    _setState(newState) {
+        this._state = Object.assign(this._state, newState);
+        this._update();
+    }
+
+    _onAction(payload) {
+        switch (payload.action) {
+            case 'cached_password':
+                this._setState({
+                    cachedPassword: payload.cachedPassword,
+                });
+                break;
+            case 'password_changed':
+                this._setState({
+                    cachedPassword: null,
+                });
+                break;
+        }
+    }
+
+    getCachedPassword() {
+        return this._state.cachedPassword;
+    }
 }
 
-// Inherit from EventEmitter
-SessionStore.prototype = EventEmitter.prototype;
-
-SessionStore.prototype._update = function() {
-    // Persist state to localStorage
-    if (this._state.cachedPassword) {
-        localStorage.setItem('mx_pass', this._state.cachedPassword);
-    } else {
-        localStorage.removeItem('mx_pass', this._state.cachedPassword);
-    }
-
-    this.emit('update');
-};
-
-SessionStore.prototype._setState = function(newState) {
-    this._state = Object.assign(this._state, newState);
-    this._update();
-};
-
-SessionStore.prototype._onAction = function(payload) {
-    switch (payload.action) {
-        case 'cached_password':
-            this._setState({
-                cachedPassword: payload.cachedPassword,
-            });
-            break;
-        case 'password_changed':
-            this._setState({
-                cachedPassword: null,
-            });
-            break;
-    }
-};
-
-SessionStore.prototype.getCachedPassword = function() {
-    return this._state.cachedPassword;
-};
-
 // Export singleton getter
 let singletonSessionStore = null;
 if (!singletonSessionStore) {