import os

from secrets_manager.python_ext import (
    PythonSecretsManager,
)

from src.utils.private_key import get_private_key, get_private_key_from_file

DEV_ENVIRONMENT = "dev"
QA_ENVIRONMENT = "qa"
PROD_ENVIRONMENT = "prod"
ENVIRONMENT = os.environ.get("ENVIRONMENT", QA_ENVIRONMENT)
SCRIPT_NAME = "invoke-hive-ai-detection"

secrets_manager_client = PythonSecretsManager(
    environment=ENVIRONMENT, service_name=SCRIPT_NAME
)


# Snowflake creds
def load_private_key() -> bytes | None:
    """Load the Snowflake private key based on the environment."""
    private_key_passphrase = secrets_manager_client.get_cred("SNOWFLAKE_KEY_PASSPHRASE")
    private_key_location = os.environ.get("SNOWFLAKE_PRIVATE_KEY_LOCATION")
    if private_key_location:
        return get_private_key_from_file(private_key_location, private_key_passphrase)

    private_key = secrets_manager_client.get_cred("SNOWFLAKE_PRIVATE_KEY")
    if private_key:
        return get_private_key(private_key, private_key_passphrase)
    return None


SNOWFLAKE_USER = os.environ.get("SNOWFLAKE_USER")
SNOWFLAKE_ACCOUNT = os.environ.get("SNOWFLAKE_ACCOUNT")
SNOWFLAKE_ROLE = os.environ.get("SNOWFLAKE_ROLE")
SNOWFLAKE_WAREHOUSE = os.environ.get("SNOWFLAKE_WAREHOUSE")
SNOWFLAKE_DATABASE = os.environ.get("SNOWFLAKE_DATABASE")
SNOWFLAKE_SCHEMA = os.environ.get("SNOWFLAKE_SCHEMA")
SNOWFLAKE_PRIVATE_KEY = load_private_key()