# plugs/grep.py
#
#

__copyright__ = 'this file is in the public domain'

from gozerbot.commands import cmnds
from gozerbot.generic import waitforqueue
from gozerbot.plughelp import plughelp
from gozerbot.examples import examples
import getopt, re

plughelp.add('grep', ' grep elements of a pipeline')

def handle_grep(bot, ievent):
    """ grep the result list """
    if not ievent.inqueue:
        ievent.reply('use grep in a pipeline')
        return
    if not ievent.rest:
        ievent.reply('grep <txt>')
        return
    try:
        (options, rest) = getopt.getopt(ievent.args, 'riv')
    except getopt.GetoptError, ex:
        ievent.reply(str(ex))
        return
    result = waitforqueue(ievent.inqueue, 30)
    if not result:
        ievent.reply('no data to grep on')
        return
    doregex = False
    docasein = False
    doinvert = False
    for i, j in options:
        if i == '-r':
            doregex = True
        if i == '-i':
            docasein = True
        if i == '-v':
            doinvert = True
    res = []
    if doregex:
        try:
            if docasein:
                reg = re.compile(' '.join(rest), re.I)
            else:
                reg = re.compile(' '.join(rest))
        except Exception, ex:
            ievent.reply("can't compile regex: %s" % str(ex))
            return
        if doinvert:
            for i in result:
                if not re.search(reg, i):
                    res.append(i)
        else:
            for i in result:
                if re.search(reg, i):
                    res.append(i)
    else:
        if docasein:
            what = ' '.join(rest).lower()
        elif doinvert:
            what = ' '.join(rest)
        else:
            what = ievent.rest
        for i in result:
            if docasein:
                if what in str(i.lower()):
                    res.append(i)
            elif doinvert:
                if what not in str(i):
                    res.append(i)
            else:
                if what in str(i):
                    res.append(i)
    if not res:
        ievent.reply('no result')
    else:
        ievent.reply(res, dot=True)

cmnds.add('grep', handle_grep, ['USER', 'WEB', 'ANON', 'CLOUD'], threaded=True)
examples.add('grep', 'grep the output of a command', 'list | grep todo')