feat: use labels to select workflow job notification channels (#58)

cd/manager/common/job/job.go

@@ -125,7 +125,7 @@ func CreateWorkflowJob(jobState JobState) (Workflow, error) {
 	} else if workflow, found := jobState.Params[WorkflowJobParam_Workflow].(string); !found {
 		return Workflow{}, fmt.Errorf("missing workflow")
 	} else {
-		workflowInputs := make(map[string]interface{}, 0)
+		workflowInputs := make(map[string]interface{})
 		if inputs, found := jobState.Params[WorkflowJobParam_Inputs].(map[string]interface{}); found {
 			workflowInputs = inputs
 		}
@@ -134,6 +134,14 @@ func CreateWorkflowJob(jobState JobState) (Workflow, error) {
 		if id, found := jobState.Params[JobParam_Id].(float64); found {
 			workflowRunId = int64(id)
 		}
-		return Workflow{org, repo, workflow, ref, workflowInputs, workflowRunUrl, workflowRunId}, nil
+		workflowLabels := make([]string, 0)
+		if labels, found := jobState.Params[WorkflowJobParam_Labels].([]interface{}); found {
+			for _, label := range labels {
+				if l, ok := label.(string); ok {
+					workflowLabels = append(workflowLabels, l)
+				}
+			}
+		}
+		return Workflow{org, repo, workflow, ref, workflowInputs, workflowRunUrl, workflowRunId, workflowLabels}, nil
 	}
 }

cd/manager/common/job/models.go

@@ -74,6 +74,12 @@ const (
 	WorkflowJobParam_Url          string = "url"
 	WorkflowJobParam_JobId        string = "job_id"
 	WorkflowJobParam_TestSelector string = "test_selector"
+	WorkflowJobParam_Labels       string = "labels"
+)
+
+const (
+	WorkflowJobLabel_Test   string = "test"
+	WorkflowJobLabel_Deploy string = "deploy"
 )
 
 // JobState represents the state of a job in the database
@@ -95,4 +101,5 @@ type Workflow struct {
 	Inputs   map[string]interface{}
 	Url      string
 	Id       int64
+	Labels   []string
 }

cd/manager/notifs/discord.go

@@ -88,6 +88,27 @@ func parseDiscordWebhookUrl(urlEnv string) (webhook.Client, error) {
 	return nil, nil
 }
 
+func webhooksForLabels(labels []string) ([]webhook.Client, error) {
+	webhooks := make([]webhook.Client, 0)
+	for _, label := range labels {
+		switch label {
+		case job.WorkflowJobLabel_Test:
+			if t, err := parseDiscordWebhookUrl("DISCORD_TESTS_WEBHOOK"); err != nil {
+				return nil, err
+			} else {
+				webhooks = append(webhooks, t)
+			}
+		case job.WorkflowJobLabel_Deploy:
+			if t, err := parseDiscordWebhookUrl("DISCORD_DEPLOYMENTS_WEBHOOK"); err != nil {
+				return nil, err
+			} else {
+				webhooks = append(webhooks, t)
+			}
+		}
+	}
+	return webhooks, nil
+}
+
 func (n JobNotifs) NotifyJob(jobs ...job.JobState) {
 	for _, jobState := range jobs {
 		if jn, err := n.getJobNotif(jobState); err != nil {

cd/manager/notifs/workflow.go

@@ -23,25 +23,25 @@ const (
 )
 
 type workflowNotif struct {
-	state           job.JobState
-	workflow        job.Workflow
-	workflowWebhook webhook.Client
+	state            job.JobState
+	workflow         job.Workflow
+	workflowWebhooks []webhook.Client
 }
 
 func newWorkflowNotif(jobState job.JobState) (jobNotif, error) {
-	if w, err := parseDiscordWebhookUrl("DISCORD_WORKFLOWS_WEBHOOK"); err != nil {
+	if workflow, err := job.CreateWorkflowJob(jobState); err != nil {
 		return nil, err
-	} else if workflow, err := job.CreateWorkflowJob(jobState); err != nil {
+	} else if webhooks, err := webhooksForLabels(workflow.Labels); err != nil {
 		return nil, err
 	} else {
-		return &workflowNotif{jobState, workflow, w}, nil
+		return &workflowNotif{jobState, workflow, webhooks}, nil
 	}
 }
 
 func (w workflowNotif) getChannels() []webhook.Client {
 	// Skip "started" notifications so that the channel doesn't get too noisy
 	if w.state.Stage != job.JobStage_Started {
-		return []webhook.Client{w.workflowWebhook}
+		return w.workflowWebhooks
 	}
 	return nil
 }