OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # Copyright (C) 2012 Google Inc. All rights reserved. | |
3 # | |
4 # Redistribution and use in source and binary forms, with or without | |
5 # modification, are permitted provided that the following conditions are | |
6 # met: | |
7 # | |
8 # * Redistributions of source code must retain the above copyright | |
9 # notice, this list of conditions and the following disclaimer. | |
10 # * Redistributions in binary form must reproduce the above | |
11 # copyright notice, this list of conditions and the following disclaimer | |
12 # in the documentation and/or other materials provided with the | |
13 # distribution. | |
14 # * Neither the name of Google Inc. nor the names of its | |
15 # contributors may be used to endorse or promote products derived from | |
16 # this software without specific prior written permission. | |
17 # | |
18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS | |
19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT | |
20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR | |
21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT | |
22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, | |
23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT | |
24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | |
25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | |
26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
29 | |
30 """A utility script for starting and stopping the web socket server with the | |
31 same configuration as used in the layout tests.""" | |
32 | |
33 import logging | |
34 import optparse | |
35 import tempfile | |
36 | |
37 from webkitpy.common.host import Host | |
38 from webkitpy.layout_tests.servers import websocket_server | |
39 | |
40 | |
41 def main(): | |
42 option_parser = optparse.OptionParser() | |
43 option_parser.add_option('--server', type='choice', | |
44 choices=['start', 'stop'], default='start', | |
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', | |
70 default=None, help='output directory.') | |
71 option_parser.add_option('-v', '--verbose', action='store_true', | |
72 default=False, | |
73 help='Include debug-level logging.') | |
74 options, args = option_parser.parse_args() | |
75 | |
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: | |
85 options.output_dir = tempfile.gettempdir() | |
86 | |
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() | |
100 # FIXME: Make this work with other ports as well. | |
101 port_obj = host.port_factory.get(port_name='chromium', options=options) | |
102 pywebsocket = websocket_server.PyWebSocket(port_obj, options.output_dir, **k
wds) | |
103 | |
104 log_level = logging.WARN | |
105 if options.verbose: | |
106 log_level = logging.DEBUG | |
107 logging.basicConfig(level=log_level) | |
108 | |
109 if 'start' == options.server: | |
110 pywebsocket.start() | |
111 else: | |
112 pywebsocket.stop() | |
113 | |
114 if '__main__' == __name__: | |
115 main() | |
OLD | NEW |