Use branch name in documentation (#3082)

Conflicts:
	src/CodeGeneration/DocGenerator/AsciiDoc/GeneratedAsciidocVisitor.cs
	src/CodeGeneration/DocGenerator/Program.cs

(cherry picked from commit 75391fa)

Author: russcam

src/CodeGeneration/DocGenerator/AsciiDoc/GeneratedAsciidocVisitor.cs

@@ -30,8 +30,8 @@ namespace DocGenerator.AsciiDoc
 	public class GeneratedAsciidocVisitor : NoopVisitor
 	{
 		private static readonly Dictionary<string,string> Ids = new Dictionary<string, string>();
-
 		private readonly FileInfo _destination;
+
 		private readonly FileInfo _source;
 	    private readonly Dictionary<string, Project> _projects;
 	    private int _topSectionTitleLevel;
@@ -76,12 +76,13 @@ public override void VisitDocument(Document document)
 
 			if (!document.Attributes.Any(a => a.Name == "ref_current"))
 			{
-				_newDocument.Attributes.Add(new AttributeEntry("ref_current", "https://www.elastic.co/guide/en/elasticsearch/reference/2.4"));
+				_newDocument.Attributes.Add(new AttributeEntry("ref_current",
+					$"https://www.elastic.co/guide/en/elasticsearch/reference/{Program.DocVersion}"));
 			}
 
 			if (document.Attributes.All(a => a.Name != "xpack_current"))
 			{
-				_newDocument.Attributes.Add(new AttributeEntry("xpack_current", "https://www.elastic.co/guide/en/x-pack/2.4"));
+				_newDocument.Attributes.Add(new AttributeEntry("xpack_current", "https://www.elastic.co/guide/en/x-pack/{Program.DocVersion}"));
 			}
 
 			var github = "https://github.com/elastic/elasticsearch-net";
@@ -96,7 +97,9 @@ public override void VisitDocument(Document document)
 				_newDocument.Attributes.Add(new AttributeEntry("nuget", "https://www.nuget.org/packages"));
 			}
 
-			var originalFile = Regex.Replace(_source.FullName.Replace("\\", "/"), @"^(.*Tests/)", $"{github}/tree/2.x/src/Tests/");
+			var originalFile = Regex.Replace(_source.FullName.Replace("\\", "/"), @"^(.*Tests/)",
+				$"{github}/tree/{Program.BranchName}/src/Tests/");
+
 			_newDocument.Insert(0, new Comment
 			{
 				Style = CommentStyle.MultiLine,

src/CodeGeneration/DocGenerator/Program.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Diagnostics;
 using System.IO;
 
 namespace DocGenerator
@@ -20,6 +21,36 @@ static Program()
 				OutputDirPath = @"..\..\..\..\..\docs";
                 BuildOutputPath = @"..\..\..\..\..\build\output";
 			}
+
+			var process = new Process
+			{
+				StartInfo = new ProcessStartInfo
+				{
+					UseShellExecute = false,
+					RedirectStandardOutput = true,
+					FileName = "git.exe",
+					CreateNoWindow = true,
+					WorkingDirectory = Environment.CurrentDirectory,
+					Arguments = "rev-parse --abbrev-ref HEAD"
+				}
+			};
+
+			try
+			{
+				process.Start();
+				BranchName = process.StandardOutput.ReadToEnd().Trim();
+				Console.WriteLine($"Using branch name {BranchName} in documentation");
+				process.WaitForExit();
+			}
+			catch (Exception)
+			{
+				BranchName = "master";
+				Console.WriteLine($"Could not get the git branch name. Assuming {BranchName}");
+			}
+			finally
+			{
+				process.Dispose();
+			}
         }
 
         public static string BuildOutputPath { get; }
@@ -28,6 +59,10 @@ static Program()
 
 		public static string OutputDirPath { get; }
 
+		public static string BranchName { get; }
+
+		public static string DocVersion => "6.1";
+
 		static int Main(string[] args)
 		{
 		    try
@@ -44,3 +79,5 @@ static int Main(string[] args)
 		}
 	}
 }
+
+