From 6e0e3607b070f5593c6e60d6acc76d7a0c2c977f Mon Sep 17 00:00:00 2001
From: dabutvin <butvinik@outlook.com>
Date: Sun, 10 Nov 2019 15:35:40 -0800
Subject: [PATCH] prevent default branch override for wikicompress

---
 CompressImagesFunction/CompressImages.cs | 31 ++++++++++++++++++------
 1 file changed, 23 insertions(+), 8 deletions(-)

diff --git a/CompressImagesFunction/CompressImages.cs b/CompressImagesFunction/CompressImages.cs
index 61b42a65d..2133aed8b 100644
--- a/CompressImagesFunction/CompressImages.cs
+++ b/CompressImagesFunction/CompressImages.cs
@@ -41,6 +41,7 @@ public static bool Run(CompressimagesParameters parameters, ICollector<CompressI
 
             var repo = new Repository(parameters.LocalPath);
             var remote = repo.Network.Remotes["origin"];
+            var isWikiCompress = parameters.CloneUrl.EndsWith(".wiki.git");
 
             // check if we have the branch already or this is empty repo
             try
@@ -64,7 +65,7 @@ public static bool Run(CompressimagesParameters parameters, ICollector<CompressI
             }
 
             // check if we should switch away from the default branch
-            if (parameters.Settings != null && !string.IsNullOrEmpty(parameters.Settings.DefaultBranchOverride))
+            if (!isWikiCompress && parameters.Settings != null && !string.IsNullOrEmpty(parameters.Settings.DefaultBranchOverride))
             {
                 logger.LogInformation(
                     "CompressImagesFunction: default branch override for {Owner}/{RepoName} is {DefaultBranchOverride}",
@@ -103,7 +104,7 @@ public static bool Run(CompressimagesParameters parameters, ICollector<CompressI
             }
 
             // Add new compressMessage if we should compress Wiki
-            if (repoConfiguration.CompressWiki && parameters.CloneUrl.Contains(".wiki.git") == false)
+            if (repoConfiguration.CompressWiki && isWikiCompress == false)
             {
                 logger.LogInformation("CompressImagesFunction: Adding Wiki image compression to queue for {Owner}/{RepoName}", parameters.RepoOwner, parameters.RepoName);
                 compressImagesMessages.Add(new CompressImagesMessage()
@@ -122,7 +123,7 @@ public static bool Run(CompressimagesParameters parameters, ICollector<CompressI
             }
 
             // Should not create branch if we are compressing Wiki
-            if (parameters.CloneUrl.Contains(".wiki.git") == false)
+            if (isWikiCompress == false)
             {
                 // check out the branch
                 repo.CreateBranch(KnownGitHubs.BranchName);
@@ -166,8 +167,12 @@ public static bool Run(CompressimagesParameters parameters, ICollector<CompressI
 
             repo.Refs.UpdateTarget(repo.Refs.Head, commitToKeep);
 
-            // Should not create branch if we are compressing Wiki
-            if (parameters.CloneUrl.Contains(".wiki.git") == false)
+            // Should use "master" if we are compressing Wiki
+            if (isWikiCompress)
+            {
+                var branchAgain = Commands.Checkout(repo, "master");
+            }
+            else
             {
                 var branchAgain = Commands.Checkout(repo, KnownGitHubs.BranchName);
             }
@@ -196,10 +201,20 @@ public static bool Run(CompressimagesParameters parameters, ICollector<CompressI
             }
 
             // push to GitHub
-            repo.Network.Push(remote, $"refs/heads/{KnownGitHubs.BranchName}", new PushOptions
+            if (isWikiCompress)
             {
-                CredentialsProvider = credentialsProvider,
-            });
+                repo.Network.Push(remote, "refs/heads/master", new PushOptions
+                {
+                    CredentialsProvider = credentialsProvider,
+                });
+            }
+            else
+            {
+                repo.Network.Push(remote, $"refs/heads/{KnownGitHubs.BranchName}", new PushOptions
+                {
+                    CredentialsProvider = credentialsProvider,
+                });
+            }
 
             return true;
         }