Artemis/src/Artemis.UI.Shared/DataModelVisualization/DataModelVisualizationRegistration.cs

using System;
using System.Collections.Generic;
using Artemis.Core;
using Artemis.UI.Shared.Services;

namespace Artemis.UI.Shared.DataModelVisualization
{
    /// <summary>
    ///     Represents a layer brush registered through
    ///     <see cref="IDataModelUIService.RegisterDataModelInput{T}" /> or
    ///     <see cref="IDataModelUIService.RegisterDataModelDisplay{T}" />
    /// </summary>
    public class DataModelVisualizationRegistration
    {
        private readonly IDataModelUIService _dataModelUIService;

        internal DataModelVisualizationRegistration(IDataModelUIService dataModelUIService,
            RegistrationType registrationType,
            Plugin plugin,
            Type supportedType,
            Type viewModelType)
        {
            _dataModelUIService = dataModelUIService;
            RegistrationType = registrationType;
            Plugin = plugin;
            SupportedType = supportedType;
            ViewModelType = viewModelType;

            if (Plugin != Constants.CorePlugin)
                Plugin.Disabled += InstanceOnDisabled;
        }

        /// <summary>
        ///     Gets the type of registration, either a display or an input
        /// </summary>
        public RegistrationType RegistrationType { get; }

        /// <summary>
        ///     Gets the plugin that registered the visualization
        /// </summary>
        public Plugin Plugin { get; }

        /// <summary>
        ///     Gets the type supported by the visualization
        /// </summary>
        public Type SupportedType { get; }

        /// <summary>
        ///     Gets the view model type of the visualization
        /// </summary>
        public Type ViewModelType { get; }

        /// <summary>
        ///     Gets a read only collection of types this visualization can convert to and from
        /// </summary>
        public IReadOnlyCollection<Type>? CompatibleConversionTypes { get; internal set; }

        internal void Unsubscribe()
        {
            if (Plugin != Constants.CorePlugin)
                Plugin.Disabled -= InstanceOnDisabled;
        }

        private void InstanceOnDisabled(object? sender, EventArgs e)
        {
            if (RegistrationType == RegistrationType.Input)
                _dataModelUIService.RemoveDataModelInput(this);
            else if (RegistrationType == RegistrationType.Display)
                _dataModelUIService.RemoveDataModelDisplay(this);
        }
    }

    /// <summary>
    ///     Represents a type of data model visualization registration
    /// </summary>
    public enum RegistrationType
    {
        /// <summary>
        ///     A visualization used for displaying values
        /// </summary>
        Display,

        /// <summary>
        ///     A visualization used for inputting values
        /// </summary>
        Input
    }
}