fix ws URL

simbafs · Aug 2, 2024 · b80cfb8 · b80cfb8
1 parent 943143d
commit b80cfb8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/frontend/src/hooks/useAttendance.ts b/frontend/src/hooks/useAttendance.ts
@@ -4,11 +4,14 @@ import { useToken } from './useToken'
 import useWS from './useWS'
 import { shouldParse } from '@/libs/util'
 import { Diff } from './useDiff'
+import { useOrigin } from './useOrigin'
 
 // useAttendance get init value of attendance from server and listen to websocket for updates
 export function useAttendance() {
 	const token = useToken()
-	const { socket, lastMessage } = useWS(`ws://localhost:3000/ws?token=${token}`)
+	const wsURL = useOrigin().replace('http', 'ws') + '/ws?token=' + token
+	console.log({ wsURL })
+	const { socket, lastMessage } = useWS(wsURL)
 	const [attendance, updateAttendance] = useReducer(
 		(
 			state: Attendance,

diff --git a/frontend/src/hooks/useOrigin.ts b/frontend/src/hooks/useOrigin.ts
@@ -0,0 +1,11 @@
+import { useEffect, useState } from 'react'
+
+export function useOrigin() {
+    const [origin, setOrigin] = useState('')
+
+    useEffect(() => {
+        setOrigin(window.location.origin)
+    }, [])
+
+    return origin
+}