from flask import Flask, current_app
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager
from flask_migrate import Migrate
from flask_jwt_extended import JWTManager
from config import Config
import logging
from logging.handlers import RotatingFileHandler
import os

db = SQLAlchemy()
migrate = Migrate()
login_manager = LoginManager()
jwt = JWTManager()

def create_app(config_class=Config):
    app = Flask(__name__)
    app.config.from_object(config_class)

    db.init_app(app)
    migrate.init_app(app, db)
    login_manager.init_app(app)
    jwt.init_app(app)

    # 设置日志记录器
    if not app.debug:
        if not os.path.exists('logs'):
            os.mkdir('logs')
        file_handler = RotatingFileHandler('logs/app.log', maxBytes=10240, backupCount=10, delay=True)
        file_handler.setFormatter(logging.Formatter(
            '%(asctime)s %(levelname)s: %(message)s [in %(pathname)s:%(lineno)d]'))
        file_handler.setLevel(logging.INFO)
        app.logger.addHandler(file_handler)

        app.logger.setLevel(logging.INFO)
        app.logger.info('Application startup')

    from app import models
    from app.models import RoleTemplate, Permission

    def init_database():
        db.create_all()
        models.UserField.init_default_fields()
        models.Department.init_default_departments()
        models.Department.init_default_fields()
        models.RoleField.init_default_fields()
        models.Role.init_default_roles()
        models.PermissionField.init_default_fields()
        models.Permission.init_default_permissions()

    def init_role_templates():
        templates = [
            {
                'name': '管理员',
                'description': '系统管理员，拥有所有权限',
                'permissions': ['用户管理', '角色管理', '部门管理', '权限管理']
            },
            {
                'name': '普通用户',
                'description': '普通用户，拥有基本权限',
                'permissions': ['查看个人信息', '修改个人信息']
            },
        ]

        for template in templates:
            if not RoleTemplate.query.filter_by(name=template['name']).first():
                permissions = [Permission.query.filter_by(name=p).first() for p in template['permissions']]
                RoleTemplate.create_template(
                    name=template['name'],
                    description=template['description'],
                    permissions=[p for p in permissions if p]
                )

    with app.app_context():
        init_database()
        init_role_templates()

    from app.auth import bp as auth_bp
    app.register_blueprint(auth_bp)

    from app.departments import bp as departments_bp
    app.register_blueprint(departments_bp, url_prefix='/departments')

    from app.roles import bp as roles_bp
    app.register_blueprint(roles_bp, url_prefix='/roles')

    from app.permissions import bp as permissions_bp
    app.register_blueprint(permissions_bp, url_prefix='/permissions')

    from app.users import bp as users_bp
    app.register_blueprint(users_bp, url_prefix='/users')

    return app