LastChanceContentFinder.cs

LastChanceContentFinder.cs

using System.Linq;
using Umbraco.Core;
using Umbraco.Core.Models;
using Umbraco.Web;
using Umbraco.Web.PublishedContentModels;
using Umbraco.Web.Routing;

namespace Web.ContentFinders
{
    public class LastChanceContentFinder : IContentFinder
    {
        public bool TryFindContent(PublishedContentRequest contentRequest)
        {
            if (!contentRequest.Is404) return contentRequest.PublishedContent != null;
            contentRequest.SetResponseStatus(404, "404 Page Not Found");
            contentRequest.PublishedContent = GetFileNotFound(contentRequest, contentRequest.Uri.Segments[1].TrimEnd("/"));
            return contentRequest.PublishedContent != null;
        } 

        private static IPublishedContent GetFileNotFound(PublishedContentRequest contentRequest, string languageCode)
        {
            var helper = new UmbracoHelper(contentRequest.RoutingContext.UmbracoContext);
            var homepage = helper.TypedContentAtRoot().FirstOrDefault(x => x.DocumentTypeAlias == Homepage.ModelTypeAlias);
            var fileNotFound = homepage?.FirstChild(x => x.DocumentTypeAlias == FileNotFound.ModelTypeAlias);
            return fileNotFound;
        }        
    }
}