from . import logging
from oslo_config import cfg
import keystoneauth1.identity.v3
import keystoneauth1.session
import requests
import yaml

LOG = logging.getLogger(__name__)

__all__ = ['get_documents']

_DECKHAND_PREFIX = 'deckhand+'

DH_TIMEOUT = 10 * 60  # 10 Minute timeout for fetching from Deckhand.


def get_documents(design_ref):
    LOG.debug('Fetching design_ref="%s"', design_ref)
    if design_ref.startswith(_DECKHAND_PREFIX):
        response = _get_from_deckhand(design_ref)
    else:
        response = _get_from_basic_web(design_ref)
    LOG.debug('Got response for design_ref="%s"', design_ref)

    response.raise_for_status()

    return list(yaml.safe_load_all(response.text))


def _get_from_basic_web(design_ref):
    return requests.get(design_ref)


def _get_from_deckhand(design_ref):
    keystone_args = {}
    for attr in ('auth_url', 'password', 'project_domain_name', 'project_name',
                 'username', 'user_domain_name'):
        keystone_args[attr] = cfg.CONF.get('keystone_authtoken', {}).get(attr)
    auth = keystoneauth1.identity.v3.Password(**keystone_args)
    session = keystoneauth1.session.Session(auth=auth)

    return session.get(design_ref[len(_DECKHAND_PREFIX):], timeout=DH_TIMEOUT)