Merge pull request #19554 from DrFaust92/r/budget_action

r/budgets_budget_action - new resource

.changelog/19554.txt

@@ -0,0 +1,3 @@
+```release-note:new-resource
+aws_budgets_budget_action
+```

aws/internal/service/budgets/finder/finder.go

@@ -0,0 +1,27 @@
+package finder
+
+import (
+ "github.com/aws/aws-sdk-go/aws"
+ "github.com/aws/aws-sdk-go/service/budgets"
+ tfbudgets "github.com/terraform-providers/terraform-provider-aws/aws/internal/service/budgets"
+)
+
+func ActionById(conn *budgets.Budgets, id string) (*budgets.DescribeBudgetActionOutput, error) {
+ accountID, actionID, budgetName, err := tfbudgets.DecodeBudgetsBudgetActionID(id)
+ if err != nil {
+ return nil, err
+ }
+
+ input := &budgets.DescribeBudgetActionInput{
+ BudgetName: aws.String(budgetName),
+ AccountId: aws.String(accountID),
+ ActionId: aws.String(actionID),
+ }
+
+ out, err := conn.DescribeBudgetAction(input)
+ if err != nil {
+ return nil, err
+ }
+
+ return out, nil
+}

aws/internal/service/budgets/id.go

@@ -0,0 +1,14 @@
+package glue
+
+import (
+ "fmt"
+ "strings"
+)
+
+func DecodeBudgetsBudgetActionID(id string) (string, string, string, error) {
+ parts := strings.Split(id, ":")
+ if len(parts) != 3 {
+ return "", "", "", fmt.Errorf("Unexpected format of ID (%q), expected AccountID:ActionID:BudgetName", id)
+ }
+ return parts[0], parts[1], parts[2], nil
+}

aws/internal/service/budgets/waiter/status.go

@@ -0,0 +1,24 @@
+package waiter
+
+import (
+ "github.com/aws/aws-sdk-go/aws"
+ "github.com/aws/aws-sdk-go/service/budgets"
+ "github.com/hashicorp/aws-sdk-go-base/tfawserr"
+ "github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+ "github.com/terraform-providers/terraform-provider-aws/aws/internal/service/budgets/finder"
+)
+
+func ActionStatus(conn *budgets.Budgets, id string) resource.StateRefreshFunc {
+ return func() (interface{}, string, error) {
+ out, err := finder.ActionById(conn, id)
+ if err != nil {
+ if tfawserr.ErrCodeEquals(err, budgets.ErrCodeNotFoundException) {
+ return nil, "", nil
+ }
+ return nil, "", err
+ }
+
+ action := out.Action
+ return action, aws.StringValue(action.Status), err
+ }
+}

aws/internal/service/budgets/waiter/waiter.go

@@ -0,0 +1,36 @@
+package waiter
+
+import (
+ "time"
+
+ "github.com/aws/aws-sdk-go/service/budgets"
+ "github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+const (
+ ActionAvailableTimeout = 2 * time.Minute
+)
+
+func ActionAvailable(conn *budgets.Budgets, id string) (*budgets.Action, error) {
+ stateConf := &resource.StateChangeConf{
+ Pending: []string{
+ budgets.ActionStatusExecutionInProgress,
+ budgets.ActionStatusStandby,
+ },
+ Target: []string{
+ budgets.ActionStatusExecutionSuccess,
+ budgets.ActionStatusExecutionFailure,
+ budgets.ActionStatusPending,
+ },
+ Refresh: ActionStatus(conn, id),
+ Timeout: ActionAvailableTimeout,
+ }
+
+ outputRaw, err := stateConf.WaitForState()
+
+ if v, ok := outputRaw.(*budgets.Action); ok {
+ return v, err
+ }
+
+ return nil, err
+}

aws/provider.go

@@ -534,6 +534,7 @@ func Provider() *schema.Provider {
  "aws_backup_vault_notifications": resourceAwsBackupVaultNotifications(),
  "aws_backup_vault_policy": resourceAwsBackupVaultPolicy(),
  "aws_budgets_budget": resourceAwsBudgetsBudget(),
+ "aws_budgets_budget_action": resourceAwsBudgetsBudgetAction(),
  "aws_cloud9_environment_ec2": resourceAwsCloud9EnvironmentEc2(),
  "aws_cloudformation_stack": resourceAwsCloudFormationStack(),
  "aws_cloudformation_stack_set": resourceAwsCloudFormationStackSet(),