Changeset 0.29.0 (#195)

server/store/postgres/workflows.go

@@ -2,10 +2,13 @@ package postgres
 
 import (
 	"context"
+	"database/sql"
 	"encoding/json"
 	"fmt"
+	"time"
 
 	"git.gobha.me/xcaliber/chat-switchboard/models"
+	"git.gobha.me/xcaliber/chat-switchboard/store"
 )
 
 // WorkflowStore implements store.WorkflowStore for Postgres.
@@ -357,3 +360,129 @@ func nullIfEmpty(s string) interface{} {
 	}
 	return s
 }
+
+// ── Assignments (v0.29.0-cs3) ───────────────────────────────────────────
+
+func (s *WorkflowStore) CreateAssignment(ctx context.Context, a *store.WorkflowAssignment) error {
+	a.ID = store.NewID()
+	_, err := DB.ExecContext(ctx, `
+		INSERT INTO workflow_assignments (id, channel_id, stage, team_id)
+		VALUES ($1, $2, $3, $4)
+	`, a.ID, a.ChannelID, a.Stage, a.TeamID)
+	return err
+}
+
+func (s *WorkflowStore) ListAssignmentsForTeam(ctx context.Context, teamID, status string) ([]store.WorkflowAssignment, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT id, channel_id, stage, team_id, assigned_to, status,
+		       created_at, claimed_at, completed_at
+		FROM workflow_assignments
+		WHERE team_id = $1 AND status = $2
+		ORDER BY created_at DESC
+	`, teamID, status)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	return scanAssignments(rows)
+}
+
+func (s *WorkflowStore) ListAssignmentsMine(ctx context.Context, userID string) ([]store.WorkflowAssignment, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT DISTINCT wa.id, wa.channel_id, wa.stage, wa.team_id, wa.assigned_to, wa.status,
+		       wa.created_at, wa.claimed_at, wa.completed_at
+		FROM workflow_assignments wa
+		LEFT JOIN team_members tm ON tm.team_id = wa.team_id AND tm.user_id = $1
+		WHERE (wa.assigned_to = $2 AND wa.status = 'claimed')
+		   OR (wa.status = 'unassigned' AND tm.user_id IS NOT NULL)
+		ORDER BY wa.created_at DESC
+	`, userID, userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	return scanAssignments(rows)
+}
+
+func (s *WorkflowStore) ClaimAssignment(ctx context.Context, assignmentID, userID string) (int64, error) {
+	res, err := DB.ExecContext(ctx, `
+		UPDATE workflow_assignments
+		SET assigned_to = $1, status = 'claimed', claimed_at = $2
+		WHERE id = $3 AND status = 'unassigned'
+	`, userID, time.Now().UTC(), assignmentID)
+	if err != nil {
+		return 0, err
+	}
+	return res.RowsAffected()
+}
+
+func (s *WorkflowStore) CompleteAssignment(ctx context.Context, assignmentID string) (int64, error) {
+	res, err := DB.ExecContext(ctx, `
+		UPDATE workflow_assignments
+		SET status = 'completed', completed_at = $1
+		WHERE id = $2 AND status = 'claimed'
+	`, time.Now().UTC(), assignmentID)
+	if err != nil {
+		return 0, err
+	}
+	return res.RowsAffected()
+}
+
+func (s *WorkflowStore) GetAssignmentChannelID(ctx context.Context, assignmentID string) (string, error) {
+	var channelID string
+	err := DB.QueryRowContext(ctx,
+		`SELECT channel_id FROM workflow_assignments WHERE id = $1`, assignmentID).Scan(&channelID)
+	return channelID, err
+}
+
+func (s *WorkflowStore) TryRoundRobin(ctx context.Context, teamID, assignmentID string) (string, error) {
+	// Find least-recently-assigned team member
+	rows, err := DB.QueryContext(ctx, `
+		SELECT m.user_id, COALESCE(MAX(wa.claimed_at), '1970-01-01T00:00:00Z') as last_claim
+		FROM team_members m
+		LEFT JOIN workflow_assignments wa ON wa.assigned_to = m.user_id AND wa.team_id = $1
+		WHERE m.team_id = $2
+		GROUP BY m.user_id
+		ORDER BY last_claim ASC
+		LIMIT 1
+	`, teamID, teamID)
+	if err != nil {
+		return "", err
+	}
+	defer rows.Close()
+
+	if !rows.Next() {
+		return "", nil // no team members
+	}
+	var userID, lastClaim string
+	if err := rows.Scan(&userID, &lastClaim); err != nil {
+		return "", err
+	}
+
+	// Claim for that user
+	_, err = DB.ExecContext(ctx, `
+		UPDATE workflow_assignments
+		SET assigned_to = $1, status = 'claimed', claimed_at = $2
+		WHERE id = $3 AND status = 'unassigned'
+	`, userID, time.Now().UTC(), assignmentID)
+	if err != nil {
+		return "", err
+	}
+	return userID, nil
+}
+
+func scanAssignments(rows *sql.Rows) ([]store.WorkflowAssignment, error) {
+	var result []store.WorkflowAssignment
+	for rows.Next() {
+		var a store.WorkflowAssignment
+		if err := rows.Scan(&a.ID, &a.ChannelID, &a.Stage, &a.TeamID,
+			&a.AssignedTo, &a.Status, &a.CreatedAt, &a.ClaimedAt, &a.CompletedAt); err != nil {
+			return nil, err
+		}
+		result = append(result, a)
+	}
+	if result == nil {
+		result = []store.WorkflowAssignment{}
+	}
+	return result, rows.Err()
+}