m + add ii

Signed-off-by: Mattias Andrée <maandree@kth.se>

1 files changed, 142 insertions, 0 deletions

diff --git a/src/plugins/ii.py b/src/plugins/ii.py
new file mode 100644
index 0000000..5ffb266
--- /dev/null
+++ b/src/plugins/ii.py
@@ -0,0 +1,142 @@
+# -*- python -*-
+'''
+xpybar – xmobar replacement written in python
+Copyright © 2014, 2015, 2016  Mattias Andrée (maandree@member.fsf.org)
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+'''
+
+
+class II:
+    '''
+    IRC interface, requires a running instance of ii <https://git.suckless.org/ii>
+    '''
+    
+    
+    def __init__(self, channel, prefix = None):
+        '''
+        Constructor
+        
+        @param  channel:str        The name of the server or the name of the
+                                   server followed by a slash and the name of
+                                   the channel including the she
+        @param  prefix:str?        The argument associated with ii's -i flag
+        '''
+        import pwd, os
+        if prefix is None:
+            prefix = pwd.getpwuid(os.getuid()).pw_dir + '/irc'
+        self.infile = '%s/%s/in' % (prefix, channel)
+        self.outfile = '%s/%s/out' % (prefix, channel)
+        self.log = []
+        self.time = '0000-00-00 00:00 '
+    
+    
+    def write(self, text):
+        '''
+        Write to the channel
+        
+        @param  text:str  The message or command
+        '''
+        text = (text + '\n').encode('utf-8')
+        with open(self.infile, 'wb') as file:
+            file.write(text)
+            file.flush()
+    
+    
+    def read(self):
+        '''
+        Fetch new messages from the channel
+        
+        @return  :list<str>  List of new messages
+        '''
+        with open(self.outfile, 'rb') as file:
+            text = file.read()
+        text = text.decode('utf-8', 'replace').split('\n')[:-1]
+        i = 0
+        for line in text:
+            if line[:17] >= self.time:
+                break
+            i += 1
+        text = text[i:]
+        i = 0
+        while i < len(text) and i < len(self.log):
+            if text[i] != self.log[i]:
+                break
+            i += 1
+        text = text[i:]
+        if len(text) == 0:
+            return text
+        last = text[-1][:17]
+        if last == self.time:
+            self.log.extend(text)
+            return text
+        self.log = []
+        self.time = last
+        i = 0
+        for line in text:
+            if line[:17] == self.time:
+                break
+            i += 1
+        self.log = text[i:]
+        return text
+    
+    
+    def wait(self, timeout = None):
+        '''
+        Wait until more messages are available
+        
+        @param  timeout:int?  The number of seconds to wait
+                              before return unconditionally
+        '''
+        import os
+        command = ['inotifywait', '-e', 'close_write,modify']
+        if timeout is not None:
+            command += ['-t', str(int(timeout + 0.5))]
+        command += ['--', self.outfile]
+        if 'PATH' in os.environ:
+            path = os.environ['PATH'].split('.')
+            for p in path:
+                p += '/pdeath'
+                if os.access(p, os.X_OK, effective_ids = True):
+                    command = [p, 'HUP'] + command
+                    break
+        spawn_read(*command)
+    
+    
+    @staticmethod
+    def list_channels(prefix = None):
+        '''
+        Fetch a list of all joined servers and channels
+        
+        @param   prefix:str?  The argument associated with ii's -i flag
+        @return  :list<list>  List of servers and channels
+        '''
+        import pwd, os
+        if prefix is None:
+            prefix = pwd.getpwuid(os.getuid()).pw_dir + '/irc'
+        ret = []
+        def isfifo(f):
+            try:
+                return os.path.stat.S_ISFIFO(os.stat('f').st_mode)
+            except:
+                return False
+        def recurse(d):
+            fs = os.listdir(d)
+            for f in fs:
+                f = d + '/' + f
+                if os.path.isfile(f + '/out') and isfifo(f + '/in'):
+                    ret.append(f[len(prefix) + 1:])
+        recurse(prefix)
+        return ret
+