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/10.8'
  34
  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
 256         # Mount the remote host on host_mount_dir
 257         command('sshfs %s:%s %s' % (self.host, self.dir_on_host, self.host_mount_dir))
 258         for g in glob.glob('%s/*' % self.host_mount_dir):
 259             rmtree(g)
 260
 261         # Environment variables
 262         self.set('CXXFLAGS', '"-I%s/include -I%s/include"' % (self.dir_on_host, OSX_ENVIRONMENT_PREFIX))
 263         self.set('LINKFLAGS', '"-L%s/lib -L%s/lib"' % (self.dir_on_host, OSX_ENVIRONMENT_PREFIX))
 264         self.set('PKG_CONFIG_PATH', '%s/lib/pkgconfig:%s/lib/pkgconfig' % (self.dir_on_host, OSX_ENVIRONMENT_PREFIX))
 265         self.set('PATH', '$PATH:/usr/local/bin:%s/bin' % OSX_ENVIRONMENT_PREFIX)
 266
 267     def work_dir_cdist(self):
 268         return self.host_mount_dir
 269
 270     def work_dir_cscript(self):
 271         return self.dir_on_host
 272
 273     def command(self, c):
 274         # Work out the cwd for the chrooted command
 275         cwd = os.getcwd()
 276         assert(cwd.startswith(self.host_mount_dir))
 277         cwd = cwd[len(self.host_mount_dir):]
 278
 279         log('ssh [%s] -> %s' % (cwd, c))
 280         command('ssh %s -- "PATH=/usr/bin:/sbin cd %s%s; %s %s"' % (self.host, self.dir_on_host, cwd, self.variables_string(True), c))
 281
 282     def cleanup(self):
 283         os.chdir('/')
 284         command('fusermount -u %s' % self.host_mount_dir)
 285         rmdir(self.host_mount_dir)
 286
 287 #
 288 # HostEnvironment
 289 #
 290
 291 class HostEnvironment(Environment):
 292     def __init__(self, directory=None):
 293         super(HostEnvironment, self).__init__()
 294         if directory is None:
 295             self.directory = tempfile.mkdtemp()
 296             self.rmdir = True
 297         else:
 298             self.directory = directory
 299             self.rmdir = False
 300
 301     def work_dir_cdist(self):
 302         return self.directory
 303
 304     def work_dir_cscript(self):
 305         return self.directory
 306
 307     def command(self, c):
 308         log('host -> %s' % c)
 309         command('%s %s' % (self.variables_string(), c))
 310
 311     def cleanup(self):
 312         if self.rmdir:
 313             rmtree(self.directory)
 314
 315
 316 def prepare_for_windows(env, bits):
 317     env.windows_prefix = '%s/%d' % (WINDOWS_ENVIRONMENT_PREFIX, bits)
 318     if not os.path.exists(env.windows_prefix):
 319         error('windows prefix %s does not exist' % env.windows_prefix)
 320
 321     if bits == 32:
 322         mingw_name = 'i686'
 323     else:
 324         mingw_name = 'x86_64'
 325
 326     mingw_path = '/mingw/%d/bin' % bits
 327     mingw_prefixes = ['/mingw/%d' % bits, '/mingw/%d/%s-w64-mingw32' % (bits, mingw_name)]
 328
 329     env.set('PKG_CONFIG_LIBDIR', '%s/lib/pkgconfig' % env.windows_prefix)
 330     env.set('PKG_CONFIG_PATH', '%s/lib/pkgconfig' % env.work_dir_cscript())
 331     env.set('PATH', '%s/bin:%s:%s' % (env.windows_prefix, mingw_path, os.environ['PATH']))
 332     env.set('CC', '%s-w64-mingw32-gcc' % mingw_name)
 333     env.set('CXX', '%s-w64-mingw32-g++' % mingw_name)
 334     env.set('LD', '%s-w64-mingw32-ld' % mingw_name)
 335     env.set('RANLIB', '%s-w64-mingw32-ranlib' % mingw_name)
 336     env.set('WINRC', '%s-w64-mingw32-windres' % mingw_name)
 337     cxx = '-I%s/include -I%s/include' % (env.windows_prefix, env.work_dir_cscript())
 338     link = '-L%s/lib -L%s/lib' % (env.windows_prefix, env.work_dir_cscript())
 339     for p in mingw_prefixes:
 340         cxx += ' -I%s/include' % p
 341         link += ' -L%s/lib' % p
 342     env.set('CXXFLAGS', '"%s"' % cxx)
 343     env.set('LINKFLAGS', '"%s"' % link)
 344
 345
 346 #
 347 # Target
 348 #
 349
 350 class Target:
 351     def __init__(self, name):
 352         self.name = name
 353         if name.startswith('ubuntu-') or name.startswith('debian-'):
 354             self.platform = 'linux'
 355             self.version = name.split('-')[1]
 356             self.bits = int(name.split('-')[2])
 357         elif name.startswith('windows-'):
 358             self.platform = 'windows'
 359             self.bits = int(name.split('-')[1])
 360         elif name == 'osx':
 361             self.platform = 'osx'
 362         elif name == 'source':
 363             self.platform = 'source'
 364
 365 def environment_for_target(target, directory):
 366     if target.platform == 'linux':
 367         return ChrootEnvironment(target.name)
 368     elif target.platform == 'windows':
 369         env = HostEnvironment(directory)
 370         prepare_for_windows(env, target.bits)
 371         return env
 372     elif target.platform == 'osx':
 373         env = RemoteEnvironment(OSX_BUILD_HOST)
 374         return env
 375     elif target.platform == 'source':
 376         return HostEnvironment()
 377
 378     return None
 379
 380 #
 381 # Project
 382 #
 383
 384 class Project(object):
 385     def __init__(self, name, directory, specifier=None):
 386         self.name = name
 387         self.directory = directory
 388         self.git_dir = GIT_DIR
 389         self.version = None
 390         self.specifier = specifier
 391         if self.specifier is None:
 392             self.specifier = 'master'
 393
 394     def checkout(self, env):
 395         flags = ''
 396         redirect = ''
 397         if args.quiet:
 398             flags = '-q'
 399             redirect = '>/dev/null'
 400         command('git clone --depth 0 %s %s/%s.git %s/src/%s' % (flags, self.git_dir, self.name, env.work_dir_cdist(), self.name))
 401         os.chdir('%s/src/%s' % (env.work_dir_cdist(), self.name))
 402         command('git checkout %s %s %s' % (flags, self.specifier, redirect))
 403         command('git submodule init')
 404         command('git submodule update')
 405         os.chdir(self.directory)
 406
 407         proj = '%s/src/%s/%s' % (env.work_dir_cdist(), self.name, self.directory)
 408
 409         self.read_cscript('%s/cscript' % proj)
 410
 411         if os.path.exists('%s/wscript' % proj):
 412             f = open('%s/wscript' % proj, 'r')
 413             version = None
 414             while 1:
 415                 l = f.readline()
 416                 if l == '':
 417                     break
 418
 419                 s = l.split()
 420                 if len(s) == 3 and s[0] == "VERSION":
 421                     self.version = Version(s[2])
 422
 423             f.close()
 424
 425     def read_cscript(self, s):
 426         self.cscript = {}
 427         execfile(s, self.cscript)
 428
 429 def set_version_in_wscript(version):
 430     f = open('wscript', 'rw')
 431     o = open('wscript.tmp', 'w')
 432     while 1:
 433         l = f.readline()
 434         if l == '':
 435             break
 436
 437         s = l.split()
 438         if len(s) == 3 and s[0] == "VERSION":
 439             print "Writing %s" % version
 440             print >>o,"VERSION = '%s'" % version
 441         else:
 442             print >>o,l,
 443     f.close()
 444     o.close()
 445
 446     os.rename('wscript.tmp', 'wscript')
 447
 448 def append_version_to_changelog(version):
 449     try:
 450         f = open('ChangeLog', 'r')
 451     except:
 452         log('Could not open ChangeLog')
 453         return
 454
 455     c = f.read()
 456     f.close()
 457
 458     f = open('ChangeLog', 'w')
 459     now = datetime.datetime.now()
 460     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))
 461     f.write(c)
 462
 463 def append_version_to_debian_changelog(version):
 464     if not os.path.exists('debian'):
 465         log('Could not find debian directory')
 466         return
 467
 468     command('dch -b -v %s-1 "New upstream release."' % version)
 469
 470 #
 471 # Command-line parser
 472 #
 473
 474 parser = argparse.ArgumentParser()
 475 parser.add_argument('command')
 476 parser.add_argument('-p', '--project', help='project name', required=True)
 477 parser.add_argument('-d', '--directory', help='directory within project repo', default='.')
 478 parser.add_argument('--beta', help='beta release', action='store_true')
 479 parser.add_argument('--full', help='full release', action='store_true')
 480 parser.add_argument('-c', '--checkout', help='string to pass to git for checkout')
 481 parser.add_argument('-o', '--output', help='output directory', default='.')
 482 parser.add_argument('-q', '--quiet', help='be quiet', action='store_true')
 483 parser.add_argument('-t', '--target', help='target')
 484 parser.add_argument('-k', '--keep', help='keep working tree', action='store_true')
 485 args = parser.parse_args()
 486
 487 args.output = os.path.abspath(args.output)
 488
 489 if args.project is None:
 490     error('you must specify -p or --project')
 491
 492 project = Project(args.project, args.directory, args.checkout)
 493
 494 if args.command == 'build':
 495     if args.target is None:
 496         error('you must specify -t or --target')
 497
 498     target = Target(args.target)
 499     env = environment_for_target(target, None)
 500     project.checkout(env)
 501     env.build_dependencies(target, project)
 502     env.build(target, project)
 503
 504     env.cleanup()
 505
 506 elif args.command == 'package':
 507     if args.target is None:
 508         error('you must specify -t or --target')
 509
 510     target = Target(args.target)
 511     env = environment_for_target(target, None)
 512
 513     packages = env.package(target, project)
 514     if hasattr(packages, 'strip') or (not hasattr(packages, '__getitem__') and not hasattr(packages, '__iter__')):
 515         packages = [packages]
 516
 517     if target.platform == 'linux':
 518         out = '%s/%s-%d' % (args.output, target.version, target.bits)
 519         try:
 520             os.makedirs(out)
 521         except:
 522             pass
 523         for p in packages:
 524             copyfile(p, '%s/%s' % (out, os.path.basename(p)))
 525     else:
 526         for p in packages:
 527             copyfile(p, '%s/%s' % (args.output, os.path.basename(p)))
 528
 529     env.cleanup()
 530
 531 elif args.command == 'release':
 532     if args.full is False and args.beta is False:
 533         error('you must specify --full or --beta')
 534
 535     env = HostEnvironment()
 536     project.checkout(env)
 537
 538     version = project.version
 539     if args.full:
 540         version.to_release()
 541     else:
 542         version.bump_beta()
 543
 544     set_version_in_wscript(version)
 545     append_version_to_changelog(version)
 546     append_version_to_debian_changelog(version)
 547
 548     command('git commit -a -m "Bump version"')
 549     command('git tag -m "v%s" v%s' % (version, version))
 550
 551     if args.full:
 552         version.bump_and_to_pre()
 553         set_version_in_wscript(version)
 554         command('git commit -a -m "Bump version"')
 555
 556     command('git push')
 557     command('git push --tags')
 558
 559     env.cleanup()
 560
 561 elif args.command == 'pot':
 562     env = HostEnvironment()
 563     project.checkout(env)
 564
 565     pots = project.cscript['make_pot'](env)
 566     for p in pots:
 567         copyfile(p, '%s/%s' % (args.output, os.path.basename(p)))
 568
 569     env.cleanup()
 570
 571 elif args.command == 'changelog':
 572     env = HostEnvironment()
 573     project.checkout(env)
 574
 575     text = open('ChangeLog', 'r')
 576     html = open('%s/changelog.html' % args.output, 'w')
 577     versions = 8
 578
 579     last = None
 580     changes = []
 581
 582     while 1:
 583         l = text.readline()
 584         if l == '':
 585             break
 586
 587         if len(l) > 0 and l[0] == "\t":
 588             s = l.split()
 589             if len(s) == 4 and s[1] == "Version" and s[3] == "released.":
 590                 if not "beta" in s[2]:
 591                     if last is not None and len(changes) > 0:
 592                         print >>html,"<h2>Changes between version %s and %s</h2>" % (s[2], last)
 593                         print >>html,"<ul>"
 594                         for c in changes:
 595                             print >>html,"<li>%s" % c
 596                         print >>html,"</ul>"
 597                     last = s[2]
 598                     changes = []
 599                     versions -= 1
 600                     if versions < 0:
 601                         break
 602             else:
 603                 c = l.strip()
 604                 if len(c) > 0:
 605                     if c[0] == '*':
 606                         changes.append(c[2:])
 607                     else:
 608                         changes[-1] += " " + c
 609
 610     env.cleanup()
 611
 612 elif args.command == 'manual':
 613     env = HostEnvironment()
 614     project.checkout(env)
 615
 616     dirs = project.cscript['make_manual'](env)
 617     for d in dirs:
 618         copytree(d, '%s/%s' % (args.output, os.path.basename(d)))
 619
 620     env.cleanup()
 621
 622 elif args.command == 'doxygen':
 623     env = HostEnvironment()
 624     project.checkout(env)
 625
 626     dirs = project.cscript['make_doxygen'](env)
 627     if hasattr(dirs, 'strip') or (not hasattr(dirs, '__getitem__') and not hasattr(dirs, '__iter__')):
 628         dirs = [dirs]
 629
 630     for d in dirs:
 631         copytree(d, '%s/%s' % (args.output, 'doc'))
 632
 633     env.cleanup()
 634
 635 elif args.command == 'latest':
 636     env = HostEnvironment()
 637     project.checkout(env)
 638
 639     f = command_and_read('git log --tags --simplify-by-decoration --pretty="%d"')
 640     t = f.readline()
 641     m = re.compile(".*\((.*)\).*").match(t)
 642     latest = None
 643     if m:
 644         tags = m.group(1).split(', ')
 645         for t in tags:
 646             if len(t) > 0 and t[0] == 'v':
 647                 latest = t[1:]
 648
 649     print latest
 650     env.cleanup()
 651
 652 elif args.command == 'test':
 653     if args.target is None:
 654         error('you must specify -t or --target')
 655
 656     target = Target(args.target)
 657     env = environment_for_target(target, '.')
 658     project.read_cscript('cscript')
 659     env.build(target, project)
 660
 661 else:
 662     error('invalid command %s' % args.command)