Add "coder_workspace" data source

kylecarbs · kylecarbs · commit 8de9e61c6b71 · 2022-02-27T18:11:11.000Z
diff --git a/provisioner/terraform/provider/provider.go b/provisioner/terraform/provider/provider.go
@@ -12,6 +12,7 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
 
+	"github.com/coder/coder/database"
 	"github.com/coder/coder/provisionersdk"
 )
 
@@ -57,6 +58,24 @@ func New() *schema.Provider {
 			}, nil
 		},
 		DataSourcesMap: map[string]*schema.Resource{
+			"coder_workspace": {
+				Description: "TODO",
+				ReadContext: func(c context.Context, rd *schema.ResourceData, i interface{}) diag.Diagnostics {
+					rd.SetId(uuid.NewString())
+					return nil
+				},
+				Schema: map[string]*schema.Schema{
+					"transition": {
+						Type:        schema.TypeString,
+						Optional:    true,
+						Description: "TODO",
+						DefaultFunc: func() (interface{}, error) {
+							return os.Getenv("CODER_WORKSPACE_TRANSITION"), nil
+						},
+						ValidateFunc: validation.StringInSlice([]string{string(database.WorkspaceTransitionStart), string(database.WorkspaceTransitionStop)}, false),
+					},
+				},
+			},
 			"coder_agent_script": {
 				Description: "TODO",
 				ReadContext: func(c context.Context, resourceData *schema.ResourceData, i interface{}) diag.Diagnostics {
diff --git a/provisioner/terraform/provider/provider_test.go b/provisioner/terraform/provider/provider_test.go
@@ -19,6 +19,35 @@ func TestProvider(t *testing.T) {
 	require.NoError(t, err)
 }
 
+func TestWorkspace(t *testing.T) {
+	t.Parallel()
+	resource.Test(t, resource.TestCase{
+		Providers: map[string]*schema.Provider{
+			"coder": provider.New(),
+		},
+		IsUnitTest: true,
+		Steps: []resource.TestStep{{
+			Config: `
+			provider "coder" {
+				url = "https://example.com"
+			}
+			data "coder_workspace" "me" {
+				transition = "start"
+			}`,
+			Check: func(state *terraform.State) error {
+				require.Len(t, state.Modules, 1)
+				require.Len(t, state.Modules[0].Resources, 1)
+				resource := state.Modules[0].Resources["data.coder_workspace.me"]
+				require.NotNil(t, resource)
+				value := resource.Primary.Attributes["transition"]
+				require.NotNil(t, value)
+				t.Log(value)
+				return nil
+			},
+		}},
+	})
+}
+
 func TestAgentScript(t *testing.T) {
 	t.Parallel()
 	resource.Test(t, resource.TestCase{
