package postgres

import (
	"context"

	"github.com/jmoiron/sqlx"

	"datarush/pkg/api/user"
)

type UserRepository struct {
	db *sqlx.DB
}

func NewUserRepository(db *sqlx.DB) *UserRepository {
	return &UserRepository{db: db}
}

func (r *UserRepository) GetProfile(ctx context.Context, userID string) (*user.User, error) {
	var u user.User
	err := r.db.GetContext(
		ctx,
		&u,
		"SELECT id, username, email, full_name, avatar_url FROM users WHERE id = $1",
		userID,
	)
	return &u, err
}

func (r *UserRepository) RegisterForCompetition(ctx context.Context, userID, competitionID string) error {
	_, err := r.db.ExecContext(
		ctx,
		"INSERT INTO user_competitions (user_id, competition_id) VALUES ($1, $2)",
		userID,
		competitionID,
	)
	return err
}

func (r *UserRepository) UnregisterFromCompetition(ctx context.Context, userID, competitionID string) error {
	_, err := r.db.ExecContext(
		ctx,
		"DELETE FROM user_competitions WHERE user_id = $1 AND competition_id = $2",
		userID,
		competitionID,
	)
	return err
}

func (r *UserRepository) ListUserCompetitions(ctx context.Context, userID string) ([]string, error) {
	var competitionIDs []string
	err := r.db.SelectContext(
		ctx,
		&competitionIDs,
		"SELECT competition_id FROM user_competitions WHERE user_id = $1",
		userID,
	)
	return competitionIDs, err
}