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

import os

from secrets_manager.lambda_ext import LambdaSecretsManager

APPLICATION_NAME = 'lambda-update-subaccount-mapping'

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

# Environment setup
QA_ENVIRONMENT = 'qa'
PROD_ENVIRONMENT = 'prod'
TEST_ENVIRONMENT = 'test'
ENVIRONMENT = os.environ.get('Environment', QA_ENVIRONMENT)

BUCKET = f'{ENVIRONMENT}-subaccount-mapping'

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

if ENVIRONMENT == TEST_ENVIRONMENT:
    NEO4J_URL = 'NEO4J_URL'
    NEO4J_USERNAME = 'NEO4J_USERNAME'
    NEO4J_PASSWORD = 'NEO4J_PASSWORD'
    NEO4J_MAX_RETRY_TIME = 2

    DATABASE_HOSTNAME = 'sqlite://'
    DATABASE_USER_NAME = 'test'
    DATABASE_PASSWORD = 'test'
    DATABASE = 'test'
else:
    # art_relations mysql configs
    DEFAULT_DATABASE = 'art_relations'
    DATABASE_HOSTNAME = os.environ.get('DATABASE_HOSTNAME')
    DATABASE_USER_NAME = os.environ.get('DATABASE_USER_NAME')
    DATABASE_PASSWORD = secrets_manager_client.get_cred('DATABASE_PASSWORD')
    DATABASE = os.environ.get('DATABASE', DEFAULT_DATABASE)

    # neo4j configs
    NEO4J_MAX_RETRY_TIME = os.environ.get('NEO4J_MAX_RETRY_TIME', 30)  # 30s
    NEO4J_URL = os.environ.get('NEO4J_URL')
    NEO4J_USERNAME = os.environ.get('NEO4J_USERNAME')
    NEO4J_PASSWORD = secrets_manager_client.get_cred('NEO4J_PASSWORD')  # noqa