Support stopping of suspended VMs (#541)

cirruslabs · Jul 7, 2023 · e8dbb86 · e8dbb86
1 parent d2ed4ef
commit e8dbb86
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/Sources/tart/Commands/Stop.swift b/Sources/tart/Commands/Stop.swift
@@ -14,6 +14,21 @@ struct Stop: AsyncParsableCommand {
 
   func run() async throws {
     let vmDir = try VMStorageLocal().open(name)
+    switch try vmDir.state() {
+    case "suspended":
+      try stopSuspended(vmDir)
+    case "running":
+      try await stopRunning(vmDir)
+    default:
+      return
+    }
+  }
+
+  func stopSuspended(_ vmDir: VMDirectory) throws {
+    try? FileManager.default.removeItem(at: vmDir.stateURL)
+  }
+
+  func stopRunning(_ vmDir: VMDirectory) async throws {
     let lock = try PIDLock(lockURL: vmDir.configURL)
 
     // Find the VM's PID