diff --git a/src/components/DownloadAll.tsx b/src/components/DownloadAll.tsx index 5975033..5641f0b 100644 --- a/src/components/DownloadAll.tsx +++ b/src/components/DownloadAll.tsx @@ -1,23 +1,43 @@ -import { DecryptedAudioFile, selectFiles } from '~/features/file-listing/fileListingSlice'; +import { DecryptedAudioFile, ProcessState, selectFiles } from '~/features/file-listing/fileListingSlice'; import { FaDownload } from 'react-icons/fa'; import { useAppSelector } from '~/hooks'; import { toast } from 'react-toastify'; export function DownloadAll() { const files = useAppSelector(selectFiles); - const filesLength = Object.keys(files).length; const onClickDownloadAll = async () => { + if (Object.keys(files).length === 0) { + toast.warning('未添加文件'); + return; + } + + //判断所有文件是否处理完成 + const allComplete = Object.values(files).every((file) => file.state !== ProcessState.PROCESSING); + if (!allComplete) { + toast.warning('请等待所有文件解密完成'); + return; + } + + //过滤处理失败的文件 + const completeFiles = Object.values(files).filter((file) => file.state === ProcessState.COMPLETE); + const filesLength = Object.keys(completeFiles).length; + + //开始下载 let dir: FileSystemDirectoryHandle | undefined; let success = 0; try { - dir = await window.showDirectoryPicker(); + dir = await window.showDirectoryPicker({ mode: 'readwrite' }); } catch (e) { console.error(e); if (e instanceof Error && e.name === 'AbortError') { return; } } - for (const [_, file] of Object.entries(files)) { + toast.warning('开始下载,请稍候'); + for (const [_, file] of Object.entries(completeFiles)) { + if (file.state !== ProcessState.COMPLETE) { + return; + } try { if (dir) { await DownloadNew(dir, file); @@ -33,7 +53,7 @@ export function DownloadAll() { if (success === filesLength) { toast.success(`成功下载: ${success}/${filesLength}首`); } else { - toast.error(`成功下载: ${success}/${filesLength}首`); + toast.warning(`成功下载: ${success}/${filesLength}首`); } };