OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 | 2 |
3 # Copyright (c) 2009-2010 The Chromium OS Authors. All rights reserved. | 3 # Copyright (c) 2009-2010 The Chromium OS Authors. All rights reserved. |
4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
6 | 6 |
7 """A CherryPy-based webserver to host images and build packages.""" | 7 """A CherryPy-based webserver to host images and build packages.""" |
8 | 8 |
9 import cherrypy | 9 import cherrypy |
10 import optparse | 10 import optparse |
11 import os | 11 import os |
12 import sys | 12 import sys |
13 | 13 |
14 import autoupdate | 14 import autoupdate |
15 | 15 |
16 CACHED_ENTRIES=12 | |
17 | |
18 # Sets up global to share between classes. | 16 # Sets up global to share between classes. |
19 global updater | 17 global updater |
20 updater = None | 18 updater = None |
21 | 19 |
22 def _GetConfig(options): | 20 def _GetConfig(options): |
23 """Returns the configuration for the devserver.""" | 21 """Returns the configuration for the devserver.""" |
24 base_config = { 'global': | 22 base_config = { 'global': |
25 { 'server.log_request_headers': True, | 23 { 'server.log_request_headers': True, |
26 'server.protocol_version': 'HTTP/1.1', | 24 'server.protocol_version': 'HTTP/1.1', |
27 'server.socket_host': '0.0.0.0', | 25 'server.socket_host': '0.0.0.0', |
(...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
108 index.exposed = True | 106 index.exposed = True |
109 | 107 |
110 | 108 |
111 if __name__ == '__main__': | 109 if __name__ == '__main__': |
112 usage = 'usage: %prog [options]' | 110 usage = 'usage: %prog [options]' |
113 parser = optparse.OptionParser(usage) | 111 parser = optparse.OptionParser(usage) |
114 parser.add_option('--archive_dir', dest='archive_dir', | 112 parser.add_option('--archive_dir', dest='archive_dir', |
115 help='serve archived builds only.') | 113 help='serve archived builds only.') |
116 parser.add_option('--board', dest='board', | 114 parser.add_option('--board', dest='board', |
117 help='When pre-generating update, board for latest image.') | 115 help='When pre-generating update, board for latest image.') |
118 parser.add_option('--clear_cache', action="store_true", default=False, | |
119 help='Clear out all cached udpates and exit') | |
120 parser.add_option('--client_prefix', dest='client_prefix', | 116 parser.add_option('--client_prefix', dest='client_prefix', |
121 help='Required prefix for client software version.', | 117 help='Required prefix for client software version.', |
122 default='MementoSoftwareUpdate') | 118 default='MementoSoftwareUpdate') |
123 parser.add_option('--factory_config', dest='factory_config', | 119 parser.add_option('--factory_config', dest='factory_config', |
124 help='Config file for serving images from factory floor.') | 120 help='Config file for serving images from factory floor.') |
125 parser.add_option('--for_vm', dest='vm', default=False, action='store_true', | 121 parser.add_option('--for_vm', dest='vm', default=False, action='store_true', |
126 help='Update is for a vm image.') | 122 help='Update is for a vm image.') |
127 parser.add_option('--image', dest='image', | 123 parser.add_option('--image', dest='image', |
128 help='Force update using this image.') | 124 help='Force update using this image.') |
129 parser.add_option('-p', '--pregenerate_update', action='store_true', | 125 parser.add_option('-p', '--pregenerate_update', action='store_true', |
130 default=False, help='Pre-generate update payload.') | 126 default=False, help='Pre-generate update payload.') |
131 parser.add_option('--port', default=8080, | 127 parser.add_option('--port', default=8080, |
132 help='Port for the dev server to use.') | 128 help='Port for the dev server to use.') |
133 parser.add_option('--src_image', default='', | 129 parser.add_option('--src_image', default='', |
134 help='Image on remote machine for generating delta update.') | 130 help='Image on remote machine for generating delta update.') |
135 parser.add_option('-t', action='store_true', dest='test_image') | 131 parser.add_option('-t', action='store_true', dest='test_image') |
136 parser.add_option('-u', '--urlbase', dest='urlbase', | 132 parser.add_option('-u', '--urlbase', dest='urlbase', |
137 help='base URL, other than devserver, for update images.') | 133 help='base URL, other than devserver, for update images.') |
| 134 parser.add_option('--use_cached', action="store_true", default=False, |
| 135 help='Prefer cached image regardless of timestamps.') |
138 parser.add_option('--validate_factory_config', action="store_true", | 136 parser.add_option('--validate_factory_config', action="store_true", |
139 dest='validate_factory_config', | 137 dest='validate_factory_config', |
140 help='Validate factory config file, then exit.') | 138 help='Validate factory config file, then exit.') |
141 parser.set_usage(parser.format_help()) | 139 parser.set_usage(parser.format_help()) |
142 (options, _) = parser.parse_args() | 140 (options, _) = parser.parse_args() |
143 | 141 |
144 devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0])) | 142 devserver_dir = os.path.dirname(os.path.abspath(sys.argv[0])) |
145 root_dir = os.path.realpath('%s/../..' % devserver_dir) | 143 root_dir = os.path.realpath('%s/../..' % devserver_dir) |
146 serve_only = False | 144 serve_only = False |
147 | 145 |
148 if options.archive_dir: | 146 if options.archive_dir: |
149 static_dir = os.path.realpath(options.archive_dir) | 147 static_dir = os.path.realpath(options.archive_dir) |
150 _PrepareToServeUpdatesOnly(static_dir) | 148 _PrepareToServeUpdatesOnly(static_dir) |
151 serve_only = True | 149 serve_only = True |
152 else: | 150 else: |
153 static_dir = os.path.realpath('%s/static' % devserver_dir) | 151 static_dir = os.path.realpath('%s/static' % devserver_dir) |
154 os.system('mkdir -p %s' % static_dir) | 152 os.system('mkdir -p %s' % static_dir) |
155 | 153 |
156 cache_dir = os.path.join(static_dir, 'cache') | |
157 cherrypy.log('Using cache directory %s' % cache_dir, 'DEVSERVER') | |
158 | |
159 if options.clear_cache: | |
160 # Clear the cache and exit | |
161 sys.exit(os.system('sudo rm -rf %s' % cache_dir)) | |
162 | |
163 if os.path.exists(cache_dir): | |
164 # Clear all but the last N cached updates | |
165 cmd = ('cd %s; ls -1tr | head --lines=-%d | xargs rm -rf' % | |
166 (cache_dir, CACHED_ENTRIES)) | |
167 if os.system(cmd) != 0: | |
168 cherrypy.log('Failed to clean up old delta cache files with %s' % cmd, | |
169 'DEVSERVER') | |
170 sys.exit(1) | |
171 | |
172 cherrypy.log('Source root is %s' % root_dir, 'DEVSERVER') | 154 cherrypy.log('Source root is %s' % root_dir, 'DEVSERVER') |
173 cherrypy.log('Serving from %s' % static_dir, 'DEVSERVER') | 155 cherrypy.log('Serving from %s' % static_dir, 'DEVSERVER') |
174 | 156 |
175 updater = autoupdate.Autoupdate( | 157 updater = autoupdate.Autoupdate( |
176 root_dir=root_dir, | 158 root_dir=root_dir, |
177 static_dir=static_dir, | 159 static_dir=static_dir, |
178 serve_only=serve_only, | 160 serve_only=serve_only, |
179 urlbase=options.urlbase, | 161 urlbase=options.urlbase, |
180 test_image=options.test_image, | 162 test_image=options.test_image, |
181 factory_config_path=options.factory_config, | 163 factory_config_path=options.factory_config, |
182 client_prefix=options.client_prefix, | 164 client_prefix=options.client_prefix, |
183 forced_image=options.image, | 165 forced_image=options.image, |
| 166 use_cached=options.use_cached, |
184 port=options.port, | 167 port=options.port, |
185 src_image=options.src_image, | 168 src_image=options.src_image, |
186 vm=options.vm, | 169 vm=options.vm, |
187 board=options.board) | 170 board=options.board) |
188 | 171 |
189 # Sanity-check for use of validate_factory_config. | 172 # Sanity-check for use of validate_factory_config. |
190 if not options.factory_config and options.validate_factory_config: | 173 if not options.factory_config and options.validate_factory_config: |
191 parser.error('You need a factory_config to validate.') | 174 parser.error('You need a factory_config to validate.') |
192 | 175 |
193 if options.factory_config: | 176 if options.factory_config: |
194 updater.ImportFactoryConfigFile(options.factory_config, | 177 updater.ImportFactoryConfigFile(options.factory_config, |
195 options.validate_factory_config) | 178 options.validate_factory_config) |
196 # We don't run the dev server with this option. | 179 # We don't run the dev server with this option. |
197 if options.validate_factory_config: | 180 if options.validate_factory_config: |
198 sys.exit(0) | 181 sys.exit(0) |
199 elif options.pregenerate_update: | 182 elif options.pregenerate_update: |
200 if not updater.PreGenerateUpdate(): | 183 if not updater.PreGenerateUpdate(): |
201 sys.exit(1) | 184 sys.exit(1) |
202 | 185 |
203 cherrypy.quickstart(DevServerRoot(), config=_GetConfig(options)) | 186 cherrypy.quickstart(DevServerRoot(), config=_GetConfig(options)) |
OLD | NEW |