Merge pull request #13 from matrix-org/markjh/vector-cleanup-client-peg

Markjh/vector cleanup client peg

src/MatrixClientPeg.js

@@ -22,18 +22,25 @@ var Matrix = require("matrix-js-sdk");
 var matrixClient = null;
 
 var localStorage = window.localStorage;
+
+function createClient(hs_url, is_url, user_id, access_token) {
+    var opts = {
+        baseUrl: hs_url,
+        idBaseUrl: is_url,
+        accessToken: access_token,
+        userId: user_id
+    };
+
+    matrixClient = Matrix.createClient(opts);
+}
+
 if (localStorage) {
     var hs_url = localStorage.getItem("mx_hs_url");
     var is_url = localStorage.getItem("mx_is_url") || 'https://matrix.org';
     var access_token = localStorage.getItem("mx_access_token");
     var user_id = localStorage.getItem("mx_user_id");
     if (access_token && user_id && hs_url) {
-        matrixClient = Matrix.createClient({
-            baseUrl: hs_url,
-            idBaseUrl: is_url,
-            accessToken: access_token,
-            userId: user_id
-        });
+        createClient(hs_url, is_url, user_id, access_token);
     }
 }
 
@@ -42,8 +49,8 @@ module.exports = {
         return matrixClient;
     },
 
-    replace: function(cli) {
-        matrixClient = cli;
+    unset: function() {
+        matrixClient = null;
     },
 
     replaceUsingUrls: function(hs_url, is_url) {
@@ -51,6 +58,23 @@ module.exports = {
             baseUrl: hs_url,
             idBaseUrl: is_url
         });
+    },
+
+    replaceUsingAccessToken: function(hs_url, is_url, user_id, access_token) {
+        createClient(hs_url, is_url, user_id, access_token);
+        if (localStorage) {
+            try {
+                localStorage.clear();
+                localStorage.setItem("mx_hs_url", hs_url);
+                localStorage.setItem("mx_is_url", is_url);
+                localStorage.setItem("mx_user_id", user_id);
+                localStorage.setItem("mx_access_token", access_token);
+            } catch (e) {
+                console.warn("Error using local storage: can't persist session!");
+            }
+        } else {
+            console.warn("No local storage available: can't persist session!");
+        }
     }
 };
 

src/controllers/pages/MatrixChat.js

@@ -91,7 +91,7 @@ module.exports = {
                 Notifier.stop();
                 Presence.stop();
                 MatrixClientPeg.get().removeAllListeners();
-                MatrixClientPeg.replace(null);
+                MatrixClientPeg.unset();
                 break;
             case 'start_registration':
                 if (this.state.logged_in) return;

src/controllers/templates/Login.js

@@ -75,27 +75,10 @@ module.exports = {
             'user': formVals.username,
             'password': formVals.password
         }).done(function(data) {
-            // XXX: we assume this means we're logged in, but there could be a next stage
-            MatrixClientPeg.replace(Matrix.createClient({
-                baseUrl: self.state.hs_url,
-                idBaseUrl: self.state.is_url,
-                userId: data.user_id,
-                accessToken: data.access_token
-            }));
-            var localStorage = window.localStorage;
-            if (localStorage) {
-                try {
-                    localStorage.clear();
-                    localStorage.setItem("mx_hs_url", self.state.hs_url);
-                    localStorage.setItem("mx_is_url", self.state.is_url);
-                    localStorage.setItem("mx_user_id", data.user_id);
-                    localStorage.setItem("mx_access_token", data.access_token);
-                } catch (e) {
-                    console.warn("Error using local storage: can't persist session!");
-                }
-            } else {
-                console.warn("No local storage available: can't persist session!");
-            }
+            MatrixClientPeg.replaceUsingAccessToken(
+                self.state.hs_url, self.state.is_url,
+                data.user_id, data.access_token
+            );
             if (self.props.onLoggedIn) {
                 self.props.onLoggedIn();
             }

src/controllers/templates/Register.js

@@ -259,20 +259,9 @@ module.exports = {
     },
 
     onRegistered: function(user_id, access_token) {
-        MatrixClientPeg.replace(Matrix.createClient({
-            baseUrl: this.state.hs_url,
-            idBaseUrl: this.state.is_url,
-            userId: user_id,
-            accessToken: access_token
-        }));
-        var localStorage = window.localStorage;
-        if (localStorage) {
-            localStorage.setItem("mx_hs_url", this.state.hs_url);
-            localStorage.setItem("mx_user_id", user_id);
-            localStorage.setItem("mx_access_token", access_token);
-        } else {
-            console.warn("No local storage available: can't persist session!");
-        }
+        MatrixClientPeg.replaceUsingAccessToken(
+            this.state.hs_url, this.state.is_url, user_id, access_token
+        );
         if (this.props.onLoggedIn) {
             this.props.onLoggedIn();
         }