| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (C) 2012 Google Inc. All rights reserved. | 2 # Copyright (C) 2012 Google Inc. All rights reserved. |
| 3 # | 3 # |
| 4 # Redistribution and use in source and binary forms, with or without | 4 # Redistribution and use in source and binary forms, with or without |
| 5 # modification, are permitted provided that the following conditions are | 5 # modification, are permitted provided that the following conditions are |
| 6 # met: | 6 # met: |
| 7 # | 7 # |
| 8 # * Redistributions of source code must retain the above copyright | 8 # * Redistributions of source code must retain the above copyright |
| 9 # notice, this list of conditions and the following disclaimer. | 9 # notice, this list of conditions and the following disclaimer. |
| 10 # * Redistributions in binary form must reproduce the above | 10 # * Redistributions in binary form must reproduce the above |
| (...skipping 25 matching lines...) Expand all Loading... |
| 36 | 36 |
| 37 from webkitpy.common.host import Host | 37 from webkitpy.common.host import Host |
| 38 from webkitpy.layout_tests.servers import websocket_server | 38 from webkitpy.layout_tests.servers import websocket_server |
| 39 | 39 |
| 40 | 40 |
| 41 def main(): | 41 def main(): |
| 42 option_parser = optparse.OptionParser() | 42 option_parser = optparse.OptionParser() |
| 43 option_parser.add_option('--server', type='choice', | 43 option_parser.add_option('--server', type='choice', |
| 44 choices=['start', 'stop'], default='start', | 44 choices=['start', 'stop'], default='start', |
| 45 help='Server action (start|stop).') | 45 help='Server action (start|stop).') |
| 46 option_parser.add_option('-p', '--port', dest='port', | |
| 47 default=None, help='Port to listen on.') | |
| 48 option_parser.add_option('-r', '--root', | |
| 49 help='Absolute path to DocumentRoot ' | |
| 50 '(overrides layout test roots).') | |
| 51 option_parser.add_option('-t', '--tls', dest='use_tls', | |
| 52 action='store_true', | |
| 53 default=False, help='use TLS (wss://).') | |
| 54 option_parser.add_option('-k', '--private_key', dest='private_key', | |
| 55 default='', help='TLS private key file.') | |
| 56 option_parser.add_option('-c', '--certificate', dest='certificate', | |
| 57 default='', help='TLS certificate file.') | |
| 58 option_parser.add_option('--ca-certificate', dest='ca_certificate', | |
| 59 default='', help='TLS CA certificate file for ' | |
| 60 'client authentication.') | |
| 61 option_parser.add_option('--chromium', action='store_true', | |
| 62 dest='chromium', | |
| 63 default=False, | |
| 64 help='Use the Chromium port.') | |
| 65 option_parser.add_option('--register_cygwin', action="store_true", | |
| 66 dest="register_cygwin", | |
| 67 help='Register Cygwin paths (on Win try bots).') | |
| 68 option_parser.add_option('--pidfile', help='path to pid file.') | |
| 69 option_parser.add_option('--output-dir', dest='output_dir', | 46 option_parser.add_option('--output-dir', dest='output_dir', |
| 70 default=None, help='output directory.') | 47 default=None, help='output directory.') |
| 71 option_parser.add_option('-v', '--verbose', action='store_true', | 48 option_parser.add_option('-v', '--verbose', action='store_true', |
| 72 default=False, | 49 default=False, |
| 73 help='Include debug-level logging.') | 50 help='Include debug-level logging.') |
| 74 options, args = option_parser.parse_args() | 51 options, args = option_parser.parse_args() |
| 75 | 52 |
| 76 if not options.port: | |
| 77 if options.use_tls: | |
| 78 # FIXME: We shouldn't grab at this private variable. | |
| 79 options.port = websocket_server._DEFAULT_WSS_PORT | |
| 80 else: | |
| 81 # FIXME: We shouldn't grab at this private variable. | |
| 82 options.port = websocket_server._DEFAULT_WS_PORT | |
| 83 | |
| 84 if not options.output_dir: | 53 if not options.output_dir: |
| 85 options.output_dir = tempfile.gettempdir() | 54 options.output_dir = tempfile.gettempdir() |
| 86 | 55 |
| 87 kwds = {'port': options.port, 'use_tls': options.use_tls} | |
| 88 if options.root: | |
| 89 kwds['root'] = options.root | |
| 90 if options.private_key: | |
| 91 kwds['private_key'] = options.private_key | |
| 92 if options.certificate: | |
| 93 kwds['certificate'] = options.certificate | |
| 94 if options.ca_certificate: | |
| 95 kwds['ca_certificate'] = options.ca_certificate | |
| 96 if options.pidfile: | |
| 97 kwds['pidfile'] = options.pidfile | |
| 98 | |
| 99 host = Host() | 56 host = Host() |
| 100 # FIXME: Make this work with other ports as well. | 57 port_obj = host.port_factory.get() |
| 101 port_obj = host.port_factory.get(port_name='chromium', options=options) | 58 pywebsocket = websocket_server.PyWebSocket(port_obj, options.output_dir) |
| 102 pywebsocket = websocket_server.PyWebSocket(port_obj, options.output_dir, **k
wds) | |
| 103 | 59 |
| 104 log_level = logging.WARN | 60 log_level = logging.WARN |
| 105 if options.verbose: | 61 if options.verbose: |
| 106 log_level = logging.DEBUG | 62 log_level = logging.DEBUG |
| 107 logging.basicConfig(level=log_level) | 63 logging.basicConfig(level=log_level) |
| 108 | 64 |
| 109 if 'start' == options.server: | 65 if 'start' == options.server: |
| 110 pywebsocket.start() | 66 pywebsocket.start() |
| 111 else: | 67 else: |
| 112 pywebsocket.stop() | 68 pywebsocket.stop() |
| 113 | 69 |
| 114 if '__main__' == __name__: | 70 if '__main__' == __name__: |
| 115 main() | 71 main() |
| OLD | NEW |