feat: add coder_ai_task_prompt data source

johnstcn · johnstcn · commit 37908cf4a0bf · 2025-11-03T21:30:26.000Z
diff --git a/docs/data-sources/ai_task_prompt.md b/docs/data-sources/ai_task_prompt.md
@@ -0,0 +1,22 @@
+---
+# generated by https://github.com/hashicorp/terraform-plugin-docs
+page_title: "coder_ai_task_prompt Data Source - terraform-provider-coder"
+subcategory: ""
+description: |-
+  Use this data source to read information about Coder tasks.
+---
+
+# coder_ai_task_prompt (Data Source)
+
+Use this data source to read information about Coder tasks.
+
+
+
+<!-- schema generated by tfplugindocs -->
+## Schema
+
+### Read-Only
+
+- `enabled` (Boolean) True when executing in a Coder Task context, false when in a Coder Workspace context
+- `id` (String) The UUID of the task, if executing in a Coder Task context. Empty in a Coder Workspace context.
+- `value` (String) The prompt text provided to the task by Coder, if executing in a Coder Task context. Empty in a Coder Workspace context.
diff --git a/integration/coder-ai-task/main.tf b/integration/coder-ai-task/main.tf
@@ -32,6 +32,8 @@ data "coder_parameter" "ai_prompt" {
   mutable     = true
 }
 
+data "coder_ai_task_prompt" "me" {}
+
 resource "coder_ai_task" "task" {
   sidebar_app {
     id = coder_app.ai_interface.id
@@ -46,6 +48,10 @@ locals {
     "ai_task.prompt"  = coder_ai_task.task.prompt
     "ai_task.enabled" = tostring(coder_ai_task.task.enabled)
     "app.id"          = coder_app.ai_interface.id
+
+    "ai_task_prompt.id"      = data.coder_ai_task_prompt.me.id
+    "ai_task_prompt.value"   = data.coder_ai_task_prompt.me.value
+    "ai_task_prompt.enabled" = tostring(data.coder_ai_task_prompt.me.enabled)
   }
 }
 
diff --git a/integration/integration_test.go b/integration/integration_test.go
@@ -215,15 +215,17 @@ func TestIntegration(t *testing.T) {
 			name:       "coder-ai-task",
 			minVersion: "v2.26.0",
 			expectedOutput: map[string]string{
-				"ai_task.id":      `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
-				"ai_task.prompt":  "",
-				"ai_task.app_id":  `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
-				"ai_task.enabled": "false",
-				"app.id":          `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
+				"ai_task.id":             `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
+				"ai_task.prompt":         "",
+				"ai_task.app_id":         `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
+				"ai_task.enabled":        "false",
+				"app.id":                 `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
+				"ai_task_prompt.id":      `^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$`,
+				"ai_task_prompt.value":   "",
+				"ai_task_prompt.enabled": "false",
 			},
 		},
 	} {
-		tt := tt
 		t.Run(tt.name, func(t *testing.T) {
 			t.Parallel()
 			if coderVersion != "latest" && semver.Compare(coderVersion, tt.minVersion) < 0 {
diff --git a/provider/ai_task.go b/provider/ai_task.go
@@ -115,3 +115,44 @@ func aiTaskResource() *schema.Resource {
 		},
 	}
 }
+
+func aiTaskPromptDatasource() *schema.Resource {
+	return &schema.Resource{
+		Description: "Use this data source to read information about Coder tasks.",
+		ReadContext: func(ctx context.Context, rd *schema.ResourceData, i interface{}) diag.Diagnostics {
+			diags := diag.Diagnostics{}
+
+			idStr := os.Getenv("CODER_TASK_ID")
+			if idStr == "" || idStr == uuid.Nil.String() {
+				rd.SetId(uuid.NewString())
+				_ = rd.Set("enabled", false)
+			} else if _, err := uuid.Parse(idStr); err == nil {
+				rd.SetId(idStr)
+				_ = rd.Set("enabled", true)
+			} else { // invalid UUID
+				diags = append(diags, errorAsDiagnostics(err)...)
+			}
+
+			prompt := os.Getenv("CODER_TASK_PROMPT")
+			_ = rd.Set("value", prompt)
+			return diags
+		},
+		Schema: map[string]*schema.Schema{
+			"id": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "The UUID of the task, if executing in a Coder Task context. Empty in a Coder Workspace context.",
+			},
+			"value": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "The prompt text provided to the task by Coder, if executing in a Coder Task context. Empty in a Coder Workspace context.",
+			},
+			"enabled": {
+				Type:        schema.TypeBool,
+				Computed:    true,
+				Description: "True when executing in a Coder Task context, false when in a Coder Workspace context",
+			},
+		},
+	}
+}
diff --git a/provider/ai_task_test.go b/provider/ai_task_test.go
@@ -4,6 +4,7 @@ import (
 	"regexp"
 	"testing"
 
+	"github.com/google/uuid"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
 	"github.com/stretchr/testify/require"
@@ -214,3 +215,83 @@ func TestAITask(t *testing.T) {
 		})
 	})
 }
