DTOs

Wabbajack.Server/DTOs/MirroredFile.cs

@@ -0,0 +1,13 @@
+using System;
+using Wabbajack.Common;
+
+namespace Wabbajack.Server.DTOs
+{
+    public class MirroredFile
+    {
+        public Hash Hash { get; set; }
+        public DateTime Created { get; set; }
+        public DateTime? Uploaded { get; set; }
+        public string Rationale { get; set; }
+    }
+}

Wabbajack.Server/DataLayer/ArchiveDownloads.cs

@@ -56,6 +56,12 @@ namespace Wabbajack.Server.DataLayer
             await using var conn = await Open();
             return (await conn.QueryAsync<(Hash, string)>("SELECT Hash, PrimaryKeyString FROM ArchiveDownloads")).ToHashSet();
         }
+        
+        public async Task<HashSet<(Hash Hash, AbstractDownloadState State)>> GetAllArchiveDownloadStates()
+        {
+            await using var conn = await Open();
+            return (await conn.QueryAsync<(Hash, AbstractDownloadState)>("SELECT Hash, DownloadState FROM ArchiveDownloads")).ToHashSet();
+        }
 
         
         public async Task<ArchiveDownload> GetArchiveDownload(Guid id)

Wabbajack.Server/DataLayer/MirroredFiles.cs

@@ -0,0 +1,71 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using AngleSharp.Common;
+using Dapper;
+using Wabbajack.Common;
+using Wabbajack.Lib.Downloaders;
+using Wabbajack.Lib.NexusApi;
+using Wabbajack.Server.DTOs;
+
+namespace Wabbajack.Server.DataLayer
+{
+    public partial class SqlService
+    {
+        public async Task<MirroredFile> GetNextMirroredFile()
+        {
+            await using var conn = await Open();
+            var results = await conn.QueryFirstOrDefaultAsync<(Hash, DateTime, DateTime, string)>(
+                "SELECT Hash, Created, Uploaded, Rationale from dbo.MirroredArchives WHERE Uploaded IS NULL");
+            return new MirroredFile
+            {
+                Hash = results.Item1, Created = results.Item2, Uploaded = results.Item3, Rationale = results.Item4
+            };
+        }
+        
+        public async Task<HashSet<Hash>> GetAllMirroredHashes()
+        {
+            await using var conn = await Open();
+            return (await conn.QueryAsync<Hash>("SELECT Hash FROM dbo.MirroredArchives")).ToHashSet();
+        }
+        
+        public async Task UpsertMirroredFile(MirroredFile file)
+        {
+            await using var conn = await Open();
+            await using var trans = await conn.BeginTransactionAsync();
+
+            await conn.ExecuteAsync("DELETE FROM dbo.MirroredArchives WHERE Hash = @Hash", new {file.Hash}, trans);
+            await conn.ExecuteAsync(
+                "INSERT INTO dbo.MirroredArchives (Hash, Created, Updated, Rationale) VALUES (@Hash, @Created, @Updated, @Rationale)",
+                file, trans);
+            await trans.CommitAsync();
+        }
+
+        public async Task InsertAllNexusMirrors()
+        {
+            var permissions = (await GetNexusPermissions()).Where(p => p.Value == HTMLInterface.PermissionValue.Yes);
+            var downloads = (await GetAllArchiveDownloadStates()).Where(a => a.State is NexusDownloader.State).ToDictionary(a =>
+            {
+                var nd = (NexusDownloader.State)a.State;
+                return (nd.Game, nd.ModID);
+            }, a => a.Hash);
+            
+            var existing = await GetAllMirroredHashes();
+
+            foreach (var (key, _) in permissions)
+            {
+                if (!downloads.TryGetValue(key, out var hash)) continue;
+                if (existing.Contains(hash)) continue;
+
+                await UpsertMirroredFile(new MirroredFile
+                {
+                    Hash = hash,
+                    Created = DateTime.UtcNow,
+                    Rationale =
+                        $"Mod ({key.Item1} {key.Item2}) has allowed re-upload permissions on the Nexus"
+                });
+            }
+        }
+    }
+}