#!/usr/bin/env python2 # Version 0.7.5 import os import sys import re import subprocess import gettext from git import Repo,Git import ConfigParser import glob import shutil import optparse import fileinput PKGHOME = "/data/pkg_repo/packages" SRCPKGHOME = "/data/pkg_repo/src_packages" mydir = os.getcwd() repolist = ['core', 'extra'] variables = ['pkgbase','pkgname','pkgver','pkgrel','arch','epoch'] cli_dict = {} cli_list = [] makepkg_cmd = ['makepkg'] REPO = "none" SFIX = "" DOCROOT = "" pkgfile = "none" pkgbase = "" pkglist = "" pkgname = "" pkgver = "" pkgrel = "" epoch = "" arch = "" CARCH = "" PKGDEST = "" TOTALPKG = "" GZPKG = "" XZPKG = "" # See what git branch we're working under git_repo = Git() branches = git_repo.branch() git_branch="didnt_find_it" for i in branches.split("\n"): if i.strip().startswith("*"): git_branch = i.split("*")[1].lstrip(' ') break print git_branch if git_branch == "R7-testing": SFIX = "-testing" elif git_branch == "master": SFIX = "" else: print "Can't determine which git branch is in use!" sys.exit(2) print "Git Branch suffix is:",SFIX # Color stuff -- does this work? Looks to bash-y to me. ALL_OFF = "$(tput sgr0)" BOLD = "$(tput bold)" BLUE = "${BOLD}$(tput setaf 4)" GREEN = "${BOLD}$(tput setaf 2)" RED = "${BOLD}$(tput setaf 1)" YELLOW = "${BOLD}$(tput setaf 3)" def plain(): mesg = list.pop() print "${BOLD} "+mesg+ "${ALL_OFF}\n"+cmdline def msg(): mesg = list.pop() print "${GREEN}==>${ALL_OFF}${BOLD} "+mesg+ "${ALL_OFF}\n"+cmdline def msg(): mesg = list.pop() print "${BLUE} ->${ALL_OFF}${BOLD} "+mesg+ "${ALL_OFF}\n"+cmdline def warning(): warning = gettext(WARNING) mesg = list.pop() print "${YELLOW}==>"+ warning+ "${ALL_OFF}${BOLD} "+mesg+ "${ALL_OFF}\n"+cmdline def error(): error = gettext(ERROR) mesg = list.pop() print "${RED}==>"+ error+ "${ALL_OFF}${BOLD} "+mesg+ "${ALL_OFF}\n"+cmdline def commandline(): global makepkg_cmd global cli_list global cli_dict clparser = optparse.OptionParser() clparser.add_option("--asroot", action="store_true", default=True, help="Allow makepkg to run as root.") clparser.add_option("-A", "--ignorearch", action="store_true", default=False, help="Ignore a missing or incomplete arch field in the build script.") clparser.add_option("-b", "--bump", action="store_true", default=False, help="Increase package release one unit.") clparser.add_option("-c", "--clean", action="store_true", default=False, help="Clean up leftover work files and directories after a successful build. ") clparser.add_option("-C", "--cleancache", action="store_true", default=False, help="Removes all cached source files from the directory specified in SRCDEST in makepkg.conf") clparser.add_option("--config", action="store", help="Use an alternate config file instead of the /etc/makepkg.conf default.") clparser.add_option("-d", "--nodeps", action="store_true", default=False, help="Do not perform any dependency checks.") clparser.add_option("-e", "--noextract", action="store_true", default=False, help="Do not extract source files; use whatever source already exists in the src/ directory.") clparser.add_option("-f", "--force", action="store_true", default=False, help="This allows a built package to be overwritten.") clparser.add_option("--forcever", action="store_true", default=False, help="This is a hidden option that should not be used unless you really know what you are doing.") clparser.add_option("-g", "--geninteg", action="store_true", default=False, help="For each source file in the source array of PKGBUILD, download the file if required and generate integrity checks.") clparser.add_option("--skipinteg", action="store_true", default=False, help="Do not perform any integrity checks, just print a warning instead.") clparser.add_option("--holdver", action="store_true", default=True, help="Prevents makepkg from automatically bumping the pkgver to the latest revision number in the package's development tree.") clparser.add_option("--nohold", action="store_false", dest="holdver", help="Automatically bump the pkgver to the latest revision number in the package's development tree.") clparser.add_option("-i", "--install", action="store_true", default=False, help="Install or upgrade the package after a successful build.") clparser.add_option("-L", "--log", action="store_true", default=True, help="Enable makepkg build logging.") clparser.add_option("-m", "--nocolor", action="store_true", default=False, help="Disable color in output messages.") clparser.add_option("-o", "--nobuild", action="store_true", default=False, help="Download and extract files only, but do not build them.") clparser.add_option("-p", action="store", help="Read the package script buildscript instead of the PKGBUILD default.", metavar='/path/to/buildscript') clparser.add_option("-r", "--rmdeps", action="store_true", default=False, help="Upon successful build, remove any dependencies installed by makepkg during dependency auto-resolution and installation when using -s.") clparser.add_option("-R", "--repackage", action="store_true", default=False, help="Repackage contents of the package without rebuilding the package.") clparser.add_option("-s", "--syncdeps", action="store_true", default=False, help="Install missing dependencies using pacman.") clparser.add_option("--allsource", action="store_true", default=False, help="Do not actually build the package, but build a source-only tarball that includes all sources, including those that are normally download via makepkg.") clparser.add_option("--source", action="store_true", default=False, help="Do not actually build the package, but build a source-only tarball that does not include sources that can be fetched via a download URL.") clparser.add_option("--pkg", action="store", help="Only build listed packages from a split package.") clparser.add_option("--noconfirm", action="store_true", default=False, help="(Passed to pacman) Prevent pacman from waiting for user input before proceeding with operations.") clparser.add_option("--noprogressbar", action="store_true", default=False, help="(Passed to pacman) Prevent pacman from displaying a progress bar.") clparser.add_option("--rmold", action="store_true", default=False, help="BETA: Remove old src and software packages from repos. Use with caution. False positives may occur (i.e. nvidia pkgs)!") (options, args) = clparser.parse_args() options1 = ['config', 'p', 'pkg'] options2 = ['asroot', 'ignorearch', 'bump', 'clean', 'cleancache', 'nodeps', 'noextract', 'force', 'forcever', 'geninteg', 'skipinteg', 'holdver', 'install', 'log', 'nocolor', 'nobuild', 'rmdeps', 'repackage', 'syncdeps', 'allsource', 'source', 'noconfirm', 'noprogressbar', 'rmold'] for o in options1: cmd1 = eval('options.'+o) if o is not 'p': if cmd1 is not None: cli_dict['--'+o] = cmd1 elif cmd1 is not None: cli_dict['-'+o] = cmd1 print "CLI Dictionary is:",cli_dict for o in options2: cmd2 = eval('options.'+o) if cmd2 is True: cli_list.append('--'+o) # Create makepkg command makepkg_cmd.extend(cli_list) for k in cli_dict.keys(): v = cli_dict.get(k) makepkg_cmd.append(k) makepkg_cmd.append(v) # Remove bump option from makepkg command if it exists if "--bump" in makepkg_cmd: makepkg_cmd.remove("--bump") # Remove rmold option from makepkg command if it exists if "--rmold" in makepkg_cmd: makepkg_cmd.remove("--rmold") # Remove "--geninteg" option (if it exists) from makepkg command. # Checking for *sums is done automaticly by mp.py if "--geninteg" in makepkg_cmd: makepkg_cmd.remove("--geninteg") print "Makepkg Command:",makepkg_cmd def bump_pkg(): # We need pkgrel to live beyond this function global SRCPKG global pkgrel # Backup the original pkgfile shutil.copy2(pkgfile, pkgfile + ".old") # Let's increase the pkgrel for line in fileinput.input(pkgfile, inplace=1): if line.strip().startswith("pkgrel"): pkgrel = line.partition("=")[2].strip('\n') # Add 1 to pkgrel new_pkgrel = int(pkgrel) + 1 line = line.replace("pkgrel=" + pkgrel, "pkgrel=" + str(new_pkgrel)) pkgrel = str(new_pkgrel) sys.stdout.write(line) print "Bumped " + pkgname + " release to " + str(new_pkgrel) if pkgbase: SRCPKG = pkgbase + "-" + epoch + pkgver + "-" + str(new_pkgrel) + ".src.tar.gz" else: SRCPKG = pkglist[0] + "-" + epoch + pkgver + "-" + str(new_pkgrel) + ".src.tar.gz" print "Source package will be: ",SRCPKG def find_repo(): global REPO global DOCROOT # Get all our possible repos into one list for item in ("chroot-devel", "mv-core", "xmpl", "local"): repolist.append(item) print "Repo List:",repolist # Create a list with the directory names of our current directory tree dir_tree = os.path.dirname(mydir).split("/") print "Dir Tree:",dir_tree # Loop through the dir_tree to see if we can find our repo for item in repolist: if item not in dir_tree: continue else: repo_name = item if repo_name == "extra": REPO = "extra" + SFIX elif repo_name == "core": REPO = "core" + SFIX elif repo_name == "xmpl": REPO = "local" elif repo_name not in repolist: print "ERROR in function find_repo: Cannot determine repository!" sys.exit(2) else: REPO = repo_name # Ensure our DOCROOT exists and if not, create it DOCROOT = "/data/pkg_repo/" + CARCH + "/" + REPO print "DOCROOT:",DOCROOT if os.path.exists(DOCROOT): print "INFO: Repository is",REPO else: os.mkdir(DOCROOT,0755) print "INFO: ",DOCROOT,"directory created. Repository is",REPO def update_repo(): # pkgname could be a list of several pkgs. Since bash array format is # loose, let bash parse the pkgname(s) first, then return a list for us. os.chdir(DOCROOT) print print "INFO: Changed working dir to",DOCROOT print # Check for --pkg in makepkg command and adjust pkglist accordingly if not "--pkg" in cli_dict.keys(): global pkglist else: v = cli_dict["--pkg"] pkglist = v.split() print "Package list:",pkglist for i in pkglist: print "Package name:",i GZPKG = i + "-" + epoch + pkgver + "-" + pkgrel + "-" + CARCH + ".pkg.tar.gz" XZPKG = i + "-" + epoch + pkgver + "-" + pkgrel + "-" + CARCH + ".pkg.tar.xz" if os.path.isfile(PKGDEST + "/" + XZPKG): TOTALPKG = XZPKG elif os.path.isfile(PKGDEST + "/" + GZPKG): TOTALPKG = GZPKG else: print "ERROR in function update_repo: Couldn't find the new package",PKGDEST + "/" + TOTALPKG sys.exit(2) print "Package file:",TOTALPKG # Remove old package(s) from local copy oldpkgcheck = re.compile( i + "-" + epoch + '(pkgver|[\d.]+)' + "-" + '(?!pkgrel)' ) dirlist = os.listdir(DOCROOT) for n in dirlist: if n.startswith(i): if not oldpkgcheck.search(n): print n,"does not match" continue else: OLDPKG = glob.glob(oldpkgcheck.search(n).group() + "*.pkg.tar.?z") print "OLDPKG =",OLDPKG if OLDPKG: for DELPKG in OLDPKG: if "--rmold" in cli_list: print "Deleting old package:",DELPKG os.remove(DELPKG) # Remove any symlinks to old packages # We make it conditional on "--force" because force will overwrite # an existing package and we want the symlink to stay, pointing to # the newly built package with the same pkgrel. if "--force" not in makepkg_cmd: if os.path.islink(mydir + "/" + DELPKG): os.remove(mydir + "/" + DELPKG) # Copy in new package print "############################################" print "Updating " + DOCROOT + " with " + TOTALPKG print "Copying " + PKGDEST + "/" + TOTALPKG shutil.copy2(PKGDEST + "/" + TOTALPKG, DOCROOT) subprocess.call(["repo-add", DOCROOT+ "/" + REPO + ".db.tar.gz", DOCROOT + "/" + TOTALPKG]) print "############################################" def update_src_pkg(): print "---------------------------------SRC------------------------------" OLDSRCPKG="" os.chdir(SRCPKGHOME + "/" + REPO) print print "INFO: Changed working dir to",SRCPKGHOME + "/" + REPO print if pkgbase: SRCPKG = pkgbase + "-" + epoch + pkgver + "-" + pkgrel + ".src.tar.gz" i = pkgbase else: SRCPKG = pkglist[0] + "-" + epoch + pkgver + "-" + pkgrel + ".src.tar.gz" i = pkglist[0] print "Source package will be:",SRCPKG if not SRCPKGHOME + "/" + REPO + "/" + SRCPKG: print "ERROR in function update_repo: Couldn't find the new package",SRCPKGHOME + "/" + REPO + "/" + SRCPKG sys.exit(2) print "SRCPKG:",SRCPKG oldpkgcheck = re.compile( i + "-" + epoch + '(pkgver|[\d.]+)' + "-" + '(?!pkgrel)' ) dirlist = os.listdir(SRCPKGHOME + "/" + REPO) if pkgbase: # Remove old src package(s) from local copy if "--rmold" in cli_list: for n in dirlist: if n.startswith(pkgbase): if not oldpkgcheck.search(n): print n,"does not match" continue else: OLDSRCPKG = glob.glob(oldpkgcheck.search(n).group() + "*.src.tar.gz") else: # Remove old src package(s) from local copy if "--rmold" in cli_list: for n in dirlist: if n.startswith(pkgname): if not oldpkgcheck.search(n): print n,"does not match" continue else: OLDSRCPKG = glob.glob(oldpkgcheck.search(n).group() + "*.src.tar.gz") if OLDSRCPKG: print "OLDSRCPKG =",OLDSRCPKG for DELSRCPKG in OLDSRCPKG: print "Deleting old source package",DELSRCPKG os.remove(DELSRCPKG) print "Copying new source package to",SRCPKGHOME + "/" + REPO + "/" + SRCPKG shutil.copy2(PKGHOME + "/" + SRCPKG, SRCPKGHOME + "/" + REPO + "/") def dup_check(): if REPO in ("local", "mv-core", "xmpl", "chroot-devel"): return for tmp_repo in ("core", "extra"): if tmp_repo + SFIX != REPO: p1 = subprocess.Popen(["pacman", "-Sl", tmp_repo + SFIX], stdout=subprocess.PIPE, stderr=subprocess.STDOUT,) p2 = subprocess.Popen(["cut", "-d", " ", "-f", "2"], stdin=p1.stdout, stdout=subprocess.PIPE,) p3 = subprocess.Popen(["grep", pkgname], stdin=p2.stdout, stdout = subprocess.PIPE,) output = p3.communicate()[0].strip('\n') if output == 0: if "--force" not in cli_list: print "#######################################" print pkgname," already exists in ",tmp_repo + SFIX print print "Use --force to overwite" print "#######################################" sys.exit(2) else: print "Force detected! Making package regardless!" print "#######################################" else: if output != 0: return def updateMD5(): # Do we have integrity checks? sums = [] # Open pkgfile for reading f = open(pkgfile, 'r') pkgfile_contents = f.readlines() f.close() # Iterate through the lines looking for a match pattern for line in pkgfile_contents: check = re.compile('sha1sums|sha256sums|sha384sums|sha512sums|md5sums') sums = check.match(line) if sums: break if not sums: # If no matches are found, append md5sums to the end of the pkgfile p = open(pkgfile, 'a') md5gen = subprocess.Popen(["makepkg", "--asroot", "-g"], stdout = subprocess.PIPE,).communicate()[0] p.writelines(md5gen) p.close() def config_file(): global pkgfile global pkgbase global pkgname global pkglist global pkgver global pkgrel global epoch global arch global SRCPKG global CARCH global PKGDEST # Check what file will be used for our PKGBUILD if "-p" in cli_dict: pkgfile = cli_dict["-p"] else: pkgfile = "PKGBUILD" print "Config File:",pkgfile # Does the file exist? if not os.path.isfile(pkgfile): print "ERROR in function config_file: Can't find ",pkgfile,"!" sys.exit(2) else: # Loop over contents to get our variables # Use bash to do it because PKGBUILDs are very loose with their format for item in variables: v = subprocess.Popen(['/bin/bash','-c', 'source ' + pkgfile + '; echo ${' + item + '[@]}'], stdout = subprocess.PIPE,) value = v.communicate()[0].strip('\n') if item == "pkgbase": pkgbase = value print "pkgbase:",pkgbase elif item == "pkgname": pkgname = value print "pkgname:",pkgname pkglist = list(value.split()) print "pkglist:",pkglist elif item == "pkgver": pkgver = value print "pkgver:",pkgver elif item == "pkgrel": pkgrel = value print "pkgrel:",pkgrel elif item == "epoch": if value: epoch = "%s:" %value print "epoch:" , epoch elif item == "arch": arch = value print "arch:",arch if pkgbase: SRCPKG = pkgbase + "-" + epoch + pkgver + "-" + pkgrel + ".src.tar.gz" else: SRCPKG = pkglist[0] + "-" + epoch + pkgver + "-" + pkgrel + ".src.tar.gz" print "Source package will be:",SRCPKG # Get needed makepkg.conf variables mpkg="/etc/makepkg.conf" f=open(mpkg,"r") # Read file contents into memory makepkg_contents=f.readlines() f.close() # Loop over contents to get our variables for mp_lines in makepkg_contents: if mp_lines.strip().startswith("CARCH"): CARCH = mp_lines.partition("=")[2].strip('\n').replace('\"','') if arch != 'any': arch = CARCH if mp_lines.strip().startswith("PKGDEST"): PKGDEST = mp_lines.partition("=")[2].strip('\n') print "CARCH is:",CARCH print "Package destination is:",PKGDEST print "Architecture is:",arch def make_package(): global arch global CARCH # Make the package! retcode = subprocess.call(makepkg_cmd) if retcode != 0: print "ERROR: makepkg failed with return code ",retcode if "--bump" in cli_list: if os.path.isfile(pkgfile + '.old'): shutil.move(pkgfile + '.old', pkgfile) print "Reverted pkgrel bump." sys.exit(2) elif "--nobuild" in makepkg_cmd: sys.exit(0) else: print "=============FINSHED CREATING PKG==================" if arch == 'any': CARCH = 'any' # MAIN PROGRAM commandline() config_file() find_repo() dup_check() # If "bump" was called, run the bump_pkg function if "--bump" in cli_list: bump_pkg() updateMD5() # Everything look good? Go for it! make_package() print "------------updating database-------------" update_repo() print "----------creating source package---------" os.chdir(mydir) retcode = subprocess.call(["makepkg", "--force", "--holdver", "--asroot", "--source"]) if retcode != 0: print "ERROR: Source package creation failed with return code",retcode sys.exit(2) else: update_src_pkg() # Clean up old PKGBUILD if it exists os.chdir(mydir) if os.path.isfile(pkgfile + '.old'): os.remove(pkgfile + '.old') print "Removed temporary backup file",pkgfile + '.old' #remove src.tar.gz so it stops cluttering up git if os.path.islink(SRCPKG): os.unlink(SRCPKG) # Remove any old -build and -package log files os.chdir(mydir) build_log = pkgname + "-" + pkgver + "-" + pkgrel + "-" + CARCH + "-build.log" pkg_log = pkgname + "-" + pkgver + "-" + pkgrel + "-" + CARCH + "-package.log" all_logs = glob.glob("*.log*") #print "All Logs:",all_logs saved_logs = [] if os.path.isfile(build_log): saved_logs.append(build_log) if os.path.isfile(pkg_log): saved_logs.append(pkg_log) #print "Saved Logs:",saved_logs for log in all_logs: if log not in saved_logs: os.remove(log)