# -*- python -*-
'''
xpybar – xmobar replacement written in python
Copyright © 2014, 2015, 2016, 2017, 2018  Mattias Andrée (maandree@kth.se)

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