cdist

   1 #!/usr/bin/python
   2
   3 #    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   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., 675 Mass Ave, Cambridge, MA 02139, USA.
  18
  19 #
  20 # Configuration
  21 #
  22
  23 LINUX_DIR_IN_CHROOT = '/home/carl'
  24 LINUX_CHROOT_PREFIX = '/home/carl/Environments'
  25 WINDOWS_ENVIRONMENT_PREFIX = '/home/carl/Environments/windows'
  26 GIT_PREFIX = 'ssh://houllier/home/carl/git'
  27 OSX_BUILD_HOST = 'carl@192.168.1.202'
  28 OSX_DIR_IN_HOST = '/Users/carl/cdist'
  29 OSX_ENVIRONMENT_PREFIX = '/Users/carl/Environments/osx'
  30 OSX_SDK_PREFIX = '/Users/carl/SDK'
  31 OSX_SDK = '10.6'
  32
  33 import os
  34 import sys
  35 import shutil
  36 import glob
  37 import tempfile
  38 import argparse
  39 import datetime
  40 import subprocess
  41 import re
  42
  43 #
  44 # Utility bits
  45 #
  46
  47 def log(m):
  48     if not args.quiet:
  49         print '\x1b[33m* %s\x1b[0m' % m
  50
  51 def error(e):
  52     print '\x1b[31mError: %s\x1b[0m' % e
  53     sys.exit(1)
  54
  55 def copytree(a, b):
  56     log('copy %s -> %s' % (a, b))
  57     shutil.copytree(a, b)
  58
  59 def copyfile(a, b):
  60     log('copy %s -> %s' % (a, b))
  61     shutil.copyfile(a, b)
  62
  63 def rmdir(a):
  64     log('remove %s' % a)
  65     os.rmdir(a)
  66
  67 def rmtree(a):
  68     log('remove %s' % a)
  69     shutil.rmtree(a, ignore_errors=True)
  70
  71 def command(c, can_fail=False):
  72     log(c)
  73     r = os.system(c)
  74     if (r >> 8) and not can_fail:
  75         error('command %s failed' % c)
  76
  77 def command_and_read(c):
  78     log(c)
  79     p = subprocess.Popen(c.split(), stdout=subprocess.PIPE)
  80     f = os.fdopen(os.dup(p.stdout.fileno()))
  81     return f
  82
  83
  84 #
  85 # Version
  86 #
  87
  88 class Version:
  89     def __init__(self, s):
  90         self.pre = False
  91         self.beta = None
  92
  93         if s.startswith("'"):
  94             s = s[1:]
  95         if s.endswith("'"):
  96             s = s[0:-1]
  97
  98         if s.endswith('pre'):
  99             s = s[0:-3]
 100             self.pre = True
 101
 102         b = s.find("beta")
 103         if b != -1:
 104             self.beta = int(s[b+4:])
 105             s = s[0:b]
 106
 107         p = s.split('.')
 108         self.major = int(p[0])
 109         self.minor = int(p[1])
 110
 111     def bump(self):
 112         self.minor += 1
 113         self.pre = False
 114         self.beta = None
 115
 116     def to_pre(self):
 117         self.pre = True
 118         self.beta = None
 119
 120     def bump_and_to_pre(self):
 121         self.bump()
 122         self.pre = True
 123         self.beta = None
 124
 125     def to_release(self):
 126         self.pre = False
 127         self.beta = None
 128
 129     def bump_beta(self):
 130         if self.pre:
 131             self.pre = False
 132             self.beta = 1
 133         elif self.beta is not None:
 134             self.beta += 1
 135         elif self.beta is None:
 136             self.beta = 1
 137
 138     def __str__(self):
 139         s = '%d.%02d' % (self.major, self.minor)
 140         if self.beta is not None:
 141             s += 'beta%d' % self.beta
 142         elif self.pre:
 143             s += 'pre'
 144
 145         return s
 146
 147 #
 148 # Targets
 149 #
 150
 151 class Target(object):
 152     def __init__(self, platform, parallel):
 153         self.platform = platform
 154         self.parallel = parallel
 155         # Environment variables that we will use when we call cscripts
 156         self.variables = {}
 157         self.debug = False
 158
 159     def build_dependencies(self, project):
 160         cwd = os.getcwd()
 161         if 'dependencies' in project.cscript:
 162             print project.cscript['dependencies'](self)
 163             for d in project.cscript['dependencies'](self):
 164                 log('Building dependency %s %s of %s' % (d[0], d[1], project.name))
 165                 dep = Project(d[0], '.', d[1])
 166                 dep.checkout(self)
 167                 self.build_dependencies(dep)
 168                 self.build(dep)
 169         os.chdir(cwd)
 170
 171     def build(self, project):
 172         project.cscript['build'](self)
 173
 174     def package(self, project):
 175         project.checkout(self)
 176         self.build_dependencies(project)
 177         project.cscript['build'](self)
 178         return project.cscript['package'](self, project.version)
 179
 180     def set(self, a, b):
 181         self.variables[a] = b
 182
 183     def get(self, a):
 184         return self.variables[a]
 185
 186     def variables_string(self, escaped_quotes=False):
 187         e = ''
 188         for k, v in self.variables.iteritems():
 189             if escaped_quotes:
 190                 v = v.replace('"', '\\"')
 191             e += '%s=%s ' % (k, v)
 192         return e
 193
 194     def cleanup(self):
 195         pass
 196
 197 #
 198 # Windows
 199 #
 200
 201 class WindowsTarget(Target):
 202     def __init__(self, bits, directory = None):
 203         super(WindowsTarget, self).__init__('windows', 2)
 204         self.bits = bits
 205         if directory is None:
 206             self.directory = tempfile.mkdtemp()
 207             self.rmdir = True
 208         else:
 209             self.directory = directory
 210             self.rmdir = False
 211
 212         self.windows_prefix = '%s/%d' % (WINDOWS_ENVIRONMENT_PREFIX, self.bits)
 213         if not os.path.exists(self.windows_prefix):
 214             error('windows prefix %s does not exist' % target.windows_prefix)
 215
 216         if self.bits == 32:
 217             mingw_name = 'i686'
 218         else:
 219             mingw_name = 'x86_64'
 220
 221         mingw_path = '/mingw/%d/bin' % self.bits
 222         mingw_prefixes = ['/mingw/%d' % self.bits, '/mingw/%d/%s-w64-mingw32' % (bits, mingw_name)]
 223
 224         self.set('PKG_CONFIG_LIBDIR', '%s/lib/pkgconfig' % self.windows_prefix)
 225         self.set('PKG_CONFIG_PATH', '%s/lib/pkgconfig' % self.work_dir_cscript())
 226         self.set('PATH', '%s/bin:%s:%s' % (self.windows_prefix, mingw_path, os.environ['PATH']))
 227         self.set('CC', '%s-w64-mingw32-gcc' % mingw_name)
 228         self.set('CXX', '%s-w64-mingw32-g++' % mingw_name)
 229         self.set('LD', '%s-w64-mingw32-ld' % mingw_name)
 230         self.set('RANLIB', '%s-w64-mingw32-ranlib' % mingw_name)
 231         self.set('WINRC', '%s-w64-mingw32-windres' % mingw_name)
 232         cxx = '-I%s/include -I%s/include' % (self.windows_prefix, self.work_dir_cscript())
 233         link = '-L%s/lib -L%s/lib' % (self.windows_prefix, self.work_dir_cscript())
 234         for p in mingw_prefixes:
 235             cxx += ' -I%s/include' % p
 236             link += ' -L%s/lib' % p
 237         self.set('CXXFLAGS', '"%s"' % cxx)
 238         self.set('LINKFLAGS', '"%s"' % link)
 239
 240     def work_dir_cdist(self):
 241         return '%s/%d' % (self.directory, self.bits)
 242
 243     def work_dir_cscript(self):
 244         return '%s/%d' % (self.directory, self.bits)
 245
 246     def command(self, c):
 247         log('host -> %s' % c)
 248         command('%s %s' % (self.variables_string(), c))
 249
 250     def cleanup(self):
 251         if self.rmdir:
 252             rmtree(self.directory)
 253
 254 #
 255 # Linux
 256 #
 257
 258 class LinuxTarget(Target):
 259     def __init__(self, distro, version, bits):
 260         super(LinuxTarget, self).__init__('linux', 2)
 261         self.distro = distro
 262         self.version = version
 263         self.bits = bits
 264         self.chroot = '%s-%s-%d' % (self.distro, self.version, self.bits)
 265
 266         for g in glob.glob('%s/*' % self.work_dir_cdist()):
 267             rmtree(g)
 268
 269         self.set('CXXFLAGS', '-I%s/include' % self.work_dir_cscript())
 270         self.set('LINKFLAGS', '-L%s/lib' % self.work_dir_cscript())
 271         self.set('PKG_CONFIG_PATH', '%s/lib/pkgconfig' % self.work_dir_cscript())
 272
 273     def work_dir_cdist(self):
 274         return '%s/%s%s' % (LINUX_CHROOT_PREFIX, self.chroot, LINUX_DIR_IN_CHROOT)
 275
 276     def work_dir_cscript(self):
 277         return LINUX_DIR_IN_CHROOT
 278
 279     def command(self, c):
 280         # Work out the cwd for the chrooted command
 281         cwd = os.getcwd()
 282         prefix = '%s/%s' % (LINUX_CHROOT_PREFIX, self.chroot)
 283         assert(cwd.startswith(prefix))
 284         cwd = cwd[len(prefix):]
 285
 286         log('schroot [%s] -> %s' % (cwd, c))
 287         command('%s schroot -c %s -d %s -p -- %s' % (self.variables_string(), self.chroot, cwd, c))
 288
 289 #
 290 # OS X
 291 #
 292
 293 class OSXTarget(Target):
 294     def __init__(self):
 295         super(OSXTarget, self).__init__('osx', 4)
 296
 297         self.host_mount_dir = tempfile.mkdtemp()
 298
 299         # Mount the remote host on host_mount_dir
 300         command('sshfs %s:%s %s' % (OSX_BUILD_HOST, OSX_DIR_IN_HOST, self.host_mount_dir))
 301         for g in glob.glob('%s/*' % self.host_mount_dir):
 302             rmtree(g)
 303
 304     def command(self, c):
 305         # Work out the cwd for the chrooted command
 306         cwd = os.getcwd()
 307         print cwd
 308         print self.host_mount_dir
 309         assert(cwd.startswith(self.host_mount_dir))
 310         cwd = cwd[len(self.host_mount_dir):]
 311
 312         log('ssh [%s] -> %s' % (cwd, c))
 313         command('ssh %s -- "cd %s%s; %s %s"' % (OSX_BUILD_HOST, OSX_DIR_IN_HOST, cwd, self.variables_string(True), c))
 314
 315     def cleanup(self):
 316         os.chdir('/')
 317         command('fusermount -u %s' % self.host_mount_dir)
 318         rmdir(self.host_mount_dir)
 319
 320
 321
 322 class OSXSingleTarget(OSXTarget):
 323     def __init__(self, bits):
 324         super(OSXSingleTarget, self).__init__()
 325         self.bits = bits
 326
 327         if bits == 32:
 328             arch = 'i386'
 329         else:
 330             arch = 'x86_64'
 331
 332         flags = '-isysroot %s/MacOSX%s.sdk -arch %s' % (OSX_SDK_PREFIX, OSX_SDK, arch)
 333         enviro = '%s/%d' % (OSX_ENVIRONMENT_PREFIX, bits)
 334
 335         # Environment variables
 336         self.set('CFLAGS', '"-I%s/include -I%s/include %s"' % (self.work_dir_cscript(), enviro, flags))
 337         self.set('CXXFLAGS', '"-I%s/include -I%s/include %s"' % (self.work_dir_cscript(), enviro, flags))
 338         self.set('LDFLAGS', '"-L%s/lib -L%s/lib %s"' % (self.work_dir_cscript(), enviro, flags))
 339         self.set('LINKFLAGS', '"-L%s/lib -L%s/lib %s"' % (self.work_dir_cscript(), enviro, flags))
 340         self.set('PKG_CONFIG_PATH', '%s/lib/pkgconfig:%s/lib/pkgconfig:/usr/lib/pkgconfig' % (self.work_dir_cscript(), enviro))
 341         self.set('PATH', '$PATH:/usr/bin:/sbin:/usr/local/bin:%s/bin' % enviro)
 342         self.set('MACOSX_DEPLOYMENT_TARGET', OSX_SDK)
 343
 344     def work_dir_cdist(self):
 345         return '%s/%d' % (self.host_mount_dir, self.bits)
 346
 347     def work_dir_cscript(self):
 348         return '%s/%d' % (OSX_DIR_IN_HOST, self.bits)
 349
 350     def package(self, project):
 351         error('cannot package non-universal OS X versions')
 352
 353
 354 class OSXUniversalTarget(OSXTarget):
 355     def __init__(self):
 356         super(OSXUniversalTarget, self).__init__()
 357         self.parts = []
 358         self.parts.append(OSXSingleTarget(32))
 359         self.parts.append(OSXSingleTarget(64))
 360
 361     def package(self, project):
 362         for p in self.parts:
 363             project.checkout(p)
 364             p.build_dependencies(project)
 365             p.build(project)
 366
 367         return project.cscript['package'](self.parts[-1], project.version)
 368
 369
 370 #
 371 # Source
 372 #
 373
 374 class SourceTarget(Target):
 375     def __init__(self):
 376         super(SourceTarget, self).__init__('source', 2)
 377         self.directory = tempfile.mkdtemp()
 378
 379     def work_dir_cdist(self):
 380         return self.directory
 381
 382     def work_dir_cscript(self):
 383         return self.directory
 384
 385     def command(self, c):
 386         log('host -> %s' % c)
 387         command('%s %s' % (self.variables_string(), c))
 388
 389     def cleanup(self):
 390         rmtree(self.directory)
 391
 392     def package(self, project):
 393         project.checkout(self)
 394         command('./waf dist')
 395         if project.directory != '.':
 396             return os.path.abspath('%s-%s.tar.bz2' % (project.directory, project.version))
 397         return os.path.abspath('%s-%s.tar.bz2' % (project.name, project.version))
 398
 399
 400 # @param s Target string:
 401 #       windows-{32,64}
 402 #    or ubuntu-version-{32,64}
 403 #    or debian-version-{32,64}
 404 #    or osx-{32,64}
 405 #    or source
 406 # @param debug True to build with debugging symbols (where possible)
 407 def target_factory(s, debug=False):
 408     target = None
 409     if s.startswith('windows-'):
 410         target = WindowsTarget(int(s.split('-')[1]))
 411     elif s.startswith('ubuntu-') or s.startswith('debian-'):
 412         p = s.split('-')
 413         target = LinuxTarget(p[0], p[1], int(p[2]))
 414     elif s.startswith('osx-'):
 415         target = OSXSingleTarget(int(s.split('-')[1]))
 416     elif s == 'osx':
 417         if args.command == 'build':
 418             target = OSXSingleTarget(64)
 419         else:
 420             target = OSXUniversalTarget()
 421     elif s == 'source':
 422         target = SourceTarget()
 423
 424     if target is not None:
 425         target.debug = debug
 426
 427     return target
 428
 429
 430 #
 431 # Project
 432 #
 433
 434 class Project(object):
 435     def __init__(self, name, directory, specifier=None):
 436         self.name = name
 437         self.directory = directory
 438         self.version = None
 439         self.specifier = specifier
 440         if self.specifier is None:
 441             self.specifier = 'master'
 442
 443     def checkout(self, target):
 444         flags = ''
 445         redirect = ''
 446         if args.quiet:
 447             flags = '-q'
 448             redirect = '>/dev/null'
 449         command('git clone --depth 0 %s %s/%s.git %s/src/%s' % (flags, GIT_PREFIX, self.name, target.work_dir_cdist(), self.name))
 450         os.chdir('%s/src/%s' % (target.work_dir_cdist(), self.name))
 451         command('git checkout %s %s %s' % (flags, self.specifier, redirect))
 452         command('git submodule init')
 453         command('git submodule update')
 454         os.chdir(self.directory)
 455
 456         proj = '%s/src/%s/%s' % (target.work_dir_cdist(), self.name, self.directory)
 457
 458         self.read_cscript('%s/cscript' % proj)
 459
 460         if os.path.exists('%s/wscript' % proj):
 461             f = open('%s/wscript' % proj, 'r')
 462             version = None
 463             while 1:
 464                 l = f.readline()
 465                 if l == '':
 466                     break
 467
 468                 s = l.split()
 469                 if len(s) == 3 and s[0] == "VERSION":
 470                     self.version = Version(s[2])
 471
 472             f.close()
 473
 474     def read_cscript(self, s):
 475         self.cscript = {}
 476         execfile(s, self.cscript)
 477
 478 def set_version_in_wscript(version):
 479     f = open('wscript', 'rw')
 480     o = open('wscript.tmp', 'w')
 481     while 1:
 482         l = f.readline()
 483         if l == '':
 484             break
 485
 486         s = l.split()
 487         if len(s) == 3 and s[0] == "VERSION":
 488             print "Writing %s" % version
 489             print >>o,"VERSION = '%s'" % version
 490         else:
 491             print >>o,l,
 492     f.close()
 493     o.close()
 494
 495     os.rename('wscript.tmp', 'wscript')
 496
 497 def append_version_to_changelog(version):
 498     try:
 499         f = open('ChangeLog', 'r')
 500     except:
 501         log('Could not open ChangeLog')
 502         return
 503
 504     c = f.read()
 505     f.close()
 506
 507     f = open('ChangeLog', 'w')
 508     now = datetime.datetime.now()
 509     f.write('%d-%02d-%02d  Carl Hetherington  <cth@carlh.net>\n\n\t* Version %s released.\n\n' % (now.year, now.month, now.day, version))
 510     f.write(c)
 511
 512 def append_version_to_debian_changelog(version):
 513     if not os.path.exists('debian'):
 514         log('Could not find debian directory')
 515         return
 516
 517     command('dch -b -v %s-1 "New upstream release."' % version)
 518
 519 #
 520 # Command-line parser
 521 #
 522
 523 parser = argparse.ArgumentParser()
 524 parser.add_argument('command')
 525 parser.add_argument('-p', '--project', help='project name', required=True)
 526 parser.add_argument('-d', '--directory', help='directory within project repo', default='.')
 527 parser.add_argument('--beta', help='beta release', action='store_true')
 528 parser.add_argument('--full', help='full release', action='store_true')
 529 parser.add_argument('-c', '--checkout', help='string to pass to git for checkout')
 530 parser.add_argument('-o', '--output', help='output directory', default='.')
 531 parser.add_argument('-q', '--quiet', help='be quiet', action='store_true')
 532 parser.add_argument('-t', '--target', help='target')
 533 parser.add_argument('-k', '--keep', help='keep working tree', action='store_true')
 534 parser.add_argument('--debug', help='build with debugging symbols where possible', action='store_true')
 535 args = parser.parse_args()
 536
 537 args.output = os.path.abspath(args.output)
 538
 539 if args.project is None:
 540     error('you must specify -p or --project')
 541
 542 project = Project(args.project, args.directory, args.checkout)
 543
 544 if args.command == 'build':
 545     if args.target is None:
 546         error('you must specify -t or --target')
 547
 548     target = target_factory(args.target, args.debug)
 549     project.checkout(target)
 550     target.build_dependencies(project)
 551     target.build(project)
 552     target.cleanup()
 553
 554 elif args.command == 'package':
 555     if args.target is None:
 556         error('you must specify -t or --target')
 557
 558     target = target_factory(args.target)
 559
 560     packages = target.package(project)
 561     if hasattr(packages, 'strip') or (not hasattr(packages, '__getitem__') and not hasattr(packages, '__iter__')):
 562         packages = [packages]
 563
 564     if target.platform == 'linux':
 565         out = '%s/%s-%d' % (args.output, target.version, target.bits)
 566         try:
 567             os.makedirs(out)
 568         except:
 569             pass
 570         for p in packages:
 571             copyfile(p, '%s/%s' % (out, os.path.basename(p)))
 572     else:
 573         for p in packages:
 574             copyfile(p, '%s/%s' % (args.output, os.path.basename(p)))
 575
 576     target.cleanup()
 577
 578 elif args.command == 'release':
 579     if args.full is False and args.beta is False:
 580         error('you must specify --full or --beta')
 581
 582     target = SourceTarget()
 583     project.checkout(target)
 584
 585     version = project.version
 586     if args.full:
 587         version.to_release()
 588     else:
 589         version.bump_beta()
 590
 591     set_version_in_wscript(version)
 592     append_version_to_changelog(version)
 593     append_version_to_debian_changelog(version)
 594
 595     command('git commit -a -m "Bump version"')
 596     command('git tag -m "v%s" v%s' % (version, version))
 597
 598     if args.full:
 599         version.bump_and_to_pre()
 600         set_version_in_wscript(version)
 601         command('git commit -a -m "Bump version"')
 602
 603     command('git push')
 604     command('git push --tags')
 605
 606     target.cleanup()
 607
 608 elif args.command == 'pot':
 609     target = SourceTarget()
 610     project.checkout(target)
 611
 612     pots = project.cscript['make_pot'](target)
 613     for p in pots:
 614         copyfile(p, '%s/%s' % (args.output, os.path.basename(p)))
 615
 616     target.cleanup()
 617
 618 elif args.command == 'changelog':
 619     target = SourceTarget()
 620     project.checkout(target)
 621
 622     text = open('ChangeLog', 'r')
 623     html = open('%s/changelog.html' % args.output, 'w')
 624     versions = 8
 625
 626     last = None
 627     changes = []
 628
 629     while 1:
 630         l = text.readline()
 631         if l == '':
 632             break
 633
 634         if len(l) > 0 and l[0] == "\t":
 635             s = l.split()
 636             if len(s) == 4 and s[1] == "Version" and s[3] == "released.":
 637                 if not "beta" in s[2]:
 638                     if last is not None and len(changes) > 0:
 639                         print >>html,"<h2>Changes between version %s and %s</h2>" % (s[2], last)
 640                         print >>html,"<ul>"
 641                         for c in changes:
 642                             print >>html,"<li>%s" % c
 643                         print >>html,"</ul>"
 644                     last = s[2]
 645                     changes = []
 646                     versions -= 1
 647                     if versions < 0:
 648                         break
 649             else:
 650                 c = l.strip()
 651                 if len(c) > 0:
 652                     if c[0] == '*':
 653                         changes.append(c[2:])
 654                     else:
 655                         changes[-1] += " " + c
 656
 657     target.cleanup()
 658
 659 elif args.command == 'manual':
 660     target = SourceTarget()
 661     project.checkout(target)
 662
 663     dirs = project.cscript['make_manual'](target)
 664     for d in dirs:
 665         copytree(d, '%s/%s' % (args.output, os.path.basename(d)))
 666
 667     target.cleanup()
 668
 669 elif args.command == 'doxygen':
 670     target = SourceTarget()
 671     project.checkout(target)
 672
 673     dirs = project.cscript['make_doxygen'](target)
 674     if hasattr(dirs, 'strip') or (not hasattr(dirs, '__getitem__') and not hasattr(dirs, '__iter__')):
 675         dirs = [dirs]
 676
 677     for d in dirs:
 678         copytree(d, '%s/%s' % (args.output, 'doc'))
 679
 680     target.cleanup()
 681
 682 elif args.command == 'latest':
 683     target = SourceTarget()
 684     project.checkout(target)
 685
 686     f = command_and_read('git log --tags --simplify-by-decoration --pretty="%d"')
 687     t = f.readline()
 688     m = re.compile(".*\((.*)\).*").match(t)
 689     latest = None
 690     if m:
 691         tags = m.group(1).split(', ')
 692         for t in tags:
 693             if len(t) > 0 and t[0] == 'v':
 694                 latest = t[1:]
 695
 696     print latest
 697     target.cleanup()
 698
 699 elif args.command == 'test':
 700     if args.target is None:
 701         error('you must specify -t or --target')
 702
 703     target = target_factory(args.target)
 704     project.read_cscript('cscript')
 705     target.build(project)
 706
 707 else:
 708     error('invalid command %s' % args.command)