# ~*~ coding: utf-8 ~*~

from rest_framework import viewsets, generics, mixins
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework_bulk import BulkModelViewSet, BulkDestroyAPIView
from django_filters.rest_framework import DjangoFilterBackend
from rest_framework_bulk import BulkListSerializer, BulkSerializerMixin, ListBulkCreateUpdateDestroyAPIView
from django.shortcuts import get_object_or_404

from common.mixins import IDInFilterMixin
from common.utils import get_object_or_none, signer
from .hands import IsSuperUserOrTerminalUser, IsSuperUser
from .models import AssetGroup, Asset, IDC, SystemUser, AdminUser, Tag
from . import serializers


class AssetViewSet(IDInFilterMixin, BulkModelViewSet):
	"""API endpoint that allows Asset to be viewed or edited."""
	queryset = Asset.objects.all()
	serializer_class = serializers.AssetSerializer
	filter_backends = (DjangoFilterBackend,)
	filter_fields = ('id', 'ip', 'hostname')
	permission_classes = (IsSuperUser,)

	def get_queryset(self):
		queryset = super(AssetViewSet, self).get_queryset()
		idc_id = self.request.query_params.get('idc_id', '')
		tags_id = self.request.query_params.get('tag_id', '')
		system_users_id = self.request.query_params.get('system_user_id', '')
		asset_group_id = self.request.query_params.get('asset_group_id', '')
		admin_user_id = self.request.query_params.get('admin_user_id', '')
		if idc_id:
			queryset = queryset.filter(idc__id=idc_id)
		if tags_id:
			queryset = queryset.filter(tags__id=tags_id)
		if system_users_id:
			queryset = queryset.filter(system_users__id=system_users_id)
		if admin_user_id:
			queryset = queryset.filter(admin_user__id=admin_user_id)
		if asset_group_id:
			queryset = queryset.filter(groups__id=asset_group_id)
		return queryset


class AssetGroupViewSet(IDInFilterMixin, BulkModelViewSet):
	queryset = AssetGroup.objects.all()
	serializer_class = serializers.AssetGroupSerializer
	permission_classes = (IsSuperUser,)

class AssetUpdateGroupApi(generics.RetrieveUpdateAPIView):
	queryset = Asset.objects.all()
	serializer_class = serializers.AssetUpdateGroupSerializer
	permission_classes = (IsSuperUser,)

## update the asset group, and add or delete the asset to the group
class AssetGroupUpdateApi(generics.RetrieveUpdateAPIView):
	queryset = AssetGroup.objects.all()
	serializer_class = serializers.AssetGroupUpdateSerializer
	permission_classes = (IsSuperUser,)

## update the asset group, and add or delete the system_user to the group
class AssetGroupUpdateSystemUserApi(generics.RetrieveUpdateAPIView):
	queryset = AssetGroup.objects.all()
	serializer_class = serializers.AssetGroupUpdateSystemUserSerializer
	permission_classes = (IsSuperUser,)

## update the IDC, and add or delete the assets to the IDC
class IDCupdateAssetsApi(generics.RetrieveUpdateAPIView):
	queryset = IDC.objects.all()
	serializer_class = serializers.IDCUpdateAssetsSerializer
	permission_classes = (IsSuperUser,)

class IDCViewSet(IDInFilterMixin, BulkModelViewSet):
	"""API endpoint that allows IDC to be viewed or edited."""
	queryset = IDC.objects.all()
	serializer_class = serializers.IDCSerializer
	permission_classes = (IsSuperUser,)

class AdminUserViewSet(IDInFilterMixin, BulkModelViewSet):
	queryset = AdminUser.objects.all()
	serializer_class = serializers.AdminUserSerializer
	permission_classes = (IsSuperUser,)


class SystemUserViewSet(IDInFilterMixin, BulkModelViewSet):
	queryset = SystemUser.objects.all()
	serializer_class = serializers.SystemUserSerializer
	permission_classes = (IsSuperUser,)


class SystemUserUpdateApi(generics.RetrieveUpdateAPIView):
	queryset = Asset.objects.all()
	serializer_class = serializers.AssetUpdateSystemUserSerializer
	permission_classes = (IsSuperUser,)

class SystemUserUpdateAssetsApi(generics.RetrieveUpdateAPIView):
	queryset = SystemUser.objects.all()
	serializer_class = serializers.SystemUserUpdateAssetsSerializer
	permission_classes = (IsSuperUser,)

class SystemUserUpdateAssetGroupApi(generics.RetrieveUpdateAPIView):
	queryset = SystemUser.objects.all()
	serializer_class = serializers.SystemUserUpdateAssetGroupSerializer
	permission_classes = (IsSuperUser,)


# class IDCAssetsApi(generics.ListAPIView):
#     model = IDC
#     serializer_class = serializers.AssetSerializer
#
#     def get(self, request, *args, **kwargs):
#         filter_kwargs = {self.lookup_field: self.kwargs[self.lookup_field]}
#         self.object = get_object_or_404(self.model, **filter_kwargs)
#         return super(IDCAssetsApi, self).get(request, *args, **kwargs)
#
#     def get_queryset(self):
#         return self.object.assets.all()


class AssetListUpdateApi(IDInFilterMixin, ListBulkCreateUpdateDestroyAPIView):
	queryset = Asset.objects.all()
	serializer_class = serializers.AssetSerializer
	permission_classes = (IsSuperUser,)


class SystemUserAuthApi(APIView):
	permission_classes = (IsSuperUserOrTerminalUser,)

	def get(self, request, *args, **kwargs):
		system_user_id = request.query_params.get('system_user_id', -1)
		system_user_username = request.query_params.get('system_user_username', '')

		system_user = get_object_or_none(SystemUser, id=system_user_id, username=system_user_username)

		if system_user:
			if system_user.password:
				password = signer.sign(system_user.password)
			else:
				password = signer.sign('')

			if system_user.private_key:
				private_key = signer.sign(system_user.private_key)
			else:
				private_key = signer.sign(None)

			response = {
				'id': system_user.id,
				'password': password,
				'private_key': private_key,
			}

			return Response(response)
		else:
			return Response({'msg': 'error system user id or username'}, status=401)


class TagViewSet(IDInFilterMixin, BulkModelViewSet):
	queryset = Tag.objects.all()
	serializer_class = serializers.TagSerializer
	permission_classes = (IsSuperUser,)

## update the IDC, and add or delete the assets to the IDC
class TagUpdateAssetsApi(generics.RetrieveUpdateAPIView):
	queryset = Tag.objects.all()
	serializer_class = serializers.TagUpdateAssetsSerializer
	permission_classes = (IsSuperUser,)