import os

from src.constants import ENV_PROD, ENVS, SERVICE

ENVIRONMENT = os.environ.get("ENVIRONMENT", ENV_PROD).lower()
SERVICE = os.environ.get("SERVICE", SERVICE).lower()

DB_NAME = os.environ.get("POSTGRES_DB_NAME")
DB_HOST = os.environ.get("POSTGRES_DB_HOST", "localhost")
DB_PORT = os.environ.get("POSTGRES_DB_PORT", 3306)
DB_USER = os.environ.get("POSTGRES_DB_USER", "root")
DB_PASS = os.environ.get("POSTGRES_DB_PASS", "root")

SQLALCHEMY_DATABASE_URI = f"postgresql+psycopg2://{DB_USER}:{DB_PASS}@{DB_HOST}:{DB_PORT}/{DB_NAME}"

SQLALCHEMY_TRACK_MODIFICATIONS = True

ATLAS_LOGIN_URL = os.environ.get("ATLAS_LOGIN_URL")
ATLAS_LOGOUT_URL = os.environ.get("ATLAS_LOGOUT_URL")
ATLAS_REFRESH_TOKEN_URL = os.environ.get("ATLAS_REFRESH_TOKEN_URL")
ATLAS_PUBLIC_KEY_URL = os.environ.get("ATLAS_PUBLIC_KEY_URL")
REFRESH_TOKEN_COOKIE_NAME = os.environ.get("REFRESH_TOKEN_COOKIE_NAME")
TOKEN_COOKIES_MAX_AGE = os.environ.get("TOKEN_COOKIES_MAX_AGE")
ATLAS_ENCODING_ALGORITHM = os.environ.get(
    "ATLAS_ENCODING_ALGORITHM", "RS256"
)
ALLOWED_HOSTS = os.environ.get("ALLOWED_HOSTS")

ATLAS_RELATED_CLAIMS_NAMESPACE = os.environ.get(
    "ATLAS_RELATED_CLAIMS_NAMESPACE", "dna"
)
ATLAS_BEARER_TOKEN_COOKIE_NAME = os.environ.get(
    "ATLAS_BEARER_TOKEN_COOKIE_NAME", "dna_bearer_token"
)
ATLAS_REFRESH_TOKEN_COOKIE_NAME = os.environ.get(
    "ATLAS_REFRESH_TOKEN_COOKIE_NAME", "dna_refresh_token"
)

# TODO: if in DEVELOPMENT_ENVS
if any(env in ENVIRONMENT for env in ENVS):
    # ATLAS config
    ATLAS_BASE_URL = os.environ.get("ATLAS_BASE_URL_PROD")

    # CMS config
    CMS_API_PROD_CONFIG: dict = {
        "client_id": os.environ.get("CMS_CLIENT_ID_PROD"),
        "client_secret": os.environ.get("CMS_CLIENT_SECRET_PROD"),
        "base_url": ATLAS_BASE_URL,
        "audience": os.environ.get("CMS_AUDIENCE_PROD"),
        "cms_service_url": os.environ.get("CMS_BASE_URL_PROD"),
    }

DEFAULT_RETRY_COUNT = int(os.environ.get("DEFAULT_RETRY_COUNT", 2))
DEFAULT_RETRY_WAIT = int(os.environ.get("DEFAULT_RETRY_WAIT", 1))
DEFAULT_RETRY_MAX_TIMEOUT = int(os.environ.get("DEFAULT_RETRY_MAX_TIMEOUT", 6))