wabbajack/Wabbajack.Lib/Downloaders/MEGADownloader.cs
2019-12-07 00:40:29 -06:00

59 lines
1.8 KiB
C#

using System;
using System.Threading.Tasks;
using CG.Web.MegaApiClient;
using Wabbajack.Common;
namespace Wabbajack.Lib.Downloaders
{
public class MegaDownloader : IDownloader, IUrlDownloader
{
public async Task<AbstractDownloadState> GetDownloaderState(dynamic archiveINI)
{
var url = archiveINI?.General?.directURL;
return GetDownloaderState(url);
}
public AbstractDownloadState GetDownloaderState(string url)
{
if (url != null && url.StartsWith(Consts.MegaPrefix))
return new State { Url = url };
return null;
}
public async Task Prepare()
{
}
public class State : HTTPDownloader.State
{
public override async Task Download(Archive a, string destination)
{
var client = new MegaApiClient();
Utils.Status("Logging into MEGA (as anonymous)");
client.LoginAnonymous();
var fileLink = new Uri(Url);
var node = client.GetNodeFromLink(fileLink);
Utils.Status($"Downloading MEGA file: {a.Name}");
client.DownloadFile(fileLink, destination);
}
public override async Task<bool> Verify()
{
var client = new MegaApiClient();
Utils.Status("Logging into MEGA (as anonymous)");
client.LoginAnonymous();
var fileLink = new Uri(Url);
try
{
var node = client.GetNodeFromLink(fileLink);
return true;
}
catch (Exception)
{
return false;
}
}
}
}
}