AddGpuUseCaseCommand.cs 1.0 KB

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