Merge pull request #227 from hmemcpy/master

Support updating a single common AssemblyInfo.cs

Author: JakeGinnivan

GitVersionExe/ArgumentParser.cs

@@ -128,6 +128,11 @@ public static Arguments ParseArguments(List<string> commandLineArguments)
                     {
                         arguments.UpdateAssemblyInfo = false;
                     }
+                    else if (!IsSwitchArgument(value))
+                    {
+                        arguments.UpdateAssemblyInfo = true;
+                        arguments.UpdateAssemblyInfoFileName = value;
+                    }
                     else
                     {
                         arguments.UpdateAssemblyInfo = true;
@@ -159,6 +164,11 @@ public static Arguments ParseArguments(List<string> commandLineArguments)
             return arguments;
         }
 
+        static bool IsSwitchArgument(string value)
+        {
+            return value != null && value.StartsWith("-") || value.StartsWith("/");
+        }
+
         static bool IsSwitch(string switchName, string value)
         {
             if (value.StartsWith("-"))

GitVersionExe/Arguments.cs

@@ -28,5 +28,6 @@ public Arguments()
         public string ExecArgs;
 
         public bool UpdateAssemblyInfo;
+        public string UpdateAssemblyInfoFileName;
     }
 }

GitVersionExe/AssemblyInfoFileUpdate.cs

@@ -16,8 +16,7 @@ public AssemblyInfoFileUpdate(Arguments args, string workingDirectory, Dictionar
             if (args.Output != OutputType.Json)
                 Console.WriteLine("Updating assembly info files");
 
-            var assemblyInfoFiles = Directory.GetFiles(workingDirectory, "AssemblyInfo.cs",
-                SearchOption.AllDirectories);
+            var assemblyInfoFiles = GetAssemblyInfoFiles(workingDirectory, args);
 
             foreach (var assemblyInfoFile in assemblyInfoFiles)
             {
@@ -44,6 +43,19 @@ public AssemblyInfoFileUpdate(Arguments args, string workingDirectory, Dictionar
             }
         }
 
+        static IEnumerable<string> GetAssemblyInfoFiles(string workingDirectory, Arguments args)
+        {
+            if (args.UpdateAssemblyInfoFileName != null)
+            {
+                if (File.Exists(args.UpdateAssemblyInfoFileName))
+                {
+                    return new[] { Path.GetFullPath(args.UpdateAssemblyInfoFileName) };
+                }
+            }
+
+            return Directory.GetFiles(workingDirectory, "AssemblyInfo.cs", SearchOption.AllDirectories);
+        }
+
         public void Dispose()
         {
             foreach (var restoreBackup in restoreBackupTasks)

Tests/ArgumentParserTests.cs

@@ -165,11 +165,37 @@ public void Unknown_argument_should_throw()
         Assert.AreEqual("Could not parse command line parameter '-x'.", exception.Message);
     }
 
+    [TestCase("-updateAssemblyInfo true")]
+    [TestCase("-updateAssemblyInfo 1")]
+    [TestCase("-updateAssemblyInfo -proj foo.sln")]
+    public void update_assembly_info_true(string command)
+    {
+        var arguments = ArgumentParser.ParseArguments(command);
+        arguments.UpdateAssemblyInfo.ShouldBe(true);
+    }
+
+    [TestCase("-updateAssemblyInfo false")]
+    [TestCase("-updateAssemblyInfo 0")]
+    public void update_assembly_info_false(string command)
+    {
+        var arguments = ArgumentParser.ParseArguments(command);
+        arguments.UpdateAssemblyInfo.ShouldBe(false);
+    }
+
+    [Test]
+    public void update_assembly_info_with_filename()
+    {
+        var arguments = ArgumentParser.ParseArguments("-updateAssemblyInfo CommonAssemblyInfo.cs");
+        arguments.UpdateAssemblyInfo.ShouldBe(true);
+        arguments.UpdateAssemblyInfoFileName.ShouldBe("CommonAssemblyInfo.cs");
+    }
+
     [Test]
-    public void update_assembly_info()
+    public void update_assembly_info_with_relative_filename()
     {
-        var arguments = ArgumentParser.ParseArguments("-updateAssemblyInfo -proj foo.sln");
+        var arguments = ArgumentParser.ParseArguments("-updateAssemblyInfo ..\\..\\CommonAssemblyInfo.cs");
         arguments.UpdateAssemblyInfo.ShouldBe(true);
+        arguments.UpdateAssemblyInfoFileName.ShouldBe("..\\..\\CommonAssemblyInfo.cs");
     }
 
     [Test]