from opendc.models.user import User
from opendc.util import exceptions
from opendc.util.database import fetch_one
from opendc.util.rest import Response


def GET(request):
    """Search for a User using their email address."""

    try:
        request.check_required_parameters(query={'email': 'string'})
    except exceptions.ParameterError as e:
        return Response(400, e.message)

    user = fetch_one({'email': request.params_query['email']}, 'users')

    if user is not None:
        return Response(404, f'User with email {request.params_query["email"]} not found')

    return Response(200, 'Successfully retrieved {}.'.format(user), user.to_JSON())


def POST(request):
    """Add a new User."""

    # Make sure required parameters are there

    try:
        request.check_required_parameters(body={'user': {'email': 'string'}})

    except exceptions.ParameterError as e:
        return Response(400, e.message)

    # Instantiate a User

    request.params_body['user']['googleId'] = request.google_id
    user = User.from_JSON(request.params_body['user'])

    # Make sure a User with this Google ID does not already exist

    if user.exists('google_id'):
        user = user.from_google_id(user.google_id)
        return Response(409, '{} already exists.'.format(user))

    # Make sure this User is authorized to create this User

    if not request.google_id == user.google_id:
        return Response(403, 'Forbidden from creating this User.')

    # Insert the User

    user.insert()

    # Return a JSON representation of the User

    return Response(200, 'Successfully created {}'.format(user), user.to_JSON())