#环境变量与安全配置：从 .env 加载配置，彻底隐藏密钥

# app/config.py（绝对不要这么写！！）
SECRET_KEY = "dev-12345678abcdefgh"
AWS_ACCESS_KEY_ID = "AKIA2V3X7Y8ZABCDEF"
AWS_SECRET_ACCESS_KEY = "rT6uI9oP0qW2eR4tY5uI6oP7aS8dF9gH0jK1l"

# app/config.py（安全！）
import os

SECRET_KEY = os.getenv("SECRET_KEY")
AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")

# app/config.py
import os
from dotenv import load_dotenv

# 第一步：加载.env文件（只在本地开发/测试环境加载）
# 如果是生产环境，直接用系统/云平台的环境变量，不需要这个文件
load_dotenv()

class BaseConfig:
    """所有环境通用的基础配置"""
    SECRET_KEY = os.getenv("SECRET_KEY", "unsafe-fallback-secret-only-for-testing")
    SQLALCHEMY_DATABASE_URI = os.getenv("DATABASE_URL", "sqlite:///app_default.db")
    SQLALCHEMY_TRACK_MODIFICATIONS = False  # 关闭这个功能可以节省内存

    # Session安全配置（生产环境必须开启！）
    SESSION_COOKIE_SECURE = os.getenv("FLASK_ENV") == "production"  # 生产环境必须用HTTPS
    SESSION_COOKIE_HTTPONLY = True  # 禁止JavaScript访问Cookie，防止XSS攻击
    SESSION_COOKIE_SAMESITE = "Lax"  # 防止CSRF攻击

    # 邮件配置（如果项目不用邮件，可以去掉）
    MAIL_SERVER = os.getenv("MAIL_SERVER")
    MAIL_PORT = int(os.getenv("MAIL_PORT", 587))
    MAIL_USE_TLS = os.getenv("MAIL_USE_TLS", "True").lower() == "true"
    MAIL_USERNAME = os.getenv("MAIL_USERNAME")
    MAIL_PASSWORD = os.getenv("MAIL_PASSWORD")


class DevelopmentConfig(BaseConfig):
    """本地开发环境专用配置"""
    DEBUG = True
    TESTING = False


class TestingConfig(BaseConfig):
    """单元测试/集成测试专用配置"""
    DEBUG = True
    TESTING = True
    SQLALCHEMY_DATABASE_URI = "sqlite:///:memory:"  # 内存数据库，测试完自动清空
    WTF_CSRF_ENABLED = False  # 测试时禁用CSRF，简化测试流程


class ProductionConfig(BaseConfig):
    """生产环境专用配置"""
    DEBUG = False
    TESTING = False

    # 生产环境必须配置真实的SECRET_KEY和DATABASE_URL，否则直接报错
    @classmethod
    def init_app(cls, app):
        super().init_app(app)
        if app.config["SECRET_KEY"] == "unsafe-fallback-secret-only-for-testing":
            raise ValueError("生产环境必须设置真实的SECRET_KEY！请检查系统环境变量。")
        if not app.config["MAIL_PASSWORD"] and app.config.get("MAIL_USE_TLS"):
            raise ValueError("生产环境邮件服务需要配置真实的密码！")


# 配置映射字典，方便根据环境切换
config = {
    "development": DevelopmentConfig,
    "testing": TestingConfig,
    "production": ProductionConfig,
    "default": DevelopmentConfig
}

# app/__init__.py
from flask import Flask
from app.config import config

def create_app(config_name="default"):
    app = Flask(__name__)
    # 加载对应环境的配置
    app.config.from_object(config[config_name])

    # 这里可以继续初始化数据库、蓝图等
    # ...

    return app