coder · Kira-Pilot · Mar 30, 2023 · Apr 10, 2023 · Apr 10, 2023 · Apr 10, 2023
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/databasefake.go b/coderd/database/dbfake/databasefake.go
@@ -1255,6 +1255,22 @@ func (q *fakeQuerier) GetAuthorizedWorkspaces(ctx context.Context, arg database.
 					continue
 				}
 
+			case database.WorkspaceStatusRestarted:
+				if !job.CompletedAt.Valid &&
+					job.CanceledAt.Valid &&
+					job.Error.Valid &&
+					build.Transition != database.WorkspaceTransitionRestart {
+					continue
+				}
+
+			case database.WorkspaceStatusRestarting:
+				if !job.CompletedAt.Valid &&
+					job.CanceledAt.Valid &&
+					job.Error.Valid &&
+					build.Transition != database.WorkspaceTransitionRestart {
+					continue
+				}
+
 			default:
 				return nil, xerrors.Errorf("unknown workspace status in filter: %q", arg.Status)
 			}

diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/migrations/000117_add_workspace_transition_restart.down.sql b/coderd/database/migrations/000117_add_workspace_transition_restart.down.sql
@@ -0,0 +1,2 @@
+-- It's not possible to drop enum values from enum types, so the UP has "IF NOT
+-- EXISTS".
diff --git a/coderd/database/migrations/000117_add_workspace_transition_restart.up.sql b/coderd/database/migrations/000117_add_workspace_transition_restart.up.sql
@@ -0,0 +1 @@
+ALTER TYPE workspace_transition ADD VALUE IF NOT EXISTS 'restart';
diff --git a/coderd/database/modelmethods.go b/coderd/database/modelmethods.go
@@ -13,24 +13,26 @@ import (
 type WorkspaceStatus string
 
 const (
-	WorkspaceStatusPending   WorkspaceStatus = "pending"
-	WorkspaceStatusStarting  WorkspaceStatus = "starting"
-	WorkspaceStatusRunning   WorkspaceStatus = "running"
-	WorkspaceStatusStopping  WorkspaceStatus = "stopping"
-	WorkspaceStatusStopped   WorkspaceStatus = "stopped"
-	WorkspaceStatusFailed    WorkspaceStatus = "failed"
-	WorkspaceStatusCanceling WorkspaceStatus = "canceling"
-	WorkspaceStatusCanceled  WorkspaceStatus = "canceled"
-	WorkspaceStatusDeleting  WorkspaceStatus = "deleting"
-	WorkspaceStatusDeleted   WorkspaceStatus = "deleted"
+	WorkspaceStatusPending    WorkspaceStatus = "pending"
+	WorkspaceStatusStarting   WorkspaceStatus = "starting"
+	WorkspaceStatusRunning    WorkspaceStatus = "running"
+	WorkspaceStatusStopping   WorkspaceStatus = "stopping"
+	WorkspaceStatusStopped    WorkspaceStatus = "stopped"
+	WorkspaceStatusFailed     WorkspaceStatus = "failed"
+	WorkspaceStatusCanceling  WorkspaceStatus = "canceling"
+	WorkspaceStatusCanceled   WorkspaceStatus = "canceled"
+	WorkspaceStatusDeleting   WorkspaceStatus = "deleting"
+	WorkspaceStatusDeleted    WorkspaceStatus = "deleted"
+	WorkspaceStatusRestarting WorkspaceStatus = "restarting"
+	WorkspaceStatusRestarted  WorkspaceStatus = "restarted"
 )
 
 func (s WorkspaceStatus) Valid() bool {
 	switch s {
 	case WorkspaceStatusPending, WorkspaceStatusStarting, WorkspaceStatusRunning,
 		WorkspaceStatusStopping, WorkspaceStatusStopped, WorkspaceStatusFailed,
 		WorkspaceStatusCanceling, WorkspaceStatusCanceled, WorkspaceStatusDeleting,
-		WorkspaceStatusDeleted:
+		WorkspaceStatusDeleted, WorkspaceStatusRestarting, WorkspaceStatusRestarted:
 		return true
 	default:
 		return false

diff --git a/coderd/database/models.go b/coderd/database/models.go
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		-- It's not possible to drop enum values from enum types, so the UP has "IF NOT
		-- EXISTS".
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		ALTER TYPE workspace_transition ADD VALUE IF NOT EXISTS 'restart';