Add task resolver

Signed-off-by: gearnode <bryan@frimin.fr>

pkg/api/console/v1/types/task.go

@@ -0,0 +1,30 @@
+package types
+
+import (
+	"github.com/getprobo/probo/pkg/probo/coredata"
+	"github.com/getprobo/probo/pkg/probo/coredata/page"
+)
+
+func NewTaskConnection(p *page.Page[*coredata.Task]) *TaskConnection {
+	var edges = make([]*TaskEdge, len(p.Data))
+
+	for i := range edges {
+		edges[i] = NewTaskEdge(p.Data[i])
+	}
+
+	return &TaskConnection{
+		Edges:    edges,
+		PageInfo: NewPageInfo(p),
+	}
+}
+
+func NewTaskEdge(f *coredata.Task) *TaskEdge {
+	return &TaskEdge{
+		Cursor: f.CursorKey(),
+		Node:   NewTask(f),
+	}
+}
+
+func NewTask(e *coredata.Task) *Task {
+	return &Task{}
+}

pkg/api/console/v1/v1_resolver.go

@@ -15,7 +15,14 @@ import (
 
 // Tasks is the resolver for the tasks field.
 func (r *controlResolver) Tasks(ctx context.Context, obj *types.Control, first *int, after *page.CursorKey, last *int, before *page.CursorKey) (*types.TaskConnection, error) {
-	panic(fmt.Errorf("not implemented: Tasks - tasks"))
+	cursor := types.NewCursor(first, after, last, before)
+
+	page, err := r.svc.ListControlTasks(ctx, obj.ID, cursor)
+	if err != nil {
+		return nil, fmt.Errorf("cannot list framework controls: %w", err)
+	}
+
+	return types.NewTaskConnection(page), nil
 }
 
 // Controls is the resolver for the controls field.

pkg/probo/coredata/task.go

@@ -1,12 +1,89 @@
 package coredata
 
-import "time"
+import (
+	"context"
+	"fmt"
+	"maps"
+	"time"
+
+	"github.com/getprobo/probo/pkg/probo/coredata/page"
+	"github.com/jackc/pgx/v5"
+	"go.gearno.de/crypto/uuid"
+	"go.gearno.de/kit/pg"
+)
 
 type (
 	Task struct {
-		ID        string
+		ID         uuid.UUID
-		ContentID string
+		ControlID  string
-		CreatedAt time.Time
+		ContentRef string
-		UpdatedAt time.Time
+		CreatedAt  time.Time
+		UpdatedAt  time.Time
 	}
+
+	Tasks []*Task
 )
+
+func (t Task) CursorKey() page.CursorKey {
+	return page.NewCursorKey(t.ID, t.CreatedAt)
+}
+
+func (t *Task) scan(r pgx.Row) error {
+	return r.Scan(
+		&t.ID,
+		&t.ControlID,
+		&t.ContentRef,
+		&t.CreatedAt,
+		&t.UpdatedAt,
+	)
+}
+
+func (t *Tasks) LoadByControlID(
+	ctx context.Context,
+	conn pg.Conn,
+	controlID string,
+	cursor *page.Cursor,
+) error {
+	q := `
+SELECT
+    task_id,
+    control_id,
+    content_ref,
+    created_at,
+    updated_at
+FROM
+    tasks
+WHERE
+    control_id = @control_id
+    AND %s
+`
+
+	q = fmt.Sprintf(q, cursor.SQLFragment())
+
+	args := pgx.NamedArgs{"control_id": controlID}
+	maps.Copy(args, cursor.SQLArguments())
+
+	r, err := conn.Query(ctx, q, args)
+	if err != nil {
+		return err
+	}
+	defer r.Close()
+
+	tasks := Tasks{}
+	for r.Next() {
+		task := &Task{}
+		if err := task.scan(r); err != nil {
+			return err
+		}
+
+		tasks = append(tasks, task)
+	}
+
+	if err := r.Err(); err != nil {
+		return err
+	}
+
+	*t = tasks
+
+	return nil
+}

pkg/probo/probo.go

@@ -81,3 +81,24 @@ func (s *Service) ListFrameworkControls(
 
 	return page.NewPage(controls, cursor), nil
 }
+
+func (s *Service) ListControlTasks(
+	ctx context.Context,
+	controlID string,
+	cursor *page.Cursor,
+) (*page.Page[*coredata.Task], error) {
+	var tasks coredata.Tasks
+
+	err := s.pg.WithConn(
+		ctx,
+		func(conn pg.Conn) error {
+			return tasks.LoadByControlID(ctx, conn, controlID, cursor)
+		},
+	)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return page.NewPage(tasks, cursor), nil
+}