import uuid

from flask import (
    Blueprint,
    jsonify,
    request,
    current_app
)

from app import api_user, encryption
from app.aws_sqs import add_notification_to_queue
from app.dao import (
    templates_dao,
    users_dao,
    services_dao,
    notifications_dao
)
from app.schemas import (
    email_notification_schema,
    sms_template_notification_schema,
    notification_status_schema
)
from app.celery.tasks import send_sms, send_email
from sqlalchemy.orm.exc import NoResultFound

notifications = Blueprint('notifications', __name__)

from app.errors import register_errors

register_errors(notifications)


def create_notification_id():
    return str(uuid.uuid4())


@notifications.route('/<string:notification_id>', methods=['GET'])
def get_notifications(notification_id):
    try:
        notification = notifications_dao.get_notification(api_user['client'], notification_id)
        return jsonify({'notification': notification_status_schema.dump(notification).data}), 200
    except NoResultFound:
        return jsonify(result="error", message="not found"), 404


@notifications.route('/sms', methods=['POST'])
def create_sms_notification():
    notification, errors = sms_template_notification_schema.load(request.get_json())
    if errors:
        return jsonify(result="error", message=errors), 400

    template = templates_dao.dao_get_template_by_id_and_service_id(
        template_id=notification['template'],
        service_id=api_user['client']
    )

    if not template:
        return jsonify(result="error", message={'template': ['Template not found']}), 400

    service = services_dao.dao_fetch_service_by_id(api_user['client'])

    if service.restricted:
        if notification['to'] not in [user.email_address for user in service.users]:
            return jsonify(result="error", message={'to': ['Invalid phone number for restricted service']}), 400

    notification_id = create_notification_id()

    send_sms.apply_async((
        api_user['client'],
        notification_id,
        encryption.encrypt(notification)),
        queue='sms')
    return jsonify({'notification_id': notification_id}), 201


@notifications.route('/email', methods=['POST'])
def create_email_notification():
    notification, errors = email_notification_schema.load(request.get_json())
    if errors:
        return jsonify(result="error", message=errors), 400

    template = templates_dao.dao_get_template_by_id_and_service_id(
        template_id=notification['template'],
        service_id=api_user['client']
    )

    if not template:
        return jsonify(result="error", message={'template': ['Template not found']}), 400

    service = services_dao.dao_fetch_service_by_id(api_user['client'])

    if service.restricted:
        if notification['to'] not in [user.email_address for user in service.users]:
            return jsonify(result="error", message={'to': ['Email address not permitted for restricted service']}), 400

    notification_id = create_notification_id()

    send_email.apply_async((
        api_user['client'],
        notification_id,
        template.subject,
        "{}@{}".format(service.email_from, current_app.config['NOTIFY_EMAIL_DOMAIN']),
        encryption.encrypt(notification)),
        queue='email')
    return jsonify({'notification_id': notification_id}), 201


@notifications.route('/sms/service/<service_id>', methods=['POST'])
def create_sms_for_service(service_id):
    resp_json = request.get_json()

    notification, errors = sms_template_notification_schema.load(resp_json)
    if errors:
        return jsonify(result="error", message=errors), 400

    template_id = notification['template']
    job_id = notification['job']

    # TODO: job/job_id is in notification and can used to update job status

    # TODO: remove once beta is reading notifications from the queue
    template = templates_dao.get_model_templates(template_id)

    if template.service.id != uuid.UUID(service_id):
        message = "Invalid template: id {} for service id: {}".format(template.id, service_id)
        return jsonify(result="error", message=message), 400

    notification_id = add_notification_to_queue(service_id, template_id, 'sms', notification)
    return jsonify({'notification_id': notification_id}), 201