Umbraco-CMS/src/Umbraco.Core/Models/Mapping/MapperContextExtensions.cs

using Umbraco.Cms.Core.Mapping;

namespace Umbraco.Extensions
{
    /// <summary>
    /// Provides extension methods for the <see cref="MapperContext"/> class.
    /// </summary>
    public static class MapperContextExtensions
    {
        private const string CultureKey = "Map.Culture";
        private const string SegmentKey = "Map.Segment";
        private const string IncludedPropertiesKey = "Map.IncludedProperties";

        /// <summary>
        /// Gets the context culture.
        /// </summary>
        public static string? GetCulture(this MapperContext context)
        {
            return context.HasItems && context.Items.TryGetValue(CultureKey, out var obj) && obj is string s ? s : null;
        }

        /// <summary>
        /// Gets the context segment.
        /// </summary>
        public static string? GetSegment(this MapperContext context)
        {
            return context.HasItems && context.Items.TryGetValue(SegmentKey, out var obj) && obj is string s ? s : null;
        }

        /// <summary>
        /// Sets a context culture.
        /// </summary>
        public static void SetCulture(this MapperContext context, string? culture)
        {
            if (culture is not null)
            {
                context.Items[CultureKey] = culture;
            }
        }

        /// <summary>
        /// Sets a context segment.
        /// </summary>
        public static void SetSegment(this MapperContext context, string? segment)
        {
            if (segment is not null)
            {
                context.Items[SegmentKey] = segment;
            }
        }

        /// <summary>
        /// Get included properties.
        /// </summary>
        public static string[]? GetIncludedProperties(this MapperContext context)
        {
            return context.HasItems && context.Items.TryGetValue(IncludedPropertiesKey, out var obj) && obj is string[] s ? s : null;
        }

        /// <summary>
        /// Sets included properties.
        /// </summary>
        public static void SetIncludedProperties(this MapperContext context, string[] properties)
        {
            context.Items[IncludedPropertiesKey] = properties;
        }
    }
}