from app import db
from app.dao.dao_utils import transactional
from app.models import ServiceSmsSender


@transactional
def insert_or_update_service_sms_sender(service, sms_sender, inbound_number_id=None):
    result = db.session.query(
        ServiceSmsSender
    ).filter(
        ServiceSmsSender.service_id == service.id
    ).update(
        {'sms_sender': sms_sender,
         'inbound_number_id': inbound_number_id
         }
    )
    if result == 0:
        new_sms_sender = ServiceSmsSender(sms_sender=sms_sender,
                                          service=service,
                                          is_default=True,
                                          inbound_number_id=inbound_number_id
                                          )
        db.session.add(new_sms_sender)


def insert_service_sms_sender(service, sms_sender):
    """
    This method is called from create_service which is wrapped in a transaction.
    """
    new_sms_sender = ServiceSmsSender(sms_sender=sms_sender,
                                      service=service,
                                      is_default=True
                                      )
    db.session.add(new_sms_sender)


def dao_get_service_sms_senders_by_id(service_id, service_sms_sender_id):
    return ServiceSmsSender.query.filter_by(
        id=service_sms_sender_id,
        service_id=service_id
    ).one()


def dao_get_sms_senders_by_service_id(service_id):
    return ServiceSmsSender.query.filter_by(service_id=service_id).all()


@transactional
def dao_add_sms_sender_for_service(service_id, sms_sender, is_default, inbound_number_id=None):
    old_default = _get_existing_default(service_id=service_id)
    if is_default:
        _reset_old_default_to_false(old_default)
    else:
        _raise_when_no_default(old_default)

    new_sms_sender = ServiceSmsSender(
        service_id=service_id,
        sms_sender=sms_sender,
        is_default=is_default,
        inbound_number_id=inbound_number_id
    )

    db.session.add(new_sms_sender)
    return new_sms_sender


@transactional
def dao_update_service_sms_sender(service_id, service_sms_sender_id, is_default, sms_sender=None):
    old_default = _get_existing_default(service_id)
    if is_default:
        _reset_old_default_to_false(old_default)
    else:
        if old_default.id == service_sms_sender_id:
            raise Exception("You must have at least one SMS sender as the default")

    sms_sender_to_update = ServiceSmsSender.query.get(service_sms_sender_id)
    sms_sender_to_update.is_default = is_default
    if not sms_sender_to_update.inbound_number_id and sms_sender:
        sms_sender_to_update.sms_sender = sms_sender
    db.session.add(sms_sender_to_update)
    return sms_sender_to_update


def _get_existing_default(service_id):
    sms_senders = dao_get_sms_senders_by_service_id(service_id=service_id)
    if sms_senders:
        old_default = [x for x in sms_senders if x.is_default]
        if len(old_default) == 1:
            return old_default[0]
        else:
            raise Exception(
                "There should only be one default sms sender for each service. Service {} has {}".format(
                    service_id,
                    len(old_default)
                )
            )
    return None


def _reset_old_default_to_false(old_default):
    if old_default:
        old_default.is_default = False
        db.session.add(old_default)


def _raise_when_no_default(old_default):
    # check that the update is not updating the only default to false
    if not old_default:
        raise Exception("You must have at least one letter contact as the default.", 400)