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

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

using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Squidex.Domain.Apps.Core;
using Squidex.Domain.Apps.Core.Contents;
using Squidex.Domain.Apps.Entities.Backup;
using Squidex.Domain.Apps.Entities.Contents.DomainObject;
using Squidex.Domain.Apps.Events.Apps;
using Squidex.Domain.Apps.Events.Contents;
using Squidex.Domain.Apps.Events.Schemas;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Commands;
using Squidex.Infrastructure.EventSourcing;
using Squidex.Infrastructure.Json.Objects;

namespace Squidex.Domain.Apps.Entities.Contents
{
    public sealed class BackupContents : IBackupHandler
    {
        private delegate void ObjectSetter(IReadOnlyDictionary<string, IJsonValue> obj, string key, IJsonValue value);

        private const string UrlsFile = "Urls.json";

        private static readonly ObjectSetter JsonSetter = (obj, key, value) =>
        {
            ((JsonObject)obj).Add(key, value);
        };

        private static readonly ObjectSetter FieldSetter = (obj, key, value) =>
        {
            ((ContentFieldData)obj)[key] = value;
        };

        private readonly Dictionary<DomainId, HashSet<DomainId>> contentIdsBySchemaId = new Dictionary<DomainId, HashSet<DomainId>>();
        private readonly Rebuilder rebuilder;
        private readonly IUrlGenerator urlGenerator;
        private Urls? assetsUrlNew;
        private Urls? assetsUrlOld;

        public string Name { get; } = "Contents";

        public sealed class Urls
        {
            public string Assets { get; set; }

            public string AssetsApp { get; set; }
        }

        public BackupContents(Rebuilder rebuilder, IUrlGenerator urlGenerator)
        {
            Guard.NotNull(rebuilder, nameof(rebuilder));
            Guard.NotNull(urlGenerator, nameof(urlGenerator));

            this.rebuilder = rebuilder;

            this.urlGenerator = urlGenerator;
        }

        public async Task BackupEventAsync(Envelope<IEvent> @event, BackupContext context)
        {
            if (@event.Payload is AppCreated appCreated)
            {
                var urls = GetUrls(appCreated.Name);

                await context.Writer.WriteJsonAsync(UrlsFile, urls);
            }
        }

        public async Task<bool> RestoreEventAsync(Envelope<IEvent> @event, RestoreContext context)
        {
            switch (@event.Payload)
            {
                case AppCreated appCreated:
                    assetsUrlNew = GetUrls(appCreated.Name);
                    assetsUrlOld = await ReadUrlsAsync(context.Reader);
                    break;
                case SchemaDeleted schemaDeleted:
                    contentIdsBySchemaId.Remove(schemaDeleted.SchemaId.Id);
                    break;
                case ContentCreated contentCreated:
                    contentIdsBySchemaId.GetOrAddNew(contentCreated.SchemaId.Id)
                        .Add(@event.Headers.AggregateId());

                    if (assetsUrlNew != null && assetsUrlOld != null)
                    {
                        ReplaceAssetUrl(contentCreated.Data);
                    }

                    break;
                case ContentUpdated contentUpdated:
                    if (assetsUrlNew != null && assetsUrlOld != null)
                    {
                        ReplaceAssetUrl(contentUpdated.Data);
                    }

                    break;
            }

            return true;
        }

        private void ReplaceAssetUrl(ContentData data)
        {
            foreach (var field in data.Values)
            {
                if (field != null)
                {
                    ReplaceAssetUrl(field, FieldSetter);
                }
            }
        }

        private void ReplaceAssetUrl(IReadOnlyDictionary<string, IJsonValue> source, ObjectSetter setter)
        {
            List<(string, string)>? replacements = null;

            foreach (var (key, value) in source)
            {
                switch (value)
                {
                    case JsonString s:
                        {
                            var newValue = s.Value;

                            newValue = newValue.Replace(assetsUrlOld!.AssetsApp, assetsUrlNew!.AssetsApp);

                            if (!ReferenceEquals(newValue, s.Value))
                            {
                                replacements ??= new List<(string, string)>();
                                replacements.Add((key, newValue));
                                break;
                            }

                            newValue = newValue.Replace(assetsUrlOld!.Assets, assetsUrlNew!.Assets);

                            if (!ReferenceEquals(newValue, s.Value))
                            {
                                replacements ??= new List<(string, string)>();
                                replacements.Add((key, newValue));
                                break;
                            }
                        }

                        break;

                    case JsonArray arr:
                        ReplaceAssetUrl(arr);
                        break;

                    case JsonObject obj:
                        ReplaceAssetUrl(obj, JsonSetter);
                        break;
                }
            }

            if (replacements != null)
            {
                foreach (var (key, newValue) in replacements)
                {
                    setter(source, key, JsonValue.Create(newValue));
                }
            }
        }

        private void ReplaceAssetUrl(JsonArray source)
        {
            for (var i = 0; i < source.Count; i++)
            {
                var value = source[i];

                switch (value)
                {
                    case JsonString s:
                        {
                            var newValue = s.Value;

                            newValue = newValue.Replace(assetsUrlOld!.AssetsApp, assetsUrlNew!.AssetsApp);

                            if (!ReferenceEquals(newValue, s.Value))
                            {
                                source[i] = JsonValue.Create(newValue);
                                break;
                            }

                            newValue = newValue.Replace(assetsUrlOld!.Assets, assetsUrlNew!.Assets);

                            if (!ReferenceEquals(newValue, s.Value))
                            {
                                source[i] = JsonValue.Create(newValue);
                                break;
                            }
                        }

                        break;

                    case JsonArray:
                        break;

                    case JsonObject obj:
                        ReplaceAssetUrl(obj, JsonSetter);
                        break;
                }
            }
        }

        public async Task RestoreAsync(RestoreContext context)
        {
            var ids = contentIdsBySchemaId.Values.SelectMany(x => x);

            if (ids.Any())
            {
                await rebuilder.InsertManyAsync<ContentDomainObject, ContentDomainObject.State>(ids);
            }
        }

        private static async Task<Urls?> ReadUrlsAsync(IBackupReader reader)
        {
            try
            {
                return await reader.ReadJsonAsync<Urls>(UrlsFile);
            }
            catch
            {
                return null;
            }
        }

        private Urls GetUrls(string appName)
        {
            return new Urls
            {
                Assets = urlGenerator.AssetContentBase(),
                AssetsApp = urlGenerator.AssetContentBase(appName)
            };
        }
    }
}