Umbraco-CMS/src/Umbraco.Web/Routing/RedirectTrackingComponent.cs

using System;
using Umbraco.Core.Models;
using Umbraco.Core.Services;
using Umbraco.Core.Events;
using System.Collections.Generic;
using Umbraco.Core.Cache;
using Umbraco.Core.Components;
using Umbraco.Core.Services.Implement;
using Umbraco.Core.Sync;
using Umbraco.Web.Cache;
using Umbraco.Web.Composing;

namespace Umbraco.Web.Redirects
{
    public class RedirectTrackingComponent : IComponent
    {
        private const string ContextKey1 = "Umbraco.Web.Redirects.RedirectTrackingEventHandler.1";
        private const string ContextKey2 = "Umbraco.Web.Redirects.RedirectTrackingEventHandler.2";
        private const string ContextKey3 = "Umbraco.Web.Redirects.RedirectTrackingEventHandler.3";

        protected RedirectTrackingComponent()
        {
            // events are weird
            // on 'published' we 'could' get the old or the new route depending on event handlers order
            // so it is not reliable. getting the old route in 'publishing' to be sure and storing in http
            // context. then for the same reason, we have to process these old items only when the cache
            // is ready
            // when moving, the moved node is also published, which is causing all sorts of troubles with
            // descendants, so when moving, we lock events so that neither 'published' nor 'publishing'
            // are processed more than once
            // we cannot rely only on ContentCacheRefresher because when CacheUpdated triggers the old
            // route is gone
            //
            // this is all verrrry weird but it seems to work

            ContentService.Publishing += ContentService_Publishing;
            ContentService.Published += ContentService_Published;
            ContentService.Moving += ContentService_Moving;
            ContentService.Moved += ContentService_Moved;
            ContentCacheRefresher.CacheUpdated += ContentCacheRefresher_CacheUpdated;

            // kill all redirects once a content is deleted
            //ContentService.Deleted += ContentService_Deleted;
            // BUT, doing it here would prevent content deletion due to FK
            // so the rows are actually deleted by the ContentRepository (see GetDeleteClauses)

            // rolled back items have to be published, so publishing will take care of that
        }

        private static void ContentCacheRefresher_CacheUpdated(ContentCacheRefresher sender, CacheRefresherEventArgs args)
        {
            // sanity checks

            if (args.MessageType != MessageType.RefreshByPayload)
                throw new InvalidOperationException("ContentCacheRefresher MessageType should be ByPayload.");

            if (args.MessageObject == null) return;
            var payloads = args.MessageObject as ContentCacheRefresher.JsonPayload[];
            if (payloads == null)
                throw new InvalidOperationException("ContentCacheRefresher MessageObject should be JsonPayload[].");

            // manage routes

            var removeKeys = new List<int>();

            foreach (var oldRoute in OldRoutes)
            {
                // assuming we cannot have 'CacheUpdated' for only part of the infos else we'd need
                // to set a flag in 'Published' to indicate which entities have been refreshed ok
                CreateRedirect(oldRoute.Key, oldRoute.Value.Item1, oldRoute.Value.Item2);
                removeKeys.Add(oldRoute.Key);
            }

            foreach (var k in removeKeys)
                OldRoutes.Remove(k);
        }

        private static Dictionary<int, Tuple<Guid, string>> OldRoutes
        {
            get
            {
                var oldRoutes = (Dictionary<int, Tuple<Guid, string>>) UmbracoContext.Current.HttpContext.Items[ContextKey3];
                if (oldRoutes == null)
                    UmbracoContext.Current.HttpContext.Items[ContextKey3] = oldRoutes = new Dictionary<int, Tuple<Guid, string>>();
                return oldRoutes;
            }
        }

        private static bool LockedEvents
        {
            get { return Moving && UmbracoContext.Current.HttpContext.Items[ContextKey2] != null; }
            set
            {
                if (Moving && value)
                    UmbracoContext.Current.HttpContext.Items[ContextKey2] = true;
                else
                    UmbracoContext.Current.HttpContext.Items.Remove(ContextKey2);
            }
        }

        private static bool Moving
        {
            get { return UmbracoContext.Current.HttpContext.Items[ContextKey1] != null; }
            set
            {
                if (value)
                    UmbracoContext.Current.HttpContext.Items[ContextKey1] = true;
                else
                {
                    UmbracoContext.Current.HttpContext.Items.Remove(ContextKey1);
                    UmbracoContext.Current.HttpContext.Items.Remove(ContextKey2);
                }
            }
        }

        private static void ContentService_Publishing(IContentService sender, PublishEventArgs<IContent> args)
        {
            if (LockedEvents) return;

            var contentCache = UmbracoContext.Current.ContentCache;
            foreach (var entity in args.PublishedEntities)
            {
                var entityContent = contentCache.GetById(entity.Id);
                if (entityContent == null) continue;
                foreach (var x in entityContent.DescendantsOrSelf())
                {
                    var route = contentCache.GetRouteById(x.Id);
                    if (IsNotRoute(route)) continue;
                    OldRoutes[x.Id] = Tuple.Create(x.Key, route);
                }
            }

            LockedEvents = true; // we only want to see the "first batch"
        }

        private static void ContentService_Published(IContentService sender, PublishEventArgs<IContent> e)
        {
            // look note in CacheUpdated
            // we might want to set a flag on the entities we are seeing here
        }

        private static void ContentService_Moving(IContentService sender, MoveEventArgs<IContent> e)
        {
            //TODO: Use the new e.EventState to track state between Moving/Moved events!
            Moving = true;
        }

        private static void ContentService_Moved(IContentService sender, MoveEventArgs<IContent> e)
        {
            Moving = false;
            LockedEvents = false;
        }

        private static void CreateRedirect(int contentId, Guid contentKey, string oldRoute)
        {
            var contentCache = UmbracoContext.Current.ContentCache;
            var newRoute = contentCache.GetRouteById(contentId);
            if (IsNotRoute(newRoute) || oldRoute == newRoute) return;
            var redirectUrlService = Current.Services.RedirectUrlService;
            redirectUrlService.Register(oldRoute, contentKey);
        }

        private static bool IsNotRoute(string route)
        {
            // null if content not found
            // err/- if collision or anomaly or ...
            return route == null || route.StartsWith("err/");
        }
    }
}