from utils.types import Path, FileName, DataSource, DataCollection
from utils.helpers import flatten

def transform_to_data_collection(path: str, file_names: list[str]) -> DataCollection:
    return list(zip([path] * len(file_names), file_names))

__daily_etf = ["GLD", "IEF", "QQQ", "SPY", "TLT"]

__daily_crypto = ["ADA_USD", "BCH_USD", "BNB_USD", "BTC_USD", "DOT_USD", "ETC_USD", "ETH_USD", "FIL_USD", "LTC_USD", "SOL_USD", "THETA_USD", "TRX_USD", "UNI_USD", "XLM_USD", "XRP_USD", "XTZ_USD"]

__daily_crypto_lightweight = ["ADA_USD", "BCH_USD"]

__hourly_crypto = ["BTC_USD", "DASH_USD", "ETC_USD", "ETH_USD", "LTC_USD", "TRX_USD", "XLM_USD", "XMR_USD", "XRP_USD"]

__daily_glassnode = ['rhodl_ratio',
    # 'cvdd',
    'nvt_ratio',
    'nvt_signal',
    'velocity',
    'supply_adjusted_cdd',
    'binary_cdd',
    'supply_adjusted_dormancy',
    'puell_multiple',
    'asopr',
    'reserve_risk',
    'sopr',
    'cdd',
    'asol',
    'msol',
    'dormancy',
    'liveliness',
    'relative_unrealized_profit',
    'relative_unrealized_loss',
    'nupl',
    'sth_nupl',
    'lth_nupl',
    'ssr',
    'bvin',
    # 'hash_rate'
]


data_collections = dict(
    daily_only_btc = transform_to_data_collection("data/daily_crypto", ['BTC_USD']),
    daily_crypto = transform_to_data_collection("data/daily_crypto", __daily_crypto),
    daily_crypto_lightweight = transform_to_data_collection("data/daily_crypto", __daily_crypto_lightweight),
    daily_etf = transform_to_data_collection("data/daily_etf", __daily_etf),
    hourly_crypto = transform_to_data_collection("data/hourly_crypto", __hourly_crypto),
    daily_glassnode =transform_to_data_collection("data/daily_glassnode",  __daily_glassnode),
)