Artemis/src/Artemis.UI/DryIoc/Factories/IVMFactory.cs

using System;
using System.Collections.ObjectModel;
using System.Reactive;
using Artemis.Core;
using Artemis.Core.LayerBrushes;
using Artemis.Core.LayerEffects;
using Artemis.Core.ScriptingProviders;
using Artemis.UI.Screens.Device;
using Artemis.UI.Screens.Plugins;
using Artemis.UI.Screens.ProfileEditor;
using Artemis.UI.Screens.ProfileEditor.DisplayCondition.ConditionTypes;
using Artemis.UI.Screens.ProfileEditor.ProfileTree;
using Artemis.UI.Screens.ProfileEditor.ProfileTree.Dialogs.AdaptionHints;
using Artemis.UI.Screens.ProfileEditor.Properties;
using Artemis.UI.Screens.ProfileEditor.Properties.DataBinding;
using Artemis.UI.Screens.ProfileEditor.Properties.Timeline;
using Artemis.UI.Screens.ProfileEditor.Properties.Tree;
using Artemis.UI.Screens.ProfileEditor.VisualEditor.Visualizers;
using Artemis.UI.Screens.Scripting;
using Artemis.UI.Screens.Settings;
using Artemis.UI.Screens.Settings.Updating;
using Artemis.UI.Screens.Sidebar;
using Artemis.UI.Screens.SurfaceEditor;
using Artemis.UI.Screens.VisualScripting;
using Artemis.UI.Screens.VisualScripting.Pins;
using DryIoc;
using ReactiveUI;

namespace Artemis.UI.DryIoc.Factories;

public interface IVmFactory
{
}

public interface IDeviceVmFactory : IVmFactory
{
    DevicePropertiesViewModel DevicePropertiesViewModel(ArtemisDevice device);
    DeviceSettingsViewModel DeviceSettingsViewModel(ArtemisDevice device, DevicesTabViewModel devicesTabViewModel);
    DeviceDetectInputViewModel DeviceDetectInputViewModel(ArtemisDevice device);
    DevicePropertiesTabViewModel DevicePropertiesTabViewModel(ArtemisDevice device);
    DeviceInfoTabViewModel DeviceInfoTabViewModel(ArtemisDevice device);
    DeviceLedsTabViewModel DeviceLedsTabViewModel(ArtemisDevice device, ObservableCollection<ArtemisLed> selectedLeds);
    InputMappingsTabViewModel InputMappingsTabViewModel(ArtemisDevice device, ObservableCollection<ArtemisLed> selectedLeds);
}
public class DeviceFactory : IDeviceVmFactory
{
    private readonly IContainer _container;
    
    public DeviceFactory(IContainer container)
    {
        _container = container;
    }
    
    public DevicePropertiesViewModel DevicePropertiesViewModel(ArtemisDevice device)
    {
        return _container.Resolve<DevicePropertiesViewModel>(new object[] { device });
    }
    
    public DeviceSettingsViewModel DeviceSettingsViewModel(ArtemisDevice device, DevicesTabViewModel devicesTabViewModel)
    {
        return _container.Resolve<DeviceSettingsViewModel>(new object[] { device, devicesTabViewModel });
    }
    
    public DeviceDetectInputViewModel DeviceDetectInputViewModel(ArtemisDevice device)
    {
        return _container.Resolve<DeviceDetectInputViewModel>(new object[] { device });
    }
    
    public DevicePropertiesTabViewModel DevicePropertiesTabViewModel(ArtemisDevice device)
    {
        return _container.Resolve<DevicePropertiesTabViewModel>(new object[] { device });
    }
    
    public DeviceInfoTabViewModel DeviceInfoTabViewModel(ArtemisDevice device)
    {
        return _container.Resolve<DeviceInfoTabViewModel>(new object[] { device });
    }
    
    public DeviceLedsTabViewModel DeviceLedsTabViewModel(ArtemisDevice device, ObservableCollection<ArtemisLed> selectedLeds)
    {
        return _container.Resolve<DeviceLedsTabViewModel>(new object[] { device, selectedLeds });
    }
    
    public InputMappingsTabViewModel InputMappingsTabViewModel(ArtemisDevice device, ObservableCollection<ArtemisLed> selectedLeds)
    {
        return _container.Resolve<InputMappingsTabViewModel>(new object[] { device, selectedLeds });
    }
}

