using System.CommandLine;
using System.Threading.Tasks;
using Toak.Commands;

namespace Toak;

public class Program
{
    public static async Task<int> Main(string[] args)
    {
        var rootCommand = new RootCommand("Toak: High-speed Linux Dictation");

        var pipeOption = new Option<bool>(new[] { "--pipe", "-p" }, "Output transcription to stdout instead of typing");
        var copyOption = new Option<bool>("--copy", "Copy to clipboard instead of typing");
        var verboseOption = new Option<bool>(new[] { "--verbose", "-v" }, "Enable detailed debug logging");

        rootCommand.AddGlobalOption(verboseOption);

        // Toggle Command
        var toggleCmd = new Command("toggle", "Starts or stops the recording");
        toggleCmd.AddOption(pipeOption);
        toggleCmd.AddOption(copyOption);
        toggleCmd.SetHandler(ToggleCommand.ExecuteAsync, pipeOption, copyOption, verboseOption);
        rootCommand.AddCommand(toggleCmd);

        // Daemon Command
        var daemonCmd = new Command("daemon", "Starts the background background service");
        daemonCmd.SetHandler(Toak.Core.DaemonService.StartAsync, verboseOption);
        rootCommand.AddCommand(daemonCmd);

        // Discard Command
        var discardCmd = new Command("discard", "Abort current recording without transcribing");
        discardCmd.AddOption(pipeOption);
        discardCmd.SetHandler(DiscardCommand.ExecuteAsync, pipeOption, verboseOption);
        rootCommand.AddCommand(discardCmd);

        // Onboard Command
        var onboardCmd = new Command("onboard", "Configure the application");
        onboardCmd.SetHandler(OnboardCommand.ExecuteAsync, verboseOption);
        rootCommand.AddCommand(onboardCmd);

        // Latency Test Command
        var latencyCmd = new Command("latency-test", "Benchmark full pipeline without recording");
        latencyCmd.SetHandler(LatencyTestCommand.ExecuteAsync, verboseOption);
        rootCommand.AddCommand(latencyCmd);

        // Show Command
        var showCmd = new Command("show", "Show current configuration");
        showCmd.SetHandler(ShowCommand.ExecuteAsync, verboseOption);
        rootCommand.AddCommand(showCmd);

        // Config Command
        var configCmd = new Command("config", "Update a specific configuration setting");
        var keyArg = new Argument<string>("key", "Configuration key (e.g., llm, whisper, lang)");
        var valArg = new Argument<string>("value", "Configuration value");
        configCmd.AddArgument(keyArg);
        configCmd.AddArgument(valArg);
        configCmd.SetHandler(ConfigUpdaterCommand.ExecuteAsync, keyArg, valArg, verboseOption);
        rootCommand.AddCommand(configCmd);

        // Skill Command
        rootCommand.AddCommand(SkillCommand.CreateCommand(verboseOption));

        return await rootCommand.InvokeAsync(args);
    }
}