package albums

import (
	"context"
	"errors"

	"github.com/jackc/pgx/v5"
	"github.com/jackc/pgx/v5/pgxpool"

	"shared-album/internal/httpx"
	"shared-album/internal/models"
)

var ErrNotFound = errors.New("album not found")

type Store struct {
	pool *pgxpool.Pool
}

func NewStore(pool *pgxpool.Pool) *Store { return &Store{pool: pool} }

func (s *Store) Create(ctx context.Context, a *models.Album) error {
	return s.pool.QueryRow(ctx,
		`INSERT INTO albums (id, owner_id, title, description, cover_photo_id)
		 VALUES ($1, $2, $3, $4, $5)
		 RETURNING created_at, updated_at`,
		a.ID, a.OwnerID, a.Title, a.Description, a.CoverPhotoID,
	).Scan(&a.CreatedAt, &a.UpdatedAt)
}

func (s *Store) OwnerID(ctx context.Context, id string) (string, error) {
	var ownerID string
	err := s.pool.QueryRow(ctx, `SELECT owner_id FROM albums WHERE id = $1`, id).Scan(&ownerID)
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return "", httpx.NotFound("album_not_found", "album not found")
		}
		return "", err
	}
	return ownerID, nil
}

func (s *Store) GetByID(ctx context.Context, id string) (*models.Album, error) {
	var a models.Album
	err := s.pool.QueryRow(ctx,
		`SELECT id, owner_id, title, description, cover_photo_id, created_at, updated_at
		 FROM albums WHERE id = $1`, id).
		Scan(&a.ID, &a.OwnerID, &a.Title, &a.Description, &a.CoverPhotoID, &a.CreatedAt, &a.UpdatedAt)
	if err != nil {
		if errors.Is(err, pgx.ErrNoRows) {
			return nil, ErrNotFound
		}
		return nil, err
	}
	return &a, nil
}

func (s *Store) ListForUser(ctx context.Context, userID string) ([]models.Album, error) {
	rows, err := s.pool.Query(ctx,
		`SELECT a.id, a.owner_id, a.title, a.description, a.cover_photo_id, a.created_at, a.updated_at
		 FROM albums a
		 LEFT JOIN album_members m ON m.album_id = a.id AND m.user_id = $1
		 WHERE a.owner_id = $1 OR m.user_id IS NOT NULL
		 ORDER BY a.created_at DESC`, userID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	out := []models.Album{}
	for rows.Next() {
		var a models.Album
		if err := rows.Scan(&a.ID, &a.OwnerID, &a.Title, &a.Description, &a.CoverPhotoID, &a.CreatedAt, &a.UpdatedAt); err != nil {
			return nil, err
		}
		out = append(out, a)
	}
	return out, rows.Err()
}

func (s *Store) Update(ctx context.Context, a *models.Album) error {
	return s.pool.QueryRow(ctx,
		`UPDATE albums
		 SET title = $2, description = $3, cover_photo_id = $4, updated_at = now()
		 WHERE id = $1
		 RETURNING updated_at`,
		a.ID, a.Title, a.Description, a.CoverPhotoID,
	).Scan(&a.UpdatedAt)
}

func (s *Store) Delete(ctx context.Context, id string) error {
	tag, err := s.pool.Exec(ctx, `DELETE FROM albums WHERE id = $1`, id)
	if err != nil {
		return err
	}
	if tag.RowsAffected() == 0 {
		return ErrNotFound
	}
	return nil
}