eliott/Nimbus
1
0
Fork 0
mirror of https://github.com/nimbusdotstorage/Nimbus synced 2026-04-22 17:45:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

IT WORKS!!!!

Browse Source
This commit is contained in:
David Bauch
2025-07-16 14:51:16 -04:00
parent 71404cecc8
commit 5fe68df251
16 changed files with 152 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
apps/web/src/hooks/useFileOperations.ts
View File

@@ -6,16 +6,17 @@ import type {
UpdateFileSchema,
UploadFileSchema,
} from "@nimbus/shared";
import { useUserInfoProvider } from "@/components/providers/user-info-provider";
import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { protectedClient } from "@/utils/client";
import type { DriveProviderClient } from "@/utils/client";
import { toast } from "sonner";
const BASE_FILE_CLIENT = protectedClient.api.files;
export function useGetFiles({ parentId, pageSize, pageToken, returnedValues }: GetFilesSchema) {
const { clientPromise } = useUserInfoProvider();
return useQuery({
queryKey: ["files", parentId, pageSize, pageToken],
queryFn: async () => {
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT.$get({
query: { parentId, pageSize: pageSize.toString(), pageToken, returnedValues },
});
@@ -27,10 +28,12 @@ export function useGetFiles({ parentId, pageSize, pageToken, returnedValues }: G
}
export function useGetFile({ fileId, returnedValues }: GetFileByIdSchema) {
const { clientPromise } = useUserInfoProvider();
return useQuery({
queryKey: ["file", fileId, returnedValues],
queryFn: async () => {
const response = await protectedClient.api.files[":id"].$get({
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT[":id"].$get({
param: { fileId },
query: { returnedValues },
});
@@ -43,8 +46,10 @@ export function useGetFile({ fileId, returnedValues }: GetFileByIdSchema) {
export function useDeleteFile() {
const queryClient = useQueryClient();
const { clientPromise } = useUserInfoProvider();
return useMutation({
mutationFn: async ({ fileId }: DeleteFileSchema) => {
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT.$delete({
query: { fileId },
});
@@ -63,8 +68,10 @@ export function useDeleteFile() {
export function useUpdateFile() {
const queryClient = useQueryClient();
const { clientPromise } = useUserInfoProvider();
return useMutation({
mutationFn: async ({ fileId, name }: UpdateFileSchema) => {
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT.$put({
query: { fileId, name },
});
@@ -84,8 +91,10 @@ export function useUpdateFile() {
export function useCreateFolder() {
const queryClient = useQueryClient();
const { clientPromise } = useUserInfoProvider();
return useMutation({
mutationFn: async ({ name, mimeType, parent }: CreateFileSchema) => {
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT.$post({
query: {
name,
@@ -109,10 +118,11 @@ export function useCreateFolder() {
export function useUploadFile() {
const queryClient = useQueryClient();
const { clientPromise } = useUserInfoProvider();
return useMutation({
// mutationFn: async ({ file, parentId, onProgress }: UploadFileParams) => {
mutationFn: async ({ file, parentId }: UploadFileSchema) => {
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT.upload.$post({
form: {
file,
@@ -145,6 +155,7 @@ export function useUploadFile() {
export function useUploadFolder() {}
export function useDownloadFile() {
const { clientPromise } = useUserInfoProvider();
return useMutation({
mutationFn: async ({
fileId,
@@ -157,6 +168,7 @@ export function useDownloadFile() {
fileName?: string;
onProgress?: (progress: number) => void;
}) => {
const BASE_FILE_CLIENT = await getBaseFileClient(clientPromise);
const response = await BASE_FILE_CLIENT.download.$get({
query: {
fileId,
@@ -237,3 +249,9 @@ export function useDownloadFile() {
},
});
}
async function getBaseFileClient(clientPromise: Promise<DriveProviderClient>) {
const client = await clientPromise;
const BASE_FILE_CLIENT = client.api.files;
return BASE_FILE_CLIENT;
}