From e3b933445fe4b18af6ec462fc40da5f482e447a0 Mon Sep 17 00:00:00 2001
From: Florent Le Coz <louiz@louiz.org>
Date: Fri, 23 Sep 2011 17:43:01 +0200
Subject: [teisenbe] first attempt at a plugin system.

---
 plugins/screen_detach.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 plugins/test.py          | 12 ++++++++++++
 2 files changed, 57 insertions(+)
 create mode 100644 plugins/screen_detach.py
 create mode 100644 plugins/test.py

(limited to 'plugins')

diff --git a/plugins/screen_detach.py b/plugins/screen_detach.py
new file mode 100644
index 00000000..6ebc77f2
--- /dev/null
+++ b/plugins/screen_detach.py
@@ -0,0 +1,45 @@
+import os
+import stat
+import pyinotify
+
+SCREEN_DIR = '/var/run/screen/S-%s' % (os.getlogin(),)
+
+class Plugin(BasePlugin):
+    def init(self):
+        self.timed_event = None
+        sock_path = None
+        self.thread = None
+        for f in os.listdir(SCREEN_DIR):
+            path = os.path.join(SCREEN_DIR, f)
+            if screen_attached(path):
+                sock_path = path
+                self.attached = True
+                break
+
+        # Only actually do something if we found an attached screen (assuming only one)
+        if sock_path:
+            wm = pyinotify.WatchManager()
+            wm.add_watch(sock_path, pyinotify.EventsCodes.ALL_FLAGS['IN_ATTRIB'])
+            self.thread = pyinotify.ThreadedNotifier(wm, default_proc_fun=HandleScreen(plugin=self))
+            self.thread.start()
+
+    def cleanup(self):
+        if self.thread:
+            self.thread.stop()
+
+    def update_screen_state(self, socket):
+        attached = screen_attached(socket)
+        if attached != self.attached:
+            self.attached = attached
+            status = 'available' if self.attached else 'away'
+            self.core.command_status(status)
+
+def screen_attached(socket):
+    return (os.stat(socket).st_mode & stat.S_IXUSR) != 0
+
+class HandleScreen(pyinotify.ProcessEvent):
+    def my_init(self, **kwargs):
+        self.plugin = kwargs['plugin']
+
+    def process_IN_ATTRIB(self, event):
+        self.plugin.update_screen_state(event.path)
diff --git a/plugins/test.py b/plugins/test.py
new file mode 100644
index 00000000..fc7aedc6
--- /dev/null
+++ b/plugins/test.py
@@ -0,0 +1,12 @@
+class Plugin(BasePlugin):
+    def init(self):
+        self.core.information("Plugin loaded")
+
+    def cleanup(self):
+        self.core.information("Plugin unloaded")
+
+    def on_message(self, message):
+        self.core.information("Test plugin received message: {}".format(message))
+
+    def command_plugintest(self, args):
+        self.core.information("Command! With args {}".format(args))
-- 
cgit v1.2.3


From f27556747896aeb891ce71cfdd0ac349d68c5b3d Mon Sep 17 00:00:00 2001
From: Florent Le Coz <louiz@louiz.org>
Date: Sat, 24 Sep 2011 22:26:31 +0200
Subject: [teisenbe] Use the imp module to import modules. Also add a simple
 translator module

---
 plugins/screen_detach.py |  1 +
 plugins/test.py          |  4 ++++
 plugins/translate.py     | 33 +++++++++++++++++++++++++++++++++
 3 files changed, 38 insertions(+)
 create mode 100644 plugins/translate.py

(limited to 'plugins')

diff --git a/plugins/screen_detach.py b/plugins/screen_detach.py
index 6ebc77f2..6ee96896 100644
--- a/plugins/screen_detach.py
+++ b/plugins/screen_detach.py
@@ -1,3 +1,4 @@
+from plugin import BasePlugin
 import os
 import stat
 import pyinotify
diff --git a/plugins/test.py b/plugins/test.py
index fc7aedc6..13ba1e9c 100644
--- a/plugins/test.py
+++ b/plugins/test.py
@@ -1,5 +1,9 @@
+from plugin import BasePlugin
+
 class Plugin(BasePlugin):
     def init(self):
+        self.add_command('plugintest', self.command_plugintest, 'Test command')
+        self.add_event_handler('message', self.on_message)
         self.core.information("Plugin loaded")
 
     def cleanup(self):
diff --git a/plugins/translate.py b/plugins/translate.py
new file mode 100644
index 00000000..625d78e7
--- /dev/null
+++ b/plugins/translate.py
@@ -0,0 +1,33 @@
+from plugin import BasePlugin
+import urllib.request
+from urllib.parse import urlencode
+import xhtml
+import json
+
+TARGET_LANG = 'en'
+
+def translate(s, target=TARGET_LANG, source=''):
+    f = urllib.request.urlopen('http://ajax.googleapis.com/ajax/services/language/translate', urlencode({ 'v': '1.0', 'q': s, 'langpair': '%s|%s' % (source, target) }))
+    response = json.loads(str(f.read(), 'utf-8'))['responseData']
+    return (response['translatedText'], response['detectedSourceLanguage'])
+
+class Plugin(BasePlugin):
+    def init(self):
+        self.add_event_handler('groupchat_message', self.on_groupchat_message)
+
+    def on_groupchat_message(self, message):
+        try:
+            room_from = message.getMucroom()
+            if message['type'] == 'error':
+                return
+
+            if room_from == 'poezio@kikoo.louiz.org':
+                nick_from = message['mucnick']
+                body = xhtml.get_body_from_message_stanza(message)
+                room = self.core.get_room_by_name(room_from)
+                text, lang = translate(body)
+                if lang != TARGET_LANG:
+                    room.add_message(text, nickname=nick_from)
+        except Exception as e:
+            import traceback
+            self.core.information("Exception in translator! %s" % (traceback.format_exc(),))
-- 
cgit v1.2.3