add VM completion for run command

cirruslabs · Apr 9, 2024 · d433871 · d433871
1 parent 13e7794
commit d433871
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/Sources/tart/Commands/Run.swift b/Sources/tart/Commands/Run.swift
@@ -11,10 +11,18 @@ var vm: VM?
 struct IPNotFound: Error {
 }
 
+func completeVM(_ args: [String]) -> [String] {
+  if let vms = try? VMStorageLocal().list() {
+    return vms.map { (name, _) in return name }
+  }
+  return []
+}
+
+
 struct Run: AsyncParsableCommand {
   static var configuration = CommandConfiguration(abstract: "Run a VM")
 
-  @Argument(help: "VM name")
+  @Argument(help: "VM name", completion: .custom(completeVM))
   var name: String
 
   @Flag(help: ArgumentHelp(