import os

import boto3

SERVICE_NAME = "lambda-collaborator-auto-report-run"

# Environment
PROD_ENVIRONMENT = "prod"
DEV_ENVIRONMENT = "dev"
QA_ENVIRONMENT = "qa"
TEST_ENVIRONMENT = "test"
ENVIRONMENT = os.environ.get("ENVIRONMENT", DEV_ENVIRONMENT)


def _get_secret(secret_name: str) -> str | None:
    if ENVIRONMENT not in [QA_ENVIRONMENT, PROD_ENVIRONMENT]:
        return None
    secret_id = f"{ENVIRONMENT}/{SERVICE_NAME}/{secret_name}"
    return boto3.client("secretsmanager").get_secret_value(SecretId=secret_id)["SecretString"]


DB_COLLABORATORS_HOST = os.environ.get("DB_COLLABORATORS_HOST")
DB_COLLABORATORS_DATABASE = "ows_collaborator"
DB_COLLABORATORS_USERNAME = os.environ.get("DB_COLLABORATORS_USERNAME")
DB_COLLABORATORS_PASSWORD = os.environ.get(
    "DB_COLLABORATORS_PASSWORD", _get_secret("DB_COLLABORATORS_PASSWORD")
)

DB_ROYALTY_ACCOUNTING_HOST = os.environ.get("DB_ROYALTY_ACCOUNTING_HOST")
DB_ROYALTY_ACCOUNTING_DATABASE = "royalty_accounting"
DB_ROYALTY_ACCOUNTING_USERNAME = os.environ.get("DB_ROYALTY_ACCOUNTING_USERNAME")
DB_ROYALTY_ACCOUNTING_PASSWORD = os.environ.get(
    "DB_ROYALTY_ACCOUNTING_PASSWORD", _get_secret("DB_ROYALTY_ACCOUNTING_PASSWORD")
)

DB_ART_RELATIONS_HOST = os.environ.get("DB_ART_RELATIONS_HOST")
DB_ART_RELATIONS_DATABASE = "art_relations"
DB_ART_RELATIONS_USERNAME = os.environ.get("DB_ART_RELATIONS_USERNAME")
DB_ART_RELATIONS_PASSWORD = os.environ.get(
    "DB_ART_RELATIONS_PASSWORD", _get_secret("DB_ART_RELATIONS_PASSWORD")
)

SQS_QUEUE_URL = os.environ.get("SQS_QUEUE_URL", "")