"""Config file for lambda function."""

import os

from secrets_manager.lambda_ext import LambdaSecretsManager

APPLICATION_NAME = "lambda-assets-hive-ai-detection"

# AWS region configuration
DEFAULT_AWS_REGION = "us-east-1"
AWS_REGION = os.environ.get("AWS_REGION", DEFAULT_AWS_REGION)

# Environment setup
TEST_ENVIRONMENT = "test"
DEV_ENVIRONMENT = "dev"
QA_ENVIRONMENT = "qa"
PROD_ENVIRONMENT = "prod"
ENVIRONMENT = os.environ["Environment"]

SENTRY_DSN = os.environ.get("SENTRY_DSN")

secrets_manager_client = LambdaSecretsManager(
    environment=ENVIRONMENT, service_name=APPLICATION_NAME
)

HIVE_CREDENTIALS = secrets_manager_client.get_cred("hive_credentials")
HIVE_RETRIES = int(os.environ.get("HIVE_RETRIES", 3))
HIVE_TIMEOUT = int(os.environ.get("HIVE_TIMEOUT", 600))
assert HIVE_CREDENTIALS is not None, "Hive credentials not found."

OWS_RETRIES = int(os.environ.get("OWS_RETRIES", 3))
OWS_TIMEOUT = int(os.environ.get("OWS_TIMEOUT", 30))

MAX_ASSET_DURATION_SECONDS = 600

# M2MTokenManager only works for QA and PROD environments.
TOKEN_MANAGER_ENVIRONMENT = (
    ENVIRONMENT
    if ENVIRONMENT not in [DEV_ENVIRONMENT, TEST_ENVIRONMENT]
    else QA_ENVIRONMENT
)