from contextlib import contextmanager
from io import BytesIO
from pathlib import Path

from lxml.etree import iterparse


__HERE__ = Path(__file__).parent.absolute()


def _parse_xml(data, strip_ns=False):
    data = bytes(data, "utf8")
    try:
        it = iterparse(BytesIO(data))
        for _, el in it:
            if "}" in el.tag and strip_ns:  # pragma: no branch
                # strip all namespaces
                el.tag = el.tag.split("}", 1)[1]
        return it.root
    except Exception as err:  # pragma: no cover
        snippet = repr(data)
        if len(snippet) > 35:
            snippet = f"{snippet[:35]} ..."

        raise ValueError(f"Unable to parse XML: {err} ({snippet})") from err


@contextmanager
def text(path, encoding="utf8"):
    with __HERE__.joinpath(path).open("r", encoding=encoding) as resource_fh:
        yield resource_fh


@contextmanager
def xml(path, encoding="utf8"):
    with __HERE__.joinpath(path).open("r", encoding=encoding) as resource_fh:
        yield _parse_xml(resource_fh.read(), strip_ns=True)