add VM affinity policies

Author: jcantrell-ntnx

examples/vm_anti_affinity_policy_v2/main.tf

@@ -0,0 +1,38 @@
+terraform {
+  required_providers {
+    nutanix = {
+      source  = "nutanix/nutanix"
+      version = "2.0.0"
+    }
+  }
+}
+
+#defining nutanix configuration
+provider "nutanix" {
+  username = var.nutanix_username
+  password = var.nutanix_password
+  endpoint = var.nutanix_endpoint
+  port     = 9440
+  insecure = true
+}
+
+#creating category
+resource "nutanix_category_v2" "category" {
+  key   = "AntiAffinityPolicy"
+  value = "tf-anti-affinity-policy"
+}
+
+# Create policy
+resource "nutanix_vm_anti_affinity_policy_v2" "policy" {
+  name        = "temp-anti-affinity"
+  description = "a description"
+  categories  = [ nutanix_category_v2.category.id ]
+}
+
+# Get all policies
+data "nutanix_vm_anti_affinity_policies_v2" "policies" {}
+
+# Get specific policy
+data "nutanix_vm_anti_affinity_policy_v2" "policy" {
+  ext_id = nutanix_vm_anti_affinity_policy_v2.policy.ext_id
+}

examples/vm_anti_affinity_policy_v2/terraform.tfvars

@@ -0,0 +1,4 @@
+#define values to the variables to be used in terraform file
+nutanix_username = "admin"
+nutanix_password = "password"
+nutanix_endpoint = "10.xx.xx.xx"

examples/vm_anti_affinity_policy_v2/variables.tf

@@ -0,0 +1,10 @@
+#define the type of variables to be used in terraform file
+variable "nutanix_username" {
+  type = string
+}
+variable "nutanix_password" {
+  type = string
+}
+variable "nutanix_endpoint" {
+  type = string
+}

examples/vm_host_affinity_policy_v2/main.tf

@@ -0,0 +1,45 @@
+terraform {
+  required_providers {
+    nutanix = {
+      source  = "nutanix/nutanix"
+      version = "2.0.0"
+    }
+  }
+}
+
+#defining nutanix configuration
+provider "nutanix" {
+  username = var.nutanix_username
+  password = var.nutanix_password
+  endpoint = var.nutanix_endpoint
+  port     = 9440
+  insecure = true
+}
+
+#creating Host category
+resource "nutanix_category_v2" "host_category" {
+  key   = "HostAffinityPolicy"
+  value = "tf-host-affinity-host-category"
+}
+
+# Create VM category
+resource "nutanix_category_v2" "vm_category" {
+  key   = "HostAffinityPolicy"
+  value = "tf-host-affinity-vm-category"
+}
+
+# Create policy
+resource "nutanix_vm_host_affinity_policy_v2" "policy" {
+  name        = "temp-host-affinity"
+  description = "a description"
+  host_categories  = [ nutanix_category_v2.host_category.id ]
+  vm_categories  = [ nutanix_category_v2.vm_category.id ]
+}
+
+# Get all policies
+data "nutanix_vm_host_affinity_policies_v2" "policies" {}
+
+# Get specific policy
+data "nutanix_vm_host_affinity_policy_v2" "policy" {
+  ext_id = nutanix_vm_host_affinity_policy_v2.policy.ext_id
+}

examples/vm_host_affinity_policy_v2/terraform.tfvars

@@ -0,0 +1,4 @@
+#define values to the variables to be used in terraform file
+nutanix_username = "admin"
+nutanix_password = "password"
+nutanix_endpoint = "10.xx.xx.xx"

examples/vm_host_affinity_policy_v2/variables.tf

@@ -0,0 +1,10 @@
+#define the type of variables to be used in terraform file
+variable "nutanix_username" {
+  type = string
+}
+variable "nutanix_password" {
+  type = string
+}
+variable "nutanix_endpoint" {
+  type = string
+}

nutanix/provider/provider.go

