3.6.0

internal/service/cloudapi/k8s/api.go

@@ -39,6 +39,8 @@ const (
 	K8sDeleteAPI      = "/restmachine/cloudapi/k8s/delete"
 	K8sListAPI        = "/restmachine/cloudapi/k8s/list"
 	K8sListDeletedAPI = "/restmachine/cloudapi/k8s/listDeleted"
+	K8sRestoreAPI     = "/restmachine/cloudapi/k8s/restore"
+	K8sEnableAPI     = "/restmachine/cloudapi/k8s/enable"
 
 	K8sWgCreateAPI = "/restmachine/cloudapi/k8s/workersGroupAdd"
 	K8sWgDeleteAPI = "/restmachine/cloudapi/k8s/workersGroupDelete"

internal/service/cloudapi/k8s/data_source_k8s_wg.go

@@ -38,51 +38,21 @@ import (
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
-	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/constants"
-	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/service/cloudapi/kvmvm"
 	log "github.com/sirupsen/logrus"
+	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/constants"
 )
 
 func dataSourceK8sWgRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
 	log.Debugf("dataSourceK8sWgRead: called with k8s id %d", d.Get("k8s_id").(int))
 
-	k8s, err := utilityDataK8sCheckPresence(ctx, d, m)
+	wg, workersComputeList, err := utilityDataK8sWgCheckPresence(ctx, d, m)
 	if err != nil {
 		return diag.FromErr(err)
 	}
+
 	d.SetId(strconv.Itoa(d.Get("wg_id").(int)))
 
-	var id int
-	if d.Id() != "" {
-		id, err = strconv.Atoi(d.Id())
-		if err != nil {
-			return diag.FromErr(err)
-		}
-	} else {
-		id = d.Get("wg_id").(int)
-	}
-
-	curWg := K8SGroup{}
-	for _, wg := range k8s.K8SGroups.Workers {
-		if wg.ID == uint64(id) {
-			curWg = wg
-			break
-		}
-	}
-	if curWg.ID == 0 {
-		return diag.Errorf("Not found wg with id: %v in k8s cluster: %v", id, k8s.ID)
-	}
-
-	workersComputeList := make([]kvmvm.ComputeGetResp, 0, 0)
-	for _, info := range curWg.DetailedInfo {
-		compute, err := utilityComputeCheckPresence(ctx, d, m, info.ID)
-		if err != nil {
-			return diag.FromErr(err)
-		}
-		workersComputeList = append(workersComputeList, *compute)
-	}
-
-	flattenWgData(d, curWg, workersComputeList)
+	flattenWg(d, *wg, workersComputeList)
 
 	return nil
 }

internal/service/cloudapi/k8s/flattens.go

