diff --git a/frontend/src/renderer/hooks/useFiles.js b/frontend/src/renderer/hooks/useFiles.js new file mode 100644 index 0000000..5df0ed1 --- /dev/null +++ b/frontend/src/renderer/hooks/useFiles.js @@ -0,0 +1,34 @@ +import { useState, useCallback } from 'react'; +import api from '../utils/api'; + +export const useFiles = () => { + const [files, setFiles] = useState([]); + const [loading, setLoading] = useState(false); + + const fetchFiles = useCallback(async (folderId = null) => { + setLoading(true); + try { + const url = folderId ? `/api/files?parentId=${folderId}` : '/api/files'; + const data = await api.get(url); + setFiles(data.files || []); + } catch (error) { + console.error('Failed to fetch files:', error); + } finally { + setLoading(false); + } + }, []); + + const uploadFile = async (file) => { + await api.upload('/api/files/upload', file); + await fetchFiles(); + }; + + const deleteFile = async (id) => { + await api.delete(`/api/files/${id}`); + await fetchFiles(); + }; + + return { files, loading, fetchFiles, uploadFile, deleteFile }; +}; + +export default useFiles;