working on multiple editor types

plugin_loader.py

@@ -71,7 +71,14 @@ class PluginLoader(Window):
         setattr(self.app, f"action_{action_name}", wrapper)
     
     def create_widget(self, widget_type: str, *args):
-        return getattr(textual.widgets, widget_type)(*args)
+        new_widget = getattr(textual.widgets, widget_type)(*args)
+    
+        def on_func(event_name: str, function):
+            self.event_handler.subscribe(new_widget, event_name, function)
+            new_widget.message_signal.subscribe(self, lambda e: self.event_handler.message_handler(new_widget, e))
+    
+        setattr(new_widget, "on", on_func)
+        return new_widget
     
     def run_on_message(self, widget, event, function):
         self.event_handler.subscribe(widget, event, function)