MyGet support (output specific to MyGet build server)

GitVersionCore/BuildServers/BuildServerList.cs

@@ -27,7 +27,8 @@ static IEnumerable<IBuildServer> DefaultSelector(Authentication authentication)
                 {
                     new ContinuaCi(authentication),
                     new TeamCity(authentication),
-                    new AppVeyor(authentication)
+                    new AppVeyor(authentication),
+                    new MyGet(authentication)
                 };
             }
 

GitVersionCore/BuildServers/MyGet.cs

@@ -0,0 +1,53 @@
+namespace GitVersion
+{
+    using System;
+    using System.Collections.Generic;
+
+    public class MyGet : BuildServerBase
+    {
+        Authentication authentication;
+
+        public MyGet(Authentication authentication)
+        {
+            this.authentication = authentication;
+        }
+
+        public override bool CanApplyToCurrentContext()
+        {
+            var buildRunner = Environment.GetEnvironmentVariable("BuildRunner");
+
+            return !string.IsNullOrEmpty(buildRunner)
+                && buildRunner.Equals("MyGet", StringComparison.InvariantCultureIgnoreCase);
+        }
+
+        public override void PerformPreProcessingSteps(string gitDirectory)
+        {
+            if (string.IsNullOrEmpty(gitDirectory))
+            {
+                throw new WarningException("Failed to find .git directory on agent.");
+            }
+
+            GitHelper.NormalizeGitDirectory(gitDirectory, authentication);
+        }
+
+        public override string[] GenerateSetParameterMessage(string name, string value)
+        {
+            var messages = new List<string>
+            {
+                string.Format("##myget[setParameter name='GitVersion.{0}' value='{1}']", name, ServiceMessageEscapeHelper.EscapeValue(value))
+            };
+
+            if (string.Equals(name, "LegacySemVerPadded", StringComparison.InvariantCultureIgnoreCase))
+            {
+                messages.Add(string.Format("##myget[buildNumber '{0}']", ServiceMessageEscapeHelper.EscapeValue(value)));
+            }
+
+            return messages.ToArray();
+        }
+
+        public override string GenerateSetVersionMessage(string versionToUseForBuildNumber)
+        {
+            return null;
+        }
+    }
+}

GitVersionCore/BuildServers/TeamCity.cs

@@ -30,32 +30,14 @@ public override string[] GenerateSetParameterMessage(string name, string value)
         {
             return new[]
             {
-                string.Format("##teamcity[setParameter name='GitVersion.{0}' value='{1}']", name, EscapeValue(value)),
-                string.Format("##teamcity[setParameter name='system.GitVersion.{0}' value='{1}']", name, EscapeValue(value))
+                string.Format("##teamcity[setParameter name='GitVersion.{0}' value='{1}']", name, ServiceMessageEscapeHelper.EscapeValue(value)),
+                string.Format("##teamcity[setParameter name='system.GitVersion.{0}' value='{1}']", name, ServiceMessageEscapeHelper.EscapeValue(value))
             };
         }
 
         public override string GenerateSetVersionMessage(string versionToUseForBuildNumber)
         {
-            return string.Format("##teamcity[buildNumber '{0}']", EscapeValue(versionToUseForBuildNumber));
-        }
-
-        static string EscapeValue(string value)
-        {
-            if (value == null)
-            {
-                return null;
-            }
-            // List of escape values from http://confluence.jetbrains.com/display/TCD8/Build+Script+Interaction+with+TeamCity
-
-            value = value.Replace("|", "||");
-            value = value.Replace("'", "|'");
-            value = value.Replace("[", "|[");
-            value = value.Replace("]", "|]");
-            value = value.Replace("\r", "|r");
-            value = value.Replace("\n", "|n");
-
-            return value;
+            return string.Format("##teamcity[buildNumber '{0}']", ServiceMessageEscapeHelper.EscapeValue(versionToUseForBuildNumber));
         }
     }
 }

GitVersionCore/GitVersionCore.csproj

@@ -62,6 +62,7 @@
     <Compile Include="BuildServers\ContinuaCi.cs" />
     <Compile Include="BuildServers\GitHelper.cs" />
     <Compile Include="BuildServers\IBuildServer.cs" />
+    <Compile Include="BuildServers\MyGet.cs" />
     <Compile Include="BuildServers\TeamCity.cs" />
     <Compile Include="GitFlow\BranchFinders\BranchCommitDifferenceFinder.cs" />
     <Compile Include="GitFlow\BranchFinders\RecentTagVersionExtractor.cs" />

GitVersionCore/Helpers/DeleteHelper.cs

@@ -2,6 +2,26 @@
 {
     using System.IO;
 
+    public static class ServiceMessageEscapeHelper
+    {
+        public static string EscapeValue(string value)
+        {
+            if (value == null)
+            {
+                return null;
+            }
+            // List of escape values from http://confluence.jetbrains.com/display/TCD8/Build+Script+Interaction+with+TeamCity
+
+            value = value.Replace("|", "||");
+            value = value.Replace("'", "|'");
+            value = value.Replace("[", "|[");
+            value = value.Replace("]", "|]");
+            value = value.Replace("\r", "|r");
+            value = value.Replace("\n", "|n");
+
+            return value;
+        }
+    }
     public static class DeleteHelper
     {
         public static void DeleteGitRepository(string directory)

GitVersionTask.Tests/BuildServers/MyGetTests.cs

@@ -0,0 +1,33 @@
+using GitVersion;
+using NUnit.Framework;
+
+[TestFixture]
+public class MyGetTests
+{
+    [Test]
+    public void Develop_branch()
+    {
+        var authentication = new Authentication();
+        var versionBuilder = new MyGet(authentication);
+        var message = versionBuilder.GenerateSetVersionMessage("0.0.0-Unstable4");
+        Assert.AreEqual(null, message);
+    }
+
+    [Test]
+    public void EscapeValues()
+    {
+        var authentication = new Authentication();
+        var versionBuilder = new MyGet(authentication);
+        var message = versionBuilder.GenerateSetParameterMessage("Foo", "0.8.0-unstable568 Branch:'develop' Sha:'ee69bff1087ebc95c6b43aa2124bd58f5722e0cb'");
+        Assert.AreEqual("##myget[setParameter name='GitVersion.Foo' value='0.8.0-unstable568 Branch:|'develop|' Sha:|'ee69bff1087ebc95c6b43aa2124bd58f5722e0cb|'']", message[0]);
+    }
+
+    [Test]
+    public void BuildNumber()
+    {
+        var authentication = new Authentication();
+        var versionBuilder = new MyGet(authentication);
+        var message = versionBuilder.GenerateSetParameterMessage("LegacySemVerPadded", "0.8.0-unstable568");
+        Assert.AreEqual("##myget[buildNumber '0.8.0-unstable568']", message[1]);
+    }
+}

GitVersionTask.Tests/GitVersionTask.Tests.csproj

@@ -89,6 +89,7 @@
     <Compile Include="BranchClassifierTests.cs" />
     <Compile Include="BranchFinders\DevelopTests.cs" />
     <Compile Include="BuildServers\BuildServerBaseTests.cs" />
+    <Compile Include="BuildServers\MyGetTests.cs" />
     <Compile Include="BuildServers\TeamCityTests.cs" />
     <Compile Include="BuildServers\ContinuaCiTests.cs" />
     <Compile Include="ConfigReaderTests.cs" />