Implement 'data_protected_from_deletion'. Fixes #97

Signed-off-by: Andy Lo-A-Foe <andy.loafoe@gmail.com>

Author: loafoe

docs/data-sources/cdl_research_study.md

@@ -35,4 +35,5 @@ In addition to all arguments above, the following attributes are exported:
 * `monitors` - The list of IAM users who have role MONITOR
 * `study_managers` - The list of IAM users who have role STUDYMANAGER
 * `research_scientists` - The list of IAM users who have role RESEARCHSCIENTIST
+* `data_protected_from_deletion` - Is data protected from deletion
 

docs/resources/cdl_research_study.md

@@ -41,6 +41,8 @@ resource "hsdp_cdl_research_study" "study_a" {
     user_id = data.hsdp_iam_user.study_manager.id
     email   = data.hsdp_iam_user.study_manager.email_address
   }
+
+  data_protected_from_deletion = false
 } 
 ```
 
@@ -69,6 +71,7 @@ The following arguments are supported:
     * `user_id` (Required) The IAM user ID of the study manager
     * `email` - (Required) The email address for this study manager (for display purposes)
     * `institute_id` - (Optional) The institute ID associated with this role
+* `data_protected_from_deletion` (Optional) Protects data from being deleted. Default is `false`
 
 
 ## Attributes Reference

hsdp/data_source_cdl_research_study.go

@@ -58,6 +58,10 @@ func dataSourceCDLResearchStudy() *schema.Resource {
 				Computed: true,
 				Elem:     &schema.Schema{Type: schema.TypeString},
 			},
+			"data_protected_from_deletion": {
+				Type:     schema.TypeBool,
+				Computed: true,
+			},
 		},
 	}
 
@@ -125,6 +129,7 @@ func dataSourceCDLResearchStudyRead(_ context.Context, d *schema.ResourceData, m
 	_ = d.Set("data_scientists", dataScientists)
 	_ = d.Set("uploaders", uploaders)
 	_ = d.Set("study_managers", studyManagers)
+	_ = d.Set("data_protected_from_deletion", study.DataProtectedFromDeletion)
 
 	return diags
 }

hsdp/resource_cdl_research_study.go

@@ -71,6 +71,11 @@ func resourceCDLResearchStudy() *schema.Resource {
 			"data_scientist": permissionSchema(),
 			"monitor":        permissionSchema(),
 			"uploader":       permissionSchema(),
+			"data_protected_from_deletion": {
+				Type:     schema.TypeBool,
+				Optional: true,
+				Default:  false,
+			},
 		},
 	}
 }
@@ -145,14 +150,16 @@ func resourceCDLResearchStudyCreate(ctx context.Context, d *schema.ResourceData,
 	description := d.Get("description").(string)
 	endsAt := d.Get("ends_at").(string)
 	studyOwner := d.Get("study_owner").(string)
+	dataProtectedFromDeletion := d.Get("data_protected_from_deletion").(bool)
 
 	createdStudy, resp, err := client.Study.CreateStudy(cdl.Study{
 		Title:       title,
 		Description: description,
 		Period: cdl.Period{
 			End: endsAt,
 		},
-		StudyOwner: studyOwner,
+		StudyOwner:                studyOwner,
+		DataProtectedFromDeletion: dataProtectedFromDeletion,
 	})
 	if err != nil {
 		if resp == nil {
@@ -169,6 +176,14 @@ func resourceCDLResearchStudyCreate(ctx context.Context, d *schema.ResourceData,
 		matchFound := false
 		for _, study := range studies {
 			if study.Title == title && study.StudyOwner == studyOwner { // TODO: check if this is sufficient for a good match
+				if study.DataProtectedFromDeletion != dataProtectedFromDeletion ||
+					study.Description != description ||
+					study.Period.End != endsAt { // Update if needed
+					study.DataProtectedFromDeletion = dataProtectedFromDeletion
+					study.Description = description
+					study.Period.End = endsAt
+					_, _, _ = client.Study.UpdateStudy(study)
+				}
 				d.SetId(study.ID)
 				matchFound = true
 				break
@@ -266,11 +281,12 @@ func resourceCDLResearchStudyUpdate(_ context.Context, d *schema.ResourceData, m
 	}
 
 	if d.HasChange("title") || d.HasChange("description") ||
-		d.HasChange("ends_at") || d.HasChange("study_owner") {
+		d.HasChange("ends_at") || d.HasChange("study_owner") || d.HasChange("data_protected_from_deletion") {
 		study.Title = d.Get("title").(string)
 		study.Description = d.Get("description").(string)
 		study.Period.End = d.Get("ends_at").(string)
 		study.StudyOwner = d.Get("study_owner").(string)
+		study.DataProtectedFromDeletion = d.Get("data_protected_from_deletion").(bool)
 
 		_, _, err = client.Study.UpdateStudy(*study)
 		if err != nil {