Add image state Downloading

harvester · Sep 18, 2021 · 1e3ea9f · 1e3ea9f
1 parent c85d5e9
commit 1e3ea9f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/pkg/constants/constants_image.go b/pkg/constants/constants_image.go
@@ -10,6 +10,7 @@ const (
 	FieldImageSize             = "size"
 	FieldImageStorageClassName = "storage_class_name"
 
-	StateImageUploading = "Uploading"
-	StateImageFailed    = "Failed"
+	StateImageUploading   = "Uploading"
+	StateImageDownloading = "Downloading"
+	StateImageFailed      = "Failed"
 )
diff --git a/pkg/importer/resource_image_importer.go b/pkg/importer/resource_image_importer.go
@@ -38,7 +38,12 @@ func ResourceImageStateGetter(obj *harvsterv1.VirtualMachineImage) (*StateGetter
 		if imported {
 			state = constants.StateCommonActive
 		} else {
-			state = constants.StateImageUploading
+			switch obj.Spec.SourceType {
+			case harvsterv1.VirtualMachineImageSourceTypeDownload:
+				state = constants.StateImageDownloading
+			default:
+				state = constants.StateImageUploading
+			}
 		}
 	} else {
 		state = constants.StateImageFailed