"""SQL query module."""

from enum import StrEnum
from functools import cache
from pathlib import Path

SQL_DIR = Path(__file__).parent


class SnowflakeQuery(StrEnum):
    """Snowflake queries."""

    # Account UPCs
    BatchAccountUpcs = 'art_relations/batch_account_upcs'
    BulkAccountUpcs = 'art_relations/bulk_account_upcs'
    CountAccountUpcs = 'art_relations/count_account_upcs'

    # Other
    GetAccountContracts = 'royalty_accounting/get_account_contracts'
    GetAccountPaymentTerms = 'royalty_accounting/get_account_payment_terms'
    GetAccounts = 'royalty_accounting/get_accounts'
    GetCloseBalanceStatuses = 'royalty_accounting/get_close_balance_statuses'
    GetFlatContractTerms = 'royalty_accounting/get_flat_contract_terms'
    GetCurrencyCodes = 'royalty_accounting/get_currency_codes'
    GetReferenceAdjustmentTypes = 'royalty_accounting/get_reference_adjustment_types'
    GetStatementPeriods = 'royalty_accounting/get_statement_periods'


@cache
def load_sql(query_name: str) -> str:
    """Load SQL query from file.

    Args:
        query_name: Name of the SQL file (e.g., 'load_account_contracts')

    Returns:
        SQL query string
    """
    sql_path = SQL_DIR / f'{query_name}.sql'
    return sql_path.read_text()