From ed33a18c9b6499ff33bacb79cebfd510b24a29c8 Mon Sep 17 00:00:00 2001
From: Pavel Boldyrev <627562+bpg@users.noreply.github.com>
Date: Sat, 27 Jan 2024 20:18:51 -0500
Subject: [PATCH] fix(pool): missing `pool_id` after import (#974)

Signed-off-by: Pavel Boldyrev <627562+bpg@users.noreply.github.com>
---
 proxmoxtf/resource/pool.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/proxmoxtf/resource/pool.go b/proxmoxtf/resource/pool.go
index 658a278a..ef9f1b91 100644
--- a/proxmoxtf/resource/pool.go
+++ b/proxmoxtf/resource/pool.go
@@ -8,6 +8,7 @@ package resource
 
 import (
 	"context"
+	"fmt"
 	"strings"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
@@ -86,7 +87,15 @@ func Pool() *schema.Resource {
 		UpdateContext: poolUpdate,
 		DeleteContext: poolDelete,
 		Importer: &schema.ResourceImporter{
-			StateContext: schema.ImportStatePassthroughContext,
+			StateContext: func(ctx context.Context, d *schema.ResourceData, i interface{}) ([]*schema.ResourceData, error) {
+				d.SetId(d.Id())
+				err := d.Set(mkResourceVirtualEnvironmentPoolPoolID, d.Id())
+				if err != nil {
+					return nil, fmt.Errorf("failed setting state during import: %w", err)
+				}
+
+				return []*schema.ResourceData{d}, nil
+			},
 		},
 	}
 }