squidex/backend/src/Squidex.Domain.Apps.Entities/Apps/DomainObject/AppDomainObject.cs

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

using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Squidex.Domain.Apps.Core.Apps;
using Squidex.Domain.Apps.Entities.Apps.Commands;
using Squidex.Domain.Apps.Entities.Apps.DomainObject.Guards;
using Squidex.Domain.Apps.Entities.Billing;
using Squidex.Domain.Apps.Events;
using Squidex.Domain.Apps.Events.Apps;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Commands;
using Squidex.Infrastructure.EventSourcing;
using Squidex.Infrastructure.Reflection;
using Squidex.Infrastructure.States;
using Squidex.Shared.Users;

#pragma warning disable MA0022 // Return Task.FromResult instead of returning null

namespace Squidex.Domain.Apps.Entities.Apps.DomainObject;

public partial class AppDomainObject : DomainObject<AppDomainObject.State>
{
    private readonly IServiceProvider serviceProvider;

    public AppDomainObject(DomainId id, IPersistenceFactory<State> persistence, ILogger<AppDomainObject> log,
        IServiceProvider serviceProvider)
        : base(id, persistence, log)
    {
        this.serviceProvider = serviceProvider;
    }

    protected override bool IsDeleted(State snapshot)
    {
        return snapshot.IsDeleted;
    }

    protected override bool CanAccept(ICommand command)
    {
        if (Snapshot.Id == default)
        {
            return true;
        }

        return command is AppCommandBase c && c.AggregateId == Snapshot.Id;
    }

    protected override bool CanAccept(ICommand command, DomainObjectState state)
    {
        switch (state)
        {
            case DomainObjectState.Undefined:
                return command is CreateApp;
            case DomainObjectState.Empty:
                return command is CreateApp;
            case DomainObjectState.Created:
                return command is not CreateApp;
            default:
                return false;
        }
    }

