From cb344d153f034a69f59e043c5c5752412ee3afee Mon Sep 17 00:00:00 2001
From: Alex Kern <alex@kern.io>
Date: Fri, 27 Dec 2024 00:04:59 -0800
Subject: [PATCH] fix types

---
 src/components/WebRTCProvider.tsx | 2 +-
 src/fs.ts                         | 2 +-
 src/hooks/useDownloader.ts        | 4 +---
 src/messages.ts                   | 5 +++++
 4 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/components/WebRTCProvider.tsx b/src/components/WebRTCProvider.tsx
index 33c838d..324aaa4 100644
--- a/src/components/WebRTCProvider.tsx
+++ b/src/components/WebRTCProvider.tsx
@@ -1,6 +1,6 @@
 'use client'
 
-import React, { useState, useEffect, useContext, useRef } from 'react'
+import React, { useState, useEffect, useContext } from 'react'
 import Loading from './Loading'
 import Peer from 'peerjs'
 
diff --git a/src/fs.ts b/src/fs.ts
index ed0ec1d..7c51985 100644
--- a/src/fs.ts
+++ b/src/fs.ts
@@ -17,7 +17,7 @@ const readDirectoryEntries = (reader: any): Promise<any[]> =>
 
 const scanDirectoryEntry = async (entry: any): Promise<File[]> => {
   const directoryReader = entry.createReader()
-  const result = []
+  const result: File[] = []
   // eslint-disable-next-line no-constant-condition
   while (true) {
     const subentries = await readDirectoryEntries(directoryReader)
diff --git a/src/hooks/useDownloader.ts b/src/hooks/useDownloader.ts
index 100b616..88f28ca 100644
--- a/src/hooks/useDownloader.ts
+++ b/src/hooks/useDownloader.ts
@@ -215,9 +215,7 @@ export function useDownloader(uploaderPeerID: string): {
   const stopDownload = useCallback(() => {
     // TODO(@kern): Continue here with stop / pause logic
     if (dataConnection) {
-      dataConnection.send({ type: MessageType.Pause } as z.infer<
-        typeof Message
-      >)
+      dataConnection.send({ type: MessageType.Pause })
       dataConnection.close()
     }
     setIsDownloading(false)
diff --git a/src/messages.ts b/src/messages.ts
index 6ed6bd6..a36d8fd 100644
--- a/src/messages.ts
+++ b/src/messages.ts
@@ -66,6 +66,10 @@ export const UsePasswordMessage = z.object({
   password: z.string(),
 })
 
+export const PauseMessage = z.object({
+  type: z.literal(MessageType.Pause),
+})
+
 export const Message = z.discriminatedUnion('type', [
   RequestInfoMessage,
   InfoMessage,
@@ -75,6 +79,7 @@ export const Message = z.discriminatedUnion('type', [
   ErrorMessage,
   PasswordRequiredMessage,
   UsePasswordMessage,
+  PauseMessage,
 ])
 
 export type Message = z.infer<typeof Message>