from django.contrib.auth.mixins import LoginRequiredMixin
from django.views import generic
from videos.models import Series, Video
from scrobbles.views import ScrobbleableListView, ScrobbleableDetailView


class MovieListView(LoginRequiredMixin, generic.ListView):
    model = Video
    template_name = "videos/movie_list.html"

    def get_queryset(self):
        return Video.objects.filter(video_type=Video.VideoType.MOVIE)


class SeriesListView(LoginRequiredMixin, generic.ListView):
    model = Series


class SeriesDetailView(LoginRequiredMixin, generic.DetailView):
    model = Series
    slug_field = "uuid"

    def get_context_data(self, **kwargs):
        user_id = self.request.user.id
        context_data = super().get_context_data(**kwargs)

        context_data["scrobbles"] = self.object.scrobbles_for_user(user_id)
        next_episode_id = self.object.last_scrobbled_episode(
            user_id
        ).next_imdb_id
        if self.object.is_episode_playing(user_id):
            next_episode_id = ""
        context_data["next_episode_id"] = "tt" + next_episode_id
        return context_data


class VideoListView(ScrobbleableListView):
    model = Video


class VideoDetailView(ScrobbleableDetailView):
    model = Video