@@ -257,9 +257,13 @@ func flattenK8sList(d *schema.ResourceData, k8sItems K8SList) {
 }
 
 func flattenResourceK8s(d *schema.ResourceData, k8s K8SRecord, masters []kvmvm.ComputeGetResp, workers []kvmvm.ComputeGetResp) {
+	wg_name := k8s.K8SGroups.Workers[0].Name
+
 	d.Set("acl", flattenAcl(k8s.ACL))
 	d.Set("account_id", k8s.AccountID)
 	d.Set("account_name", k8s.AccountName)
+	d.Set("k8sci_id", k8s.CIID)
+	d.Set("wg_name", wg_name)
 	d.Set("bservice_id", k8s.BServiceID)
 	d.Set("created_by", k8s.CreatedBy)
 	d.Set("created_time", k8s.CreatedTime)
@@ -268,7 +272,9 @@ func flattenResourceK8s(d *schema.ResourceData, k8s K8SRecord, masters []kvmvm.C
 	d.Set("k8s_ci_name", k8s.K8CIName)
 	d.Set("masters", flattenMasterGroup(k8s.K8SGroups.Masters, masters))
 	d.Set("workers", flattenK8sGroup(k8s.K8SGroups.Workers, workers))
+	d.Set("with_lb", k8s.LBID != 0)
 	d.Set("lb_id", k8s.LBID)
+	d.Set("name", k8s.Name)
 	d.Set("rg_id", k8s.RGID)
 	d.Set("rg_name", k8s.RGName)
 	d.Set("status", k8s.Status)
@@ -278,7 +284,7 @@ func flattenResourceK8s(d *schema.ResourceData, k8s K8SRecord, masters []kvmvm.C
 	d.Set("default_wg_id", k8s.K8SGroups.Workers[0].ID)
 }
 
-func flattenWgData(d *schema.ResourceData, wg K8SGroup, computes []kvmvm.ComputeGetResp) {
+func flattenWg(d *schema.ResourceData, wg K8SGroup, computes []kvmvm.ComputeGetResp) {
 	d.Set("annotations", wg.Annotations)
 	d.Set("cpu", wg.CPU)
 	d.Set("detailed_info", flattenDetailedInfo(wg.DetailedInfo, computes))

internal/service/cloudapi/k8s/resource_check_input_values.go

@@ -0,0 +1,143 @@
+package k8s
+
+import (
+	"context"
+	"encoding/json"
+	"net/url"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/controller"
+)
+
+func existK8sID(ctx context.Context, d *schema.ResourceData, m interface{}) (bool, error) {
+	c := m.(*controller.ControllerCfg)
+	urlValues := &url.Values{}
+
+	k8sList := []struct {
+		ID int `json:"id"`
+	}{}
+
+	k8sListAPI := "/restmachine/cloudapi/k8s/list"
+
+	k8sListRaw, err := c.DecortAPICall(ctx, "POST", k8sListAPI, urlValues)
+	if err != nil {
+		return false, err
+	}
+
+	err = json.Unmarshal([]byte(k8sListRaw), &k8sList)
+	if err != nil {
+		return false, err
+	}
+
+	haveK8s := false
+	k8sID := d.Get("k8s_id").(int)
+	for _, k8s := range k8sList {
+		if k8s.ID == k8sID {
+			haveK8s = true
+			break
+		}
+	}
+
+	return haveK8s, nil
+}
+
+func existK8sCIID(ctx context.Context, d *schema.ResourceData, m interface{}) (bool, error) {
+	c := m.(*controller.ControllerCfg)
+	urlValues := &url.Values{}
+
+	k8sciList := []struct {
+		ID int `json:"id"`
+	}{}
+
+	k8sciListAPI := "/restmachine/cloudapi/k8ci/list"
+
+	k8sciListRaw, err := c.DecortAPICall(ctx, "POST", k8sciListAPI, urlValues)
+	if err != nil {
+		return false, err
+	}
+
+	err = json.Unmarshal([]byte(k8sciListRaw), &k8sciList)
+	if err != nil {
+		return false, err
+	}
+
+	haveK8sCI := false
+	k8sciID := d.Get("k8sci_id").(int)
+	for _, k8ci := range k8sciList {
+		if k8ci.ID == k8sciID {
+			haveK8sCI = true
+			break
+		}
+	}
+
+	return haveK8sCI, nil
+}
+
+func existRGID(ctx context.Context, d *schema.ResourceData, m interface{}) (bool, error) {
+	c := m.(*controller.ControllerCfg)
+	urlValues := &url.Values{}
+
+	rgList := []struct {
+		ID int `json:"id"`
+	}{}
+
+	rgListAPI := "/restmachine/cloudapi/rg/list"
+
+	rgListRaw, err := c.DecortAPICall(ctx, "POST", rgListAPI, urlValues)
+	if err != nil {
+		return false, err
+	}
+
+	err = json.Unmarshal([]byte(rgListRaw), &rgList)
+	if err != nil {
+		return false, err
+	}
+
+	haveRG := false
+	rgId := d.Get("rg_id").(int)
+	for _, rg := range rgList {
+		if rg.ID == rgId {
+			haveRG = true
+			break
+		}
+	}
+
+	return haveRG, nil
+}
+
+func existExtNetID(ctx context.Context, d *schema.ResourceData, m interface{}) (bool, error) {
+	extNetID := d.Get("extnet_id").(int)
+
+	if extNetID == 0 {
+		return true, nil
+	}
+
+	c := m.(*controller.ControllerCfg)
+	urlValues := &url.Values{}
+
+	extNetList := []struct {
+		ID int `json:"id"`
+	}{}
+
+	extNetListAPI := "/restmachine/cloudapi/extnet/list"
+
+	extNetListRaw, err := c.DecortAPICall(ctx, "POST", extNetListAPI, urlValues)
+	if err != nil {
+		return false, err
+	}
+
+	err = json.Unmarshal([]byte(extNetListRaw), &extNetList)
+	if err != nil {
+		return false, err
+	}
+
+	haveExtNet := false
+	for _, extNet := range extNetList {
+		if extNet.ID == extNetID {
+			haveExtNet = true
+			break
+		}
+	}
+
+	return haveExtNet, nil
+}

internal/service/cloudapi/k8s/resource_k8s.go

@@ -43,15 +43,45 @@ import (
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	log "github.com/sirupsen/logrus"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/constants"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/controller"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/service/cloudapi/kvmvm"
-	log "github.com/sirupsen/logrus"
+	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/status"
 )
 
 func resourceK8sCreate(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
 	log.Debugf("resourceK8sCreate: called with name %s, rg %d", d.Get("name").(string), d.Get("rg_id").(int))
 
+	haveRGID, err := existRGID(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if !haveRGID {
+		return diag.Errorf("resourceK8sCreate: can't create k8s cluster because RGID %d is not allowed or does not exist", d.Get("rg_id").(int))
+	}
+
+	haveK8sciID, err := existK8sCIID(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if !haveK8sciID {
+		return diag.Errorf("resourceK8sCreate: can't create k8s cluster because K8sCIID %d is not allowed or does not exist", d.Get("k8sci_id").(int))
+	}
+
+	if _, ok := d.GetOk("extnet_id"); ok {
+		haveExtNetID, err := existExtNetID(ctx, d, m)
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		if !haveExtNetID {
+			return diag.Errorf("resourceK8sCreate: can't create k8s cluster because ExtNetID %d is not allowed or does not exist", d.Get("extnet_id").(int))
+		}
+	}
+
 	c := m.(*controller.ControllerCfg)
 	urlValues := &url.Values{}
 	urlValues.Add("name", d.Get("name").(string))
@@ -158,12 +188,61 @@ func resourceK8sCreate(ctx context.Context, d *schema.ResourceData, m interface{
 }
 
 func resourceK8sRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
-
-	k8s, err := utilityDataK8sCheckPresence(ctx, d, m)
+	k8s, err := utilityK8sCheckPresence(ctx, d, m)
 	if err != nil {
 		d.SetId("")
 		return diag.FromErr(err)
 	}
+
+	c := m.(*controller.ControllerCfg)
+
+	hasChanged := false
+
+	switch k8s.Status {
+	case status.Modeled:
+		return diag.Errorf("The k8s cluster is in status: %s, please, contact support for more information", k8s.Status)
+	case status.Creating:
+	case status.Created:
+	case status.Deleting:
+	case status.Deleted:
+		urlVal := &url.Values{}
+		urlVal.Add("k8sId", d.Id())
+
+		_, err := c.DecortAPICall(ctx, "POST", K8sRestoreAPI, urlVal)
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		_, err = c.DecortAPICall(ctx, "POST", K8sEnableAPI, urlVal)
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		hasChanged = true
+	case status.Destroying:
+		return diag.Errorf("The k8s cluster is in progress with status: %s", k8s.Status)
+	case status.Destroyed:
+		d.SetId("")
+		return resourceK8sCreate(ctx, d, m)
+	case status.Enabling:
+	case status.Enabled:
+	case status.Disabling:
+	case status.Disabled:
+		log.Debugf("The k8s cluster is in status: %s, troubles may occur with update. Please, enable compute first.", k8s.Status)
+	case status.Restoring:
+	}
+
+	if hasChanged {
+		k8s, err = utilityK8sCheckPresence(ctx, d, m)
+		if k8s == nil {
+			d.SetId("")
+			if err != nil {
+				return diag.FromErr(err)
+			}
+			return nil
+		}
+	}
+
 	k8sList, err := utilityK8sListCheckPresence(ctx, d, m, K8sListAPI)
 	if err != nil {
 		d.SetId("")
@@ -201,7 +280,6 @@ func resourceK8sRead(ctx context.Context, d *schema.ResourceData, m interface{})
 
 	flattenResourceK8s(d, *k8s, masterComputeList, workersComputeList)
 
-	c := m.(*controller.ControllerCfg)
 	urlValues := &url.Values{}
 	urlValues.Add("lbId", strconv.FormatUint(k8s.LBID, 10))
 
@@ -233,6 +311,76 @@ func resourceK8sUpdate(ctx context.Context, d *schema.ResourceData, m interface{
 
 	c := m.(*controller.ControllerCfg)
 
+	haveRGID, err := existRGID(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if !haveRGID {
+		return diag.Errorf("resourceK8sUpdate: can't update k8s cluster because RGID %d is not allowed or does not exist", d.Get("rg_id").(int))
+	}
+
+	haveK8sciID, err := existK8sCIID(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if !haveK8sciID {
+		return diag.Errorf("resourceK8sUpdate: can't update k8s cluster because K8sCIID %d is not allowed or does not exist", d.Get("k8sci_id").(int))
+	}
+
+	k8s, err := utilityK8sCheckPresence(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	hasChanged := false
+
+	switch k8s.Status {
+	case status.Modeled:
+		return diag.Errorf("The k8s cluster is in status: %s, please, contact support for more information", k8s.Status)
+	case status.Creating:
+	case status.Created:
+	case status.Deleting:
+	case status.Deleted:
+		urlVal := &url.Values{}
+		urlVal.Add("k8sId", d.Id())
+
+		_, err := c.DecortAPICall(ctx, "POST", K8sRestoreAPI, urlVal)
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		_, err = c.DecortAPICall(ctx, "POST", K8sEnableAPI, urlVal)
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		hasChanged = true
+	case status.Destroying:
+		return diag.Errorf("The k8s cluster is in progress with status: %s", k8s.Status)
+	case status.Destroyed:
+		d.SetId("")
+		return resourceK8sCreate(ctx, d, m)
+	case status.Enabling:
+	case status.Enabled:
+	case status.Disabling:
+	case status.Disabled:
+		log.Debugf("The k8s cluster is in status: %s, troubles may occur with update. Please, enable compute first.", k8s.Status)
+	case status.Restoring:
+	}
+
+	if hasChanged {
+		k8s, err = utilityK8sCheckPresence(ctx, d, m)
+		if k8s == nil {
+			d.SetId("")
+			if err != nil {
+				return diag.FromErr(err)
+			}
+			return nil
+		}
+	}
+
 	if d.HasChange("name") {
 		urlValues := &url.Values{}
 		urlValues.Add("k8sId", d.Id())
@@ -245,11 +393,6 @@ func resourceK8sUpdate(ctx context.Context, d *schema.ResourceData, m interface{
 	}
 
 	if d.HasChange("workers") {
-		k8s, err := utilityK8sCheckPresence(ctx, d, m)
-		if err != nil {
-			return diag.FromErr(err)
-		}
-
 		wg := k8s.K8SGroups.Workers[0]
 		urlValues := &url.Values{}
 		urlValues.Add("k8sId", d.Id())
@@ -321,7 +464,6 @@ func resourceK8sSchemaMake() map[string]*schema.Schema {
 		"wg_name": {
 			Type:        schema.TypeString,
 			Required:    true,
-			ForceNew:    true,
 			Description: "Name for first worker group created with cluster.",
 		},
 		"labels": {
@@ -369,7 +511,6 @@ func resourceK8sSchemaMake() map[string]*schema.Schema {
 		"with_lb": {
 			Type:        schema.TypeBool,
 			Optional:    true,
-			ForceNew:    true,
 			Default:     true,
 			Description: "Create k8s with load balancer if true.",
 		},

internal/service/cloudapi/k8s/resource_k8s_wg.go

@@ -40,15 +40,24 @@ import (
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	log "github.com/sirupsen/logrus"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/constants"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/controller"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/service/cloudapi/kvmvm"
-	log "github.com/sirupsen/logrus"
 )
 
 func resourceK8sWgCreate(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
 	log.Debugf("resourceK8sWgCreate: called with k8s id %d", d.Get("k8s_id").(int))
 
+	haveK8sID, err := existK8sID(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if !haveK8sID {
+		return diag.Errorf("resourceK8sCreate: can't create k8s cluster because K8sID %d is not allowed or does not exist", d.Get("k8s_id").(int))
+	}
+
 	c := m.(*controller.ControllerCfg)
 	urlValues := &url.Values{}
 	urlValues.Add("k8sId", strconv.Itoa(d.Get("k8s_id").(int)))
@@ -56,7 +65,12 @@ func resourceK8sWgCreate(ctx context.Context, d *schema.ResourceData, m interfac
 	urlValues.Add("workerNum", strconv.Itoa(d.Get("num").(int)))
 	urlValues.Add("workerCpu", strconv.Itoa(d.Get("cpu").(int)))
 	urlValues.Add("workerRam", strconv.Itoa(d.Get("ram").(int)))
-	urlValues.Add("workerDisk", strconv.Itoa(d.Get("disk").(int)))
+
+	if d.Get("disk") == nil {
+		urlValues.Add("workerDisk", strconv.Itoa(0))
+	} else {
+		urlValues.Add("workerDisk", strconv.Itoa(d.Get("disk").(int)))
+	}
 
 	resp, err := c.DecortAPICall(ctx, "POST", K8sWgCreateAPI, urlValues)
 	if err != nil {
@@ -100,34 +114,13 @@ func resourceK8sWgCreate(ctx context.Context, d *schema.ResourceData, m interfac
 func resourceK8sWgRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
 	log.Debugf("resourceK8sWgRead: called with %v", d.Id())
 
-	k8s, err := utilityDataK8sCheckPresence(ctx, d, m)
+	wg, err := utilityK8sWgCheckPresence(ctx, d, m)
 	if err != nil {
 		return diag.FromErr(err)
 	}
 
-	var id int
-	if d.Id() != "" {
-		id, err = strconv.Atoi(strings.Split(d.Id(), "#")[0])
-		if err != nil {
-			return diag.FromErr(err)
-		}
-	} else {
-		id = d.Get("wg_id").(int)
-	}
-
-	curWg := K8SGroup{}
-	for _, wg := range k8s.K8SGroups.Workers {
-		if wg.ID == uint64(id) {
-			curWg = wg
-			break
-		}
-	}
-	if curWg.ID == 0 {
-		return diag.Errorf("Not found wg with id: %v in k8s cluster: %v", id, k8s.ID)
-	}
-
 	workersComputeList := make([]kvmvm.ComputeGetResp, 0, 0)
-	for _, info := range curWg.DetailedInfo {
+	for _, info := range wg.DetailedInfo {
 		compute, err := utilityComputeCheckPresence(ctx, d, m, info.ID)
 		if err != nil {
 			return diag.FromErr(err)
@@ -135,10 +128,20 @@ func resourceK8sWgRead(ctx context.Context, d *schema.ResourceData, m interface{
 		workersComputeList = append(workersComputeList, *compute)
 	}
 
+	d.Set("wg_id", wg.ID)
+	if strings.Contains(d.Id(), "#") {
+		k8sId, err := strconv.Atoi(strings.Split(d.Id(), "#")[1])
+		if err != nil {
+			return diag.FromErr(err)
+		}
+
+		d.Set("k8s_id", k8sId)
+	} else {
+		d.Set("k8s_id", d.Get("k8s_id"))
+	}
 	d.SetId(strings.Split(d.Id(), "#")[0])
-	d.Set("k8s_id", k8s.ID)
-	d.Set("wg_id", curWg.ID)
-	flattenWgData(d, curWg, workersComputeList)
+
+	flattenWg(d, *wg, workersComputeList)
 
 	return nil
 }
@@ -148,6 +151,15 @@ func resourceK8sWgUpdate(ctx context.Context, d *schema.ResourceData, m interfac
 
 	c := m.(*controller.ControllerCfg)
 
+	haveK8sID, err := existK8sID(ctx, d, m)
+	if err != nil {
+		return diag.FromErr(err)
+	}
+
+	if !haveK8sID {
+		return diag.Errorf("resourceK8sUpdate: can't update k8s cluster because K8sID %d is not allowed or does not exist", d.Get("k8s_id").(int))
+	}
+
 	wg, err := utilityK8sWgCheckPresence(ctx, d, m)
 	if err != nil {
 		return diag.FromErr(err)
@@ -242,8 +254,7 @@ func resourceK8sWgSchemaMake() map[string]*schema.Schema {
 		"disk": {
 			Type:        schema.TypeInt,
 			Optional:    true,
-			ForceNew:    true,
-			Default:     0,
+			Computed:    true,
 			Description: "Worker node boot disk size. If unspecified or 0, size is defined by OS image size.",
 		},
 		"wg_id": {

internal/service/cloudapi/k8s/utility_k8s.go

@@ -37,7 +37,6 @@ import (
 	"encoding/json"
 	"net/url"
 	"strconv"
-	"strings"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/controller"
@@ -89,15 +88,8 @@ func utilityComputeCheckPresence(ctx context.Context, d *schema.ResourceData, m
 func utilityDataK8sCheckPresence(ctx context.Context, d *schema.ResourceData, m interface{}) (*K8SRecord, error) {
 	c := m.(*controller.ControllerCfg)
 	urlValues := &url.Values{}
-	if d.Get("k8s_id") != 0 && d.Get("k8s_id") != nil {
-		urlValues.Add("k8sId", strconv.Itoa(d.Get("k8s_id").(int)))
-	} else if id := d.Id(); id != "" {
-		if strings.Contains(id, "#") {
-			urlValues.Add("k8sId", strings.Split(d.Id(), "#")[1])
-		} else {
-			urlValues.Add("k8sId", d.Id())
-		}
-	}
+	urlValues.Add("k8sId", strconv.Itoa(d.Get("k8s_id").(int)))
+
 	k8sRaw, err := c.DecortAPICall(ctx, "POST", K8sGetAPI, urlValues)
 	if err != nil {
 		return nil, err

internal/service/cloudapi/k8s/utility_k8s_wg.go

@@ -38,15 +38,82 @@ import (
 	"fmt"
 	"net/url"
 	"strconv"
+	"strings"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/controller"
+	"repository.basistech.ru/BASIS/terraform-provider-decort/internal/service/cloudapi/kvmvm"
 )
 
+func utilityDataK8sWgCheckPresence(ctx context.Context, d *schema.ResourceData, m interface{}) (*K8SGroup, []kvmvm.ComputeGetResp, error) {
+	c := m.(*controller.ControllerCfg)
+	urlValues := &url.Values{}
+
+	k8sId := d.Get("k8s_id").(int)
+	wgId := d.Get("wg_id").(int)
+
+	urlValues.Add("k8sId", strconv.Itoa(k8sId))
+
+	k8sRaw, err := c.DecortAPICall(ctx, "POST", K8sGetAPI, urlValues)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	k8s := K8SRecord{}
+	err = json.Unmarshal([]byte(k8sRaw), &k8s)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	curWg := K8SGroup{}
+	for _, wg := range k8s.K8SGroups.Workers {
+		if wg.ID == uint64(wgId) {
+			curWg = wg
+			break
+		}
+	}
+	if curWg.ID == 0 {
+		return nil, nil, fmt.Errorf("WG with id %v in k8s cluster %v not found", wgId, k8sId)
+	}
+
+	workersComputeList := make([]kvmvm.ComputeGetResp, 0, 0)
+	for _, info := range curWg.DetailedInfo {
+		compute, err := utilityComputeCheckPresence(ctx, d, m, info.ID)
+		if err != nil {
+			return nil, nil, err
+		}
+
+		workersComputeList = append(workersComputeList, *compute)
+	}
+
+	return &curWg, workersComputeList, nil
+}
+
 func utilityK8sWgCheckPresence(ctx context.Context, d *schema.ResourceData, m interface{}) (*K8SGroup, error) {
 	c := m.(*controller.ControllerCfg)
 	urlValues := &url.Values{}
-	urlValues.Add("k8sId", strconv.Itoa(d.Get("k8s_id").(int)))
+	var wgId int
+	var k8sId int
+	var err error
+
+	if strings.Contains(d.Id(), "#") {
+		wgId, err = strconv.Atoi(strings.Split(d.Id(), "#")[0])
+		if err != nil {
+			return nil, err
+		}
+		k8sId, err = strconv.Atoi(strings.Split(d.Id(), "#")[1])
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		wgId, err = strconv.Atoi(d.Id())
+		if err != nil {
+			return nil, err
+		}
+		k8sId = d.Get("k8s_id").(int)
+	}
+
+	urlValues.Add("k8sId", strconv.Itoa(k8sId))
 
 	resp, err := c.DecortAPICall(ctx, "POST", K8sGetAPI, urlValues)
 	if err != nil {
@@ -62,21 +129,11 @@ func utilityK8sWgCheckPresence(ctx context.Context, d *schema.ResourceData, m in
 		return nil, err
 	}
 
-	var id int
-	if d.Id() != "" {
-		id, err = strconv.Atoi(d.Id())
-		if err != nil {
-			return nil, err
-		}
-	} else {
-		id = d.Get("wg_id").(int)
-	}
-
 	for _, wg := range k8s.K8SGroups.Workers {
-		if wg.ID == uint64(id) {
+		if wg.ID == uint64(wgId) {
 			return &wg, nil
 		}
 	}
 
-	return nil, fmt.Errorf("Not found wg with id: %v in k8s cluster: %v", id, k8s.ID)
+	return nil, fmt.Errorf("Not found wg with id: %v in k8s cluster: %v", wgId, k8s.ID)
 }