刷新续传
This commit is contained in:
mtvpls
@@ -1,8 +1,9 @@
|
||||
'use client';
|
||||
|
||||
import React, { createContext, useCallback, useContext,useState } from 'react';
|
||||
import React, { createContext, useCallback, useContext, useState, useEffect } from 'react';
|
||||
|
||||
import { M3U8Downloader, M3U8DownloadTask } from '@/lib/m3u8-downloader';
|
||||
import Toast from '@/components/Toast';
|
||||
|
||||
interface DownloadContextType {
|
||||
downloader: M3U8Downloader;
|
||||
@@ -33,6 +34,7 @@ export function DownloadProvider({ children }: { children: React.ReactNode }) {
|
||||
const [tasks, setTasks] = useState<M3U8DownloadTask[]>([]);
|
||||
const [showDownloadPanel, setShowDownloadPanel] = useState(false);
|
||||
const [startingTaskIds, setStartingTaskIds] = useState<Set<string>>(new Set());
|
||||
const [toast, setToast] = useState<{ message: string; type: 'success' | 'error' | 'info' } | null>(null);
|
||||
|
||||
// 自动启动下一个等待的任务
|
||||
const startNextPendingTask = useCallback((currentDownloader: M3U8Downloader) => {
|
||||
@@ -57,20 +59,155 @@ export function DownloadProvider({ children }: { children: React.ReactNode }) {
|
||||
const [downloader] = useState(() => new M3U8Downloader({
|
||||
onProgress: (task) => {
|
||||
setTasks(downloader.getAllTasks());
|
||||
// 保存任务状态
|
||||
saveTasks(downloader.getAllTasks());
|
||||
},
|
||||
onComplete: (task) => {
|
||||
setTasks(downloader.getAllTasks());
|
||||
// 保存任务状态
|
||||
saveTasks(downloader.getAllTasks());
|
||||
// 任务完成后，尝试启动下一个等待的任务
|
||||
startNextPendingTask(downloader);
|
||||
},
|
||||
onError: (task, error) => {
|
||||
console.error('下载错误:', error);
|
||||
setTasks(downloader.getAllTasks());
|
||||
// 保存任务状态
|
||||
saveTasks(downloader.getAllTasks());
|
||||
// 任务出错后，尝试启动下一个等待的任务
|
||||
startNextPendingTask(downloader);
|
||||
},
|
||||
}));
|
||||
|
||||
// 保存任务到 localStorage
|
||||
const saveTasks = useCallback((tasks: M3U8DownloadTask[]) => {
|
||||
if (typeof window === 'undefined') return;
|
||||
|
||||
try {
|
||||
// 只保存必要的信息，不保存 ArrayBuffer 等无法序列化的数据
|
||||
const tasksToSave = tasks.map(task => ({
|
||||
id: task.id,
|
||||
url: task.url,
|
||||
title: task.title,
|
||||
type: task.type,
|
||||
status: task.status,
|
||||
finishList: task.finishList,
|
||||
downloadIndex: task.downloadIndex,
|
||||
finishNum: task.finishNum,
|
||||
errorNum: task.errorNum,
|
||||
source: task.source,
|
||||
videoId: task.videoId,
|
||||
episodeIndex: task.episodeIndex,
|
||||
downloadMode: task.downloadMode,
|
||||
rangeDownload: task.rangeDownload,
|
||||
m3u8Content: task.m3u8Content,
|
||||
}));
|
||||
|
||||
localStorage.setItem('downloadTasks', JSON.stringify(tasksToSave));
|
||||
} catch (error) {
|
||||
console.error('保存任务失败:', error);
|
||||
}
|
||||
}, []);
|
||||
|
||||
// 从 localStorage 恢复任务
|
||||
useEffect(() => {
|
||||
if (typeof window === 'undefined') return;
|
||||
|
||||
const restoreTasks = async () => {
|
||||
try {
|
||||
const savedTasks = localStorage.getItem('downloadTasks');
|
||||
if (!savedTasks) return;
|
||||
|
||||
const tasks = JSON.parse(savedTasks);
|
||||
|
||||
// 读取下载模式和目录句柄
|
||||
const downloadMode = localStorage.getItem('downloadMode') as 'browser' | 'filesystem' || 'browser';
|
||||
let dirHandle: FileSystemDirectoryHandle | undefined;
|
||||
|
||||
if (downloadMode === 'filesystem') {
|
||||
const dbName = 'MoonTVPlus';
|
||||
const storeName = 'dirHandles';
|
||||
|
||||
dirHandle = await new Promise<FileSystemDirectoryHandle | undefined>((resolve) => {
|
||||
const request = indexedDB.open(dbName, 1);
|
||||
|
||||
request.onsuccess = (event) => {
|
||||
const db = (event.target as IDBOpenDBRequest).result;
|
||||
|
||||
if (!db.objectStoreNames.contains(storeName)) {
|
||||
db.close();
|
||||
resolve(undefined);
|
||||
return;
|
||||
}
|
||||
|
||||
const transaction = db.transaction([storeName], 'readonly');
|
||||
const store = transaction.objectStore(storeName);
|
||||
const getRequest = store.get('downloadDir');
|
||||
|
||||
getRequest.onsuccess = () => {
|
||||
const handle = getRequest.result as FileSystemDirectoryHandle | undefined;
|
||||
db.close();
|
||||
resolve(handle);
|
||||
};
|
||||
|
||||
getRequest.onerror = () => {
|
||||
db.close();
|
||||
resolve(undefined);
|
||||
};
|
||||
};
|
||||
|
||||
request.onerror = () => {
|
||||
resolve(undefined);
|
||||
};
|
||||
});
|
||||
}
|
||||
|
||||
// 恢复任务
|
||||
for (const savedTask of tasks) {
|
||||
// 只恢复未完成的任务
|
||||
if (savedTask.status === 'downloading' || savedTask.status === 'pause' || savedTask.status === 'ready') {
|
||||
try {
|
||||
const taskId = await downloader.createTask(
|
||||
savedTask.url,
|
||||
savedTask.title,
|
||||
savedTask.type,
|
||||
{
|
||||
source: savedTask.source,
|
||||
videoId: savedTask.videoId,
|
||||
episodeIndex: savedTask.episodeIndex,
|
||||
}
|
||||
);
|
||||
|
||||
const task = downloader.getTask(taskId);
|
||||
if (task) {
|
||||
// 恢复任务状态
|
||||
task.status = savedTask.status === 'downloading' ? 'pause' : savedTask.status; // 将 downloading 改为 pause
|
||||
task.finishList = savedTask.finishList;
|
||||
task.downloadIndex = savedTask.downloadIndex;
|
||||
task.finishNum = savedTask.finishNum;
|
||||
task.errorNum = savedTask.errorNum;
|
||||
task.downloadMode = savedTask.downloadMode;
|
||||
task.rangeDownload = savedTask.rangeDownload;
|
||||
|
||||
if (dirHandle) {
|
||||
task.filesystemDirHandle = dirHandle;
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('恢复任务失败:', savedTask.title, error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
setTasks(downloader.getAllTasks());
|
||||
} catch (error) {
|
||||
console.error('恢复任务失败:', error);
|
||||
}
|
||||
};
|
||||
|
||||
restoreTasks();
|
||||
}, [downloader]);
|
||||
|
||||
const addDownloadTask = useCallback(async (
|
||||
url: string,
|
||||
title: string,
|
||||
@@ -82,13 +219,80 @@ export function DownloadProvider({ children }: { children: React.ReactNode }) {
|
||||
}
|
||||
) => {
|
||||
try {
|
||||
const taskId = await downloader.createTask(url, title, type, metadata);
|
||||
|
||||
// 读取下载模式设置
|
||||
const downloadMode = typeof window !== 'undefined'
|
||||
? (localStorage.getItem('downloadMode') as 'browser' | 'filesystem') || 'browser'
|
||||
: 'browser';
|
||||
|
||||
// 如果是 filesystem 模式，检查是否已经下载过
|
||||
if (downloadMode === 'filesystem' && typeof window !== 'undefined' && metadata?.source && metadata?.videoId && metadata?.episodeIndex !== undefined) {
|
||||
try {
|
||||
const dbName = 'MoonTVPlus';
|
||||
const storeName = 'dirHandles';
|
||||
|
||||
const alreadyDownloaded = await new Promise<boolean>((resolve) => {
|
||||
const request = indexedDB.open(dbName, 1);
|
||||
|
||||
request.onsuccess = async (event) => {
|
||||
const db = (event.target as IDBOpenDBRequest).result;
|
||||
|
||||
if (!db.objectStoreNames.contains(storeName)) {
|
||||
db.close();
|
||||
resolve(false);
|
||||
return;
|
||||
}
|
||||
|
||||
const transaction = db.transaction([storeName], 'readonly');
|
||||
const store = transaction.objectStore(storeName);
|
||||
const getRequest = store.get('downloadDir');
|
||||
|
||||
getRequest.onsuccess = async () => {
|
||||
const dirHandle = getRequest.result as FileSystemDirectoryHandle | undefined;
|
||||
db.close();
|
||||
|
||||
if (!dirHandle) {
|
||||
resolve(false);
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
// 检查子目录和 playlist.m3u8 是否存在
|
||||
const sourceDirHandle = await dirHandle.getDirectoryHandle(metadata.source!, { create: false });
|
||||
const videoIdDirHandle = await sourceDirHandle.getDirectoryHandle(metadata.videoId!, { create: false });
|
||||
const epDirHandle = await videoIdDirHandle.getDirectoryHandle(`ep${metadata.episodeIndex! + 1}`, { create: false });
|
||||
|
||||
// 只检查 playlist.m3u8 是否存在（它是最后生成的）
|
||||
await epDirHandle.getFileHandle('playlist.m3u8', { create: false });
|
||||
resolve(true);
|
||||
} catch {
|
||||
// 目录或文件不存在
|
||||
resolve(false);
|
||||
}
|
||||
};
|
||||
|
||||
getRequest.onerror = () => {
|
||||
db.close();
|
||||
resolve(false);
|
||||
};
|
||||
};
|
||||
|
||||
request.onerror = () => {
|
||||
resolve(false);
|
||||
};
|
||||
});
|
||||
|
||||
if (alreadyDownloaded) {
|
||||
console.log('视频已下载，跳过:', title);
|
||||
setToast({ message: `${title} 已经下载过了，无需重复下载`, type: 'info' });
|
||||
return;
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('检查下载状态失败:', error);
|
||||
}
|
||||
}
|
||||
|
||||
const taskId = await downloader.createTask(url, title, type, metadata);
|
||||
|
||||
// 如果是 filesystem 模式，从 IndexedDB 读取目录句柄
|
||||
if (downloadMode === 'filesystem' && typeof window !== 'undefined') {
|
||||
try {
|
||||
@@ -250,6 +454,13 @@ export function DownloadProvider({ children }: { children: React.ReactNode }) {
|
||||
}}
|
||||
>
|
||||
{children}
|
||||
{toast && (
|
||||
<Toast
|
||||
message={toast.message}
|
||||
type={toast.type}
|
||||
onClose={() => setToast(null)}
|
||||
/>
|
||||
)}
|
||||
</DownloadContext.Provider>
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user