fix(archive): close metadata websocket on completion

src/utils/download.ts

@@ -1,17 +1,22 @@
-import WebSocket from 'ws';
+import WebSocket from 'ws'
 
-async function downloadVideo(ws: any, id: string): Promise<{ fail: boolean, message: string, size: number }> {
+interface WsLike {
-  return new Promise(async (resolve, reject) => {
+  send: (msg: string) => void
+}
+
+async function downloadVideo(ws: WsLike, id: string): Promise<{ fail: boolean, message: string, size: number }> {
+  const { promise, resolve } = Promise.withResolvers<{ fail: boolean, message: string, size: number }>()
   let isDownloading = true
   const downloader = new WebSocket(`ws://${(process.env.METADATA!).replace('http://', '')}/download/${id}`)
-    let size: number = 0
+  let size = 0
 
-    downloader.on('message', async function message(data: any) {
+  downloader.on('message', function message(data: unknown) {
-      const text = data.toString()
+    const text = String(data)
     if (text.startsWith('VIDEOSIZE-')) {
       size = parseInt(text.replace('VIDEOSIZE-', ''))
-      } else if (text == 'done') {
+    } else if (text === 'done') {
       isDownloading = false
+      downloader.close()
       return resolve({
         fail: false,
         message: '',
@@ -24,14 +29,26 @@ async function downloadVideo(ws: any, id: string): Promise<{ fail: boolean, mess
 
   downloader.on('close', function close() {
     if (!isDownloading) return
-
+    isDownloading = false
     return resolve({
       fail: true,
       message: 'The metadata server unexpectedly closed the websocket. Please try again.',
       size
     })
   })
+
+  downloader.on('error', function error(err: Error) {
+    if (!isDownloading) return
+    isDownloading = false
+    downloader.close()
+    return resolve({
+      fail: true,
+      message: `WebSocket error: ${err?.message || String(err)}`,
+      size
     })
+  })
+
+  return promise
 }
 
 export { downloadVideo }