MetaPhase-Consulting/State-TalentMAP-API

import logging

from rest_condition import Or
from rest_framework.permissions import IsAuthenticated, IsAuthenticatedOrReadOnly
from rest_framework.response import Response
from rest_framework import status

from django.contrib.auth.models import Group
from django.http import HttpResponse
from drf_yasg.utils import swagger_auto_schema
from drf_yasg import openapi

from talentmap_api.common.permissions import isDjangoGroupMemberOrReadOnly, isDjangoGroupMember
from talentmap_api.fsbid.views.base import BaseView
from rest_framework.views import APIView
from talentmap_api.common.common_helpers import user_in_any_group
import talentmap_api.fsbid.services.employee as services
import talentmap_api.fsbid.services.client as client_services

logger = logging.getLogger(__name__)


class FSBidEmployeePerdetSeqNumActionView(BaseView):

    permission_classes = (IsAuthenticated,)

    def put(self, request, *args, **kwargs):
        '''
        Sets the employee perdet_seq_num (emp_id) for the user
        '''
        jwt = request.META['HTTP_JWT']
        emp_id = services.get_employee_perdet_seq_num(jwt)

        if emp_id is not None:
            user = request.user.profile
            user.emp_id = str(int(emp_id))
            user.save()

        auth_user = request.user

        # Get the valid and mapped roles from the token
        user_roles = services.map_group_to_fsbid_role(jwt)

        # Add roles
        for current_role in user_roles:
            auth_user.groups.add(current_role)

        # Remove any roles that the user has lost since the last time they logged in
        for role in services.ROLE_MAPPING.values():
            if role not in user_roles.values_list('name', flat=True):
                auth_user.groups.remove(Group.objects.filter(name=role).first())

        auth_user.save()

        return Response(status=status.HTTP_204_NO_CONTENT)


class FSBidBureauUserPermissionsView(BaseView):
    permission_classes = (IsAuthenticatedOrReadOnly, isDjangoGroupMemberOrReadOnly('bureau_user'))
    '''
    Get an employee's assigned bureaus
    '''
    def get(self, request):
        result = services.get_bureau_permissions(request.META['HTTP_JWT'], f"{request.scheme}://{request.get_host()}")
        if result is None:
            return Response(status=status.HTTP_404_NOT_FOUND)
        return Response(result)


class FSBidOrgUserPermissionsView(BaseView):
    permission_classes = (IsAuthenticatedOrReadOnly, isDjangoGroupMemberOrReadOnly('post_user'))
    '''
    Get an employee's assigned organizations
    '''
    def get(self, request):
        result = services.get_org_permissions(request.META['HTTP_JWT'], f"{request.scheme}://{request.get_host()}")
        if result is None:
            return Response(status=status.HTTP_404_NOT_FOUND)
        return Response(result)


class FSBidSeparationsView(BaseView):
    permission_classes = [Or(isDjangoGroupMember('ao_user'), isDjangoGroupMember('cdo'))]

    @swagger_auto_schema(
        manual_parameters=[
            openapi.Parameter("page", openapi.IN_QUERY, type=openapi.TYPE_INTEGER, description='A page number within the paginated result set.'),
            openapi.Parameter("limit", openapi.IN_QUERY, type=openapi.TYPE_INTEGER, description='Number of results to return per page.'),
        ])

    def get(self, request, pk):
        '''
        Get an employee's separations
        '''
        return Response(services.get_separations(request.query_params, request.META['HTTP_JWT'], pk))

class FSBidAssignmentSeparationsBidsView(BaseView):
    permission_classes = [Or(isDjangoGroupMember('ao_user'), isDjangoGroupMember('cdo'))]

    @swagger_auto_schema(
        manual_parameters=[
            openapi.Parameter("page", openapi.IN_QUERY, type=openapi.TYPE_INTEGER, description='A page number within the paginated result set.'),
            openapi.Parameter("limit", openapi.IN_QUERY, type=openapi.TYPE_INTEGER, description='Number of results to return per page.'),
        ])

    def get(self, request, pk):
        '''
        Get an employee's assignments,separations, and bids
        '''
        return Response(services.get_assignments_separations_bids(request.query_params, request.META['HTTP_JWT'], pk))


class FSBidEmployeeProfileReportView(APIView):
    permission_classes = [Or(isDjangoGroupMember('ao_user'), isDjangoGroupMember('bureau_user'), isDjangoGroupMember('post_user'), isDjangoGroupMember('cdo'))]

    def get(self, request, pk):
        '''
        Get an employee's profile report
        '''
        # pk is hru id
        try:
            return services.get_employee_profile_report(request.query_params, pk, request.META['HTTP_JWT'])
        except:
            return Response(status=status.HTTP_400_BAD_REQUEST)


class SECREFUserInfoView(APIView):

    def get(self, request, pk):
        '''
        Get user's info
        '''
        return Response(client_services.get_user_information(request.META['HTTP_JWT'], pk))