102 lines
2.7 KiB
TypeScript
102 lines
2.7 KiB
TypeScript
'use client'
|
|
|
|
import { useState, useRef } from 'react';
|
|
import { replaceFile, uploadFile } from '@/lib/hooks';
|
|
import { toast } from 'sonner';
|
|
import { useAuth } from '@/components/context/auth';
|
|
import { resizeImage } from '@/lib/hooks';
|
|
|
|
export type Replace =
|
|
| { replace: true, path: string }
|
|
| false;
|
|
|
|
export type uploadToStorageProps = {
|
|
file: File;
|
|
bucket: string;
|
|
resize: boolean;
|
|
options?: {
|
|
maxWidth?: number;
|
|
maxHeight?: number;
|
|
quality?: number;
|
|
};
|
|
replace?: Replace;
|
|
};
|
|
|
|
export const useFileUpload = () => {
|
|
const [isUploading, setIsUploading] = useState(false);
|
|
const fileInputRef = useRef<HTMLInputElement>(null);
|
|
const { profile, isAuthenticated } = useAuth();
|
|
|
|
const uploadToStorage = async ({
|
|
file,
|
|
bucket,
|
|
resize = false,
|
|
options = {},
|
|
replace = false,
|
|
}: uploadToStorageProps) => {
|
|
try {
|
|
if (!isAuthenticated) throw new Error('User is not authenticated');
|
|
|
|
setIsUploading(true);
|
|
if (replace) {
|
|
const updateResult = await replaceFile({
|
|
bucket,
|
|
path: replace.path,
|
|
file,
|
|
options: {
|
|
contentType: file.type,
|
|
},
|
|
});
|
|
if (!updateResult.success) {
|
|
console.error('Error updating file:', updateResult.error);
|
|
} else {
|
|
console.log('We used the new update function hopefully it worked!');
|
|
return { success: true, path: updateResult.data };
|
|
}
|
|
}
|
|
|
|
let fileToUpload = file;
|
|
if (resize && file.type.startsWith('image/'))
|
|
fileToUpload = await resizeImage({ file, options });
|
|
|
|
// Generate a unique filename to avoid collisions
|
|
const fileExt = file.name.split('.').pop();
|
|
const fileName = `${Date.now()}-${profile?.id}.${fileExt}`;
|
|
|
|
// Upload the file to Supabase storage
|
|
const uploadResult = await uploadFile({
|
|
bucket,
|
|
path: fileName,
|
|
file: fileToUpload,
|
|
options: {
|
|
contentType: file.type,
|
|
},
|
|
});
|
|
|
|
if (!uploadResult.success) {
|
|
throw new Error(uploadResult.error || `Failed to upload to ${bucket}`);
|
|
}
|
|
|
|
return { success: true, path: uploadResult.data };
|
|
} catch (error) {
|
|
console.error(`Error uploading to ${bucket}:`, error);
|
|
toast.error(
|
|
error instanceof Error
|
|
? error.message
|
|
: `Failed to upload to ${bucket}`,
|
|
);
|
|
return { success: false, error };
|
|
} finally {
|
|
setIsUploading(false);
|
|
// Clear the input value so the same file can be selected again
|
|
if (fileInputRef.current) fileInputRef.current.value = '';
|
|
}
|
|
};
|
|
|
|
return {
|
|
isUploading,
|
|
fileInputRef,
|
|
uploadToStorage,
|
|
};
|
|
};
|