feat: add enabled computed field to coder_ai_task

As discussed, we want to enable a way for consumers to know if their
template is being provisioned as a task or not. We know that
`CODER_TASK_ID` is set by the provisioner, it is being provisioned as
a task, and not as a usual workspace. We use this knowledge to set
a computed field `enabled` to reflect this.

Author: DanielleMaywood

provider/ai_task.go

@@ -34,19 +34,16 @@ func aiTaskResource() *schema.Resource {
 
 			if idStr := os.Getenv("CODER_TASK_ID"); idStr != "" {
 				resourceData.SetId(idStr)
+				resourceData.Set("enabled", true)
 			} else {
 				resourceData.SetId(uuid.NewString())
-
-				diags = append(diags, diag.Diagnostic{
-					Severity: diag.Warning,
-					Summary:  "`CODER_TASK_ID` should be set. If you are seeing this message, the version of the Coder Terraform provider you are using is likely too new for your current Coder version.",
-				})
+				resourceData.Set("enabled", false)
 			}
 
 			if prompt := os.Getenv("CODER_TASK_PROMPT"); prompt != "" {
 				resourceData.Set("prompt", prompt)
 			} else {
-				resourceData.Set("prompt", "default")
+				resourceData.Set("prompt", "")
 			}
 
 			var (
@@ -112,6 +109,11 @@ func aiTaskResource() *schema.Resource {
 				ValidateFunc:  validation.IsUUID,
 				ConflictsWith: []string{"sidebar_app"},
 			},
+			"enabled": {
+				Type:        schema.TypeBool,
+				Description: "The field is true if the template is being provisioned as a task.",
+				Computed:    true,
+			},
 		},
 	}
 }

provider/ai_task_test.go

@@ -9,6 +9,62 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+func TestAITask_Enabled(t *testing.T) {
+	t.Run("EnabledWhenTask", func(t *testing.T) {
+		t.Setenv("CODER_TASK_ID", "7d8d4c2e-fb57-44f9-a183-22509819c2e7")
+
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {
+				}
+				resource "coder_ai_task" "test" {
+					app_id = "9a3ff7b4-4b3f-48c6-8d3a-a8118ac921fc"
+				}
+				`,
+				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)
+
+					require.Equal(t, "true", resource.Primary.Attributes["enabled"])
+
+					return nil
+				},
+			}},
+		})
+	})
+
+	t.Run("DisabledWhenWorkspace", func(t *testing.T) {
+		t.Setenv("CODER_TASK_ID", "")
+
+		resource.Test(t, resource.TestCase{
+			ProviderFactories: coderFactory(),
+			IsUnitTest:        true,
+			Steps: []resource.TestStep{{
+				Config: `
+				provider "coder" {
+				}
+				resource "coder_ai_task" "test" {
+					app_id = "9a3ff7b4-4b3f-48c6-8d3a-a8118ac921fc"
+				}
+				`,
+				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)
+
+					require.Equal(t, "false", resource.Primary.Attributes["enabled"])
+
+					return nil
+				},
+			}},
+		})
+	})
+}
+
 func TestAITask(t *testing.T) {
 	t.Setenv("CODER_TASK_ID", "7d8d4c2e-fb57-44f9-a183-22509819c2e7")
 	t.Setenv("CODER_TASK_PROMPT", "some task prompt")
@@ -35,6 +91,7 @@ func TestAITask(t *testing.T) {
 						"id",
 						"prompt",
 						"app_id",
+						"enabled",
 					} {
 						value := resource.Primary.Attributes[key]
 						require.NotNil(t, value)
@@ -97,6 +154,7 @@ func TestAITask(t *testing.T) {
 						"id",
 						"prompt",
 						"app_id",
+						"enabled",
 					} {
 						value := resource.Primary.Attributes[key]
 						require.NotNil(t, value)