StabilityMatrix/StabilityMatrix.Core/Models/CivitPostDownloadContextAction.cs

using System.Diagnostics;
using System.Text.Json;
using StabilityMatrix.Core.Models.Api;
using StabilityMatrix.Core.Services;

namespace StabilityMatrix.Core.Models;

public class CivitPostDownloadContextAction : IContextAction
{
    /// <inheritdoc />
    public object? Context { get; set; }
    
    public static CivitPostDownloadContextAction FromCivitFile(CivitFile file)
    {
        return new CivitPostDownloadContextAction
        {
            Context = file.Hashes.BLAKE3
        };
    }
    
    public void Invoke(ISettingsManager settingsManager)
    {
        var result = Context as string;
        
        if (Context is JsonElement jsonElement)
        {
            result = jsonElement.GetString();
        }
        
        if (result is null)
        {
            Debug.WriteLine($"Context {Context} is not a string.");
            return;
        }
        
        Debug.WriteLine($"Adding {result} to installed models.");
        settingsManager.Transaction(
            s =>
            {
                s.InstalledModelHashes ??= new HashSet<string>();
                s.InstalledModelHashes.Add(result);
            });
    }
}