from flask import Blueprint, Flask, redirect, render_template, url_for
from flask_assets import Environment

from starfall.web.blueprints.base import BaseBlueprint
from starfall.web.controllers.secure.login import LoginController
from starfall.web.controllers.secure.register import RegisterController


class AdminBlueprint(BaseBlueprint):
    def __init__(self, blueprint: Blueprint, assets: Environment, app: Flask):
        super().__init__(blueprint, assets, app)
        blueprint.add_url_rule("/secure/", view_func=self.secure)
        blueprint.add_url_rule("/secure/login/", methods=["GET", "POST"], view_func=self.login)
        blueprint.add_url_rule("/secure/register/", methods=["GET", "POST"], view_func=self.register)

    def secure(self):
        return redirect(url_for("starfall.login"))

    def login(self):
        LoginController.apply(self)
        return render_template("secure/login.jinja", bp=self)

    def register(self):
        RegisterController.apply(self)
        return render_template("secure/register.jinja", bp=self)