data_source_datadog_synthetics_locations.go

package datadog

import (
	"context"

	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
)

func dataSourceDatadogSyntheticsLocations() *schema.Resource {
	return &schema.Resource{
		Description: "Use this data source to retrieve Datadog's Synthetics Locations (to be used in Synthetics tests).",
		ReadContext: dataSourceDatadogSyntheticsLocationsRead,

		SchemaFunc: func() map[string]*schema.Schema {
			// Locations are a map of IDs to names
			return map[string]*schema.Schema{
				"locations": {
					Description: "A map of available Synthetics location IDs to names for Synthetics tests.",
					Type:        schema.TypeMap,
					Computed:    true,
				},
			}
		},
	}
}

func dataSourceDatadogSyntheticsLocationsRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {

	providerConf := meta.(*ProviderConfiguration)
	apiInstances := providerConf.DatadogApiInstances
	auth := providerConf.Auth

	syntheticsLocations, _, err := apiInstances.GetSyntheticsApiV1().ListLocations(auth)
	if err != nil {
		return diag.FromErr(err)
	}

	locationsMap := make(map[string]string)
	for _, location := range syntheticsLocations.GetLocations() {
		locationsMap[location.GetId()] = location.GetName()
	}

	if len(locationsMap) > 0 {
		d.SetId("datadog-synthetics-location")
	}
	d.Set("locations", locationsMap)

	return nil
}