aur_repo/package/Package.py

#!/usr/bin/env python3
import sh
from sh import pacman
from sh import git
from sh import sudo
import requests
import re
import os
import shutil
import glob

class BuildError(Exception):
  def __init__(self):
    pass

class Package(object):
  def __init__(self, name, buildPath, repoPath, repoName):
    self.name = name
    self.buildPath = buildPath
    self.repoPath = repoPath
    self.repoName = repoName
    self.path = os.path.join(self.buildPath, name)
    self.repo = False
    self.aur = False
    self.group = False
    self.upToDate = False
    self.aurdeps = []
    self.pkgs = ""
    self.inaur()
    if not self.aur:
      self.inrepo()
    if self.aur:
      self.isgroup()
      if not os.path.exists(self.path):
        self.clone()
      else:
        self.pull()
        if self.upToDate:
          self.checkGit()
      try:
        self.getDeps()
      except BuildError:
        print("Aborting build of %s" % self.name)

  def inrepo(self):
    try:
      results = pacman("-Ssq", self.name)
    except sh.ErrorReturnCode_1:
      return
    for result in results.split("\n"):
      if self.name == result:
        self.repo = True

  def inaur(self):
    response = requests.get("https://aur.archlinux.org/packages/%s" % (self.name))
    if response.status_code < 400:
      self.aur = True

  def isgroup(self):
    response = requests.get("https://aur.archlinux.org/pkgbase/%s" % (self.name))
    if response.status_code < 400:
      self.group = True

  def clone(self):
    git.clone("https://aur.archlinux.org/%s.git" % (self.name), self.path)

  def pull(self):
    results = git("-C", self.path, "pull")
    if "Already up-to-date" in results:
      self.upToDate = True

  def checkGit(self):
    pass

  def getDeps(self):
    pkgbuild = open(os.path.join(self.path, "PKGBUILD"), errors="surrogateescape").read()
    depends = []
    m = re.findall("^makedepends.*?=\((.*?)\)\s*$", pkgbuild, re.MULTILINE | re.DOTALL)
    if m:
      m = " ".join(m)
      depends.extend(m.replace("'", "").replace('"', '').split())
    for dep in depends:
      tmp = Package(dep, self.buildPath, self.repoPath, self.repoName)
      if not tmp.aur and not tmp.repo:
        print("Could not find make dependency %s" % (dep))
      if tmp.aur:
        self.aurdeps.append(tmp)
      else:
        try:
          pacman("-Qi", dep)
        except sh.ErrorReturnCode_1:
          try:
            print("Installing make dependency %s" % (dep))
            results = sudo.pacman("--noconfirm", "-S", dep)
          except sh.ErrorReturnCode_1:
            print("Could not install make dependency %s" % (dep))
            raise BuildError
    depends = []
    m = re.findall("^depends.*?=\((.*?)\)\s*$", pkgbuild, re.MULTILINE | re.DOTALL)
    if m:
      m = " ".join(m)
      depends.extend(m.replace("'", "").replace('"', '').split())
    for dep in depends:
      tmp = Package(dep, self.buildPath, self.repoPath, self.repoName)
      if not tmp.aur and not tmp.repo:
        print("Could not find dependency %s" % (dep))
      elif tmp.aur:
        self.aurdeps.append(tmp)
      else:
        try:
          pacman("-Qi", dep)
        except sh.ErrorReturnCode_1:
          try:
            print("Installing dependency %s" % (dep))
            results = sudo.pacman("--noconfirm", "-S", dep)
          except sh.ErrorReturnCode_1:
            print("Could not install dependency %s" % (dep))
            raise BuildError

  def build(self):
    for dep in self.aurdeps:
      try:
        dep.build()
      except BuildError:
        print("could not build dependency %s" % (dep.name))
        return
    print("Building", self.name)
    os.chdir(self.path)
    try:
      results = sh.makepkg("-d", "--noconfirm", _err="/var/log/aur_repo/%s.log" % self.name)
    except sh.ErrorReturnCode_1:
      with open("/var/log/aur_repo/%s.log" % self.name) as log:
        if "A package has already been built" in log.read():
          print("%s is up to date" % (self.name))
          return
      raise BuildError
    except sh.ErrorReturnCode_2:
      print("Error building %s, see /var/log/aur_repo/%s.log" % (self.name, self.name))
      raise BuildError
    for line in open("/var/log/aur_repo/%s.log" % self.name).read().split("\n"):
      if "Finished making" in line:
        if(self.group):
          self.pkgs = sh.glob("%s/*.pkg.*" % (self.path))
        else:
          self.pkgs = sh.glob("%s/%s*.pkg.*" % (self.path,self.name))[0]
    self.add()

  def add(self):
    if self.pkgs:
      for pkg in self.pkgs:
        dbPath = os.path.join(self.repoPath, "%s.db.tar.gz" % (self.repoName))
        pkgPath = os.path.join(self.repoPath, os.path.basename(pkg))
        shutil.copy(pkg, pkgPath)
        repoAdd = sh.Command("repo-add")
        repoAdd(dbPath, pkgPath)