From 5f434cd31cda9d64523810777e4be48d3ce293d6 Mon Sep 17 00:00:00 2001
From: Travis Ralston <travpc@gmail.com>
Date: Thu, 13 Dec 2018 14:45:08 -0700
Subject: [PATCH] Don't break the UI when something goes wrong

---
 src/components/structures/MatrixChat.js  | 38 ++++++++++++++----------
 src/components/structures/login/Login.js |  5 +++-
 src/i18n/strings/en_EN.json              |  1 +
 3 files changed, 28 insertions(+), 16 deletions(-)

diff --git a/src/components/structures/MatrixChat.js b/src/components/structures/MatrixChat.js
index e93234c679..c8b2737cc9 100644
--- a/src/components/structures/MatrixChat.js
+++ b/src/components/structures/MatrixChat.js
@@ -1761,22 +1761,30 @@ export default React.createClass({
     },
 
     _tryDiscoverDefaultHomeserver: async function(serverName) {
-        const discovery = await AutoDiscovery.findClientConfig(serverName);
-        const state = discovery["m.homeserver"].state;
-        if (state !== AutoDiscovery.SUCCESS) {
-            console.error("Failed to discover homeserver on startup:", discovery);
+        try {
+            const discovery = await AutoDiscovery.findClientConfig(serverName);
+            const state = discovery["m.homeserver"].state;
+            if (state !== AutoDiscovery.SUCCESS) {
+                console.error("Failed to discover homeserver on startup:", discovery);
+                this.setState({
+                    defaultServerDiscoveryError: discovery["m.homeserver"].error,
+                    loadingDefaultHomeserver: false,
+                });
+            } else {
+                const hsUrl = discovery["m.homeserver"].base_url;
+                const isUrl = discovery["m.identity_server"].state === AutoDiscovery.SUCCESS
+                    ? discovery["m.identity_server"].base_url
+                    : "https://vector.im";
+                this.setState({
+                    defaultHsUrl: hsUrl,
+                    defaultIsUrl: isUrl,
+                    loadingDefaultHomeserver: false,
+                });
+            }
+        } catch (e) {
+            console.error(e);
             this.setState({
-                defaultServerDiscoveryError: discovery["m.homeserver"].error,
-                loadingDefaultHomeserver: false,
-            });
-        } else {
-            const hsUrl = discovery["m.homeserver"].base_url;
-            const isUrl = discovery["m.identity_server"].state === AutoDiscovery.SUCCESS
-                ? discovery["m.identity_server"].base_url
-                : "https://vector.im";
-            this.setState({
-                defaultHsUrl: hsUrl,
-                defaultIsUrl: isUrl,
+                defaultServerDiscoveryError: _t("Unknown error discovering homeserver"),
                 loadingDefaultHomeserver: false,
             });
         }
diff --git a/src/components/structures/login/Login.js b/src/components/structures/login/Login.js
index bfaab8fdb8..b94a1759cf 100644
--- a/src/components/structures/login/Login.js
+++ b/src/components/structures/login/Login.js
@@ -343,7 +343,10 @@ module.exports = React.createClass({
             }
         } catch (e) {
             console.error(e);
-            throw e;
+            this.setState({
+                findingHomeserver: false,
+                discoveryError: _t("Unknown error discovering homeserver"),
+            });
         }
     },
 
diff --git a/src/i18n/strings/en_EN.json b/src/i18n/strings/en_EN.json
index 8c5f3f5351..56109fa8db 100644
--- a/src/i18n/strings/en_EN.json
+++ b/src/i18n/strings/en_EN.json
@@ -1128,6 +1128,7 @@
     "Review terms and conditions": "Review terms and conditions",
     "Old cryptography data detected": "Old cryptography data detected",
     "Data from an older version of Riot has been detected. This will have caused end-to-end cryptography to malfunction in the older version. End-to-end encrypted messages exchanged recently whilst using the older version may not be decryptable in this version. This may also cause messages exchanged with this version to fail. If you experience problems, log out and back in again. To retain message history, export and re-import your keys.": "Data from an older version of Riot has been detected. This will have caused end-to-end cryptography to malfunction in the older version. End-to-end encrypted messages exchanged recently whilst using the older version may not be decryptable in this version. This may also cause messages exchanged with this version to fail. If you experience problems, log out and back in again. To retain message history, export and re-import your keys.",
+    "Unknown error discovering homeserver": "Unknown error discovering homeserver",
     "Logout": "Logout",
     "Your Communities": "Your Communities",
     "Did you know: you can use communities to filter your Riot.im experience!": "Did you know: you can use communities to filter your Riot.im experience!",