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

import os

from dotenv import load_dotenv
from secrets_manager.lambda_ext import LambdaSecretsManager

load_dotenv()

APPLICATION_NAME = 'lambda-abacus-schedule-auto-add'

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

# Environment setup
DEV_ENVIRONMENT = 'dev'
QA_ENVIRONMENT = 'qa'
UAT_ENVIRONMENT = 'uat'
PROD_ENVIRONMENT = 'prod'
ENVIRONMENT = os.environ.get('ENVIRONMENT', QA_ENVIRONMENT)

GRAPHQL_GATEWAY_URL = os.environ.get(
    'GRAPHQL_GATEWAY_URL', 'https://qa-graphql-gateway.theorchard.io/graphql')

GRAPHQL_HEADERS = {
    'Orchard-Profile-Id': os.environ.get('ORCHARD_PROFILE_ID'),
    'Orchard-Profile-UUID': os.environ.get('ORCHARD_PROFILE_UUID'),
    'Orchard-Identity-Id': os.environ.get('ORCHARD_IDENTITY_ID'),
    'Orchard-Roles': 'administrator',
    'Orchard-Profile-Type': 'AbacusProfile'
}

# # Pull QA & Prod credentials from secrets manager
if ENVIRONMENT in [QA_ENVIRONMENT, PROD_ENVIRONMENT, UAT_ENVIRONMENT]:
    client = LambdaSecretsManager(
        environment=ENVIRONMENT, service_name=APPLICATION_NAME)
    SENTRY_DSN = client.get_cred('SENTRY_DSN')

    GRAPHQL_HEADERS['Orchard-Profile-Id'] = client.get_cred('ORCHARD_PROFILE_ID')
    GRAPHQL_HEADERS['Orchard-Profile-UUID'] = client.get_cred('ORCHARD_PROFILE_UUID')
    GRAPHQL_HEADERS['Orchard-Identity-Id'] = client.get_cred('ORCHARD_IDENTITY_ID')