public interface ISettingsVmFactory : IVmFactory
{
    PluginSettingsViewModel PluginSettingsViewModel(Plugin plugin);
    PluginViewModel PluginViewModel(Plugin plugin, ReactiveCommand<Unit, Unit>? reload);
    PluginFeatureViewModel PluginFeatureViewModel(PluginFeatureInfo pluginFeatureInfo, bool showShield);
}
public class SettingsVmFactory : ISettingsVmFactory
{
    private readonly IContainer _container;
    
    public SettingsVmFactory(IContainer container)
    {
        _container = container;
    }
    
    public PluginSettingsViewModel PluginSettingsViewModel(Plugin plugin)
    {
        return _container.Resolve<PluginSettingsViewModel>(new object[] { plugin });
    }
    
    public PluginViewModel PluginViewModel(Plugin plugin, ReactiveCommand<Unit, Unit>? reload)
    {
        return _container.Resolve<PluginViewModel>(new object?[] { plugin, reload });
    }
    
    public PluginFeatureViewModel PluginFeatureViewModel(PluginFeatureInfo pluginFeatureInfo, bool showShield)
    {
        return _container.Resolve<PluginFeatureViewModel>(new object[] { pluginFeatureInfo, showShield });
    }
}

public interface ISidebarVmFactory : IVmFactory
{
    SidebarViewModel? SidebarViewModel(IScreen hostScreen);
    SidebarCategoryViewModel SidebarCategoryViewModel(ProfileCategory profileCategory);
    SidebarProfileConfigurationViewModel SidebarProfileConfigurationViewModel(ProfileConfiguration profileConfiguration);
}
public class SidebarVmFactory : ISidebarVmFactory
{
    private readonly IContainer _container;
    
    public SidebarVmFactory(IContainer container)
    {
        _container = container;
    }
    
    public SidebarViewModel? SidebarViewModel(IScreen hostScreen)
    {
        return _container.Resolve<SidebarViewModel>(new object[] { hostScreen });
    }
    
    public SidebarCategoryViewModel SidebarCategoryViewModel(ProfileCategory profileCategory)
    {
        return _container.Resolve<SidebarCategoryViewModel>(new object[] { profileCategory });
    }
    
    public SidebarProfileConfigurationViewModel SidebarProfileConfigurationViewModel(ProfileConfiguration profileConfiguration)
    {
        return _container.Resolve<SidebarProfileConfigurationViewModel>(new object[] { profileConfiguration });
    }
}

public interface ISurfaceVmFactory : IVmFactory
{
    SurfaceDeviceViewModel SurfaceDeviceViewModel(ArtemisDevice device, SurfaceEditorViewModel surfaceEditorViewModel);
    ListDeviceViewModel ListDeviceViewModel(ArtemisDevice device, SurfaceEditorViewModel surfaceEditorViewModel);
}
public class SurfaceVmFactory : ISurfaceVmFactory
{
    private readonly IContainer _container;

    public SurfaceVmFactory(IContainer container)
    {
        _container = container;
    }

    public SurfaceDeviceViewModel SurfaceDeviceViewModel(ArtemisDevice device, SurfaceEditorViewModel surfaceEditorViewModel)
    {
        return _container.Resolve<SurfaceDeviceViewModel>(new object[] { device, surfaceEditorViewModel });
    }

    public ListDeviceViewModel ListDeviceViewModel(ArtemisDevice device, SurfaceEditorViewModel surfaceEditorViewModel)
    {
        return _container.Resolve<ListDeviceViewModel>(new object[] { device, surfaceEditorViewModel });
    }
}

public interface IPrerequisitesVmFactory : IVmFactory
{
    PluginPrerequisiteViewModel PluginPrerequisiteViewModel(PluginPrerequisite pluginPrerequisite, bool uninstall);
}
public class PrerequisitesVmFactory : IPrerequisitesVmFactory
{
    private readonly IContainer _container;

    public PrerequisitesVmFactory(IContainer container)
    {
        _container = container;
    }

    public PluginPrerequisiteViewModel PluginPrerequisiteViewModel(PluginPrerequisite pluginPrerequisite, bool uninstall)
    {
        return _container.Resolve<PluginPrerequisiteViewModel>(new object[] { pluginPrerequisite, uninstall });
    }
}

