using CliFx;
using CliFx.Attributes;
using CliFx.Exceptions;
using CliFx.Infrastructure;
using Microsoft.Extensions.DependencyInjection;
using System.Threading.Tasks;

namespace App;

public static class Program
{
    public static async Task<int> Main(params string[] args)
    {
        ServiceCollection services = [];

        services.AddSingleton<NotifyCommand>();

        services.AddKeyedSingleton<INotificationService, SmsNotificationService>("sms");
        services.AddKeyedSingleton<INotificationService, EmailNotificationService>("email");

        ServiceProvider serviceProvider = services.BuildServiceProvider();

        int exitCode = await new CliApplicationBuilder()
            .AddCommandsFromThisAssembly()
            .UseTypeActivator(serviceProvider)
            .Build()
            .RunAsync(args)
            .ConfigureAwait(false);

        return exitCode;
    }
}

[Command(Description = "Notify user")]
public class NotifyCommand(IKeyedServiceProvider keyedServiceProvider) : ICommand
{
    [CommandParameter(0, Description = "Notification Service")]
    public required string NotificationService { get; init; }

    [CommandOption("message", 'm', Description = "Notification Message")]
    public required string Message { get; init; }

    public ValueTask ExecuteAsync(IConsole console)
    {
        var service = keyedServiceProvider.GetKeyedService<INotificationService>(NotificationService)
            ?? throw new CommandException($"Invalid Notification Service '{NotificationService}'");

        service.Notify(Message);

        return default;
    }
}

public interface INotificationService
{
    string Notify(string message);
}

public class SmsNotificationService : INotificationService
{
    public string Notify(string message) => $"[SMS] {message}";
}

public class EmailNotificationService : INotificationService
{
    public string Notify(string message) => $"[Email] {message}";
}