breakpilot-compliance/ai-compliance-sdk/internal/roadmap/store_items.go

package roadmap

import (
	"context"
	"encoding/json"
	"fmt"
	"time"

	"github.com/google/uuid"
	"github.com/jackc/pgx/v5"
)

// ============================================================================
// RoadmapItem CRUD Operations
// ============================================================================

// CreateItem creates a new roadmap item
func (s *Store) CreateItem(ctx context.Context, item *RoadmapItem) error {
	item.ID = uuid.New()
	item.CreatedAt = time.Now().UTC()
	item.UpdatedAt = item.CreatedAt
	if item.Status == "" {
		item.Status = ItemStatusPlanned
	}
	if item.Priority == "" {
		item.Priority = ItemPriorityMedium
	}
	if item.Category == "" {
		item.Category = ItemCategoryTechnical
	}

	dependsOn, _ := json.Marshal(item.DependsOn)
	blockedBy, _ := json.Marshal(item.BlockedBy)
	evidenceReq, _ := json.Marshal(item.EvidenceRequired)
	evidenceProv, _ := json.Marshal(item.EvidenceProvided)

	_, err := s.pool.Exec(ctx, `
		INSERT INTO roadmap_items (
			id, roadmap_id, title, description, category, priority, status,
			control_id, regulation_ref, gap_id,
			effort_days, effort_hours, estimated_cost,
			assignee_id, assignee_name, department,
			planned_start, planned_end, actual_start, actual_end,
			depends_on, blocked_by,
			evidence_required, evidence_provided,
			notes, risk_notes,
			source_row, source_file, sort_order,
			created_at, updated_at
		) VALUES (
			$1, $2, $3, $4, $5, $6, $7,
			$8, $9, $10,
			$11, $12, $13,
			$14, $15, $16,
			$17, $18, $19, $20,
			$21, $22,
			$23, $24,
			$25, $26,
			$27, $28, $29,
			$30, $31
		)
	`,
		item.ID, item.RoadmapID, item.Title, item.Description, string(item.Category), string(item.Priority), string(item.Status),
		item.ControlID, item.RegulationRef, item.GapID,
		item.EffortDays, item.EffortHours, item.EstimatedCost,
		item.AssigneeID, item.AssigneeName, item.Department,
		item.PlannedStart, item.PlannedEnd, item.ActualStart, item.ActualEnd,
		dependsOn, blockedBy,
		evidenceReq, evidenceProv,
		item.Notes, item.RiskNotes,
		item.SourceRow, item.SourceFile, item.SortOrder,
		item.CreatedAt, item.UpdatedAt,
	)

	return err
}

// GetItem retrieves a roadmap item by ID
func (s *Store) GetItem(ctx context.Context, id uuid.UUID) (*RoadmapItem, error) {
	var item RoadmapItem
	var category, priority, status string
	var dependsOn, blockedBy, evidenceReq, evidenceProv []byte

	err := s.pool.QueryRow(ctx, `
		SELECT
			id, roadmap_id, title, description, category, priority, status,
			control_id, regulation_ref, gap_id,
			effort_days, effort_hours, estimated_cost,
			assignee_id, assignee_name, department,
			planned_start, planned_end, actual_start, actual_end,
			depends_on, blocked_by,
			evidence_required, evidence_provided,
			notes, risk_notes,
			source_row, source_file, sort_order,
			created_at, updated_at
		FROM roadmap_items WHERE id = $1
	`, id).Scan(
		&item.ID, &item.RoadmapID, &item.Title, &item.Description, &category, &priority, &status,
		&item.ControlID, &item.RegulationRef, &item.GapID,
		&item.EffortDays, &item.EffortHours, &item.EstimatedCost,
		&item.AssigneeID, &item.AssigneeName, &item.Department,
		&item.PlannedStart, &item.PlannedEnd, &item.ActualStart, &item.ActualEnd,
		&dependsOn, &blockedBy,
		&evidenceReq, &evidenceProv,
		&item.Notes, &item.RiskNotes,
		&item.SourceRow, &item.SourceFile, &item.SortOrder,
		&item.CreatedAt, &item.UpdatedAt,
	)

	if err == pgx.ErrNoRows {
		return nil, nil
	}
	if err != nil {
		return nil, err
	}

	item.Category = ItemCategory(category)
	item.Priority = ItemPriority(priority)
	item.Status = ItemStatus(status)
	json.Unmarshal(dependsOn, &item.DependsOn)
	json.Unmarshal(blockedBy, &item.BlockedBy)
	json.Unmarshal(evidenceReq, &item.EvidenceRequired)
	json.Unmarshal(evidenceProv, &item.EvidenceProvided)

	return &item, nil
}

