From ce8ad89d925df4a32536f166a8dfbcdf2570549e Mon Sep 17 00:00:00 2001
From: Fornax <wim@fornaxian.tech>
Date: Wed, 24 Apr 2024 18:47:18 +0200
Subject: [PATCH] Use new websocket proto for view counting

---
 svelte/src/file_viewer/FileStats.svelte  | 48 ++++++++++++++++--------
 svelte/src/file_viewer/FileViewer.svelte |  9 +----
 2 files changed, 34 insertions(+), 23 deletions(-)

diff --git a/svelte/src/file_viewer/FileStats.svelte b/svelte/src/file_viewer/FileStats.svelte
index cda3976..c9229a6 100644
--- a/svelte/src/file_viewer/FileStats.svelte
+++ b/svelte/src/file_viewer/FileStats.svelte
@@ -1,4 +1,5 @@
 <script>
+import { onMount } from "svelte";
 import { formatDataVolume, formatThousands } from "../util/Formatting.svelte"
 
 export let file = {
@@ -9,6 +10,7 @@ export let file = {
 	bandwidth_used: 0,
 	bandwidth_used_paid: 0,
 }
+export let view_token = ""
 
 let views = 0
 let downloads = 0
@@ -30,18 +32,28 @@ let update_stats = async id => {
 	}
 	size = file.size
 
-	// If the socket is already active we need to close it
-	if (socket !== null) {
-		// Disable the error handler so it doesn't start retrying the connection
-		socket.onerror = null
-		socket.close()
-		socket = null
+	send_watch_command()
+}
+
+const send_watch_command = () => {
+	if (socket.readyState === WebSocket.OPEN) {
+		socket.send(
+			JSON.stringify(
+				{cmd: "watch_file", a1: file.id, a2: view_token}
+			)
+		)
+	}
+}
+
+const init_socket = () => {
+	if (socket !== null || socket.readyState !== WebSocket.CLOSED) {
+		return
 	}
 
-	console.log("opening socket for", id)
-	socket = new WebSocket(
-		location.origin.replace(/^http/, 'ws') + "/api/file/" + id + "/stats"
-	)
+	console.log("initializing socket")
+	socket = new WebSocket(location.origin.replace(/^http/, 'ws') + "/api/file_stats")
+
+	socket.onopen = () => send_watch_command()
 	socket.onmessage = msg => {
 		let j = JSON.parse(msg.data)
 		console.debug("WS update", j)
@@ -63,14 +75,20 @@ let update_stats = async id => {
 		socket = null
 		error_msg = "failed to get stats, retrying..."
 
-		window.setTimeout(() => {
-			if (socket === null) {
-				update_stats(file.id)
-			}
-		}, 3000)
+		window.setTimeout(init_socket, 2000)
 	}
 }
 
+onMount(() => {
+	init_socket()
+
+	return () => {
+		if (socket !== null) {
+			socket.close()
+			socket = null
+		}
+	}
+})
 </script>
 
 <div>
diff --git a/svelte/src/file_viewer/FileViewer.svelte b/svelte/src/file_viewer/FileViewer.svelte
index f2d31f1..1d207df 100644
--- a/svelte/src/file_viewer/FileViewer.svelte
+++ b/svelte/src/file_viewer/FileViewer.svelte
@@ -207,13 +207,6 @@ const open_file_index = async index => {
 	}
 
 	apply_customizations(file)
-
-	// Register a file view
-	fetch(window.api_endpoint + "/file/" + file.id + "/view", {
-		method: "POST",
-		headers: { "Content-Type": "application/x-www-form-urlencoded" },
-		body: "token=" + view_token
-	})
 }
 const toggle_gallery = () => {
 	if (view === "gallery") {
@@ -425,7 +418,7 @@ const keyboard_event = evt => {
 	<div class="file_preview_row">
 		<div class="toolbar" class:toolbar_visible>
 			{#if view === "file"}
-				<FileStats file={file}/>
+				<FileStats file={file} view_token={view_token}/>
 			{:else if view === "gallery"}
 				<ListStats list={list}/>
 			{/if}