package store

import (
	"context"
	"sync"

	"5803025029/w04/internal/models"

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

type UserStore struct {
	conn *pgx.Conn
	mu   *sync.Mutex
}

func NewUserStore(conn *pgx.Conn, optMu ...*sync.Mutex) *UserStore {
	var mu *sync.Mutex
	if len(optMu) > 0 && optMu[0] != nil {
		mu = optMu[0]
	} else {
		mu = &sync.Mutex{}
	}
	return &UserStore{conn: conn, mu: mu}
}

func (s *UserStore) Create(ctx context.Context, user *models.User) error {
	s.mu.Lock()
	defer s.mu.Unlock()

	if user.ID == "" {
		user.ID = uuid.New().String()
	}

	query := `INSERT INTO users (id, username, password_hash) VALUES ($1, $2, $3)`
	_, err := s.conn.Exec(ctx, query, user.ID, user.Username, user.PasswordHash)
	return err
}

func (s *UserStore) GetByUsername(ctx context.Context, username string) (*models.User, error) {
	s.mu.Lock()
	defer s.mu.Unlock()

	var u models.User
	query := `SELECT id, username, password_hash FROM users WHERE username = $1`
	err := s.conn.QueryRow(ctx, query, username).Scan(&u.ID, &u.Username, &u.PasswordHash)
	if err != nil {
		return nil, err
	}
	return &u, nil
}

func (s *UserStore) GetByID(ctx context.Context, id string) (*models.User, error) {
	s.mu.Lock()
	defer s.mu.Unlock()

	var u models.User
	query := `SELECT id, username, password_hash FROM users WHERE id = $1`
	err := s.conn.QueryRow(ctx, query, id).Scan(&u.ID, &u.Username, &u.PasswordHash)
	if err != nil {
		return nil, err
	}
	return &u, nil
}