AI-Studio/app/MindWork AI Studio/Dialogs/ConfigurationSnippetImportDialog.razor.cs

using AIStudio.Components;
using AIStudio.Tools.PluginSystem;
using AIStudio.Tools.ToolCallingSystem;

using Microsoft.AspNetCore.Components;

namespace AIStudio.Dialogs;

public partial class ConfigurationSnippetImportDialog : MSGComponentBase
{
    [CascadingParameter]
    private IMudDialogInstance MudDialog { get; set; } = null!;

    [Parameter]
    public PluginConfigurationObjectType Type { get; set; } = PluginConfigurationObjectType.NONE;

    [Parameter]
    public string ImportLabel { get; set; } = string.Empty;

    [Inject]
    private ToolRegistry ToolRegistry { get; init; } = null!;

    private string snippet = string.Empty;
    private string issue = string.Empty;

    /// <summary>
    /// What the snippet imports, while the user still has to read its hints.
    /// </summary>
    private ConfigurationSnippetImportResult? importWithHints;

    private void SnippetChanged(string? text)
    {
        // The issue and the hints belong to the snippet they were found in:
        this.snippet = text ?? string.Empty;
        this.issue = string.Empty;
        this.importWithHints = null;
    }

    private async Task Import()
    {
        if (!ConfigurationSnippetImport.TryImport(this.snippet, this.Type, out var imported, out this.issue))
            return;

        imported = await ConfigurationSnippetImport.ResolveReferencesAsync(imported, this.SettingsManager, this.ToolRegistry);
        if (imported.Hints.Count > 0)
        {
            this.importWithHints = imported;
            return;
        }

        this.Close(imported);
    }

    private void ContinueAnyway()
    {
        if (this.importWithHints is not null)
            this.Close(this.importWithHints);
    }

    private void Close(ConfigurationSnippetImportResult imported)
    {
        //
        // An organization can lock the import at any time, e.g., with a new version of its
        // configuration plugin, and the user may take a while to read the hints. Hence, the
        // permission is checked when the dialog hands over the imported object:
        //
        if (!this.SettingsManager.ConfigurationData.App.CanImportConfigurationSnippet(this.Type))
        {
            this.issue = T("Your organization does not allow importing this type of configuration.");
            this.importWithHints = null;
            return;
        }

        this.MudDialog.Close(DialogResult.Ok(imported));
    }

    private void Cancel() => this.MudDialog.Cancel();

    /// <summary>
    /// Lets the user paste a configuration snippet and imports it as an object of the user's own.
    /// </summary>
    /// <param name="service">The dialog service.</param>
    /// <param name="type">The type of configuration object to import.</param>
    /// <param name="title">The title of the dialog, which also labels its import button.</param>
    /// <returns>The imported object, or null when the user canceled.</returns>
    public static async Task<ConfigurationSnippetImportResult?> ShowAsync(IDialogService service, PluginConfigurationObjectType type, string title)
    {
        var parameters = new DialogParameters<ConfigurationSnippetImportDialog>
        {
            { x => x.Type, type },
            { x => x.ImportLabel, title },
        };
        var dialog = await service.ShowAsync<ConfigurationSnippetImportDialog>(title, parameters, DialogOptions.FULLSCREEN);
        var result = await dialog.Result;
        return result is { Canceled: false, Data: ConfigurationSnippetImportResult imported } ? imported : null;
    }
}