stable reconnect

thewh1teagle · Nov 19, 2023 · 831e41c · 831e41c
1 parent a4d85f3
commit 831e41c
Show file tree

Hide file tree

Showing 2 changed files with 85 additions and 27 deletions.
diff --git a/desktop/src/App.tsx b/desktop/src/App.tsx
@@ -27,7 +27,7 @@ function App() {
   const [id, ] = useLocalStorage('id', uuidv4())
   console.log('localstorage id => ', id)
   const [loading, setLoading] = useState(true);
-  const [peer, ] = useState(new Peer(id, { pingInterval: 1000 }));
+  const [peer, ] = useState(new Peer(id, { pingInterval: 2000 , host: '1.peerjs.com'}));
   const [conn, setConn] = useState<DataConnection | null>(null);
   const qrDiv = useRef<HTMLDivElement>(null);
 
@@ -99,7 +99,7 @@ function App() {
       <span className="text-3xl mb-5">Ready to connect</span>
       <div ref={qrDiv} />
       {loading && (
-        <div className="flex items-center justify-center w-[300px] h-[300px] rounded-xl bg-neutral">
+        <div className="flex items-center justify-center w-[300px] h-[300px] rounded-xl bg-transparent shadow-xl">
           <span className="loading loading-spinner loading-lg p-0"></span>
         </div>
       )}

diff --git a/web/src/App.tsx b/web/src/App.tsx
@@ -1,11 +1,10 @@
-import Peer, { DataConnection } from "peerjs";
-import { useEffect, useState } from "react";
 import NoSleep from "nosleep.js";
-import { v4 as uuidv4 } from 'uuid';
+import { useEffect, useRef, useState } from "react";
 import { useLongPress } from 'use-long-press';
-const noSleep = new NoSleep();
 
+import Peer, {DataConnection} from "peerjs";
 
+const noSleep = new NoSleep();
 enum Action {
   VOL_UP,
   VOL_DN,
@@ -18,17 +17,18 @@ interface Message {
   action: Action;
 }
 
-
-
 function App() {
 
-
   const params = new URLSearchParams(window.location.search);
-  const [loading, setLoading] = useState(true);
-  const [id,] = useState(uuidv4())
+
+
+  const peerRef = useRef<Peer>()
+  const connRef = useRef<DataConnection>()
+  const addressRef = useRef<null | string>('')
+  const checkConnectionIntervalRef = useRef<number | null>(null)
+  const connectIntervalRef = useRef<number | null>(null)
+  const [loading ,setLoading] = useState(true)
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  const [peer] = useState(new Peer(id, { pingInterval: 1000 }));
-  const [conn, setConn] = useState<DataConnection | null>(null);
 
   function sendMessage(data: Message) {
     console.log('sending => ', data)
@@ -38,25 +38,83 @@ function App() {
       noSleep.enable()
     }
 
-    conn?.send(JSON.stringify(data));
+    connRef.current?.send(JSON.stringify(data));
   }
 
-  useEffect(() => {
-    const address = params.get("id");
-    if (!address) {
-      alert('Wrong address, please scan again.')
-      return
+
+  function reconnect() {
+
+    if (checkConnectionIntervalRef.current) {
+      clearInterval(checkConnectionIntervalRef.current);
+    }
+    if (connectIntervalRef.current) {
+      clearInterval(connectIntervalRef.current);
+    }
+      // reconnect
+      peerRef.current?.destroy()
+      connRef.current?.close()
+      connRef.current = undefined
+      peerRef.current = undefined
+      setLoading(true)
+      if (checkConnectionIntervalRef.current) {
+        clearInterval(checkConnectionIntervalRef.current)
+      }
+      connectIntervalRef.current = setInterval(() => connect(), 5000)
+
+  }
+
+  function checkConnection() {
+    if (!connRef.current?.peerConnection) {
+      // reconnect
+      reconnect()
     }
-    console.log("connecting to ", address);
-    peer.on("open", () => {
-      const connection = peer.connect(address!);
-      connection.on('open', () => {
-        setLoading(false);
+  }
+
+
+  async function createConnection() {
+    return new Promise<void>((resolve, reject) => {
+      console.log("connecting to ", addressRef.current);
+      peerRef.current = new Peer({host: '1.peerjs.com', debug: 3, pingInterval: 2000})
+      peerRef.current.on('open', () => {
+        if (addressRef.current) {
+          peerRef?.current?.on('error', (error) => reject(error))
+          connRef.current = peerRef.current?.connect(addressRef.current)
+          connRef.current?.on('open', () => resolve())
+        }
       })
-      setConn(connection);
-    });
+    })
+  }
 
-
+  async function connect() {
+
+    try {
+      await createConnection()
+
+      connRef.current?.once('iceStateChanged', state => {
+        if (state === 'disconnected' || state == 'closed' || state == 'failed') {
+          reconnect()
+        }
+      })
+      if (connectIntervalRef.current) {
+        clearInterval(connectIntervalRef.current!)
+      }
+      checkConnectionIntervalRef.current = setInterval(checkConnection, 1000)
+      setLoading(false)
+    } catch (e) {
+      console.log(e)
+    }
+  }
+
+  useEffect(() => {
+    async function init() {
+      addressRef.current = params.get("id");
+      if (!addressRef.current) {
+        alert('Wrong address, please scan again.')
+        return
+      }
+      await reconnect()
+    }
+    init()
 
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, []);