Changeset 0.29.0 (#195)

server/store/sqlite/workflows.go

@@ -364,3 +364,130 @@ 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 (?, ?, ?, ?)
+	`, 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 = ? AND status = ?
+		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 = ?
+		WHERE (wa.assigned_to = ? 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 = ?, status = 'claimed', claimed_at = ?
+		WHERE id = ? AND status = 'unassigned'
+	`, userID, time.Now().UTC().Format(timeFmt), 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 = ?
+		WHERE id = ? AND status = 'claimed'
+	`, time.Now().UTC().Format(timeFmt), 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 = ?`, assignmentID).Scan(&channelID)
+	return channelID, err
+}
+
+func (s *WorkflowStore) TryRoundRobin(ctx context.Context, teamID, assignmentID string) (string, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT m.user_id, COALESCE(MAX(wa.claimed_at), '1970-01-01 00:00:00') as last_claim
+		FROM team_members m
+		LEFT JOIN workflow_assignments wa ON wa.assigned_to = m.user_id AND wa.team_id = ?
+		WHERE m.team_id = ?
+		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
+	}
+	var userID, lastClaim string
+	if err := rows.Scan(&userID, &lastClaim); err != nil {
+		return "", err
+	}
+
+	_, err = DB.ExecContext(ctx, `
+		UPDATE workflow_assignments
+		SET assigned_to = ?, status = 'claimed', claimed_at = ?
+		WHERE id = ? AND status = 'unassigned'
+	`, userID, time.Now().UTC().Format(timeFmt), 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
+		var claimedAt, completedAt *time.Time
+		if err := rows.Scan(&a.ID, &a.ChannelID, &a.Stage, &a.TeamID,
+			&a.AssignedTo, &a.Status, st(&a.CreatedAt), stN(&claimedAt), stN(&completedAt)); err != nil {
+			return nil, err
+		}
+		a.ClaimedAt = claimedAt
+		a.CompletedAt = completedAt
+		result = append(result, a)
+	}
+	if result == nil {
+		result = []store.WorkflowAssignment{}
+	}
+	return result, rows.Err()
+}