from typing import Union, List

import requests
from urllib.parse import urljoin

from .simplenode import SimpleNode


# TODO validation

"""
http://localhost:4502/crx/de/init.jsp?_dc=1549392939742
http://localhost:4502/crx/de/nodetypes.jsp?_dc=1549392939958
http://localhost:4502/crx/server/crx.default/jcr%3aroot/libs.1.json?_dc=1549392123434&node=xnode-265
http://localhost:4502/crx/de/query.jsp?_dc=1549392245191&_charset_=utf-8&type=xpath&stmt=%2Fjcr%3Aroot%2Fbin%2F%2F*%5Bjcr%3Acontains(.%2C%20%27asdf%27)%5D%20order%20by%20%40jcr%3Ascore&showResults=true
http://app30-prd-asd.sbnl.vancis.nl:4502/bin/wcm/references.json?path=%2Fcontent%2Fdam%2Fbeeldbank%2F_0005_home_algemeen.png&predicate=wcmcontent&_charset_=utf-8

Download:
http://app30-prd-asd.sbnl.vancis.nl:4502/crx/server/crx.default/jcr:root/content/dam/beeldbank/vrouw-direct-naar.jpg/jcr:content/renditions/original/jcr:content/jcr:data
OR
download.jsp?path=%2Fcontent%2Fdam%2Fbeeldbank%2Fvrouw-direct-naar.jpg%2Fjcr%3Acontent%2Frenditions%2Foriginal%2Fjcr%3Acontent%2Fjcr%3Adata&index=0
"""

CRX_SERVER_ROOT = '/crx/server/crx.default/jcr:root/'
CRX_QUERY = '/crx/de/query.jsp'
WCM_REFERENCES = '/bin/wcm/references.json'

JSON_DATA_EXTENSION = '.1.json'

QUERY_TYPES = {
    'XPATH': 'xpath',
    'SQL': 'sql',
    'SQL2': 'JCR-SQL2'
}


class CrxException(ValueError):
    pass


class Connection:
    def __init__(self,
                 host: str = 'localhost',
                 port: int = 4502,
                 protocol: str = 'http',
                 root: str = CRX_SERVER_ROOT,
                 query: str = CRX_QUERY,
                 image_references: str = WCM_REFERENCES):

        self.host = f'{protocol}://{host}:{port}'
        self.data_root = self.host + root
        self.query_path = self.host + query
        self.image_references = self.host + image_references

        self.session = requests.session()

    def login_basic(self, username: str, password: str):
        self.session.auth = (username, password)

    def query(self, query: str, query_type: str = 'SQL2') -> List[str]:
        response = self.session.get(self.query_path, params={
            '_charset': 'utf-8',
            'type': QUERY_TYPES.get(query_type, query_type),
            'stmt': query,
            'showResults': 'true'
        })
        data = response.json()

        # TODO check for error

        return list(map(lambda node: node['path'], data['results']))

    def get_image_references(self, path: str):
        response = self.session.get(self.image_references, params={
            'path': path
        })
        return response.json()['pages']

    def get_node_raw(self, path: str):
        url = urljoin(self.data_root, '.' + path + JSON_DATA_EXTENSION)
        try:
            response = self.session.get(url)
        except requests.exceptions.RequestException as exception:
            raise CrxException()  # todo more specific exceptions

        try:
            data = response.json()
        except ValueError:
            raise  # todo

        return data

    def download_binary(self, path: str) -> bytes:
        # TODO verify if it is not b64 encoded. for some reason it is in FireFox
        resp = self.session.get(
            urljoin(self.data_root, '.' + path)
        )
        return resp.content

    def get_simple_node(self, path: str) -> SimpleNode:
        return SimpleNode(path, self.get_node_raw(path), self)

    def rename_node(self, old_path: str, new_path: str):
        diff = f'>{old_path} : {new_path}'
        resp = self.session.post(self.data_root, data={':diff': diff})
        resp.raise_for_status()

    def apply_diff(self, diff: Union[str, bytes]):
        files = {
            ':diff': (
                None,
                diff,
                'text/plain; charset=utf-8'
            )
        }
        # todo check for exception
        resp = self.session.post(self.data_root, files=files)
        resp.raise_for_status()