# from distutils.util import strtobool
import os

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization
from dotenv import load_dotenv

load_dotenv('.env')

# Envirnoment
PROD_ENVIRONMENT = 'prod'
DEV_ENVIRONMENT = 'dev'
QA_ENVIRONMENT = 'qa'
TEST_ENVIRONMENT = 'test'

ENVIRONMENT = os.environ.get('ENVIRONMENT', 'DEV')

# Application Vars #
APPLICATION_NAME = 'oa-contract-ingest-to-abacus'
S3_BUCKET = os.environ.get('S3_BUCKET', 'qa-abacus-json-contract-file-import')
BATCH_SIZE = int(os.environ.get('BATCH_SIZE', 500))
LAMBDA_FUNCTION_NAME = os.environ.get('LAMBDA_FUNCTION_NAME', 'qa-lambda-abacus-json-contract-file-import')
LAMBDA_MAX_PARALLEL = int(os.environ.get('LAMBDA_MAX_PARALLEL', 4))
IMPORT_USER = os.environ.get('IMPORT_USER')
JIRA_TICKET = os.environ.get('JIRA_TICKET')

# Log to Screen level
CONSOLE_LOG_LEVEL = str(os.environ.get('CONSOLE_LOG_LEVEL', 'INFO'))

# Application Version
APP_VERSION = os.environ.get('APP_VERSION', '1.0')

# Snowflake ------------------------------------------------------------------
SNOWFLAKE_ACCOUNT = os.environ.get('SNOWFLAKE_ACCOUNT', 'orchard')
SNOWFLAKE_USER = os.environ.get('SNOWFLAKE_USER', '')
SNOWFLAKE_ROLE = os.environ.get('SNOWFLAKE_ROLE', 'DEV_ENGINEERING')
SNOWFLAKE_PASSWORD = os.environ.get('SNOWFLAKE_PASSWORD', '')
SNOWFLAKE_AUTHENTICATOR = os.environ.get('SNOWFLAKE_AUTHENTICATOR', '')
SNOWFLAKE_WAREHOUSE = os.environ.get(
    'SNOWFLAKE_WAREHOUSE', 'DEV_OWS_WAREHOUSE')
SNOWFLAKE_DATABASE = os.environ.get('SNOWFLAKE_DATABASE', '')
SNOWFLAKE_SCHEMA = os.environ.get('SNOWFLAKE_SCHEMA', '')

SNOWFLAKE_KEY_PASSPHRASE = os.environ.get('SNOWFLAKE_KEY_PASSPHRASE', '')
SNOWFLAKE_PRIVATE_KEY_PATH = os.environ.get('SNOWFLAKE_PRIVATE_KEY_PATH', '')

# ----------- Snowflake PEM-key RSA auth -------------------------------------
try:
    HOME = os.environ['HOME']
except KeyError:
    HOME = str(
        os.environ['HOMEDRIVE'] + os.environ['HOMEPATH']
    ).replace('\\', '/')

# Snowflake key
if not SNOWFLAKE_PRIVATE_KEY_PATH:
    SNOWFLAKE_PRIVATE_KEY_PATH = '{}/.ssh/snowflake/rsa_key.p8'.format(
        HOME)

with open(SNOWFLAKE_PRIVATE_KEY_PATH, 'rb') as key:
    p_key = serialization.load_pem_private_key(
        key.read(),
        password=SNOWFLAKE_KEY_PASSPHRASE.encode(),
        backend=default_backend())

SNOWFLAKE_PRIVATE_KEY = p_key.private_bytes(
    encoding=serialization.Encoding.DER,
    format=serialization.PrivateFormat.PKCS8,
    encryption_algorithm=serialization.NoEncryption())

# AWS
AWS_ACCESS_KEY_ID = os.environ.get('AWS_ACCESS_KEY_ID')
AWS_SECRET_ACCESS_KEY = os.environ.get('AWS_SECRET_ACCESS_KEY')