From c80041b152bd476fad51f3370eacfaf5e17ecbaf Mon Sep 17 00:00:00 2001
From: Danny Kopping <dannykopping@gmail.com>
Date: Thu, 19 Jun 2025 20:24:39 +0200
Subject: [PATCH] feat: add `coder_ai_task` resource (#413)

---
 docs/resources/ai_task.md | 31 +++++++++++++++
 provider/ai_task.go       | 61 +++++++++++++++++++++++++++++
 provider/ai_task_test.go  | 82 +++++++++++++++++++++++++++++++++++++++
 provider/provider.go      |  1 +
 4 files changed, 175 insertions(+)
 create mode 100644 docs/resources/ai_task.md
 create mode 100644 provider/ai_task.go
 create mode 100644 provider/ai_task_test.go

diff --git a/docs/resources/ai_task.md b/docs/resources/ai_task.md
new file mode 100644
index 00000000..1922ef59
--- /dev/null
+++ b/docs/resources/ai_task.md
@@ -0,0 +1,31 @@
+---
+# generated by https://github.com/hashicorp/terraform-plugin-docs
+page_title: "coder_ai_task Resource - terraform-provider-coder"
+subcategory: ""
+description: |-
+  Use this resource to define Coder tasks.
+---
+
+# coder_ai_task (Resource)
+
+Use this resource to define Coder tasks.
+
+
+
+<!-- schema generated by tfplugindocs -->
+## Schema
+
+### Required
+
+- `sidebar_app` (Block Set, Min: 1, Max: 1) The coder_app to display in the sidebar. Usually a chat interface with the AI agent running in the workspace, like https://github.com/coder/agentapi. (see [below for nested schema](#nestedblock--sidebar_app))
+
+### Read-Only
+
+- `id` (String) A unique identifier for this resource.
+
+<a id="nestedblock--sidebar_app"></a>
+### Nested Schema for `sidebar_app`
+
+Required:
+
+- `id` (String) A reference to an existing `coder_app` resource in your template.
diff --git a/provider/ai_task.go b/provider/ai_task.go
new file mode 100644
index 00000000..76b19f3c
--- /dev/null
+++ 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 = "AI Prompt"
+
+func aiTask() *schema.Resource {
+	return &schema.Resource{
+		SchemaVersion: 1,
+
+		Description: "Use this resource to define Coder tasks.",
+		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. Usually a chat interface with the AI agent running in the workspace, like https://github.com/coder/agentapi.",
+				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
new file mode 100644
index 00000000..5f7a8a49
--- /dev/null
+++ 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
index cc2644ef..43e3a6ac 100644
--- 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(),

<!DOCTYPE html PUBLIC '-//W3C//DTD XHTML 1.0 Transitional//EN' 'http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd'>
<html xmlns='http://www.w3.org/1999/xhtml'>
<head>
<title>pFad - Phonifier reborn</title>
<meta http-equiv='Content-Type' content='text/html; charset=utf-8' />
</head>
<body>
<h1>Pfad - The Proxy pFad of &#169; 2024 Garber Painting. All rights reserved.</h1>


<!-- Disclaimer -->
<p>Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.</p>
<br>
<p>Alternative Proxies:</p><p><a href="http://clevelandohioweatherforecast.com/php-proxy/index.php?q=https://github.com/coder/terraform-provider-coder/compare/v2.6.0...v2.7.0.patch" target="_blank">Alternative Proxy</a></p><p><a href="http://clevelandohioweatherforecast.com/pFad/index.php?u=https://github.com/coder/terraform-provider-coder/compare/v2.6.0...v2.7.0.patch" target="_blank">pFad Proxy</a></p><p><a href="http://clevelandohioweatherforecast.com/pFad/v3index.php?u=https://github.com/coder/terraform-provider-coder/compare/v2.6.0...v2.7.0.patch" target="_blank">pFad v3 Proxy</a></p><p><a href="http://clevelandohioweatherforecast.com/pFad/v4index.php?u=https://github.com/coder/terraform-provider-coder/compare/v2.6.0...v2.7.0.patch" target="_blank">pFad v4 Proxy</a></p></body>
</html>