homeio

1"use client";
2
3import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
4import { queryKeys } from "@/lib/shared/query-keys";
5import type { DriveBrowseResponse, DriveFileEntry, DriveUploadResponse } from "@/lib/shared/contracts/google-drive";
6
7type ApiResponse<T> = { data: T };
8type ApiError = { error?: string; code?: string };
9
10function mapError(status: number, body: ApiError) {
11  return body.error ?? `Request failed (${status})${body.code ? ` [${body.code}]` : ""}`;
12}
13
14async function fetchDriveBrowse(
15  connectionId: string,
16  folderId: string,
17): Promise<DriveBrowseResponse> {
18  const params = new URLSearchParams({ folderId });
19  const res = await fetch(
20    `/api/v1/files/google-drive/${encodeURIComponent(connectionId)}/browse?${params.toString()}`,
21    { cache: "no-store" },
22  );
23  if (!res.ok) {
24    const body = (await res.json().catch(() => ({}))) as ApiError;
25    throw new Error(mapError(res.status, body));
26  }
27  const json = (await res.json()) as ApiResponse<DriveBrowseResponse>;
28  return json.data;
29}
30
31async function deleteDriveFile(connectionId: string, fileId: string): Promise<void> {
32  const res = await fetch(
33    `/api/v1/files/google-drive/${encodeURIComponent(connectionId)}/files/${encodeURIComponent(fileId)}`,
34    { method: "DELETE" },
35  );
36  if (!res.ok) {
37    const body = (await res.json().catch(() => ({}))) as ApiError;
38    throw new Error(mapError(res.status, body));
39  }
40}
41
42async function createDriveFolder(
43  connectionId: string,
44  parentId: string,
45  name: string,
46): Promise<DriveFileEntry> {
47  const res = await fetch(
48    `/api/v1/files/google-drive/${encodeURIComponent(connectionId)}/folders`,
49    {
50      method: "POST",
51      headers: { "Content-Type": "application/json" },
52      body: JSON.stringify({ parentId, name }),
53    },
54  );
55  if (!res.ok) {
56    const body = (await res.json().catch(() => ({}))) as ApiError;
57    throw new Error(mapError(res.status, body));
58  }
59  const json = (await res.json()) as ApiResponse<DriveFileEntry>;
60  return json.data;
61}
62
63async function uploadDriveFiles(
64  connectionId: string,
65  folderId: string,
66  files: File[],
67  onProgress?: (loaded: number, total: number) => void,
68): Promise<DriveUploadResponse> {
69  const formData = new FormData();
70  formData.append("folderId", folderId);
71  let totalSize = 0;
72  for (const file of files) {
73    formData.append("file", file);
74    totalSize += file.size;
75  }
76
77  return new Promise((resolve, reject) => {
78    const xhr = new XMLHttpRequest();
79    xhr.open("POST", `/api/v1/files/google-drive/${encodeURIComponent(connectionId)}/upload`);
80
81    if (onProgress) {
82      xhr.upload.addEventListener("progress", (e) => {
83        if (e.lengthComputable) onProgress(e.loaded, e.total);
84        else onProgress(0, totalSize);
85      });
86    }
87
88    xhr.addEventListener("load", () => {
89      if (xhr.status >= 200 && xhr.status < 300) {
90        try {
91          const json = JSON.parse(xhr.responseText) as ApiResponse<DriveUploadResponse>;
92          resolve(json.data);
93        } catch {
94          reject(new Error("Invalid response from upload"));
95        }
96      } else {
97        try {
98          const body = JSON.parse(xhr.responseText) as ApiError;
99          reject(new Error(mapError(xhr.status, body)));
100        } catch {
101          reject(new Error(`Upload failed: ${xhr.status}`));
102        }
103      }
104    });
105
106    xhr.addEventListener("error", () => reject(new Error("Network error during upload")));
107    xhr.send(formData);
108  });
109}
110
111export function buildDriveDownloadUrl(
112  connectionId: string,
113  fileId: string,
114  name: string,
115  mimeType: string,
116): string {
117  const params = new URLSearchParams({ fileId, name, mimeType });
118  return `/api/v1/files/google-drive/${encodeURIComponent(connectionId)}/download?${params.toString()}`;
119}
120
121export function useGoogleDriveFiles(connectionId: string, folderId: string) {
122  return useQuery({
123    queryKey: queryKeys.googleDriveBrowse(connectionId, folderId),
124    queryFn: () => fetchDriveBrowse(connectionId, folderId),
125    staleTime: 10_000,
126    enabled: Boolean(connectionId),
127  });
128}
129
130export function useGoogleDriveDeleteFile(connectionId: string, folderId: string) {
131  const queryClient = useQueryClient();
132  return useMutation({
133    mutationFn: (fileId: string) => deleteDriveFile(connectionId, fileId),
134    onSuccess: () => {
135      void queryClient.invalidateQueries({
136        queryKey: queryKeys.googleDriveBrowse(connectionId, folderId),
137      });
138    },
139  });
140}
141
142export function useGoogleDriveCreateFolder(connectionId: string, folderId: string) {
143  const queryClient = useQueryClient();
144  return useMutation({
145    mutationFn: ({ parentId, name }: { parentId: string; name: string }) =>
146      createDriveFolder(connectionId, parentId, name),
147    onSuccess: () => {
148      void queryClient.invalidateQueries({
149        queryKey: queryKeys.googleDriveBrowse(connectionId, folderId),
150      });
151    },
152  });
153}
154
155export function useGoogleDriveUpload(connectionId: string, folderId: string) {
156  const queryClient = useQueryClient();
157  return useMutation({
158    mutationFn: ({
159      files,
160      onProgress,
161    }: {
162      files: File[];
163      onProgress?: (loaded: number, total: number) => void;
164    }) => uploadDriveFiles(connectionId, folderId, files, onProgress),
165    onSuccess: () => {
166      void queryClient.invalidateQueries({
167        queryKey: queryKeys.googleDriveBrowse(connectionId, folderId),
168      });
169    },
170  });
171}
172