Sheng
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import logging
|
|
import tornado.web
|
|
import tornado.ioloop
|
|
|
|
from tornado.options import options
|
|
from webssh.handler import IndexHandler, WsockHandler, NotFoundHandler
|
|
from webssh.settings import (
|
|
get_app_settings, get_host_keys_settings, get_policy_setting,
|
|
get_ssl_context, get_server_settings, detect_is_open_to_public
|
|
)
|
|
|
|
|
|
def make_handlers(loop, options):
|
|
host_keys_settings = get_host_keys_settings(options)
|
|
policy = get_policy_setting(options, host_keys_settings)
|
|
|
|
handlers = [
|
|
(r'/', IndexHandler, dict(loop=loop, policy=policy,
|
|
host_keys_settings=host_keys_settings)),
|
|
(r'/ws', WsockHandler, dict(loop=loop))
|
|
]
|
|
return handlers
|
|
|
|
|
|
def make_app(handlers, settings):
|
|
settings.update(default_handler_class=NotFoundHandler)
|
|
return tornado.web.Application(handlers, **settings)
|
|
|
|
|
|
def main():
|
|
options.parse_command_line()
|
|
loop = tornado.ioloop.IOLoop.current()
|
|
app = make_app(make_handlers(loop, options), get_app_settings(options))
|
|
ssl_ctx = get_ssl_context(options)
|
|
server_settings = get_server_settings(options)
|
|
app.listen(options.port, options.address, **server_settings)
|
|
logging.info('Listening on {}:{}'.format(options.address, options.port))
|
|
if ssl_ctx:
|
|
server_settings.update(ssl_options=ssl_ctx)
|
|
app.listen(options.sslport, options.ssladdress, **server_settings)
|
|
logging.info('Listening on ssl {}:{}'.format(options.ssladdress,
|
|
options.sslport))
|
|
detect_is_open_to_public(options)
|
|
loop.start()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|