feat: add display_apps field to agent

coder · sreya · Aug 30, 2023 · Aug 15, 2023 · Aug 28, 2023 · Aug 29, 2023
commit a71e3c8ed08a5cdeb7870b99ab1a720672c4a2af
diff --git a/coderd/apidoc/docs.go b/coderd/apidoc/docs.go
diff --git a/coderd/apidoc/swagger.json b/coderd/apidoc/swagger.json
diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go
@@ -4428,6 +4428,7 @@ func (q *FakeQuerier) InsertWorkspaceAgent(_ context.Context, arg database.Inser
 		MOTDFile:                 arg.MOTDFile,
 		LifecycleState:           database.WorkspaceAgentLifecycleStateCreated,
 		ShutdownScript:           arg.ShutdownScript,
+		DisplayApps:              arg.DisplayApps,
 	}
 
 	q.workspaceAgents = append(q.workspaceAgents, agent)

diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/migrations/000152_agent_default_apps.down.sql b/coderd/database/migrations/000152_agent_default_apps.down.sql
@@ -0,0 +1,5 @@
+BEGIN;
+ALTER TABLE workspace_agents DROP COLUMN display_apps;
+DROP TYPE display_app; 
+COMMIT;
+
diff --git a/coderd/database/migrations/000152_agent_default_apps.up.sql b/coderd/database/migrations/000152_agent_default_apps.up.sql
@@ -0,0 +1,4 @@
+BEGIN;
+CREATE TYPE display_app AS ENUM ('vscode', 'vscode_insiders', 'web_terminal', 'ssh_helper', 'port_forwarding_helper');
+ALTER TABLE workspace_agents ADD column display_apps display_app[] DEFAULT '{vscode, vscode_insiders, web_terminal, ssh_helper, port_forwarding_helper}';
+COMMIT;
diff --git a/coderd/database/modelmethods.go b/coderd/database/modelmethods.go
@@ -10,6 +10,8 @@ import (
 	"github.com/coder/coder/v2/coderd/rbac"
 )
 
+var DefaultDisplayApps = []DisplayApp{DisplayAppVscode, DisplayAppWebTerminal, DisplayAppSSHHelper, DisplayAppPortForwardingHelper}
+
 type WorkspaceStatus string
 
 const (

diff --git a/coderd/database/models.go b/coderd/database/models.go
-Original file line number
+Diff line change
@@ Expand Up / @@ -10,6 +10,8 @@ import ( @@
     	"github.com/coder/coder/v2/coderd/rbac"
     )
+    var DefaultDisplayApps = []DisplayApp{DisplayAppVscode, DisplayAppWebTerminal, DisplayAppSSHHelper, DisplayAppPortForwardingHelper}
     type WorkspaceStatus string
     const (
@@ Expand Down @@