import tempfile
from contextlib import contextmanager
from shutil import rmtree

from galaxy.util import parse_xml_string
from galaxy.managers.citations import (
    parse_citation,
    BibtexCitation,
    CitationCollection,
)

EXAMPLE_BIBTEX_CITATION = """<citation type="bibtex">@article{goecks2010galaxy,
    title={Galaxy: a comprehensive approach for supporting accessible, reproducible, and transparent computational research in the life sciences},
    author={Goecks, Jeremy and Nekrutenko, Anton and Taylor, James and The Galaxy Team},
    journal={Genome Biol},
    volume={11},
    number={8},
    pages={R86},
    year={2010}
}</citation>"""


def test_parse_citation():
    xml_text = EXAMPLE_BIBTEX_CITATION
    citation_elem = parse_xml_string(xml_text)
    with temp_directory() as test_directory:
        citation = parse_citation(citation_elem, test_directory, None)
    bibtex = citation.to_bibtex()
    assert "title={Galaxy" in bibtex


def test_citation_collection():
    citation_collection = CitationCollection()
    assert len( citation_collection ) == 0
    cite1 = QuickBibtexCitation("@article{'test1'}")
    cite1dup = QuickBibtexCitation("@article{'test1'}")
    cite2 = QuickBibtexCitation("@article{'test2'}")
    assert citation_collection.add(cite1)
    assert not citation_collection.add(cite1dup)
    assert citation_collection.add(cite2)
    assert len( citation_collection ) == 2


@contextmanager
def temp_directory():
    base_path = tempfile.mkdtemp()
    try:
        yield base_path
    finally:
        rmtree(base_path)


class QuickBibtexCitation( BibtexCitation ):

    def __init__( self, raw_bibtex ):
        self._set_raw_bibtex( raw_bibtex )