Added support for patching named *.fs files to the AssemblyInfoFileUpdate class.

Author: asbjornu

src/GitVersionCore/VersionAssemblyInfoResources/AssemblyVersionInfoTemplates.cs

@@ -10,10 +10,18 @@
     public class AssemblyVersionInfoTemplates
     {
         static readonly IDictionary<string, FileInfo> assemblyInfoSourceList;
+        static readonly IDictionary<string, string> assemblyInfoAddFormats;
 
         static AssemblyVersionInfoTemplates()
         {
             assemblyInfoSourceList = GetEmbeddedVersionAssemblyFiles().ToDictionary(k => k.Extension, v => v);
+            // TODO: It would be nice to do something a bit more clever here, like reusing the VersionAssemblyInfo.* templates somehow. @asbjornu
+            assemblyInfoAddFormats = new Dictionary<string, string>
+            {
+                {".cs", "[assembly: {0}]"},
+                {".vb", "<assembly: {0}>"},
+                {".fs", "[<assembly: {0}>]"}
+            };
         }
 
         public static string GetAssemblyInfoTemplateFor(string assemblyInfoFile)
@@ -35,18 +43,19 @@ public static string GetAssemblyInfoAddFormatFor([NotNull] string fileExtension)
             if (fileExtension == null)
                 throw new ArgumentNullException("fileExtension");
 
-            // TODO: It would be nice to do something a bit more clever here, like reusing the VersionAssemblyInfo.* templates somehow. @asbjornu
-            switch (fileExtension.ToLowerInvariant())
-            {
-                case ".cs":
-                    return "[assembly: {0}]";
-                case ".vb":
-                    return "<assembly: {0}>";
-                case ".fs":
-                    return "[<assembly: {0}>]";
-            }
+            string assemblyInfoAddFormat;
+            if (!assemblyInfoAddFormats.TryGetValue(fileExtension, out assemblyInfoAddFormat))
+                throw new NotSupportedException(string.Format("Unknown file extension '{0}'.", fileExtension));
+
+            return assemblyInfoAddFormat;
+        }
+
+        public static bool IsSupported([NotNull] string fileExtension)
+        {
+            if (fileExtension == null)
+                throw new ArgumentNullException("fileExtension");
 
-            throw new NotSupportedException(string.Format("Unknown file extension '{0}'.", fileExtension));
+            return assemblyInfoAddFormats.Keys.Contains(fileExtension, StringComparer.OrdinalIgnoreCase);
         }
 
         private static IEnumerable<FileInfo> GetEmbeddedVersionAssemblyFiles()

src/GitVersionExe/AssemblyInfoFileUpdate.cs

@@ -34,7 +34,7 @@ public AssemblyInfoFileUpdate(Arguments args, string workingDirectory, VersionVa
             var assemblyFileVersionRegex = new Regex(@"AssemblyFileVersion\(""[^""]*""\)");
             var assemblyFileVersionString = string.Format("AssemblyFileVersion(\"{0}\")", assemblyFileVersion);
 
-            foreach (var assemblyInfoFile in assemblyInfoFiles.Select(f => new FileInfo(f)))
+            foreach (var assemblyInfoFile in assemblyInfoFiles)
             {
                 var backupAssemblyInfo = assemblyInfoFile.FullName + ".bak";
                 var localAssemblyInfo = assemblyInfoFile.FullName;
@@ -73,7 +73,7 @@ static string ReplaceOrAppend(Regex replaceRegex, string inputString, string rep
         }
 
 
-        static IEnumerable<string> GetAssemblyInfoFiles(string workingDirectory, Arguments args, IFileSystem fileSystem)
+        static IEnumerable<FileInfo> GetAssemblyInfoFiles(string workingDirectory, Arguments args, IFileSystem fileSystem)
         {
             if (args.UpdateAssemblyInfoFileName != null && args.UpdateAssemblyInfoFileName.Any(x => !string.IsNullOrWhiteSpace(x)))
             {
@@ -83,16 +83,18 @@ static IEnumerable<string> GetAssemblyInfoFiles(string workingDirectory, Argumen
 
                     if (EnsureVersionAssemblyInfoFile(args, fileSystem, fullPath))
                     {
-                        yield return fullPath;
+                        yield return new FileInfo(fullPath);
                     }
                 }
             }
             else
             {
-                foreach (var item in fileSystem.DirectoryGetFiles(workingDirectory, "AssemblyInfo.*", SearchOption.AllDirectories)
-                    .Where(f => f.EndsWith(".cs", StringComparison.OrdinalIgnoreCase) || f.EndsWith(".vb", StringComparison.OrdinalIgnoreCase)))
+                foreach (var item in fileSystem.DirectoryGetFiles(workingDirectory, "AssemblyInfo.*", SearchOption.AllDirectories))
                 {
-                    yield return item;
+                    var assemblyInfoFile = new FileInfo(item);
+
+                    if (AssemblyVersionInfoTemplates.IsSupported(assemblyInfoFile.Extension))
+                        yield return assemblyInfoFile;
                 }
             }
         }