X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-dev.git/blobdiff_plain/929f1553a7e60441bee2b53fed3530445a7299cf..3d81f14142b8b4e1216a8bfe54166c901fea5412:/build-hooks/post-build?ds=sidebyside diff --git a/build-hooks/post-build b/build-hooks/post-build index a727cf1..277f8f2 100755 --- a/build-hooks/post-build +++ b/build-hooks/post-build @@ -53,27 +53,70 @@ from email.mime import text from invirt import common, database, builder from invirt.config import structs as config -def make_msg(build, values, verbose=True, success=lambda x: x, failure=lambda x: x): +def build_completion_msg(succeeded, values, verbose=True, success=lambda x: x, failure=lambda x: x): + """Format a message reporting the results of a build""" values = dict(values) if not verbose and values['traceback'] is not None: + # TODO: better heuristic values['traceback'] = textwrap.fill('\n'.join(values['traceback'].split('\n')[-2:])) - if build.succeeded: + if succeeded: values['result'] = success(values['result']) - msg = """Build of %(package)s v%(version)s in %(pocket)s %(result)s. + msg = """Build of %(package)s %(version)s in %(pocket)s %(result)s. Branch %(pocket)s has been advanced to %(short_commit)s. (Build %(build_id)s was submitted by %(principal)s at %(inserted_at)s.)""" % values else: values['result'] = failure(values['result']) - msg = """Build of %(package)s v%(version)s in %(pocket)s %(result)s while %(failed_stage)s. + msg = """Build of %(package)s %(version)s in %(pocket)s %(result)s while %(failed_stage)s. %(traceback)s (Build %(build_id)s was submitted by %(principal)s at %(inserted_at)s.)""" % values return msg +def submit_completion_msg(succeeded, values, verbose=True, success=lambda x: x, failure=lambda x: x): + values = dict(values) + if succeeded: + values['result'] = success(values['result']) + else: + values['result'] = failure(values['result']) + msg = """Submission of %(commit)s to be built in %(pocket)s %(result)s. +Build submitted by %(principal)s.""" % values + return msg + +def repo_creation_msg(succeeded, values, verbose=True, success=lambda x: x, failure=lambda x: x): + values = dict(values) + assert succeeded + msg = '%(principal)s just created a new repository, %(category)s/%(name)s.git' % values + return msg + +# Names of hooks +POST_BUILD = 'post-build' +FAILED_BUILD = 'failed-build' +POST_SUBMIT = 'post-submit' +FAILED_SUBMIT = 'failed-submit' +POST_ADD_REPO = 'post-add-repo' + +# Types of communication + +ZEPHYR = 'zephyr' +MAIL = 'mail' + +message_generators = { + ZEPHYR : { POST_BUILD : build_completion_msg, + FAILED_BUILD : build_completion_msg, + POST_SUBMIT : submit_completion_msg, + FAILED_SUBMIT : submit_completion_msg, + POST_ADD_REPO : repo_creation_msg }, + MAIL : { POST_BUILD : build_completion_msg, + FAILED_BUILD : build_completion_msg, + POST_SUBMIT : submit_completion_msg, + FAILED_SUBMIT : submit_completion_msg, + POST_ADD_REPO : repo_creation_msg } + } + def zephyr_escape(m): m = re.sub('@', '@@', m) m = re.sub('}', '@(})', m) @@ -86,52 +129,97 @@ def zephyr_failure(m): return '}@{@color(red)%s}@{' % zephyr_escape(m) def main(): - parser = optparse.OptionParser('Usage: %prog build_id') + parser = optparse.OptionParser('Usage: %prog [options] [arguments]') opts, args = parser.parse_args() - if len(args) != 1: - parser.print_help() - return 1 prog = os.path.basename(sys.argv[0]) - database.connect() - build = database.Build.query().get(args[0]) - short_commit = builder.canonicalize_commit(build.package, build.commit, shorten=True) - values = { 'build_id' : build.build_id, - 'commit' : build.commit, - 'failed_stage' : build.failed_stage, - 'inserted_at' : build.inserted_at, - 'package' : build.package, - 'pocket' : build.pocket, - 'principal' : build.principal, - 'short_commit' : short_commit, - 'traceback' : build.traceback, - 'version' : build.version } - if build.succeeded: - values['result'] = 'succeeded' - else: - values['result'] = 'failed' - try: - if prog == 'post-build': + if prog == POST_BUILD: hook_config = config.build.hooks.post_build - elif prog == 'failed-build': + elif prog == FAILED_BUILD: hook_config = config.build.hooks.failed_build + elif prog == POST_SUBMIT: + hook_config = config.build.hooks.post_submit + elif prog == FAILED_SUBMIT: + hook_config = config.build.hooks.failed_submit + elif prog == POST_ADD_REPO: + hook_config = config.build.hooks.post_add_repo else: - print >>sys.stderr, '{post,failed}-build invoke with unrecognized name %s' % prog + parser.error('hook script invoked with unrecognized name %s' % prog) return 2 except common.InvirtConfigError: print >>sys.stderr, 'No hook configuration found for %s.' % prog return 1 + if prog in [POST_BUILD, FAILED_BUILD]: + if len(args) != 1: + parser.set_usage('Usage: %prog [options] build_id') + parser.print_help() + return 1 + database.connect() + build = database.Build.query().get(args[0]) + short_commit = builder.canonicalize_commit(build.package, build.commit, shorten=True) + values = { 'build_id' : build.build_id, + 'commit' : build.commit, + 'failed_stage' : build.failed_stage, + 'inserted_at' : build.inserted_at, + 'package' : build.package, + 'pocket' : build.pocket, + 'principal' : build.principal, + 'short_commit' : short_commit, + 'traceback' : build.traceback, + 'version' : build.version, + 'default_instance' : 'build_%(build_id)s', + 'default_subject' : 'XVM build %(result)s: %(package)s %(version)s in %(pocket)s'} + if build.succeeded: + assert prog == POST_BUILD + values['result'] = 'succeeded' + succeeded = True + else: + assert prog == FAILED_BUILD + values['result'] = 'failed' + succeeded = False + elif prog in [POST_SUBMIT, FAILED_SUBMIT]: + if len(args) != 4: + parser.set_usage('Usage: %prog [options] pocket package commit principal') + parser.print_help() + return 2 + values = { 'pocket' : args[0], + 'package' : args[1], + 'commit' : args[2], + 'principal' : args[3], + 'default_instance' : 'submission', + 'default_subject' : 'Submission %(result)s: %(package)s %(version)s in %(pocket)s'} + if prog == POST_SUBMIT: + values['result'] = 'succeeded' + succeeded = True + else: + values['result'] = 'failed' + succeeded = False + elif prog in [POST_ADD_REPO]: + if len(args) != 3: + parser.set_usage('Usage: %prog [options] category name principal') + parser.print_help() + return 3 + values = { 'category' : args[0], + 'name' : args[1], + 'principal' : args[2], + 'default_instance' : 'new-repo', + 'default_subject' : 'New repository %(category)s/%(name)s'} + succeeded = True + else: + raise AssertionError('Impossible state') + try: zephyr_config = hook_config.zephyr klass = zephyr_config['class'] % values except common.InvirtConfigError: print >>sys.stderr, 'No zephyr configuration specified for %s.' % prog else: - msg = '@{%s}' % make_msg(build, values, verbose=False, + make_msg = message_generators[ZEPHYR][prog] + msg = '@{%s}' % make_msg(succeeded, values, verbose=False, success=zephyr_success, failure=zephyr_failure) - instance = zephyr_config.get('instance', 'build_%(build_id)s') % values + instance = zephyr_config.get('instance', values['default_instance']) % values zsig = zephyr_config.get('zsig', 'XVM Buildbot') % values common.captureOutput(['zwrite', '-c', klass, '-i', instance, '-s', zsig, '-d', '-m', msg], @@ -144,11 +232,12 @@ def main(): except common.InvirtConfigError: print >>sys.stderr, 'No email configuration specified for %s.' % prog else: - msg = make_msg(build, values) + make_msg = message_generators[MAIL][prog] + msg = make_msg(succeeded, values) email = text.MIMEText(msg) email['To'] = to % values email['From'] = sender % values - email['Subject'] = mail_config.get('subject', 'XVM build %(build_id)s has %(result)s') % values + email['Subject'] = mail_config.get('subject', values['default_subject']) % values common.captureOutput(['sendmail', '-t'], email.as_string(), stdout=None, stderr=None)