@@ -306,6 +306,10 @@ func Provider() *schema.Provider {
 			"nutanix_ngt_configuration_v2":                    vmmv2.DatasourceNutanixNGTConfigurationV4(),
 			"nutanix_image_placement_policy_v2":               vmmv2.DatasourceNutanixImagePlacementV4(),
 			"nutanix_image_placement_policies_v2":             vmmv2.DatasourceNutanixImagePlacementsV4(),
+			"nutanix_vm_anti_affinity_policy_v2":              vmmv2.DatasourceNutanixVMAntiAffinityPolicyV2(),
+			"nutanix_vm_anti_affinity_policies_v2":            vmmv2.DatasourceNutanixVMAntiAffinityPoliciesV2(),
+			"nutanix_vm_host_affinity_policy_v2":              vmmv2.DatasourceNutanixVMHostAffinityPolicyV2(),
+			"nutanix_vm_host_affinity_policies_v2":            vmmv2.DatasourceNutanixVMHostAffinityPoliciesV2(),
 			"nutanix_cluster_v2":                              clustersv2.DatasourceNutanixClusterEntityV2(),
 			"nutanix_clusters_v2":                             clustersv2.DatasourceNutanixClusterEntitiesV2(),
 			"nutanix_host_v2":                                 clustersv2.DatasourceNutanixHostEntityV2(),
@@ -427,6 +431,8 @@ func Provider() *schema.Provider {
 			"nutanix_vm_network_device_assign_ip_v2":          vmmv2.ResourceNutanixVmsNetworkDeviceAssignIPV2(),
 			"nutanix_vm_network_device_migrate_v2":            vmmv2.ResourceNutanixVmsNetworkDeviceMigrateV2(),
 			"nutanix_image_placement_policy_v2":               vmmv2.ResourceNutanixImagePlacementV2(),
+			"nutanix_vm_host_affinity_policy_v2":              vmmv2.ResourceNutanixVMHostAffinityPolicyV2(),
+			"nutanix_vm_anti_affinity_policy_v2":              vmmv2.ResourceNutanixVMAntiAffinityPolicyV2(),
 			"nutanix_cluster_v2":                              clustersv2.ResourceNutanixClusterV2(),
 			"nutanix_cluster_add_node_v2":                     clustersv2.ResourceNutanixClusterAddNodeV2(),
 			"nutanix_clusters_discover_unconfigured_nodes_v2": clustersv2.ResourceNutanixClusterDiscoverUnconfiguredNodesV2(),

nutanix/sdks/v4/vmm/vmm.go

@@ -7,10 +7,12 @@ import (
 )
 
 type Client struct {
-	ImagesAPIInstance          *api.ImagesApi
-	TemplatesAPIInstance       *api.TemplatesApi
-	VMAPIInstance              *api.VmApi
-	ImagesPlacementAPIInstance *api.ImagePlacementPoliciesApi
+	ImagesAPIInstance               *api.ImagesApi
+	TemplatesAPIInstance            *api.TemplatesApi
+	VMAPIInstance                   *api.VmApi
+	ImagesPlacementAPIInstance      *api.ImagePlacementPoliciesApi
+	VMAntiAffinityPolicyAPIInstance *api.VmAntiAffinityPoliciesApi
+	VMHostAffinityPolicyAPIInstance *api.VmHostAffinityPoliciesApi
 }
 
 func NewVmmClient(credentials client.Credentials) (*Client, error) {
@@ -30,10 +32,12 @@ func NewVmmClient(credentials client.Credentials) (*Client, error) {
 	}
 
 	f := &Client{
-		ImagesAPIInstance:          api.NewImagesApi(baseClient),
-		TemplatesAPIInstance:       api.NewTemplatesApi(baseClient),
-		VMAPIInstance:              api.NewVmApi(baseClient),
-		ImagesPlacementAPIInstance: api.NewImagePlacementPoliciesApi(baseClient),
+		ImagesAPIInstance:               api.NewImagesApi(baseClient),
+		TemplatesAPIInstance:            api.NewTemplatesApi(baseClient),
+		VMAPIInstance:                   api.NewVmApi(baseClient),
+		ImagesPlacementAPIInstance:      api.NewImagePlacementPoliciesApi(baseClient),
+		VMAntiAffinityPolicyAPIInstance: api.NewVmAntiAffinityPoliciesApi(baseClient),
+		VMHostAffinityPolicyAPIInstance: api.NewVmHostAffinityPoliciesApi(baseClient),
 	}
 
 	return f, nil

nutanix/services/vmmv2/data_source_nutanix_vm_anti_affinity_policies_v2.go

@@ -0,0 +1,189 @@
+package vmmv2
+
+import (
+	"context"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/nutanix/ntnx-api-golang-clients/vmm-go-client/v4/models/vmm/v4/ahv/policies"
+	conns "github.com/terraform-providers/terraform-provider-nutanix/nutanix"
+	"github.com/terraform-providers/terraform-provider-nutanix/utils"
+)
+
+func DatasourceNutanixVMAntiAffinityPoliciesV2() *schema.Resource {
+	return &schema.Resource{
+		ReadContext: DatasourceNutanixVMAntiAffinityPoliciesV2Read,
+		Schema: map[string]*schema.Schema{
+			"page": {
+				Type:     schema.TypeInt,
+				Optional: true,
+			},
+			"limit": {
+				Type:     schema.TypeInt,
+				Optional: true,
+			},
+			"filter": {
+				Type:     schema.TypeString,
+				Optional: true,
+			},
+			"order_by": {
+				Type:     schema.TypeString,
+				Optional: true,
+			},
+			"policies": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"ext_id": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"name": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"description": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"create_time": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"update_time": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+						"created_by": {
+							Type:     schema.TypeMap,
+							Computed: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+						"updated_by": {
+							Type:     schema.TypeMap,
+							Computed: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+						"categories": {
+							Type:     schema.TypeSet,
+							Computed: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func DatasourceNutanixVMAntiAffinityPoliciesV2Read(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*conns.Client).VmmAPI
+
+	var filter, orderBy *string
+	var page, limit *int
+
+	if pagef, ok := d.GetOk("page"); ok {
+		page = utils.IntPtr(pagef.(int))
+	} else {
+		page = nil
+	}
+	if limitf, ok := d.GetOk("limit"); ok {
+		limit = utils.IntPtr(limitf.(int))
+	} else {
+		limit = nil
+	}
+	if filterf, ok := d.GetOk("filter"); ok {
+		filter = utils.StringPtr(filterf.(string))
+	} else {
+		filter = nil
+	}
+	if order, ok := d.GetOk("order_by"); ok {
+		orderBy = utils.StringPtr(order.(string))
+	} else {
+		orderBy = nil
+	}
+
+	resp, err := conn.VMAntiAffinityPolicyAPIInstance.ListVmAntiAffinityPolicies(page, limit, filter, orderBy)
+
+	if err != nil {
+		return diag.Errorf("error while fetching policies : %v", err)
+	}
+
+	if resp.Data == nil {
+		if err := d.Set("policies", make([]interface{}, 0)); err != nil {
+			return diag.FromErr(err)
+		}
+
+		d.SetId(utils.GenUUID())
+
+		return diag.Diagnostics{{
+			Severity: diag.Warning,
+			Summary:  "🫙 No data found.",
+			Detail:   "The API returned an empty list of policies.",
+		}}
+	}
+
+	getResp := resp.Data.GetValue().([]policies.VmAntiAffinityPolicy)
+
+	if err := d.Set("policies", flattenVMAntiAffinityPolicyEntities(getResp)); err != nil {
+		return diag.FromErr(err)
+	}
+
+	d.SetId(resource.UniqueId())
+
+	return nil
+}
+
+func flattenVMAntiAffinityPolicyEntities(pr []policies.VmAntiAffinityPolicy) []interface{} {
+	if len(pr) > 0 {
+		policies := make([]interface{}, len(pr))
+
+		for k, v := range pr {
+			policy := make(map[string]interface{})
+
+			if v.ExtId != nil {
+				policy["ext_id"] = v.ExtId
+			}
+			if v.Name != nil {
+				policy["name"] = v.Name
+			}
+			if v.Description != nil {
+				policy["description"] = v.Description
+			}
+			if v.CreateTime != nil {
+				policy["create_time"] = v.CreateTime.String()
+			}
+			if v.UpdateTime != nil {
+				policy["update_time"] = v.UpdateTime.String()
+			}
+			if v.CreatedBy != nil {
+				if v.CreatedBy.ExtId != nil {
+					policy["created_by"] = map[string]string{
+						"ext_id": *v.CreatedBy.ExtId,
+					}
+				}
+			}
+			if v.UpdatedBy != nil {
+				if v.UpdatedBy.ExtId != nil {
+					policy["updated_by"] = map[string]string{
+						"ext_id": *v.UpdatedBy.ExtId,
+					}
+				}
+			}
+			if v.Categories != nil {
+				policy["categories"] = flattenPolicyCategoryReference(v.Categories)
+			}
+			policies[k] = policy
+		}
+		return policies
+	}
+	return nil
+}