Move webide and terminado event handlers to lib/tfw/components

lib/tfw/components/source_code_event_handler.py

@@ -0,0 +1,103 @@
+from os.path import splitext, isfile, join, relpath
+from glob import glob
+
+from tfw.util import SupervisorMixin
+from tfw.event_handler_base import EventHandlerBase
+
+from tfw.config.logs import logging
+log = logging.getLogger(__name__)
+
+
+class FileManager:
+    def __init__(self, working_directory, selected_file=None):
+        self.exclude = ['__pycache__']
+        self._workdir = working_directory
+        self.filename = selected_file or self._relpath(self.files[0])
+        self.language = map_file_extension_to_language(self.filename)
+
+    def select_file(self, filename):
+        if not filename in self.files:
+           raise EnvironmentError('No such file in workdir!')
+        self.filename = filename
+        self.language = map_file_extension_to_language(self.filename)
+
+    @property
+    def files(self):
+        return [self._relpath(file) for file in glob(join(self._workdir, '**/*'), recursive=True)
+                if isfile(file) and
+                not any(word in file for word in self.exclude)]
+
+    @property
+    def file_contents(self):
+        with open(self._filepath(self.filename), 'r', errors='surrogateescape') as ifile:
+            return ifile.read()
+
+    @file_contents.setter
+    def file_contents(self, value):
+        with open(self._filepath(self.filename), 'w', errors='surrogateescape') as ofile:
+            ofile.write(value)
+
+    def _filepath(self, filename):
+        return join(self._workdir, filename)
+
+    def _relpath(self, filename):
+        return relpath(self._filepath(filename), start=self._workdir)
+
+
+class SourceCodeEventHandler(EventHandlerBase, SupervisorMixin):
+    def __init__(self, anchor, directory, process_name, selected_file=None):
+        super().__init__(anchor)
+        self.filemanager = FileManager(directory, selected_file=selected_file)
+        self.process_name = process_name
+
+        self.commands = {
+            'read': self.read,
+            'write': self.write,
+            'select': self.select
+        }
+
+    def read(self, data):
+        try: data['content'] = self.filemanager.file_contents
+        except PermissionError: data['content'] = 'You have no permission to open that file :('
+        except FileNotFoundError: data['content'] = 'This file was removed :('
+        except Exception: data['content'] = 'Failed to read file :('
+        return data
+
+    def write(self, data):
+        try: self.filemanager.file_contents = data['content']
+        except Exception: log.exception('Error writing file!')
+        self.restart_process()
+        return data
+
+    def select(self, data):
+        try: self.filemanager.select_file(data['filename'])
+        except EnvironmentError: log.exception('Failed to select file "{}"'.format(data['filename']))
+        return data
+
+    def attach_fileinfo(self, data):
+        data['filename'] = self.filemanager.filename
+        data['language'] = self.filemanager.language
+        data['files'] = self.filemanager.files
+
+    def handle_event(self, anchor, data_json):
+        data = data_json['data']
+        data_json['data'] = self.commands[data['command']](data)
+        self.attach_fileinfo(data)
+        return data_json
+
+
+def map_file_extension_to_language(filename):
+    language_map = {
+        '.py': 'python',
+        '.js': 'javascript',
+        '.html': 'html',
+        '.css': 'css',
+        '.java': 'java',
+        '.cpp': 'c_cpp',
+        '.hpp': 'c_cpp',
+        '.c': 'c_cpp',
+        '.h': 'c_cpp',
+        '.cs': 'csharp'
+    }
+    _, extension = splitext(filename)
+    return language_map.get(extension, 'text')

lib/tfw/components/terminado_event_handler.py

@@ -0,0 +1,17 @@
+from tfw.event_handler_base import EventHandlerBase
+from tfw.util import SupervisorMixin
+from tfw.config import tfwenv
+from tfw.config.logs import logging
+log = logging.getLogger(__name__)
+
+
+class TerminadoEventHandler(EventHandlerBase, SupervisorMixin):
+    def __init__(self, anchor, process_name):
+        super().__init__(anchor)
+        self.working_directory = tfwenv.TERMINADO_DIR
+        self.process_name = process_name
+        self.start_process()
+
+    def handle_event(self, anchor, data_json):
+        log.debug('TerminadoEventHandler received event for anchor {}'.format(anchor))
+        # TODO: wat do?

lib/tfw/components/terminado_mini_server.py

@@ -0,0 +1,33 @@
+from tornado.ioloop import IOLoop
+from tornado.web import Application
+from terminado import TermSocket, UniqueTermManager
+
+from tfw.config import tfwenv
+from tfw.config.logs import logging
+log = logging.getLogger(__name__)
+
+
+class TerminadoMiniServer:
+    def __init__(self, url, port, workdir, shellcmd):
+        self.port = port
+        self.application = Application(
+            [(
+                url,
+                TerminadoMiniServer.CORSTermSocket,
+                {'term_manager': UniqueTermManager(shell_command=shellcmd,
+                                                   term_settings={'cwd': workdir})}
+            )]
+        )
+
+    class CORSTermSocket(TermSocket):
+        def check_origin(self, origin):
+            return True
+
+    def listen(self):
+        self.application.listen(self.port)
+        IOLoop.instance().start()
+
+
+if __name__ == '__main__':
+    log.info('Terminado Mini Server listening on {}'.format(tfwenv.TERMINADO_PORT))
+    TerminadoMiniServer('/terminal', tfwenv.TERMINADO_PORT, tfwenv.TERMINADO_WD, ['bash']).listen()