Umbraco-CMS/src/Umbraco.Core/Models/ContentBaseExtensions.cs

using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Core.Strings;

namespace Umbraco.Extensions;

/// <summary>
///     Provides extension methods to IContentBase to get URL segments.
/// </summary>
public static class ContentBaseExtensions
{
    private static DefaultUrlSegmentProvider? _defaultUrlSegmentProvider;

    /// <summary>
    ///     Gets the URL segment for a specified content and culture.
    /// </summary>
    /// <param name="content">The content.</param>
    /// <param name="shortStringHelper"></param>
    /// <param name="urlSegmentProviders"></param>
    /// <param name="culture">The culture.</param>
    /// <param name="published">Whether to get the published or draft.</param>
    /// <returns>The URL segment.</returns>
    public static string? GetUrlSegment(this IContentBase content, IShortStringHelper shortStringHelper, IEnumerable<IUrlSegmentProvider> urlSegmentProviders, string? culture = null, bool published = true)
    {
        if (content == null)
        {
            throw new ArgumentNullException(nameof(content));
        }

        if (urlSegmentProviders == null)
        {
            throw new ArgumentNullException(nameof(urlSegmentProviders));
        }

        var url = urlSegmentProviders.Select(p => p.GetUrlSegment(content, published, culture)).FirstOrDefault(u => u != null);
        if (url == null)
        {
            if (_defaultUrlSegmentProvider == null)
            {
                _defaultUrlSegmentProvider = new DefaultUrlSegmentProvider(shortStringHelper);
            }

            url = _defaultUrlSegmentProvider.GetUrlSegment(content, culture); // be safe
        }

        return url;
    }
}