From 63dc5cb8f6dbb6d273bd519c7768893df02a3b97 Mon Sep 17 00:00:00 2001
From: Pavel Boldyrev <627562+bpg@users.noreply.github.com>
Date: Sat, 10 Dec 2022 11:51:29 -0500
Subject: [PATCH] fix: handling `datastore_id` in LXC template (#180)

---
 example/resource_virtual_environment_container.tf   | 8 ++++++++
 proxmoxtf/resource_virtual_environment_container.go | 4 ++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/example/resource_virtual_environment_container.tf b/example/resource_virtual_environment_container.tf
index 9a158de3..3f887399 100644
--- a/example/resource_virtual_environment_container.tf
+++ b/example/resource_virtual_environment_container.tf
@@ -1,6 +1,10 @@
 resource "proxmox_virtual_environment_container" "example_template" {
   description = "Managed by Terraform"
 
+  disk {
+    datastore_id = element(data.proxmox_virtual_environment_datastores.example.datastore_ids, index(data.proxmox_virtual_environment_datastores.example.datastore_ids, "local-lvm"))
+  }
+
   initialization {
     dns {
       server = "1.1.1.1"
@@ -37,6 +41,10 @@ resource "proxmox_virtual_environment_container" "example_template" {
 }
 
 resource "proxmox_virtual_environment_container" "example" {
+  disk {
+    datastore_id = element(data.proxmox_virtual_environment_datastores.example.datastore_ids, index(data.proxmox_virtual_environment_datastores.example.datastore_ids, "local-lvm"))
+  }
+
   clone {
     vm_id = proxmox_virtual_environment_container.example_template.id
   }
diff --git a/proxmoxtf/resource_virtual_environment_container.go b/proxmoxtf/resource_virtual_environment_container.go
index 67ceda81..d33a392a 100644
--- a/proxmoxtf/resource_virtual_environment_container.go
+++ b/proxmoxtf/resource_virtual_environment_container.go
@@ -1393,12 +1393,12 @@ func resourceVirtualEnvironmentContainerRead(ctx context.Context, d *schema.Reso
 
 	if len(clone) > 0 {
 		if len(currentDisk) > 0 {
-			err := d.Set(mkResourceVirtualEnvironmentContainerDiskDatastoreID, []interface{}{disk})
+			err := d.Set(mkResourceVirtualEnvironmentContainerDisk, []interface{}{disk})
 			diags = append(diags, diag.FromErr(err)...)
 		}
 	} else if len(currentDisk) > 0 ||
 		disk[mkResourceVirtualEnvironmentContainerDiskDatastoreID] != dvResourceVirtualEnvironmentContainerDiskDatastoreID {
-		err := d.Set(mkResourceVirtualEnvironmentContainerDiskDatastoreID, []interface{}{disk})
+		err := d.Set(mkResourceVirtualEnvironmentContainerDisk, []interface{}{disk})
 		diags = append(diags, diag.FromErr(err)...)
 	}