schema

incident-io · Apr 23, 2024 · 420ad67 · 420ad67
1 parent e8faaaf
commit 420ad67
Show file tree

Hide file tree

Showing 3 changed files with 124 additions and 25 deletions.
diff --git a/internal/provider/engine.go b/internal/provider/engine.go
@@ -28,7 +28,7 @@ type IncidentEngineExpression struct {
 	ElseBranch    *IncidentEngineElseBranch           `tfsdk:"else_branch"`
 	ID            types.String                        `tfsdk:"id"` // rmv?
 	Label         types.String                        `tfsdk:"label"`
-	Operations    []IncidentEngineExpressionOperation `tfsdk:"operation"`
+	Operations    []IncidentEngineExpressionOperation `tfsdk:"operations"`
 	Reference     types.String                        `tfsdk:"reference"`
 	RootReference types.String                        `tfsdk:"root_reference"`
 }

diff --git a/internal/provider/incident_workflow_resource.go b/internal/provider/incident_workflow_resource.go
@@ -60,24 +60,53 @@ func (r *IncidentWorkflowResource) Schema(ctx context.Context, req resource.Sche
 		},
 	}
 
+	paramBindingAttributes := map[string]schema.Attribute{
+		"array_value": schema.SetNestedAttribute{
+			Optional: true,
+			NestedObject: schema.NestedAttributeObject{
+				Attributes: paramBindingValueAttributes,
+			},
+		},
+		"value": schema.SingleNestedAttribute{
+			Optional:   true,
+			Attributes: paramBindingValueAttributes,
+		},
+	}
+
 	paramBindingsAttribute := schema.ListNestedAttribute{
+		Required: true,
+		NestedObject: schema.NestedAttributeObject{
+			Attributes: paramBindingAttributes,
+		},
+	}
+
+	conditionsAttribute := schema.SetNestedAttribute{
 		Required: true,
 		NestedObject: schema.NestedAttributeObject{
 			Attributes: map[string]schema.Attribute{
-				"array_value": schema.SetNestedAttribute{
-					Optional: true,
-					NestedObject: schema.NestedAttributeObject{
-						Attributes: paramBindingValueAttributes,
-					},
+				"operation": schema.StringAttribute{
+					Required: true,
 				},
-				"value": schema.SingleNestedAttribute{
-					Optional:   true,
-					Attributes: paramBindingValueAttributes,
+				"param_bindings": paramBindingsAttribute,
+				"subject": schema.StringAttribute{
+					Required: true,
 				},
 			},
 		},
 	}
 
+	returnsAttribute := schema.SingleNestedAttribute{
+		Required: true,
+		Attributes: map[string]schema.Attribute{
+			"array": schema.BoolAttribute{
+				Required: true,
+			},
+			"type": schema.StringAttribute{
+				Required: true,
+			},
+		},
+	}
+
 	resp.Schema = schema.Schema{
 		MarkdownDescription: apischema.TagDocstring("Workflows V2"),
 		Attributes: map[string]schema.Attribute{
@@ -103,20 +132,7 @@ func (r *IncidentWorkflowResource) Schema(ctx context.Context, req resource.Sche
 				Required: true,
 				NestedObject: schema.NestedAttributeObject{
 					Attributes: map[string]schema.Attribute{
-						"conditions": schema.SetNestedAttribute{
-							Required: true,
-							NestedObject: schema.NestedAttributeObject{
-								Attributes: map[string]schema.Attribute{
-									"operation": schema.StringAttribute{
-										Required: true,
-									},
-									"param_bindings": paramBindingsAttribute,
-									"subject": schema.StringAttribute{
-										Required: true,
-									},
-								},
-							},
-						},
+						"conditions": conditionsAttribute,
 					},
 				},
 			},
@@ -137,7 +153,82 @@ func (r *IncidentWorkflowResource) Schema(ctx context.Context, req resource.Sche
 					},
 				},
 			},
-			// TODO add expressions to schema, test locally, unit test
+			"expressions": schema.ListNestedAttribute{
+				Required: true,
+				NestedObject: schema.NestedAttributeObject{
+					Attributes: map[string]schema.Attribute{
+						"else_branch": schema.SingleNestedAttribute{
+							Optional: true,
+							Attributes: map[string]schema.Attribute{
+								"result": schema.SingleNestedAttribute{
+									Required:   true,
+									Attributes: paramBindingAttributes,
+								},
+							},
+						},
+						"id": schema.StringAttribute{
+							Computed: true,
+						},
+						"label": schema.StringAttribute{
+							Required: true,
+						},
+						"operations": schema.ListNestedAttribute{
+							Required: true,
+							NestedObject: schema.NestedAttributeObject{
+								Attributes: map[string]schema.Attribute{
+									"branches": schema.SingleNestedAttribute{
+										Optional: true,
+										Attributes: map[string]schema.Attribute{
+											"branches": schema.ListNestedAttribute{
+												Required: true,
+												NestedObject: schema.NestedAttributeObject{
+													Attributes: map[string]schema.Attribute{
+														"conditions": conditionsAttribute,
+														"result":     paramBindingsAttribute,
+													},
+												},
+											},
+											"returns": returnsAttribute,
+										},
+									},
+									"filter": schema.SingleNestedAttribute{
+										Optional: true,
+										Attributes: map[string]schema.Attribute{
+											"conditions": conditionsAttribute,
+										},
+									},
+									"navigate": schema.SingleNestedAttribute{
+										Optional: true,
+										Attributes: map[string]schema.Attribute{
+											"reference": schema.StringAttribute{
+												Required: true,
+											},
+										},
+									},
+									"operation_type": schema.StringAttribute{
+										Required: true,
+									},
+									"parse": schema.SingleNestedAttribute{
+										Optional: true,
+										Attributes: map[string]schema.Attribute{
+											"returns": returnsAttribute,
+											"source": schema.StringAttribute{
+												Required: true,
+											},
+										},
+									},
+								},
+							},
+						},
+						"reference": schema.StringAttribute{
+							Required: true,
+						},
+						"root_reference": schema.StringAttribute{
+							Required: true,
+						},
+					},
+				},
+			},
 		},
 	}
 }

diff --git a/internal/provider/incident_workflow_resource_test.go b/internal/provider/incident_workflow_resource_test.go
@@ -118,8 +118,16 @@ resource "incident_workflow" "example" {
 			]
 		}
 	]
+	expressions = [
+		{
+			"label" = "My Expression"
+			operations = []
+			reference = "status"
+			root_reference = "incident"
+		}
+	]
 }
-`))
+`)) // TODO test locally and UT
 
 func incidentWorkflowDefault() workflowTemplateOverrides {
 	return workflowTemplateOverrides{