diff --git a/examples/data-sources/netbox_dcim_location/data-source.tf b/examples/data-sources/netbox_dcim_location/data-source.tf new file mode 100644 index 000000000..293b8b3fe --- /dev/null +++ b/examples/data-sources/netbox_dcim_location/data-source.tf @@ -0,0 +1,3 @@ +data "netbox_dcim_location" "location_test" { + slug = "TestLocation" +} diff --git a/netbox/dcim/data_netbox_dcim_location.go b/netbox/dcim/data_netbox_dcim_location.go new file mode 100644 index 000000000..982276834 --- /dev/null +++ b/netbox/dcim/data_netbox_dcim_location.go @@ -0,0 +1,66 @@ +package dcim + +import ( + "context" + "regexp" + "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/dcim" + "github.com/smutel/terraform-provider-netbox/v6/netbox/internal/util" +) + +func DataNetboxDcimLocation() *schema.Resource { + return &schema.Resource{ + Description: "Get info about location (dcim module) from netbox.", + ReadContext: dataNetboxDcimLocationRead, + + Schema: map[string]*schema.Schema{ + "content_type": { + Type: schema.TypeString, + Computed: true, + Description: "The content type of this location (dcim module).", + }, + "slug": { + Type: schema.TypeString, + Required: true, + ValidateFunc: validation.StringMatch( + regexp.MustCompile("^[-a-zA-Z0-9_]{1,50}$"), + "Must be like ^[-a-zA-Z0-9_]{1,50}$"), + Description: "The slug of the location (dcim module).", + }, + }, + } +} + +func dataNetboxDcimLocationRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics { + client := m.(*netboxclient.NetBoxAPI) + + slug := d.Get("slug").(string) + + p := dcim.NewDcimLocationsListParams().WithSlug(&slug) + + list, err := client.Dcim.DcimLocationsList(p, nil) + if err != nil { + return diag.FromErr(err) + } + + if *list.Payload.Count < 1 { + return diag.Errorf("Your query returned no results. " + + "Please change your search criteria and try again.") + } else if *list.Payload.Count > 1 { + return diag.Errorf("Your query returned more than one result. " + + "Please try a more specific search criteria.") + } + + r := list.Payload.Results[0] + d.SetId(strconv.FormatInt(r.ID, 10)) + if err = d.Set("content_type", util.ConvertURIContentType(r.URL)); err != nil { + return diag.FromErr(err) + } + + return nil +} diff --git a/netbox/provider.go b/netbox/provider.go index ba89048bd..3a984b205 100644 --- a/netbox/provider.go +++ b/netbox/provider.go @@ -149,6 +149,7 @@ func Provider() *schema.Provider { "netbox_json_wireless_wireless_lan_groups_list": json.DataNetboxJSONWirelessWirelessLanGroupsList(), "netbox_json_wireless_wireless_lans_list": json.DataNetboxJSONWirelessWirelessLansList(), "netbox_json_wireless_wireless_links_list": json.DataNetboxJSONWirelessWirelessLinksList(), + "netbox_dcim_location": dcim.DataNetboxDcimLocation(), "netbox_dcim_platform": dcim.DataNetboxDcimPlatform(), "netbox_dcim_rack": dcim.DataNetboxDcimRack(), "netbox_dcim_rack_role": dcim.DataNetboxDcimRackRole(),