// ListItems lists items for a roadmap with optional filters
func (s *Store) ListItems(ctx context.Context, roadmapID uuid.UUID, filters *RoadmapItemFilters) ([]RoadmapItem, error) {
	query := `
		SELECT
			id, roadmap_id, title, description, category, priority, status,
			control_id, regulation_ref, gap_id,
			effort_days, effort_hours, estimated_cost,
			assignee_id, assignee_name, department,
			planned_start, planned_end, actual_start, actual_end,
			depends_on, blocked_by,
			evidence_required, evidence_provided,
			notes, risk_notes,
			source_row, source_file, sort_order,
			created_at, updated_at
		FROM roadmap_items WHERE roadmap_id = $1`

	args := []interface{}{roadmapID}
	argIdx := 2

	if filters != nil {
		if filters.Status != "" {
			query += fmt.Sprintf(" AND status = $%d", argIdx)
			args = append(args, string(filters.Status))
			argIdx++
		}
		if filters.Priority != "" {
			query += fmt.Sprintf(" AND priority = $%d", argIdx)
			args = append(args, string(filters.Priority))
			argIdx++
		}
		if filters.Category != "" {
			query += fmt.Sprintf(" AND category = $%d", argIdx)
			args = append(args, string(filters.Category))
			argIdx++
		}
		if filters.AssigneeID != nil {
			query += fmt.Sprintf(" AND assignee_id = $%d", argIdx)
			args = append(args, *filters.AssigneeID)
			argIdx++
		}
		if filters.ControlID != "" {
			query += fmt.Sprintf(" AND control_id = $%d", argIdx)
			args = append(args, filters.ControlID)
			argIdx++
		}
		if filters.SearchQuery != "" {
			query += fmt.Sprintf(" AND (title ILIKE $%d OR description ILIKE $%d)", argIdx, argIdx)
			args = append(args, "%"+filters.SearchQuery+"%")
			argIdx++
		}
	}

	query += " ORDER BY sort_order ASC, priority ASC, created_at ASC"

	if filters != nil && filters.Limit > 0 {
		query += fmt.Sprintf(" LIMIT $%d", argIdx)
		args = append(args, filters.Limit)
		argIdx++

		if filters.Offset > 0 {
			query += fmt.Sprintf(" OFFSET $%d", argIdx)
			args = append(args, filters.Offset)
		}
	}

	rows, err := s.pool.Query(ctx, query, args...)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	var items []RoadmapItem
	for rows.Next() {
		var item RoadmapItem
		var category, priority, status string
		var dependsOn, blockedBy, evidenceReq, evidenceProv []byte

		err := rows.Scan(
			&item.ID, &item.RoadmapID, &item.Title, &item.Description, &category, &priority, &status,
			&item.ControlID, &item.RegulationRef, &item.GapID,
			&item.EffortDays, &item.EffortHours, &item.EstimatedCost,
			&item.AssigneeID, &item.AssigneeName, &item.Department,
			&item.PlannedStart, &item.PlannedEnd, &item.ActualStart, &item.ActualEnd,
			&dependsOn, &blockedBy,
			&evidenceReq, &evidenceProv,
			&item.Notes, &item.RiskNotes,
			&item.SourceRow, &item.SourceFile, &item.SortOrder,
			&item.CreatedAt, &item.UpdatedAt,
		)
		if err != nil {
			return nil, err
		}

		item.Category = ItemCategory(category)
		item.Priority = ItemPriority(priority)
		item.Status = ItemStatus(status)
		json.Unmarshal(dependsOn, &item.DependsOn)
		json.Unmarshal(blockedBy, &item.BlockedBy)
		json.Unmarshal(evidenceReq, &item.EvidenceRequired)
		json.Unmarshal(evidenceProv, &item.EvidenceProvided)

		items = append(items, item)
	}

	return items, nil
}

