[Enhancement] - Add os_disk.storage_type and data_disk.storage_type to azurerm_image

internal/services/compute/image_resource.go

@@ -143,6 +143,14 @@ func resourceImage() *pluginsdk.Resource {
 							ForceNew:     true,
 							ValidateFunc: validate.DiskEncryptionSetID,
 						},
+
+						"storage_type": {
+							Type:         pluginsdk.TypeString,
+							Description:  "The type of storage disk",
+							Optional:     true,
+							ForceNew:     true,
+							ValidateFunc: validation.StringInSlice(images.PossibleValuesForStorageAccountTypes(), false),
+						},
 					},
 				},
 			},
@@ -188,6 +196,14 @@ func resourceImage() *pluginsdk.Resource {
 							Computed:     true,
 							ValidateFunc: validation.NoZeroValues,
 						},
+
+						"storage_type": {
+							Type:         pluginsdk.TypeString,
+							Description:  "The type of storage disk",
+							Optional:     true,
+							ForceNew:     true,
+							ValidateFunc: validation.StringInSlice(images.PossibleValuesForStorageAccountTypes(), false),
+						},
 					},
 				},
 			},
@@ -372,6 +388,10 @@ func expandImageOSDisk(input []interface{}) *images.ImageOSDisk {
 			}
 		}
 
+		if storageAccountType := config["storage_type"].(string); storageAccountType != "" {
+			out.StorageAccountType = pointer.To(images.StorageAccountTypes(storageAccountType))
+		}
+
 		return out
 	}
 
@@ -403,6 +423,10 @@ func expandImageDataDisks(disks []interface{}) *[]images.ImageDataDisk {
 			item.ManagedDisk = managedDisk
 		}
 
+		if storageAccountType := config["storage_type"].(string); storageAccountType != "" {
+			item.StorageAccountType = pointer.To(images.StorageAccountTypes(storageAccountType))
+		}
+
 		output = append(output, item)
 	}
 
@@ -434,6 +458,10 @@ func flattenImageOSDisk(input *images.ImageStorageProfile) []interface{} {
 			if set := v.DiskEncryptionSet; set != nil && set.Id != nil {
 				diskEncryptionSetId = *set.Id
 			}
+			storageAccountType := ""
+			if v.StorageAccountType != nil {
+				storageAccountType = string(*v.StorageAccountType)
+			}
 			output = append(output, map[string]interface{}{
 				"blob_uri":               blobUri,
 				"caching":                caching,
@@ -442,6 +470,7 @@ func flattenImageOSDisk(input *images.ImageStorageProfile) []interface{} {
 				"os_state":               string(v.OsState),
 				"size_gb":                diskSizeGB,
 				"disk_encryption_set_id": diskEncryptionSetId,
+				"storage_type":           storageAccountType,
 			})
 		}
 	}
@@ -471,12 +500,17 @@ func flattenImageDataDisks(input *images.ImageStorageProfile) []interface{} {
 				if disk.ManagedDisk != nil && disk.ManagedDisk.Id != nil {
 					managedDiskId = *disk.ManagedDisk.Id
 				}
+				storageAccountType := ""
+				if disk.StorageAccountType != nil {
+					storageAccountType = string(*disk.StorageAccountType)
+				}
 				output = append(output, map[string]interface{}{
 					"blob_uri":        blobUri,
 					"caching":         caching,
 					"lun":             int(disk.Lun),
 					"managed_disk_id": managedDiskId,
 					"size_gb":         diskSizeGb,
+					"storage_type":    storageAccountType,
 				})
 			}
 		}

internal/services/compute/image_resource_test.go

