from django.contrib import messages
from django.shortcuts import redirect
from django.utils.translation import ugettext_lazy as _

from misago.acl import version as acl_version
from misago.acl.forms import get_permissions_forms
from misago.acl.models import Role
from misago.acl.views import RoleAdmin, RolesList
from misago.admin.views import generic
from misago.categories.forms import CategoryRoleForm, CategoryRolesACLFormFactory, RoleCategoryACLFormFactory
from misago.categories.models import Category, CategoryRole, RoleCategoryACL
from misago.categories.views.categoriesadmin import CategoriesList, CategoryAdmin


class CategoryRoleAdmin(generic.AdminBaseMixin):
    root_link = 'misago:admin:permissions:categories:index'
    Model = CategoryRole
    templates_dir = 'misago/admin/categoryroles'
    message_404 = _("Requested role does not exist.")


class CategoryRolesList(CategoryRoleAdmin, generic.ListView):
    ordering = (('name', None),)


class RoleFormMixin(object):
    def real_dispatch(self, request, target):
        form = CategoryRoleForm(instance=target)

        perms_forms = get_permissions_forms(target)

        if request.method == 'POST':
            perms_forms = get_permissions_forms(target, request.POST)
            valid_forms = 0
            for permissions_form in perms_forms:
                if permissions_form.is_valid():
                    valid_forms += 1

            form = CategoryRoleForm(request.POST, instance=target)
            if form.is_valid() and len(perms_forms) == valid_forms:
                new_permissions = {}
                for permissions_form in perms_forms:
                    cleaned_data = permissions_form.cleaned_data
                    new_permissions[permissions_form.prefix] = cleaned_data

                form.instance.permissions = new_permissions
                form.instance.save()

                messages.success(
                    request, self.message_submit % {'name': target.name})

                if 'stay' in request.POST:
                    return redirect(request.path)
                else:
                    return redirect(self.root_link)
            elif form.is_valid() and len(perms_forms) != valid_forms:
                form.add_error(None, _("Form contains errors."))

        return self.render(
            request,
            {
                'form': form,
                'target': target,
                'perms_forms': perms_forms,
            })


class NewCategoryRole(RoleFormMixin, CategoryRoleAdmin, generic.ModelFormView):
    message_submit = _('New role "%(name)s" has been saved.')


class EditCategoryRole(RoleFormMixin, CategoryRoleAdmin, generic.ModelFormView):
    message_submit = _('Role "%(name)s" has been changed.')


class DeleteCategoryRole(CategoryRoleAdmin, generic.ButtonView):
    def check_permissions(self, request, target):
        if target.special_role:
            message = _('Role "%(name)s" is special '
                        'role and can\'t be deleted.')
            return message % {'name': target.name}

    def button_action(self, request, target):
        target.delete()
        message = _('Role "%(name)s" has been deleted.')
        messages.success(request, message % {'name': target.name})


"""
Create category roles view for assinging roles to category,
add link to it in categories list
"""
class CategoryPermissions(CategoryAdmin, generic.ModelFormView):
    templates_dir = 'misago/admin/categoryroles'
    template = 'categoryroles.html'

    def real_dispatch(self, request, target):
        category_roles = CategoryRole.objects.order_by('name')

        assigned_roles = {}
        for acl in target.category_role_set.select_related('category_role'):
            assigned_roles[acl.role_id] = acl.category_role

        forms = []
        forms_are_valid = True
        for role in Role.objects.order_by('name'):
            FormType = CategoryRolesACLFormFactory(
                role, category_roles, assigned_roles.get(role.pk))

            if request.method == 'POST':
                forms.append(FormType(request.POST, prefix=role.pk))
                if not forms[-1].is_valid():
                    forms_are_valid = False
            else:
                forms.append(FormType(prefix=role.pk))

        if request.method == 'POST' and forms_are_valid:
            target.category_role_set.all().delete()
            new_permissions = []
            for form in forms:
                if form.cleaned_data['category_role']:
                    new_permissions.append(
                        RoleCategoryACL(
                            role=form.role,
                            category=target,
                            category_role=form.cleaned_data['category_role']
                        ))
            if new_permissions:
                RoleCategoryACL.objects.bulk_create(new_permissions)

            acl_version.invalidate()

            message = _("Category %(name)s permissions have been changed.")
            messages.success(request, message % {'name': target.name})
            if 'stay' in request.POST:
                return redirect(request.path)
            else:
                return redirect(self.root_link)

        return self.render(request, {
            'forms': forms,
            'target': target,
        })

CategoriesList.add_item_action(
    name=_("Category permissions"),
    icon='fa fa-adjust',
    link='misago:admin:categories:nodes:permissions',
    style='success'
)


"""
Create role categories view for assinging categories to role,
add link to it in user roles list
"""
class RoleCategoriesACL(RoleAdmin, generic.ModelFormView):
    templates_dir = 'misago/admin/categoryroles'
    template = 'rolecategories.html'

    def real_dispatch(self, request, target):
        categories = Category.objects.all_categories()
        roles = CategoryRole.objects.order_by('name')

        if not categories:
            messages.info(request, _("No categories exist."))
            return redirect(self.root_link)

        choices = {}
        for choice in target.categories_acls.select_related('category_role'):
            choices[choice.category_id] = choice.category_role

        forms = []
        forms_are_valid = True
        for category in categories:
            category.level_range = range(category.level - 1)
            FormType = RoleCategoryACLFormFactory(category,
                                               roles,
                                               choices.get(category.pk))

            if request.method == 'POST':
                forms.append(FormType(request.POST, prefix=category.pk))
                if not forms[-1].is_valid():
                    forms_are_valid = False
            else:
                forms.append(FormType(prefix=category.pk))

        if request.method == 'POST' and forms_are_valid:
            target.categories_acls.all().delete()
            new_permissions = []
            for form in forms:
                if form.cleaned_data['role']:
                    new_permissions.append(
                        RoleCategoryACL(role=target,
                                     category=form.category,
                                     category_role=form.cleaned_data['role']))
            if new_permissions:
                RoleCategoryACL.objects.bulk_create(new_permissions)

            acl_version.invalidate()

            message = _("Category permissions for role "
                        "%(name)s have been changed.")
            messages.success(request, message % {'name': target.name})
            if 'stay' in request.POST:
                return redirect(request.path)
            else:
                return redirect(self.root_link)

        return self.render(request, {
            'forms': forms,
            'target': target,
        })

RolesList.add_item_action(
    name=_("Categories permissions"),
    icon='fa fa-comments-o',
    link='misago:admin:permissions:users:categories',
    style='success'
)