homeio

1"use client";
2
3import type { UsbDrive, UsbSseEvent } from "@/lib/shared/contracts/usb";
4import { queryKeys } from "@/lib/shared/query-keys";
5import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
6import { useEffect } from "react";
7
8async function fetchUsbDrives(): Promise<UsbDrive[]> {
9  const res = await fetch("/api/v1/files/usb");
10  if (!res.ok) return [];
11  const data = await res.json() as { drives: UsbDrive[] };
12  return data.drives;
13}
14
15export function useUsbDrives() {
16  const queryClient = useQueryClient();
17
18  const { data: drives = [] } = useQuery({
19    queryKey: queryKeys.usbDrives,
20    queryFn: fetchUsbDrives,
21    staleTime: 10_000,
22    refetchInterval: 15_000,
23  });
24
25  useEffect(() => {
26    const source = new EventSource("/api/v1/files/usb/stream");
27
28    const handleEvent = () => {
29      queryClient.invalidateQueries({ queryKey: queryKeys.usbDrives });
30    };
31
32    const eventTypes: UsbSseEvent["type"][] = [
33      "usb.connected",
34      "usb.disconnected",
35      "usb.mounted",
36      "usb.unmounted",
37    ];
38    for (const type of eventTypes) {
39      source.addEventListener(type, handleEvent);
40    }
41
42    return () => {
43      for (const type of eventTypes) {
44        source.removeEventListener(type, handleEvent);
45      }
46      source.close();
47    };
48  }, [queryClient]);
49
50  const mountMutation = useMutation({
51    mutationFn: async (driveId: string) => {
52      const res = await fetch(`/api/v1/files/usb/${driveId}/mount`, { method: "POST" });
53      if (!res.ok) throw new Error("Mount failed");
54    },
55    onSuccess: () => queryClient.invalidateQueries({ queryKey: queryKeys.usbDrives }),
56  });
57
58  const unmountMutation = useMutation({
59    mutationFn: async (driveId: string) => {
60      const res = await fetch(`/api/v1/files/usb/${driveId}/unmount`, { method: "POST" });
61      if (!res.ok) throw new Error("Unmount failed");
62    },
63    onSuccess: () => queryClient.invalidateQueries({ queryKey: queryKeys.usbDrives }),
64  });
65
66  const ejectMutation = useMutation({
67    mutationFn: async (driveId: string) => {
68      const res = await fetch(`/api/v1/files/usb/${driveId}/eject`, { method: "POST" });
69      if (!res.ok) throw new Error("Eject failed");
70    },
71    onSuccess: () => queryClient.invalidateQueries({ queryKey: queryKeys.usbDrives }),
72  });
73
74  return {
75    drives,
76    mount: mountMutation.mutate,
77    unmount: unmountMutation.mutate,
78    eject: ejectMutation.mutate,
79    isMounting: mountMutation.isPending,
80    isUnmounting: unmountMutation.isPending,
81    isEjecting: ejectMutation.isPending,
82  };
83}
84