from .models import *
from .serializers import *

from django.shortcuts import render
from rest_framework.decorators import api_view
from rest_framework.decorators import api_view
from rest_framework.response import Response
from rest_framework import status
from django.contrib.auth.models import User
from django.contrib.auth import authenticate
from rest_framework.authentication import TokenAuthentication
from rest_framework.permissions import IsAuthenticated
from rest_framework.decorators import authentication_classes
from rest_framework.decorators import permission_classes
from django.shortcuts import get_object_or_404
from datetime import datetime
import json
import os
from django.core.cache import cache
from django.http import JsonResponse

from django.conf import settings
import ffmpeg_streaming
from ffmpeg_streaming import Formats, Bitrate, Representation, Size
from django.core.files.storage import default_storage
from .tasks import create_hls_tasks


# Create your views here.
@api_view(['GET'])
@authentication_classes([TokenAuthentication])
@permission_classes([IsAuthenticated])
def nsfw_videos(request):
    if 'nsfw_videos' in cache:
        data = cache.get('nsfw_videos')
        if len(data) > 0:
            return Response(data=data, status=status.HTTP_200_OK)
    
    users = Video.objects.all()
    serializer = VideoSerializer(users, many=True)
    data = serializer.data
    cache.set('nsfw_videos', data, timeout=None)
    return Response(data=data, status=status.HTTP_200_OK)


@api_view(['GET'])
@authentication_classes([TokenAuthentication])
@permission_classes([IsAuthenticated])
def hls_videos(request):
    video_url = request.GET.get('video_name', '')
    type = request.GET.get('video_type', '')

    end_location = str(settings.MEDIA_ROOT) + '/hls/'+ video_url +'.m3u8'
    end_file_name = '/media/hls/'+ video_url +'_720p.m3u8'

    if default_storage.exists(end_location):
        return JsonResponse({'file_location': end_file_name})

    media_location = os.path.join(settings.MEDIA_ROOT) + "/" + type + "/" + video_url
    video = ffmpeg_streaming.input(media_location)
    
    hls = video.hls(Formats.h264())
    #_720p  = Representation(Size(1280, 720), Bitrate(2048 * 1024, 320 * 1024))
    hls.auto_generate_representations()
    hls.output(end_location)
    # {{url}}/videos/hls_video?video_name=Spiderman_Stretch.mp4&video_type=exercise_videos
    # {{url}}/videos/hls_video?video_name=Recover_24.mp4&video_type=videos
    return JsonResponse({'file_location': end_file_name})

@api_view(['GET'])
@authentication_classes([TokenAuthentication])
@permission_classes([IsAuthenticated])
def create_hls(request):
    create_hls_tasks.delay()
    return JsonResponse({'running': "running"})