using System.ComponentModel;
using System.Text.Json;
using Sbroenne.ExcelMcp.CLI.Daemon;
using Sbroenne.ExcelMcp.CLI.Infrastructure;
using Sbroenne.ExcelMcp.Core.Models.Actions;
using Spectre.Console;
using Spectre.Console.Cli;
namespace Sbroenne.ExcelMcp.CLI.Commands;
/// <summary>
/// Connection commands - thin wrapper that sends requests to daemon.
/// Actions: list, view, create, test, refresh, delete, load-to, get-properties, set-properties
/// </summary>
internal sealed class ConnectionCommand : AsyncCommand<ConnectionCommand.Settings>
{
public override async Task<int> ExecuteAsync(CommandContext context, Settings settings, CancellationToken cancellationToken)
{
if (string.IsNullOrWhiteSpace(settings.SessionId))
{
AnsiConsole.MarkupLine("[red]Session ID is required. Use --session <id>[/]");
return 1;
}
if (string.IsNullOrWhiteSpace(settings.Action))
{
AnsiConsole.MarkupLine("[red]Action is required.[/]");
return 1;
}
if (!ActionValidator.TryNormalizeAction<ConnectionAction>(settings.Action, out var action, out var errorMessage))
{
AnsiConsole.MarkupLine($"[red]{errorMessage}[/]");
return 1;
}
var command = $"connection.{action}";
var connectionString = ResolveFileOrValue(settings.ConnectionString, settings.ConnectionStringFile);
var commandText = ResolveFileOrValue(settings.CommandText, settings.CommandTextFile);
object? args = action switch
{
"list" => null,
"view" => new { connectionName = settings.ConnectionName },
"create" => new { connectionName = settings.ConnectionName, connectionType = settings.ConnectionType, connectionString, commandText },
"test" => new { connectionName = settings.ConnectionName },
"refresh" => new { connectionName = settings.ConnectionName },
"delete" => new { connectionName = settings.ConnectionName },
"load-to" => new { connectionName = settings.ConnectionName, loadDestination = settings.LoadDestination, sheetName = settings.SheetName, targetCell = settings.TargetCell },
"get-properties" => new { connectionName = settings.ConnectionName },
"set-properties" => new { connectionName = settings.ConnectionName, connectionString, commandText, description = settings.Description, backgroundQuery = settings.BackgroundQuery, refreshOnFileOpen = settings.RefreshOnOpen, savePassword = settings.SavePassword, refreshPeriod = settings.RefreshPeriod },
_ => new { connectionName = settings.ConnectionName }
};
using var client = new DaemonClient();
var response = await client.SendAsync(new DaemonRequest
{
Command = command,
SessionId = settings.SessionId,
Args = args != null ? JsonSerializer.Serialize(args, DaemonProtocol.JsonOptions) : null
}, cancellationToken);
if (response.Success)
{
Console.WriteLine(!string.IsNullOrEmpty(response.Result) ? response.Result : JsonSerializer.Serialize(new { success = true }, DaemonProtocol.JsonOptions));
return 0;
}
else
{
Console.WriteLine(JsonSerializer.Serialize(new { success = false, error = response.ErrorMessage }, DaemonProtocol.JsonOptions));
return 1;
}
}
/// <summary>
/// Returns file contents if filePath is provided, otherwise returns the direct value.
/// </summary>
private static string? ResolveFileOrValue(string? directValue, string? filePath)
{
if (!string.IsNullOrWhiteSpace(filePath))
{
if (!File.Exists(filePath))
{
throw new FileNotFoundException($"File not found: {filePath}");
}
return File.ReadAllText(filePath);
}
return directValue;
}
internal sealed class Settings : CommandSettings
{
[CommandArgument(0, "<ACTION>")]
[Description("The action to perform (e.g., list, view, create, refresh)")]
public string Action { get; init; } = string.Empty;
[CommandOption("-s|--session <SESSION>")]
[Description("Session ID from 'session open' command")]
public string SessionId { get; init; } = string.Empty;
[CommandOption("--connection <NAME>")]
[Description("Connection name")]
public string? ConnectionName { get; init; }
[CommandOption("--connection-type <TYPE>")]
[Description("Connection type (e.g., ODBC, OleDb)")]
public string? ConnectionType { get; init; }
[CommandOption("--connection-string <STRING>")]
[Description("Database connection string")]
public string? ConnectionString { get; init; }
[CommandOption("--connection-string-file <PATH>")]
[Description("Read connection string from file instead of command line")]
public string? ConnectionStringFile { get; init; }
[CommandOption("--command-text <TEXT>")]
[Description("SQL command or query text")]
public string? CommandText { get; init; }
[CommandOption("--command-text-file <PATH>")]
[Description("Read SQL command from file instead of command line")]
public string? CommandTextFile { get; init; }
[CommandOption("--load-destination <DEST>")]
[Description("Load destination: worksheet, data-model, both, connection-only")]
public string? LoadDestination { get; init; }
[CommandOption("--sheet <NAME>")]
[Description("Target worksheet name")]
public string? SheetName { get; init; }
[CommandOption("--target-cell <ADDRESS>")]
[Description("Target cell address for data load")]
public string? TargetCell { get; init; }
[CommandOption("--refresh-on-open")]
[Description("Refresh connection when workbook opens")]
public bool? RefreshOnOpen { get; init; }
[CommandOption("--enable-refresh")]
[Description("Enable manual refresh")]
public bool? EnableRefresh { get; init; }
[CommandOption("--description <TEXT>")]
[Description("Connection description")]
public string? Description { get; init; }
[CommandOption("--background-query")]
[Description("Run query in background")]
public bool? BackgroundQuery { get; init; }
[CommandOption("--save-password")]
[Description("Save password in connection string")]
public bool? SavePassword { get; init; }
[CommandOption("--refresh-period <MINUTES>")]
[Description("Auto-refresh period in minutes (0 to disable)")]
public int? RefreshPeriod { get; init; }
}
}
