package postgres

import (
	"context"

	"github.com/jmoiron/sqlx"

	"datarush/pkg/api/task"
)

type TaskRepository struct {
	db *sqlx.DB
}

func NewTaskRepository(db *sqlx.DB) *TaskRepository {
	return &TaskRepository{db: db}
}

func (r *TaskRepository) CreateTask(ctx context.Context, t *task.Task) (*task.Task, error) {
	query := `INSERT INTO tasks (competition_id, title, description, in_competition_position, max_points, max_attempts, type) 
	VALUES ($1, $2, $3, $4, $5, $6, $7) RETURNING id, created_at, updated_at`

	var createdTask task.Task
	err := r.db.QueryRowxContext(ctx, query, t.CompetitionId, t.Title, t.Description, t.InCompetitionPosition, t.MaxPoints, t.MaxAttempts, t.Type).
		StructScan(&createdTask)
	if err != nil {
		return nil, err
	}
	createdTask.CompetitionId = t.CompetitionId
	createdTask.Title = t.Title
	createdTask.Description = t.Description
	createdTask.InCompetitionPosition = t.InCompetitionPosition
	createdTask.MaxPoints = t.MaxPoints
	createdTask.MaxAttempts = t.MaxAttempts
	createdTask.Type = t.Type
	return &createdTask, nil
}

func (r *TaskRepository) GetTask(ctx context.Context, id string) (*task.Task, error) {
	var t task.Task
	err := r.db.GetContext(ctx, &t, "SELECT * FROM tasks WHERE id = $1", id)
	return &t, err
}

func (r *TaskRepository) EditTask(ctx context.Context, t *task.Task) (*task.Task, error) {
	query := `UPDATE tasks SET title = $1, description = $2, in_competition_position = $3, max_points = $4, max_attempts = $5, type = $6, updated_at = now()
	WHERE id = $7 RETURNING updated_at`

	var updatedTask task.Task
	err := r.db.QueryRowxContext(ctx, query, t.Title, t.Description, t.InCompetitionPosition, t.MaxPoints, t.MaxAttempts, t.Type, t.Id).
		StructScan(&updatedTask)
	if err != nil {
		return nil, err
	}
	t.UpdatedAt = updatedTask.UpdatedAt
	return t, nil
}

func (r *TaskRepository) DeleteTask(ctx context.Context, id string) error {
	_, err := r.db.ExecContext(ctx, "DELETE FROM tasks WHERE id = $1", id)
	return err
}

func (r *TaskRepository) ListCompetitionTasks(ctx context.Context, competitionID string) ([]*task.Task, error) {
	var tasks []*task.Task
	err := r.db.SelectContext(ctx, &tasks, "SELECT * FROM tasks WHERE competition_id = $1", competitionID)
	return tasks, err
}

func (r *TaskRepository) GetTaskAttachments(
	ctx context.Context,
	taskID string,
	showPrivate bool,
) ([]*task.TaskAttachment, error) {
	var attachments []*task.TaskAttachment
	query := "SELECT * FROM task_attachments WHERE task_id = $1"
	args := []interface{}{taskID}

	if !showPrivate {
		query += " AND is_public = true"
	}

	err := r.db.SelectContext(ctx, &attachments, query, args...)
	return attachments, err
}