Module supertokens_python.recipe.passwordless.api.create_code

# Copyright (c) 2021, VRAI Labs and/or its affiliates. All rights reserved.
#
# This software is licensed under the Apache License, Version 2.0 (the
# "License") as published by the Apache Software Foundation.
#
# You may not use this file except in compliance with the License. You may
# obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from typing import Union

import phonenumbers  # type: ignore
from phonenumbers import format_number, parse  # type: ignore
from supertokens_python.exceptions import raise_bad_input_exception
from supertokens_python.recipe.passwordless.interfaces import (
    APIInterface, APIOptions, CreateCodePostGeneralErrorResponse)
from supertokens_python.recipe.passwordless.utils import (
    ContactEmailOnlyConfig, ContactEmailOrPhoneConfig, ContactPhoneOnlyConfig)


async def create_code(api_implementation: APIInterface, api_options: APIOptions):
    if api_implementation.disable_create_code_post:
        return None

    body = await api_options.request.json()

    if body is None:
        raise_bad_input_exception("Please provide a JSON body")

    email: Union[str, None] = None
    phone_number: Union[str, None] = None

    if ('email' in body and 'phoneNumber' in body) or (
            'email' not in body and 'phoneNumber' not in body):
        raise_bad_input_exception(
            'Please provide exactly one of email or phoneNumber')

    if 'email' not in body and isinstance(
            api_options.config.contact_config, ContactEmailOnlyConfig):
        raise_bad_input_exception(
            'Please provide an email since you have set the contactMethod to "EMAIL"')

    if 'phoneNumber' not in body and isinstance(
            api_options.config.contact_config, ContactPhoneOnlyConfig):
        raise_bad_input_exception(
            'Please provide a phoneNumber since you have set the contactMethod to "PHONE"')

    if 'email' in body:
        email = body['email']
    if 'phoneNumber' in body:
        phone_number = body['phoneNumber']

    if email is not None and (
        isinstance(api_options.config.contact_config, (ContactEmailOnlyConfig, ContactEmailOrPhoneConfig))
    ):
        email = email.strip()
        validation_error = await api_options.config.contact_config.validate_email_address(email)
        if validation_error is not None:
            api_options.response.set_json_content(
                CreateCodePostGeneralErrorResponse(validation_error).to_json())
            return api_options.response

    if phone_number is not None and (
        isinstance(api_options.config.contact_config, (ContactEmailOrPhoneConfig, ContactPhoneOnlyConfig))
    ):
        validation_error = await api_options.config.contact_config.validate_phone_number(phone_number)
        if validation_error is not None:
            api_options.response.set_json_content(
                CreateCodePostGeneralErrorResponse(validation_error).to_json())
            return api_options.response
        try:
            phone_number_formatted: str = format_number(
                parse(phone_number, None),
                phonenumbers.PhoneNumberFormat.E164)  # type: ignore
            phone_number = phone_number_formatted
        except Exception:
            phone_number = phone_number.strip()
    result = await api_implementation.create_code_post(
        email=email, phone_number=phone_number, api_options=api_options, user_context={})
    api_options.response.set_json_content(result.to_json())

    return api_options.response