From a7845802bb7b149e076664dfa96280036407be99 Mon Sep 17 00:00:00 2001
From: Justin Swanson <justin.c.swanson@gmail.com>
Date: Fri, 20 Dec 2019 14:32:04 -0600
Subject: [PATCH 1/2] Added path of a missing patch to error logging

---
 Wabbajack.Lib/MO2Compiler.cs | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/Wabbajack.Lib/MO2Compiler.cs b/Wabbajack.Lib/MO2Compiler.cs
index f93985eb..be8b7da7 100644
--- a/Wabbajack.Lib/MO2Compiler.cs
+++ b/Wabbajack.Lib/MO2Compiler.cs
@@ -117,7 +117,7 @@ namespace Wabbajack.Lib
             
             if (Directory.Exists(lootPath))
             {
-                    lootFiles = Directory.EnumerateFiles(lootPath, "userlist.yaml", SearchOption.AllDirectories)
+                lootFiles = Directory.EnumerateFiles(lootPath, "userlist.yaml", SearchOption.AllDirectories)
                     .Where(p => p.FileExists())
                     .Select(p => new RawSourceFile(VFS.Index.ByRootPath[p])
                         { Path = Path.Combine(Consts.LOOTFolderFilesDir, p.RelativeTo(lootPath)) });
@@ -415,8 +415,9 @@ namespace Wabbajack.Lib
             var absolutePaths = AllFiles.ToDictionary(e => e.Path, e => e.AbsolutePath);
             await groups.PMap(Queue, group => BuildArchivePatches(group.Key, group, absolutePaths));
 
-            if (InstallDirectives.OfType<PatchedFromArchive>().FirstOrDefault(f => f.PatchID == null) != null)
-                Error("Missing patches after generation, this should not happen");
+            var firstFailedPatch = InstallDirectives.OfType<PatchedFromArchive>().FirstOrDefault(f => f.PatchID == null);
+            if (firstFailedPatch != null)
+                Error($"Missing patches after generation, this should not happen. First failure: {firstFailedPatch.FullPath}");
         }
 
         private async Task BuildArchivePatches(string archiveSha, IEnumerable<PatchedFromArchive> group,

From 790acec2d0de0be73c32ed32e43ae4e16488bf36 Mon Sep 17 00:00:00 2001
From: Justin Swanson <justin.c.swanson@gmail.com>
Date: Fri, 20 Dec 2019 14:50:02 -0600
Subject: [PATCH 2/2] Added missing PMap overload for Func<Task> work actions

---
 Wabbajack.Common/Utils.cs | 40 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/Wabbajack.Common/Utils.cs b/Wabbajack.Common/Utils.cs
index 73d2e57d..99139225 100644
--- a/Wabbajack.Common/Utils.cs
+++ b/Wabbajack.Common/Utils.cs
@@ -1,4 +1,4 @@
-using System;
+using System;
 using System.Collections.Generic;
 using System.Data.HashFunction.xxHash;
 using System.Diagnostics;
@@ -611,6 +611,44 @@ namespace Wabbajack.Common
             return await Task.WhenAll(tasks);
         }
 
+        public static async Task PMap<TI>(this IEnumerable<TI> coll, WorkQueue queue,
+            Func<TI, Task> f)
+        {
+            var colllst = coll.ToList();
+
+            var remainingTasks = colllst.Count;
+
+            var tasks = colllst.Select(i =>
+            {
+                var tc = new TaskCompletionSource<bool>();
+                queue.QueueTask(async () =>
+                {
+                    try
+                    {
+                        await f(i);
+                        tc.SetResult(true);
+                    }
+                    catch (Exception ex)
+                    {
+                        tc.SetException(ex);
+                    }
+                    Interlocked.Decrement(ref remainingTasks);
+                });
+                return tc.Task;
+            }).ToList();
+
+            // To avoid thread starvation, we'll start to help out in the work queue
+            if (WorkQueue.WorkerThread)
+                while (remainingTasks > 0)
+                    if (queue.Queue.TryTake(out var a, 500))
+                    {
+                        WorkQueue.AsyncLocalCurrentQueue.Value = WorkQueue.ThreadLocalCurrentQueue.Value;
+                        await a();
+                    }
+
+            await Task.WhenAll(tasks);
+        }
+
         public static async Task PMap<TI>(this IEnumerable<TI> coll, WorkQueue queue, Action<TI> f)
         {
             await coll.PMap(queue, i =>