public interface IProfileEditorVmFactory : IVmFactory
{
    ProfileEditorViewModel ProfileEditorViewModel(IScreen hostScreen);
    FolderTreeItemViewModel FolderTreeItemViewModel(TreeItemViewModel? parent, Folder folder);
    LayerTreeItemViewModel LayerTreeItemViewModel(TreeItemViewModel? parent, Layer layer);
    LayerShapeVisualizerViewModel LayerShapeVisualizerViewModel(Layer layer);
    LayerVisualizerViewModel LayerVisualizerViewModel(Layer layer);
}
public class ProfileEditorVmFactory : IProfileEditorVmFactory
{
    private readonly IContainer _container;

    public ProfileEditorVmFactory(IContainer container)
    {
        _container = container;
    }

    public FolderTreeItemViewModel FolderTreeItemViewModel(TreeItemViewModel? parent, Folder folder)
    {
        return _container.Resolve<FolderTreeItemViewModel>(new object?[] { parent, folder });
    }

    public LayerShapeVisualizerViewModel LayerShapeVisualizerViewModel(Layer layer)
    {
        return _container.Resolve<LayerShapeVisualizerViewModel>(new object[] { layer });
    }

    public LayerTreeItemViewModel LayerTreeItemViewModel(TreeItemViewModel? parent, Layer layer)
    {
        return _container.Resolve<LayerTreeItemViewModel>(new object?[] { parent, layer });
    }

    public LayerVisualizerViewModel LayerVisualizerViewModel(Layer layer)
    {
        return _container.Resolve<LayerVisualizerViewModel>(new object[] { layer });
    }

    public ProfileEditorViewModel ProfileEditorViewModel(IScreen hostScreen)
    {
        return _container.Resolve<ProfileEditorViewModel>(new object[] { hostScreen });
    }
}

public interface ILayerPropertyVmFactory : IVmFactory
{
    PropertyViewModel PropertyViewModel(ILayerProperty layerProperty);
    PropertyGroupViewModel PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup);
    PropertyGroupViewModel PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, BaseLayerBrush layerBrush);
    PropertyGroupViewModel PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, BaseLayerEffect layerEffect);

    TreeGroupViewModel TreeGroupViewModel(PropertyGroupViewModel propertyGroupViewModel);

    TimelineViewModel TimelineViewModel(ObservableCollection<PropertyGroupViewModel> propertyGroupViewModels);
    TimelineGroupViewModel TimelineGroupViewModel(PropertyGroupViewModel propertyGroupViewModel);
}
public class LayerPropertyVmFactory : ILayerPropertyVmFactory
{
    private readonly IContainer _container;

    public LayerPropertyVmFactory(IContainer container)
    {
        _container = container;
    }

    public PropertyViewModel PropertyViewModel(ILayerProperty layerProperty)
    {
        return _container.Resolve<PropertyViewModel>(new object[] { layerProperty });
    }

    public PropertyGroupViewModel PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup)
    {
        return _container.Resolve<PropertyGroupViewModel>(new object[] { layerPropertyGroup });
    }

    public PropertyGroupViewModel PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, BaseLayerBrush layerBrush)
    {
        return _container.Resolve<PropertyGroupViewModel>(new object[] { layerPropertyGroup, layerBrush });
    }

    public PropertyGroupViewModel PropertyGroupViewModel(LayerPropertyGroup layerPropertyGroup, BaseLayerEffect layerEffect)
    {
        return _container.Resolve<PropertyGroupViewModel>(new object[] { layerPropertyGroup, layerEffect });
    }

    public TreeGroupViewModel TreeGroupViewModel(PropertyGroupViewModel propertyGroupViewModel)
    {
        return _container.Resolve<TreeGroupViewModel>(new object[] { propertyGroupViewModel });
    }

    public TimelineViewModel TimelineViewModel(ObservableCollection<PropertyGroupViewModel> propertyGroupViewModels)
    {
        return _container.Resolve<TimelineViewModel>(new object[] { propertyGroupViewModels });
    }

    public TimelineGroupViewModel TimelineGroupViewModel(PropertyGroupViewModel propertyGroupViewModel)
    {
        return _container.Resolve<TimelineGroupViewModel>(new object[] { propertyGroupViewModel });
    }
}

public interface IDataBindingVmFactory : IVmFactory
{
    DataBindingViewModel DataBindingViewModel();
}
public class DataBindingVmFactory : IDataBindingVmFactory
{
    private readonly IContainer _container;

