|
| 1 | +import { useState, useCallback } from 'react'; |
| 2 | + |
| 3 | +const ACCEPTED_TYPES = ['image/jpeg', 'image/png', 'image/gif', 'image/webp', 'application/pdf', 'text/plain', 'text/csv']; |
| 4 | +const MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB |
| 5 | +const MAX_FILES = 10; |
| 6 | + |
| 7 | +function validateFile(file) { |
| 8 | + if (!ACCEPTED_TYPES.includes(file.type)) return `"${file.name}": unsupported file type`; |
| 9 | + if (file.size > MAX_FILE_SIZE) return `"${file.name}": exceeds 10MB limit`; |
| 10 | + return null; |
| 11 | +} |
| 12 | + |
| 13 | +async function compressImage(file) { |
| 14 | + if (!file.type.startsWith('image/')) return file; |
| 15 | + return new Promise((resolve) => { |
| 16 | + const img = new Image(); |
| 17 | + const url = URL.createObjectURL(file); |
| 18 | + img.onload = () => { |
| 19 | + const canvas = document.createElement('canvas'); |
| 20 | + const scale = Math.min(1, 1200 / Math.max(img.width, img.height)); |
| 21 | + canvas.width = img.width * scale; |
| 22 | + canvas.height = img.height * scale; |
| 23 | + canvas.getContext('2d').drawImage(img, 0, 0, canvas.width, canvas.height); |
| 24 | + URL.revokeObjectURL(url); |
| 25 | + canvas.toBlob((blob) => resolve(new File([blob], file.name, { type: 'image/jpeg' })), 'image/jpeg', 0.85); |
| 26 | + }; |
| 27 | + img.onerror = () => { URL.revokeObjectURL(url); resolve(file); }; |
| 28 | + img.src = url; |
| 29 | + }); |
| 30 | +} |
| 31 | + |
| 32 | +export function useFileUpload(uploadFn) { |
| 33 | + const [files, setFiles] = useState([]); // { id, file, preview, status, progress, error } |
| 34 | + const [errors, setErrors] = useState([]); |
| 35 | + |
| 36 | + const addFiles = useCallback((incoming) => { |
| 37 | + const list = Array.from(incoming); |
| 38 | + const validationErrors = []; |
| 39 | + const valid = []; |
| 40 | + |
| 41 | + for (const file of list) { |
| 42 | + const err = validateFile(file); |
| 43 | + if (err) { validationErrors.push(err); continue; } |
| 44 | + if (files.length + valid.length >= MAX_FILES) { |
| 45 | + validationErrors.push(`Max ${MAX_FILES} files allowed`); |
| 46 | + break; |
| 47 | + } |
| 48 | + const preview = file.type.startsWith('image/') ? URL.createObjectURL(file) : null; |
| 49 | + valid.push({ id: `${Date.now()}-${Math.random()}`, file, preview, status: 'pending', progress: 0, error: null }); |
| 50 | + } |
| 51 | + |
| 52 | + setErrors(validationErrors); |
| 53 | + if (valid.length) setFiles((prev) => [...prev, ...valid]); |
| 54 | + }, [files.length]); |
| 55 | + |
| 56 | + const removeFile = useCallback((id) => { |
| 57 | + setFiles((prev) => { |
| 58 | + const entry = prev.find((f) => f.id === id); |
| 59 | + if (entry?.preview) URL.revokeObjectURL(entry.preview); |
| 60 | + return prev.filter((f) => f.id !== id); |
| 61 | + }); |
| 62 | + }, []); |
| 63 | + |
| 64 | + const uploadAll = useCallback(async () => { |
| 65 | + const pending = files.filter((f) => f.status === 'pending'); |
| 66 | + if (!pending.length || !uploadFn) return; |
| 67 | + |
| 68 | + for (const entry of pending) { |
| 69 | + setFiles((prev) => prev.map((f) => f.id === entry.id ? { ...f, status: 'uploading', progress: 0 } : f)); |
| 70 | + try { |
| 71 | + const compressed = await compressImage(entry.file); |
| 72 | + await uploadFn(compressed, (progress) => { |
| 73 | + setFiles((prev) => prev.map((f) => f.id === entry.id ? { ...f, progress } : f)); |
| 74 | + }); |
| 75 | + setFiles((prev) => prev.map((f) => f.id === entry.id ? { ...f, status: 'done', progress: 100 } : f)); |
| 76 | + } catch (err) { |
| 77 | + setFiles((prev) => prev.map((f) => f.id === entry.id ? { ...f, status: 'error', error: err.message } : f)); |
| 78 | + } |
| 79 | + } |
| 80 | + }, [files, uploadFn]); |
| 81 | + |
| 82 | + const clearAll = useCallback(() => { |
| 83 | + setFiles((prev) => { prev.forEach((f) => f.preview && URL.revokeObjectURL(f.preview)); return []; }); |
| 84 | + setErrors([]); |
| 85 | + }, []); |
| 86 | + |
| 87 | + return { files, errors, addFiles, removeFile, uploadAll, clearAll }; |
| 88 | +} |
0 commit comments