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