homeio

1"use client";
2
3import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
4import { queryKeys } from "@/lib/shared/query-keys";
5
6export type GoogleDriveConnection = {
7  id: string;
8  email: string;
9  displayName: string | null;
10  connectedAt: string;
11};
12
13type ConnectionsApiResponse = { data: GoogleDriveConnection[]; configured: boolean };
14
15function mapError(status: number, body: { error?: string; code?: string }) {
16  return body.error ?? `Request failed (${status})${body.code ? ` [${body.code}]` : ""}`;
17}
18
19type ConnectionsResult = { connections: GoogleDriveConnection[]; configured: boolean };
20
21async function listConnections(): Promise<ConnectionsResult> {
22  const res = await fetch("/api/v1/files/google-drive/connections", { cache: "no-store" });
23  if (!res.ok) {
24    const body = (await res.json().catch(() => ({}))) as { error?: string; code?: string };
25    throw new Error(mapError(res.status, body));
26  }
27  const json = (await res.json()) as ConnectionsApiResponse;
28  return { connections: json.data, configured: json.configured };
29}
30
31async function removeConnection(id: string): Promise<void> {
32  const res = await fetch(`/api/v1/files/google-drive/connections/${encodeURIComponent(id)}`, {
33    method: "DELETE",
34  });
35  if (!res.ok) {
36    const body = (await res.json().catch(() => ({}))) as { error?: string; code?: string };
37    throw new Error(mapError(res.status, body));
38  }
39}
40
41export function useGoogleDriveConnections() {
42  return useQuery({
43    queryKey: queryKeys.googleDriveConnections,
44    queryFn: listConnections,
45    staleTime: 5_000,
46  });
47}
48
49export function useRemoveGoogleDriveConnection() {
50  const queryClient = useQueryClient();
51  return useMutation({
52    mutationFn: removeConnection,
53    onSuccess: () => {
54      void queryClient.invalidateQueries({ queryKey: queryKeys.googleDriveConnections });
55    },
56  });
57}
58