AddGpuUseCaseCommand.cs 1.1 KB

1234567891011121314151617181920212223242526272829303132333435
  1. using Microsoft.Extensions.DependencyInjection;
  2. using RackPeek.Domain.Resources.Servers;
  3. using RackPeek.Domain.UseCases.Gpus;
  4. using Spectre.Console;
  5. using Spectre.Console.Cli;
  6. namespace Shared.Rcl.Commands.Servers.Gpus;
  7. public class ServerGpuAddSettings : ServerNameSettings
  8. {
  9. [CommandOption("--model <MODEL>")] public string Model { get; set; }
  10. [CommandOption("--vram <VRAM>")] public int Vram { get; set; }
  11. }
  12. public class ServerGpuAddCommand(IServiceProvider serviceProvider)
  13. : AsyncCommand<ServerGpuAddSettings>
  14. {
  15. public override async Task<int> ExecuteAsync(
  16. CommandContext context,
  17. ServerGpuAddSettings settings,
  18. CancellationToken cancellationToken)
  19. {
  20. using var scope = serviceProvider.CreateScope();
  21. var useCase = scope.ServiceProvider.GetRequiredService<IAddGpuUseCase<Server>>();
  22. await useCase.ExecuteAsync(
  23. settings.Name,
  24. settings.Model,
  25. settings.Vram);
  26. AnsiConsole.MarkupLine($"[green]GPU added to '{settings.Name}'.[/]");
  27. return 0;
  28. }
  29. }