import os

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


SNOWFLAKE_PRIVATE_KEY_PATH = os.path.join(
    os.environ["HOME"], ".ssh", "snowflake", "rsa_key.p8"
)

with open(SNOWFLAKE_PRIVATE_KEY_PATH, "rb") as key:
    p_key = serialization.load_pem_private_key(
        key.read(),
        password=os.environ["SNOWFLAKE_KEY_PASSPHRASE"].encode(),
        backend=default_backend(),
    )

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

SNOWFLAKE_DB_CONFIG = {
    "role": os.environ["SNOWFLAKE_ROLE"],
    "warehouse": os.environ["SNOWFLAKE_WAREHOUSE"],
    "db": os.environ["SNOWFLAKE_DATABASE"],
    "database": os.environ["SNOWFLAKE_DATABASE"],
    "schema": os.environ["SNOWFLAKE_SCHEMA"],
    "user": os.environ["SNOWFLAKE_USER"],
    "password": os.environ["SNOWFLAKE_PASSWORD"],
    "account": os.environ["SNOWFLAKE_ACCOUNT"],
    "private_key": pkb,
}

RDS_DB_CONFIG = {
    "user": os.environ["COLLABORATOR_DB_USER"],
    "password": os.environ["COLLABORATOR_DB_PASSWORD"],
    "host": os.environ["COLLABORATOR_DB_HOST"],
    "database": os.environ["COLLABORATOR_DB_DATABASE"],
}