+
+func TestAITaskPromptDatasource(t *testing.T) {
+	t.Run("Exists", func(t *testing.T) {
+		t.Setenv("CODER_TASK_ID", "7d8d4c2e-fb57-44f9-a183-22509819c2e7")
+		t.Setenv("CODER_TASK_PROMPT", "some task prompt")
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {}
+				data "coder_ai_task_prompt" "me" {}
+				`,
+				Check: func(s *terraform.State) error {
+					require.Len(t, s.Modules, 1)
+					require.Len(t, s.Modules[0].Resources, 1)
+					resource := s.Modules[0].Resources["data.coder_ai_task_prompt.me"]
+					require.NotNil(t, resource)
+
+					taskID := resource.Primary.Attributes["id"]
+					require.Equal(t, "7d8d4c2e-fb57-44f9-a183-22509819c2e7", taskID)
+
+					taskPromptValue := resource.Primary.Attributes["value"]
+					require.Equal(t, "some task prompt", taskPromptValue)
+
+					enabledValue := resource.Primary.Attributes["enabled"]
+					require.Equal(t, "true", enabledValue)
+					return nil
+				},
+			}},
+		})
+	})
+
+	t.Run("NotExists", func(t *testing.T) {
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {}
+				data "coder_ai_task_prompt" "me" {}
+				`,
+				Check: func(s *terraform.State) error {
+					require.Len(t, s.Modules, 1)
+					require.Len(t, s.Modules[0].Resources, 1)
+					resource := s.Modules[0].Resources["data.coder_ai_task_prompt.me"]
+					require.NotNil(t, resource)
+
+					taskID := resource.Primary.Attributes["id"]
+					require.NotEmpty(t, taskID)
+					require.NotEqual(t, uuid.Nil.String(), taskID)
+					_, err := uuid.Parse(taskID)
+					require.NoError(t, err)
+
+					taskPromptValue := resource.Primary.Attributes["value"]
+					require.Empty(t, taskPromptValue)
+
+					enabledValue := resource.Primary.Attributes["enabled"]
+					require.Equal(t, "false", enabledValue)
+					return nil
+				},
+			}},
+		})
+	})
+
+	t.Run("InvalidTaskID", func(t *testing.T) {
+		t.Setenv("CODER_TASK_ID", "not a valid UUID")
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {}
+				data "coder_ai_task_prompt" "me" {}
+				`,
+				ExpectError: regexp.MustCompile(`invalid UUID`),
+			}},
+		})
+	})
+}
diff --git a/provider/provider.go b/provider/provider.go
@@ -64,6 +64,7 @@ func New() *schema.Provider {
 			"coder_external_auth":    externalAuthDataSource(),
 			"coder_workspace_owner":  workspaceOwnerDataSource(),
 			"coder_workspace_preset": workspacePresetDataSource(),
+			"coder_ai_task_prompt":   aiTaskPromptDatasource(),
 		},
 		ResourcesMap: map[string]*schema.Resource{
 			"coder_agent":          agentResource(),
diff --git a/provider/provider_test.go b/provider/provider_test.go
@@ -37,7 +37,8 @@ func TestProviderEmpty(t *testing.T) {
 			}
 			data "coder_parameter" "param" {
 				name = "hey"
-			}`,
+			}
+			data "coder_ai_task_prompt" "me" {}`,
 			Check: func(state *terraform.State) error {
 				return nil
 			},

Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,8 @@ data "coder_parameter" "ai_prompt" {`
`32`	`32`	`mutable = true`
`33`	`33`	`}`
`34`	`34`
	`35`	`+data "coder_ai_task_prompt" "me" {}`
	`36`	`+`
`35`	`37`	`resource "coder_ai_task" "task" {`
`36`	`38`	`sidebar_app {`
`37`	`39`	`id = coder_app.ai_interface.id`
`@@ -46,6 +48,10 @@ locals {`
`46`	`48`	`"ai_task.prompt" = coder_ai_task.task.prompt`
`47`	`49`	`"ai_task.enabled" = tostring(coder_ai_task.task.enabled)`
`48`	`50`	`"app.id" = coder_app.ai_interface.id`
	`51`	`+`
	`52`	`+ "ai_task_prompt.id" = data.coder_ai_task_prompt.me.id`
	`53`	`+ "ai_task_prompt.value" = data.coder_ai_task_prompt.me.value`
	`54`	`+ "ai_task_prompt.enabled" = tostring(data.coder_ai_task_prompt.me.enabled)`
`49`	`55`	`}`
`50`	`56`	`}`
`51`	`57`