"""Citation handling for LaTeX output."""

#-----------------------------------------------------------------------------
# Copyright (c) 2013, the IPython Development Team.
#
# Distributed under the terms of the Modified BSD License.
#
# The full license is in the file COPYING.txt, distributed with this software.
#-----------------------------------------------------------------------------

#-----------------------------------------------------------------------------
# Code
#-----------------------------------------------------------------------------


__all__ = ['citation2latex']


def citation2latex(s):
    """Parse citations in Markdown cells.
    
    This looks for HTML tags having a data attribute names `data-cite`
    and replaces it by the call to LaTeX cite command. The tranformation
    looks like this:
    
    `<cite data-cite="granger">(Granger, 2013)</cite>`
    
    Becomes
    
    `\\cite{granger}`
    
    Any HTML tag can be used, which allows the citations to be formatted
    in HTML in any manner.
    """
    try:
        from lxml import html
    except ImportError:
        return s

    tree = html.fragment_fromstring(s, create_parent='div')
    _process_node_cite(tree)
    s = html.tostring(tree, encoding='unicode')
    if s.endswith('</div>'):
        s = s[:-6]
    if s.startswith('<div>'):
        s = s[5:]
    return s


def _process_node_cite(node):
    """Do the citation replacement as we walk the lxml tree."""
    
    def _get(o, name):
        value = getattr(o, name, None)
        return '' if value is None else value
    
    if 'data-cite' in node.attrib:
        cite = '\cite{%(ref)s}' % {'ref': node.attrib['data-cite']}
        prev = node.getprevious()
        if prev is not None:
            prev.tail = _get(prev, 'tail') + cite + _get(node, 'tail')
        else:
            parent = node.getparent()
            if parent is not None:
                parent.text = _get(parent, 'text') + cite + _get(node, 'tail')
        try:
            node.getparent().remove(node)
        except AttributeError:
            pass
    else:
        for child in node:
            _process_node_cite(child)