// UpdateItem updates a roadmap item
func (s *Store) UpdateItem(ctx context.Context, item *RoadmapItem) error {
	item.UpdatedAt = time.Now().UTC()

	dependsOn, _ := json.Marshal(item.DependsOn)
	blockedBy, _ := json.Marshal(item.BlockedBy)
	evidenceReq, _ := json.Marshal(item.EvidenceRequired)
	evidenceProv, _ := json.Marshal(item.EvidenceProvided)

	_, err := s.pool.Exec(ctx, `
		UPDATE roadmap_items SET
			title = $2, description = $3, category = $4, priority = $5, status = $6,
			control_id = $7, regulation_ref = $8, gap_id = $9,
			effort_days = $10, effort_hours = $11, estimated_cost = $12,
			assignee_id = $13, assignee_name = $14, department = $15,
			planned_start = $16, planned_end = $17, actual_start = $18, actual_end = $19,
			depends_on = $20, blocked_by = $21,
			evidence_required = $22, evidence_provided = $23,
			notes = $24, risk_notes = $25,
			sort_order = $26, updated_at = $27
		WHERE id = $1
	`,
		item.ID, item.Title, item.Description, string(item.Category), string(item.Priority), string(item.Status),
		item.ControlID, item.RegulationRef, item.GapID,
		item.EffortDays, item.EffortHours, item.EstimatedCost,
		item.AssigneeID, item.AssigneeName, item.Department,
		item.PlannedStart, item.PlannedEnd, item.ActualStart, item.ActualEnd,
		dependsOn, blockedBy,
		evidenceReq, evidenceProv,
		item.Notes, item.RiskNotes,
		item.SortOrder, item.UpdatedAt,
	)

	return err
}

// DeleteItem deletes a roadmap item
func (s *Store) DeleteItem(ctx context.Context, id uuid.UUID) error {
	_, err := s.pool.Exec(ctx, "DELETE FROM roadmap_items WHERE id = $1", id)
	return err
}

// BulkCreateItems creates multiple items in a transaction
func (s *Store) BulkCreateItems(ctx context.Context, items []RoadmapItem) error {
	tx, err := s.pool.Begin(ctx)
	if err != nil {
		return err
	}
	defer tx.Rollback(ctx)

	for i := range items {
		item := &items[i]
		item.ID = uuid.New()
		item.CreatedAt = time.Now().UTC()
		item.UpdatedAt = item.CreatedAt

		dependsOn, _ := json.Marshal(item.DependsOn)
		blockedBy, _ := json.Marshal(item.BlockedBy)
		evidenceReq, _ := json.Marshal(item.EvidenceRequired)
		evidenceProv, _ := json.Marshal(item.EvidenceProvided)

		_, err := tx.Exec(ctx, `
			INSERT INTO roadmap_items (
				id, roadmap_id, title, description, category, priority, status,
				control_id, regulation_ref, gap_id,
				effort_days, effort_hours, estimated_cost,
				assignee_id, assignee_name, department,
				planned_start, planned_end, actual_start, actual_end,
				depends_on, blocked_by,
				evidence_required, evidence_provided,
				notes, risk_notes,
				source_row, source_file, sort_order,
				created_at, updated_at
			) VALUES (
				$1, $2, $3, $4, $5, $6, $7,
				$8, $9, $10,
				$11, $12, $13,
				$14, $15, $16,
				$17, $18, $19, $20,
				$21, $22,
				$23, $24,
				$25, $26,
				$27, $28, $29,
				$30, $31
			)
		`,
			item.ID, item.RoadmapID, item.Title, item.Description, string(item.Category), string(item.Priority), string(item.Status),
			item.ControlID, item.RegulationRef, item.GapID,
			item.EffortDays, item.EffortHours, item.EstimatedCost,
			item.AssigneeID, item.AssigneeName, item.Department,
			item.PlannedStart, item.PlannedEnd, item.ActualStart, item.ActualEnd,
			dependsOn, blockedBy,
			evidenceReq, evidenceProv,
			item.Notes, item.RiskNotes,
			item.SourceRow, item.SourceFile, item.SortOrder,
			item.CreatedAt, item.UpdatedAt,
		)
		if err != nil {
			return err
		}
	}

	return tx.Commit(ctx)
}