Move inference logic to a service and use it

Author: cotti

config/products.yml

@@ -91,18 +91,25 @@ products:
     versioning: 'stack'
   edot-ios:
     display: 'Elastic Distribution of OpenTelemetry iOS'
+    repository: 'apm-agent-ios'
   edot-android:
     display: 'Elastic Distribution of OpenTelemetry Android'
+    repository: 'apm-agent-android'
   edot-dotnet:
     display: 'Elastic Distribution of OpenTelemetry .NET'
+    repository: 'elastic-otel-dotnet'
   edot-java:
     display: 'Elastic Distribution of OpenTelemetry Java'
+    repository: 'elastic-otel-java'
   edot-node:
     display: 'Elastic Distribution of OpenTelemetry Node'
+    repository: 'elastic-otel-node'
   edot-php:
     display: 'Elastic Distribution of OpenTelemetry PHP'
+    repository: 'elastic-otel-php'
   edot-python:
     display: 'Elastic Distribution of OpenTelemetry Python'
+    repository: 'elastic-otel-python'
   edot-cf-aws:
     display: 'EDOT Cloud Forwarder for AWS'
   edot-cf-azure:

src/Elastic.Documentation.Configuration/Products/Product.cs

@@ -19,5 +19,6 @@ public record Product
 	public required string Id { get; init; }
 	public required string DisplayName { get; init; }
 	public VersioningSystem? VersioningSystem { get; init; }
+	public string? Repository { get; init; }
 }
 

src/Elastic.Documentation.Configuration/Products/ProductExtensions.cs

@@ -21,7 +21,8 @@ public static ProductsConfiguration CreateProducts(this ConfigurationFileProvide
 			{
 				Id = kvp.Key,
 				DisplayName = kvp.Value.Display,
-				VersioningSystem = versionsConfiguration.GetVersioningSystem(VersionsConfigurationExtensions.ToVersioningSystemId(kvp.Value.Versioning ?? kvp.Key))
+				VersioningSystem = versionsConfiguration.GetVersioningSystem(VersionsConfigurationExtensions.ToVersioningSystemId(kvp.Value.Versioning ?? kvp.Key)),
+				Repository = kvp.Value.Repository
 			});
 
 		return new ProductsConfiguration
@@ -41,4 +42,5 @@ internal sealed record ProductDto
 {
 	public string Display { get; set; } = string.Empty;
 	public string? Versioning { get; set; }
+	public string? Repository { get; set; }
 }

src/Elastic.Documentation.Configuration/Versions/VersionInference.cs

@@ -0,0 +1,43 @@
+// Licensed to Elasticsearch B.V under one or more agreements.
+// Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
+// See the LICENSE file in the project root for more information
+
+using Elastic.Documentation.Configuration.LegacyUrlMappings;
+using Elastic.Documentation.Configuration.Products;
+
+namespace Elastic.Documentation.Configuration.Versions;
+
+public interface IVersionInferrerService
+{
+	VersioningSystem InferVersion(IReadOnlyCollection<LegacyPageMapping>? legacyPages);
+}
+
+public class ProductVersionInferrerService(ProductsConfiguration productsConfiguration, VersionsConfiguration versionsConfiguration, string repositoryName) : IVersionInferrerService
+{
+	private ProductsConfiguration ProductsConfiguration { get; } = productsConfiguration;
+	private VersionsConfiguration VersionsConfiguration { get; } = versionsConfiguration;
+	private string RepositoryName { get; } = repositoryName;
+	public VersioningSystem InferVersion(IReadOnlyCollection<LegacyPageMapping>? legacyPages)
+	{
+		var versioning = legacyPages is not null && legacyPages.Count > 0
+			? legacyPages.ElementAt(0).Product.VersioningSystem! // If the page has a legacy page mapping, use the versioning system of the legacy page
+			: ProductsConfiguration.Products.TryGetValue(RepositoryName, out var belonging)
+				? belonging.VersioningSystem! //If the page's docset has a name with a direct product match, use the versioning system of the product
+				: ProductsConfiguration.Products.Values.SingleOrDefault(p =>
+					p.Repository is not null && p.Repository.Equals(RepositoryName, StringComparison.OrdinalIgnoreCase)) is { } repositoryMatch
+					? repositoryMatch.VersioningSystem! // Verify if the page belongs to a repository linked to a product, and if so, use the versioning system of the product
+					: VersionsConfiguration.VersioningSystems[VersioningSystemId.Stack]; // Fallback to the stack versioning system
+
+		return versioning;
+	}
+}
+
+public class NoopVersionInferrer : IVersionInferrerService
+{
+	public VersioningSystem InferVersion(IReadOnlyCollection<LegacyPageMapping>? legacyPages) => new()
+	{
+		Id = VersioningSystemId.Stack,
+		Base = new SemVersion(0, 0, 0),
+		Current = new SemVersion(0, 0, 0)
+	};
+}

