From 8c287ffcb0c2505f3063dc4a50c654cdc4b0a1d2 Mon Sep 17 00:00:00 2001
From: Robert Long <robert@robertlong.me>
Date: Fri, 10 Dec 2021 13:27:27 -0800
Subject: [PATCH] Add room alias redirect

---
 src/App.jsx  | 66 +++++++++++++++++++++++++++++++++++++++++++++++++---
 src/Home.jsx |  2 +-
 2 files changed, 64 insertions(+), 4 deletions(-)

diff --git a/src/App.jsx b/src/App.jsx
index 72461ce..26508b1 100644
--- a/src/App.jsx
+++ b/src/App.jsx
@@ -14,8 +14,14 @@ See the License for the specific language governing permissions and
 limitations under the License.
 */
 
-import React, { useEffect } from "react";
-import { BrowserRouter as Router, Switch, Route } from "react-router-dom";
+import React, { useEffect, useState } from "react";
+import {
+  BrowserRouter as Router,
+  Switch,
+  Route,
+  useLocation,
+  useHistory,
+} from "react-router-dom";
 import * as Sentry from "@sentry/react";
 import { OverlayProvider } from "@react-aria/overlays";
 import { Home } from "./Home";
@@ -24,8 +30,8 @@ import { RegisterPage } from "./RegisterPage";
 import { Room } from "./Room";
 import { ClientProvider } from "./ConferenceCallManagerHooks";
 import { useFocusVisible } from "@react-aria/interactions";
-import classNames from "classnames";
 import styles from "./App.module.css";
+import { ErrorModal } from "./ErrorModal";
 
 const SentryRoute = Sentry.withSentryRouting(Route);
 
@@ -68,9 +74,63 @@ export default function App() {
             <SentryRoute path="/room/:roomId?">
               <Room />
             </SentryRoute>
+            <SentryRoute path="*">
+              <RoomRedirect />
+            </SentryRoute>
           </Switch>
         </OverlayProvider>
       </ClientProvider>
     </Router>
   );
 }
+
+function RoomRedirect() {
+  const { pathname } = useLocation();
+  const history = useHistory();
+  const [error, setError] = useState();
+
+  useEffect(() => {
+    async function redirect() {
+      let roomId = pathname;
+
+      if (pathname.startsWith("/")) {
+        roomId = roomId.substr(1, roomId.length);
+      }
+
+      if (!roomId.startsWith("#") && !roomId.startsWith("!")) {
+        let loginHomeserverUrl = homeserverUrl.trim();
+
+        if (!loginHomeserverUrl.includes("://")) {
+          loginHomeserverUrl = "https://" + loginHomeserverUrl;
+        }
+
+        try {
+          const wellKnownUrl = new URL(
+            "/.well-known/matrix/client",
+            window.location
+          );
+          const response = await fetch(wellKnownUrl);
+          const config = await response.json();
+
+          if (config["m.homeserver"]) {
+            loginHomeserverUrl = config["m.homeserver"];
+          }
+        } catch (error) {}
+
+        const { host } = new URL(loginHomeserverUrl);
+
+        roomId = `#${roomId}:${host}`;
+      }
+
+      history.replace(`/room/${roomId}`);
+    }
+
+    redirect().catch(setError);
+  }, [history, pathname]);
+
+  if (error) {
+    return <ErrorModal error={error} />;
+  }
+
+  return <div>Loading...</div>;
+}
diff --git a/src/Home.jsx b/src/Home.jsx
index 255e901..a1af772 100644
--- a/src/Home.jsx
+++ b/src/Home.jsx
@@ -58,7 +58,7 @@ export function Home() {
       e.preventDefault();
       const data = new FormData(e.target);
       const roomId = data.get("roomId");
-      history.push(`/room/${roomId}`);
+      history.push(`/${roomId}`);
     },
     [history]
   );