feat: add session token data source

sreya · sreya · commit 8f726142a3f9 · 2023-05-09T00:43:51.000Z
diff --git a/provider/workspace.go b/provider/workspace.go
@@ -51,6 +51,9 @@ func workspaceDataSource() *schema.Resource {
 			}
 			rd.Set("name", name)
 
+			sessionToken := os.Getenv("CODER_SESSION_TOKEN")
+			_ = rd.Set("coder_session_token", sessionToken)
+
 			id := os.Getenv("CODER_WORKSPACE_ID")
 			if id == "" {
 				id = uuid.NewString()
@@ -131,6 +134,11 @@ func workspaceDataSource() *schema.Resource {
 				Computed:    true,
 				Description: "Name of the workspace.",
 			},
+			"coder_session_token": {
+				Type:        schema.TypeString,
+				Computed:    true,
+				Description: "Session token for interfacing with a Coder deployment. It is regenerated everytime a workspace is started.",
+			},
 		},
 	}
 }
diff --git a/provider/workspace_test.go b/provider/workspace_test.go
@@ -3,16 +3,18 @@ package provider_test
 import (
 	"testing"
 
-	"github.com/coder/terraform-provider-coder/provider"
 	"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 TestWorkspace(t *testing.T) {
 	t.Setenv("CODER_WORKSPACE_OWNER", "owner123")
 	t.Setenv("CODER_WORKSPACE_OWNER_EMAIL", "owner123@example.com")
+	t.Setenv("CODER_SESSION_TOKEN", "abc123")
 
 	resource.Test(t, resource.TestCase{
 		Providers: map[string]*schema.Provider{
@@ -39,6 +41,7 @@ func TestWorkspace(t *testing.T) {
 				require.Equal(t, "8080", attribs["access_port"])
 				require.Equal(t, "owner123", attribs["owner"])
 				require.Equal(t, "owner123@example.com", attribs["owner_email"])
+				require.Equal(t, "abc123", attribs["coder_session_token"])
 				return nil
 			},
 		}},
