from django.contrib.auth import decorators as auth_decorator
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.db.models import Count, Q
from django.urls import reverse
from django.utils.decorators import method_decorator
from django.views import generic

from access.mixin import OrganizationPermission

from core.forms.comment import AddNoteForm
from core.models.notes import Notes

from itam.models.device import DeviceSoftware
from itam.models.software import Software, SoftwareVersion
from itam.forms.software.update import Update as SoftwareUpdate_Form

from settings.models.user_settings import UserSettings



class IndexView(PermissionRequiredMixin, OrganizationPermission, generic.ListView):
    model = Software
    permission_required = 'itam.view_software'
    template_name = 'itam/software_index.html.j2'
    context_object_name = "softwares"
    paginate_by = 10


    def get_queryset(self):

        if self.request.user.is_superuser:

            return Software.objects.filter().order_by('name')

        else:

            return Software.objects.filter(Q(organization__in=self.user_organizations()) | Q(is_global = True)).order_by('name')




class View(OrganizationPermission, generic.UpdateView):
    model = Software
    permission_required = [
        'itam.view_software',
        'itam.change_software'
    ]
    template_name = 'itam/software.html.j2'

    form_class = SoftwareUpdate_Form

    context_object_name = "software"


    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)

        software_versions = SoftwareVersion.objects.filter(
            software=self.kwargs['pk']
        ).annotate(
            installs=Count("installedversion")
        )

        context['software_versions'] = software_versions

        context['notes_form'] = AddNoteForm(prefix='note')
        context['notes'] = Notes.objects.filter(software=self.kwargs['pk'])

        context['model_pk'] = self.kwargs['pk']
        context['model_name'] = self.model._meta.verbose_name.replace(' ', '')

        context['model_delete_url'] = reverse('ITAM:_software_delete', args=(self.kwargs['pk'],))

        context['content_title'] = self.object.name

        if self.request.user.is_superuser:

            context['device_software'] = DeviceSoftware.objects.filter(
                software=self.kwargs['pk']
            ).order_by(
                'device',
                'organization'
            )

        elif not self.request.user.is_superuser:
            context['device_software'] = DeviceSoftware.objects.filter(
                Q(device__in=self.user_organizations(),
                software=self.kwargs['pk'])
            ).order_by(
                'device',
                'organization'
            )

        return context


    @method_decorator(auth_decorator.permission_required("itam.change_software", raise_exception=True))
    def post(self, request, *args, **kwargs):

        software = Software.objects.get(pk=self.kwargs['pk'])

        notes = AddNoteForm(request.POST, prefix='note')

        if notes.is_bound and notes.is_valid() and notes.instance.note != '':

            notes.instance.organization = software.organization
            notes.instance.software = software
            notes.instance.usercreated = request.user

            notes.save()



        return super().post(request, *args, **kwargs)


    def get_success_url(self, **kwargs):

        return f"/itam/software/{self.kwargs['pk']}/"



class Add(OrganizationPermission, generic.CreateView):
    model = Software
    permission_required = [
        'itam.add_software',
    ]
    template_name = 'form.html.j2'
    fields = [
        'name',
        'publisher',
        'category',
        'organization',
        'is_global'
    ]


    def get_initial(self):

        return {
            'organization': UserSettings.objects.get(user = self.request.user).default_organization
        }


    def get_success_url(self, **kwargs):

        return f"/itam/software/"


    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)

        context['content_title'] = 'Add Software'

        return context

class Delete(OrganizationPermission, generic.DeleteView):
    model = Software
    permission_required = [
        'itam.delete_software',
    ]
    template_name = 'form.html.j2'

    def get_success_url(self, **kwargs):

        return f"/itam/software/"


    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)

        context['content_title'] = 'Delete ' + self.object.name

        return context