config.py

import os
from pathlib import Path


class BaseConfig:
    SECRET_KEY = os.getenv("SECRET_KEY", "change-me")
    DB_PATH = os.getenv("DB_PATH")  # None until CSV import creates a database
    ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD", "admin123")
    STATION_DEFAULT_PIN = os.getenv("STATION_DEFAULT_PIN")
    STATION_DEFAULT_NAME = os.getenv("STATION_DEFAULT_NAME", "Station")
    STATION_DEFAULT_MAX_LOGINS = int(os.getenv("STATION_DEFAULT_MAX_LOGINS", "1"))
    STATION_DEFAULT_PIN_LENGTH = int(os.getenv("STATION_DEFAULT_PIN_LENGTH", "6"))

    WTF_CSRF_ENABLED = True
    SESSION_COOKIE_SAMESITE = os.getenv("SESSION_COOKIE_SAMESITE", "Lax")
    SESSION_COOKIE_HTTPONLY = True
    SESSION_COOKIE_SECURE = False  # override in production
    PREFERRED_URL_SCHEME = os.getenv("PREFERRED_URL_SCHEME", "http")

    # Flask
    DEBUG = False
    TESTING = False

    @classmethod
    def ensure_paths(cls) -> None:
        """Ensure directories for DB_PATH exist."""
        if cls.DB_PATH:
            Path(cls.DB_PATH).parent.mkdir(parents=True, exist_ok=True)


class DevelopmentConfig(BaseConfig):
    DEBUG = True
    SESSION_COOKIE_SECURE = False


class ProductionConfig(BaseConfig):
    DEBUG = False
    SESSION_COOKIE_SECURE = os.getenv("SESSION_COOKIE_SECURE", "true").lower() == "true"


class TestingConfig(BaseConfig):
    TESTING = True
    DEBUG = True
    WTF_CSRF_ENABLED = False
    SESSION_COOKIE_SECURE = False
    DB_PATH = os.getenv("TEST_DB_PATH", "alles_neu/app/database/test.db")


CONFIG_MAP: dict[str, type[BaseConfig]] = {
    "development": DevelopmentConfig,
    "dev": DevelopmentConfig,
    "production": ProductionConfig,
    "prod": ProductionConfig,
    "testing": TestingConfig,
    "test": TestingConfig,
}


def get_config(env_name: str | None = None) -> type[BaseConfig]:
    """
    Returns a config class based on the environment name.

    Priority:
    1) Explicit env_name argument
    2) ENV or FLASK_ENV environment variable
    3) Default to ProductionConfig
    """
    env = (env_name or os.getenv("ENV") or os.getenv("FLASK_ENV") or "").lower()
    config_cls = CONFIG_MAP.get(env, ProductionConfig)
    config_cls.ensure_paths()
    return config_cls