from flask import request, render_template, redirect, url_for, flash
from flask_login import login_required

from app.main import main
from app.main.forms import TemplateForm
from app.main.dao.services_dao import get_service_by_id
from app.main.dao import templates_dao as dao
from client.errors import HTTPError


@main.route("/services/<int:service_id>/templates")
@login_required
def manage_service_templates(service_id):
    try:
        templates = dao.get_service_templates(service_id)['data']
    except HTTPError as e:
        if e.status_code == 404:
            abort(404)
        else:
            raise e
    return render_template(
        'views/manage-templates.html',
        service_id=service_id,
        templates=templates)


@main.route("/services/<int:service_id>/templates/add", methods=['GET', 'POST'])
@login_required
def add_service_template(service_id):
    try:
        service = dao.get_service_by_id(service_id)['data']
    except HTTPError as e:
        if e.status_code == 404:
            abort(404)
        else:
            raise e

    form = TemplateForm()

    if form.validate_on_submit():
        dao.insert_service_template(
            form.name.data, form.template_type.data, form.content.data, service_id)
        return redirect(url_for(
            '.manage_service_templates', service_id=service_id))
    return render_template(
        'views/edit-template.html',
        h1='Add template',
        form=form,
        service_id=service_id)


@main.route("/services/<int:service_id>/templates/<int:template_id>", methods=['GET', 'POST'])
@login_required
def edit_service_template(service_id, template_id):
    try:
        template = dao.get_service_template(service_id, template_id)['data']
    except HTTPError as e:
        if e.status_code == 404:
            abort(404)
        else:
            raise e
    form = TemplateForm(form_data=template)

    if form.validate_on_submit():
        dao.update_service_template(
            template_id, form.name.data, form.template_type.data,
            form.content.data, service_id)
        return redirect(url_for('.manage_service_templates', service_id=service_id))

    return render_template(
        'views/edit-template.html',
        h1='Edit template',
        form=form,
        service_id=service_id,
        template_id=template_id)


@main.route("/services/<int:service_id>/templates/<int:template_id>/delete", methods=['GET', 'POST'])
@login_required
def delete_service_template(service_id, template_id):
    try:
        template = dao.get_service_template(service_id, template_id)['data']
    except HTTPError as e:
        if e.status_code == 404:
            abort(404)
        else:
            raise e

    form = TemplateForm(form_data=template)

    if form.validate_on_submit():
        dao.delete_service_template(service_id, template_id)
        return redirect(url_for('.manage_service_templates', service_id=service_id))

    flash('Are you sure you want to delete ‘{}’?'.format(form.name.data), 'delete')
    return render_template(
        'views/edit-template.html',
        h1='Edit template',
        form=form,
        service_id=service_id,
        template_id=template_id)