auto_debuild.py

   1 #!/usr/bin/python
   2 # auto-debuild - Automatic Generation of Debian Packages
   3 # Copyright (C) 2012 Ralf Jung <post@ralfj.de>
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18
  19 import os, shutil, stat, time, subprocess, sys, shlex, tempfile, argparse
  20 from collections import OrderedDict
  21
  22 # a dict with some useful additional getters which can convert types and handle one-element lists like their single member
  23 class ConfigDict(dict):
  24         def getstr(self, name, default = None):
  25                 if not name in self: return default
  26                 val = self[name]
  27                 if isinstance(val, list):
  28                         if len(val) != 1: raise Exception('%s is a list, but it should not' % name)
  29                         return val[0]
  30                 else:
  31                         return val
  32
  33         def getint(self, name, default = None):
  34                 return int(self.getstr(name, default))
  35
  36         def getbool(self, name, default = None):
  37                 val = self.getstr(name, default)
  38                 if isinstance(val, bool): return val # already a bool
  39                 return val.lower() in ('true', 'yes', 'on', '1')
  40
  41 # create a safe-to-call shell command from the array
  42 def safeCall(*args):
  43         res = ""
  44         for arg in args:
  45                 assert arg.find("'") < 0 # ' is not supported
  46                 if len(res): res += " "
  47                 res += "'"+arg+"'"
  48         return res
  49
  50 # Load a section-less config file: maps parameter names to space-separated lists of strings (with shell quotation)
  51 # Lines starting with spaces are continuation lines
  52 def loadConfigFile(file):
  53         # read config file
  54         linenr = 0
  55         with open(file) as file:
  56                 result = ConfigDict()
  57                 curKey = None
  58                 for line in file:
  59                         linenr += 1
  60                         isCont = len(line) and line[0].isspace() # remember if we were a continuation line
  61                         if isCont and curKey is None:
  62                                 raise Exception("Invalid config, line %d: Starting with continuation line" % linenr)
  63                         line = line.strip()
  64                         if not len(line) or line.startswith("#"): continue # skip empty and comment lines
  65                         try:
  66                                 if isCont:
  67                                         # continuation line
  68                                         result[curKey] += shlex.split(line)
  69                                 else:
  70                                         # option line
  71                                         pos = line.index("=") # will raise exception when substring is not found
  72                                         curKey = line[:pos].strip()
  73                                         result[curKey] = shlex.split(line[pos+1:]) # shlex.split also strips
  74                         except Exception:
  75                                 raise Exception("Invalid config, line %d: Error parsing line (quoting issue?)" % linenr)
  76         # add some convencience get functions
  77         return result
  78
  79 # representation of a build system
  80 class BuildSystem:
  81         def __init__(self, ruleMaker, buildDepends = [], binaryDepends = []):
  82                 self.ruleMaker = ruleMaker
  83                 self.buildDepends = buildDepends
  84                 self.binaryDepends = binaryDepends
  85
  86 # abstract representation of rules file
  87 class RulesFile:
  88         def __init__(self):
  89                 self.env = {}
  90                 self.dh = []
  91                 self.dhWith = set()
  92                 self.rules = OrderedDict()
  93
  94         def env2str(self):
  95                 res = ""
  96                 for name in self.env: # we rely on the name being sane (i.e., no special characters)
  97                         val = self.env[name]
  98                         assert val.find("'") < 0 # ' is not supported
  99                         if len(res): res += " "
 100                         res += name+"='"+val+"'"
 101                 return res
 102
 103         def write(self, f):
 104                 print >>f, "#!/usr/bin/make -f"
 105                 print >>f, ""
 106                 print >>f, ".PHONY: build" # there may be a directory called "build"
 107                 print >>f, ""
 108                 print >>f, "build %:" # need to mention "build" here again explicitly so PHONY takes effect
 109                 # write proper dh call
 110                 dh = self.dh
 111                 if self.dhWith:
 112                         dh.append('--with='+','.join(self.dhWith))
 113                 print >>f, '\t'+self.env2str()+' dh $@ '+safeCall(*dh)
 114                 # write remaining rules
 115                 for rule in self.rules:
 116                         print >>f, ""
 117                         print >>f, "override_dh_"+rule+":"
 118                         for line in self.rules[rule]:
 119                                 print >>f, "\t"+line
 120
 121 # rule-makers
 122 def cmakeRules(r, config):
 123         buildDir = config.getstr('buildDir', 'build')
 124         srcDir = os.getcwd()
 125         r.dh += ["--buildsystem=cmake", "--builddirectory="+buildDir] # dh parameters
 126         r.rules['auto_configure'] = [
 127                 safeCall("mkdir", "-p", buildDir),
 128                 safeCall("cd", buildDir) + " && " +
 129                   safeCall("cmake", srcDir, "-DCMAKE_INSTALL_PREFIX=/usr", *config.get('cmakeParameters', []))
 130         ]
 131         r.rules['auto_clean'] = [safeCall('rm', '-f', os.path.join(buildDir, 'CMakeCache.txt'))] # clean old cmake cache
 132
 133 def automakeRules(r, config):
 134         buildDir = config.getstr('buildDir', 'build')
 135         srcDir = os.getcwd()
 136         # "build" is what we are building *on*, and "host" is what we are building *for* (and GNU is weird...)
 137         # also, we don't really support cross-building... ;-) (to do so, we'd have to write shell code that checks whether BUILD_GNU_TYPE
 138         #  and HOST_GNU_TYPE are equal, and if they are not, add a --host parameter)
 139         r.dh += ["--buildsystem=autoconf", "--builddirectory="+config.getstr('buildDir')]
 140         r.rules['auto_configure'] = [
 141                 safeCall("mkdir", "-p", buildDir),
 142                 safeCall("cd", buildDir) + " && " +
 143                 'BUILD_TYPE=$$(dpkg-architecture -qDEB_BUILD_GNU_TYPE) && ' + # doing the expansion beforehand ensures that we cancel if it fails
 144                 'MULTIARCH=$$(dpkg-architecture -qDEB_HOST_MULTIARCH) && '+
 145                 safeCall(srcDir+"/configure") +
 146                         ' --build=$$BUILD_TYPE '+ # do the escape manually here so we can use the variables (there's no user-controlled string in here anyway)
 147                         '--prefix=/usr --includedir=/usr/include --mandir=/usr/share/man --infodir=/usr/share/info '+
 148                         '--libdir=/usr/lib/$$MULTIARCH --libexecdir=/usr/lib/$$MULTIARCH '+
 149                         safeCall('--docdir=/usr/share/doc/'+config['binaryName'], '--sysconfdir=/etc', '--localstatedir=/var', *config.get('automakeParameters', []))
 150         ]
 151         r.rules['auto_clean'] = [safeCall("rm", "-f", "--", buildDir+"/config.status")] # do not re-use old configuration
 152
 153 def makefileRules(r, config):
 154         r.dh += ["--buildsystem=makefile"]
 155         r.rules['auto_configure'] = []
 156
 157 def noneRules(r, config):
 158         r.dh += ["--buildsystem=makefile"] # makefile does the least possible harm
 159         r.rules['auto_configure'] = []
 160         r.rules['auto_build'] = []
 161         r.rules['auto_clean'] = []
 162         return r
 163
 164 # build systems
 165 buildSystems = {
 166         'cmake': BuildSystem(cmakeRules, ["cmake"]),
 167         'automake': BuildSystem(automakeRules),
 168         'makefile': BuildSystem(makefileRules),
 169         'none': BuildSystem(noneRules),
 170 }
 171
 172 # utility functions
 173 def commandInBuildEnv(config, command):
 174         schroot = config.getstr('schroot')
 175         if schroot is not None: command = ['schroot', '-c', schroot, '--'] + command
 176         return command
 177
 178 def getArchitecture(config):
 179         cmd = commandInBuildEnv(config, ['dpkg-architecture', '-qDEB_HOST_ARCH'])
 180         p = subprocess.Popen(cmd, stdout=subprocess.PIPE)
 181         res = p.communicate()[0] # get only stdout
 182         if p.returncode != 0: raise Exception("Querying dpkg for the architecture failed")
 183         return res[0:len(res)-1] # chop of the \n at the end
 184
 185 def writeDependency(f, name, list):
 186         if len(list):
 187                 print >>f, name+": "+', '.join(list)
 188
 189 def deleteDebianFolder():
 190         if os.path.islink('debian'):
 191                 target = os.readlink('debian')
 192                 if os.path.exists(target):
 193                         shutil.rmtree(target)
 194                 os.remove('debian')
 195         else:
 196                 shutil.rmtree('debian')
 197
 198 # actual work functions
 199 def createDebianFiles(config):
 200         if not isinstance(config, ConfigDict):
 201                 config = ConfigDict(config)
 202         sourceName = config.getstr('sourceName')
 203         binaryName = config.getstr('binaryName', sourceName+'-local')
 204         config['binaryName'] = binaryName # make it usable by build systems
 205         name = config.getstr('name', os.getenv('USER')) # os.getlogin() fails in minimal chroots
 206         email = config.getstr('email', os.getenv('USER')+'@'+os.uname()[1]) # user@hostname
 207         debDir = os.path.expanduser(config.getstr('debDir'))
 208         buildSystem = buildSystems[config.getstr('buildSystem')] # get the data immediately
 209         version = config.getstr('version') # version name excluding epoch (used for filenames)
 210         fullVersion = str(config.getint('epoch'))+':'+version if 'epoch' in config else version # version name including epoch
 211         dbgPackage = config.getbool('dbgPackage', False)
 212         parallelJobs = config.getint('parallelJobs', 2)
 213         packageArchitecture = config.getstr('architecture', 'any')
 214         withPython2 = config.getbool('withPython2', False)
 215         withSIP = config.getbool('withSIP', False)
 216         withAutoreconf = config.getbool('withAutoreconf', False)
 217         # add some build dependencies (a bit hacky adding it to the build system...)
 218         if withSIP:
 219                 withPython2 = True
 220                 buildSystem.buildDepends.append("python-sip")
 221                 buildSystem.binaryDepends.append("${sip:Depends}")
 222         if withPython2:
 223                 buildSystem.buildDepends.append("python")
 224                 buildSystem.binaryDepends.append("${python:Depends}")
 225         if withAutoreconf:
 226                 buildSystem.buildDepends.append("dh-autoreconf")
 227         # we return the list of files generated, so we need to know the architecture
 228         arch = getArchitecture(config)
 229         files = []
 230         # create folders
 231         if os.path.exists('debian') or os.path.islink('debian'): raise Exception('debian folder already exists?')
 232         if config.getbool('useTmp', True):
 233                 tempdir = tempfile.mkdtemp(prefix='auto-debuild-')
 234                 os.symlink(tempdir, 'debian')
 235         else:
 236                 os.mkdir('debian')
 237         if not os.path.exists(debDir): os.makedirs(debDir)
 238         # source format file
 239         os.mkdir('debian/source')
 240         with open('debian/source/format', 'w') as f:
 241                 print >>f, "3.0 (native)"
 242         # compat file
 243         with open('debian/compat', 'w') as f:
 244                 print >>f, "9"
 245         # copyright file
 246         with open('debian/copyright', 'w') as f:
 247                 print >>f, "Auto-generated by auto-debuild, not suited for distribution"
 248         # changelog file
 249         with open('debian/changelog', 'w') as f:
 250                 print >>f, sourceName,"("+fullVersion+")","UNRELEASED; urgency=low"
 251                 print >>f, ""
 252                 print >>f, "  * Auto-generated by auto-debuild"
 253                 print >>f, ""
 254                 print >>f, " --",name,"<"+email+">  "+time.strftime('%a, %d %b %Y %H:%M:%S %z')
 255         # control file
 256         with open('debian/control', 'w') as f:
 257                 # source package
 258                 print >>f, "Source:",sourceName
 259                 print >>f, "Section:",config.getstr('section', 'misc')
 260                 print >>f, "Priority: extra"
 261                 print >>f, "Maintainer: %s <%s>" % (name, email)
 262                 writeDependency(f, 'Build-Depends', ["debhelper (>= 9)"] + buildSystem.buildDepends + config.get('buildDepends', []))
 263                 print >>f, "Standards-Version: 3.9.3"
 264                 # main binary package
 265                 print >>f, ""
 266                 print >>f, "Package:",binaryName
 267                 print >>f, "Architecture:",packageArchitecture
 268                 if 'binaryMultiArch' in config:
 269                         print >>f, "Multi-Arch:",config.getstr('binaryMultiArch')
 270                 writeDependency(f, "Pre-Depends", ["${misc:Pre-Depends}"] + config.get('binaryPreDepends', []))
 271                 writeDependency(f, "Depends", ["${shlibs:Depends}", "${misc:Depends}"] + buildSystem.binaryDepends +
 272                         config.get('binaryDepends', []))
 273                 writeDependency(f, "Recommends", config.get('binaryRecommends', []))
 274                 writeDependency(f, "Provides", config.get('binaryProvides', [sourceName]))
 275                 writeDependency(f, "Conflicts", config.get('binaryConflicts', []))
 276                 writeDependency(f, "Breaks", config.get('binaryBreaks', []) + config.get('binaryBreaksReplaces', []))
 277                 writeDependency(f, "Replaces", config.get('binaryReplaces', []) + config.get('binaryBreaksReplaces', []))
 278                 print >>f, "Description:",sourceName,"(auto-debuild)"
 279                 print >>f, " Package auto-generated by auto-debuild."
 280                 files.append(os.path.join(debDir, "%s_%s_%s.deb" % (binaryName, version, arch)))
 281                 # debug package
 282                 if dbgPackage:
 283                         print >>f, ""
 284                         print >>f, "Package:",binaryName+"-dbg"
 285                         print >>f, "Section: debug"
 286                         print >>f, "Priority: extra"
 287                         print >>f, "Architecture:",packageArchitecture
 288                         writeDependency(f, "Depends", ["${misc:Depends}", binaryName+" (= ${binary:Version})"])
 289                         print >>f, "Description:",sourceName,"debug smbols (auto-debuild)"
 290                         print >>f, " Package containing debug symbols for "+sourceName+", auto-generated by auto-debuild."
 291                         files.append(os.path.join(debDir, "%s-dbg_%s_%s.deb" % (binaryName, version, arch)))
 292                 # shim packages
 293                 for shim in config.get('binaryShims', []):
 294                         print >>f, ""
 295                         print >>f, "Package:",shim
 296                         print >>f, "Section:",config.getstr('section', 'misc')
 297                         print >>f, "Priority: extra"
 298                         print >>f, "Architecture:",packageArchitecture
 299                         writeDependency(f, "Depends", ["${misc:Depends}", binaryName+" (= ${binary:Version})"])
 300                         print >>f, "Description:",sourceName,"shim for",shim,"(auto-debuild)"
 301                         print >>f, " Package pretending to be "+shim+", auto-generated by auto-debuild."
 302                         files.append(os.path.join(debDir, "%s_%s_%s.deb" % (shim, version, arch)))
 303         # install file
 304         with open('debian/'+binaryName+'.install', 'w') as f:
 305                 for line in config.get('binaryInstallFiles', []):
 306                         if line.startswith('/'): # a file from within the package, not from the source tree
 307                                 line = 'debian/'+binaryName+line
 308                         print >>f, line
 309         # maintainer scripts for alternatives
 310         if 'alternatives' in config:
 311                 with open('debian/'+binaryName+'.postinst', 'w') as f:
 312                         print >>f, "#!/bin/sh"
 313                         print >>f, "set -e"
 314                         print >>f, 'if [ "$1" = "configure" ]; then'
 315                         for alternative in config.get('alternatives'):
 316                                 alternative = shlex.split(alternative)
 317                                 print >>f, safeCall('update-alternatives', '--install', alternative[0], alternative[1], alternative[2], alternative[3])
 318                         print >>f, 'fi'
 319                         print >>f, ''
 320                         print >>f, '#DEBHELPER#'
 321                         print >>f, ''
 322                         print >>f, 'exit 0'
 323                 with open('debian/'+binaryName+'.prerm', 'w') as f:
 324                         print >>f, "#!/bin/sh"
 325                         print >>f, "set -e"
 326                         print >>f, 'if [ "$1" = "remove" ]; then'
 327                         for alternative in config.get('alternatives'):
 328                                 alternative = shlex.split(alternative)
 329                                 print >>f, safeCall('update-alternatives', '--remove', alternative[1], alternative[2])
 330                         print >>f, 'fi'
 331                         print >>f, ''
 332                         print >>f, '#DEBHELPER#'
 333                         print >>f, ''
 334                         print >>f, 'exit 0'
 335         # rules file: build system specific
 336         with open('debian/rules', 'w') as f:
 337                 # pre-fill rule file with our global defaults
 338                 r = RulesFile()
 339                 r.rules['auto_test'] = []
 340                 r.rules['auto_install'] = [safeCall('dh_auto_install', '--destdir=debian/'+binaryName)] # install everything into the binary package
 341                 r.rules['auto_configure'] = ['dh_auto_configure'] # make sure there is an override (we may just append to it later)
 342                 # patch rule file for build system: may only touch auto_* rules and the dh options
 343                 buildSystem.ruleMaker(r, config)
 344                 # global rules
 345                 r.env["DEB_BUILD_OPTIONS"] = 'parallel='+str(parallelJobs)
 346                 if not dbgPackage:
 347                         # disable debug information
 348                         r.env["DEB_CFLAGS_APPEND"] = '-g0'
 349                         r.env["DEB_CXXFLAGS_APPEND"] = '-g0'
 350                 r.dh += ['--parallel']
 351                 if withPython2:
 352                         r.dhWith.add('python2')
 353                         r.rules['python2'] = ['dh_python2 --no-guessing-versions --no-shebang-rewrite']
 354                         if withSIP:
 355                                 r.rules['python2'].append(safeCall('dh_sip', '-p'+binaryName))
 356                 if withAutoreconf:
 357                         r.dhWith.add('autoreconf')
 358                 r.rules['builddeb'] = [safeCall('dh_builddeb', "--destdir="+debDir)] # passing this gobally to dh results in weird problems (like stuff being installed there, and not in the package...)
 359                 # installation rule
 360                 if 'binarySkipFiles' in config:
 361                         r.rules['auto_install'].append(safeCall('cd', 'debian/'+binaryName) + " && " + safeCall('rm', *config.get('binarySkipFiles')))
 362                 # debug packages
 363                 if dbgPackage:
 364                         r.rules['strip'] = [safeCall('dh_strip', '--dbg-package='+binaryName+"-dbg")] # put debug files in appropriate package
 365                 # make the doc folder of the other packages a symlink (dbg, shims)
 366                 r.rules['installdocs'] = [safeCall('dh_installdocs', '--link-doc='+binaryName)]
 367                 # wait after configuration?
 368                 if config.getbool('waitAfterConfig', False):
 369                         r.rules['auto_configure'].append("@"+safeCall('read', '-p', 'Configuration done. Hit "Enter" to build the package. ', 'DUMMY_VAR')) # if we run in dash, we need to tell it which variable to use for the result...
 370                 # dump it to a file
 371                 r.write(f)
 372         mode = os.stat('debian/rules').st_mode
 373         os.chmod('debian/rules', mode | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
 374         # return list of files that will be created
 375         return files
 376
 377 def buildDebianPackage(config):
 378         if not isinstance(config, ConfigDict):
 379                 config = ConfigDict(config)
 380         commands = ['dpkg-checkbuilddeps', 'debian/rules clean', 'debian/rules build', 'fakeroot debian/rules binary']
 381         command = ['bash', '-c', ' && '.join(commands)] # make it all one command, so we don't have to open and close the chroot too often
 382         subprocess.check_call(commandInBuildEnv(config, command))
 383         deleteDebianFolder() # cleanup: the debian folder only contains what we just created
 384
 385 ###################################################################
 386 # if we are called directly as script
 387 if __name__ == "__main__":
 388         try:
 389                 # read command-line arguments
 390                 parser = argparse.ArgumentParser(description='Automatic Generation of Debian Packages')
 391                 parser.add_argument("-w", "--wait-after-config",
 392                                                         action="store_true", dest="wait_after_config",
 393                                                         help="Wait for user confirmation after configuration is finished")
 394                 args = parser.parse_args()
 395                 # get config
 396                 config = loadConfigFile('auto-debuild.conf')
 397                 config['waitAfterConfig'] = args.wait_after_config
 398                 # generate debian files
 399                 if os.path.exists('debian') or os.path.islink('debian'):
 400                         if raw_input("A debian folder already exists, do you want to remove it and whatever it links to (y/N)? ").lower() != "y":
 401                                 sys.exit(1)
 402                         deleteDebianFolder()
 403                 files = createDebianFiles(config)
 404                 # check if a file is overwritten
 405                 for file in files:
 406                         if os.path.exists(file):
 407                                 if raw_input("Do you want to overwrite %s (y/N)? " % file).lower() != "y":
 408                                         sys.exit(1)
 409                 # run compilation
 410                 buildDebianPackage(config)
 411                 # install files
 412                 print "Installing created deb files..."
 413                 subprocess.check_call(['sudo', 'dpkg', '--install'] + files)
 414         except (subprocess.CalledProcessError, KeyboardInterrupt) as e: # for some exceptions, a stackrace is usually pointless
 415                 print >> sys.stderr
 416                 print >> sys.stderr
 417                 if isinstance(e, KeyboardInterrupt): # str(e) would be the empty string
 418                         print >> sys.stderr, "Interruped by user"
 419                 else:
 420                         print >> sys.stderr, "Error during package creation: %s" % str(e)
 421                 print >> sys.stderr
 422                 sys.exit(1)