src/Elastic.Markdown/DocumentationGenerator.cs

@@ -7,6 +7,7 @@
 using Elastic.Documentation;
 using Elastic.Documentation.Configuration;
 using Elastic.Documentation.Configuration.LegacyUrlMappings;
+using Elastic.Documentation.Configuration.Versions;
 using Elastic.Documentation.Links;
 using Elastic.Documentation.Links.CrossLinks;
 using Elastic.Documentation.Serialization;
@@ -72,7 +73,7 @@ public DocumentationGenerator(
 		Context = docSet.Context;
 		CrossLinkResolver = docSet.CrossLinkResolver;
 		HtmlWriter = new HtmlWriter(DocumentationSet, _writeFileSystem, new DescriptionGenerator(), navigationHtmlWriter, legacyUrlMapper,
-			positionalNavigation);
+			positionalNavigation, new ProductVersionInferrerService(DocumentationSet.Context.ProductsConfiguration, DocumentationSet.Context.VersionsConfiguration, DocumentationSet.Context.Git.RepositoryName));
 		_documentationFileExporter =
 			docSet.Context.AvailableExporters.Contains(Exporter.Html)
 				? docSet.EnabledExtensions.FirstOrDefault(e => e.FileExporter != null)?.FileExporter

src/Elastic.Markdown/HtmlWriter.cs

@@ -7,7 +7,6 @@
 using Elastic.Documentation;
 using Elastic.Documentation.Configuration.LegacyUrlMappings;
 using Elastic.Documentation.Configuration.Products;
-using Elastic.Documentation.Configuration.TableOfContents;
 using Elastic.Documentation.Configuration.Versions;
 using Elastic.Documentation.Site.FileProviders;
 using Elastic.Documentation.Site.Navigation;
@@ -26,7 +25,8 @@ public class HtmlWriter(
 	IDescriptionGenerator descriptionGenerator,
 	INavigationHtmlWriter? navigationHtmlWriter = null,
 	ILegacyUrlMapper? legacyUrlMapper = null,
-	IPositionalNavigation? positionalNavigation = null
+	IPositionalNavigation? positionalNavigation = null,
+	IVersionInferrerService? versionInferrerService = null
 )
 	: IMarkdownStringRenderer
 {
@@ -39,6 +39,8 @@ public class HtmlWriter(
 	private ILegacyUrlMapper LegacyUrlMapper { get; } = legacyUrlMapper ?? new NoopLegacyUrlMapper();
 	private IPositionalNavigation PositionalNavigation { get; } = positionalNavigation ?? documentationSet;
 
+	private IVersionInferrerService VersionInferrerService { get; } = versionInferrerService ?? new NoopVersionInferrer();
+
 	/// <inheritdoc />
 	public string Render(string markdown, IFileInfo? source)
 	{
@@ -103,24 +105,7 @@ private async Task<RenderResult> RenderLayout(MarkdownFile markdown, MarkdownDoc
 			fullNavigationRenderResult
 		);
 
-		// Acquiring the versioning system for the current page:
-		// 1. If the page has a legacy page mapping, use the versioning system of the legacy page
-		// 2. If the page's docset has a name with a direct product maatch, use the versioning system of the product
-		// 3. If the page's docset has a table of contents entry with a direct product match, use the versioning system of the product
-		// 4. Fallback to the stack versioning system
-		VersioningSystem pageVersioning = null!;
-		if (legacyPages is not null && legacyPages.Count > 0)
-			pageVersioning = legacyPages.ElementAt(0).Product.VersioningSystem!;
-		else if (DocumentationSet.Context.ProductsConfiguration.Products.TryGetValue(DocumentationSet.Name, out var belonging))
-			pageVersioning = belonging.VersioningSystem!;
-		else if (DocumentationSet.Configuration.TableOfContents.FirstOrDefault(t => t is TocReference tRef && !tRef.Source.LocalPath.Equals("/")) is TocReference tocRef)
-		{
-			var productFound = DocumentationSet.Context.ProductsConfiguration.Products.TryGetValue(tocRef.Source.LocalPath.Trim('/'), out var tocProduct);
-			if (productFound && tocProduct is not null)
-				pageVersioning = tocProduct.VersioningSystem!;
-		}
-		else
-			pageVersioning = DocumentationSet.Context.VersionsConfiguration.VersioningSystems[VersioningSystemId.Stack];
+		var pageVersioning = VersionInferrerService.InferVersion(legacyPages);
 
 		var currentBaseVersion = $"{pageVersioning.Base.Major}.{pageVersioning.Base.Minor}+";