Uninstalling if cost of repair is larger than cost of content

Author: Dzejkop

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

@@ -71,22 +71,22 @@ AppContentSummary latestVersionContentSummary
             checkIntegrity.Prepare(_status);
             checkIntegrity.Execute(cancellationToken);
 
-            int missingFilesCount = checkIntegrity.Results.Files
-                .Select(f => f.Status == FileIntegrityStatus.MissingData)
-                .Count();
+            var missingFiles = checkIntegrity.Results.Files
+                .Select(f => f.Status == FileIntegrityStatus.MissingData);
+
+            int missingFilesCount = missingFiles.Count();
 
-            int invalidSizeFilesCount = checkIntegrity.Results.Files
-                .Select(f => f.Status == FileIntegrityStatus.InvalidSize)
-                .Count();
+            var invalidSizeFiles = checkIntegrity.Results.Files
+                .Select(f => f.Status == FileIntegrityStatus.InvalidSize);
+
+            int invalidSizeFilesCount = invalidSizeFiles.Count();
 
             if (missingFilesCount + invalidSizeFilesCount == 0)
             {
                 DebugLogger.Log("No missing or invalid size files.");
                 return;
             }
 
-            var repairStrategy = new AppUpdaterRepairAndDiffStrategy(Context, _status, performDiff: false);
-
             double repairCost = (missingFilesCount + invalidSizeFilesCount) * 2;
             if (isNewVersionAvailable)
             {
@@ -100,11 +100,15 @@ AppContentSummary latestVersionContentSummary
             if (repairCost < contentSize)
             {
                 DebugLogger.Log(string.Format("Repair cost {0} is smaller than content cost {1}, repairing...", repairCost, contentSize));
+                IAppUpdaterStrategy repairStrategy = _strategyResolver.Create(StrategyType.RepairAndDiff, Context);
                 repairStrategy.Update(cancellationToken);
             }
             else
             {
-                DebugLogger.Log("Content cost is smaller than repair.");
+                DebugLogger.Log("Content cost is smaller than repair. Uninstalling to prepare for content strategy.");
+                IUninstallCommand uninstall = commandFactory.CreateUninstallCommand(Context);
+                uninstall.Prepare(_status);
+                uninstall.Execute(cancellationToken);
             }
         }