from urllib import urlencode, quote as urlquote, unquote
try:
    from urlparse import parse_qs
except ImportError:
    from cgi import parse_qs

from django import template
from django.utils.html import escape

register = template.Library()

def link_encode(url, query):
    # massage the data into all utf-8 encoded strings first, so urlencode
    # doesn't barf at the data we pass it
    query = dict((k, unicode(v).encode('utf-8')) for k, v in query.items())
    data = urlencode(query).replace('&', '&')
    return "%s?%s" % (url, data)

@register.filter
def url_unquote(original_url):
    try:
        url = original_url
        if isinstance(url, unicode):
            url = url.encode('ascii')
        url = unquote(url).decode('utf-8')
        return url
    except UnicodeError:
        return original_url

class BuildQueryStringNode(template.Node):
    def __init__(self, sortfield):
        self.sortfield = sortfield
        super(BuildQueryStringNode, self).__init__()

    def render(self, context):
        qs = parse_qs(context['current_query'])
        if qs.has_key('sort') and self.sortfield in qs['sort']:
            if self.sortfield.startswith('-'):
                qs['sort'] = [self.sortfield[1:]]
            else:
                qs['sort'] = ['-' + self.sortfield]
        else:
            qs['sort'] = [self.sortfield]
        return urlencode(qs, True).replace('&', '&')

@register.tag(name='buildsortqs')
def do_buildsortqs(parser, token):
    try:
        tagname, sortfield = token.split_contents()
    except ValueError:
        raise template.TemplateSyntaxError(
                "%r tag requires a single argument" % tagname)
    if not (sortfield[0] == sortfield[-1] and sortfield[0] in ('"', "'")):
        raise template.TemplateSyntaxError(
                "%r tag's argument should be in quotes" % tagname)
    return BuildQueryStringNode(sortfield[1:-1])

@register.simple_tag
def pkg_details_link(pkg):
    link = '<a href="%s" title="View package details for %s">%s</a>'
    return link % (pkg.get_absolute_url(), pkg.pkgname, pkg.pkgname)

@register.simple_tag
def multi_pkg_details(pkgs):
    return ', '.join([pkg_details_link(pkg) for pkg in pkgs])

@register.simple_tag
def maintainer_link(user):
    if user:
        # TODO don't hardcode
        title = escape('View packages maintained by ' + user.get_full_name())
        return '<a href="/packages/?maintainer=%s" title="%s">%s</a>' % (
                user.username,
                title,
                user.get_full_name(),
        )
    return ''

@register.simple_tag
def get_download_link(package):
    parts = {
        "repo": package.repo.name.lower(),
        "arch": package.arch.name,
        "pkgfile": package.filename
    }
    if parts["arch"] == "any":
        parts["arch"] = "i686"
    linkbase = "//repo.parabolagnulinux.org/%(repo)s/os/%(arch)s/%(pkgfile)s"
    return linkbase % parts

@register.simple_tag
def packager_link(user):
    if user:
        # TODO don't hardcode
        title = escape('View packages packaged by ' + user.get_full_name())
        return '<a href="/packages/?packager=%s" title="%s">%s</a>' % (
                user.username,
                title,
                user.get_full_name(),
        )
    return ''

@register.simple_tag
def get_wiki_link(package):
    url = "//wiki.parabolagnulinux.org/index.php"
    data = {
        'title': "Special:Search",
        'search': package.pkgname,
    }
    return link_encode(url, data)

@register.simple_tag
def svn_arch(package):
    repo = package.repo.name.lower()
    return svn_link(package, "repos/%s-%s" % (repo, package.arch.name))

@register.simple_tag
def svn_trunk(package):
    return svn_link(package, "trunk")

@register.simple_tag
def bugs_list(package):
    url = "//bugs.parabolagnulinux.org/bugs/issue"
    data = {
        '@action': 'search',
        'title': package.pkgname,
    }
    return link_encode(url, data)

@register.simple_tag
def bug_report(package):
    url = "//bugs.parabolagnulinux.org/bugs/issue"
    data = {
        '@template': 'item',
        'keyword': 'packages',
        'title': '[%s] PLEASE ENTER SUMMARY' % package.pkgname,
    }
    return link_encode(url, data)

@register.simple_tag
def flag_unfree(package):
    url = "//bugs.parabolagnulinux.org/bugs/issue"
    data = {
        '@template': 'item',
        'keyword': 'packages,unfree',
        'nosy': 'dev_list',
        'priority': 'critical',
        'title': '[%s] Please put your reasons here (register first if you haven\'t)' % package.pkgname,
    }
    return "//bugs.parabolagnulinux.org/bugs/issue?%s" % urlencode(data) 
# vim: set ts=4 sw=4 et: