Artemis/src/Artemis.UI.Shared/Services/ProfileEditor/Commands/ChangeElementDisplayCondition.cs

using System;
using Artemis.Core;

namespace Artemis.UI.Shared.Services.ProfileEditor.Commands;

/// <summary>
///     Represents a profile editor command that can be used to change the display condition of a profile element.
/// </summary>
public class ChangeElementDisplayCondition : IProfileEditorCommand, IDisposable
{
    private readonly ICondition _condition;
    private readonly ICondition _oldCondition;
    private readonly RenderProfileElement _profileElement;
    private bool _executed;

    /// <summary>
    ///     Creates a new instance of the <see cref="ChangeElementDisplayCondition" /> class.
    /// </summary>
    /// <param name="profileElement">The render profile element whose display condition to change.</param>
    /// <param name="condition">The new display condition.</param>
    public ChangeElementDisplayCondition(RenderProfileElement profileElement, ICondition condition)
    {
        _profileElement = profileElement;
        _condition = condition;
        _oldCondition = profileElement.DisplayCondition;
    }

    /// <inheritdoc />
    public void Dispose()
    {
        if (_executed)
            _oldCondition?.Dispose();
        else
            _condition?.Dispose();
    }

    /// <inheritdoc />
    public string DisplayName => "Change display condition mode";

    /// <inheritdoc />
    public void Execute()
    {
        _profileElement.DisplayCondition = _condition;
        _executed = true;
    }

    /// <inheritdoc />
    public void Undo()
    {
        _profileElement.DisplayCondition = _oldCondition;
        _executed = false;
    }
}