# Module imports
from plane.authentication.adapter.credential import CredentialAdapter
from plane.db.models import User
from plane.authentication.adapter.error import (
    AUTHENTICATION_ERROR_CODES,
    AuthenticationException,
)


class EmailProvider(CredentialAdapter):

    provider = "email"

    def __init__(
        self,
        request,
        key=None,
        code=None,
        is_signup=False,
    ):
        super().__init__(request, self.provider)
        self.key = key
        self.code = code
        self.is_signup = is_signup

    def set_user_data(self):
        if self.is_signup:
            # Check if the user already exists
            if User.objects.filter(email=self.key).exists():
                raise AuthenticationException(
                    error_message="USER_ALREADY_EXIST",
                    error_code=AUTHENTICATION_ERROR_CODES[
                        "USER_ALREADY_EXIST"
                    ],
                )

            super().set_user_data(
                {
                    "email": self.key,
                    "user": {
                        "avatar": "",
                        "first_name": "",
                        "last_name": "",
                        "provider_id": "",
                        "is_password_autoset": False,
                    },
                }
            )
            return
        else:
            user = User.objects.filter(
                email=self.key,
            ).first()

            # User does not exists
            if not user:
                raise AuthenticationException(
                    error_message="USER_DOES_NOT_EXIST",
                    error_code=AUTHENTICATION_ERROR_CODES[
                        "USER_DOES_NOT_EXIST"
                    ],
                    payload={
                        "email": self.key,
                    },
                )

            # Check user password
            if not user.check_password(self.code):
                raise AuthenticationException(
                    error_message=(
                        "AUTHENTICATION_FAILED_SIGN_UP"
                        if self.is_signup
                        else "AUTHENTICATION_FAILED_SIGN_IN"
                    ),
                    error_code=AUTHENTICATION_ERROR_CODES[
                        (
                            "AUTHENTICATION_FAILED_SIGN_UP"
                            if self.is_signup
                            else "AUTHENTICATION_FAILED_SIGN_IN"
                        )
                    ],
                    payload={"email": self.key},
                )

            super().set_user_data(
                {
                    "email": self.key,
                    "user": {
                        "avatar": "",
                        "first_name": "",
                        "last_name": "",
                        "provider_id": "",
                        "is_password_autoset": False,
                    },
                }
            )
            return