clean up temporary directory on error

beringresearch · Jun 28, 2023 · 19368d7 · 19368d7
1 parent 26b9c19
commit 19368d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cmd/import.go b/cmd/import.go
@@ -78,6 +78,7 @@ func importMachine(cmd *cobra.Command, args []string) {
 	machineConfig, err := qemu.GetMachineConfig(importName)
 	if err != nil {
 		os.RemoveAll(targetDir)
+		os.RemoveAll(tempArchive)
 		log.Fatal("unable to import: " + err.Error())
 	}
 
@@ -87,6 +88,7 @@ func importMachine(cmd *cobra.Command, args []string) {
 	err = qemu.SaveMachineConfig(machineConfig)
 	if err != nil {
 		os.RemoveAll(targetDir)
+		os.RemoveAll(tempArchive)
 		log.Fatal("unable to import: " + err.Error())
 	}
 }