feat: Add "projects list" command to the CLI (#333)

This adds a WorkspaceOwnerCount parameter returned from the
projects API. It's helpful to display the amount of usage
a specific project has.

database/databasefake/databasefake.go

@@ -195,6 +195,41 @@ func (q *fakeQuerier) GetWorkspaceByUserIDAndName(_ context.Context, arg databas
 	return database.Workspace{}, sql.ErrNoRows
 }
 
+func (q *fakeQuerier) GetWorkspaceOwnerCountsByProjectIDs(_ context.Context, projectIDs []uuid.UUID) ([]database.GetWorkspaceOwnerCountsByProjectIDsRow, error) {
+	counts := map[string]map[string]struct{}{}
+	for _, projectID := range projectIDs {
+		found := false
+		for _, workspace := range q.workspace {
+			if workspace.ProjectID.String() != projectID.String() {
+				continue
+			}
+			countByOwnerID, ok := counts[projectID.String()]
+			if !ok {
+				countByOwnerID = map[string]struct{}{}
+			}
+			countByOwnerID[workspace.OwnerID] = struct{}{}
+			counts[projectID.String()] = countByOwnerID
+			found = true
+			break
+		}
+		if !found {
+			counts[projectID.String()] = map[string]struct{}{}
+		}
+	}
+	res := make([]database.GetWorkspaceOwnerCountsByProjectIDsRow, 0)
+	for key, value := range counts {
+		uid := uuid.MustParse(key)
+		res = append(res, database.GetWorkspaceOwnerCountsByProjectIDsRow{
+			ProjectID: uid,
+			Count:     int64(len(value)),
+		})
+	}
+	if len(res) == 0 {
+		return nil, sql.ErrNoRows
+	}
+	return res, nil
+}
+
 func (q *fakeQuerier) GetWorkspaceResourcesByHistoryID(_ context.Context, workspaceHistoryID uuid.UUID) ([]database.WorkspaceResource, error) {
 	q.mutex.Lock()
 	defer q.mutex.Unlock()

database/querier.go

@@ -39,6 +39,7 @@ type querier interface {
 	GetWorkspaceHistoryByWorkspaceID(ctx context.Context, workspaceID uuid.UUID) ([]WorkspaceHistory, error)
 	GetWorkspaceHistoryByWorkspaceIDAndName(ctx context.Context, arg GetWorkspaceHistoryByWorkspaceIDAndNameParams) (WorkspaceHistory, error)
 	GetWorkspaceHistoryByWorkspaceIDWithoutAfter(ctx context.Context, workspaceID uuid.UUID) (WorkspaceHistory, error)
+	GetWorkspaceOwnerCountsByProjectIDs(ctx context.Context, ids []uuid.UUID) ([]GetWorkspaceOwnerCountsByProjectIDsRow, error)
 	GetWorkspaceResourcesByHistoryID(ctx context.Context, workspaceHistoryID uuid.UUID) ([]WorkspaceResource, error)
 	GetWorkspacesByProjectAndUserID(ctx context.Context, arg GetWorkspacesByProjectAndUserIDParams) ([]Workspace, error)
 	GetWorkspacesByUserID(ctx context.Context, ownerID string) ([]Workspace, error)

database/query.sql

@@ -278,6 +278,18 @@ WHERE
   owner_id = $1
   AND project_id = $2;
 
+-- name: GetWorkspaceOwnerCountsByProjectIDs :many
+SELECT
+  project_id,
+  COUNT(DISTINCT owner_id)
+FROM
+  workspace
+WHERE
+  project_id = ANY(@ids :: uuid [ ])
+GROUP BY
+  project_id,
+  owner_id;
+
 -- name: GetWorkspaceHistoryByID :one
 SELECT
   *

database/query.sql.go

@@ -1070,6 +1070,47 @@ func (q *sqlQuerier) GetWorkspaceHistoryByWorkspaceIDWithoutAfter(ctx context.Co
 	return i, err
 }
 
+const getWorkspaceOwnerCountsByProjectIDs = `-- name: GetWorkspaceOwnerCountsByProjectIDs :many
+SELECT
+  project_id,
+  COUNT(DISTINCT owner_id)
+FROM
+  workspace
+WHERE
+  project_id = ANY($1 :: uuid [ ])
+GROUP BY
+  project_id,
+  owner_id
+`
+
+type GetWorkspaceOwnerCountsByProjectIDsRow struct {
+	ProjectID uuid.UUID `db:"project_id" json:"project_id"`
+	Count     int64     `db:"count" json:"count"`
+}
+
+func (q *sqlQuerier) GetWorkspaceOwnerCountsByProjectIDs(ctx context.Context, ids []uuid.UUID) ([]GetWorkspaceOwnerCountsByProjectIDsRow, error) {
+	rows, err := q.db.QueryContext(ctx, getWorkspaceOwnerCountsByProjectIDs, pq.Array(ids))
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []GetWorkspaceOwnerCountsByProjectIDsRow
+	for rows.Next() {
+		var i GetWorkspaceOwnerCountsByProjectIDsRow
+		if err := rows.Scan(&i.ProjectID, &i.Count); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Close(); err != nil {
+		return nil, err
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
 const getWorkspaceResourcesByHistoryID = `-- name: GetWorkspaceResourcesByHistoryID :many
 SELECT
   id, created_at, workspace_history_id, type, name, workspace_agent_token, workspace_agent_id