feat: workflow job template notifications (#3)

Tioborto · web-flow · commit 00db9157df52 · 2022-04-21T11:18:35.000+02:00
diff --git a/awx/provider.go b/awx/provider.go
@@ -37,36 +37,39 @@ func Provider() *schema.Provider {
 			},
 		},
 		ResourcesMap: map[string]*schema.Resource{
-			"awx_credential_azure_key_vault":                 resourceCredentialAzureKeyVault(),
-			"awx_credential_google_compute_engine":           resourceCredentialGoogleComputeEngine(),
-			"awx_credential_input_source":                    resourceCredentialInputSource(),
-			"awx_credential":                                 resourceCredential(),
-			"awx_credential_type":                            resourceCredentialType(),
-			"awx_credential_machine":                         resourceCredentialMachine(),
-			"awx_credential_scm":                             resourceCredentialSCM(),
-			"awx_execution_environment":                      resourceExecutionEnvironment(),
-			"awx_host":                                       resourceHost(),
-			"awx_inventory_group":                            resourceInventoryGroup(),
-			"awx_inventory_source":                           resourceInventorySource(),
-			"awx_inventory":                                  resourceInventory(),
-			"awx_job_template_credential":                    resourceJobTemplateCredentials(),
-			"awx_job_template":                               resourceJobTemplate(),
-			"awx_job_template_launch":                        resourceJobTemplateLaunch(),
-			"awx_job_template_notification_template_error":   resourceJobTemplateNotificationTemplateError(),
-			"awx_job_template_notification_template_started": resourceJobTemplateNotificationTemplateStarted(),
-			"awx_job_template_notification_template_success": resourceJobTemplateNotificationTemplateSuccess(),
-			"awx_notification_template":                      resourceNotificationTemplate(),
-			"awx_organization":                               resourceOrganization(),
-			"awx_project":                                    resourceProject(),
-			"awx_schedule":                                   resourceSchedule(),
-			"awx_settings_ldap_team_map":                     resourceSettingsLDAPTeamMap(),
-			"awx_team":                                       resourceTeam(),
-			"awx_workflow_job_template_node_always":          resourceWorkflowJobTemplateNodeAlways(),
-			"awx_workflow_job_template_node_failure":         resourceWorkflowJobTemplateNodeFailure(),
-			"awx_workflow_job_template_node_success":         resourceWorkflowJobTemplateNodeSuccess(),
-			"awx_workflow_job_template_node":                 resourceWorkflowJobTemplateNode(),
-			"awx_workflow_job_template":                      resourceWorkflowJobTemplate(),
-			"awx_workflow_job_template_schedule":             resourceWorkflowJobTemplateSchedule(),
+			"awx_credential_azure_key_vault":                          resourceCredentialAzureKeyVault(),
+			"awx_credential_google_compute_engine":                    resourceCredentialGoogleComputeEngine(),
+			"awx_credential_input_source":                             resourceCredentialInputSource(),
+			"awx_credential":                                          resourceCredential(),
+			"awx_credential_type":                                     resourceCredentialType(),
+			"awx_credential_machine":                                  resourceCredentialMachine(),
+			"awx_credential_scm":                                      resourceCredentialSCM(),
+			"awx_execution_environment":                               resourceExecutionEnvironment(),
+			"awx_host":                                                resourceHost(),
+			"awx_inventory_group":                                     resourceInventoryGroup(),
+			"awx_inventory_source":                                    resourceInventorySource(),
+			"awx_inventory":                                           resourceInventory(),
+			"awx_job_template_credential":                             resourceJobTemplateCredentials(),
+			"awx_job_template":                                        resourceJobTemplate(),
+			"awx_job_template_launch":                                 resourceJobTemplateLaunch(),
+			"awx_job_template_notification_template_error":            resourceJobTemplateNotificationTemplateError(),
+			"awx_job_template_notification_template_started":          resourceJobTemplateNotificationTemplateStarted(),
+			"awx_job_template_notification_template_success":          resourceJobTemplateNotificationTemplateSuccess(),
+			"awx_notification_template":                               resourceNotificationTemplate(),
+			"awx_organization":                                        resourceOrganization(),
+			"awx_project":                                             resourceProject(),
+			"awx_schedule":                                            resourceSchedule(),
+			"awx_settings_ldap_team_map":                              resourceSettingsLDAPTeamMap(),
+			"awx_team":                                                resourceTeam(),
+			"awx_workflow_job_template_node_always":                   resourceWorkflowJobTemplateNodeAlways(),
+			"awx_workflow_job_template_node_failure":                  resourceWorkflowJobTemplateNodeFailure(),
+			"awx_workflow_job_template_node_success":                  resourceWorkflowJobTemplateNodeSuccess(),
+			"awx_workflow_job_template_node":                          resourceWorkflowJobTemplateNode(),
+			"awx_workflow_job_template":                               resourceWorkflowJobTemplate(),
+			"awx_workflow_job_template_schedule":                      resourceWorkflowJobTemplateSchedule(),
+			"awx_workflow_job_template_notification_template_error":   resourceWorkflowJobTemplateNotificationTemplateError(),
+			"awx_workflow_job_template_notification_template_started": resourceWorkflowJobTemplateNotificationTemplateStarted(),
+			"awx_workflow_job_template_notification_template_success": resourceWorkflowJobTemplateNotificationTemplateSuccess(),
 		},
 		DataSourcesMap: map[string]*schema.Resource{
 			"awx_credential_azure_key_vault": dataSourceCredentialAzure(),
diff --git a/awx/resource_workflow_job_template_notification_template_error.go b/awx/resource_workflow_job_template_notification_template_error.go
@@ -0,0 +1,39 @@
+/*
+*TBD*
+
+Example Usage
+
+```hcl
+resource "awx_workflow_job_template_notification_template_error" "baseconfig" {
+  workflow_job_template_id   = awx_workflow_job_template.baseconfig.id
+  notification_template_id   = awx_notification_template.default.id
+}
+```
+
+*/
+package awx
+
+import (
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func resourceWorkflowJobTemplateNotificationTemplateError() *schema.Resource {
+	return &schema.Resource{
+		CreateContext: resourceWorkflowJobTemplateNotificationTemplateCreateForType("error"),
+		DeleteContext: resourceWorkflowJobTemplateNotificationTemplateDeleteForType("error"),
+		ReadContext:   resourceWorkflowJobTemplateNotificationTemplateRead,
+
+		Schema: map[string]*schema.Schema{
+			"workflow_job_template_id": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: true,
+			},
+			"notification_template_id": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: true,
+			},
+		},
+	}
+}
diff --git a/awx/resource_workflow_job_template_notification_template_shared.go b/awx/resource_workflow_job_template_notification_template_shared.go
@@ -0,0 +1,95 @@
+package awx
+
+import (
+	"context"
+	"strconv"
+
+	awx "github.com/denouche/goawx/client"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func getResourceWorkflowJobTemplateNotificationTemplateAssociateFuncForType(client *awx.WorkflowJobTemplateNotificationTemplatesService, typ string) func(workflowJobTemplateID int, notificationTemplateID int) (*awx.NotificationTemplate, error) {
+	switch typ {
+	case "error":
+		return client.AssociateWorkflowJobTemplateNotificationTemplatesError
+	case "success":
+		return client.AssociateWorkflowJobTemplateNotificationTemplatesSuccess
+	case "started":
+		return client.AssociateWorkflowJobTemplateNotificationTemplatesStarted
+	}
+	return nil
+}
+
+func getResourceWorkflowJobTemplateNotificationTemplateDisassociateFuncForType(client *awx.WorkflowJobTemplateNotificationTemplatesService, typ string) func(workflowJobTemplateID int, notificationTemplateID int) (*awx.NotificationTemplate, error) {
+	switch typ {
+	case "error":
+		return client.DisassociateWorkflowJobTemplateNotificationTemplatesError
+	case "success":
+		return client.DisassociateWorkflowJobTemplateNotificationTemplatesSuccess
+	case "started":
+		return client.DisassociateWorkflowJobTemplateNotificationTemplatesStarted
+	}
+	return nil
+}
+
+func resourceWorkflowJobTemplateNotificationTemplateCreateForType(typ string) func(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
+	return func(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
+		var diags diag.Diagnostics
+		client := m.(*awx.AWX)
+		awxWorkflowJobTemplateService := client.WorkflowJobTemplateService
+		workflowJobTemplateID := d.Get("workflow_job_template_id").(int)
+		_, err := awxWorkflowJobTemplateService.GetWorkflowJobTemplateByID(workflowJobTemplateID, make(map[string]string))
+		if err != nil {
+			return buildDiagNotFoundFail("workflow job template", workflowJobTemplateID, err)
+		}
+
+		awxWorkflowJobTemplateNotifService := client.WorkflowJobTemplateNotificationTemplatesService
+		notificationTemplateID := d.Get("notification_template_id").(int)
+		associationFunc := getResourceWorkflowJobTemplateNotificationTemplateAssociateFuncForType(awxWorkflowJobTemplateNotifService, typ)
+		if associationFunc == nil {
+			return buildDiagnosticsMessage("Create: WorkflowJobTemplate not AssociateWorkflowJobTemplateNotificationTemplates", "Fail to find association function for notification_template type %s", typ)
+		}
+
+		result, err := associationFunc(workflowJobTemplateID, notificationTemplateID)
+		if err != nil {
+			return buildDiagnosticsMessage("Create: WorkflowJobTemplate not AssociateWorkflowJobTemplateNotificationTemplates", "Fail to associate notification_template credentials with ID %v, for workflow_job_template ID %v, got error: %s", notificationTemplateID, workflowJobTemplateID, err.Error())
+		}
+
+		d.SetId(strconv.Itoa(result.ID))
+		return diags
+	}
+}
+
+func resourceWorkflowJobTemplateNotificationTemplateRead(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
+	var diags diag.Diagnostics
+	return diags
+}
+
+func resourceWorkflowJobTemplateNotificationTemplateDeleteForType(typ string) func(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
+	return func(ctx context.Context, d *schema.ResourceData, m interface{}) diag.Diagnostics {
+		var diags diag.Diagnostics
+		client := m.(*awx.AWX)
+		awxWorkflowJobTemplateService := client.WorkflowJobTemplateService
+		workflowJobTemplateID := d.Get("workflow_job_template_id").(int)
+		_, err := awxWorkflowJobTemplateService.GetWorkflowJobTemplateByID(workflowJobTemplateID, make(map[string]string))
+		if err != nil {
+			return buildDiagNotFoundFail("workflow job template", workflowJobTemplateID, err)
+		}
+
+		awxWorkflowJobTemplateNotifService := client.WorkflowJobTemplateNotificationTemplatesService
+		notificationTemplateID := d.Get("notification_template_id").(int)
+		disassociationFunc := getResourceWorkflowJobTemplateNotificationTemplateDisassociateFuncForType(awxWorkflowJobTemplateNotifService, typ)
+		if disassociationFunc == nil {
+			return buildDiagnosticsMessage("Create: WorkflowJobTemplate not DisassociateWorkflowJobTemplateNotificationTemplates", "Fail to find disassociation function for notification_template type %s", typ)
+		}
+
+		_, err = disassociationFunc(workflowJobTemplateID, notificationTemplateID)
+		if err != nil {
+			return buildDiagnosticsMessage("Create: WorkflowJobTemplate not DisassociateWorkflowJobTemplateNotificationTemplates", "Fail to associate notification_template credentials with ID %v, for job_template ID %v, got error: %s", notificationTemplateID, workflowJobTemplateID, err.Error())
+		}
+
+		d.SetId("")
+		return diags
+	}
+}
diff --git a/awx/resource_workflow_job_template_notification_template_started.go b/awx/resource_workflow_job_template_notification_template_started.go
@@ -0,0 +1,39 @@
+/*
+*TBD*
+
+Example Usage
+
+```hcl
+resource "awx_workflow_job_template_notification_template_started" "baseconfig" {
+  workflow_job_template_id   = awx_workflow_job_template.baseconfig.id
+  notification_template_id   = awx_notification_template.default.id
+}
+```
+
+*/
+package awx
+
+import (
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func resourceWorkflowJobTemplateNotificationTemplateStarted() *schema.Resource {
+	return &schema.Resource{
+		CreateContext: resourceWorkflowJobTemplateNotificationTemplateCreateForType("started"),
+		DeleteContext: resourceWorkflowJobTemplateNotificationTemplateDeleteForType("started"),
+		ReadContext:   resourceWorkflowJobTemplateNotificationTemplateRead,
+
+		Schema: map[string]*schema.Schema{
+			"workflow_job_template_id": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: true,
+			},
+			"notification_template_id": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: true,
+			},
+		},
+	}
+}
diff --git a/awx/resource_workflow_job_template_notification_template_success.go b/awx/resource_workflow_job_template_notification_template_success.go
@@ -0,0 +1,39 @@
+/*
+*TBD*
+
+Example Usage
+
+```hcl
+resource "awx_workflow_job_template_notification_template_success" "baseconfig" {
+  workflow_job_template_id   = awx_workflow_job_template.baseconfig.id
+  notification_template_id   = awx_notification_template.default.id
+}
+```
+
+*/
+package awx
+
+import (
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+)
+
+func resourceWorkflowJobTemplateNotificationTemplateSuccess() *schema.Resource {
+	return &schema.Resource{
+		CreateContext: resourceWorkflowJobTemplateNotificationTemplateCreateForType("success"),
+		DeleteContext: resourceWorkflowJobTemplateNotificationTemplateDeleteForType("success"),
+		ReadContext:   resourceWorkflowJobTemplateNotificationTemplateRead,
+
+		Schema: map[string]*schema.Schema{
+			"workflow_job_template_id": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: true,
+			},
+			"notification_template_id": {
+				Type:     schema.TypeInt,
+				Required: true,
+				ForceNew: true,
+			},
+		},
+	}
+}
