feat: add coder_ai_task resource

dannykopping · dannykopping · commit 97e45841d2c8 · 2025-06-18T12:30:53.000+02:00
Signed-off-by: Danny Kopping &lt;dannykopping@gmail.com&gt;
diff --git a/provider/ai_task.go b/provider/ai_task.go
@@ -0,0 +1,61 @@
+package provider
+
+import (
+	"context"
+
+	"github.com/google/uuid"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
+)
+
+type AITask struct {
+	ID         string             `mapstructure:"id"`
+	SidebarApp []AITaskSidebarApp `mapstructure:"sidebar_app"`
+}
+
+type AITaskSidebarApp struct {
+	ID string `mapstructure:"id"`
+}
+
+// TaskPromptParameterName is the name of the parameter which is *required* to be defined when a coder_ai_task is used.
+const TaskPromptParameterName = "task_prompt"
+
+func aiTask() *schema.Resource {
+	return &schema.Resource{
+		SchemaVersion: 1,
+
+		Description: "Use this resource to define Coder tasks.", // TODO: docs link.
+		CreateContext: func(c context.Context, resourceData *schema.ResourceData, i any) diag.Diagnostics {
+			resourceData.SetId(uuid.NewString())
+			return nil
+		},
+		ReadContext:   schema.NoopContext,
+		DeleteContext: schema.NoopContext,
+		Schema: map[string]*schema.Schema{
+			"id": {
+				Type:        schema.TypeString,
+				Description: "A unique identifier for this resource.",
+				Computed:    true,
+			},
+			"sidebar_app": {
+				Type:        schema.TypeSet,
+				Description: "The coder_app to display in the sidebar.", // TODO: need some clear guidance on what type of app to use here.
+				ForceNew:    true,
+				Required:    true,
+				MaxItems:    1,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"id": {
+							Type:         schema.TypeString,
+							Description:  "A reference to an existing `coder_app` resource in your template.",
+							Required:     true,
+							ForceNew:     true,
+							ValidateFunc: validation.IsUUID,
+						},
+					},
+				},
+			},
+		},
+	}
+}
diff --git a/provider/ai_task_test.go b/provider/ai_task_test.go
@@ -0,0 +1,82 @@
+package provider_test
+
+import (
+	"regexp"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
+	"github.com/stretchr/testify/require"
+)
+
+func TestAITask(t *testing.T) {
+	t.Parallel()
+
+	t.Run("OK", func(t *testing.T) {
+		t.Parallel()
+
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {
+				}
+				resource "coder_agent" "dev" {
+					os = "linux"
+					arch = "amd64"
+				}
+				resource "coder_app" "code-server" {
+					agent_id = coder_agent.dev.id
+					slug = "code-server"
+					display_name = "code-server"
+					icon = "builtin:vim"
+					url = "http://localhost:13337"
+					open_in = "slim-window"
+				}
+				resource "coder_ai_task" "test" {
+					sidebar_app {
+						id = coder_app.code-server.id
+					}
+				}
+				`,
+				Check: func(state *terraform.State) error {
+					require.Len(t, state.Modules, 1)
+					resource := state.Modules[0].Resources["coder_ai_task.test"]
+					require.NotNil(t, resource)
+					for _, key := range []string{
+						"id",
+						"sidebar_app.#",
+					} {
+						value := resource.Primary.Attributes[key]
+						require.NotNil(t, value)
+						require.Greater(t, len(value), 0)
+					}
+					require.Equal(t, "1", resource.Primary.Attributes["sidebar_app.#"])
+					return nil
+				},
+			}},
+		})
+	})
+
+	t.Run("InvalidSidebarAppID", func(t *testing.T) {
+		t.Parallel()
+
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {
+				}
+				resource "coder_ai_task" "test" {
+					sidebar_app {
+						id = "not-a-uuid"
+					}
+				}
+				`,
+				ExpectError: regexp.MustCompile(`expected "sidebar_app.0.id" to be a valid UUID`),
+			}},
+		})
+	})
+}
diff --git a/provider/provider.go b/provider/provider.go
@@ -72,6 +72,7 @@ func New() *schema.Provider {
 		ResourcesMap: map[string]*schema.Resource{
 			"coder_agent":          agentResource(),
 			"coder_agent_instance": agentInstanceResource(),
+			"coder_ai_task":        aiTask(),
 			"coder_app":            appResource(),
 			"coder_metadata":       metadataResource(),
 			"coder_script":         scriptResource(),