    public override Task<CommandResult> ExecuteAsync(IAggregateCommand command,
        CancellationToken ct)
    {
        switch (command)
        {
            case CreateApp create:
                return ApplyReturn(create, c =>
                {
                    GuardApp.CanCreate(c);

                    Create(c);

                    return Snapshot;
                }, ct);

            case UpdateApp update:
                return ApplyReturn(update, c =>
                {
                    GuardApp.CanUpdate(c);

                    Update(c);

                    return Snapshot;
                }, ct);

            case TransferToTeam transfer:
                return ApplyReturnAsync(transfer, async (c, ct) =>
                {
                    await GuardApp.CanTransfer(c, Snapshot, AppProvider, ct);

                    Transfer(c);

                    return Snapshot;
                }, ct);

            case UpdateAppSettings updateSettings:
                return ApplyReturn(updateSettings, c =>
                {
                    GuardApp.CanUpdateSettings(c);

                    UpdateSettings(c);

                    return Snapshot;
                }, ct);

            case UploadAppImage uploadImage:
                return ApplyReturn(uploadImage, c =>
                {
                    GuardApp.CanUploadImage(c);

                    UploadImage(c);

                    return Snapshot;
                }, ct);

            case RemoveAppImage removeImage:
                return ApplyReturn(removeImage, c =>
                {
                    GuardApp.CanRemoveImage(c);

                    RemoveImage(c);

                    return Snapshot;
                }, ct);

            case ConfigureAssetScripts configureAssetScripts:
                return ApplyReturn(configureAssetScripts, c =>
                {
                    GuardApp.CanUpdateAssetScripts(c);

                    ConfigureAssetScripts(c);

                    return Snapshot;
                }, ct);

            case AssignContributor assignContributor:
                return ApplyReturnAsync(assignContributor, async (c, ct) =>
                {
                    var (plan, _, _) = await UsageGate.GetPlanForAppAsync(Snapshot, false, ct);

                    await GuardAppContributors.CanAssign(c, Snapshot, Users, plan);

                    AssignContributor(c, !Snapshot.Contributors.ContainsKey(assignContributor.ContributorId));

                    return Snapshot;
                }, ct);

            case RemoveContributor removeContributor:
                return ApplyReturn(removeContributor, c =>
                {
                    GuardAppContributors.CanRemove(c, Snapshot);

                    RemoveContributor(c);

                    return Snapshot;
                }, ct);

            case AttachClient attachClient:
                return ApplyReturn(attachClient, c =>
                {
                    GuardAppClients.CanAttach(c, Snapshot);

                    AttachClient(c);

                    return Snapshot;
                }, ct);

            case UpdateClient updateClient:
                return ApplyReturn(updateClient, c =>
                {
                    GuardAppClients.CanUpdate(c, Snapshot);

                    UpdateClient(c);

                    return Snapshot;
                }, ct);

            case RevokeClient revokeClient:
                return ApplyReturn(revokeClient, c =>
                {
                    GuardAppClients.CanRevoke(c, Snapshot);

                    RevokeClient(c);

                    return Snapshot;
                }, ct);

            case AddWorkflow addWorkflow:
                return ApplyReturn(addWorkflow, c =>
                {
                    GuardAppWorkflows.CanAdd(c);

                    AddWorkflow(c);

                    return Snapshot;
                }, ct);

            case UpdateWorkflow updateWorkflow:
                return ApplyReturn(updateWorkflow, c =>
                {
                    GuardAppWorkflows.CanUpdate(c, Snapshot);

                    UpdateWorkflow(c);

                    return Snapshot;
                }, ct);

            case DeleteWorkflow deleteWorkflow:
                return ApplyReturn(deleteWorkflow, c =>
                {
                    GuardAppWorkflows.CanDelete(c, Snapshot);

                    DeleteWorkflow(c);

                    return Snapshot;
                }, ct);

            case AddLanguage addLanguage:
                return ApplyReturn(addLanguage, c =>
                {
                    GuardAppLanguages.CanAdd(c, Snapshot);

                    AddLanguage(c);

                    return Snapshot;
                }, ct);

            case RemoveLanguage removeLanguage:
                return ApplyReturn(removeLanguage, c =>
                {
                    GuardAppLanguages.CanRemove(c, Snapshot);

                    RemoveLanguage(c);

                    return Snapshot;
                }, ct);

            case UpdateLanguage updateLanguage:
                return ApplyReturn(updateLanguage, c =>
                {
                    GuardAppLanguages.CanUpdate(c, Snapshot);

                    UpdateLanguage(c);

                    return Snapshot;
                }, ct);

            case AddRole addRole:
                return ApplyReturn(addRole, c =>
                {
                    GuardAppRoles.CanAdd(c, Snapshot);

                    AddRole(c);

                    return Snapshot;
                }, ct);

            case DeleteRole deleteRole:
                return ApplyReturn(deleteRole, c =>
                {
                    GuardAppRoles.CanDelete(c, Snapshot);

                    DeleteRole(c);

                    return Snapshot;
                }, ct);

            case UpdateRole updateRole:
                return ApplyReturn(updateRole, c =>
                {
                    GuardAppRoles.CanUpdate(c, Snapshot);

                    UpdateRole(c);

                    return Snapshot;
                }, ct);

            case DeleteApp delete:
                return ApplyAsync(delete, async (c, ct) =>
                {
                    await BillingManager.UnsubscribeAsync(c.Actor.Identifier, Snapshot, default);

                    DeleteApp(c);
                }, ct);

            case ChangePlan changePlan:
                return ApplyReturnAsync(changePlan, async (c, ct) =>
                {
                    GuardApp.CanChangePlan(c, Snapshot, BillingPlans);

                    if (string.Equals(FreePlan?.Id, c.PlanId, StringComparison.Ordinal))
                    {
                        if (!c.FromCallback)
                        {
                            await BillingManager.UnsubscribeAsync(c.Actor.Identifier, Snapshot, default);
                        }

                        ResetPlan(c);

                        return new PlanChangedResult(c.PlanId, true, null);
                    }
                    else
                    {
                        if (!c.FromCallback)
                        {
                            var redirectUri = await BillingManager.MustRedirectToPortalAsync(c.Actor.Identifier, Snapshot, c.PlanId, ct);

                            if (redirectUri != null)
                            {
                                return new PlanChangedResult(c.PlanId, false, redirectUri);
                            }

                            await BillingManager.SubscribeAsync(c.Actor.Identifier, Snapshot, changePlan.PlanId, default);
                        }

                        ChangePlan(c);

                        return new PlanChangedResult(c.PlanId);
                    }
                }, ct);

            default:
                ThrowHelper.NotSupportedException();
                return default!;
        }
    }

