package db

import (
	"context"
	"strings"
)

func (q *Queries) GetCommandPaletteContinueWatching(ctx context.Context, userID string, query string, limit int64) ([]GetContinueWatchingEntriesRow, error) {
	if userID == "" {
		return nil, nil
	}
	if limit <= 0 {
		limit = 5
	}

	needle, pattern := commandPalettePattern(query)
	rows, err := q.db.QueryContext(ctx, `
SELECT
    c.id,
    c.user_id,
    c.anime_id,
    c.current_episode,
    c.current_time_seconds,
    c.duration_seconds,
    c.created_at,
    c.updated_at,
    a.title_original,
    a.title_english,
    a.title_japanese,
    a.image_url,
    a.duration_seconds as anime_duration_seconds
FROM continue_watching_entry c
JOIN anime a ON c.anime_id = a.id
WHERE c.user_id = ?
  AND (
    ? = ''
    OR lower(a.title_original) LIKE ?
    OR lower(coalesce(a.title_english, '')) LIKE ?
    OR lower(coalesce(a.title_japanese, '')) LIKE ?
    OR lower('Continue watching') LIKE ?
  )
ORDER BY c.updated_at DESC
LIMIT ?`, userID, needle, pattern, pattern, pattern, pattern, limit)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	items := make([]GetContinueWatchingEntriesRow, 0, int(limit))
	for rows.Next() {
		var item GetContinueWatchingEntriesRow
		if err := rows.Scan(
			&item.ID,
			&item.UserID,
			&item.AnimeID,
			&item.CurrentEpisode,
			&item.CurrentTimeSeconds,
			&item.DurationSeconds,
			&item.CreatedAt,
			&item.UpdatedAt,
			&item.TitleOriginal,
			&item.TitleEnglish,
			&item.TitleJapanese,
			&item.ImageUrl,
			&item.AnimeDurationSeconds,
		); err != nil {
			return nil, err
		}
		items = append(items, item)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}

	return items, nil
}

func (q *Queries) GetCommandPaletteWatchlist(ctx context.Context, userID string, query string, limit int64) ([]GetUserWatchListRow, error) {
	if userID == "" {
		return nil, nil
	}
	if limit <= 0 {
		limit = 5
	}

	needle, pattern := commandPalettePattern(query)
	rows, err := q.db.QueryContext(ctx, `
SELECT 
    e.id,
    e.user_id,
    e.anime_id,
    e.status,
    e.created_at,
    e.updated_at,
    e.current_episode,
    e.last_episode_at,
    e.current_time_seconds,
    a.title_original,
    a.title_english,
    a.title_japanese,
    a.image_url,
    a.airing
FROM watch_list_entry e
JOIN anime a ON e.anime_id = a.id
WHERE e.user_id = ?
  AND e.status IN ('watching', 'plan_to_watch')
  AND (
    ? = ''
    OR lower(a.title_original) LIKE ?
    OR lower(coalesce(a.title_english, '')) LIKE ?
    OR lower(coalesce(a.title_japanese, '')) LIKE ?
    OR lower(e.status) LIKE ?
  )
ORDER BY
  CASE e.status
    WHEN 'watching' THEN 0
    WHEN 'plan_to_watch' THEN 1
    ELSE 2
  END,
  e.updated_at DESC
LIMIT ?`, userID, needle, pattern, pattern, pattern, pattern, limit)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	items := make([]GetUserWatchListRow, 0, int(limit))
	for rows.Next() {
		var item GetUserWatchListRow
		if err := rows.Scan(
			&item.ID,
			&item.UserID,
			&item.AnimeID,
			&item.Status,
			&item.CreatedAt,
			&item.UpdatedAt,
			&item.CurrentEpisode,
			&item.LastEpisodeAt,
			&item.CurrentTimeSeconds,
			&item.TitleOriginal,
			&item.TitleEnglish,
			&item.TitleJapanese,
			&item.ImageUrl,
			&item.Airing,
		); err != nil {
			return nil, err
		}
		items = append(items, item)
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}

	return items, nil
}

func commandPalettePattern(query string) (string, string) {
	needle := strings.ToLower(strings.TrimSpace(query))
	return needle, "%" + needle + "%"
}