squidex/backend/src/Squidex.Domain.Apps.Entities/Contents/ReferencesFluidExtension.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text.Encodings.Web;
using System.Threading.Tasks;
using Fluid;
using Fluid.Ast;
using Fluid.Values;
using Microsoft.Extensions.DependencyInjection;
using Squidex.Domain.Apps.Core.Rules.EnrichedEvents;
using Squidex.Domain.Apps.Core.Templates;
using Squidex.Infrastructure;

#pragma warning disable CA1826 // Do not use Enumerable methods on indexable collections

namespace Squidex.Domain.Apps.Entities.Contents
{
    public sealed class ReferencesFluidExtension : IFluidExtension
    {
        private static readonly FluidValue ErrorNullReference = FluidValue.Create(null);
        private readonly IServiceProvider serviceProvider;

        private sealed class ReferenceTag : AppTag
        {
            private readonly IContentQueryService contentQuery;

            public ReferenceTag(IServiceProvider serviceProvider)
                : base(serviceProvider)
            {
                contentQuery = serviceProvider.GetRequiredService<IContentQueryService>();
            }

            public override async ValueTask<Completion> WriteToAsync(TextWriter writer, TextEncoder encoder, TemplateContext context, FilterArgument[] arguments)
            {
                if (arguments.Length == 2 && context.GetValue("event")?.ToObjectValue() is EnrichedEvent enrichedEvent)
                {
                    var id = await arguments[1].Expression.EvaluateAsync(context);

                    var content = await ResolveContentAsync(AppProvider, contentQuery, enrichedEvent.AppId.Id, id);

                    if (content != null)
                    {
                        var name = (await arguments[0].Expression.EvaluateAsync(context)).ToStringValue();

                        context.SetValue(name, content);
                    }
                }

                return Completion.Normal;
            }
        }

        public ReferencesFluidExtension(IServiceProvider serviceProvider)
        {
            Guard.NotNull(serviceProvider, nameof(serviceProvider));

            this.serviceProvider = serviceProvider;
        }

        public void RegisterGlobalTypes(IMemberAccessStrategy memberAccessStrategy)
        {
            memberAccessStrategy.Register<IContentEntity>();
            memberAccessStrategy.Register<IEntity>();
            memberAccessStrategy.Register<IEntityWithCreatedBy>();
            memberAccessStrategy.Register<IEntityWithLastModifiedBy>();
            memberAccessStrategy.Register<IEntityWithVersion>();
            memberAccessStrategy.Register<IEnrichedContentEntity>();

            AddReferenceFilter();
        }

        private void AddReferenceFilter()
        {
            var appProvider = serviceProvider.GetRequiredService<IAppProvider>();

            var contentQuery = serviceProvider.GetRequiredService<IContentQueryService>();

            TemplateContext.GlobalFilters.AddAsyncFilter("reference", async (input, arguments, context) =>
            {
                if (context.GetValue("event")?.ToObjectValue() is EnrichedEvent enrichedEvent)
                {
                    var content = await ResolveContentAsync(appProvider, contentQuery, enrichedEvent.AppId.Id, input);

                    if (content == null)
                    {
                        return ErrorNullReference;
                    }

                    return FluidValue.Create(content);
                }

                return ErrorNullReference;
            });
        }

        public void RegisterLanguageExtensions(FluidParserFactory factory)
        {
            factory.RegisterTag("reference", new ReferenceTag(serviceProvider));
        }

        private static async Task<IContentEntity?> ResolveContentAsync(IAppProvider appProvider, IContentQueryService contentQuery, DomainId appId, FluidValue id)
        {
            var app = await appProvider.GetAppAsync(appId);

            if (app == null)
            {
                return null;
            }

            var domainId = DomainId.Create(id.ToStringValue());
            var domainIds = new List<DomainId> { domainId };

            var requestContext =
                Context.Admin(app).Clone(b => b
                    .WithUnpublished()
                    .WithoutTotal());

            var contents = await contentQuery.QueryAsync(requestContext, Q.Empty.WithIds(domainIds));
            var content = contents.FirstOrDefault();

            return content;
        }
    }
}