StabilityMatrix/StabilityMatrix.Core/Models/Api/Comfy/ComfyWebSocketResponse.cs

using System.Text.Json;
using System.Text.Json.Nodes;
using System.Text.Json.Serialization;
using StabilityMatrix.Core.Models.Api.Comfy.WebSocketData;

namespace StabilityMatrix.Core.Models.Api.Comfy;

public class ComfyWebSocketResponse
{
    [JsonPropertyName("type")]
    public required ComfyWebSocketResponseType Type { get; set; }
    
    /// <summary>
    /// Depending on the value of <see cref="Type"/>,
    /// this property will be one of these types
    /// <list type="bullet">
    /// <item>Status - <see cref="ComfyWebSocketStatusData"/></item>
    /// <item>Progress - <see cref="ComfyWebSocketProgressData"/></item>
    /// <item>Executing - <see cref="ComfyWebSocketExecutingData"/></item>
    /// </list>
    /// </summary>
    [JsonPropertyName("data")]
    public required JsonObject Data { get; set; }
    
    public T? GetDataAsType<T>() where T : class
    {
        return Data.Deserialize<T>();
    }
}