72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
|
import webbrowser
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import signal
|
||
|
|
from http.server import HTTPServer, SimpleHTTPRequestHandler
|
||
|
|
from socketserver import TCPServer
|
||
|
|
import threading
|
||
|
|
|
||
|
|
|
||
|
|
class QuietHTTPRequestHandler(SimpleHTTPRequestHandler):
|
||
|
|
|
||
|
|
def log_message(self, format, *args):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def log_request(self, code='-', size='-'):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
class DeTTECTEditor:
|
||
|
|
|
||
|
|
def __init__(self, port):
|
||
|
|
"""
|
||
|
|
Constructor of the DeTTECTEditor class. Sets the SIGTERM (clean quit) en SIGINT (Ctrl+C) handlers and the default variables.
|
||
|
|
:param port: The port for the webserver to listen on
|
||
|
|
"""
|
||
|
|
signal.signal(signal.SIGTERM, self._signal_handler)
|
||
|
|
signal.signal(signal.SIGINT, self._signal_handler)
|
||
|
|
self.port = port
|
||
|
|
self.httpd = None
|
||
|
|
|
||
|
|
def _signal_handler(self, signal, frame):
|
||
|
|
"""
|
||
|
|
Handles the termination of the application.
|
||
|
|
:param signum: Indicator of the termination signal
|
||
|
|
:param frame:
|
||
|
|
"""
|
||
|
|
print("Shutting down webserver")
|
||
|
|
self.httpd.server_close()
|
||
|
|
self.httpd.shutdown()
|
||
|
|
|
||
|
|
def _run_webserver(self):
|
||
|
|
"""
|
||
|
|
Starts the webserver on the given port.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
os.chdir('./editor/dist')
|
||
|
|
self.httpd = TCPServer(('', self.port), QuietHTTPRequestHandler)
|
||
|
|
|
||
|
|
print("Editor started at port %d" % self.port)
|
||
|
|
url = 'http://localhost:%d/dettect-editor' % self.port
|
||
|
|
|
||
|
|
if not os.getenv('DeTTECT_DOCKER_CONTAINER'):
|
||
|
|
print("Opening webbrowser: " + url)
|
||
|
|
webbrowser.open_new_tab(url)
|
||
|
|
else:
|
||
|
|
print("You can open the Editor on: " + url)
|
||
|
|
|
||
|
|
self.httpd.serve_forever()
|
||
|
|
except Exception as e:
|
||
|
|
print("Could not start webserver: " + str(e))
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
"""
|
||
|
|
Starts the Editor by starting a thread where the webserver runs in.
|
||
|
|
"""
|
||
|
|
thread = threading.Thread(target=self._run_webserver)
|
||
|
|
thread.start()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
print("Please use dettect.py for running the DeTT&CT Editor. Run 'python dettect.py e -h' for more information.")
|