import os
import sys
from .utils import *
from .endpoints import Endpoints


class GlassnodeClient:
    def __init__(
            self,
            api_key=None,
            asset='BTC',
            resolution='24h',
            currency='native',
            since=None,
            until=None
    ):
        """
        Glassnode API client.

        :param asset: Asset to which the metric refers. (ex. BTC)
        :param resolution: Temporal resolution of the data received. Can be '10m', '1h', '24h', '1w' or '1month'.
        :param currency: NATIVE, USD
        :param since: Start date as a string (ex. 2015-11-27)
        :param until: Start date as a string (ex. 2018-05-03)
        """
        if api_key:
            self._api_key = api_key
        elif 'GLASSNODE_API_KEY' in os.environ:
            self._api_key = os.environ.get('GLASSNODE_API_KEY')
        else:
            # API key is required for every endpoint!
            print(f'\033[91m ERROR: Glassnode API key required!\033[0m')
            sys.exit()

        self.endpoints = Endpoints()
        self.endpoints.endpoints = self._api_key
        self._asset = asset
        self._resolution = resolution
        self._since = since
        self._until = until
        self._currency = currency

    @property
    def asset(self):
        return self._asset

    @property
    def resolution(self):
        return self._resolution

    def get(self, endpoint, params=None):
        return fetch(endpoint, self.__prepare_request_params(params))

    def __prepare_request_params(self, params):
        p = dict()
        p['api_key'] = self._api_key
        p['a'] = self._asset
        p['i'] = self._resolution

        if self._since is not None:
            try:
                p['s'] = unix_timestamp(self._since)
            except Exception as e:
                print(e)

        if self._until is not None:
            try:
                p['u'] = unix_timestamp(self._until)
            except Exception as e:
                print(e)

        # Set domain specific query parameters if available
        if params:
            if 'e' in p:
                p['e'] = params['e']
            if 'm' in p:
                p['miner'] = params['m']

        return p