package sharelinks

import (
	"context"
	"database/sql"
	"errors"
	"time"

	"galo/internal/models"
	"galo/internal/sqliteutil"
)

// SQLiteStore is the SQLite implementation of Repository.
type SQLiteStore struct {
	db *sql.DB
}

func NewSQLiteStore(db *sql.DB) *SQLiteStore { return &SQLiteStore{db: db} }

func (s *SQLiteStore) Create(ctx context.Context, l *models.ShareLink) error {
	var createdAt string
	err := s.db.QueryRowContext(ctx,
		`INSERT INTO share_links (id, album_id, token, created_by, expires_at)
		 VALUES ($1, $2, $3, $4, $5)
		 RETURNING created_at`,
		l.ID, l.AlbumID, l.Token, l.CreatedBy, sqliteutil.FormatTime(l.ExpiresAt),
	).Scan(&createdAt)
	if err != nil {
		return err
	}
	l.CreatedAt = sqliteutil.ParseTime(createdAt)
	return nil
}

func (s *SQLiteStore) ListByAlbum(ctx context.Context, albumID string) ([]models.ShareLink, error) {
	rows, err := s.db.QueryContext(ctx,
		`SELECT id, album_id, token, created_by, created_at, expires_at, revoked_at
		 FROM share_links WHERE album_id = $1 ORDER BY created_at DESC`, albumID)
	if err != nil {
		return nil, err
	}
	defer rows.Close()
	out := []models.ShareLink{}
	for rows.Next() {
		var l models.ShareLink
		var createdAt, expiresAt string
		var revokedAt sql.NullString
		if err := rows.Scan(&l.ID, &l.AlbumID, &l.Token, &l.CreatedBy, &createdAt, &expiresAt, &revokedAt); err != nil {
			return nil, err
		}
		l.CreatedAt = sqliteutil.ParseTime(createdAt)
		l.ExpiresAt = sqliteutil.ParseTime(expiresAt)
		l.RevokedAt = sqliteutil.ParseNullTime(revokedAt)
		out = append(out, l)
	}
	return out, rows.Err()
}

func (s *SQLiteStore) GetByToken(ctx context.Context, token string) (*models.ShareLink, error) {
	var l models.ShareLink
	var createdAt, expiresAt string
	var revokedAt sql.NullString
	err := s.db.QueryRowContext(ctx,
		`SELECT id, album_id, token, created_by, created_at, expires_at, revoked_at
		 FROM share_links WHERE token = $1`, token).
		Scan(&l.ID, &l.AlbumID, &l.Token, &l.CreatedBy, &createdAt, &expiresAt, &revokedAt)
	if err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return nil, ErrNotFound
		}
		return nil, err
	}
	l.CreatedAt = sqliteutil.ParseTime(createdAt)
	l.ExpiresAt = sqliteutil.ParseTime(expiresAt)
	l.RevokedAt = sqliteutil.ParseNullTime(revokedAt)
	return &l, nil
}

func (s *SQLiteStore) Revoke(ctx context.Context, id, albumID string) error {
	res, err := s.db.ExecContext(ctx,
		`UPDATE share_links SET revoked_at = $3 WHERE id = $1 AND album_id = $2 AND revoked_at IS NULL`,
		id, albumID, sqliteutil.FormatTime(time.Now()))
	if err != nil {
		return err
	}
	if n, _ := res.RowsAffected(); n == 0 {
		return ErrNotFound
	}
	return nil
}