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

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

using System.Runtime.CompilerServices;
using Squidex.Domain.Apps.Core.Contents;
using Squidex.Domain.Apps.Core.HandleRules;
using Squidex.Domain.Apps.Core.Rules.EnrichedEvents;
using Squidex.Domain.Apps.Core.Rules.Triggers;
using Squidex.Domain.Apps.Core.Scripting;
using Squidex.Domain.Apps.Entities.Contents.Repositories;
using Squidex.Domain.Apps.Events;
using Squidex.Domain.Apps.Events.Contents;
using Squidex.Infrastructure;
using Squidex.Infrastructure.EventSourcing;
using Squidex.Infrastructure.Reflection;
using Squidex.Text;

namespace Squidex.Domain.Apps.Entities.Contents
{
    public sealed class ContentChangedTriggerHandler : IRuleTriggerHandler
    {
        private readonly IScriptEngine scriptEngine;
        private readonly IContentLoader contentLoader;
        private readonly IContentRepository contentRepository;

        public bool CanCreateSnapshotEvents => true;

        public Type TriggerType => typeof(ContentChangedTriggerV2);

        public bool Handles(AppEvent appEvent)
        {
            return appEvent is ContentEvent;
        }

        public ContentChangedTriggerHandler(
            IScriptEngine scriptEngine,
            IContentLoader contentLoader,
            IContentRepository contentRepository)
        {
            this.scriptEngine = scriptEngine;
            this.contentLoader = contentLoader;
            this.contentRepository = contentRepository;
        }

        public async IAsyncEnumerable<EnrichedEvent> CreateSnapshotEventsAsync(RuleContext context,
            [EnumeratorCancellation] CancellationToken ct)
        {
            var trigger = (ContentChangedTriggerV2)context.Rule.Trigger;

            var schemaIds =
                trigger.Schemas?.Count > 0 ?
                trigger.Schemas.Select(x => x.SchemaId).Distinct().ToHashSet() :
                null;

            await foreach (var content in contentRepository.StreamAll(context.AppId.Id, schemaIds, ct))
            {
                var result = new EnrichedContentEvent
                {
                    Type = EnrichedContentEventType.Created
                };

                SimpleMapper.Map(content, result);

                result.Actor = content.LastModifiedBy;
                result.Name = $"ContentQueried({content.SchemaId.Name.ToPascalCase()})";

                yield return result;
            }
        }

        public async IAsyncEnumerable<EnrichedEvent> CreateEnrichedEventsAsync(Envelope<AppEvent> @event, RuleContext context,
            [EnumeratorCancellation] CancellationToken ct)
        {
            var contentEvent = (ContentEvent)@event.Payload;

            var result = new EnrichedContentEvent();

            var content =
                await contentLoader.GetAsync(
                    contentEvent.AppId.Id,
                    contentEvent.ContentId,
                    @event.Headers.EventStreamNumber());

            if (content != null)
            {
                SimpleMapper.Map(content, result);
            }

            switch (@event.Payload)
            {
                case ContentCreated:
                    result.Type = EnrichedContentEventType.Created;
                    break;
                case ContentDeleted:
                    result.Type = EnrichedContentEventType.Deleted;
                    break;
                case ContentStatusChanged e when e.Change == StatusChange.Published:
                    result.Type = EnrichedContentEventType.Published;
                    break;
                case ContentStatusChanged e when e.Change == StatusChange.Unpublished:
                    result.Type = EnrichedContentEventType.Unpublished;
                    break;
                case ContentStatusChanged e when e.Change == StatusChange.Change:
                    result.Type = EnrichedContentEventType.StatusChanged;
                    break;
                case ContentUpdated:
                    {
                        result.Type = EnrichedContentEventType.Updated;

                        if (content != null)
                        {
                            var previousContent =
                                await contentLoader.GetAsync(
                                    content.AppId.Id,
                                    content.Id,
                                    content.Version - 1);

                            if (previousContent != null)
                            {
                                result.DataOld = previousContent.Data;
                            }
                        }

                        break;
                    }
            }

            yield return result;
        }

        public string? GetName(AppEvent @event)
        {
            switch (@event)
            {
                case ContentCreated e:
                    return $"{e.SchemaId.Name.ToPascalCase()}Created";
                case ContentDeleted e:
                    return $"{e.SchemaId.Name.ToPascalCase()}Deleted";
                case ContentStatusChanged e when e.Change == StatusChange.Published:
                    return $"{e.SchemaId.Name.ToPascalCase()}Published";
                case ContentStatusChanged e when e.Change == StatusChange.Unpublished:
                    return $"{e.SchemaId.Name.ToPascalCase()}Unpublished";
                case ContentStatusChanged e when e.Change == StatusChange.Change:
                    return $"{e.SchemaId.Name.ToPascalCase()}StatusChanged";
                case ContentUpdated e:
                    return $"{e.SchemaId.Name.ToPascalCase()}Updated";
            }

            return null;
        }

        public bool Trigger(Envelope<AppEvent> @event, RuleContext context)
        {
            var trigger = (ContentChangedTriggerV2)context.Rule.Trigger;

            if (trigger.HandleAll)
            {
                return true;
            }

            if (trigger.Schemas != null)
            {
                var contentEvent = (ContentEvent)@event.Payload;

                foreach (var schema in trigger.Schemas)
                {
                    if (MatchsSchema(schema, contentEvent.SchemaId))
                    {
                        return true;
                    }
                }
            }

            return false;
        }

        public bool Trigger(EnrichedEvent @event, RuleContext context)
        {
            var trigger = (ContentChangedTriggerV2)context.Rule.Trigger;

            if (trigger.HandleAll)
            {
                return true;
            }

            if (trigger.Schemas != null)
            {
                var contentEvent = (EnrichedContentEvent)@event;

                foreach (var schema in trigger.Schemas)
                {
                    if (MatchsSchema(schema, contentEvent.SchemaId) && MatchsCondition(schema, contentEvent))
                    {
                        return true;
                    }
                }
            }

            return false;
        }

        private static bool MatchsSchema(ContentChangedTriggerSchemaV2? schema, NamedId<DomainId> schemaId)
        {
            return schemaId != null && schemaId.Id == schema?.SchemaId;
        }

        private bool MatchsCondition(ContentChangedTriggerSchemaV2 schema, EnrichedSchemaEventBase @event)
        {
            if (string.IsNullOrWhiteSpace(schema.Condition))
            {
                return true;
            }

            // Script vars are just wrappers over dictionaries for better performance.
            var vars = new EventScriptVars
            {
                Event = @event
            };

            return scriptEngine.Evaluate(vars, schema.Condition);
        }
    }
}