    public DataBindingVmFactory(IContainer container)
    {
        _container = container;
    }

    public DataBindingViewModel DataBindingViewModel()
    {
        return _container.Resolve<DataBindingViewModel>();
    }
}

public interface IPropertyVmFactory
{
    ITreePropertyViewModel TreePropertyViewModel(ILayerProperty layerProperty, PropertyViewModel propertyViewModel);
    ITimelinePropertyViewModel TimelinePropertyViewModel(ILayerProperty layerProperty, PropertyViewModel propertyViewModel);
}

public interface INodeVmFactory : IVmFactory
{
    NodeScriptViewModel NodeScriptViewModel(NodeScript nodeScript, bool isPreview);
    NodePickerViewModel NodePickerViewModel(NodeScript nodeScript);
    NodeViewModel NodeViewModel(NodeScriptViewModel nodeScriptViewModel, INode node);
    CableViewModel CableViewModel(NodeScriptViewModel nodeScriptViewModel, IPin from, IPin to);
    DragCableViewModel DragCableViewModel(PinViewModel pinViewModel);
    InputPinViewModel InputPinViewModel(IPin inputPin, NodeScriptViewModel nodeScriptViewModel);
    OutputPinViewModel OutputPinViewModel(IPin outputPin, NodeScriptViewModel nodeScriptViewModel);
    InputPinCollectionViewModel InputPinCollectionViewModel(IPinCollection inputPinCollection, NodeScriptViewModel nodeScriptViewModel);
    OutputPinCollectionViewModel OutputPinCollectionViewModel(IPinCollection outputPinCollection, NodeScriptViewModel nodeScriptViewModel);
}
public class NodeVmFactory : INodeVmFactory
{
    private readonly IContainer _container;

    public NodeVmFactory(IContainer container)
    {
        _container = container;
    }

    public NodeScriptViewModel NodeScriptViewModel(NodeScript nodeScript, bool isPreview)
    {
        return _container.Resolve<NodeScriptViewModel>(new object[] { nodeScript, isPreview });
    }

    public NodePickerViewModel NodePickerViewModel(NodeScript nodeScript)
    {
        return _container.Resolve<NodePickerViewModel>(new object[] { nodeScript });
    }

    public NodeViewModel NodeViewModel(NodeScriptViewModel nodeScriptViewModel, INode node)
    {
        return _container.Resolve<NodeViewModel>(new object[] { nodeScriptViewModel, node });
    }

    public CableViewModel CableViewModel(NodeScriptViewModel nodeScriptViewModel, IPin from, IPin to)
    {
        return _container.Resolve<CableViewModel>(new object[] { nodeScriptViewModel, from, to });
    }

    public DragCableViewModel DragCableViewModel(PinViewModel pinViewModel)
    {
        return _container.Resolve<DragCableViewModel>(new object[] { pinViewModel });
    }

    public InputPinViewModel InputPinViewModel(IPin inputPin, NodeScriptViewModel nodeScriptViewModel)
    {
        return _container.Resolve<InputPinViewModel>(new object[] { inputPin, nodeScriptViewModel });
    }

    public OutputPinViewModel OutputPinViewModel(IPin outputPin, NodeScriptViewModel nodeScriptViewModel)
    {
        return _container.Resolve<OutputPinViewModel>(new object[] { outputPin, nodeScriptViewModel });
    }

    public InputPinCollectionViewModel InputPinCollectionViewModel(IPinCollection inputPinCollection, NodeScriptViewModel nodeScriptViewModel)
    {
        return _container.Resolve<InputPinCollectionViewModel>(new object[] { inputPinCollection, nodeScriptViewModel });
    }

    public OutputPinCollectionViewModel OutputPinCollectionViewModel(IPinCollection outputPinCollection, NodeScriptViewModel nodeScriptViewModel)
    {
        return _container.Resolve<OutputPinCollectionViewModel>(new object[] { outputPinCollection, nodeScriptViewModel });
    }
}

public interface IConditionVmFactory : IVmFactory
{
    AlwaysOnConditionViewModel AlwaysOnConditionViewModel(AlwaysOnCondition alwaysOnCondition);
    PlayOnceConditionViewModel PlayOnceConditionViewModel(PlayOnceCondition playOnceCondition);
    StaticConditionViewModel StaticConditionViewModel(StaticCondition staticCondition);
    EventConditionViewModel EventConditionViewModel(EventCondition eventCondition);
}
public class ConditionVmFactory : IConditionVmFactory
{
    private readonly IContainer _container;

