StabilityMatrix/StabilityMatrix.Avalonia/Models/InferenceProjectDocument.cs

using System;
using System.Text.Json;
using System.Text.Json.Nodes;
using System.Text.Json.Serialization;
using StabilityMatrix.Avalonia.ViewModels.Inference;

namespace StabilityMatrix.Avalonia.Models;

/// <summary>
/// This is the project file for inference tabs
/// </summary>
[JsonSerializable(typeof(InferenceProjectDocument))]
public class InferenceProjectDocument
{
    [JsonIgnore]
    private static readonly JsonSerializerOptions SerializerOptions =
        new() { IgnoreReadOnlyProperties = true, WriteIndented = true, };

    public int Version { get; set; } = 2;

    [JsonConverter(typeof(JsonStringEnumConverter))]
    public InferenceProjectType ProjectType { get; set; }

    public JsonObject? State { get; set; }

    public static InferenceProjectDocument FromLoadable(IJsonLoadableState loadableModel)
    {
        return new InferenceProjectDocument
        {
            ProjectType = loadableModel switch
            {
                InferenceTextToImageViewModel => InferenceProjectType.TextToImage,
                _
                    => throw new InvalidOperationException(
                        $"Unknown loadable model type: {loadableModel.GetType()}"
                    )
            },
            State = loadableModel.SaveStateToJsonObject()
        };
    }

    public Type GetViewModelType()
    {
        return ProjectType switch
        {
            InferenceProjectType.TextToImage => typeof(InferenceTextToImageViewModel),
            InferenceProjectType.Unknown => throw new InvalidOperationException(),
            _ => throw new ArgumentOutOfRangeException(nameof(ProjectType), ProjectType, null)
        };
    }
}