Skip to content

Commit

Permalink
feat: Add netbox_virtualization_cluster_group resource
Browse files Browse the repository at this point in the history
  • Loading branch information
amhn committed Nov 18, 2022
1 parent 87fbc92 commit c2782ce
Show file tree
Hide file tree
Showing 4 changed files with 469 additions and 0 deletions.
82 changes: 82 additions & 0 deletions examples/resources/netbox_virtualization_cluster_group/resource.tf
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
resource "netbox_virtualization_cluster_type" "test" {
name = "test-{{ .namesuffix }}"
slug = "test-{{ .namesuffix }}"
description = "Test device role"

tag {
name = "tag1"
slug = "tag1"
}

custom_field {
name = "cf_boolean"
type = "boolean"
value = "true"
}

custom_field {
name = "cf_date"
type = "date"
value = "2020-12-25"
}

custom_field {
name = "cf_text"
type = "text"
value = "some text"
}

custom_field {
name = "cf_integer"
type = "integer"
value = "10"
}

custom_field {
name = "cf_selection"
type = "select"
value = "1"
}

custom_field {
name = "cf_url"
type = "url"
value = "https://github.com"
}

custom_field {
name = "cf_multi_selection"
type = "multiselect"
value = jsonencode([
"0",
"1"
])
}

custom_field {
name = "cf_json"
type = "json"
value = jsonencode({
stringvalue = "string"
boolvalue = false
dictionary = {
numbervalue = 5
}
})
}

custom_field {
name = "cf_object"
type = "object"
value = 1
}

custom_field {
name = "cf_multi_object"
type = "multiobject"
value = jsonencode([
1,
2
])
}
}
1 change: 1 addition & 0 deletions netbox/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,7 @@ func Provider() *schema.Provider {
"netbox_tenancy_contact_role": tenancy.ResourceNetboxTenancyContactRole(),
"netbox_tenancy_tenant": tenancy.ResourceNetboxTenancyTenant(),
"netbox_tenancy_tenant_group": tenancy.ResourceNetboxTenancyTenantGroup(),
"netbox_virtualization_cluster_group": virtualization.ResourceNetboxVirtualizationClusterGroup(),
"netbox_virtualization_interface": virtualization.ResourceNetboxVirtualizationInterface(),
"netbox_virtualization_vm": virtualization.ResourceNetboxVirtualizationVM(),
"netbox_virtualization_vm_primary_ip": virtualization.ResourceNetboxVirtualizationVMPrimaryIP(),
Expand Down
265 changes: 265 additions & 0 deletions netbox/virtualization/resource_netbox_virtualization_cluster_group.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,265 @@
package virtualization

import (
"context"
"strconv"

"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
netboxclient "github.com/smutel/go-netbox/v3/netbox/client"
"github.com/smutel/go-netbox/v3/netbox/client/virtualization"
"github.com/smutel/go-netbox/v3/netbox/models"
"github.com/smutel/terraform-provider-netbox/v4/netbox/internal/customfield"
"github.com/smutel/terraform-provider-netbox/v4/netbox/internal/requestmodifier"
"github.com/smutel/terraform-provider-netbox/v4/netbox/internal/tag"
"github.com/smutel/terraform-provider-netbox/v4/netbox/internal/util"
)

func ResourceNetboxVirtualizationClusterGroup() *schema.Resource {
return &schema.Resource{
Description: "Manage a cluster group (virtualization module) within Netbox.",
CreateContext: resourceNetboxVirtualizationClusterGroupCreate,
ReadContext: resourceNetboxVirtualizationClusterGroupRead,
UpdateContext: resourceNetboxVirtualizationClusterGroupUpdate,
DeleteContext: resourceNetboxVirtualizationClusterGroupDelete,
Exists: resourceNetboxVirtualizationClusterGroupExists,
Importer: &schema.ResourceImporter{
StateContext: schema.ImportStatePassthroughContext,
},

Schema: map[string]*schema.Schema{
"content_type": {
Type: schema.TypeString,
Computed: true,
Description: "The content type of this cluster group (virtualization module).",
},
"cluster_count": {
Type: schema.TypeInt,
Computed: true,
Description: "The number of clusters in this cluster group (virtualization module).",
},
"created": {
Type: schema.TypeString,
Computed: true,
Description: "Date when this cluster group was created.",
},
"custom_field": &customfield.CustomFieldSchema,
"description": {
Type: schema.TypeString,
Optional: true,
ValidateFunc: validation.StringLenBetween(0, 100),
Description: "The description of this cluster group (virtualization module).",
},
"last_updated": {
Type: schema.TypeString,
Computed: true,
Description: "Date when this cluster group was last updated.",
},
"name": {
Type: schema.TypeString,
Required: true,
ValidateFunc: validation.StringLenBetween(1, 100),
Description: "The name of this cluster group (virtualization module).",
},
"slug": {
Type: schema.TypeString,
Required: true,
ValidateFunc: validation.StringLenBetween(1, 100),
Description: "The slug of this cluster group (virtualization module).",
},
"tag": &tag.TagSchema,
"url": {
Type: schema.TypeString,
Computed: true,
Description: "The link to this cluster group (virtualization module).",
},
},
}
}

var clusterGroupRequiredFields = []string{
"created",
"last_updated",
"name",
"slug",
"tags",
}

func resourceNetboxVirtualizationClusterGroupCreate(ctx context.Context, d *schema.ResourceData,
m interface{}) diag.Diagnostics {
client := m.(*netboxclient.NetBoxAPI)

resourceCustomFields := d.Get("custom_field").(*schema.Set).List()
customFields := customfield.ConvertCustomFieldsFromTerraformToAPI(nil, resourceCustomFields)
name := d.Get("name").(string)
slug := d.Get("slug").(string)
tags := d.Get("tag").(*schema.Set).List()

newResource := &models.ClusterGroup{
CustomFields: customFields,
Description: d.Get("description").(string),
Name: &name,
Slug: &slug,
Tags: tag.ConvertTagsToNestedTags(tags),
}

resource := virtualization.NewVirtualizationClusterGroupsCreateParams().WithData(newResource)

resourceCreated, err := client.Virtualization.VirtualizationClusterGroupsCreate(resource, nil)
if err != nil {
return diag.FromErr(err)
}

d.SetId(strconv.FormatInt(resourceCreated.Payload.ID, 10))

return resourceNetboxVirtualizationClusterGroupRead(ctx, d, m)
}

func resourceNetboxVirtualizationClusterGroupRead(ctx context.Context, d *schema.ResourceData,
m interface{}) diag.Diagnostics {
client := m.(*netboxclient.NetBoxAPI)

resourceID := d.Id()
params := virtualization.NewVirtualizationClusterGroupsListParams().WithID(&resourceID)
resources, err := client.Virtualization.VirtualizationClusterGroupsList(params, nil)
if err != nil {
return diag.FromErr(err)
}

if len(resources.Payload.Results) != 1 {
d.SetId("")
return nil
}

resource := resources.Payload.Results[0]

if err = d.Set("cluster_count", resource.ClusterCount); err != nil {
return diag.FromErr(err)
}
if err = d.Set("content_type", util.ConvertURIContentType(resource.URL)); err != nil {
return diag.FromErr(err)
}
if err = d.Set("created", resource.Created.String()); err != nil {
return diag.FromErr(err)
}
resourceCustomFields := d.Get("custom_field").(*schema.Set).List()
customFields := customfield.UpdateCustomFieldsFromAPI(resourceCustomFields, resource.CustomFields)

if err = d.Set("custom_field", customFields); err != nil {
return diag.FromErr(err)
}
if err = d.Set("description", resource.Description); err != nil {
return diag.FromErr(err)
}
if err = d.Set("last_updated", resource.LastUpdated.String()); err != nil {
return diag.FromErr(err)
}
if err = d.Set("name", resource.Name); err != nil {
return diag.FromErr(err)
}
if err = d.Set("slug", resource.Slug); err != nil {
return diag.FromErr(err)
}
if err = d.Set("tag", tag.ConvertNestedTagsToTags(resource.Tags)); err != nil {
return diag.FromErr(err)
}
if err = d.Set("url", resource.URL); err != nil {
return diag.FromErr(err)
}

return nil
}

func resourceNetboxVirtualizationClusterGroupUpdate(ctx context.Context, d *schema.ResourceData,
m interface{}) diag.Diagnostics {
client := m.(*netboxclient.NetBoxAPI)
modifiedFields := make(map[string]interface{})

resourceID, err := strconv.ParseInt(d.Id(), 10, 64)
if err != nil {
return diag.Errorf("Unable to convert ID into int64")
}
params := &models.ClusterGroup{}

if d.HasChange("custom_field") {
stateCustomFields, resourceCustomFields := d.GetChange("custom_field")
customFields := customfield.ConvertCustomFieldsFromTerraformToAPI(stateCustomFields.(*schema.Set).List(), resourceCustomFields.(*schema.Set).List())
params.CustomFields = &customFields
}
if d.HasChange("description") {
description := d.Get("description").(string)
params.Description = description
modifiedFields["description"] = description
}
if d.HasChange("name") {
name := d.Get("name").(string)
params.Name = &name
}
if d.HasChange("slug") {
slug := d.Get("slug").(string)
params.Slug = &slug
}
if d.HasChange("tag") {
tags := d.Get("tag").(*schema.Set).List()
params.Tags = tag.ConvertTagsToNestedTags(tags)
}

resource := virtualization.NewVirtualizationClusterGroupsPartialUpdateParams().WithData(params)

resource.SetID(resourceID)

_, err = client.Virtualization.VirtualizationClusterGroupsPartialUpdate(resource, nil, requestmodifier.NewNetboxRequestModifier(modifiedFields, clusterGroupRequiredFields))
if err != nil {
return diag.FromErr(err)
}

return resourceNetboxVirtualizationClusterGroupRead(ctx, d, m)
}

func resourceNetboxVirtualizationClusterGroupDelete(ctx context.Context, d *schema.ResourceData,
m interface{}) diag.Diagnostics {
client := m.(*netboxclient.NetBoxAPI)

resourceExists, err := resourceNetboxVirtualizationClusterGroupExists(d, m)
if err != nil {
return diag.FromErr(err)
}

if !resourceExists {
return nil
}

id, err := strconv.ParseInt(d.Id(), 10, 64)
if err != nil {
return diag.Errorf("Unable to convert ID into int64")
}

resource := virtualization.NewVirtualizationClusterGroupsDeleteParams().WithID(id)
if _, err := client.Virtualization.VirtualizationClusterGroupsDelete(resource, nil); err != nil {
return diag.FromErr(err)
}

return nil
}

func resourceNetboxVirtualizationClusterGroupExists(d *schema.ResourceData,
m interface{}) (b bool, e error) {
client := m.(*netboxclient.NetBoxAPI)
resourceExist := false

resourceID := d.Id()
params := virtualization.NewVirtualizationClusterGroupsListParams().WithID(&resourceID)
resources, err := client.Virtualization.VirtualizationClusterGroupsList(params, nil)
if err != nil {
return resourceExist, err
}

for _, resource := range resources.Payload.Results {
if strconv.FormatInt(resource.ID, 10) == d.Id() {
resourceExist = true
}
}

return resourceExist, nil
}
Loading

0 comments on commit c2782ce

Please sign in to comment.