StabilityMatrix/StabilityMatrix.Core/Models/Api/CivitModel.cs

using System.Text.Json.Serialization;
using LiteDB;

namespace StabilityMatrix.Core.Models.Api;

public class CivitModel
{
    [JsonPropertyName("id")]
    public int Id { get; set; }

    [JsonPropertyName("name")]
    public string Name { get; set; }

    [JsonPropertyName("description")]
    public string? Description { get; set; }

    [JsonPropertyName("type")]
    public CivitModelType Type { get; set; }

    [JsonPropertyName("nsfw")]
    public bool Nsfw { get; set; }

    [JsonPropertyName("tags")]
    public string[] Tags { get; set; }

    [JsonPropertyName("mode")]
    public CivitMode? Mode { get; set; }

    [JsonPropertyName("creator")]
    public CivitCreator Creator { get; set; }

    [JsonPropertyName("stats")]
    public CivitModelStats Stats { get; set; }

    [BsonRef("ModelVersions")]
    [JsonPropertyName("modelVersions")]
    public List<CivitModelVersion>? ModelVersions { get; set; }

    private FileSizeType? fullFilesSize;
    public FileSizeType FullFilesSize
    {
        get
        {
            if (fullFilesSize != null)
                return fullFilesSize;
            var kbs = 0.0;

            var latestVersion = ModelVersions?.FirstOrDefault();
            if (latestVersion?.Files != null && latestVersion.Files.Any())
            {
                var latestModelFile = latestVersion.Files.FirstOrDefault(
                    x => x.Type == CivitFileType.Model
                );
                kbs = latestModelFile?.SizeKb ?? 0;
            }
            fullFilesSize = new FileSizeType(kbs);
            return fullFilesSize;
        }
    }

    public string LatestModelVersionName =>
        ModelVersions != null && ModelVersions.Any() ? ModelVersions[0].Name : string.Empty;

    public string? BaseModelType =>
        ModelVersions != null && ModelVersions.Any()
            ? ModelVersions[0].BaseModel?.Replace("SD", "").Trim()
            : string.Empty;
}