Add task priority field

Introduce a rank-style priority on tasks, scoped by
(organization_id, state). New tasks auto-assign the next
priority. Reordering uses the same CTE-based algorithm as
trust center references and compliance external URLs.
Exposed through GraphQL, MCP, and the PRIORITY order field.

Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>

Author: SachaProbo

pkg/coredata/migrations/20260326T150000Z.sql

@@ -0,0 +1,19 @@
+ALTER TABLE tasks ADD COLUMN priority INTEGER;
+
+WITH ranked_tasks AS (
+    SELECT
+        id,
+        ROW_NUMBER() OVER (PARTITION BY organization_id, state ORDER BY created_at DESC, id DESC) AS rn
+    FROM tasks
+)
+UPDATE tasks t
+SET priority = rt.rn
+FROM ranked_tasks rt
+WHERE t.id = rt.id;
+
+ALTER TABLE tasks ALTER COLUMN priority SET NOT NULL;
+
+ALTER TABLE tasks
+ADD CONSTRAINT tasks_organization_id_state_priority_key
+    UNIQUE (organization_id, state, priority)
+    DEFERRABLE INITIALLY DEFERRED;

pkg/coredata/task.go

@@ -41,6 +41,7 @@ type (
 		TimeEstimate   *time.Duration `db:"time_estimate"`
 		AssignedToID   *gid.GID       `db:"assigned_to_profile_id"`
 		Deadline       *time.Time     `db:"deadline"`
+		Priority       int            `db:"priority"`
 		CreatedAt      time.Time      `db:"created_at"`
 		UpdatedAt      time.Time      `db:"updated_at"`
 	}
