"""Config for the datadog_caching_monitor SWF workflow."""

import base64
import os

ENVIRONMENT = os.environ.get('Environment', 'dev')
FEED_NAME = 'datadog_caching_monitor'

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

DATADOG_API_KEY = os.environ.get('DATADOG_API_KEY')

SF_CONFIG = {
    'role': os.environ.get('SNOWFLAKE_ROLE'),
    'warehouse': os.environ.get('SNOWFLAKE_WAREHOUSE'),
    'db': os.environ.get('SNOWFLAKE_DATABASE'),
    'schema': os.environ.get('SNOWFLAKE_SCHEMA'),
    'user': os.environ.get('SNOWFLAKE_USER'),
    'account': os.environ.get('SNOWFLAKE_ACCOUNT'),
    'password': '',  # we're using a keypair
    'private_key': base64.b64decode(
        bytes(os.environ.get(
            'SNOWFLAKE_PRIVATE_KEY_STRING', ''), encoding='utf-8'))
}


def get_snowflake_users_to_monitor():
    """Get Snowflake users to monitor."""
    return ['PROD_OWS_ANALYTICS', 'PROD_BFF_MOBILE', 'PROD_VAPI']


def get_snowflake_warehouses_to_monitor():
    """Get Snowflake warehouses to monitor."""
    return [
        warehouse.strip() for warehouse in os.environ.get(
            'SNOWFLAKE_WAREHOUSES_TO_MONITOR',
            'PROD_OWS_WAREHOUSE').split(',')
    ]