@@ -527,11 +527,12 @@ resource "azurerm_image" "test" {
   %s
 
   os_disk {
-    os_type  = "Linux"
-    os_state = "Generalized"
-    blob_uri = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
-    size_gb  = 30
-    caching  = "None"
+    os_type      = "Linux"
+    os_state     = "Generalized"
+    blob_uri     = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
+    size_gb      = 30
+    caching      = "None"
+    storage_type = "StandardSSD_LRS"
   }
 
   tags = {
@@ -553,11 +554,12 @@ resource "azurerm_image" "import" {
   resource_group_name = azurerm_image.test.resource_group_name
 
   os_disk {
-    os_type  = "Linux"
-    os_state = "Generalized"
-    blob_uri = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
-    size_gb  = 30
-    caching  = "None"
+    os_type      = "Linux"
+    os_state     = "Generalized"
+    blob_uri     = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
+    size_gb      = 30
+    caching      = "None"
+    storage_type = "StandardSSD_LRS"
   }
 
   tags = {
@@ -579,11 +581,12 @@ resource "azurerm_image" "testdestination" {
   resource_group_name = azurerm_resource_group.test.name
 
   os_disk {
-    os_type  = "Linux"
-    os_state = "Generalized"
-    blob_uri = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
-    size_gb  = 30
-    caching  = "None"
+    os_type      = "Linux"
+    os_state     = "Generalized"
+    blob_uri     = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
+    size_gb      = 30
+    caching      = "None"
+    storage_type = "StandardSSD_LRS"
   }
 
   tags = {
@@ -718,11 +721,12 @@ resource "azurerm_image" "testdestination" {
   resource_group_name = azurerm_resource_group.test.name
 
   os_disk {
-    os_type  = "Linux"
-    os_state = "Generalized"
-    blob_uri = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
-    size_gb  = 30
-    caching  = "None"
+    os_type      = "Linux"
+    os_state     = "Generalized"
+    blob_uri     = "${azurerm_storage_account.test.primary_blob_endpoint}${azurerm_storage_container.test.name}/myosdisk1.vhd"
+    size_gb      = 30
+    caching      = "None"
+    storage_type = "StandardSSD_LRS"
   }
 
   tags = {
@@ -873,6 +877,7 @@ resource "azurerm_image" "test" {
     size_gb                = 30
     caching                = "None"
     disk_encryption_set_id = azurerm_disk_encryption_set.test.id
+    storage_type           = "StandardSSD_LRS"
   }
 
   tags = {

website/docs/functions/parse_resource_id.html.markdown

@@ -66,4 +66,4 @@ normalise_resource_id(id string) string
 
 ## Arguments
 
-1. `id` (String) Azure Resource Manager ID.
+1. `id` (String) Azure Resource Manager ID.

website/docs/r/image.html.markdown

@@ -55,6 +55,7 @@ The `os_disk` block supports the following:
 * `caching` - (Optional) Specifies the caching mode as `ReadWrite`, `ReadOnly`, or `None`. The default is `None`.
 * `size_gb` - (Optional) Specifies the size of the image to be created. Changing this forces a new resource to be created.
 * `disk_encryption_set_id` - (Optional) The ID of the Disk Encryption Set which should be used to encrypt this image. Changing this forces a new resource to be created.
+* `storage_type` - (Optional) The type of storage disk to use. See [Azure API Docs](https://learn.microsoft.com/en-us/rest/api/compute/images/create-or-update?view=rest-compute-2024-03-02&tabs=HTTP#storageaccounttypes) for valid values.
 
 ---
 
@@ -65,6 +66,7 @@ The `data_disk` block supports the following:
 * `blob_uri` - (Optional) Specifies the URI in Azure storage of the blob that you want to use to create the image.
 * `caching` - (Optional) Specifies the caching mode as `ReadWrite`, `ReadOnly`, or `None`. Defaults to `None`.
 * `size_gb` - (Optional) Specifies the size of the image to be created. The target size can't be smaller than the source size.
+* `storage_type` - (Optional) The type of storage disk to use. See [Azure API Docs](https://learn.microsoft.com/en-us/rest/api/compute/images/create-or-update?view=rest-compute-2024-03-02&tabs=HTTP#storageaccounttypes) for valid values.
 
 ## Attributes Reference