New Resource: azure_consumption_budget_*

.teamcity/components/generated/services.kt

@@ -18,6 +18,7 @@ var services = mapOf(
         "cognitive" to "Cognitive Services",
         "communication" to "Communication",
         "compute" to "Compute",
+        "consumption" to "Consumption",
         "containers" to "Container Services",
         "cosmos" to "CosmosDB",
         "costmanagement" to "Cost Management",

azurerm/internal/clients/client.go

@@ -23,6 +23,7 @@ import (
 	cognitiveServices "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cognitive/client"
 	communication "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/communication/client"
 	compute "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/compute/client"
+	consumption "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/consumption/client"
 	containerServices "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/containers/client"
 	cosmosdb "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cosmos/client"
 	costmanagement "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/costmanagement/client"
@@ -123,6 +124,7 @@ type Client struct {
 	Cognitive             *cognitiveServices.Client
 	Communication         *communication.Client
 	Compute               *compute.Client
+	Consumption           *consumption.Client
 	Containers            *containerServices.Client
 	Cosmos                *cosmosdb.Client
 	CostManagement        *costmanagement.Client
@@ -225,6 +227,7 @@ func (client *Client) Build(ctx context.Context, o *common.ClientOptions) error
 	client.Cognitive = cognitiveServices.NewClient(o)
 	client.Communication = communication.NewClient(o)
 	client.Compute = compute.NewClient(o)
+	client.Consumption = consumption.NewClient(o)
 	client.Containers = containerServices.NewClient(o)
 	client.Cosmos = cosmosdb.NewClient(o)
 	client.CostManagement = costmanagement.NewClient(o)

azurerm/internal/provider/services.go

@@ -19,6 +19,7 @@ import (
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cognitive"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/communication"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/compute"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/consumption"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/containers"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/cosmos"
 	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/costmanagement"
@@ -127,6 +128,7 @@ func SupportedUntypedServices() []sdk.UntypedServiceRegistration {
 		communication.Registration{},
 		compute.Registration{},
 		containers.Registration{},
+		consumption.Registration{},
 		cosmos.Registration{},
 		costmanagement.Registration{},
 		customproviders.Registration{},

azurerm/internal/services/consumption/client/client.go

@@ -0,0 +1,19 @@
+package client
+
+import (
+	"github.com/Azure/azure-sdk-for-go/services/consumption/mgmt/2019-10-01/consumption"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/common"
+)
+
+type Client struct {
+	BudgetsClient *consumption.BudgetsClient
+}
+
+func NewClient(o *common.ClientOptions) *Client {
+	budgetsClient := consumption.NewBudgetsClientWithBaseURI(o.ResourceManagerEndpoint, o.SubscriptionId)
+	o.ConfigureClient(&budgetsClient.Client, o.ResourceManagerAuthorizer)
+
+	return &Client{
+		BudgetsClient: &budgetsClient,
+	}
+}

azurerm/internal/services/consumption/consumption_budget_common.go

@@ -0,0 +1,110 @@
+package consumption
+
+import (
+	"fmt"
+
+	"github.com/Azure/azure-sdk-for-go/services/consumption/mgmt/2019-10-01/consumption"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+	"github.com/shopspring/decimal"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/helpers/tf"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/clients"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/timeouts"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/utils"
+)
+
+func resourceArmConsumptionBudgetRead(d *schema.ResourceData, meta interface{}, scope, name string) error {
+	client := meta.(*clients.Client).Consumption.BudgetsClient
+	ctx, cancel := timeouts.ForCreateUpdate(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	resp, err := client.Get(ctx, scope, name)
+	if err != nil {
+		if utils.ResponseWasNotFound(resp.Response) {
+			d.SetId("")
+			return nil
+		}
+		return fmt.Errorf("error making read request on Azure Consumption Budget %q for scope %q: %+v", name, scope, err)
+	}
+
+	d.Set("name", resp.Name)
+	if resp.Amount != nil {
+		amount, _ := resp.Amount.Float64()
+		d.Set("amount", amount)
+	}
+	d.Set("time_grain", string(resp.TimeGrain))
+	d.Set("time_period", FlattenConsumptionBudgetTimePeriod(resp.TimePeriod))
+	d.Set("notification", schema.NewSet(schema.HashResource(SchemaConsumptionBudgetNotificationElement()), FlattenConsumptionBudgetNotifications(resp.Notifications)))
+	d.Set("filter", FlattenConsumptionBudgetFilter(resp.Filter))
+
+	return nil
+}
+
+func resourceArmConsumptionBudgetDelete(d *schema.ResourceData, meta interface{}, scope string) error {
+	client := meta.(*clients.Client).Consumption.BudgetsClient
+	ctx, cancel := timeouts.ForDelete(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	name := d.Get("name").(string)
+	resp, err := client.Delete(ctx, scope, name)
+
+	if err != nil {
+		if !utils.ResponseWasNotFound(resp) {
+			return fmt.Errorf("error issuing delete request on Azure Consumption Budget %q for scope %q: %+v", name, scope, err)
+		}
+	}
+
+	return nil
+}
+
+func resourceArmConsumptionBudgetCreateUpdate(d *schema.ResourceData, meta interface{}, resourceName, scope string) error {
+	client := meta.(*clients.Client).Consumption.BudgetsClient
+	ctx, cancel := timeouts.ForCreateUpdate(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	name := d.Get("name").(string)
+
+	if d.IsNewResource() {
+		existing, err := client.Get(ctx, scope, name)
+		if err != nil {
+			if !utils.ResponseWasNotFound(existing.Response) {
+				return fmt.Errorf("error checking for presence of existing Consumption Budget %q for scope %q: %s", name, scope, err)
+			}
+		}
+
+		if existing.ID != nil && *existing.ID != "" {
+			return tf.ImportAsExistsError(resourceName, *existing.ID)
+		}
+	}
+
+	amount := decimal.NewFromFloat(d.Get("amount").(float64))
+	timePeriod, err := ExpandConsumptionBudgetTimePeriod(d.Get("time_period").([]interface{}))
+	if err != nil {
+		return fmt.Errorf("error expanding `time_period`: %+v", err)
+	}
+
+	// The Consumption Budget API requires the category type field to be set in a budget's properties.
+	// 'Cost' is the only valid Budget type today according to the API spec.
+	category := "Cost"
+	parameters := consumption.Budget{
+		Name: utils.String(name),
+		BudgetProperties: &consumption.BudgetProperties{
+			Amount:        &amount,
+			Category:      &category,
+			Filter:        ExpandConsumptionBudgetFilter(d.Get("filter").([]interface{})),
+			Notifications: ExpandConsumptionBudgetNotifications(d.Get("notification").(*schema.Set).List()),
+			TimeGrain:     consumption.TimeGrainType(d.Get("time_grain").(string)),
+			TimePeriod:    timePeriod,
+		},
+	}
+
+	read, err := client.CreateOrUpdate(ctx, scope, name, parameters)
+	if err != nil {
+		return err
+	}
+
+	if read.ID == nil {
+		return fmt.Errorf("cannot read Azure Consumption Budget %q for scope %q", name, scope)
+	}
+
+	return nil
+}

azurerm/internal/services/consumption/consumption_budget_resource_group_resource.go

@@ -0,0 +1,79 @@
+package consumption
+
+import (
+	"time"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/consumption/parse"
+	resourceParse "github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/services/resource/parse"
+	"github.com/terraform-providers/terraform-provider-azurerm/azurerm/internal/tf/pluginsdk"
+)
+
+func resourceArmConsumptionBudgetResourceGroup() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceArmConsumptionBudgetResourceGroupCreateUpdate,
+		Read:   resourceArmConsumptionBudgetResourceGroupRead,
+		Update: resourceArmConsumptionBudgetResourceGroupCreateUpdate,
+		Delete: resourceArmConsumptionBudgetResourceGroupDelete,
+		Importer: pluginsdk.ImporterValidatingResourceId(func(id string) error {
+			_, err := parse.ConsumptionBudgetResourceGroupID(id)
+			return err
+		}),
+
+		Timeouts: &schema.ResourceTimeout{
+			Create: schema.DefaultTimeout(30 * time.Minute),
+			Read:   schema.DefaultTimeout(5 * time.Minute),
+			Update: schema.DefaultTimeout(30 * time.Minute),
+			Delete: schema.DefaultTimeout(30 * time.Minute),
+		},
+
+		Schema: SchemaConsumptionBudgetResourceGroupResource(),
+	}
+}
+
+func resourceArmConsumptionBudgetResourceGroupCreateUpdate(d *schema.ResourceData, meta interface{}) error {
+	name := d.Get("name").(string)
+	resourceGroupId, err := resourceParse.ResourceGroupID(d.Get("resource_group_id").(string))
+	if err != nil {
+		return err
+	}
+
+	err = resourceArmConsumptionBudgetCreateUpdate(d, meta, consumptionBudgetResourceGroupName, resourceGroupId.ID())
+	if err != nil {
+		return err
+	}
+
+	d.SetId(parse.NewConsumptionBudgetResourceGroupID(resourceGroupId.SubscriptionId, resourceGroupId.ResourceGroup, name).ID())
+
+	return resourceArmConsumptionBudgetResourceGroupRead(d, meta)
+}
+
+func resourceArmConsumptionBudgetResourceGroupRead(d *schema.ResourceData, meta interface{}) error {
+	consumptionBudgetId, err := parse.ConsumptionBudgetResourceGroupID(d.Id())
+	if err != nil {
+		return err
+	}
+
+	resourceGroupId := resourceParse.NewResourceGroupID(consumptionBudgetId.SubscriptionId, consumptionBudgetId.ResourceGroup)
+
+	err = resourceArmConsumptionBudgetRead(d, meta, resourceGroupId.ID(), consumptionBudgetId.BudgetName)
+	if err != nil {
+		return err
+	}
+
+	// The scope of a Resource Group consumption budget is the Resource Group ID
+	d.Set("resource_group_id", resourceGroupId.ID())
+
+	return nil
+}
+
+func resourceArmConsumptionBudgetResourceGroupDelete(d *schema.ResourceData, meta interface{}) error {
+	consumptionBudgetId, err := parse.ConsumptionBudgetResourceGroupID(d.Id())
+	if err != nil {
+		return err
+	}
+
+	resourceGroupId := resourceParse.NewResourceGroupID(consumptionBudgetId.SubscriptionId, consumptionBudgetId.ResourceGroup)
+
+	return resourceArmConsumptionBudgetDelete(d, meta, resourceGroupId.ID())
+}