Calculating valid repair cost in pre update

Author: Dzejkop

Assets/PatchKit Patcher/Scripts/AppUpdater/AppUpdater.cs

@@ -3,6 +3,7 @@
 using System.Linq;
 using System.Linq.Expressions;
 using System.Threading;
+using System.Collections.Generic;
 using PatchKit.Unity.Patcher.Cancellation;
 using PatchKit.Unity.Patcher.Debug;
 using PatchKit.Unity.Patcher.AppUpdater.Commands;
@@ -72,12 +73,12 @@ AppContentSummary latestVersionContentSummary
             checkIntegrity.Execute(cancellationToken);
 
             var missingFiles = checkIntegrity.Results.Files
-                .Select(f => f.Status == FileIntegrityStatus.MissingData);
+                .Where(f => f.Status == FileIntegrityStatus.MissingData);
 
             int missingFilesCount = missingFiles.Count();
 
             var invalidSizeFiles = checkIntegrity.Results.Files
-                .Select(f => f.Status == FileIntegrityStatus.InvalidSize);
+                .Where(f => f.Status == FileIntegrityStatus.InvalidSize);
 
             int invalidSizeFilesCount = invalidSizeFiles.Count();
 
@@ -87,15 +88,7 @@ AppContentSummary latestVersionContentSummary
                 return;
             }
 
-            double repairCost = (missingFilesCount + invalidSizeFilesCount) * 2;
-            if (isNewVersionAvailable)
-            {
-                repairCost *= latestVersionContentSummary.Chunks.Size;
-            }
-            else
-            {
-                repairCost *= installedVersionContentSummary.Chunks.Size;
-            }
+            double repairCost = CalculateRepairCost(installedVersionContentSummary, missingFiles.Concat(invalidSizeFiles));
 
             if (repairCost < contentSize)
             {
@@ -112,6 +105,13 @@ AppContentSummary latestVersionContentSummary
             }
         }
 
+        private long CalculateRepairCost(AppContentSummary contentSummary, IEnumerable<FileIntegrity> filesToRepair)
+        {
+            return filesToRepair
+                .Select(f => contentSummary.Files.FirstOrDefault(e => e.Path == f.FileName))
+                .Sum(f => f.Size);
+        }
+
         public void Update(CancellationToken cancellationToken)
         {
             Assert.MethodCalledOnlyOnce(ref _updateHasBeenCalled, "Update");