feat(coderd): add tasks list and get endpoints

Fixes coder/internal#899
coder · mafredri · Aug 22, 2025 · Aug 21, 2025 · Aug 21, 2025 · Aug 22, 2025
commit bd14dddd00c964c31b009a9c2a4c8a7fe8eefe6e
diff --git a/coderd/aitasks.go b/coderd/aitasks.go
@@ -8,6 +8,7 @@ import (
 	"slices"
 	"strings"
 
+	"github.com/go-chi/chi/v5"
 	"github.com/google/uuid"
 
 	"cdr.dev/slog"
@@ -17,6 +18,8 @@ import (
 	"github.com/coder/coder/v2/coderd/httpapi"
 	"github.com/coder/coder/v2/coderd/httpmw"
 	"github.com/coder/coder/v2/coderd/rbac"
+	"github.com/coder/coder/v2/coderd/rbac/policy"
+	"github.com/coder/coder/v2/coderd/searchquery"
 	"github.com/coder/coder/v2/coderd/taskname"
 	"github.com/coder/coder/v2/codersdk"
 )
@@ -186,3 +189,267 @@ func (api *API) tasksCreate(rw http.ResponseWriter, r *http.Request) {
 	defer commitAudit()
 	createWorkspace(ctx, aReq, apiKey.UserID, api, owner, createReq, rw, r)
 }
+
+// tasksListResponse wraps a list of experimental tasks.
+//
+// Experimental: Response shape is experimental and may change.
+type tasksListResponse struct {
+	Tasks []codersdk.Task `json:"tasks"`
+	Count int             `json:"count"`
+}
+
+func mapTaskStatus(ws codersdk.Workspace) codersdk.TaskStatus {
+	if ws.LatestAppStatus != nil {
+		switch ws.LatestAppStatus.State {
+		case codersdk.WorkspaceAppStatusStateWorking:
+			return codersdk.TaskStatusWorking
+		case codersdk.WorkspaceAppStatusStateIdle:
+			return codersdk.TaskStatusIdle
+		case codersdk.WorkspaceAppStatusStateComplete:
+			return codersdk.TaskStatusCompleted
+		case codersdk.WorkspaceAppStatusStateFailure:
+			return codersdk.TaskStatusFailed
+		}
+	}
+
+	switch ws.LatestBuild.Status {
+	case codersdk.WorkspaceStatusPending, codersdk.WorkspaceStatusStarting, codersdk.WorkspaceStatusRunning:
+		return codersdk.TaskStatusWorking
+	case codersdk.WorkspaceStatusStopping, codersdk.WorkspaceStatusStopped, codersdk.WorkspaceStatusDeleting, codersdk.WorkspaceStatusDeleted:
+		return codersdk.TaskStatusCompleted
+	case codersdk.WorkspaceStatusFailed, codersdk.WorkspaceStatusCanceling, codersdk.WorkspaceStatusCanceled:
+		return codersdk.TaskStatusFailed
+	default:
+		return codersdk.TaskStatusWorking
+	}
+}
+
+// tasksList is an experimental endpoint to list AI tasks by mapping
+// workspaces to a task-shaped response.
+func (api *API) tasksList(rw http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	apiKey := httpmw.APIKey(r)
+
+	// Support standard pagination/filters for workspaces.
+	page, ok := ParsePagination(rw, r)
+	if !ok {
+		return
+	}
+	queryStr := r.URL.Query().Get("q")
+	filter, errs := searchquery.Workspaces(ctx, api.Database, queryStr, page, api.AgentInactiveDisconnectTimeout)
+	if len(errs) > 0 {
+		httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{
+			Message:     "Invalid workspace search query.",
+			Validations: errs,
+		})
+		return
+	}
+
+	// Ensure that we only include AI task workspaces in the results.
+	filter.HasAITask = sql.NullBool{Valid: true, Bool: true}
+
+	if filter.OwnerUsername == "me" || filter.OwnerUsername == "" {
+		filter.OwnerID = apiKey.UserID
+		filter.OwnerUsername = ""
+	}
+
+	prepared, err := api.HTTPAuth.AuthorizeSQLFilter(r, policy.ActionRead, rbac.ResourceWorkspace.Type)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error preparing sql filter.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+
+	// Order with requester's favorites first, include summary row.
+	filter.RequesterID = apiKey.UserID
+	filter.WithSummary = true
+
+	workspaceRows, err := api.Database.GetAuthorizedWorkspaces(ctx, filter, prepared)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching workspaces.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+	if len(workspaceRows) == 0 {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching workspaces.",
+			Detail:  "Workspace summary row is missing.",
+		})
+		return
+	}
+	if len(workspaceRows) == 1 {
+		httpapi.Write(ctx, rw, http.StatusOK, tasksListResponse{
+			Tasks: []codersdk.Task{},
+			Count: 0,
+		})
+		return
+	}
+
+	// Skip summary row.
+	workspaceRows = workspaceRows[:len(workspaceRows)-1]
+
+	workspaces := database.ConvertWorkspaceRows(workspaceRows)
+
+	// Gather associated data and convert to API workspaces.
+	data, err := api.workspaceData(ctx, workspaces)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching workspace resources.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+	apiWorkspaces, err := convertWorkspaces(apiKey.UserID, workspaces, data)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error converting workspaces.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+
+	// Fetch prompts for each workspace build and map by build ID.
+	buildIDs := make([]uuid.UUID, 0, len(apiWorkspaces))
+	for _, ws := range apiWorkspaces {
+		buildIDs = append(buildIDs, ws.LatestBuild.ID)
+	}
+	parameters, err := api.Database.GetWorkspaceBuildParametersByBuildIDs(ctx, buildIDs)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching task prompts.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+	promptsByBuildID := make(map[uuid.UUID]string, len(parameters))
+	for _, p := range parameters {
+		if p.Name == codersdk.AITaskPromptParameterName {
+			promptsByBuildID[p.WorkspaceBuildID] = p.Value
+		}
+	}
+
+	tasks := make([]codersdk.Task, 0, len(apiWorkspaces))
+	for _, ws := range apiWorkspaces {
+		tasks = append(tasks, codersdk.Task{
+			ID:             ws.ID,
+			OrganizationID: ws.OrganizationID,
+			OwnerID:        ws.OwnerID,
+			Name:           ws.Name,
+			TemplateID:     ws.TemplateID,
+			WorkspaceID:    uuid.NullUUID{Valid: true, UUID: ws.ID},
+			Prompt:         promptsByBuildID[ws.LatestBuild.ID],
+			Status:         mapTaskStatus(ws),
+			CreatedAt:      ws.CreatedAt,
+			UpdatedAt:      ws.UpdatedAt,
+		})
+	}
+
+	httpapi.Write(ctx, rw, http.StatusOK, tasksListResponse{
+		Tasks: tasks,
+		Count: len(tasks),
+	})
+}
+
+// taskGet is an experimental endpoint to fetch a single AI task by ID
+// (workspace ID). It returns a synthesized task response including
+// prompt and status.
+func (api *API) taskGet(rw http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	apiKey := httpmw.APIKey(r)
+
+	idStr := chi.URLParam(r, "id")
+	taskID, err := uuid.Parse(idStr)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{
+			Message: fmt.Sprintf("Invalid UUID %q for task ID.", idStr),
+		})
+		return
+	}
+
+	workspace, err := api.Database.GetWorkspaceByID(ctx, taskID)
+	if httpapi.Is404Error(err) {
+		httpapi.ResourceNotFound(rw)
+		return
+	}
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching workspace.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+
+	data, err := api.workspaceData(ctx, []database.Workspace{workspace})
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching workspace resources.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+	if len(data.builds) == 0 || len(data.templates) == 0 {
+		httpapi.ResourceNotFound(rw)
+		return
+	}
+	if data.builds[0].HasAITask == nil || !*data.builds[0].HasAITask {
+		httpapi.ResourceNotFound(rw)
+		return
+	}
+
+	appStatus := codersdk.WorkspaceAppStatus{}
+	if len(data.appStatuses) > 0 {
+		appStatus = data.appStatuses[0]
+	}
+
+	ws, err := convertWorkspace(
+		apiKey.UserID,
+		workspace,
+		data.builds[0],
+		data.templates[0],
+		api.Options.AllowWorkspaceRenames,
+		appStatus,
+	)
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error converting workspace.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+
+	// Fetch the AI prompt from the build parameters.
+	params, err := api.Database.GetWorkspaceBuildParametersByBuildIDs(ctx, []uuid.UUID{ws.LatestBuild.ID})
+	if err != nil {
+		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
+			Message: "Internal error fetching task prompt.",
+			Detail:  err.Error(),
+		})
+		return
+	}
+	prompt := ""
+	for _, p := range params {
+		if p.Name == codersdk.AITaskPromptParameterName {
+			prompt = p.Value
+			break
+		}
+	}
+
+	resp := codersdk.Task{
+		ID:             ws.ID,
+		OrganizationID: ws.OrganizationID,
+		OwnerID:        ws.OwnerID,
+		Name:           ws.Name,
+		TemplateID:     ws.TemplateID,
+		WorkspaceID:    uuid.NullUUID{Valid: true, UUID: ws.ID},
+		Prompt:         prompt,
+		Status:         mapTaskStatus(ws),
+		CreatedAt:      ws.CreatedAt,
+		UpdatedAt:      ws.UpdatedAt,
+	}
+
+	httpapi.Write(ctx, rw, http.StatusOK, resp)
+}