from django.views.generic import TemplateView
from django.shortcuts import get_object_or_404
from django.http import FileResponse as BaseFileResponse
from .models import SharedFile, FileToken
import mimetypes
from django.utils import timezone
from django.contrib.auth.decorators import login_required


mimetypes.init()


class SharedFileDetails(TemplateView):
    template_name = "file.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['file'] = get_object_or_404(SharedFile, short_id=kwargs['id'])
        context['token'] = FileToken.objects.create(file=context['file'])
        return context


def FileResponse(file):
    response = BaseFileResponse(file.file)
    response['Content-Type'] = mimetypes.guess_type(file.get_original_filename())[0]
    response['Content-Disposition'] = 'attachment; filename="{0}"'.format(file.get_original_filename())
    return response


def file_download(request, id, token):
    time_threshold = timezone.now() - FileToken.valid_time
    FileToken.objects.filter(created__lt=time_threshold)
    file = get_object_or_404(SharedFile, short_id=id, published=True)
    token = get_object_or_404(FileToken, token=token, file=file)
    token.delete()  # delete after used
    return FileResponse(file)


def hotlink_file_download(request, id):
    file = get_object_or_404(SharedFile, short_id=id, hotlink=True, published=True)
    return FileResponse(file)


@login_required
def uploaded_file(request, filename):
    file = get_object_or_404(SharedFile, file=request.get_full_path()[1:])  # strip preceding slash
    response = FileResponse(file)
    return response