143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
import logging, time, cgi, string, random
|
|
from gevent.pywsgi import WSGIServer
|
|
from gevent.pywsgi import WSGIHandler
|
|
from lib.geventwebsocket.handler import WebSocketHandler
|
|
from UiRequest import UiRequest
|
|
from Site import SiteManager
|
|
from Config import config
|
|
from Debug import Debug
|
|
|
|
|
|
# Skip websocket handler if not necessary
|
|
class UiWSGIHandler(WSGIHandler):
|
|
def __init__(self, *args, **kwargs):
|
|
self.server = args[2]
|
|
super(UiWSGIHandler, self).__init__(*args, **kwargs)
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
|
|
|
|
def run_application(self):
|
|
self.server.sockets[self.client_address] = self.socket
|
|
if "HTTP_UPGRADE" in self.environ: # Websocket request
|
|
try:
|
|
ws_handler = WebSocketHandler(*self.args, **self.kwargs)
|
|
ws_handler.__dict__ = self.__dict__ # Match class variables
|
|
ws_handler.run_application()
|
|
except Exception, err:
|
|
logging.error("UiWSGIHandler websocket error: %s" % Debug.formatException(err))
|
|
if config.debug: # Allow websocket errors to appear on /Debug
|
|
import sys
|
|
del self.server.sockets[self.client_address]
|
|
sys.modules["main"].DebugHook.handleError()
|
|
else: # Standard HTTP request
|
|
#print self.application.__class__.__name__
|
|
try:
|
|
super(UiWSGIHandler, self).run_application()
|
|
except Exception, err:
|
|
logging.error("UiWSGIHandler error: %s" % Debug.formatException(err))
|
|
if config.debug: # Allow websocket errors to appear on /Debug
|
|
import sys
|
|
del self.server.sockets[self.client_address]
|
|
sys.modules["main"].DebugHook.handleError()
|
|
del self.server.sockets[self.client_address]
|
|
|
|
|
|
class UiServer:
|
|
def __init__(self):
|
|
self.ip = config.ui_ip
|
|
self.port = config.ui_port
|
|
if self.ip == "*": self.ip = "" # Bind all
|
|
#self.sidebar_websockets = [] # Sidebar websocket connections
|
|
#self.auth_key = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(12)) # Global admin auth key
|
|
self.sites = SiteManager.site_manager.list()
|
|
self.log = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
# Handle WSGI request
|
|
def handleRequest(self, env, start_response):
|
|
path = env["PATH_INFO"]
|
|
if env.get("QUERY_STRING"):
|
|
get = dict(cgi.parse_qsl(env['QUERY_STRING']))
|
|
else:
|
|
get = {}
|
|
ui_request = UiRequest(self, get, env, start_response)
|
|
if config.debug: # Let the exception catched by werkezung
|
|
return ui_request.route(path)
|
|
else: # Catch and display the error
|
|
try:
|
|
return ui_request.route(path)
|
|
except Exception, err:
|
|
logging.debug("UiRequest error: %s" % Debug.formatException(err))
|
|
return ui_request.error500("Err: %s" % Debug.formatException(err))
|
|
|
|
|
|
# Reload the UiRequest class to prevent restarts in debug mode
|
|
def reload(self):
|
|
global UiRequest
|
|
import imp, sys
|
|
reload(sys.modules["User.UserManager"])
|
|
reload(sys.modules["Ui.UiWebsocket"])
|
|
UiRequest = imp.load_source("UiRequest", "src/Ui/UiRequest.py").UiRequest
|
|
#UiRequest.reload()
|
|
|
|
|
|
# Bind and run the server
|
|
def start(self):
|
|
handler = self.handleRequest
|
|
|
|
if config.debug:
|
|
# Auto reload UiRequest on change
|
|
from Debug import DebugReloader
|
|
DebugReloader(self.reload)
|
|
|
|
# Werkzeug Debugger
|
|
try:
|
|
from werkzeug.debug import DebuggedApplication
|
|
handler = DebuggedApplication(self.handleRequest, evalex=True)
|
|
except Exception, err:
|
|
self.log.info("%s: For debugging please download Werkzeug (http://werkzeug.pocoo.org/)" % err)
|
|
from Debug import DebugReloader
|
|
self.log.write = lambda msg: self.log.debug(msg.strip()) # For Wsgi access.log
|
|
self.log.info("--------------------------------------")
|
|
self.log.info("Web interface: http://%s:%s/" % (config.ui_ip, config.ui_port))
|
|
self.log.info("--------------------------------------")
|
|
|
|
if config.open_browser:
|
|
logging.info("Opening browser: %s...", config.open_browser)
|
|
import webbrowser
|
|
if config.open_browser == "default_browser":
|
|
browser = webbrowser.get()
|
|
else:
|
|
browser = webbrowser.get(config.open_browser)
|
|
browser.open("http://%s:%s" % (config.ui_ip, config.ui_port), new=2)
|
|
|
|
self.server = WSGIServer((self.ip.replace("*", ""), self.port), handler, handler_class=UiWSGIHandler, log=self.log)
|
|
self.server.sockets = {}
|
|
self.server.serve_forever()
|
|
self.log.debug("Stopped.")
|
|
|
|
|
|
def stop(self):
|
|
self.log.debug("Stopping...")
|
|
# Close WS sockets
|
|
if "clients" in dir(self.server):
|
|
for client in self.server.clients.values():
|
|
client.ws.close()
|
|
# Close http sockets
|
|
sock_closed = 0
|
|
for sock in self.server.sockets.values():
|
|
try:
|
|
sock._sock.close()
|
|
sock.close()
|
|
sock_closed += 1
|
|
except Exception, err:
|
|
pass
|
|
self.log.debug("Socket closed: %s" % sock_closed)
|
|
|
|
self.server.socket.close()
|
|
self.server.stop()
|
|
time.sleep(1)
|
|
|