homeio

1"use client";
2
3import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
4import type {
5  CreateNetworkShareRequest,
6  DiscoverServersResponse,
7  DiscoverSharesRequest,
8  DiscoverSharesResponse,
9  NetworkShareStatus,
10} from "@/lib/shared/contracts/files";
11import { queryKeys } from "@/lib/shared/query-keys";
12
13type NetworkSharesResponse = {
14  data: NetworkShareStatus[];
15  meta?: {
16    count: number;
17  };
18};
19
20type NetworkShareResponse = {
21  data: NetworkShareStatus;
22};
23
24type DiscoverServersApiResponse = {
25  data: DiscoverServersResponse;
26};
27
28type DiscoverSharesApiResponse = {
29  data: DiscoverSharesResponse;
30};
31
32function mapError(responseStatus: number, errorBody: { error?: string; code?: string }) {
33  return (
34    errorBody.error ??
35    `Request failed (${responseStatus})${errorBody.code ? ` [${errorBody.code}]` : ""}`
36  );
37}
38
39async function listShares() {
40  const response = await fetch("/api/v1/files/network/shares", {
41    cache: "no-store",
42  });
43
44  if (!response.ok) {
45    const errorBody = (await response.json().catch(() => ({}))) as {
46      error?: string;
47      code?: string;
48    };
49    throw new Error(mapError(response.status, errorBody));
50  }
51
52  const json = (await response.json()) as NetworkSharesResponse;
53  return json.data;
54}
55
56async function createShare(payload: CreateNetworkShareRequest) {
57  const response = await fetch("/api/v1/files/network/shares", {
58    method: "POST",
59    headers: {
60      "Content-Type": "application/json",
61    },
62    body: JSON.stringify(payload),
63  });
64
65  if (!response.ok) {
66    const errorBody = (await response.json().catch(() => ({}))) as {
67      error?: string;
68      code?: string;
69    };
70    throw new Error(mapError(response.status, errorBody));
71  }
72
73  const json = (await response.json()) as NetworkShareResponse;
74  return json.data;
75}
76
77async function deleteShare(shareId: string) {
78  const response = await fetch(`/api/v1/files/network/shares/${encodeURIComponent(shareId)}`, {
79    method: "DELETE",
80  });
81
82  if (!response.ok) {
83    const errorBody = (await response.json().catch(() => ({}))) as {
84      error?: string;
85      code?: string;
86    };
87    throw new Error(mapError(response.status, errorBody));
88  }
89
90  return true;
91}
92
93async function mountShare(shareId: string) {
94  const response = await fetch(
95    `/api/v1/files/network/shares/${encodeURIComponent(shareId)}/mount`,
96    {
97      method: "POST",
98    },
99  );
100
101  if (!response.ok) {
102    const errorBody = (await response.json().catch(() => ({}))) as {
103      error?: string;
104      code?: string;
105    };
106    throw new Error(mapError(response.status, errorBody));
107  }
108
109  const json = (await response.json()) as NetworkShareResponse;
110  return json.data;
111}
112
113async function unmountShare(shareId: string) {
114  const response = await fetch(
115    `/api/v1/files/network/shares/${encodeURIComponent(shareId)}/unmount`,
116    {
117      method: "POST",
118    },
119  );
120
121  if (!response.ok) {
122    const errorBody = (await response.json().catch(() => ({}))) as {
123      error?: string;
124      code?: string;
125    };
126    throw new Error(mapError(response.status, errorBody));
127  }
128
129  const json = (await response.json()) as NetworkShareResponse;
130  return json.data;
131}
132
133async function discoverServers() {
134  const response = await fetch("/api/v1/files/network/discover/servers", {
135    cache: "no-store",
136  });
137
138  if (!response.ok) {
139    const errorBody = (await response.json().catch(() => ({}))) as {
140      error?: string;
141      code?: string;
142    };
143    throw new Error(mapError(response.status, errorBody));
144  }
145
146  const json = (await response.json()) as DiscoverServersApiResponse;
147  return json.data;
148}
149
150async function discoverHostShares(payload: DiscoverSharesRequest) {
151  const response = await fetch("/api/v1/files/network/discover/shares", {
152    method: "POST",
153    headers: {
154      "Content-Type": "application/json",
155    },
156    body: JSON.stringify(payload),
157  });
158
159  if (!response.ok) {
160    const errorBody = (await response.json().catch(() => ({}))) as {
161      error?: string;
162      code?: string;
163    };
164    throw new Error(mapError(response.status, errorBody));
165  }
166
167  const json = (await response.json()) as DiscoverSharesApiResponse;
168  return json.data;
169}
170
171function invalidateNetworkQueries(queryClient: ReturnType<typeof useQueryClient>) {
172  void queryClient.invalidateQueries({
173    queryKey: queryKeys.networkShares,
174  });
175  void queryClient.invalidateQueries({
176    queryKey: queryKeys.filesList("Network"),
177  });
178}
179
180export function useNetworkShares() {
181  return useQuery({
182    queryKey: queryKeys.networkShares,
183    queryFn: listShares,
184    staleTime: 5_000,
185  });
186}
187
188export function useCreateNetworkShare() {
189  const queryClient = useQueryClient();
190
191  return useMutation({
192    mutationFn: createShare,
193    onSuccess: () => {
194      invalidateNetworkQueries(queryClient);
195    },
196  });
197}
198
199export function useRemoveNetworkShare() {
200  const queryClient = useQueryClient();
201
202  return useMutation({
203    mutationFn: deleteShare,
204    onSuccess: () => {
205      invalidateNetworkQueries(queryClient);
206    },
207  });
208}
209
210export function useMountNetworkShare() {
211  const queryClient = useQueryClient();
212
213  return useMutation({
214    mutationFn: mountShare,
215    onSuccess: () => {
216      invalidateNetworkQueries(queryClient);
217    },
218  });
219}
220
221export function useUnmountNetworkShare() {
222  const queryClient = useQueryClient();
223
224  return useMutation({
225    mutationFn: unmountShare,
226    onSuccess: () => {
227      invalidateNetworkQueries(queryClient);
228    },
229  });
230}
231
232export function useDiscoverNetworkServers() {
233  return useMutation({
234    mutationFn: discoverServers,
235  });
236}
237
238export function useDiscoverNetworkShares() {
239  return useMutation({
240    mutationFn: discoverHostShares,
241  });
242}
243