from collections import OrderedDict

from django.contrib.auth import user_logged_in

from rest_framework import status
from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from rest_framework.reverse import reverse
from rest_framework.viewsets import ViewSet

from djoser.views import (LoginView, PasswordResetView,
                          PasswordResetConfirmView)


class Login(LoginView):
    def action(self, serializer):
        user = serializer.user
        token, _ = Token.objects.get_or_create(user=user)
        user_logged_in.send(sender=user.__class__,
                            request=self.request,
                            user=user)
        return Response(
            data={'token': token.key, 'id': token.user_id},
            status=status.HTTP_200_OK,
        )


class PasswordReset(PasswordResetView):
    """Overriding to return empty object, for ember-ajax"""
    def action(self, serializer):
        response = super(PasswordReset, self).action(serializer)
        response.data = {}
        return response


class PasswordResetConfirm(PasswordResetConfirmView):
    """Overriding to return empty object, for ember-ajax"""
    def action(self, serializer):
        response = super(PasswordResetConfirm, self).action(serializer)
        response.data = {}
        return response


class AdminURLs(ViewSet):
    def get_view_name(self):
        return 'Admin URLs List'

    def list(self, request, *args, **kwargs):
        urls = OrderedDict([
            ('projects', [
                ['project', 'projects', 'project'],
                ['grant', 'projects', 'grant'],
                ['grant-report', 'projects', 'grantreport'],
            ]),

            ('collections', [
                ['collection-type', 'collections_ccdb', 'collectiontype'],
            ]),
        ])

        data = OrderedDict()
        for category, group in urls.items():
            paths = []
            for url in group:
                lookup = 'admin:{}_{}_changelist'.format(url[1], url[2])
                path = reverse(lookup, request=request)
                paths.append({'id': url[0], 'url': path})
            data[category] = paths
        return Response(data)