Umbraco-CMS/src/Umbraco.Web/Models/PublishedContent/PublishedValueLanguageFallback.cs

using Umbraco.Core.Models;
using Umbraco.Core.Models.PublishedContent;
using Umbraco.Core.Services;

namespace Umbraco.Web.Models.PublishedContent
{
    /// <summary>
    /// Provides a default implementation for <see cref="IPublishedValueFallback"/> that allows
    /// for use of fall-back languages
    /// </summary>
    /// <remarks>
    /// Inherits from <see cref="PublishedValueFallback" /> that implments what was available in v7.
    /// </remarks>
    public class PublishedValueLanguageFallback : PublishedValueFallback
    {
        private readonly ILocalizationService _localizationService;

        public PublishedValueLanguageFallback(ServiceContext services)
        {
            _localizationService = services.LocalizationService;
        }

        /// <inheritdoc />
        public override object GetValue(IPublishedProperty property, string culture, string segment, object defaultValue)
        {
            object value;
            if (TryGetValueFromFallbackLanguage(property, culture, segment, defaultValue, out value))
            {
                return value;
            }

            return base.GetValue(property, culture, segment, defaultValue);
        }

        /// <inheritdoc />
        public override T GetValue<T>(IPublishedProperty property, string culture, string segment, T defaultValue)
        {
            T value;
            if (TryGetValueFromFallbackLanguage(property, culture, segment, defaultValue, out value))
            {
                return value;
            }

            return base.GetValue(property, culture, segment, defaultValue);
        }

        /// <inheritdoc />
        public override object GetValue(IPublishedElement content, string alias, string culture, string segment, object defaultValue)
        {
            object value;
            if (TryGetValueFromFallbackLanguage(content, alias, culture, segment, defaultValue, out value))
            {
                return value;
            }

            return base.GetValue(content, alias, culture, segment, defaultValue);
        }

        /// <inheritdoc />
        public override T GetValue<T>(IPublishedElement content, string alias, string culture, string segment, T defaultValue)
        {
            T value;
            if (TryGetValueFromFallbackLanguage(content, alias, culture, segment, defaultValue, out value))
            {
                return value;
            }

            return base.GetValue(content, alias, culture, segment, defaultValue);
        }

        /// <inheritdoc />
        public override object GetValue(IPublishedContent content, string alias, string culture, string segment, object defaultValue, bool recurse, PublishedValueFallbackPriority fallbackPriority)
        {
            return GetValue<object>(content, alias, culture, segment, defaultValue, recurse, fallbackPriority);
        }

        /// <inheritdoc />
        public override T GetValue<T>(IPublishedContent content, string alias, string culture, string segment, T defaultValue, bool recurse, PublishedValueFallbackPriority fallbackPriority)
        {
            if (fallbackPriority == PublishedValueFallbackPriority.RecursiveTree)
            {
                var result = base.GetValue<T>(content, alias, culture, segment, defaultValue, recurse, PublishedValueFallbackPriority.RecursiveTree);
                if (ValueIsNotNullEmptyOrDefault(result, defaultValue))
                {
                    // We've prioritised recursive tree search and found a value, so can return it.
                    return result;
                }

                if (TryGetValueFromFallbackLanguage(content, alias, culture, segment, defaultValue, recurse, out result))
                {
                    return result;
                }

                return defaultValue;
            }

            if (fallbackPriority == PublishedValueFallbackPriority.FallbackLanguage)
            {
                T result;
                if (TryGetValueFromFallbackLanguage(content, alias, culture, segment, defaultValue, recurse, out result))
                {
                    return result;
                }
            }

            // No language fall back content found, so use base implementation
            return base.GetValue<T>(content, alias, culture, segment, defaultValue, recurse, fallbackPriority);
        }

        private bool TryGetValueFromFallbackLanguage<T>(IPublishedProperty property, string culture, string segment, T defaultValue, out T value)
        {
            if (string.IsNullOrEmpty(culture))
            {
                value = defaultValue;
                return false;
            }

            var language = _localizationService.GetLanguageByIsoCode(culture);
            if (language.FallbackLanguageId.HasValue == false)
            {
                value = defaultValue;
                return false;
            }

            var fallbackLanguageId = language.FallbackLanguageId;
            while (fallbackLanguageId.HasValue)
            {
                var fallbackLanguage = GetLanguageById(fallbackLanguageId.Value);
                value = property.Value(fallbackLanguage.IsoCode, segment, defaultValue);
                if (ValueIsNotNullEmptyOrDefault(value, defaultValue))
                {
                    return true;
                }

                fallbackLanguageId = fallbackLanguage.FallbackLanguageId;
            }

            value = defaultValue;
            return false;
        }

        private bool TryGetValueFromFallbackLanguage<T>(IPublishedElement content, string alias, string culture, string segment, T defaultValue, out T value)
        {
            if (string.IsNullOrEmpty(culture))
            {
                value = defaultValue;
                return false;
            }

            var language = _localizationService.GetLanguageByIsoCode(culture);
            if (language.FallbackLanguageId.HasValue == false)
            {
                value = defaultValue;
                return false;
            }

            var fallbackLanguageId = language.FallbackLanguageId;
            while (fallbackLanguageId.HasValue)
            {
                var fallbackLanguage = GetLanguageById(fallbackLanguageId.Value);
                value = content.Value(alias, fallbackLanguage.IsoCode, segment, defaultValue);
                if (ValueIsNotNullEmptyOrDefault(value, defaultValue))
                {
                    return true;
                }

                fallbackLanguageId = fallbackLanguage.FallbackLanguageId;
            }

            value = defaultValue;
            return false;
        }

        private bool TryGetValueFromFallbackLanguage<T>(IPublishedContent content, string alias, string culture, string segment, T defaultValue, bool recurse, out T value)
        {
            if (string.IsNullOrEmpty(culture))
            {
                value = defaultValue;
                return false;
            }

            var language = _localizationService.GetLanguageByIsoCode(culture);
            if (language.FallbackLanguageId.HasValue == false)
            {
                value = defaultValue;
                return false;
            }

            var fallbackLanguageId = language.FallbackLanguageId;
            while (fallbackLanguageId.HasValue)
            {
                var fallbackLanguage = GetLanguageById(fallbackLanguageId.Value);
                value = content.Value(alias, fallbackLanguage.IsoCode, segment, defaultValue, recurse);
                if (ValueIsNotNullEmptyOrDefault(value, defaultValue))
                {
                    return true;
                }

                fallbackLanguageId = fallbackLanguage.FallbackLanguageId;
            }

            value = defaultValue;
            return false;
        }

        private ILanguage GetLanguageById(int id)
        {
            return _localizationService.GetLanguageById(id);
        }

        private static bool ValueIsNotNullEmptyOrDefault<T>(T value, T defaultValue)
        {
            return value != null &&
                   string.IsNullOrEmpty(value.ToString()) == false &&
                   value.Equals(defaultValue) == false;
        }
    }
}