    public ConditionVmFactory(IContainer container)
    {
        _container = container;
    }

    public AlwaysOnConditionViewModel AlwaysOnConditionViewModel(AlwaysOnCondition alwaysOnCondition)
    {
        return _container.Resolve<AlwaysOnConditionViewModel>(new object[] { alwaysOnCondition });
    }

    public PlayOnceConditionViewModel PlayOnceConditionViewModel(PlayOnceCondition playOnceCondition)
    {
        return _container.Resolve<PlayOnceConditionViewModel>(new object[] { playOnceCondition });
    }

    public StaticConditionViewModel StaticConditionViewModel(StaticCondition staticCondition)
    {
        return _container.Resolve<StaticConditionViewModel>(new object[] { staticCondition });
    }

    public EventConditionViewModel EventConditionViewModel(EventCondition eventCondition)
    {
        return _container.Resolve<EventConditionViewModel>(new object[] { eventCondition });
    }
}

public interface ILayerHintVmFactory : IVmFactory
{
    CategoryAdaptionHintViewModel CategoryAdaptionHintViewModel(Layer layer, CategoryAdaptionHint adaptionHint);
    DeviceAdaptionHintViewModel DeviceAdaptionHintViewModel(Layer layer, DeviceAdaptionHint adaptionHint);
    KeyboardSectionAdaptionHintViewModel KeyboardSectionAdaptionHintViewModel(Layer layer, KeyboardSectionAdaptionHint adaptionHint);
}
public class LayerHintVmFactory : ILayerHintVmFactory
{
    private readonly IContainer _container;

    public LayerHintVmFactory(IContainer container)
    {
        _container = container;
    }

    public CategoryAdaptionHintViewModel CategoryAdaptionHintViewModel(Layer layer, CategoryAdaptionHint adaptionHint)
    {
        return _container.Resolve<CategoryAdaptionHintViewModel>(new object[] { layer, adaptionHint });
    }

    public DeviceAdaptionHintViewModel DeviceAdaptionHintViewModel(Layer layer, DeviceAdaptionHint adaptionHint)
    {
        return _container.Resolve<DeviceAdaptionHintViewModel>(new object[] { layer, adaptionHint });
    }

    public KeyboardSectionAdaptionHintViewModel KeyboardSectionAdaptionHintViewModel(Layer layer, KeyboardSectionAdaptionHint adaptionHint)
    {
        return _container.Resolve<KeyboardSectionAdaptionHintViewModel>(new object[] { layer, adaptionHint });
    }
}

public interface IScriptVmFactory : IVmFactory
{
    ScriptConfigurationViewModel ScriptConfigurationViewModel(ScriptConfiguration scriptConfiguration);
    ScriptConfigurationViewModel ScriptConfigurationViewModel(Profile profile, ScriptConfiguration scriptConfiguration);
}
public class ScriptVmFactory : IScriptVmFactory
{
    private readonly IContainer _container;

    public ScriptVmFactory(IContainer container)
    {
        _container = container;
    }

    public ScriptConfigurationViewModel ScriptConfigurationViewModel(ScriptConfiguration scriptConfiguration)
    {
        return _container.Resolve<ScriptConfigurationViewModel>(new object[] { scriptConfiguration });
    }

    public ScriptConfigurationViewModel ScriptConfigurationViewModel(Profile profile, ScriptConfiguration scriptConfiguration)
    {
        return _container.Resolve<ScriptConfigurationViewModel>(new object[] { profile, scriptConfiguration });
    }
}

public interface IReleaseVmFactory : IVmFactory
{
    ReleaseViewModel ReleaseListViewModel(string releaseId, string version, DateTimeOffset createdAt);
}
public class ReleaseVmFactory : IReleaseVmFactory
{
    private readonly IContainer _container;

    public ReleaseVmFactory(IContainer container)
    {
        _container = container;
    }
    
    public ReleaseViewModel ReleaseListViewModel(string releaseId, string version, DateTimeOffset createdAt)
    {
        return _container.Resolve<ReleaseViewModel>(new object[] { releaseId, version, createdAt });
    }
}