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

"""Query and install ArchLinux User Repository (AUR) packages.

You can search for packages and open their URLs. This extension is also intended to be used to \

quickly install the packages. If you are missing your favorite AUR helper tool send a PR.

Synopsis: <trigger> <pkg_name>"""

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.2"

__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.loads(response.read().decode())

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)

results_json = data['results']

results_json.sort(key=lambda item: item['Name'])

results_json.sort(key=lambda item: len(item['Name']))

for entry in results_json:

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/")])