|
|
|
|
@ -0,0 +1,440 @@
|
|
|
|
|
using DownKyi.Core.Aria2cNet;
|
|
|
|
|
using DownKyi.Core.Aria2cNet.Client;
|
|
|
|
|
using DownKyi.Core.Aria2cNet.Client.Entity;
|
|
|
|
|
using DownKyi.Core.Aria2cNet.Server;
|
|
|
|
|
using DownKyi.Core.BiliApi.VideoStream.Models;
|
|
|
|
|
using DownKyi.Core.Logging;
|
|
|
|
|
using DownKyi.Core.Settings;
|
|
|
|
|
using DownKyi.Core.Utils;
|
|
|
|
|
using DownKyi.Models;
|
|
|
|
|
using DownKyi.Utils;
|
|
|
|
|
using DownKyi.ViewModels.DownloadManager;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Collections.ObjectModel;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
namespace DownKyi.Services.Download
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 音视频采用Aria下载，其余采用WebClient下载
|
|
|
|
|
/// </summary>
|
|
|
|
|
public class CustomAriaDownloadService : DownloadService, IDownloadService
|
|
|
|
|
{
|
|
|
|
|
public CustomAriaDownloadService(ObservableCollection<DownloadingItem> downloadingList, ObservableCollection<DownloadedItem> downloadedList) : base(downloadingList, downloadedList)
|
|
|
|
|
{
|
|
|
|
|
Tag = "AriaDownloadService";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#region 音视频
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 下载音频，返回下载文件路径
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
public override string DownloadAudio(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
PlayUrlDashVideo downloadAudio = BaseDownloadAudio(downloading);
|
|
|
|
|
|
|
|
|
|
return DownloadVideo(downloading, downloadAudio);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 下载视频，返回下载文件路径
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
public override string DownloadVideo(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
PlayUrlDashVideo downloadVideo = BaseDownloadVideo(downloading);
|
|
|
|
|
|
|
|
|
|
return DownloadVideo(downloading, downloadVideo);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 将下载音频和视频的函数中相同代码抽象出来
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <param name="downloadVideo"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
private string DownloadVideo(DownloadingItem downloading, PlayUrlDashVideo downloadVideo)
|
|
|
|
|
{
|
|
|
|
|
// 如果为空，说明没有匹配到可下载的音频视频
|
|
|
|
|
if (downloadVideo == null) { return null; }
|
|
|
|
|
|
|
|
|
|
// 下载链接
|
|
|
|
|
List<string> urls = new List<string>();
|
|
|
|
|
if (downloadVideo.BaseUrl != null) { urls.Add(downloadVideo.BaseUrl); }
|
|
|
|
|
if (downloadVideo.BackupUrl != null) { urls.AddRange(downloadVideo.BackupUrl); }
|
|
|
|
|
|
|
|
|
|
// 路径
|
|
|
|
|
downloading.DownloadBase.FilePath = downloading.DownloadBase.FilePath.Replace("\\", "/");
|
|
|
|
|
string[] temp = downloading.DownloadBase.FilePath.Split('/');
|
|
|
|
|
//string path = downloading.DownloadBase.FilePath.Replace(temp[temp.Length - 1], "");
|
|
|
|
|
string path = downloading.DownloadBase.FilePath.TrimEnd(temp[temp.Length - 1].ToCharArray());
|
|
|
|
|
|
|
|
|
|
// 下载文件名
|
|
|
|
|
string fileName = Guid.NewGuid().ToString("N");
|
|
|
|
|
string key = $"{downloadVideo.Id}_{downloadVideo.Codecs}";
|
|
|
|
|
|
|
|
|
|
// 老版本数据库没有这一项，会变成null
|
|
|
|
|
if (downloading.Downloading.DownloadedFiles == null)
|
|
|
|
|
{
|
|
|
|
|
downloading.Downloading.DownloadedFiles = new List<string>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (downloading.Downloading.DownloadFiles.ContainsKey(key))
|
|
|
|
|
{
|
|
|
|
|
// 如果存在，表示下载过，
|
|
|
|
|
// 则继续使用上次下载的文件名
|
|
|
|
|
fileName = downloading.Downloading.DownloadFiles[key];
|
|
|
|
|
|
|
|
|
|
// 还要检查一下文件有没有被人删掉，删掉的话重新下载
|
|
|
|
|
// 如果下载视频之后音频文件被人删了。此时gid还是视频的，会下错文件
|
|
|
|
|
if (downloading.Downloading.DownloadedFiles.Contains(key) && File.Exists(Path.Combine(path, fileName)))
|
|
|
|
|
{
|
|
|
|
|
return Path.Combine(path, fileName);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// 记录本次下载的文件
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
downloading.Downloading.DownloadFiles.Add(key, fileName);
|
|
|
|
|
}
|
|
|
|
|
catch (ArgumentException) { }
|
|
|
|
|
// Gid最好能是每个文件单独存储，现在复用有可能会混
|
|
|
|
|
// 不过好消息是下载是按固定顺序的，而且下载了两个音频会混流不过
|
|
|
|
|
downloading.Downloading.Gid = null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 启用https
|
|
|
|
|
AllowStatus useSSL = SettingsManager.GetInstance().UseSSL();
|
|
|
|
|
if (useSSL == AllowStatus.YES)
|
|
|
|
|
{
|
|
|
|
|
for (int i = 0; i < urls.Count; i++)
|
|
|
|
|
{
|
|
|
|
|
string url = urls[i];
|
|
|
|
|
if (url.StartsWith("http://"))
|
|
|
|
|
{
|
|
|
|
|
urls[i] = url.Replace("http://", "https://");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
for (int i = 0; i < urls.Count; i++)
|
|
|
|
|
{
|
|
|
|
|
string url = urls[i];
|
|
|
|
|
if (url.StartsWith("https://"))
|
|
|
|
|
{
|
|
|
|
|
urls[i] = url.Replace("https://", "http://");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 开始下载
|
|
|
|
|
DownloadResult downloadStatus = DownloadByAria(downloading, urls, path, fileName);
|
|
|
|
|
switch (downloadStatus)
|
|
|
|
|
{
|
|
|
|
|
case DownloadResult.SUCCESS:
|
|
|
|
|
downloading.Downloading.DownloadedFiles.Add(key);
|
|
|
|
|
downloading.Downloading.Gid = null;
|
|
|
|
|
return Path.Combine(path, fileName);
|
|
|
|
|
case DownloadResult.FAILED:
|
|
|
|
|
case DownloadResult.ABORT:
|
|
|
|
|
default:
|
|
|
|
|
return nullMark;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endregion
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 下载封面
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
public override string DownloadCover(DownloadingItem downloading, string coverUrl, string fileName)
|
|
|
|
|
{
|
|
|
|
|
return BaseDownloadCover(downloading, coverUrl, fileName);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 下载弹幕
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
public override string DownloadDanmaku(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
return BaseDownloadDanmaku(downloading);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 下载字幕
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
public override List<string> DownloadSubtitle(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
return BaseDownloadSubtitle(downloading);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 混流音频和视频
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <param name="audioUid"></param>
|
|
|
|
|
/// <param name="videoUid"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
public override string MixedFlow(DownloadingItem downloading, string audioUid, string videoUid)
|
|
|
|
|
{
|
|
|
|
|
if (videoUid == nullMark)
|
|
|
|
|
{
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
return BaseMixedFlow(downloading, audioUid, videoUid);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 解析视频流的下载链接
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
public override void Parse(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
BaseParse(downloading);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 停止下载服务(转换await和Task.Wait两种调用形式)
|
|
|
|
|
/// </summary>
|
|
|
|
|
private async Task EndTask()
|
|
|
|
|
{
|
|
|
|
|
// 停止基本任务
|
|
|
|
|
await BaseEndTask();
|
|
|
|
|
|
|
|
|
|
// 关闭Aria服务器
|
|
|
|
|
await CloseAriaServer();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 停止下载服务
|
|
|
|
|
/// </summary>
|
|
|
|
|
public void End()
|
|
|
|
|
{
|
|
|
|
|
Task.Run(EndTask).Wait();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 启动下载服务
|
|
|
|
|
/// </summary>
|
|
|
|
|
public void Start()
|
|
|
|
|
{
|
|
|
|
|
// 设置aria token
|
|
|
|
|
AriaClient.SetToken(SettingsManager.GetInstance().GetAriaToken());
|
|
|
|
|
// 设置aria host
|
|
|
|
|
AriaClient.SetHost(SettingsManager.GetInstance().GetAriaHost());
|
|
|
|
|
|
|
|
|
|
// 启动基本服务
|
|
|
|
|
BaseStart();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 强制暂停
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <exception cref="OperationCanceledException"></exception>
|
|
|
|
|
protected override void Pause(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
|
|
|
|
|
|
downloading.DownloadStatusTitle = DictionaryResource.GetString("Pausing");
|
|
|
|
|
if (downloading.Downloading.DownloadStatus == DownloadStatus.PAUSE)
|
|
|
|
|
{
|
|
|
|
|
throw new OperationCanceledException("Stop thread by pause");
|
|
|
|
|
}
|
|
|
|
|
// 是否存在
|
|
|
|
|
var isExist = IsExist(downloading);
|
|
|
|
|
if (!isExist.Result)
|
|
|
|
|
{
|
|
|
|
|
throw new OperationCanceledException("Task is deleted");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 是否存在于下载列表中
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
private async Task<bool> IsExist(DownloadingItem downloading)
|
|
|
|
|
{
|
|
|
|
|
bool isExist = downloadingList.Contains(downloading);
|
|
|
|
|
if (isExist)
|
|
|
|
|
{
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// 先恢复为waiting状态，暂停状态下Remove会导致文件重新下载，原因暂不清楚
|
|
|
|
|
await AriaClient.UnpauseAsync(downloading.Downloading.Gid);
|
|
|
|
|
// 移除下载项
|
|
|
|
|
var ariaRemove = await AriaClient.RemoveAsync(downloading.Downloading.Gid);
|
|
|
|
|
if (ariaRemove == null || ariaRemove.Result == downloading.Downloading.Gid)
|
|
|
|
|
{
|
|
|
|
|
// 从内存中删除下载项
|
|
|
|
|
await AriaClient.RemoveDownloadResultAsync(downloading.Downloading.Gid);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 关闭Aria服务器
|
|
|
|
|
/// </summary>
|
|
|
|
|
private async Task CloseAriaServer()
|
|
|
|
|
{
|
|
|
|
|
// 暂停所有下载
|
|
|
|
|
var ariaPause = await AriaClient.PauseAllAsync();
|
|
|
|
|
#if DEBUG
|
|
|
|
|
Core.Utils.Debugging.Console.PrintLine(ariaPause.ToString());
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
// 关闭服务器
|
|
|
|
|
bool close = AriaServer.CloseServer();
|
|
|
|
|
#if DEBUG
|
|
|
|
|
Core.Utils.Debugging.Console.PrintLine(close);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// 采用Aria下载文件
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="downloading"></param>
|
|
|
|
|
/// <returns></returns>
|
|
|
|
|
private DownloadResult DownloadByAria(DownloadingItem downloading, List<string> urls, string path, string localFileName)
|
|
|
|
|
{
|
|
|
|
|
// path已斜杠结尾，去掉斜杠
|
|
|
|
|
path = path.TrimEnd('/').TrimEnd('\\');
|
|
|
|
|
|
|
|
|
|
//检查gid对应任务，如果已创建那么直接使用
|
|
|
|
|
//但是代理设置会出现不能随时更新的问题
|
|
|
|
|
|
|
|
|
|
if (downloading.Downloading.Gid != null)
|
|
|
|
|
{
|
|
|
|
|
Task<AriaTellStatus> status = AriaClient.TellStatus(downloading.Downloading.Gid);
|
|
|
|
|
if (status == null || status.Result == null)
|
|
|
|
|
downloading.Downloading.Gid = null;
|
|
|
|
|
else if (status.Result.Result == null && status.Result.Error != null)
|
|
|
|
|
{
|
|
|
|
|
if (status.Result.Error.Message.Contains("is not found"))
|
|
|
|
|
{
|
|
|
|
|
downloading.Downloading.Gid = null;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (downloading.Downloading.Gid == null)
|
|
|
|
|
{
|
|
|
|
|
AriaSendOption option = new AriaSendOption
|
|
|
|
|
{
|
|
|
|
|
//HttpProxy = $"http://{Settings.GetAriaHttpProxy()}:{Settings.GetAriaHttpProxyListenPort()}",
|
|
|
|
|
Dir = path,
|
|
|
|
|
Out = localFileName,
|
|
|
|
|
//Header = $"cookie: {LoginHelper.GetLoginInfoCookiesString()}\nreferer: https://www.bilibili.com",
|
|
|
|
|
//UseHead = "true",
|
|
|
|
|
UserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.54 Safari/537.36",
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
//// 如果设置了代理，则增加HttpProxy
|
|
|
|
|
//if (SettingsManager.GetInstance().IsAriaHttpProxy() == AllowStatus.YES)
|
|
|
|
|
//{
|
|
|
|
|
// option.HttpProxy = $"http://{SettingsManager.GetInstance().GetAriaHttpProxy()}:{SettingsManager.GetInstance().GetAriaHttpProxyListenPort()}";
|
|
|
|
|
//}
|
|
|
|
|
|
|
|
|
|
// 添加一个下载
|
|
|
|
|
Task<AriaAddUri> ariaAddUri = AriaClient.AddUriAsync(urls, option);
|
|
|
|
|
if (ariaAddUri == null || ariaAddUri.Result == null || ariaAddUri.Result.Result == null)
|
|
|
|
|
{
|
|
|
|
|
return DownloadResult.FAILED;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 保存gid
|
|
|
|
|
string gid = ariaAddUri.Result.Result;
|
|
|
|
|
downloading.Downloading.Gid = gid;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
Task<AriaPause> ariaUnpause = AriaClient.UnpauseAsync(downloading.Downloading.Gid);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 管理下载
|
|
|
|
|
AriaManager ariaManager = new AriaManager();
|
|
|
|
|
ariaManager.TellStatus += AriaTellStatus;
|
|
|
|
|
ariaManager.DownloadFinish += AriaDownloadFinish;
|
|
|
|
|
return ariaManager.GetDownloadStatus(downloading.Downloading.Gid, new Action(() =>
|
|
|
|
|
{
|
|
|
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
|
switch (downloading.Downloading.DownloadStatus)
|
|
|
|
|
{
|
|
|
|
|
case DownloadStatus.PAUSE:
|
|
|
|
|
Task<AriaPause> ariaPause = AriaClient.PauseAsync(downloading.Downloading.Gid);
|
|
|
|
|
// 通知UI，并阻塞当前线程
|
|
|
|
|
Pause(downloading);
|
|
|
|
|
break;
|
|
|
|
|
case DownloadStatus.DOWNLOADING:
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
}));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void AriaTellStatus(long totalLength, long completedLength, long speed, string gid)
|
|
|
|
|
{
|
|
|
|
|
// 当前的下载视频
|
|
|
|
|
DownloadingItem video = null;
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
video = downloadingList.FirstOrDefault(it => it.Downloading.Gid == gid);
|
|
|
|
|
}
|
|
|
|
|
catch (InvalidOperationException e)
|
|
|
|
|
{
|
|
|
|
|
Core.Utils.Debugging.Console.PrintLine("AriaTellStatus()发生异常: {0}", e);
|
|
|
|
|
LogManager.Error("AriaTellStatus()", e);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (video == null) { return; }
|
|
|
|
|
|
|
|
|
|
// 下载进度百分比
|
|
|
|
|
float percent = 0;
|
|
|
|
|
if (totalLength != 0)
|
|
|
|
|
{
|
|
|
|
|
percent = (float)completedLength / totalLength * 100;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 根据进度判断本次是否需要更新UI
|
|
|
|
|
if (Math.Abs(percent - video.Progress) < 0.01) { return; }
|
|
|
|
|
|
|
|
|
|
// 下载进度
|
|
|
|
|
video.Progress = percent;
|
|
|
|
|
|
|
|
|
|
// 下载大小
|
|
|
|
|
video.DownloadingFileSize = Format.FormatFileSize(completedLength) + "/" + Format.FormatFileSize(totalLength);
|
|
|
|
|
|
|
|
|
|
// 下载速度
|
|
|
|
|
video.SpeedDisplay = Format.FormatSpeed(speed);
|
|
|
|
|
|
|
|
|
|
// 最大下载速度
|
|
|
|
|
if (video.Downloading.MaxSpeed < speed)
|
|
|
|
|
{
|
|
|
|
|
video.Downloading.MaxSpeed = speed;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void AriaDownloadFinish(bool isSuccess, string downloadPath, string gid, string msg)
|
|
|
|
|
{
|
|
|
|
|
//throw new NotImplementedException();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
leiurayer
2 years ago