@@ -50,6 +51,8 @@ type (
 
 func (c Task) CursorKey(orderBy TaskOrderField) page.CursorKey {
 	switch orderBy {
+	case TaskOrderFieldPriority:
+		return page.NewCursorKey(c.ID, c.Priority)
 	case TaskOrderFieldCreatedAt:
 		return page.NewCursorKey(c.ID, c.CreatedAt)
 	}
@@ -89,6 +92,7 @@ SELECT
     time_estimate,
     assigned_to_profile_id,
     deadline,
+    priority,
     created_at,
     updated_at
 FROM
@@ -141,6 +145,7 @@ SELECT
     time_estimate,
     assigned_to_profile_id,
     deadline,
+    priority,
     created_at,
     updated_at
 FROM
@@ -170,12 +175,17 @@ WHERE
 	return nil
 }
 
-func (c Task) Insert(
+func (c *Task) Insert(
 	ctx context.Context,
 	conn pg.Conn,
 	scope Scoper,
 ) error {
 	q := `
+WITH next_priority AS (
+    SELECT COALESCE(MAX(priority), 0) + 1 AS value
+    FROM tasks
+    WHERE organization_id = @organization_id AND state = @state
+)
 INSERT INTO
     tasks (
         tenant_id,
@@ -189,6 +199,7 @@ INSERT INTO
         time_estimate,
         assigned_to_profile_id,
         deadline,
+        priority,
         created_at,
         updated_at
     )
@@ -204,9 +215,11 @@ VALUES (
     @time_estimate,
     @assigned_to_profile_id,
     @deadline,
+    (SELECT value FROM next_priority),
     @created_at,
     @updated_at
-);
+)
+RETURNING priority;
 `
 
 	args := pgx.StrictNamedArgs{
@@ -224,8 +237,8 @@ VALUES (
 		"created_at":             c.CreatedAt,
 		"updated_at":             c.UpdatedAt,
 	}
-	_, err := conn.Exec(ctx, q, args)
 
+	err := conn.QueryRow(ctx, q, args).Scan(&c.Priority)
 	if err != nil {
 		var pgErr *pgconn.PgError
 		if errors.As(err, &pgErr) {
@@ -245,6 +258,11 @@ func (c *Task) Upsert(
 	scope Scoper,
 ) error {
 	q := `
+WITH next_priority AS (
+    SELECT COALESCE(MAX(priority), 0) + 1 AS value
+    FROM tasks
+    WHERE organization_id = @organization_id AND state = @state
+)
 INSERT INTO
     tasks (
         tenant_id,
@@ -258,6 +276,7 @@ INSERT INTO
         time_estimate,
         assigned_to_profile_id,
         deadline,
+        priority,
         created_at,
         updated_at
     )
@@ -273,6 +292,7 @@ VALUES (
     @time_estimate,
     @assigned_to_profile_id,
     @deadline,
+    (SELECT value FROM next_priority),
     @created_at,
     @updated_at
 )
@@ -292,6 +312,7 @@ RETURNING
     time_estimate,
     assigned_to_profile_id,
     deadline,
+    priority,
     created_at,
     updated_at
 `
@@ -377,6 +398,7 @@ func (c *Tasks) LoadByOrganizationID(
 		time_estimate,
 		assigned_to_profile_id,
 		deadline,
+		priority,
 		created_at,
 		updated_at
 	FROM
@@ -458,6 +480,7 @@ SELECT
     time_estimate,
     assigned_to_profile_id,
     deadline,
+    priority,
     created_at,
     updated_at
 FROM
@@ -525,6 +548,59 @@ WHERE %s
 	return err
 }
 
+func (c *Task) UpdatePriority(
+	ctx context.Context,
+	conn pg.Conn,
+	scope Scoper,
+) error {
+	q := `
+WITH old AS (
+  SELECT
+	priority AS old_priority
+  FROM tasks
+  WHERE %s AND id = @id AND organization_id = @organization_id AND state = @state
+)
+
+UPDATE tasks
+SET
+    priority = CASE
+        WHEN id = @id THEN @new_priority
+        ELSE priority + CASE
+            WHEN @new_priority < old.old_priority THEN 1
+            WHEN @new_priority > old.old_priority THEN -1
+        END
+    END,
+    updated_at = @updated_at
+FROM old
+WHERE %s
+  AND organization_id = @organization_id
+  AND state = @state
+  AND (
+    id = @id
+    OR (priority BETWEEN LEAST(old.old_priority, @new_priority) AND GREATEST(old.old_priority, @new_priority))
+  );
+`
+
+	scopeFragment := scope.SQLFragment()
+	q = fmt.Sprintf(q, scopeFragment, scopeFragment)
+
+	args := pgx.StrictNamedArgs{
+		"id":              c.ID,
+		"new_priority":    c.Priority,
+		"organization_id": c.OrganizationID,
+		"state":           c.State,
+		"updated_at":      c.UpdatedAt,
+	}
+	maps.Copy(args, scope.SQLArguments())
+
+	_, err := conn.Exec(ctx, q, args)
+	if err != nil {
+		return fmt.Errorf("cannot update task priority: %w", err)
+	}
+
+	return nil
+}
+
 func (c *Task) Delete(
 	ctx context.Context,
 	conn pg.Conn,

pkg/coredata/task_order_field.go

@@ -19,11 +19,19 @@ type (
 )
 
 const (
+	TaskOrderFieldPriority  TaskOrderField = "PRIORITY"
 	TaskOrderFieldCreatedAt TaskOrderField = "CREATED_AT"
 )
 
 func (p TaskOrderField) Column() string {
-	return string(p)
+	switch p {
+	case TaskOrderFieldPriority:
+		return "priority"
+	case TaskOrderFieldCreatedAt:
+		return "created_at"
+	default:
+		return string(p)
+	}
 }
 
 func (p TaskOrderField) String() string {

pkg/probo/task_service.go

@@ -51,6 +51,7 @@ type (
 		Deadline     **time.Time
 		AssignedToID **gid.GID
 		MeasureID    **gid.GID
+		Priority     *int
 	}
 )
 
@@ -319,6 +320,13 @@ func (s TaskService) Update(
 
 			task.UpdatedAt = time.Now()
 
+			if req.Priority != nil {
+				task.Priority = *req.Priority
+				if err := task.UpdatePriority(ctx, conn, s.svc.scope); err != nil {
+					return fmt.Errorf("cannot update task priority: %w", err)
+				}
+			}
+
 			if err := task.Update(ctx, conn, s.svc.scope); err != nil {
 				return fmt.Errorf("cannot update task: %w", err)
 			}

pkg/server/api/console/v1/schema.graphql

@@ -499,6 +499,7 @@ enum MeasureOrderField
 
 enum TaskOrderField
     @goModel(model: "go.probo.inc/probo/pkg/coredata.TaskOrderField") {
+    PRIORITY
     CREATED_AT
 }
 
@@ -2392,6 +2393,7 @@ type Task implements Node {
     name: String!
     description: String
     state: TaskState!
+    priority: Int!
     timeEstimate: Duration
     deadline: Datetime
     assignedTo: Profile @goField(forceResolver: true)
@@ -4211,6 +4213,7 @@ input UpdateTaskInput {
     name: String
     description: String @goField(omittable: true)
     state: TaskState
+    priority: Int
     timeEstimate: Duration @goField(omittable: true)
     deadline: Datetime @goField(omittable: true)
     assignedToId: ID @goField(omittable: true)

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

@@ -70,6 +70,7 @@ func NewTask(t *coredata.Task) *Task {
 		Name:         t.Name,
 		Description:  t.Description,
 		State:        t.State,
+		Priority:     t.Priority,
 		TimeEstimate: t.TimeEstimate,
 		Deadline:     t.Deadline,
 		CreatedAt:    t.CreatedAt,

pkg/server/api/console/v1/v1_resolver.go

@@ -3766,6 +3766,7 @@ func (r *mutationResolver) UpdateTask(ctx context.Context, input types.UpdateTas
 			Name:         input.Name,
 			Description:  gqlutils.UnwrapOmittable(input.Description),
 			State:        input.State,
+			Priority:     input.Priority,
 			TimeEstimate: gqlutils.UnwrapOmittable(input.TimeEstimate),
 			Deadline:     gqlutils.UnwrapOmittable(input.Deadline),
 			AssignedToID: gqlutils.UnwrapOmittable(input.AssignedToID),

pkg/server/api/mcp/v1/schema.resolvers.go

@@ -1904,6 +1904,7 @@ func (r *Resolver) UpdateTaskTool(ctx context.Context, req *mcp.CallToolRequest,
 			Name:         input.Name,
 			Description:  UnwrapOmittable(input.Description),
 			State:        input.State,
+			Priority:     input.Priority,
 			TimeEstimate: UnwrapOmittable(input.TimeEstimate),
 			Deadline:     UnwrapOmittable(input.Deadline),
 			AssignedToID: UnwrapOmittable(input.AssignedToID),

pkg/server/api/mcp/v1/specification.yaml

@@ -4619,6 +4619,7 @@ components:
     TaskOrderField:
       type: string
       enum:
+        - PRIORITY
         - CREATED_AT
       go.probo.inc/mcpgen/type: go.probo.inc/probo/pkg/coredata.TaskOrderField
 
@@ -4642,6 +4643,7 @@ components:
         - organization_id
         - name
         - state
+        - priority
         - created_at
         - updated_at
       properties:
@@ -4671,6 +4673,9 @@ components:
         state:
           $ref: "#/components/schemas/TaskState"
           description: Task state
+        priority:
+          type: integer
+          description: Task priority within state
         time_estimate:
           anyOf:
             - $ref: "#/components/schemas/Duration"
@@ -4822,6 +4827,9 @@ components:
             - type: "null"
               description: No state
           description: Task state
+        priority:
+          type: integer
+          description: Task priority within state
         time_estimate:
           anyOf:
             - $ref: "#/components/schemas/Duration"

pkg/server/api/mcp/v1/types/task.go

@@ -25,6 +25,7 @@ func NewTask(t *coredata.Task) *Task {
 		Name:         t.Name,
 		Description:  t.Description,
 		State:        t.State,
+		Priority:     t.Priority,
 		TimeEstimate: t.TimeEstimate,
 		CreatedAt:    t.CreatedAt,
 		UpdatedAt:    t.UpdatedAt,