# -*- coding: utf-8 -*-

"""Arch User Repository extension

This extension adapts the AUR web interface. You can search for packages and open their URLs.

This extension is also intended to be used to quickly install the packages. Currently yaourt and
pacaur can be used. If you are missing your favorite AUR helper tool send a PR."""

from albertv0 import *
from shutil import which
from datetime import datetime
from shlex import split
from urllib import request, parse
import json
import os
import re

__iid__ = "PythonInterface/v0.1"
__prettyname__ = "Archlinux User Repository"
__version__ = "1.1"
__trigger__ = "aur "
__author__ = "Manuel Schneider"
__dependencies__ = []

iconPath = os.path.dirname(__file__)+"/arch.svg"
baseurl = 'https://aur.archlinux.org/rpc/'
install_cmdline = None

if which("yaourt"):
    install_cmdline = "yaourt -S aur/%s"
elif which("pacaur"):
    install_cmdline = "pacaur -S aur/%s"


def handleQuery(query):
    if not query.isTriggered:
        return

    stripped = query.string.strip()

    if stripped:
        params = {
            'v': '5',
            'type': 'search',
            'by': 'name',
            'arg': stripped
        }
        url = "%s?%s" % (baseurl, parse.urlencode(params))
        req = request.Request(url)

        with request.urlopen(req) as response:
            data = json.load(response)
            if data['type'] == "error":
                return Item(
                    id=__prettyname__,
                    icon=iconPath,
                    text="Error",
                    subtext=data['error'],
                    completion=query.rawString
                )
            else:
                results = []
                pattern = re.compile(query.string, re.IGNORECASE)
                for entry in data['results']:
                    name = entry['Name']

                    item = Item(
                        id=__prettyname__,
                        icon=iconPath,
                        text="<b>%s</b> <i>%s</i> (%s)" % (pattern.sub(lambda m: "<u>%s</u>" % m.group(0), name), entry['Version'], entry['NumVotes']),
                        completion=query.rawString
                    )
                    subtext = entry['Description'] if entry['Description'] else "<No description>"
                    if entry['OutOfDate']:
                        subtext = '<font color="red">[Out of date: %s]</font> %s' % (datetime.fromtimestamp(entry['OutOfDate']).strftime("%F"), subtext)
                    if entry['Maintainer'] is None:
                        subtext = '<font color="red">[Orphan]</font> %s' % subtext
                    item.subtext = subtext

                    if install_cmdline:
                        tokens = split(install_cmdline % name)
                        item.addAction(TermAction("Install with %s" % tokens[0], tokens))
                        item.addAction(TermAction("Install with %s (noconfirm)" % tokens[0], tokens + ["--noconfirm"]))

                    item.addAction(UrlAction("Open AUR website", "https://aur.archlinux.org/packages/%s/" % name))

                    if entry['URL']:
                        item.addAction(UrlAction("Open project website", entry['URL']))

                    results.append(item)
                return results
    else:
        return Item(id=__prettyname__,
                    icon=iconPath,
                    text=__prettyname__,
                    subtext="Enter a query to search the AUR",
                    completion=query.rawString,
                    actions=[UrlAction("Open AUR packages website", "https://aur.archlinux.org/packages/")])