from ast import literal_eval from tornado.ioloop import IOLoop from tfw.components import IdeEventHandler, TerminalEventHandler, ProcessManagingEventHandler, BashMonitor from tfw.components import TerminalCommands, LogMonitoringEventHandler from tfw.networking import MessageSender, TFWServerConnector from tfw.config import TFWENV from tfw.config.logs import logging from tao.config import TAOENV LOG = logging.getLogger(__name__) def cenator(history): LOG.debug('User executed command: "%s"', history[-1]) MessageSender().send('JOHN CENA', f'You\'ve executed "{history[-1]}"') class TestCommands(TerminalCommands): """ Some example commands useful for debugging. Please remove from production code and inherit your own class from TerminalCommands if you need to define custom commands in your challenge. """ # pylint: disable=unused-argument, attribute-defined-outside-init, no-self-use def command_sendmessage(self, *args): """ Insert TFW message template as first argument if executed without args. Evaluate first argumen as a dict and send it to the frontend. This is useful for playing around with frontend APIs. """ if not args: message_template = """'{"key": "", "data": {"command": ""}}'""" TFWServerConnector().send_to_eventhandler({ 'key': 'shell', 'data': { 'command': 'write', 'shellcmd': f'sendmessage {message_template}' } }) else: TFWServerConnector().send(literal_eval(args[0])) def command_seppuku_tfw(self, *args): """ Restart tfw_server.py and event_handler_main.py. This can speed up development when combined with mounting volumes from host to container. """ seppuku = ('nohup sh -c "supervisorctl restart tfwserver event_handler_main" &> /dev/null & ' 'clear && echo "Committed seppuku! :)" && sleep infinity') uplink = TFWServerConnector() uplink.send_to_eventhandler({ 'key': 'shell', 'data': { 'command': 'write', 'shellcmd': f'{seppuku}\n' } }) uplink.send({ 'key': 'dashboard', 'data': { 'command': 'reload_frontend' } }) if __name__ == '__main__': ide = IdeEventHandler( # Web IDE backend key='ide', allowed_directories=[TFWENV.IDE_WD, TFWENV.WEBSERVICE_DIR], directory=TFWENV.IDE_WD, exclude=['*.pyc'], additional_watched_directories=[TFWENV.WEBSERVICE_DIR] ) terminal = TerminalEventHandler( # Web shell backend key='shell', monitor=BashMonitor(TFWENV.HISTFILE) ) processmanager = ProcessManagingEventHandler( # Handles 'deploy' button clicks key='processmanager', dirmonitor=ide.monitor, log_tail=2000 ) logmonitor = LogMonitoringEventHandler( # Sends live logs of webservice process to frontend 'webservice', log_tail=2000 ) eventhandlers = {ide, terminal, processmanager, logmonitor} terminal.historymonitor.subscribe_callback(cenator) commands = TestCommands(bashrc=f'/home/{TAOENV.USER}/.bashrc') terminal.historymonitor.subscribe_callback(commands.callback) try: IOLoop.instance().start() finally: for eh in eventhandlers: eh.cleanup()