From 7a60d1f13bd30adcece519d2d8c09866ffd57ff9 Mon Sep 17 00:00:00 2001
From: MrExplode <me@sunstorm.rocks>
Date: Sun, 3 Nov 2024 01:52:05 +0100
Subject: [PATCH] feat(web): some websocket logic

---
 webapp/src/lib/data/control.ts   |  3 +-
 webapp/src/lib/data/websocket.ts | 50 ++++++++++++++++++++++++++++++++
 2 files changed, 52 insertions(+), 1 deletion(-)
 create mode 100644 webapp/src/lib/data/websocket.ts

diff --git a/webapp/src/lib/data/control.ts b/webapp/src/lib/data/control.ts
index be7d8f1..368f26b 100644
--- a/webapp/src/lib/data/control.ts
+++ b/webapp/src/lib/data/control.ts
@@ -1,11 +1,12 @@
 import { writable } from 'svelte/store'
 import { get } from './api'
 
-export const connected = writable(false)
 export const playing = writable(false)
 export const paused = writable(false)
 export const currentTime = writable('00 : 00 : 00 / 00')
 export const logs = writable<string[]>([])
+export const connected = writable(false)
+export const retryCountdown = writable(5)
 
 export const syncPlaying = async () => {
     const data = await get('/control/play')
diff --git a/webapp/src/lib/data/websocket.ts b/webapp/src/lib/data/websocket.ts
new file mode 100644
index 0000000..961b140
--- /dev/null
+++ b/webapp/src/lib/data/websocket.ts
@@ -0,0 +1,50 @@
+import { connected, retryCountdown } from '$lib/data/control'
+import { get } from 'svelte/store'
+
+let socket: WebSocket | null = null
+let pingTaskId = -1
+let retryTaskId = -1
+
+export const load = () => {
+    socket = new WebSocket('')
+    socket.onopen = open
+    socket.onclose = close
+    socket.onmessage = message
+}
+
+const open = () => {
+    console.log('[WS] Connected')
+    pingTaskId = setInterval(() => socket?.send(JSON.stringify({ type: 'ping' })), 10000)
+}
+
+const close = (event: CloseEvent) => {
+    console.log('[WS] Disconnected: ', event.reason)
+    connected.set(false)
+    if (pingTaskId != -1) clearInterval(pingTaskId)
+
+    retryConnection()
+}
+
+const message = (event: MessageEvent) => {
+    try {
+        const payload = JSON.parse(event.data)
+        // todo handling
+        console.log(payload)
+    } catch (error: unknown) {
+        console.log('Failed to handle WS message', error)
+    }
+}
+
+const retryConnection = () => {
+    retryCountdown.set(5)
+    retryTaskId = setInterval(() => {
+        const count = get(retryCountdown)
+        if (count > 0) {
+            retryCountdown.set(count - 1)
+        } else {
+            clearInterval(retryTaskId)
+            socket = null
+            load()
+        }
+    }, 1000)
+}