baseimage-tutorial-framework/lib/tfw/builtins/terminal_event_handler.py

90 lines
2.8 KiB
Python
Raw Normal View History

# Copyright (C) 2018 Avatao.com Innovative Learning Kft.
# All Rights Reserved. See LICENSE file for details.
2019-06-10 13:32:45 +00:00
import logging
2019-06-28 13:11:02 +00:00
from tfw.networking import Scope
from tfw.components import BashMonitor
from tfw.components.terminado_mini_server import TerminadoMiniServer
from tfw.config import TFWENV
from tao.config import TAOENV
2019-06-28 13:11:02 +00:00
from .event_handler import EventHandler
LOG = logging.getLogger(__name__)
2019-06-28 13:11:02 +00:00
class TerminalEventHandler(EventHandler):
2018-04-18 17:44:26 +00:00
"""
Event handler responsible for managing terminal sessions for frontend xterm
sessions to connect to. You need to instanciate this in order for frontend
terminals to work.
2018-06-01 14:20:20 +00:00
This EventHandler accepts messages that have a data['command'] key specifying
2018-04-18 17:44:26 +00:00
a command to be executed.
2018-06-01 14:20:20 +00:00
The API of each command is documented in their respective handler.
2018-04-18 17:44:26 +00:00
"""
def __init__(self, key):
2018-04-18 17:44:26 +00:00
"""
:param key: key this EventHandler listens to
:param monitor: tfw.components.HistoryMonitor instance to read command history from
"""
2019-06-28 13:11:02 +00:00
super().__init__(key, scope=Scope.WEBSOCKET)
self._historymonitor = BashMonitor(self.server_connector, TFWENV.HISTFILE)
bash_as_user_cmd = ['sudo', '-u', TAOENV.USER, 'bash']
2018-06-01 15:19:58 +00:00
self.terminado_server = TerminadoMiniServer(
'/terminal',
TFWENV.TERMINADO_PORT,
TFWENV.TERMINADO_WD,
bash_as_user_cmd
2018-06-01 15:19:58 +00:00
)
self.commands = {
'write': self.write,
'read': self.read
}
self._historymonitor.start()
self.terminado_server.listen()
@property
def historymonitor(self):
return self._historymonitor
def handle_event(self, message):
try:
data = message['data']
message['data'] = self.commands[data['command']](data)
self.send_message(message)
except KeyError:
LOG.error('IGNORING MESSAGE: Invalid message received: %s', message)
2018-03-03 13:14:44 +00:00
def write(self, data):
2018-04-18 17:44:26 +00:00
"""
Writes a string to the terminal session (on the pty level).
Useful for pre-typing and executing commands for the user.
2018-06-01 14:20:20 +00:00
:param data: TFW message data containing 'value'
(command to be written to the pty)
2018-04-18 17:44:26 +00:00
"""
2018-05-31 12:08:29 +00:00
self.terminado_server.pty.write(data['value'])
return data
2018-03-03 21:46:14 +00:00
def read(self, data):
2018-04-18 17:44:26 +00:00
"""
Reads the history of commands executed.
2018-06-01 14:20:20 +00:00
:param data: TFW message data containing 'count'
(the number of history elements to return)
:return dict: message with list of commands in data['history']
2018-04-18 17:44:26 +00:00
"""
data['count'] = int(data.get('count', 1))
if self.historymonitor:
data['history'] = self.historymonitor.history[-data['count']:]
2018-03-03 21:46:14 +00:00
return data
def cleanup(self):
2019-05-31 11:10:33 +00:00
self.terminado_server.stop()
self.historymonitor.stop()