from flask import redirect, url_for, abort

import screens

from core.flask_app import app


@app.route("/game/<slug>/elements/<int:element_id>/delete", methods=["POST"])
def element_delete(slug, element_id):
    element = screens.get_element(slug, element_id)
    if not element:
        abort(404)
    screen_id = element["screen_id"]
    parent_id = element.get("parent_id")
    screens.delete_element(slug, element_id)
    # Resélectionne le conteneur parent (s'il y en a un) après suppression,
    # plutôt que de laisser le panneau de droite vide : sinon, supprimer un
    # élément fait brusquement "sauter" l'éditeur vers un état sans rien de
    # sélectionné.
    if parent_id:
        return redirect(url_for("screen_edit", slug=slug, screen_id=screen_id, selected=parent_id))
    return redirect(url_for("screen_edit", slug=slug, screen_id=screen_id))