[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 1 | # Copyright (c) 2009 The Chromium OS Authors. All rights reserved. |
| 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | import autoupdate |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 6 | import buildutil |
| 7 | import optparse |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 8 | import os |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 9 | import SimpleHTTPServer |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 10 | import web |
[email protected] | 4dc2581 | 2009-10-27 23:46:26 | [diff] [blame] | 11 | import sys |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 12 | |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 13 | global updater |
| 14 | updater = None |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 15 | |
[email protected] | 6424466 | 2009-11-12 00:52:08 | [diff] [blame] | 16 | global buildbot |
| 17 | buildbot = None |
| 18 | |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 19 | class index: |
| 20 | def GET(self): |
[email protected] | 6424466 | 2009-11-12 00:52:08 | [diff] [blame] | 21 | pkgs = buildbot.GetPackages() |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 22 | return render.index(pkgs) |
| 23 | |
| 24 | class update: |
| 25 | """ |
| 26 | Processes updates from the client machine. If an update is found, the url |
| 27 | references a static link that can be served automagically from web.py. |
| 28 | """ |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 29 | def POST(self, args=None): |
| 30 | return updater.HandleUpdatePing(web.data(), args) |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 31 | |
[email protected] | 6424466 | 2009-11-12 00:52:08 | [diff] [blame] | 32 | class build: |
| 33 | """ |
| 34 | builds the package specified by the pkg parameter and returns the name |
| 35 | of the output file. |
| 36 | """ |
Ryan Cairns | dd1ceb8 | 2010-03-03 05:35:01 | [diff] [blame] | 37 | def POST(self): |
[email protected] | 6424466 | 2009-11-12 00:52:08 | [diff] [blame] | 38 | input = web.input() |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 39 | web.debug('emerging %s ' % input.pkg) |
| 40 | emerge_command = 'emerge-%s %s' % (input.board, input.pkg) |
Ryan Cairns | dd1ceb8 | 2010-03-03 05:35:01 | [diff] [blame] | 41 | err = os.system(emerge_command) |
| 42 | if err != 0: |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 43 | raise Exception('failed to execute %s' % emerge_command) |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 44 | |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 45 | if __name__ == '__main__': |
| 46 | usage = 'usage: %prog [options]' |
| 47 | parser = optparse.OptionParser(usage) |
| 48 | parser.add_option('-a', '--archive_dir', dest='archive_dir', |
| 49 | help='serve archived builds only.') |
| 50 | parser.add_option("-t", action="store_true", dest="test_image") |
| 51 | options, args = parser.parse_args() |
| 52 | # clean up the args, due to httpserver's hardcoded use of sys.argv |
| 53 | if options.archive_dir: |
| 54 | sys.argv.remove('-a') |
| 55 | sys.argv.remove(options.archive_dir) |
| 56 | if options.test_image: |
| 57 | sys.argv.remove('-t') |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 58 | |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 59 | |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 60 | root_dir = os.path.realpath('%s/../..' % |
| 61 | os.path.dirname(os.path.abspath(sys.argv[0]))) |
| 62 | if options.archive_dir: |
| 63 | static_dir = os.path.realpath(options.archive_dir) |
| 64 | assert os.path.exists(static_dir), '%s must exist.' % options.archive_dir |
| 65 | web.debug('using archive dir: %s' % static_dir) |
| 66 | else: |
| 67 | static_dir = os.path.realpath('%s/static' % |
| 68 | os.path.dirname(os.path.abspath(sys.argv[0]))) |
| 69 | web.debug('dev root is %s' % root_dir) |
| 70 | os.system('mkdir -p %s' % static_dir) |
| 71 | web.debug('Serving images from %s' % static_dir) |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 72 | |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 73 | updater = autoupdate.Autoupdate(root_dir=root_dir, |
| 74 | static_dir=static_dir, |
| 75 | serve_only=options.archive_dir, |
| 76 | test_image=options.test_image) |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 77 | urls = ('/', 'index', |
[email protected] | 6424466 | 2009-11-12 00:52:08 | [diff] [blame] | 78 | '/update', 'update', |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 79 | '/update/(.+)', 'update', |
[email protected] | 6424466 | 2009-11-12 00:52:08 | [diff] [blame] | 80 | '/webbuild', 'webbuild', |
| 81 | '/build', 'build') |
| 82 | |
Sean O'Connor | 14b6a0a | 2010-03-21 06:23:48 | [diff] [blame] | 83 | app = web.application(urls, globals(), autoreload=True) |
[email protected] | 21a5ca3 | 2009-11-04 18:23:23 | [diff] [blame] | 84 | render = web.template.render('templates/') |
[email protected] | ded2240 | 2009-10-26 22:36:21 | [diff] [blame] | 85 | app.run() |