support on-call rotations for ssmcontacts

Author: MacoTasu

internal/service/ssmcontacts/contact.go

@@ -15,6 +15,7 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	"github.com/hashicorp/terraform-provider-aws/internal/conns"
 	"github.com/hashicorp/terraform-provider-aws/internal/create"
+	"github.com/hashicorp/terraform-provider-aws/internal/flex"
 	tftags "github.com/hashicorp/terraform-provider-aws/internal/tags"
 	"github.com/hashicorp/terraform-provider-aws/internal/tfresource"
 	"github.com/hashicorp/terraform-provider-aws/names"
@@ -49,6 +50,13 @@ func ResourceContact() *schema.Resource {
 				Type:     schema.TypeString,
 				Optional: true,
 			},
+			"rotation_ids": {
+				Type:     schema.TypeList,
+				Optional: true,
+				Elem: &schema.Schema{
+					Type: schema.TypeString,
+				},
+			},
 			names.AttrType: {
 				Type:     schema.TypeString,
 				Required: true,
@@ -68,12 +76,23 @@ func resourceContactCreate(ctx context.Context, d *schema.ResourceData, meta any
 	var diags diag.Diagnostics
 	client := meta.(*conns.AWSClient).SSMContactsClient(ctx)
 
+	contactType := types.ContactType(d.Get(names.AttrType).(string))
+
 	input := &ssmcontacts.CreateContactInput{
 		Alias:       aws.String(d.Get(names.AttrAlias).(string)),
 		DisplayName: aws.String(d.Get(names.AttrDisplayName).(string)),
-		Plan:        &types.Plan{Stages: []types.Stage{}},
 		Tags:        getTagsIn(ctx),
-		Type:        types.ContactType(d.Get(names.AttrType).(string)),
+		Type:        contactType,
+	}
+
+	if contactType == types.ContactTypeOncallSchedule {
+		plan := &types.Plan{}
+		if v, ok := d.GetOk("rotation_ids"); ok {
+			plan.RotationIds = flex.ExpandStringValueList(v.([]any))
+		}
+		input.Plan = plan
+	} else {
+		input.Plan = &types.Plan{Stages: []types.Stage{}}
 	}
 
 	output, err := client.CreateContact(ctx, input)
@@ -117,10 +136,23 @@ func resourceContactUpdate(ctx context.Context, d *schema.ResourceData, meta any
 	var diags diag.Diagnostics
 	conn := meta.(*conns.AWSClient).SSMContactsClient(ctx)
 
-	if d.HasChanges(names.AttrDisplayName) {
+	if d.HasChanges(names.AttrDisplayName, "rotation_ids") {
+		contactType := types.ContactType(d.Get(names.AttrType).(string))
+
 		in := &ssmcontacts.UpdateContactInput{
-			ContactId:   aws.String(d.Id()),
-			DisplayName: aws.String(d.Get(names.AttrDisplayName).(string)),
+			ContactId: aws.String(d.Id()),
+		}
+
+		if d.HasChange(names.AttrDisplayName) {
+			in.DisplayName = aws.String(d.Get(names.AttrDisplayName).(string))
+		}
+
+		if d.HasChange("rotation_ids") && contactType == types.ContactTypeOncallSchedule {
+			plan := &types.Plan{}
+			if v, ok := d.GetOk("rotation_ids"); ok {
+				plan.RotationIds = flex.ExpandStringValueList(v.([]any))
+			}
+			in.Plan = plan
 		}
 
 		_, err := conn.UpdateContact(ctx, in)

internal/service/ssmcontacts/contact_test.go

@@ -241,6 +241,58 @@ func testAccContact_updateDisplayName(t *testing.T) {
 	})
 }
 
+func testAccContact_oncallSchedule(t *testing.T) {
+	if testing.Short() {
+		t.Skip("skipping long-running test in short mode")
+	}
+
+	ctx := acctest.Context(t)
+	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
+	resourceName := "aws_ssmcontacts_contact.test"
+
+	resource.Test(t, resource.TestCase{
+		PreCheck: func() {
+			acctest.PreCheck(ctx, t)
+			testAccContactPreCheck(ctx, t)
+		},
+		ErrorCheck:               acctest.ErrorCheck(t, names.SSMContactsServiceID),
+		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
+		CheckDestroy:             testAccCheckContactDestroy(ctx),
+		Steps: []resource.TestStep{
+			{
+				Config: testAccContactConfig_oncallSchedule(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckContactExists(ctx, resourceName),
+					resource.TestCheckResourceAttr(resourceName, names.AttrAlias, rName),
+					resource.TestCheckResourceAttr(resourceName, names.AttrType, "ONCALL_SCHEDULE"),
+					resource.TestCheckResourceAttr(resourceName, "rotation_ids.#", "1"),
+					acctest.MatchResourceAttrRegionalARN(ctx, resourceName, names.AttrARN, "ssm-contacts", regexache.MustCompile(`contact/.+$`)),
+				),
+			},
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+			{
+				Config: testAccContactConfig_oncallScheduleUpdated(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckContactExists(ctx, resourceName),
+					resource.TestCheckResourceAttr(resourceName, names.AttrAlias, rName),
+					resource.TestCheckResourceAttr(resourceName, names.AttrType, "ONCALL_SCHEDULE"),
+					resource.TestCheckResourceAttr(resourceName, "rotation_ids.#", "2"),
+					acctest.MatchResourceAttrRegionalARN(ctx, resourceName, names.AttrARN, "ssm-contacts", regexache.MustCompile(`contact/.+$`)),
+				),
+			},
+			{
+				Config: testAccContactConfig_none(),
+				Check:  testAccCheckContactDestroy(ctx),
+			},
+		},
+	})
+}
+
+
 func testAccCheckContactDestroy(ctx context.Context) resource.TestCheckFunc {
 	return func(s *terraform.State) error {
 		conn := acctest.Provider.Meta().(*conns.AWSClient).SSMContactsClient(ctx)
@@ -383,3 +435,105 @@ resource "aws_ssmcontacts_contact" "contact_one" {
 }
 `, alias, displayName))
 }
+
+func testAccContactConfig_oncallSchedule(alias string) string {
+	return acctest.ConfigCompose(
+		testAccContactConfig_base(),
+		fmt.Sprintf(`
+resource "aws_ssmcontacts_contact" "test_contact" {
+  alias = "%[1]s-contact"
+  type  = "PERSONAL"
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+
+resource "aws_ssmcontacts_rotation" "test" {
+  contact_ids = [aws_ssmcontacts_contact.test_contact.arn]
+  name        = %[1]q
+  recurrence {
+    number_of_on_calls    = 1
+    recurrence_multiplier = 1
+    daily_settings {
+      hour_of_day    = 9
+      minute_of_hour = 0
+    }
+  }
+  time_zone_id = "America/Los_Angeles"
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+
+resource "aws_ssmcontacts_contact" "test" {
+  alias        = %[1]q
+  display_name = %[1]q
+  type         = "ONCALL_SCHEDULE"
+  rotation_ids = [aws_ssmcontacts_rotation.test.arn]
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+`, alias))
+}
+
+func testAccContactConfig_oncallScheduleUpdated(alias string) string {
+	return acctest.ConfigCompose(
+		testAccContactConfig_base(),
+		fmt.Sprintf(`
+resource "aws_ssmcontacts_contact" "test_contact" {
+  alias = "%[1]s-contact"
+  type  = "PERSONAL"
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+
+resource "aws_ssmcontacts_contact" "test_contact_2" {
+  alias = "%[1]s-contact-2"
+  type  = "PERSONAL"
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+
+resource "aws_ssmcontacts_rotation" "test" {
+  contact_ids = [aws_ssmcontacts_contact.test_contact.arn]
+  name        = %[1]q
+  recurrence {
+    number_of_on_calls    = 1
+    recurrence_multiplier = 1
+    daily_settings {
+      hour_of_day    = 9
+      minute_of_hour = 0
+    }
+  }
+  time_zone_id = "America/Los_Angeles"
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+
+resource "aws_ssmcontacts_rotation" "test_2" {
+  contact_ids = [aws_ssmcontacts_contact.test_contact_2.arn]
+  name        = "%[1]s-2"
+  recurrence {
+    number_of_on_calls    = 1
+    recurrence_multiplier = 1
+    daily_settings {
+      hour_of_day    = 14
+      minute_of_hour = 30
+    }
+  }
+  time_zone_id = "America/New_York"
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+
+resource "aws_ssmcontacts_contact" "test" {
+  alias        = %[1]q
+  display_name = %[1]q
+  type         = "ONCALL_SCHEDULE"
+  rotation_ids = [
+    aws_ssmcontacts_rotation.test.arn,
+    aws_ssmcontacts_rotation.test_2.arn
+  ]
+
+  depends_on = [aws_ssmincidents_replication_set.test]
+}
+`, alias))
+}

internal/service/ssmcontacts/helper.go

@@ -16,6 +16,9 @@ func setContactResourceData(d *schema.ResourceData, getContactOutput *ssmcontact
 	d.Set(names.AttrAlias, getContactOutput.Alias)
 	d.Set(names.AttrType, getContactOutput.Type)
 	d.Set(names.AttrDisplayName, getContactOutput.DisplayName)
+	if getContactOutput.Plan != nil {
+		d.Set("rotation_ids", getContactOutput.Plan.RotationIds)
+	}
 
 	return nil
 }

internal/service/ssmcontacts/ssmcontacts_test.go

@@ -22,6 +22,7 @@ func TestAccSSMContacts_serial(t *testing.T) {
 			"updateDisplayName":  testAccContact_updateDisplayName,
 			"tags":               testAccSSMContactsContact_tagsSerial,
 			"updateType":         testAccContact_updateType,
+			"oncallSchedule":     testAccContact_oncallSchedule,
 		},
 		"ContactDataSource": {
 			acctest.CtBasic: testAccContactDataSource_basic,

website/docs/d/ssmcontacts_contact.html.markdown

@@ -32,6 +32,7 @@ This data source supports the following arguments:
 This data source exports the following attributes in addition to the arguments above:
 
 * `alias` - A unique and identifiable alias of the contact or escalation plan.
-* `type` - The type of contact engaged. A single contact is type `PERSONAL` and an escalation plan is type `ESCALATION`.
+* `type` - The type of contact engaged. A single contact is type `PERSONAL`, an escalation plan is type `ESCALATION`, and an on-call schedule is type `ONCALL_SCHEDULE`.
 * `display_name` - Full friendly name of the contact or escalation plan.
+* `rotation_ids` - List of rotation IDs associated with the contact.
 * `tags` - Map of tags to assign to the resource.

website/docs/r/ssmcontacts_contact.html.markdown

@@ -41,18 +41,56 @@ resource "aws_ssmcontacts_contact" "example" {
 }
 ```
 
+### On-call Schedule Usage
+
+```terraform
+resource "aws_ssmcontacts_contact" "oncall_contact" {
+  alias = "oncall-contact"
+  type  = "PERSONAL"
+
+  depends_on = [aws_ssmincidents_replication_set.example]
+}
+
+resource "aws_ssmcontacts_rotation" "example" {
+  contact_ids = [aws_ssmcontacts_contact.oncall_contact.arn]
+  name        = "example-rotation"
+
+  recurrence {
+    number_of_on_calls    = 1
+    recurrence_multiplier = 1
+    daily_settings {
+      hour_of_day    = 9
+      minute_of_hour = 0
+    }
+  }
+
+  time_zone_id = "America/Los_Angeles"
+
+  depends_on = [aws_ssmincidents_replication_set.example]
+}
+
+resource "aws_ssmcontacts_contact" "example" {
+  alias        = "oncall-schedule"
+  display_name = "Example On-call Schedule"
+  type         = "ONCALL_SCHEDULE"
+  rotation_ids = [aws_ssmcontacts_rotation.example.arn]
+
+  depends_on = [aws_ssmincidents_replication_set.example]
+}
+```
+
 ## Argument Reference
 
 The following arguments are required:
 
 - `alias` - (Required) A unique and identifiable alias for the contact or escalation plan. Must be between 1 and 255 characters, and may contain alphanumerics, underscores (`_`), and hyphens (`-`).
-- `type` - (Required) The type of contact engaged. A single contact is type PERSONAL and an escalation
-  plan is type ESCALATION.
+- `type` - (Required) The type of contact engaged. A single contact is type `PERSONAL`, an escalation plan is type `ESCALATION`, and an on-call schedule is type `ONCALL_SCHEDULE`.
 
 The following arguments are optional:
 
 - `region` - (Optional) Region where this resource will be [managed](https://docs.aws.amazon.com/general/latest/gr/rande.html#regional-endpoints). Defaults to the Region set in the [provider configuration](https://registry.terraform.io/providers/hashicorp/aws/latest/docs#aws-configuration-reference).
 - `display_name` - (Optional) Full friendly name of the contact or escalation plan. If set, must be between 1 and 255 characters, and may contain alphanumerics, underscores (`_`), hyphens (`-`), periods (`.`), and spaces.
+- `rotation_ids` - (Optional) List of rotation IDs associated with the contact. Required when `type` is `ONCALL_SCHEDULE`.
 - `tags` - (Optional) Key-value tags for the monitor. If configured with a provider [`default_tags` configuration block](https://registry.terraform.io/providers/hashicorp/aws/latest/docs#default_tags-configuration-block) present, tags with matching keys will overwrite those defined at the provider-level.
 
 ## Attribute Reference