import os

from flask_dance.consumer import OAuth2ConsumerBlueprint


CREDENTIALS_DIR = os.getenv('CREDENTIALS_DIR', '')


class OAuth2ConsumerBlueprintWithClientRefresh(OAuth2ConsumerBlueprint):
    '''Same as flask_dance.consumer.OAuth2ConsumerBlueprint, but loads client credentials from file'''

    def refresh_credentials(self):
        with open(os.path.join(CREDENTIALS_DIR, 'authcode-client-id')) as fd:
            # note that we need to set two attributes because of how OAuth2ConsumerBlueprint works :-/
            self._client_id = self.client_id = fd.read().strip()
        with open(os.path.join(CREDENTIALS_DIR, 'authcode-client-secret')) as fd:
            self.client_secret = fd.read().strip()

    def login(self):
        self.refresh_credentials()
        return super().login()