i think i have something working
This commit is contained in:
brent s
226
ARB/package.py
Normal file
226
ARB/package.py
Normal file
@@ -0,0 +1,226 @@
|
||||
import copy
|
||||
import os
|
||||
import re
|
||||
import shutil
|
||||
import subprocess
|
||||
import tarfile
|
||||
import tempfile
|
||||
import warnings
|
||||
##
|
||||
import Namcap
|
||||
import requests
|
||||
##
|
||||
import arb_util
|
||||
|
||||
|
||||
# TODO: should this be a configuration option?
|
||||
aurbase = 'https://aur.archlinux.org'
|
||||
|
||||
# Maps the AUR API attribute names to their pkgbuild equivalents.
|
||||
# Remote all attributes not present here.
|
||||
_attrmap = {'Description': 'desc',
|
||||
'Depends': 'depends',
|
||||
'License': 'licenses',
|
||||
'Name': 'name',
|
||||
'URL': 'url',
|
||||
'URLPath': 'pkgurl',
|
||||
'Version': 'version'}
|
||||
# And this is a blank dict as returned by Namcap.package (with useless values removed)
|
||||
_pkgattrs = {k: None for k in ('base', 'desc', 'install', 'name', 'url', 'version')}
|
||||
_pkgattrs.update({k: [] for k in ('arch', 'backup', 'conflicts', 'depends', 'groups', 'licenses', 'makedepends',
|
||||
'md5sums', 'names', 'optdepends', 'options', 'orig_depends', 'orig_makedepends',
|
||||
'orig_optdepends', 'orig_provides', 'provides', 'replaces', 'sha1sums',
|
||||
'sha224sums', 'sha256sums', 'sha384sums', 'sha512sums', 'source', 'split',
|
||||
'validgpgkeys')})
|
||||
# This be custom.
|
||||
_pkgattrs.update({'pkgurl': None})
|
||||
|
||||
|
||||
class Package(object):
|
||||
def __init__(self, pkg_xml, ns = '', gpgobj = None, *args, **kwargs):
|
||||
self.xml = pkg_xml
|
||||
self.always_build = arb_util.xmlBool(pkg_xml.attrib.get('alwaysBuild', True))
|
||||
self.cleanup = arb_util.xmlBool(pkg_xml.attrib.get('cleanUp', True))
|
||||
self.name = pkg_xml.text
|
||||
self.gpg = gpgobj
|
||||
self.pkginfo = None
|
||||
self.srcdir = None
|
||||
|
||||
def build(self, destdir):
|
||||
if not self.srcdir:
|
||||
raise RuntimeError('You must run .extract() before running .build()')
|
||||
prebuild_files = []
|
||||
postbuild_files = []
|
||||
for root, dirs, files in os.walk(self.srcdir):
|
||||
for f in files:
|
||||
prebuild_files.append(os.path.join(root, f))
|
||||
os.chdir(self.srcdir)
|
||||
# customizepkg-scripting in AUR
|
||||
try:
|
||||
custpkg_out = subprocess.run(['/usr/bin/customizepkg',
|
||||
'-m'],
|
||||
stdout = subprocess.PIPE,
|
||||
stderr = subprocess.PIPE)
|
||||
except FileNotFoundError:
|
||||
pass # Not installed
|
||||
build_out = subprocess.run(['/usr/bin/multilib-build',
|
||||
'-c',
|
||||
'--',
|
||||
'--',
|
||||
'--skippgpcheck',
|
||||
'--syncdeps',
|
||||
'--noconfirm',
|
||||
'--log',
|
||||
'--holdver',
|
||||
'--skipinteg'],
|
||||
stdout = subprocess.PIPE,
|
||||
stderr = subprocess.PIPE)
|
||||
for root, dirs, files in os.walk(self.srcdir):
|
||||
for f in files:
|
||||
fpath = os.path.join(root, f)
|
||||
if fpath in prebuild_files:
|
||||
continue
|
||||
if fpath.endswith('.log'):
|
||||
continue
|
||||
postbuild_files.append(fpath)
|
||||
postbuild_files = [i for i in postbuild_files if i.endswith('.pkg.tar.xz')]
|
||||
if not postbuild_files:
|
||||
warnings.warn('Could not reliably find any built packages for {0}; skipping'.format(self.name))
|
||||
else:
|
||||
for f in postbuild_files:
|
||||
fdest = os.path.join(destdir,
|
||||
os.path.basename(f))
|
||||
if os.path.isfile(fdest):
|
||||
os.remove(fdest)
|
||||
shutil.move(f, fdest)
|
||||
if self.cleanup:
|
||||
shutil.rmtree(self.srcdir)
|
||||
return([os.path.basename(f) for f in postbuild_files])
|
||||
|
||||
def extract(self, dest):
|
||||
# no-op; this is handled in the subclasses since it's unique to them.
|
||||
pass
|
||||
return(True)
|
||||
|
||||
def getPkgInfo(self):
|
||||
# no-op; this is handled in the subclasses since it's unique to them.
|
||||
pass
|
||||
return(True)
|
||||
|
||||
|
||||
class LocalPkg(Package):
|
||||
def __init__(self, pkg_xml, ns = '', *args, **kwargs):
|
||||
super().__init__(pkg_xml, ns = ns, *args, **kwargs)
|
||||
self.source = os.path.abspath(os.path.expanduser(pkg_xml.attrib.get('path',
|
||||
'.')))
|
||||
|
||||
def extract(self, dest):
|
||||
self.getPkgInfo()
|
||||
if os.path.isfile(self.source):
|
||||
try:
|
||||
with tarfile.open(name = self.source, mode = 'r|*') as tar:
|
||||
tar.extractall(dest)
|
||||
dest = os.path.join(dest, self.name)
|
||||
except tarfile.ReadError as e:
|
||||
if str(e) != 'invalid header':
|
||||
# TODO: log instead
|
||||
raise (e)
|
||||
# "invalid header" means it isn't a tarball. Contextually, that means a PKGBUILD file.
|
||||
dest = os.path.join(dest, self.name)
|
||||
os.makedirs(dest, exist_ok = True)
|
||||
shutil.copy2(self.source, dest)
|
||||
elif os.path.isdir(self.source):
|
||||
os.makedirs(dest, exist_ok = True)
|
||||
# Already "extracted".
|
||||
shutil.copytree(self.source, dest)
|
||||
pkg = dict(Namcap.package.load_from_pkgbuild(os.path.join(dest, 'PKGBUILD')))
|
||||
del(pkg['setvars'])
|
||||
self.pkginfo.update(pkg)
|
||||
self.srcdir = dest
|
||||
return(True)
|
||||
|
||||
def getPkgInfo(self):
|
||||
pkgbuild = None
|
||||
pkgbld_re = re.compile(r'(^|/)PKGBUILD$')
|
||||
is_temp = False
|
||||
if os.path.isfile(self.source):
|
||||
try:
|
||||
with tarfile.open(name = self.source, mode = 'r:*') as tar:
|
||||
for f in tar.getmembers():
|
||||
if pkgbld_re.search(f.name):
|
||||
pkgbuild = tempfile.mkstemp()[1]
|
||||
with open(pkgbuild, 'wb') as fh:
|
||||
fh.write(tar.extractfile(f).read())
|
||||
is_temp = True
|
||||
break
|
||||
except tarfile.ReadError as e:
|
||||
if str(e) != 'file could not be opened successfully':
|
||||
# TODO: log instead
|
||||
raise(e)
|
||||
# "file could not be opened successfully" means it isn't a tarball.
|
||||
# Contextually, that means a PKGBUILD file.
|
||||
pkgbuild = self.source
|
||||
elif os.path.isdir(self.source):
|
||||
pkgbuild = os.path.join(self.source, 'PKGBUILD')
|
||||
if not pkgbuild:
|
||||
raise RuntimeError('Could not find a PKGBUILD for {0}'.format(self.name))
|
||||
pkg = copy.deepcopy(_pkgattrs)
|
||||
pkg.update(dict(Namcap.package.load_from_pkgbuild(pkgbuild)))
|
||||
del(pkg['setvars'])
|
||||
if is_temp:
|
||||
os.remove(pkgbuild)
|
||||
if self.pkginfo and isinstance(self.pkginfo, dict):
|
||||
self.pkginfo.update(pkg)
|
||||
else:
|
||||
self.pkginfo = pkg
|
||||
return()
|
||||
|
||||
|
||||
class AURPkg(Package):
|
||||
def __init__(self, pkg_xml, ns = '', *args, **kwargs):
|
||||
super().__init__(pkg_xml, ns = ns, *args, **kwargs)
|
||||
|
||||
def extract(self, dest):
|
||||
dl_url = None
|
||||
self.getPkgInfo()
|
||||
if self.pkginfo['name'] == self.name:
|
||||
dl_url = os.path.join(aurbase, re.sub(r'^/+', r'', self.pkginfo['pkgurl']))
|
||||
if not dl_url:
|
||||
# TODO: log instead?
|
||||
warnings.warn('Could not find a download path for {0}; skipping'.format(self.name))
|
||||
return(False)
|
||||
with requests.get(dl_url, stream = True) as url:
|
||||
try:
|
||||
with tarfile.open(mode = 'r|*', fileobj = io.BytesIO(url.content)) as tar:
|
||||
tar.extractall(dest)
|
||||
dest = os.path.join(dest, self.name)
|
||||
# This *technically* does nothing unless the AUR is *very* broken.
|
||||
except tarfile.ReadError as e:
|
||||
# "invalid header" means it isn't a tarball
|
||||
if str(e) != 'invalid header':
|
||||
# TODO: log instead
|
||||
raise(e)
|
||||
pkg = Namcap.package.load_from_pkgbuild(os.path.join(dest, 'PKGBUILD'))
|
||||
del(pkg['setvars'])
|
||||
self.pkginfo.update(pkg)
|
||||
self.srcdir = dest
|
||||
return(True)
|
||||
|
||||
def getPkgInfo(self):
|
||||
pkg_srch = requests.get(os.path.join(aurbase, 'rpc'),
|
||||
params = {'v': 5,
|
||||
'type': 'info',
|
||||
'arg': self.name}).json()
|
||||
if 'results' not in pkg_srch:
|
||||
raise RuntimeError(('AUR request for {0} was unsuccessful.'
|
||||
'Check {1} for status').format(self.name, aurbase))
|
||||
if len(pkg_srch['results']) != 1:
|
||||
# TODO: log instead?
|
||||
warnings.warn('Package {0} not found in the AUR'.format(self.name))
|
||||
return(False)
|
||||
pkginfo = copy.deepcopy(_pkgattrs)
|
||||
for k, v in pkg_srch['results'][0].items():
|
||||
if k in _attrmap:
|
||||
pkginfo[_attrmap[k]] = v
|
||||
self.pkginfo = pkginfo
|
||||
return()
|
||||
Reference in New Issue
Block a user