# -*- coding: utf-8 -*-
from flask import Flask
from models import db
from config import Config

def create_admin_note():
    """顯示 LDAP 管理員設定說明。"""
    print("✅ 資料庫已初始化！")
    print("   ===================================")
    print("   📋 LDAP 管理員設定說明")
    print("   ===================================")
    print("   由於系統使用 LDAP 驗證，管理員權限需要在首次登入後手動設定。")
    print("   ")
    print("   步驟:")
    print("   1. 使用 AD 帳號登入系統")
    print("   2. 直接在資料庫中將該用戶的 role 更新為 'admin'")
    print("   3. 或修改 auth.py 中新用戶的預設權限設定")
    print("   ")
    print("   SQL 範例:")
    print("   UPDATE ts_user SET role='admin' WHERE username='你的AD帳號';")
    print("   ===================================")

def init_database(app):
    """初始化資料庫：刪除所有現有資料表並重新建立。"""
    with app.app_context():
        print("🔄 開始進行資料庫初始化...")
        # 為了安全，先刪除所有表格，再重新建立
        db.drop_all()
        print("   - 所有舊資料表已刪除。")
        db.create_all()
        print("   - 所有新資料表已根據 models.py 建立。")
        print("✅ 資料庫結構已成功初始化！")

if __name__ == '__main__':
    # 建立一個暫時的 Flask app 來提供資料庫操作所需的應用程式上下文
    app = Flask(__name__)
    app.config.from_object(Config)
    
    # 將資料庫物件與 app 綁定
    db.init_app(app)

    print("=================================================")
    print("          ⚠️  資料庫初始化腳本  ⚠️")
    print("=================================================")
    print("此腳本將會刪除所有現有的資料，並重新建立資料庫結構。")
    print("這個操作是不可逆的！")
    
    # 讓使用者確認操作
    confirmation = input("👉 您確定要繼續嗎？ (yes/no): ")

    if confirmation.lower() == 'yes':
        init_database(app)
        create_admin_note()
        print("\n🎉 全部完成！")
    else:
        print("❌ 操作已取消。")