from __future__ import annotations

import json
import logging
import os
import re
from dataclasses import dataclass, field
from pathlib import Path, PurePath
from test.data import TEST_DATA_DIR
from test.utils import GraphHelper
from typing import (
    ClassVar,
    Collection,
    Dict,
    Iterable,
    List,
    Optional,
    OrderedDict,
    Pattern,
    Tuple,
    Union,
    cast,
)

import pytest
from _pytest.mark.structures import Mark, MarkDecorator, ParameterSet

import rdflib.compare
import rdflib.util
from rdflib.graph import ConjunctiveGraph
from rdflib.namespace import XSD
from rdflib.term import URIRef
from rdflib.util import guess_format

TEST_DIR = Path(__file__).parent.parent.absolute()
VARIANTS_DIR = TEST_DATA_DIR / "variants"

# Put files from other directories in here.
EXTRA_FILES: List[Path] = []

SUFFIX_FORMAT_MAP = {**rdflib.util.SUFFIX_FORMAT_MAP, "hext": "hext"}


@dataclass
class GraphAsserts:
    """
    A specification of asserts that must be checked against a graph. This is
    read in from a JSON dict.
    """

    quad_count: Optional[int] = None
    exact_match: bool = False
    has_subject_iris: Optional[List[str]] = None

    def check(
        self, first_graph: Optional[ConjunctiveGraph], graph: ConjunctiveGraph
    ) -> None:
        """
        if `first_graph` is `None` then this is the first check before any
        other graphs have been processed.
        """
        if self.quad_count is not None:
            assert self.quad_count == len(list(graph.quads()))
        if first_graph is not None and self.exact_match:
            GraphHelper.assert_quad_sets_equals(first_graph, graph)
        if first_graph is None and self.has_subject_iris is not None:
            subjects_iris = {
                f"{subject}"
                for subject in graph.subjects()
                if isinstance(subject, URIRef)
            }
            assert set(self.has_subject_iris) == subjects_iris

    @classmethod
    def from_path(cls, path: Path) -> GraphAsserts:
        with path.open("r") as f:
            return cls(**json.load(f))


@dataclass(order=True)
class GraphVariants:
    """
    Represents a graph with multiple variants in different files.
    """

    key: str
    variants: Dict[str, Path] = field(default_factory=OrderedDict)
    asserts: GraphAsserts = field(default_factory=lambda: GraphAsserts())

    _variant_regex: ClassVar[Pattern[str]] = re.compile(
        r"^(.*?)(|[-]variant-[^/]+|[-]asserts)$"
    )

    def pytest_param(
        self,
        marks: Optional[
            Union[MarkDecorator, Collection[Union[MarkDecorator, Mark]]]
        ] = None,
    ) -> ParameterSet:
        if marks is None:
            marks = cast(Tuple[MarkDecorator], tuple())
        logging.debug("self = %s", self)
        return pytest.param(self, id=self.key, marks=marks)

    @classmethod
    def _decompose_path(cls, file_path: Path, basedir: Optional[Path]):
        if basedir:
            file_path = file_path.absolute().resolve().relative_to(basedir)
        name_noext, ext = os.path.splitext(file_path)
        name_noext_path = PurePath(name_noext)
        match = cls._variant_regex.match("/".join(name_noext_path.parts))
        if match is None:
            raise RuntimeError(f"{cls._variant_regex!r} did not match {name_noext}")
        file_key = match.group(1)
        variant_key = f"{match.group(2)}{ext}"
        return (file_key, variant_key)

    @classmethod
    def for_files(
        cls, file_paths: Iterable[Path], basedir: Optional[Path] = None
    ) -> Dict[str, "GraphVariants"]:
        graph_varaint_dict: Dict[str, GraphVariants] = {}
        for file_path in file_paths:
            logging.debug("file_path = %s", file_path)
            file_key, variant_key = cls._decompose_path(file_path, basedir)
            # file_key = f"{file_path.parent / stem}"
            if file_key not in graph_varaint_dict:
                graph_variant = graph_varaint_dict[file_key] = GraphVariants(file_key)
            else:
                graph_variant = graph_varaint_dict[file_key]
            if variant_key.endswith("-asserts.json"):
                graph_variant.asserts = GraphAsserts.from_path(file_path)
            else:
                graph_variant.variants[variant_key] = file_path
        return graph_varaint_dict

    @classmethod
    def for_directory(
        cls, directory: Path, basedir: Optional[Path] = None
    ) -> Dict[str, "GraphVariants"]:
        file_paths = []
        for file_path in directory.glob("**/*"):
            if not file_path.is_file():
                continue
            if file_path.name.endswith(".md"):
                continue
            file_paths.append(file_path)
        logging.debug("file_paths = %s", file_paths)
        return cls.for_files(file_paths, basedir)


GRAPH_VARIANT_DICT = {
    **GraphVariants.for_directory(VARIANTS_DIR, TEST_DATA_DIR),
    **GraphVariants.for_files(EXTRA_FILES, TEST_DIR),
}

