using System;
using System.Collections.Generic;
using StabilityMatrix.Core.Models.Api.Comfy.NodeTypes;
using StabilityMatrix.Core.Models.Api.Comfy.Nodes;

namespace StabilityMatrix.Avalonia.Models.Inference;

/// <summary>
/// Event args for applying a <see cref="IComfyStep"/>.
/// </summary>
public class ModuleApplyStepEventArgs : EventArgs
{
    public required ComfyNodeBuilder Builder { get; init; }

    public NodeDictionary Nodes => Builder.Nodes;

    public ModuleApplyStepTemporaryArgs Temp { get; } = new();

    /// <summary>
    /// Index of the step in the pipeline.
    /// </summary>
    public int StepIndex { get; init; }

    /// <summary>
    /// Index
    /// </summary>
    public int StepTypeIndex { get; init; }

    /// <summary>
    /// Generation overrides (like hires fix generate, current seed generate, etc.)
    /// </summary>
    public IReadOnlyDictionary<Type, bool> IsEnabledOverrides { get; init; } = new Dictionary<Type, bool>();

    public class ModuleApplyStepTemporaryArgs
    {
        /// <summary>
        /// Temporary conditioning apply step, used by samplers to apply control net.
        /// </summary>
        public ConditioningConnections? Conditioning { get; set; }

        /// <summary>
        /// Temporary refiner conditioning apply step, used by samplers to apply control net.
        /// </summary>
        public ConditioningConnections? RefinerConditioning { get; set; }

        /// <summary>
        /// Temporary model apply step, used by samplers to apply control net.
        /// </summary>
        public ModelNodeConnection? Model { get; set; }
    }
}