from flask import redirect, render_template, request, session, url_for
from notifications_utils.broadcast_areas import broadcast_area_libraries
from notifications_utils.template import BroadcastPreviewTemplate
from orderedset import OrderedSet

from app import current_service
from app.main import main
from app.main.forms import BroadcastAreaForm, SearchByNameForm
from app.utils import service_has_permission, user_has_permissions


@main.route('/services/<uuid:service_id>/broadcast-dashboard')
@user_has_permissions()
@service_has_permission('broadcast')
def broadcast_dashboard(service_id):
    return render_template(
        'views/broadcast/dashboard.html',
    )


@main.route('/services/<uuid:service_id>/broadcast')
@user_has_permissions('send_messages')
@service_has_permission('broadcast')
def broadcast(service_id):
    if 'broadcast_areas' in session:
        session.pop('broadcast_areas')
    return redirect(url_for(
        '.preview_broadcast_areas',
        service_id=current_service.id,
    ))


@main.route('/services/<uuid:service_id>/broadcast/areas')
@user_has_permissions('send_messages')
@service_has_permission('broadcast')
def preview_broadcast_areas(service_id):
    selected_areas_ids = session.get('broadcast_areas', [])
    return render_template(
        'views/broadcast/preview-areas.html',
        selected=list(broadcast_area_libraries.get_areas(
            *selected_areas_ids
        )),
        area_polygons=broadcast_area_libraries.get_polygons_for_areas_lat_long(
            *selected_areas_ids
        )
    )


@main.route('/services/<uuid:service_id>/broadcast/libraries')
@user_has_permissions('send_messages')
@service_has_permission('broadcast')
def choose_broadcast_library(service_id):
    return render_template(
        'views/broadcast/libraries.html',
        libraries=broadcast_area_libraries,
        selected=broadcast_area_libraries.get_areas(
            *session.get('broadcast_areas', [])
        ),
    )


@main.route('/services/<uuid:service_id>/broadcast/libraries/<library_slug>', methods=['GET', 'POST'])
@user_has_permissions('send_messages')
@service_has_permission('broadcast')
def choose_broadcast_area(service_id, library_slug):
    library = broadcast_area_libraries.get(library_slug)
    form = BroadcastAreaForm.from_library(library)
    if form.validate_on_submit():
        if not session.get('broadcast_areas'):
            session['broadcast_areas'] = []
        session['broadcast_areas'] = session['broadcast_areas'] + form.areas.data
        session['broadcast_areas'] = list(OrderedSet(
            session['broadcast_areas']
        ))
        return redirect(url_for(
            '.preview_broadcast_areas',
            service_id=current_service.id,
        ))
    return render_template(
        'views/broadcast/areas.html',
        form=form,
        search_form=SearchByNameForm(),
        show_search_form=(len(form.areas.choices) > 7),
        page_title=library.name,
    )


@main.route('/services/<uuid:service_id>/broadcast/remove/<area_slug>')
@user_has_permissions('send_messages')
@service_has_permission('broadcast')
def remove_broadcast_area(service_id, area_slug):
    session['broadcast_areas'] = list(filter(
        lambda saved_area_id: saved_area_id != area_slug,
        session.get('broadcast_areas', []),
    ))

    return redirect(url_for(
        '.preview_broadcast_areas',
        service_id=current_service.id,
    ))


@main.route('/services/<uuid:service_id>/broadcast/preview', methods=['GET', 'POST'])
@user_has_permissions('send_messages')
@service_has_permission('broadcast')
def preview_broadcast_message(service_id):
    if request.method == 'POST':
        return 'OK'
    selected_areas = session.get('broadcast_areas', [])
    return render_template(
        'views/broadcast/preview-message.html',
        selected=list(broadcast_area_libraries.get_areas(
            *selected_areas
        )),
        template=BroadcastPreviewTemplate({
            'content': 'Message here',
            'template_type': 'broadcast',
        })
    )