EXPECTED_FAILURES = {
    ("variants/schema_only_base"): pytest.mark.xfail(
        reason="Some issue with handling base URI that does not end with a slash",
        raises=ValueError,
    ),
    ("variants/rdf11trig_eg2"): pytest.mark.xfail(
        reason="""
    This fails randomly, passing less than 10% of the time, and always failing
    with comparing hext against trig. Not clear why, it may be a big with hext
    parsing.

    AssertionError: checking rdf11trig_eg2.hext against rdf11trig_eg2.trig
    in both:
        (rdflib.term.BNode('cb0'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/mbox'), rdflib.term.URIRef('mailto:bob@oldcorp.example.org'))
        (rdflib.term.BNode('cb0'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/name'), rdflib.term.Literal('Bob'))
        (rdflib.term.URIRef('http://example.org/bob'), rdflib.term.URIRef('http://purl.org/dc/terms/publisher'), rdflib.term.Literal('Bob'))
        (rdflib.term.URIRef('http://example.org/alice'), rdflib.term.URIRef('http://purl.org/dc/terms/publisher'), rdflib.term.Literal('Alice'))
    only in first:
        (rdflib.term.BNode('cb0'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/knows'), rdflib.term.BNode('cbb5eb12b5dcf688537b0298cce144c6dd68cf047530d0b4a455a8f31f314244fd'))
        (rdflib.term.BNode('cbb5eb12b5dcf688537b0298cce144c6dd68cf047530d0b4a455a8f31f314244fd'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/mbox'), rdflib.term.URIRef('mailto:alice@work.example.org'))
        (rdflib.term.BNode('cbb5eb12b5dcf688537b0298cce144c6dd68cf047530d0b4a455a8f31f314244fd'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/name'), rdflib.term.Literal('Alice'))
    only in second:
        (rdflib.term.BNode('cb0'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/knows'), rdflib.term.BNode('cbcd41774964510991c01701d8430149bc373e1f23734d9c938c81a40b1429aa33'))
        (rdflib.term.BNode('cbcd41774964510991c01701d8430149bc373e1f23734d9c938c81a40b1429aa33'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/mbox'), rdflib.term.URIRef('mailto:alice@work.example.org'))
        (rdflib.term.BNode('cbcd41774964510991c01701d8430149bc373e1f23734d9c938c81a40b1429aa33'), rdflib.term.URIRef('http://xmlns.com/foaf/0.1/name'), rdflib.term.Literal('Alice'))
        """,
        raises=AssertionError,
    ),
    ("variants/diverse_quads"): pytest.mark.xfail(
        reason="""
    TriG parsing gets confused about what graph 'XSD string' appears in:
        (rdflib.term.URIRef('example:subject'),
        rdflib.term.URIRef('http://example.com/predicate'),
        rdflib.term.Literal('XSD string'),
    -   rdflib.term.URIRef('example:graph')),
    +   rdflib.term.URIRef('urn:example:graph')),
    ?                       ++++
        """,
        raises=AssertionError,
    ),
}


def tests_found() -> None:
    logging.debug("VARIANTS_DIR = %s", VARIANTS_DIR)
    logging.debug("EXTRA_FILES = %s", EXTRA_FILES)
    assert len(GRAPH_VARIANT_DICT) >= 1
    logging.debug("ALL_VARIANT_GRAPHS = %s", GRAPH_VARIANT_DICT)
    xml_literal = GRAPH_VARIANT_DICT.get("variants/xml_literal")
    assert xml_literal is not None
    assert len(xml_literal.variants) >= 5
    assert xml_literal.asserts.quad_count == 1


@pytest.mark.parametrize(
    "graph_variant",
    [
        graph_variant.pytest_param(EXPECTED_FAILURES.get(graph_variant.key))
        for graph_variant in GRAPH_VARIANT_DICT.values()
    ],
)
def test_variants(graph_variant: GraphVariants) -> None:
    """
    All variants of a graph are isomorphic with the first variant, and thus
    eachother.
    """
    logging.debug("graph_variant = %s", graph_variant)
    public_id = URIRef(f"example:{graph_variant.key}")
    assert len(graph_variant.variants) > 0
    first_graph: Optional[ConjunctiveGraph] = None
    first_path: Optional[Path] = None
    logging.debug("graph_variant.asserts = %s", graph_variant.asserts)

    for variant_key, variant_path in graph_variant.variants.items():
        logging.debug("variant_path = %s", variant_path)
        format = guess_format(variant_path.name, fmap=SUFFIX_FORMAT_MAP)
        assert format is not None, f"could not determine format for {variant_path.name}"
        graph = ConjunctiveGraph()
        graph.parse(variant_path, format=format, publicID=public_id)
        # Stripping data types as different parsers (e.g. hext) have different
        # opinions of when a bare string is of datatype XSD.string or not.
        # Probably something that needs more investigation.
        GraphHelper.strip_literal_datatypes(graph, {XSD.string})
        graph_variant.asserts.check(first_graph, graph)
        if first_graph is None:
            first_graph = graph
            first_path = variant_path
        else:
            assert first_path is not None
            GraphHelper.assert_isomorphic(
                first_graph,
                graph,
                f"checking {variant_path.relative_to(VARIANTS_DIR)} against {first_path.relative_to(VARIANTS_DIR)}",
            )