squidex/backend/src/Squidex.Domain.Apps.Core.Model/Contents/ContentData.cs

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

using Squidex.Infrastructure;

namespace Squidex.Domain.Apps.Core.Contents;

public sealed class ContentData : Dictionary<string, ContentFieldData?>, IEquatable<ContentData>
{
    public ContentData()
        : base(StringComparer.Ordinal)
    {
    }

    public ContentData(int capacity)
        : base(capacity, StringComparer.Ordinal)
    {
    }

    public ContentData(IDictionary<string, ContentFieldData?> source)
        : base(source, StringComparer.Ordinal)
    {
    }

    public ContentData AddField(string name, ContentFieldData? data)
    {
        Guard.NotNullOrEmpty(name);

        this[name] = data;

        return this;
    }

    public ContentData UseSameFields(ContentData? other)
    {
        if (other is not { Count: > 0 })
        {
            return this;
        }

        foreach (var (fieldName, fieldData) in this.ToList())
        {
            if (fieldData == null)
            {
                continue;
            }

            if (!other.TryGetValue(fieldName, out var otherField) || otherField == null)
            {
                continue;
            }

            if (otherField.Equals(fieldData))
            {
                this[fieldName] = otherField;
            }
            else
            {
                foreach (var (language, value) in fieldData.ToList())
                {
                    if (!otherField.TryGetValue(language, out var otherValue))
                    {
                        continue;
                    }

                    if (otherValue.Equals(value))
                    {
                        fieldData[language] = otherValue;
                    }
                }
            }
        }

        return this;
    }

    private static ContentData MergeTo(ContentData target, params ContentData[] sources)
    {
        Guard.NotEmpty(sources);

        if (sources.Length == 1 || sources.Skip(1).All(x => ReferenceEquals(x, sources[0])))
        {
            return sources[0];
        }

        foreach (var source in sources)
        {
            foreach (var (fieldName, sourceFieldData) in source)
            {
                if (sourceFieldData == null)
                {
                    continue;
                }

                if (Updates.IsUnset(sourceFieldData))
                {
                    target.Remove(fieldName);
                    continue;
                }

                var targetFieldData = target.GetOrAdd(fieldName, _ => []);

                if (targetFieldData == null)
                {
                    continue;
                }

                foreach (var (partition, sourceValue) in sourceFieldData)
                {
                    if (Updates.IsUnset(sourceValue))
                    {
                        targetFieldData.Remove(partition);
                    }
                    else
                    {
                        targetFieldData[partition] = sourceValue;
                    }
                }
            }
        }

        return target;
    }

    public static ContentData Merge(params ContentData[] contents)
    {
        return MergeTo([], contents);
    }

    public ContentData MergeInto(ContentData target)
    {
        return Merge(target, this);
    }

    public override bool Equals(object? obj)
    {
        return Equals(obj as ContentData);
    }

    public bool Equals(ContentData? other)
    {
        return other != null && (ReferenceEquals(this, other) || this.EqualsDictionary(other));
    }

    public override int GetHashCode()
    {
        return this.DictionaryHashCode();
    }

    public override string ToString()
    {
        return $"{{{string.Join(", ", this.Select(x => $"\"{x.Key}\":{x.Value}"))}}}";
    }

    public ContentData Clone()
    {
        var clone = new ContentData(Count);

        foreach (var (key, value) in this)
        {
            clone[key] = value?.Clone()!;
        }

        return clone;
    }
}