from django.db import transaction

from drf_spectacular.utils import extend_schema

from rest_framework.generics import GenericAPIView  # type: ignore
from rest_framework import status  # type: ignore
from rest_framework.request import HttpRequest  # type: ignore
from rest_framework.response import Response  # type: ignore
from rest_framework.permissions import IsAuthenticated  # type: ignore

from core.utils.views import BaseApiViewMixin
from core.apps.companies.serializers import (
    RetrieveCompanySerializer,
    CreateCompanySerializer,
)
from core.apps.companies.models import CompanyModel, CompanyAccountModel


######################################################################
# @api-view  |  POST, GET - me/companies
######################################################################
@extend_schema(tags=["Me"])
class MeCompanyApiView(BaseApiViewMixin, GenericAPIView):  # type: ignore
    permission_classes = [IsAuthenticated]

    method_permission_classes = {}
    method_serializer_class = {
        "post": CreateCompanySerializer,
        "get": RetrieveCompanySerializer,
    }

    def get(self, request: HttpRequest, *args: object, **kwargs: object) -> Response:
        companies = CompanyModel.objects.filter(accounts__user=request.user)
        ser = RetrieveCompanySerializer(instance=companies, many=True)
        return Response(ser.data, status.HTTP_200_OK)
    
    def create(self, request: HttpRequest, *args: object, **kwargs: object) -> Response:
        with transaction.atomic():
            serializer = CreateCompanySerializer(data=request.data)  # type: ignore
            serializer.is_valid(raise_exception=True)
            company = serializer.save()  # type: ignore

            account = CompanyAccountModel(company=company, user=request.user)
            account.save()
        return Response(serializer.data, status.HTTP_201_CREATED)