SPT-AKI-Installer/Aki.Helper/DownloadHelper.cs

64 lines
2.0 KiB
C#
Raw Normal View History

2022-07-09 13:14:03 -04:00
using HttpClientProgress;
using SPT_AKI_Installer.Aki.Core.Model;
using System;
using System.IO;
2022-07-11 21:19:47 -04:00
using System.Linq;
using System.Net.Http;
2022-07-11 21:19:47 -04:00
using System.Security.Cryptography;
using System.Threading.Tasks;
namespace SPT_AKI_Installer.Aki.Helper
{
public static class DownloadHelper
{
2022-07-09 00:33:55 -04:00
private static HttpClient _httpClient = new HttpClient() { Timeout = TimeSpan.FromHours(1) };
2022-07-11 21:19:47 -04:00
public static bool FileHashCheck(FileInfo file, string expectedHash)
{
using (MD5 md5Service = MD5.Create())
using (var sourceStream = file.OpenRead())
{
byte[] sourceHash = md5Service.ComputeHash(sourceStream);
byte[] expectedHashBytes = Convert.FromBase64String(expectedHash);
bool matched = Enumerable.SequenceEqual(sourceHash, expectedHashBytes);
return matched;
}
}
public static async Task<GenericResult> DownloadFile(FileInfo outputFile, string targetLink, IProgress<double> progress, string expectedHash = null)
{
try
{
2022-07-09 00:33:55 -04:00
outputFile.Refresh();
2022-07-09 00:33:55 -04:00
if (outputFile.Exists) outputFile.Delete();
2022-07-09 00:33:55 -04:00
// Use the provided extension method
using (var file = new FileStream(outputFile.FullName, FileMode.Create, FileAccess.Write, FileShare.None))
await _httpClient.DownloadDataAsync(targetLink, file, progress);
2022-07-09 00:33:55 -04:00
outputFile.Refresh();
2022-07-09 13:14:03 -04:00
if (!outputFile.Exists)
{
2022-07-09 00:33:55 -04:00
return GenericResult.FromError($"Failed to download {outputFile.Name}");
}
2022-07-11 21:19:47 -04:00
if (expectedHash != null && !FileHashCheck(outputFile, expectedHash))
{
return GenericResult.FromError("Hash mismatch");
}
2022-07-09 00:33:55 -04:00
return GenericResult.FromSuccess();
}
2022-07-09 13:14:03 -04:00
catch (Exception ex)
{
2022-07-09 00:33:55 -04:00
return GenericResult.FromError(ex.Message);
}
}
}
}