Ticket #7728: hrefexample.py

""" Copyright (c) 2008 Martin Scharrer <martin@scharrer-online.de>
    v0.1 - Oct 2008
    This is Free Software under the GPL v3!
""" 
from trac.core import *
from trac.wiki.api import parse_args
from trac.wiki.macros import WikiMacroBase
from genshi.builder import Element
from trac.wiki.formatter import extract_link
from StringIO import StringIO


class HrefExtractionExampleMacro(WikiMacroBase):
    """ Provides the macro
    """

    def expand_macro(self, formatter, name, content):
        args, kwargs = parse_args(content)
        traclink = args[0]

        link = extract_link(self.env, formatter.context, traclink)
        if isinstance(link, Element):
            href = link.attrib.get('href')
        else:
            href = traclink

        if href == None:
            raise TracError('Link "%s" invalid!' % traclink)

        # Raw href is needed:
        if href.startswith('/browser/'):
            if href.find('?') == -1:
                href = '%s?format=raw' % href
            else:
                href = '%s&format=raw' % href
        elif href.startswith('/attachment/'):
            href = href.replace('/attachment/','/raw-attachment/', 1)

        out = StringIO()
        # Test output:
        out.write( '||%s| => |%s||<br />\n' % (link,href) );
        return out.getvalue()