learnhouse/config/config.py

from typing import Optional
from pydantic import BaseModel
import os
import yaml


class SentryConfig(BaseModel):
    dsn: str
    environment: str
    release: str


class CookieConfig(BaseModel):
    domain: str


class HostingConfig(BaseModel):
    domain: str
    ssl: bool
    use_default_org: bool
    allowed_origins: list
    allowed_regexp: str
    self_hosted: bool
    sentry_config: Optional[SentryConfig]
    cookie_config: CookieConfig


class DatabaseConfig(BaseModel):
    mongodb_connection_string: Optional[str]


class LearnHouseConfig(BaseModel):
    site_name: str
    site_description: str
    contact_email: str
    hosting_config: HostingConfig
    database_config: DatabaseConfig


def get_learnhouse_config() -> LearnHouseConfig:
    # Get the YAML file
    yaml_path = os.path.join(os.path.dirname(__file__), "config.yaml")

    # Load the YAML file
    with open(yaml_path, "r") as f:
        yaml_config = yaml.safe_load(f)
  
    # Check if environment variables are defined
    env_site_name = os.environ.get("LEARNHOUSE_SITE_NAME")
    env_site_description = os.environ.get("LEARNHOUSE_SITE_DESCRIPTION")
    env_contact_email = os.environ.get("LEARNHOUSE_CONTACT_EMAIL")
    env_domain = os.environ.get("LEARNHOUSE_DOMAIN")
    os.environ.get("LEARNHOUSE_PORT")
    env_ssl = os.environ.get("LEARNHOUSE_SSL")
    env_use_default_org = os.environ.get("LEARNHOUSE_USE_DEFAULT_ORG")
    env_allowed_origins = os.environ.get("LEARNHOUSE_ALLOWED_ORIGINS")
    env_cookie_domain = os.environ.get("LEARNHOUSE_COOKIE_DOMAIN")
    # Allowed origins should be a comma separated string
    if env_allowed_origins:
        env_allowed_origins = env_allowed_origins.split(",")
    env_allowed_regexp = os.environ.get("LEARNHOUSE_ALLOWED_REGEXP")
    env_self_hosted = os.environ.get("LEARNHOUSE_SELF_HOSTED")
    env_mongodb_connection_string = os.environ.get(
        "LEARNHOUSE_MONGODB_CONNECTION_STRING"
    )

    # Sentry Config
    env_sentry_dsn = os.environ.get("LEARNHOUSE_SENTRY_DSN")
    env_sentry_environment = os.environ.get("LEARNHOUSE_SENTRY_ENVIRONMENT")
    env_sentry_release = os.environ.get("LEARNHOUSE_SENTRY_RELEASE")

    # Fill in values with YAML file if they are not provided
    site_name = env_site_name or yaml_config.get("site_name")
    site_description = env_site_description or yaml_config.get("site_description")
    contact_email = env_contact_email or yaml_config.get("contact_email")

    domain = env_domain or yaml_config.get("hosting_config", {}).get("domain")
    ssl = env_ssl or yaml_config.get("hosting_config", {}).get("ssl")
    use_default_org = env_use_default_org or yaml_config.get("hosting_config", {}).get(
        "use_default_org"
    )
    allowed_origins = env_allowed_origins or yaml_config.get("hosting_config", {}).get(
        "allowed_origins"
    )
    allowed_regexp = env_allowed_regexp or yaml_config.get("hosting_config", {}).get(
        "allowed_regexp"
    )
    self_hosted = env_self_hosted or yaml_config.get("hosting_config", {}).get(
        "self_hosted"
    )

    cookies_domain = env_cookie_domain or yaml_config.get("hosting_config", {}).get("cookies_config", {}).get("domain")
    cookie_config = CookieConfig(
        domain=cookies_domain
    )

    # Database config
    mongodb_connection_string = env_mongodb_connection_string or yaml_config.get(
        "database_config", {}
    ).get("mongodb_connection_string")

    # Sentry config
    # check if the sentry config is provided in the YAML file
    sentry_config_verif = (
        yaml_config.get("hosting_config", {}).get("sentry_config")
        or env_sentry_dsn
        or env_sentry_environment
        or env_sentry_release
        or None
    )

    sentry_dsn = env_sentry_dsn or yaml_config.get("hosting_config", {}).get(
        "sentry_config", {}
    ).get("dsn")
    sentry_environment = env_sentry_environment or yaml_config.get(
        "hosting_config", {}
    ).get("sentry_config", {}).get("environment")
    sentry_release = env_sentry_release or yaml_config.get("hosting_config", {}).get(
        "sentry_config", {}
    ).get("release")

    if sentry_config_verif:
        sentry_config = SentryConfig(
            dsn=sentry_dsn, environment=sentry_environment, release=sentry_release
        )
    else:
        sentry_config = None

    # Create HostingConfig and DatabaseConfig objects
    hosting_config = HostingConfig(
        domain=domain,
        ssl=bool(ssl),
        use_default_org=bool(use_default_org),
        allowed_origins=list(allowed_origins),
        allowed_regexp=allowed_regexp,
        self_hosted=bool(self_hosted),
        sentry_config=sentry_config,
        cookie_config=cookie_config,
    )
    database_config = DatabaseConfig(
        mongodb_connection_string=mongodb_connection_string
    )

    # Create LearnHouseConfig object
    config = LearnHouseConfig(
        site_name=site_name,
        site_description=site_description,
        contact_email=contact_email,
        hosting_config=hosting_config,
        database_config=database_config,
    )

    return config