import glob
import importlib
import json
import os
from flask import Blueprint, Flask, jsonify, request
from flask_migrate import Migrate

from config import getConfig

from controllers import R
from enums.load_enum import LoadEnum
from mauth import HasPerm
from mauth.login_user_holder import LoginUserHolder
from mauth.mtoken import redis_client
from mexception import ExceptionConfig
from mlogging import LoggingConfig
from mlogging.request_log import RequestLog
from models import db
from services import LowCodeServiceUtil
from datetime import datetime
from werkzeug.middleware.proxy_fix import ProxyFix

from tools import ant_matcher

app = Flask(__name__)
# 应用 ProxyFix 中间件
# x_for=1 表示信任代理服务器的 X-Forwarded-For 头中的第一个 IP 地址
app.wsgi_app = ProxyFix(app.wsgi_app, x_for=1)

# 获取controllers目录下所有的.py文件路径
controllers_dir = 'controllers'
controller_files = glob.glob(os.path.join(controllers_dir, '*.py'))

# 过滤掉__init__.py文件
controller_files = [f for f in controller_files if '__init__' not in f]

# 动态导入每个控制器模块并注册蓝图
for file_path in controller_files:
    module_name = os.path.splitext(os.path.basename(file_path))[0]  # 去掉'.py'后缀
    blueprint_name = module_name.replace('_controller','')
    spec = importlib.util.spec_from_file_location(module_name, file_path)
    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    
    # 检查模块是否有对应的blueprint属性，如果有，则注册它
    if hasattr(module, blueprint_name) and isinstance(getattr(module, blueprint_name), Blueprint):
        app.register_blueprint(getattr(module, blueprint_name))

# 从配置对象中加载
app.config.from_object(getConfig())

# 初始化db
db.init_app(app)
# 初始化redis
redis_client.init_app(app)
app.before_request(LoginUserHolder.before_request)

# 自定义日期转换器，格式化日期
class CustomJSONEncoder(json.JSONEncoder):
    def default(self, obj):
        try:
            if isinstance(obj, datetime):
                return obj.strftime('%Y-%m-%d %H:%M:%S')  # 自定义日期时间格式
            iterable = iter(obj)
        except TypeError:
            pass
        else:
            return list(iterable)
        return json.JSONEncoder.default(self, obj)

app.json_encoder = CustomJSONEncoder

# 配置字典
LoadEnum(app)
# 配置日志
LoggingConfig(app)
# 配置请求日志
RequestLog(app)
# 配置异常处理
ExceptionConfig(app)
# 执行数据库迁移相关操作。flask db指令不能直接使用，需要获取Migrate实例，这里注册Migrate实例
migrate = Migrate(app, db)
# 配置代码生成器-仅开发环境配置
if app.config.get("ENV") == "dev":
    from generator import CodeGenerator
    CodeGenerator(app)
# 低代码-通用下拉接口
@app.route('/<moduleName>/<tableName>/select', methods=['POST'])
@HasPerm(name="通过下拉")
def lowcode_select(moduleName, tableName):
    return R.data(LowCodeServiceUtil.select(moduleName=moduleName, tableName=tableName))

# 拦截器-demo环境拦截
if app.config.get("ENV") == "demo":
    @app.before_request
    def check_blacklist():
        path = request.path
        BLACK_LIST = app.config.get("BLACK_LIST")
        if any(ant_matcher(path, blacklisted_path) for blacklisted_path in BLACK_LIST):
            response = jsonify({"code": 99990406, "msg": "您没有资源访问权限，请联系管理员！", "data": None})
            return response


if __name__ == '__main__':
    app.run(host="0.0.0.0", port=app.config.get("PORT", 5000), debug=app.config.get("DEBUG", True))