From 70f69a3cffb42d33d29ad4b1b05facaab3dfc691 Mon Sep 17 00:00:00 2001
From: Neil Dwyer <neildwyer1991@gmail.com>
Date: Mon, 6 May 2024 17:44:53 -0700
Subject: [PATCH] Re-introduce connection mode

---
 src/components/PlaygroundConnect.tsx |  5 +++--
 src/hooks/useConnection.tsx          | 21 ++++++++++++++-------
 src/pages/index.tsx                  | 17 ++++++++---------
 3 files changed, 25 insertions(+), 18 deletions(-)

diff --git a/src/components/PlaygroundConnect.tsx b/src/components/PlaygroundConnect.tsx
index 63e8567..cc82b1b 100644
--- a/src/components/PlaygroundConnect.tsx
+++ b/src/components/PlaygroundConnect.tsx
@@ -2,10 +2,11 @@ import { useConfig } from "@/hooks/useConfig";
 import { CLOUD_ENABLED, CloudConnect } from "../cloud/CloudConnect";
 import { Button } from "./button/Button";
 import { useState } from "react";
+import { ConnectionMode } from "@/hooks/useConnection";
 
 type PlaygroundConnectProps = {
   accentColor: string;
-  onConnectClicked: () => void;
+  onConnectClicked: (mode: ConnectionMode) => void;
 };
 
 const ConnectTab = ({ active, onClick, children }: any) => {
@@ -57,7 +58,7 @@ const TokenConnect = ({
             newSettings.ws_url = url;
             newSettings.token = token;
             setUserSettings(newSettings);
-            onConnectClicked();
+            onConnectClicked("manual");
           }}
         >
           Connect
diff --git a/src/hooks/useConnection.tsx b/src/hooks/useConnection.tsx
index 85c4715..2addebc 100644
--- a/src/hooks/useConnection.tsx
+++ b/src/hooks/useConnection.tsx
@@ -1,17 +1,19 @@
 "use client"
 
-import { CLOUD_ENABLED } from "@/cloud/CloudConnect";
 import { useCloud } from "@/cloud/useCloud";
 import React, { createContext, useState } from "react";
 import { useCallback } from "react";
 import { useConfig } from "./useConfig";
 
+export type ConnectionMode = "cloud" | "manual" | "env"
+
 type TokenGeneratorData = {
   shouldConnect: boolean;
   wsUrl: string;
   token: string;
+  mode: ConnectionMode;
   disconnect: () => Promise<void>;
-  connect: () => Promise<void>;
+  connect: (mode: ConnectionMode) => Promise<void>;
 };
 
 const ConnectionContext = createContext<TokenGeneratorData | undefined>(undefined);
@@ -26,16 +28,20 @@ export const ConnectionProvider = ({
   const [connectionDetails, setConnectionDetails] = useState<{
     wsUrl: string;
     token: string;
+    mode: ConnectionMode;
     shouldConnect: boolean;
-  }>({ wsUrl: "", token: "", shouldConnect: false });
+  }>({ wsUrl: "", token: "", shouldConnect: false, mode: "manual" });
 
-  const connect = useCallback(async () => {
+  const connect = useCallback(async (mode: ConnectionMode) => {
     let token = "";
     let url = "";
-    if (CLOUD_ENABLED) {
+    if (mode === "cloud") {
       token = await generateToken();
       url = cloudWSUrl;
-    } else if (process.env.NEXT_PUBLIC_LIVEKIT_URL) {
+    } else if (mode === "env") {
+      if(!process.env.NEXT_PUBLIC_LIVEKIT_URL) {
+        throw new Error("NEXT_PUBLIC_LIVEKIT_URL is not set");
+      }
       url = process.env.NEXT_PUBLIC_LIVEKIT_URL;
       const {accessToken} = await fetch("/api/token").then((res) => res.json());
       token = accessToken;
@@ -43,7 +49,7 @@ export const ConnectionProvider = ({
       token = config.settings.token;
       url = config.settings.ws_url;
     }
-    setConnectionDetails({ wsUrl: url, token, shouldConnect: true });
+    setConnectionDetails({ wsUrl: url, token, shouldConnect: true, mode });
   }, [
     cloudWSUrl,
     config.settings.token,
@@ -61,6 +67,7 @@ export const ConnectionProvider = ({
         wsUrl: connectionDetails.wsUrl,
         token: connectionDetails.token,
         shouldConnect: connectionDetails.shouldConnect,
+        mode: connectionDetails.mode,
         connect,
         disconnect,
       }}
diff --git a/src/pages/index.tsx b/src/pages/index.tsx
index 6e40c9d..7517966 100644
--- a/src/pages/index.tsx
+++ b/src/pages/index.tsx
@@ -12,10 +12,8 @@ import { PlaygroundConnect } from "@/components/PlaygroundConnect";
 import Playground from "@/components/playground/Playground";
 import { PlaygroundToast, ToastType } from "@/components/toast/PlaygroundToast";
 import { ConfigProvider, useConfig } from "@/hooks/useConfig";
-import { ConnectionProvider, useConnection } from "@/hooks/useConnection";
+import { ConnectionMode, ConnectionProvider, useConnection } from "@/hooks/useConnection";
 import { useMemo } from "react";
-import { CLOUD_ENABLED } from "@/cloud/CloudConnect";
-import { useCloud } from "@/cloud/useCloud";
 
 const themeColors = [
   "cyan",
@@ -45,14 +43,14 @@ export function HomeInner() {
     message: string;
     type: ToastType;
   } | null>(null);
-  const { shouldConnect, wsUrl, token, connect, disconnect } =
+  const { shouldConnect, wsUrl, token, mode, connect, disconnect } =
     useConnection();
   
   const {config} = useConfig();
 
   const handleConnect = useCallback(
-    async (c: boolean) => {
-      c ? connect() : disconnect();
+    async (c: boolean, mode: ConnectionMode) => {
+      c ? connect(mode) : disconnect();
     },
     [connect, disconnect]
   );
@@ -119,7 +117,8 @@ export function HomeInner() {
             <Playground
               themeColors={themeColors}
               onConnect={(c) => {
-                handleConnect(c);
+                const m = process.env.NEXT_PUBLIC_LIVEKIT_URL ? "env" : mode;
+                handleConnect(c, m);
               }}
             />
             <RoomAudioRenderer />
@@ -128,8 +127,8 @@ export function HomeInner() {
         ) : (
           <PlaygroundConnect
             accentColor={themeColors[0]}
-            onConnectClicked={() => {
-              handleConnect(true);
+            onConnectClicked={(mode) => {
+              handleConnect(true, mode);
             }}
           />
         )}