UpdateGpuUseCaseCommand.cs 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  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 ServerGpuUpdateSettings : ServerNameSettings
  7. {
  8. [CommandOption("--index <INDEX>")] public int Index { get; set; }
  9. [CommandOption("--model <MODEL>")] public string Model { get; set; }
  10. [CommandOption("--vram <VRAM>")] public int Vram { get; set; }
  11. }
  12. public class ServerGpuUpdateCommand(IServiceProvider serviceProvider)
  13. : AsyncCommand<ServerGpuUpdateSettings>
  14. {
  15. public override async Task<int> ExecuteAsync(
  16. CommandContext context,
  17. ServerGpuUpdateSettings settings,
  18. CancellationToken cancellationToken)
  19. {
  20. using var scope = serviceProvider.CreateScope();
  21. var useCase = scope.ServiceProvider.GetRequiredService<UpdateGpuUseCase>();
  22. await useCase.ExecuteAsync(
  23. settings.Name,
  24. settings.Index,
  25. settings.Model,
  26. settings.Vram);
  27. AnsiConsole.MarkupLine($"[green]GPU {settings.Index} updated on '{settings.Name}'.[/]");
  28. return 0;
  29. }
  30. }