app.py

from flask import Flask, redirect, url_for
from flask_mail import Mail
from flask_login import LoginManager, login_required
from dotenv import load_dotenv
from auth import auth_bp  # 匯入藍圖
from models import db, User
from dashboard import dashboard_bp
from utils.password_manager import password_manager_bp
from utils.password_strength_checker import strength_checker_bp
from utils.password_generator import generator_bp
from utils.password_encryptor import encryptor_bp
import os
from datetime import timedelta

app = Flask(__name__)
app.jinja_env.add_extension('jinja2.ext.do')  # 啟用 do 擴展

# 載入環境變數
load_dotenv()

# 配置 Flask 應用
app.config['SECRET_KEY'] = os.getenv('SECRET_KEY', 'fallback_secret_key')
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///users.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.config['PERMANENT_SESSION_LIFETIME'] = timedelta(minutes=20)  # session 有效期 20 分鐘

# 設定郵件參數
app.config['MAIL_SERVER'] = 'smtp.gmail.com'
app.config['MAIL_PORT'] = 587
app.config['MAIL_USE_TLS'] = True
app.config['MAIL_USE_SSL'] = False
MAIL_USERNAME = os.getenv('MAIL_USERNAME')  # 發送的Gmail信箱
app.config['MAIL_USERNAME'] = MAIL_USERNAME  
app.config['MAIL_PASSWORD'] = os.getenv('MAIL_PASSWORD')  # Gmail應用專用密碼

# 初始化 Flask-Mail
mail = Mail(app)

# 初始化擴展
db.init_app(app)

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'auth.login'

# 加載使用者
@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))

# 註冊藍圖
app.register_blueprint(auth_bp)
app.register_blueprint(dashboard_bp)
app.register_blueprint(password_manager_bp)
app.register_blueprint(strength_checker_bp)
app.register_blueprint(generator_bp)
app.register_blueprint(encryptor_bp)

# 初始化資料庫
with app.app_context():
    db.create_all()

# 主頁路由
# 在 app.py 內加入一個公開的首頁路由
@login_required
@app.route('/')
def index():
    return redirect(url_for('dashboard.dashboard'))



if __name__ == '__main__':
    app.run(debug=True)