wabbajack/Wabbajack.Lib/Downloaders/AbstractDownloadState.cs

103 lines
3.4 KiB
C#
Raw Normal View History

2019-12-29 22:57:01 +00:00
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Alphaleonis.Win32.Filesystem;
using Wabbajack.Lib.Validation;
2019-10-12 18:03:45 +00:00
namespace Wabbajack.Lib.Downloaders
2019-10-12 18:03:45 +00:00
{
public interface IMetaState
{
string URL { get; }
string Name { get; set; }
string Author { get; set; }
string Version { get; set; }
string ImageURL { get; set; }
bool IsNSFW { get; set; }
string Description { get; set; }
Task<bool> LoadMetaData();
}
2019-10-12 18:03:45 +00:00
public abstract class AbstractDownloadState
{
2019-12-29 22:57:01 +00:00
public static List<Type> KnownSubTypes = new List<Type>
2019-12-29 22:57:01 +00:00
{
typeof(HTTPDownloader.State),
typeof(GameFileSourceDownloader.State),
typeof(GoogleDriveDownloader.State),
typeof(LoversLabDownloader.State),
typeof(ManualDownloader.State),
typeof(MediaFireDownloader.State),
typeof(MegaDownloader.State),
typeof(ModDBDownloader.State),
typeof(NexusDownloader.State),
typeof(SteamWorkshopDownloader.State),
typeof(VectorPlexusDownloader.State),
2020-01-22 09:41:28 +00:00
typeof(DeadlyStreamDownloader.State),
2020-01-22 09:50:09 +00:00
typeof(AFKModsDownloader.State),
typeof(TESAllianceDownloader.State),
typeof(BethesdaNetDownloader.State),
typeof(YouTubeDownloader.State)
2019-12-29 22:57:01 +00:00
};
public static Dictionary<string, Type> NameToType { get; set; }
public static Dictionary<Type, string> TypeToName { get; set; }
static AbstractDownloadState()
{
NameToType = KnownSubTypes.ToDictionary(t => t.FullName.Substring(t.Namespace.Length + 1), t => t);
TypeToName = NameToType.ToDictionary(k => k.Value, k => k.Key);
}
public abstract object[] PrimaryKey { get; }
2020-01-10 13:25:01 +00:00
public string PrimaryKeyString
{
get
{
var pk = new List<object>();
pk.Add(AbstractDownloadState.TypeToName[GetType()]);
pk.AddRange(PrimaryKey);
var pk_str = string.Join("|",pk.Select(p => p.ToString()));
return pk_str;
}
}
2019-12-29 22:57:01 +00:00
2019-10-12 18:03:45 +00:00
/// <summary>
/// Returns true if this file is allowed to be downloaded via whitelist
/// </summary>
/// <param name="whitelist"></param>
/// <returns></returns>
public abstract bool IsWhitelisted(ServerWhitelist whitelist);
/// <summary>
/// Downloads this file to the given destination location
/// </summary>
/// <param name="destination"></param>
public abstract Task<bool> Download(Archive a, string destination);
2019-10-12 18:03:45 +00:00
public async Task<bool> Download(string destination)
{
2020-01-03 19:39:36 +00:00
var path = Path.GetDirectoryName(destination);
if (!string.IsNullOrEmpty(path) && !Directory.Exists(path))
Directory.CreateDirectory(path);
return await Download(new Archive {Name = Path.GetFileName(destination)}, destination);
}
2019-10-12 18:03:45 +00:00
/// <summary>
/// Returns true if this link is still valid
/// </summary>
/// <returns></returns>
public abstract Task<bool> Verify(Archive archive);
public abstract IDownloader GetDownloader();
public abstract string GetManifestURL(Archive a);
public abstract string[] GetMetaIni();
2019-10-12 18:03:45 +00:00
}
}