feat: support workspace tags (#223)

mtojek · web-flow · commit 041e000b50a7 · 2024-05-16T17:29:09.000+02:00
diff --git a/examples/resources/coder_workspace_tags/resource.tf b/examples/resources/coder_workspace_tags/resource.tf
@@ -0,0 +1,50 @@
+provider "coder" {}
+
+data "coder_parameter" "os_selector" {
+  name         = "os_selector"
+  display_name = "Operating System"
+  mutable      = false
+
+  default = "osx"
+
+  option {
+    icon  = "/icons/linux.png"
+    name  = "Linux"
+    value = "linux"
+  }
+  option {
+    icon  = "/icons/osx.png"
+    name  = "OSX"
+    value = "osx"
+  }
+  option {
+    icon  = "/icons/windows.png"
+    name  = "Windows"
+    value = "windows"
+  }
+}
+
+data "coder_parameter" "feature_cache_enabled" {
+  name         = "feature_cache_enabled"
+  display_name = "Enable cache?"
+  type         = "bool"
+
+  default = false
+}
+
+data "coder_parameter" "feature_debug_enabled" {
+  name         = "feature_debug_enabled"
+  display_name = "Enable debug?"
+  type         = "bool"
+
+  default = true
+}
+
+data "coder_workspace_tags" "custom_workspace_tags" {
+  tags = {
+    "cluster" = "developers"
+    "os"      = data.coder_parameter.os_selector.value
+    "debug"   = "${data.coder_parameter.feature_debug_enabled.value}+12345"
+    "cache"   = data.coder_parameter.feature_cache_enabled.value == "true" ? "nix-with-cache" : "no-cache"
+  }
+}
diff --git a/provider/examples_test.go b/provider/examples_test.go
@@ -14,19 +14,25 @@ import (
 func TestExamples(t *testing.T) {
 	t.Parallel()
 
-	t.Run("coder_parameter", func(t *testing.T) {
-		t.Parallel()
+	for _, testDir := range []string{
+		"coder_parameter",
+		"coder_workspace_tags",
+	} {
+		t.Run(testDir, func(t *testing.T) {
+			testDir := testDir
+			t.Parallel()
 
-		resource.Test(t, resource.TestCase{
-			Providers: map[string]*schema.Provider{
-				"coder": provider.New(),
-			},
-			IsUnitTest: true,
-			Steps: []resource.TestStep{{
-				Config: mustReadFile(t, "../examples/resources/coder_parameter/resource.tf"),
-			}},
+			resource.Test(t, resource.TestCase{
+				Providers: map[string]*schema.Provider{
+					"coder": provider.New(),
+				},
+				IsUnitTest: true,
+				Steps: []resource.TestStep{{
+					Config: mustReadFile(t, "../examples/resources/"+testDir+"/resource.tf"),
+				}},
+			})
 		})
-	})
+	}
 }
 
 func mustReadFile(t *testing.T, path string) string {
diff --git a/provider/provider.go b/provider/provider.go
@@ -68,11 +68,12 @@ func New() *schema.Provider {
 			}, nil
 		},
 		DataSourcesMap: map[string]*schema.Resource{
-			"coder_workspace":     workspaceDataSource(),
-			"coder_provisioner":   provisionerDataSource(),
-			"coder_parameter":     parameterDataSource(),
-			"coder_git_auth":      gitAuthDataSource(),
-			"coder_external_auth": externalAuthDataSource(),
+			"coder_workspace":      workspaceDataSource(),
+			"coder_workspace_tags": workspaceTagDataSource(),
+			"coder_provisioner":    provisionerDataSource(),
+			"coder_parameter":      parameterDataSource(),
+			"coder_git_auth":       gitAuthDataSource(),
+			"coder_external_auth":  externalAuthDataSource(),
 		},
 		ResourcesMap: map[string]*schema.Resource{
 			"coder_agent":          agentResource(),
diff --git a/provider/workspace_tags.go b/provider/workspace_tags.go
@@ -0,0 +1,32 @@
+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"
+)
+
+type WorkspaceTags struct {
+	Tags map[string]string
+}
+
+func workspaceTagDataSource() *schema.Resource {
+	return &schema.Resource{
+		Description: "Use this data source to configure workspace tags to select provisioners.",
+		ReadContext: func(ctx context.Context, rd *schema.ResourceData, i interface{}) diag.Diagnostics {
+			rd.SetId(uuid.NewString())
+			return nil
+		},
+		Schema: map[string]*schema.Schema{
+			"tags": {
+				Type:        schema.TypeMap,
+				Description: `Key-value map with workspace tags`,
+				ForceNew:    true,
+				Optional:    true,
+				Elem:        &schema.Schema{Type: schema.TypeString},
+			},
+		},
+	}
+}
diff --git a/provider/workspace_tags_test.go b/provider/workspace_tags_test.go
@@ -0,0 +1,48 @@
+package provider_test
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/terraform-provider-coder/provider"
+)
+
+func TestWorkspaceTags(t *testing.T) {
+	resource.Test(t, resource.TestCase{
+		Providers: map[string]*schema.Provider{
+			"coder": provider.New(),
+		},
+		IsUnitTest: true,
+		Steps: []resource.TestStep{{
+			Config: `
+			provider "coder" {
+			}
+			data "coder_parameter" "animal" {
+				name = "animal"
+				type = "string"
+				default = "chris"
+			}
+			data "coder_workspace_tags" "wt" {
+				tags = {
+					"cat" = "james"
+					"dog" = data.coder_parameter.animal.value
+				}
+			}`,
+			Check: func(state *terraform.State) error {
+				require.Len(t, state.Modules, 1)
+				require.Len(t, state.Modules[0].Resources, 2)
+				resource := state.Modules[0].Resources["data.coder_workspace_tags.wt"]
+				require.NotNil(t, resource)
+
+				attribs := resource.Primary.Attributes
+				require.Equal(t, "james", attribs["tags.cat"])
+				require.Equal(t, "chris", attribs["tags.dog"])
+				return nil
+			},
+		}},
+	})
+}
