Umbraco-CMS/tests/Umbraco.Tests.Common/Builders/ContentEditingBuilder.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using Umbraco.Cms.Core.Models;
using Umbraco.Cms.Core.Models.ContentEditing;
using Umbraco.Cms.Tests.Common.Builders.Extensions;
using Umbraco.Cms.Tests.Common.Builders.Interfaces;
using Umbraco.Cms.Tests.Common.Builders.Interfaces.ContentCreateModel;

namespace Umbraco.Cms.Tests.Common.Builders;

public class ContentEditingBuilder
    : BuilderBase<ContentCreateModel>,
        IWithInvariantNameBuilder,
        IWithInvariantPropertiesBuilder,
        IWithVariantsBuilder,
        IWithKeyBuilder,
        IWithContentTypeKeyBuilder,
        IWithParentKeyBuilder,
        IWithTemplateKeyBuilder
{
    private IContentType _contentType;
    private ContentTypeBuilder _contentTypeBuilder;
    private IEnumerable<PropertyValueModel> _invariantProperties = [];
    private IEnumerable<VariantModel> _variants = [];
    private Guid _contentTypeKey;
    private Guid? _parentKey;
    private Guid? _templateKey;
    private Guid? _key;
    private string _invariantName;

    Guid? IWithKeyBuilder.Key
    {
        get => _key;
        set => _key = value;
    }

    string IWithInvariantNameBuilder.InvariantName
    {
        get => _invariantName;
        set => _invariantName = value;
    }

    IEnumerable<PropertyValueModel> IWithInvariantPropertiesBuilder.InvariantProperties
    {
        get => _invariantProperties;
        set => _invariantProperties = value;
    }

    IEnumerable<VariantModel> IWithVariantsBuilder.Variants
    {
        get => _variants;
        set => _variants = value;
    }

    Guid? IWithParentKeyBuilder.ParentKey
    {
        get => _parentKey;
        set => _parentKey = value;
    }

    Guid IWithContentTypeKeyBuilder.ContentTypeKey
    {
        get => _contentTypeKey;
        set => _contentTypeKey = value;
    }

    Guid? IWithTemplateKeyBuilder.TemplateKey
    {
        get => _templateKey;
        set => _templateKey = value;
    }

    public ContentEditingBuilder WithInvariantName(string invariantName)
    {
        _invariantName = invariantName;
        return this;
    }

    public ContentEditingBuilder WithInvariantProperty(string alias, object value)
    {
        var property = new PropertyValueModel { Alias = alias, Value = value };
        _invariantProperties = _invariantProperties.Concat(new[] { property });
        return this;
    }

    public ContentEditingBuilder AddVariant(string culture, string segment, string name,
        IEnumerable<PropertyValueModel> properties)
    {
        var variant = new VariantModel { Culture = culture, Segment = segment, Name = name, Properties = properties };
        _variants = _variants.Concat(new[] { variant });
        return this;
    }

    public ContentEditingBuilder WithParentKey(Guid parentKey)
    {
        _parentKey = parentKey;
        return this;
    }

    public ContentEditingBuilder WithTemplateKey(Guid templateKey)
    {
        _templateKey = templateKey;
        return this;
    }

    public ContentEditingBuilder WithContentType(IContentType contentType)
    {
        _contentTypeBuilder = null;
        _contentType = contentType;
        return this;
    }

    public override ContentCreateModel Build()
    {
        var key = _key ?? Guid.NewGuid();
        var parentKey = _parentKey;
        var templateKey = _templateKey;
        var invariantName = _invariantName ?? Guid.NewGuid().ToString();
        var invariantProperties = _invariantProperties;
        var variants = _variants;

        if (_contentTypeBuilder is null && _contentType is null)
        {
            throw new InvalidOperationException(
                "A content item cannot be constructed without providing a content type. Use AddContentType() or WithContentType().");
        }

        var contentType = _contentType ?? _contentTypeBuilder.Build();
        var content = new ContentCreateModel();

        content.InvariantName = invariantName;
        if (parentKey is not null)
        {
            content.ParentKey = parentKey;
        }

        if (templateKey is not null)
        {
            content.TemplateKey = templateKey;
        }

        content.ContentTypeKey = contentType.Key;
        content.Key = key;
        content.InvariantProperties = invariantProperties;
        content.Variants = variants;

        return content;
    }

    public static ContentCreateModel CreateBasicContent(IContentType contentType, Guid? key) =>
        new ContentEditingBuilder()
            .WithKey(key)
            .WithContentType(contentType)
            .WithInvariantName("Home")
            .Build();

    public static ContentCreateModel CreateSimpleContent(IContentType contentType) =>
        new ContentEditingBuilder()
            .WithContentType(contentType)
            .WithInvariantName("Home")
            .WithInvariantProperty("title", "Welcome to our Home page")
            .Build();

    public static ContentCreateModel CreateSimpleContent(IContentType contentType, string name, Guid? parentKey) =>
        new ContentEditingBuilder()
            .WithContentType(contentType)
            .WithInvariantName(name)
            .WithParentKey(parentKey)
            .WithInvariantProperty("title", "Welcome to our Home page")
            .Build();
}