from flask import render_template, request, redirect, url_for, flash
from flask_login import login_user, login_required, logout_user, current_user
from scripts.models import db, User

def init_auth_routes(app):
    @app.route('/login', methods=['GET', 'POST'])
    def login():
        if request.method == 'POST':
            phone = request.form['phone']
            user = User.query.filter_by(phone=phone).first()
            if user:
                login_user(user)
                app.logger.info(f'用户 {user.name}({user.phone}) 登录成功')
                # 无论是统计员还是装维员，都跳转到仪表盘
                return redirect(url_for('dashboard'))
            else:
                app.logger.warning(f'登录失败：用户不存在 (手机号: {phone})')
                flash('用户不存在')
        return render_template('login.html')

    @app.route('/logout')
    @login_required
    def logout():
        app.logger.info(f'用户 {current_user.name}({current_user.phone}) 登出')
        logout_user()
        return redirect(url_for('login'))