from flask import Blueprint, render_template, request, redirect, url_for, flash, session
import requests
from frontend.config import Config

auth_bp = Blueprint('auth', __name__)

@auth_bp.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        remember_me = request.form.get('remember_me') == 'on'

        response = requests.post(f'{Config.BASE_API_URL}/admin/login/', json={"username": username, "password": password}, params={"remember_me": remember_me})

        if response.status_code == 200:
            session['token'] = response.json().get('access_token')
            flash("登录成功", "success")
            return render_template("refresh.html")
        else:
            flash("登录失败，请检查用户名或密码", "danger")

    return render_template('login.html')

@auth_bp.route('/logout')
def logout():
    session.clear()
    flash("已退出登录", "info")
    return redirect(url_for('auth.login'))