    private void Create(CreateApp command)
    {
        var appId = NamedId.Of(command.AppId, command.Name);

        void RaiseInitial<T>(T @event) where T : AppEvent
        {
            Raise(command, @event, appId);
        }

        RaiseInitial(new AppCreated());

        var actor = command.Actor;

        if (actor.IsUser)
        {
            RaiseInitial(new AppContributorAssigned { ContributorId = actor.Identifier, Role = Role.Owner });
        }

        var settings = serviceProvider.GetService<InitialSettings>()?.Settings;

        if (settings != null)
        {
            RaiseInitial(new AppSettingsUpdated { Settings = settings });
        }
    }

    private void ChangePlan(ChangePlan command)
    {
        Raise(command, new AppPlanChanged());
    }

    private void ResetPlan(ChangePlan command)
    {
        Raise(command, new AppPlanReset());
    }

    private void Update(UpdateApp command)
    {
        Raise(command, new AppUpdated());
    }

    private void Transfer(TransferToTeam command)
    {
        Raise(command, new AppTransfered());
    }

    private void UpdateSettings(UpdateAppSettings command)
    {
        Raise(command, new AppSettingsUpdated());
    }

    private void ConfigureAssetScripts(ConfigureAssetScripts command)
    {
        Raise(command, new AppAssetsScriptsConfigured());
    }

    private void UpdateClient(UpdateClient command)
    {
        Raise(command, new AppClientUpdated());
    }

    private void UploadImage(UploadAppImage command)
    {
        Raise(command, new AppImageUploaded { Image = new AppImage(command.File.MimeType) });
    }

    private void RemoveImage(RemoveAppImage command)
    {
        Raise(command, new AppImageRemoved());
    }

    private void UpdateLanguage(UpdateLanguage command)
    {
        Raise(command, new AppLanguageUpdated());
    }

    private void AssignContributor(AssignContributor command, bool isAdded)
    {
        Raise(command, new AppContributorAssigned { IsAdded = isAdded });
    }

    private void RemoveContributor(RemoveContributor command)
    {
        Raise(command, new AppContributorRemoved());
    }

    private void AttachClient(AttachClient command)
    {
        Raise(command, new AppClientAttached());
    }

    private void RevokeClient(RevokeClient command)
    {
        Raise(command, new AppClientRevoked());
    }

    private void AddWorkflow(AddWorkflow command)
    {
        Raise(command, new AppWorkflowAdded());
    }

    private void UpdateWorkflow(UpdateWorkflow command)
    {
        Raise(command, new AppWorkflowUpdated());
    }

    private void DeleteWorkflow(DeleteWorkflow command)
    {
        Raise(command, new AppWorkflowDeleted());
    }

    private void AddLanguage(AddLanguage command)
    {
        Raise(command, new AppLanguageAdded());
    }

    private void RemoveLanguage(RemoveLanguage command)
    {
        Raise(command, new AppLanguageRemoved());
    }

    private void AddRole(AddRole command)
    {
        Raise(command, new AppRoleAdded());
    }

    private void DeleteRole(DeleteRole command)
    {
        Raise(command, new AppRoleDeleted());
    }

    private void UpdateRole(UpdateRole command)
    {
        Raise(command, new AppRoleUpdated());
    }

    private void DeleteApp(DeleteApp command)
    {
        Raise(command, new AppDeleted());
    }

    private void Raise<T, TEvent>(T command, TEvent @event, NamedId<DomainId>? id = null) where T : class where TEvent : AppEvent
    {
        SimpleMapper.Map(command, @event);

        @event.AppId ??= id ?? Snapshot.NamedId();

        RaiseEvent(Envelope.Create(@event));
    }

    private IAppProvider AppProvider
    {
        get => serviceProvider.GetRequiredService<IAppProvider>();
    }

    private IBillingPlans BillingPlans
    {
        get => serviceProvider.GetRequiredService<IBillingPlans>();
    }

    private IBillingManager BillingManager
    {
        get => serviceProvider.GetRequiredService<IBillingManager>();
    }

    private IUsageGate UsageGate
    {
        get => serviceProvider.GetRequiredService<IUsageGate>();
    }

    private IUserResolver Users
    {
        get => serviceProvider.GetRequiredService<IUserResolver>();
    }

    private Plan FreePlan
    {
        get => BillingPlans.GetFreePlan();
    }
}