Example #1
0
def writeout(botname, type, channel, txt, eventjson):
    """ output the watched event to the channel. """
    event = EventBase().load(eventjson)
    watchbot = getfleet().byname(botname)
    if not watchbot: watchbot = getfleet().makebot(type, botname)
    if watchbot:
         watchbot.outnocb(channel, txt, event=event)
Example #2
0
def handle_adminsave(bot, ievent):
    """ no arguments - boot the bot .. do some initialisation. """
    ievent.reply("saving mainconfig")
    getmainconfig().save()
    ievent.reply("saving fleet bots")
    getfleet().save()
    ievent.reply("saving all plugins")
    plugs.save()
    ievent.done()
Example #3
0
File: irc.py Project: code2u/jsb
def handle_broadcast(bot, ievent):
    """ arguments: <txt> - broadcast txt to all joined channels. """
    if not ievent.rest:
         ievent.missing('<txt>')
         return
    ievent.reply('broadcasting')
    getfleet().broadcast(ievent.rest)
    partyline.say_broadcast(ievent.rest)
    ievent.reply('done')
Example #4
0
def handle_adminsave(bot, ievent):
    """ no arguments - boot the bot ..  do some initialisation.  """
    ievent.reply("saving mainconfig")
    getmainconfig().save()
    ievent.reply("saving fleet bots")
    getfleet().save()
    ievent.reply("saving all plugins")
    plugs.save()
    ievent.done()
Example #5
0
def handle_broadcast(bot, ievent):
    """ arguments: <txt> - broadcast txt to all joined channels. """
    if not ievent.rest:
         ievent.missing('<txt>')
         return
    ievent.reply('broadcasting')
    getfleet().broadcast(ievent.rest)
    partyline.say_broadcast(ievent.rest)
    ievent.reply('done')
Example #6
0
File: alarm.py Project: code2u/jsb
 def alarmsay(self, item):
     """ say alarm txt """
     logging.warn("trying to deliver on %s" % str(item))
     bot = getfleet().byname(item.botname)
     if not bot: bot = getfleet().makebot(None, item.botname)
     if bot:
         if item.printto: bot.say(item.printto, "[%s] %s" % (item.nick, item.txt), speed=1)
         else: bot.say(item.nick, item.txt, speed=1)
     else: logging.warn("can't find %s bot in fleet" % item.botname)
     self.delete(item.idnr)
Example #7
0
File: bot.py Project: code2u/jsb
 def doreconnect(self):
     """ reconnect to the server. """
     cfg = cpy(self.cfg)
     botjid = cfg.user
     newbot = getfleet().makebot('sxmpp', cfg.name, config=cfg, showerror=True)
     if not newbot: raise Exception("can't make bot %s with config %s" % (self.cfg.name, self.cfg.tojson()))
     newbot.reconnectcount = self.reconnectcount
     newbot.start()
     newbot.joinchannels()
     getfleet().replace(botjid, newbot)
     return True
Example #8
0
def writeout(botname, type, channel, txt, eventjson):
    """ output the watched event to the channel. """
    event = EventBase().load(eventjson)
    global bots
    watchbot = None
    if botname in bots: watchbot = bots[botname]
    if not watchbot: watchbot = getfleet().byname(botname)
    if not watchbot: watchbot = getfleet().makebot(type, botname)
    if watchbot:
         if not botname in bots: bots[botname] = watchbot
         txt = watchbot.normalize(txt)
         txt = stripcolor(txt)
         watchbot.outnocb(channel, txt, event=event)
Example #9
0
def handle_reboot(bot, ievent):
    """ no arguments - reboot the bot. """
    ievent.reply("rebooting")
    #time.sleep(3)
    if ievent.rest == "cold": stateful = False
    else: stateful = True
    if stateful:
        if bot.type == "tornado": 
            callback = functools.partial(reboot_stateful, bot, ievent, getfleet(), partyline)
            bot.server.io_loop.add_callback(callback)
        else: mainhandler.put(0, reboot_stateful, bot, ievent, getfleet(), partyline)
    else:
        getfleet().exit()
        mainhandler.put(0, reboot)
Example #10
0
File: core.py Project: code2u/jsb
def handle_reboot(bot, ievent):
    """ no arguments - reboot the bot. """
    if bot.isgae:
        ievent.reply("this command doesn't work on the GAE")
        return
    ievent.reply("rebooting")
    time.sleep(3)
    if ievent.rest == "cold": stateful = False
    else: stateful = True
    if stateful:
        if bot.type == "tornado": bot.ioloop.add_callback(lambda: reboot_stateful(bot, ievent, getfleet(), partyline))
        else: mainhandler.put(0, reboot_stateful, bot, ievent, getfleet(), partyline)
    else:
        getfleet().exit()
        mainhandler.put(0, reboot)
Example #11
0
 def _resume(self, data, reto=None):
     """ resume a party line connection after reboot. """
     fleet = getfleet()
     for i in data['partyline']:
         logging.warn("partyline - resuming %s" % i)
         bot = fleet.byname(i['botname'])
         if not bot:
             logging.error("partyline - can't find bot")
             continue
         sock = socket.fromfd(i['fileno'], socket.AF_INET,
                              socket.SOCK_STREAM)
         sock.setblocking(1)
         nick = i['nick']
         userhost = i['userhost']
         channel = i['channel']
         if not bot:
             logging.error("partyline - can't find %s bot in fleet" %
                           i['botname'])
             continue
         self.socks.append({
             'bot': bot,
             'sock': sock,
             'nick': nick,
             'userhost': userhost,
             'channel': channel,
             'silent': i['silent']
         })
         bot._dccresume(sock, nick, userhost, channel)
         if reto: self.say_nick(nick, 'rebooting done')
Example #12
0
def fleet_add(bot, ievent):
    """ add a fleet bot. """
    try:
        name, type, server, nick = ievent.rest.split()
    except ValueError:
        ievent.missing("<name> <type> <server>|<botjid> <nick>|<passwd>")
        return
    bots = ievent.rest.split()
    fleet = getfleet()
    bot = fleet.byname(name)
    if bot:
        event.reply("%s bot already exists" % name)
        return
    cfg = Config('fleet' + os.sep + stripname(name) + os.sep + 'config')
    cfg.disable = 0
    if type == "irc":
        cfg.port = 6667
        cfg.server = server
        cfg.nick = nick
    elif type in ["xmpp", "sxmpp"]:
        cfg.port = 4442
        cfg.host = server
        try:
            n, serv = cfg.host.split("@")
        except (ValueError, TypeError):
            pass
        cfg.server = serv
        cfg.password = nick
    cfg.save()
    bot = fleet.makebot(type, name, cfg)
    ievent.reply('enabled and started %s bot - %s' % (name, cfg.filename))
    start_new_thread(bot.start, ())
Example #13
0
 def _listen(self):
     """ listen for udp messages .. /msg via bot"""
     if not cfg['udp']: return
     fleet = getfleet()
     for botname in cfg['udpbots']:
         if not fleet.byname(botname): logging.info("udp - can't find %s bot" % botname)
     try:
         fleet.startok.wait(5)
         logging.warn('udp listening on %s %s' % (cfg['udphost'], cfg['udpport']))
         self.sock.bind((cfg['udphost'], cfg['udpport']))
         self.stop = 0
     except IOError:
         handle_exception()
         self.sock = None
         self.stop = 1
         return
     # loop on listening udp socket
     while not self.stop:
         try: input, addr = self.sock.recvfrom(64000)
         except socket.timeout: continue
         except Exception, ex:
             try: (errno, errstr) = ex
             except ValueError: errno = 0 ; errstr = str(ex)
             if errno == 4: logging.warn("udp - %s - %s" % (self.name, str(ex))) ; break
             if errno == 35: continue
             else: handle_exception() ; break
         if self.stop: break
         self.queue.put((input, addr))
Example #14
0
 def alarmsay(self, item):
     """ say alarm txt """
     logging.warn("trying to deliver on %s" % str(item))
     bot = getfleet().byname(item.botname)
     if not bot: bot = getfleet().makebot(None, item.botname)
     if bot:
         if bot.type == "irc": bot.connectok.wait()
         if item.printto:
             bot.say(item.printto,
                     "[%s] %s" % (item.nick, item.txt),
                     speed=1)
         else:
             bot.say(item.nick, item.txt, speed=1)
     else:
         logging.warn("can't find %s bot in fleet" % item.botname)
     self.delete(item.idnr)
Example #15
0
def handle_relay(bot, event):
    """ arguments: [<botname>] [<type>] <target> .. open a relay to a user. all input from us will be relayed. """
    try:
        (botname, type, target) = event.args
    except ValueError:
        try:
            botname = bot.cfg.name
            (type, target) = event.args
        except ValueError:
            try:
                botname = bot.cfg.name
                type = bot.cfg.bottype
                target = event.args[0]
            except IndexError:
                event.missing('[<botname>] [<bottype>] <target>')
                return
    origin = unicode((bot.cfg.name, event.channel))
    if not getfleet().byname(botname):
        event.reply("no %s bot in fleet." % botname)
        return
    if not relay.data.has_key(origin): relay.data[origin] = []
    try:
        if not [type, target] in relay.data[origin]:
            relay.data[origin].append([botname, type, target])
            relay.save()
    except KeyError:
        relay.data[origin] = [
            [botname, type, target],
        ]
        relay.save()
    event.done()
Example #16
0
def fleet_cmnd(bot, ievent):
    """ co cmnd on fleet bot(s). """
    try:
        (name, cmndtxt) = ievent.rest.split(' ', 1)
    except ValueError:
        ievent.missing("<name> <cmndstring>")
        return
    fleet = getfleet()
    if name == "all": do = fleet.list()
    else: do = [
        name,
    ]
    for botname in do:
        bot = fleet.byname(botname)
        if not bot:
            ievent.reply("%s bot is not in fleet" % botname)
            return
        result = bot.docmnd(ievent.userhost,
                            ievent.channel,
                            cmndtxt,
                            wait=1,
                            nooutput=True)
        if result: res = waitforqueue(result.outqueue, 60000)
        else: ievent.reply("no result")
        ievent.reply("[%s] %s" % (botname, ", ".join(res)))
    ievent.reply("done")
Example #17
0
def plusscan(skip=False):
    global teller
    teller += 1
    if teller % 5 != 0: return
    logging.warn("running plus scan")
    fleet = getfleet()
    for id, channels in state.data.ids.iteritems():
        if not id in state.data.seen: state.data.seen[id] = []
        for botname, chan in channels:
            try:
                res = getplus(id)
                if not res:
                    logging.warn("no result from %s" % id)
                    continue
                bot = fleet.byname(botname)
                if bot:
                    todo = []
                    for r in res:
                        stamp = uuid.uuid3(uuid.NAMESPACE_URL, str(r)).hex
                        if stamp not in state.data.seen[id]:
                            state.data.seen[id].append(stamp)
                            todo.append(r)
                    if todo: bot.say(chan, "new plus update: ", todo)
                else: logging.warn("no %s bot in fleet" % botname)
            except AttributeError, ex:
                logging.error(str(ex))
            except Exception, ex:
                handle_exception()
Example #18
0
def forwardoutcb(bot, event):
    """ forward callback. """
    e = cpy(event)
    logging.debug("forward - cbtype is %s - %s" % (event.cbtype, event.how))
    e.forwarded = True
    e.source = bot.jid
    e.botname = bot.server or bot.name
    if not event.chan: event.bind(bot)
    if event.chan: e.allowwatch = event.chan.data.allowwatch
    fleet = getfleet()
    for jid in forward.data.channels[event.channel.lower()]:
        logging.info("forward - sending to %s" % jid)
        if jid == "twitter":
            try:
                postmsg(forward.data.outs[jid], e.txt)
            except Exception, ex:
                handle_exception()
            continue
        outbot = fleet.getfirstjabber(bot.isgae)
        if not outbot and bot.isgae:
            outbot = fleet.makebot('xmpp', 'forwardbot')
        if outbot:
            e.source = outbot.jid
            txt = outbot.normalize(e.tojson())
            txt = stripcolor(txt)
            #txt = e.tojson()
            container = Container(outbot.jid, txt)
            outbot.outnocb(jid, container.tojson())
        else:
            logging.info("forward - no xmpp bot found in fleet".upper())
Example #19
0
def reboot_stateful(bot, ievent, fleet, partyline):
    """ reboot the bot, but keep the connections (IRC only). """
    logging.warn("reboot - doing statefull reboot")
    session = {'bots': {}, 'name': bot.cfg.name, 'channel': ievent.channel, 'partyline': []}
    fleet = getfleet()
    for i in fleet.bots:
        logging.warn("reboot - updating %s" % i.cfg.name)
        data = i._resumedata()
        if not data: continue
        session['bots'].update(data)
        if i.type == "sxmpp": i.exit() ; continue
        if i.type == "tornado":
            i.exit()
            time.sleep(0.1)
            for socketlist in i.websockets.values():
                for sock in socketlist: sock.stream.close()
    session['partyline'] = partyline._resumedata()
    sfile, sessionfile = tempfile.mkstemp('-session', 'jsb-', text=True)
    logging.warn("writing session file %s" % sessionfile)
    json.dump(session, open(sessionfile, "w"))
    args = []
    skip = False
    for a in sys.argv[1:]:
        if skip: skip = False ; continue
        if a == "-r": skip = True ; continue
        args.append(a)
    os.execl(sys.argv[0], sys.argv[0], '-r', sessionfile, *args)
Example #20
0
def topicset(state):
    retval = None
    fleet = getfleet()
    for botname in fleet.list():
        bot = fleet.byname(botname)
        if testing: bot.say('#nurdbottest',botname)
        if bot:
            if not testing: topicdata = bot.gettopic('#nurds')
            if testing: topicdata = bot.gettopic('#nurdbottest')
#            time.sleep(5)
#            if not testing: topicdata = bot.gettopic('#nurds')
#            if testing: topicdata = bot.gettopic('#nurdbottest')
#            if testing: bot.say('#nurdbottest',str(topicdata))
#            if not topicdata and not testing: bot.say('#nurds',"can't get topic data, but the status should be toggled to "+statusStr(state)+'!') ; return
            if not topicdata: bot.say('#nurdbottest',"can't get topic data, but the status should be toggled to "+statusStr(state)+'!') ; return
    splitted = topicdata[0].split(' | ')
    statusline = splitted[0]
    if 'Space is ' in statusline:
        del splitted[0]
    if state:
        splitted.insert(0,'Space is OPEN')
        retval = True
    else:
        splitted.insert(0,'Space is CLOSED')
        retval = False
    newtopic = ' | '.join(splitted)
    for botname in fleet.list():
        bot = fleet.byname(botname)
        if bot and newtopic != topicdata[0]:
            if not testing: bot.settopic('#nurds', newtopic)
            if testing: bot.settopic('#nurdbottest', newtopic)
    return retval
Example #21
0
def doorsense_output(msg):
    '''The listener code for the sensor server.'''
    pswd = tcppassword.data['password']
    if msg[:len(pswd)] == pswd:
        msg = msg[len(pswd):]
        rsensor, rvalue = tuple(msg.split(':'))
        fleet = getfleet()
        for botname in fleet.list():
            bot = fleet.byname(botname)
            if bot:
                if not testing: bot.say('#nurds',rsensor+' is '+rvalue)
                bot.say('#nurdbottest',rsensor+' is '+rvalue)
        for type in sensorlist.data:
            for n, sensor in enumerate(sensorlist.data[type]):
                if rsensor.lower() == sensor['name'].lower():
                    try:
                        if type == 'door_locked': 
                            if rvalue.lower() == 'true': sensor['value']=True
                            elif rvalue.lower() == 'false': sensor['value']=False
                            else: sensor['value']=None
                        else: sensor['value']=float(rvalue)
                    except:
                        pass
                    sensorlist.data[type][n] = sensor
                    sensorlist.save()
                    apiupdate()
                    if type == 'door_locked': statuscheck()
Example #22
0
def askcallback(bot, event):
    """ this is the callbacks that handles the responses to questions. """
    sendto = questions.data[event.userhost]
    jid = []
    channels = []
    try:
        (printto, txt) = event.txt.split(':', 1)
    except ValueError:
        printto = False
        txt = event.txt
    txt = txt.strip()
    done = []
    fleet = getfleet()
    for botname, type, userhost, channel in sendto:
        if not printto or userhost != printto: continue
        askbot = fleet.makebot(type)
        if not askbot: askbot = fleet.makebot('xmpp', 'askxmppbot')
        logging.debug("ask - %s %s %s %s" % (botname, type, userhost, channel))
        if askbot:
            for jid in channel:
                askbot.say(channel, "%s says: %s" % (event.userhost, txt))
        else:
            logging.warn("ask - can't find %s bot in fleet" % type)
            continue
        try:
            questions.data[event.userhost].remove(
                [botname, type, userhost, channel])
            questions.save()
        except ValueError:
            pass
        done.append(channel)
        break
    if done: event.reply('answer sent to ', done)
Example #23
0
def handle_fleetconnect(bot, ievent):
    """ arguments: <botname> - connect a fleet bot to it's server. """
    try:
        botname = ievent.args[0]
    except IndexError:
        ievent.missing('<botname>')
        return
    try:
        fleet = getfleet()
        fleetbot = fleet.byname(botname)
        if fleetbot:
            start_new_thread(fleetbot.connect, ())
            ievent.reply('%s connect thread started' % botname)
        else:
            ievent.reply("can't connect %s .. trying enable" % botname)
            cfg = Config('fleet' + os.sep + stripname(botname) + os.sep +
                         'config')
            cfg['disable'] = 0
            if not cfg.name: cfg.name = botname
            cfg.save()
            bot = fleet.makebot(cfg.type, cfg.name, cfg)
            if bot:
                ievent.reply('enabled and started %s bot' % botname)
                start_new_thread(bot.start, ())
            else:
                ievent.reply("can't make %s bot" % cfg.name)
    except Exception, ex:
        ievent.reply(str(ex))
Example #24
0
File: relay.py Project: code2u/jsb
def relaycallback(bot, event):
    """ this is the callbacks that handles the responses to questions. """
    # determine where the event came from
    e = cpy(event)
    origin = e.channel
    e.isrelayed = True
    e.headlines = True
    try:
        # loop over relays for origin
        for botname, type, target in relay.data[origin]:
            try:
                logging.debug('trying relay of %s to (%s,%s)' % (origin, type, target))
                #if target == origin: continue
                # tests to prevent looping
                if botname == bot.cfg.name and origin == target: continue
                # check whether relay is blocked
                if block.data.has_key(origin):
                    if [botname, type, target] in block.data[origin]: continue
                # retrieve the bot from fleet (based on type)
                fleet = getfleet()
                outbot = fleet.byname(botname)
                if not outbot: outbot = fleet.makebot(type, botname)
                if outbot:
                    logging.info('relay - outbot found - %s - %s' % (outbot.cfg.name, outbot.type))
                    # we got bot .. use it to send the relayed message
                    if e.nick == bot.cfg.nick: txt = "[!] %s" % e.txt
                    else: txt = "[%s] %s" % (e.nick, e.txt)
                    if event: t = "[%s]" % outbot.cfg.nick
                    logging.debug("relay - sending to %s (%s)" % (target, outbot.cfg.name)) 
                    txt = stripcolor(txt)
                    outbot.outnocb(target, txt, "normal", event=e)
                else: logging.info("can't find bot for (%s,%s,%s)" % (botname, type, target))
            except Exception, ex: handle_exception()
    except KeyError: pass
Example #25
0
File: fleet.py Project: code2u/jsb
def fleet_enable(bot, ievent):
    """ arguments: <list of botnames> - enable a fleet bot. """
    if not ievent.rest:
        ievent.missing("<list of botnames>")
        return
    bots = ievent.rest.split()
    fleet = getfleet()
    for name in bots:
        bot = fleet.byname(name)
        if bot:
            bot.cfg.load()
            bot.cfg['disable'] = 0
            if not bot.cfg.name: bot.cfg.name = name
            bot.cfg.save()
            ievent.reply('enabled %s' % name)
            #start_new_thread(bot.connect, ())
        elif name in fleet.avail():
            cfg = Config('fleet' + os.sep + stripname(name) + os.sep + 'config')
            cfg['disable'] = 0
            if not cfg.name: cfg.name = name
            cfg.save()
            bot = fleet.makebot(cfg.type, cfg.name, cfg)
            if not bot: ievent.reply("can't make %s bot - %s" % (cfg.name, cfg.type)) ; return
            ievent.reply('enabled %s bot' % name)
            #start_new_thread(bot.start, ())
        else: ievent.reply('no %s bot in fleet' % name)
Example #26
0
def plusscan(skip=False):
    global teller
    teller += 1
    if teller % 5 != 0: return 
    logging.warn("running plus scan")
    fleet = getfleet()
    for id, channels in state.data.ids.iteritems():
        if not id in state.data.seen: state.data.seen[id] = []
        for botname, chan in channels:
            try:
                res = getplus(id)
                if not res: logging.warn("no result from %s" % id) ; continue
                bot = fleet.byname(botname)
                if bot:
                    todo = []
                    for r in res:
                        stamp = uuid.uuid3(uuid.NAMESPACE_URL, str(r)).hex
                        if stamp not in state.data.seen[id]:
                            state.data.seen[id].append(stamp)
                            todo.append(r)
                    if todo: bot.say(chan, "new plus update: " , todo)
                else: logging.warn("no %s bot in fleet" % botname)
            except AttributeError, ex: logging.error(str(ex))
            except Exception, ex: handle_exception()
    state.save()
Example #27
0
File: udp.py Project: code2u/jsb
 def _listen(self):
     """ listen for udp messages .. /msg via bot"""
     if not cfg['udp']: return
     fleet = getfleet()
     for botname in cfg['udpbots']:
         if not fleet.byname(botname): logging.info("udp - can't find %s bot" % botname)
     try:
         fleet.startok.wait(5)
         logging.warn('udp listening on %s %s' % (cfg['udphost'], cfg['udpport']))
         self.sock.bind((cfg['udphost'], cfg['udpport']))
         self.stop = 0
     except IOError:
         handle_exception()
         self.sock = None
         self.stop = 1
         return
     # loop on listening udp socket
     while not self.stop:
         try: input, addr = self.sock.recvfrom(64000)
         except socket.timeout: continue
         except Exception, ex:
             try: (errno, errstr) = ex
             except ValueError: errno = 0 ; errstr = str(ex)
             if errno == 4: logging.warn("udp - %s - %s" % (self.name, str(ex))) ; break
             if errno == 35: continue
             else: handle_exception() ; break
         if self.stop: break
         self.queue.put((input, addr))
Example #28
0
File: fleet.py Project: code2u/jsb
def fleet_add(bot, ievent):
    """ arguments: <name> <type> <server>|<botjid> <nick>|<passwd> - add a newly created bot to the fleet. """
    try:
        name, type, server, nick = ievent.rest.split()
    except ValueError: ievent.missing("<name> <type> <server>|<botjid> <nick>|<passwd>") ; return
    type = type.lower()
    fleet = getfleet()
    bot = fleet.byname(name)
    if bot: event.reply("%s bot already exists" % name) ; return
    cfg = Config('fleet' + os.sep + stripname(name) + os.sep + 'config')
    cfg.disable = 0
    if type == "irc":
        cfg.port = 6667
        cfg.server = server
        cfg.nick = nick
    elif type in ["xmpp", "sxmpp"]:
        cfg.port = 4442
        cfg.host = server
        try: n, serv = cfg.host.split("@")
        except (ValueError, TypeError): pass
        cfg.server = serv
        cfg.password = nick
    cfg.save()
    bot = fleet.makebot(type, name, cfg)
    fleet.addbot(bot)
    if bot:
        ievent.reply('enabled and started %s bot - %s' % (name, cfg.filename))
        start_new_thread(bot.start, ())
    else: ievent.reply("can't make %s bot" % cfg.name)
Example #29
0
def scan(skip=False):
    global teller, dostop
    if dostop: return
    teller += 1
    try: do = int(cfg.sleep)
    except ValueError: do = 5
    if do < 1: do = 5
    if teller % do != 0: return 
    logging.info("running")
    fleet = getfleet()
    todo = sync()
    if not todo: logging.info("nothing todo") ; return
    for b in todo:
        uid = str(b.user_id)
        if not uid in state.data.ids: logging.warn("we don't follow id %s" % uid) ; continue
        for channel in state.data.ids[uid]:
            if dostop: return
            botname, chan = channel
            bot = fleet.byname(botname)
            if bot:
                if b.post_id: url = ("http://stackoverflow.com/questions/%s" % b.post_id) or "no url found"
                bot.say(chan, "*%s* %s - *%s* - %s - %s - %s (%s)" % (state.data.names[uid].upper(), b.action, b.description, url, time.ctime(b.creation_date), b.detail or "no detail", b.post_type))
                if b.action == "answered":
                    aa = getanswers(b.post_id)
                    if aa:
                        a = aa[-1]
                        try: body = a['body']
                        except KeyError: continue
                        (urls, c) = geturls(body)
                        if c: bot.say(chan, u"> " + c)
                        else: bot.say(chan, "can't find answers")
                        if urls: bot.say(chan, "urls: %s" % " -=- ".join(urls))
            else: logging.warn("no %s bot in fleet" % botname)
Example #30
0
def topicset(state):
    retval = None
    try:
        fleet = getfleet()
        for botname in fleet.list():
             bot = fleet.byname(botname)
             if bot:
                 topicdata = bot.gettopic('#nurds')
        splitted = topicdata[0].split(' | ')
        statusline = splitted[0]
        if 'Space is ' in statusline:
            del splitted[0]
        if state:
            splitted.insert(0,'Space is OPEN')
            retval = True
        else:
            splitted.insert(0,'Space is CLOSED')
            retval = False
        newtopic = ' | '.join(splitted)
        for botname in fleet.list():
            bot = fleet.byname(botname)
            if bot and newtopic != topicdata[0]:
                if not testing: bot.settopic('#nurds', newtopic)
#            if testing: bot.settopic('#nurdbottest', newtopic)
    except:
        pass
    return retval
Example #31
0
 def work(self, botname, type, channel, entries, url, *args, **kwargs):
     try:
         item = self.byurl(url)
         name = item.data.name
         try:
             fleet = getfleet()
             bot = fleet.byname(botname)
             if not bot and type:
                 bot = fleet.makebot(type, botname)
             if not bot:
                 bot = fleet.makebot('xmpp', botname)
         except NoSuchBotType, ex: logging.warn("hubbub - %s" % str(ex)) ; return
         if not bot: logging.error("hubbub - can't find %s bot in fleet" % type) ; return
         res2 = entries
         if not res2: logging.info("no updates for %s (%s) feed available" % (item.data.name, channel)) ; return
         if item.markup.get(jsonstring([name, channel]), 'reverse-order'): res2 = res2[::-1]
         if item.markup.get(jsonstring([name, channel]), 'all-lines'):
             for i in res2:
                 response = self.makeresponse(name, [i, ], channel)
                 try: bot.say(channel, response)
                 except Exception, ex: handle_exception()
         else:
             sep =  item.markup.get(jsonstring([name, channel]), 'separator')
             if sep: response = self.makeresponse(name, res2, channel, sep=sep)
             else: response = self.makeresponse(name, res2, channel)
             bot.say(channel, response)
     except Exception, ex: handle_exception()
Example #32
0
File: ask.py Project: code2u/jsb
def handle_ask(bot, event):
    """ 
        arguments: <subject> <question> - this command lets you ask a question that gets dispatched to jabber 
        users that have registered themselves for that particular subject. 

    """
    try: subject, question = event.rest.split(' ', 1)
    except ValueError:
        event.missing('<subject> <question>')
        return
    try: expertslist = experts.data[subject]
    except KeyError:
        if '@' in subject: expertslist = [subject, ]
        else: expertslist = [defaultJID, ]
    fleet = getfleet()
    xmppbot = fleet.getfirstjabber()
    if xmppbot:
        for expert in expertslist: xmppbot.say(expert, "%s (%s) asks you: %s" % (event.userhost, bot.cfg.name, question))
    else:
        logging.warn("ask - can't find jabber bot in fleet")
        return
    asker = event.userhost
    for expert in expertslist:
        if not questions.data.has_key(expert): questions.data[expert] = []
        questions.data[expert].append([bot.cfg.name, bot.type, event.userhost, event.channel])
    questions.save()
    event.reply('question is sent to %s' % ' .. '.join(expertslist))
Example #33
0
def fleet_enable(bot, ievent):
    """ arguments: <list of botnames> - enable a fleet bot. """
    if not ievent.rest:
        ievent.missing("<list of botnames>")
        return
    bots = ievent.rest.split()
    fleet = getfleet()
    for name in bots:
        bot = fleet.byname(name)
        if bot:
            bot.cfg.load_config()
            bot.cfg['disable'] = 0
            if not bot.cfg.name: bot.cfg.name = name
            bot.cfg.save()
            ievent.reply('enabled %s' % name)
            #start_new_thread(bot.connect, ())
        elif name in fleet.avail():
            cfg = Config('fleet' + os.sep + stripname(name) + os.sep +
                         'config')
            cfg['disable'] = 0
            if not cfg.name: cfg.name = name
            cfg.save()
            bot = fleet.makebot(cfg.type, cfg.name, cfg)
            if not bot:
                ievent.reply("can't make %s bot - %s" % (cfg.name, cfg.type))
                return
            ievent.reply('enabled %s bot' % name)
            #start_new_thread(bot.start, ())
        else:
            ievent.reply('no %s bot in fleet' % name)
Example #34
0
def forwardoutcb(bot, event): 
    """ forward callback. """
    e = cpy(event)
    logging.debug("forward - cbtype is %s - %s" % (event.cbtype, event.how))
    e.forwarded = True
    e.source = bot.cfg.user
    e.botname = bot.cfg.server or bot.cfg.name
    if not event.chan: event.bind(bot)
    if event.chan: e.allowwatch = event.chan.data.allowwatch
    fleet = getfleet()
    for jid in forward.data.channels[event.channel.lower()]:
        if not "@" in jid: logging.error("forward - %s is not a valid JID" % jid) ; continue
        logging.info("forward - sending to %s" % jid)
        if jid == "twitter":
            try: postmsg(forward.data.outs[jid], e.txt)
            except Exception, ex: handle_exception()
            continue
        outbot = fleet.getfirstjabber(bot.isgae)
        if not outbot and bot.isgae: outbot = fleet.makebot('xmpp', 'forwardbot')
        if outbot:
            e.source = outbot.cfg.user
            txt = outbot.normalize(e.tojson())
            txt = stripcolor(txt)
            #txt = e.tojson()
            container = Container(outbot.cfg.user, txt)
            outbot.outnocb(jid, container.tojson()) 
        else: logging.info("forward - no xmpp bot found in fleet".upper())
Example #35
0
File: ask.py Project: code2u/jsb
def askcallback(bot, event):
    """ this is the callbacks that handles the responses to questions. """
    sendto = questions.data[event.userhost]
    jid = []
    channels = []
    try: (printto, txt) = event.txt.split(':', 1)
    except ValueError: printto = False ; txt = event.txt
    txt = txt.strip()
    done = []
    fleet = getfleet()
    for botname, type, userhost, channel in sendto:
        if not printto or userhost != printto: continue
        askbot = fleet.makebot(type)
        if not askbot: askbot = fleet.makebot('xmpp', 'askxmppbot')
        logging.debug("ask - %s %s %s %s" % (botname, type, userhost, channel))
        if askbot:
            for jid in channel: askbot.say(channel, "%s says: %s" % (event.userhost, txt))
        else: logging.warn("ask - can't find %s bot in fleet" % type) ; continue
        try:
            questions.data[event.userhost].remove([botname, type, userhost, channel])
            questions.save()
        except ValueError: pass
        done.append(channel)
        break
    if done: event.reply('answer sent to ', done)
Example #36
0
File: fleet.py Project: code2u/jsb
def handle_fleetconnect(bot, ievent):
    """ arguments: <botname> - connect a fleet bot to it's server. """
    try: botname = ievent.args[0]
    except IndexError:
        ievent.missing('<botname>')
        return
    try:
        fleet = getfleet()
        fleetbot = fleet.byname(botname)
        if fleetbot:
            start_new_thread(fleetbot.connect, ())
            ievent.reply('%s connect thread started' % botname)
        else:
            ievent.reply("can't connect %s .. trying enable" % botname)
            cfg = Config('fleet' + os.sep + stripname(botname) + os.sep + 'config')
            cfg['disable'] = 0
            if not cfg.name: cfg.name = botname
            cfg.save()
            bot = fleet.makebot(cfg.type, cfg.name, cfg)
            if bot:
                ievent.reply('enabled and started %s bot' % botname)
                start_new_thread(bot.start, ())
            else: ievent.reply("can't make %s bot" % cfg.name)
    except Exception, ex:
        ievent.reply(str(ex))
Example #37
0
File: fleet.py Project: code2u/jsb
def fleet_cmnd(bot, ievent):
    """ arguments: <botname> <cmndstring> - do cmnd on fleet bot(s). """
    try:
        (name, cmndtxt) = ievent.rest.split(' ', 1)
    except ValueError: ievent.missing("<botname> <cmndstring>") ; return
    fleet = getfleet()
    if name == "all": fleet.cmndall(ievent, cmndtxt)
    else: fleet.cmnd(ievent, name, cmndtxt)
Example #38
0
def handle_reboot(bot, ievent):
    """ no arguments - reboot the bot. """
    ievent.reply("rebooting")
    #time.sleep(3)
    if ievent.rest == "cold": stateful = False
    else: stateful = True
    if stateful:
        if bot.type == "tornado":
            callback = functools.partial(reboot_stateful, bot, ievent,
                                         getfleet(), partyline)
            bot.server.io_loop.add_callback(callback)
        else:
            mainhandler.put(0, reboot_stateful, bot, ievent, getfleet(),
                            partyline)
    else:
        getfleet().exit()
        mainhandler.put(0, reboot)
Example #39
0
File: fleet.py Project: code2u/jsb
def handle_fleetdel(bot, ievent):
    """ arguments: <botname> - delete bot from fleet. """
    try: name = ievent.args[0]
    except IndexError:
        ievent.missing('<name>')
        return
    try:
        if getfleet().delete(name): ievent.reply('%s deleted' % name)
        else: ievent.reply('%s delete failed' % name)
    except Exception, ex: ievent.reply(str(ex))
Example #40
0
def handle_xmppinvite(bot, event):
    """ arguments: <list of JIDs> - invite (subscribe to) a different user. """
    if not event.rest:
        event.missing("<list of JIDs>")
        return
    bot = getfleet().getfirstjabber()
    if bot:
        for jid in event.args: bot.invite(jid)
        event.done()
    else: event.reply("can't find jabber bot in fleet")
Example #41
0
def reboot_stateful(bot, ievent, fleet, partyline):
    """ reboot the bot, but keep the connections (IRC only). """
    logging.warn("reboot - doing statefull reboot")
    session = {'bots': {}, 'name': bot.name, 'channel': ievent.channel, 'partyline': []}
    for i in getfleet().bots:
        logging.warn("reboot - updating %s" % i.name)
        data = i._resumedata()
        if not data: continue
        session['bots'].update(data)
        if i.bottype == "sxmpp": i.exit()
    session['partyline'] = partyline._resumedata()
    sessionfile = tempfile.mkstemp('-session', 'jsb-')[1]
    json.dump(session, open(sessionfile, 'w'))
    getfleet().save()
    args = []
    if len(sys.argv) > 1:
        os.execl(sys.argv[0], sys.argv[0], '-r', sessionfile, *sys.argv[1:])
    else:
        os.execl(sys.argv[0], sys.argv[0], '-r', sessionfile)
Example #42
0
File: udp.py Project: code2u/jsb
 def dosay(self, printto, txt):
     """ send txt to printto .. do some checks. """
     if cfg['udpparty'] and partyline.is_on(printto): partyline.say_nick(printto, txt) ; return
     if not cfg['udpbots']: bots = [cfg['udpbot'], ]
     else: bots = cfg['udpbots']
     for botname in bots:
         bot = getfleet().byname(botname)
         if not bot: logging.warn("udp - can't find %s bot in fleet" % botname) ; continue
         bot.connectok.wait()
         bot.say(printto, txt)
         for i in self.loggers: i.log(printto, txt)
Example #43
0
 def dosay(self, printto, txt):
     """ send txt to printto .. do some checks. """
     if cfg['udpparty'] and partyline.is_on(printto): partyline.say_nick(printto, txt) ; return
     if not cfg['udpbots']: bots = [cfg['udpbot'], ]
     else: bots = cfg['udpbots']
     for botname in bots:
         bot = getfleet().byname(botname)
         if not bot: logging.warn("udp - can't find %s bot in fleet" % botname) ; continue
         bot.connectok.wait()
         bot.say(printto, txt)
         for i in self.loggers: i.log(printto, txt)
Example #44
0
 def doreconnect(self):
     """ reconnect to the server. """
     botjid = self.jid
     newbot = getfleet().makebot('sxmpp', self.name, cfg=self.cfg)
     newbot.reconnectcount = self.reconnectcount
     self.exit()
     if newbot.start():
         self.jid += '.old'
         #newbot.joinchannels()
         if fleet.replace(botjid, newbot): return True
     return False
Example #45
0
def handle_fleetdel(bot, ievent):
    """ arguments: <botname> - delete bot from fleet. """
    try:
        name = ievent.args[0]
    except IndexError:
        ievent.missing('<name>')
        return
    try:
        if getfleet().delete(name): ievent.reply('%s deleted' % name)
        else: ievent.reply('%s delete failed' % name)
    except Exception, ex:
        ievent.reply(str(ex))
Example #46
0
File: reboot.py Project: code2u/jsb
def reboot_stateful(bot, ievent, fleet, partyline):
    """ reboot the bot, but keep the connections (IRC only). """
    logging.warn("reboot - doing statefull reboot")
    session = {'bots': {}, 'name': bot.cfg.name, 'channel': ievent.channel, 'partyline': []}
    for i in getfleet().bots:
        logging.warn("reboot - updating %s" % i.cfg.name)
        data = i._resumedata()
        if not data: continue
        session['bots'].update(data)
        if i.type == "sxmpp": i.exit() ; continue
        if i.type == "convore": i.exit() ; continue
        if i.type == "tornado":
            i.exit()
            time.sleep(0.1)
            for socketlist in i.websockets.values():
                for sock in socketlist: sock.stream.close()
    session['partyline'] = partyline._resumedata()
    sessionfile = tempfile.mkstemp('-session', 'jsb-')[1]
    json.dump(session, open(sessionfile, 'w'))
    getfleet().save()
    os.execl(sys.argv[0], sys.argv[0], '-r', sessionfile, *sys.argv[1:])
Example #47
0
def fleet_disable(bot, ievent):
    """ arguments: <list of botnames> - disable a fleet bot. """
    if not ievent.rest:
        ievent.missing("<list of botnames>")
        return
    fleet = getfleet()
    bots = fleet.loadall(ievent.args)
    for bot in bots:
        bot.cfg['disable'] = 1
        bot.cfg.save()
        ievent.reply('disabled %s bot.' % bot.cfg.name)
        fleet.exit(bot.cfg.name)
Example #48
0
def handle_xmppinvite(bot, event):
    """ arguments: <list of JIDs> - invite (subscribe to) a different user. """
    if not event.rest:
        event.missing("<list of JIDs>")
        return
    bot = getfleet().getfirstjabber()
    if bot:
        for jid in event.args:
            bot.invite(jid)
        event.done()
    else:
        event.reply("can't find jabber bot in fleet")
Example #49
0
def handle_reboot(bot, ievent):
    """ reboot the bot. """
    if bot.isgae:
        ievent.reply("this command doesn't work on the GAE")
        return
    ievent.reply("rebooting")
    time.sleep(3)
    if ievent.rest == "cold": stateful = False
    else: stateful = True
    if stateful:
        mainhandler.put(0, reboot_stateful, bot, ievent, getfleet(), partyline)
    else:
        bot.exit()
        mainhandler.put(0, reboot)
Example #50
0
def irccat_output(msg):
    fleet = getfleet()
    dest, msg = splitMsg(msg)
    for chan in dest:
        logging.info("sending to %s" % chan)
        for botname in fleet.list():
            if botname not in cfg.data['botnames']: continue
            bot = fleet.byname(botname)
            if bot:
                if chan[0] == "#" and chan not in bot.state['joinedchannels']:
                    logging.warn("someone tried to make me say something in a channel i'm not in! (%s - %s)" % (bot.cfg.name, chan))
                else:
                    bot.say(chan, msg)
            else: logging.error("can't find %s bot in fleet" % botname)
Example #51
0
def handle_fleetdisconnect(bot, ievent):
    """ arguments: <botname> - disconnect a fleet bot from server. """
    try:
        botname = ievent.args[0]
    except IndexError:
        ievent.missing('<botname>')
        return
    ievent.reply('exiting %s' % botname)
    try:
        fleet = getfleet()
        if fleet.exit(botname): ievent.reply("%s bot stopped" % botname)
        else: ievent.reply("can't stop %s bot" % botname)
    except Exception, ex:
        ievent.reply("fleet - %s" % str(ex))
Example #52
0
def handle_nsauth(bot, ievent):
    """ perform an auth request. """
    if bot.jabber: return
    if len(ievent.args) != 1: name = bot.name
    else: name = ievent.args[0]
    fbot = getfleet().byname(name)
    if not fbot:
        ievent.reply('fleet bot %s not found' % name)
        return
    if not nsauth.has(fbot):
        ievent.reply('nickserv not configured on %s' % fbot.name)
        return
    nsauth.identify(fbot)
    ievent.reply('ok')
Example #53
0
def reboot_stateful(bot, ievent, fleet, partyline):
    """ reboot the bot, but keep the connections (IRC only). """
    logging.warn("reboot - doing statefull reboot")
    session = {
        'bots': {},
        'name': bot.name,
        'channel': ievent.channel,
        'partyline': []
    }
    for i in getfleet().bots:
        logging.warn("reboot - updating %s" % i.name)
        data = i._resumedata()
        if not data: continue
        session['bots'].update(data)
        if i.bottype == "sxmpp": i.exit()
    session['partyline'] = partyline._resumedata()
    sessionfile = tempfile.mkstemp('-session', 'jsb-')[1]
    json.dump(session, open(sessionfile, 'w'))
    getfleet().save()
    args = []
    if len(sys.argv) > 1:
        os.execl(sys.argv[0], sys.argv[0], '-r', sessionfile, *sys.argv[1:])
    else:
        os.execl(sys.argv[0], sys.argv[0], '-r', sessionfile)
Example #54
0
 def handle(self):
     try:
         fleet = getfleet()
         msg = unicode(self.rfile.readline().strip())
         logging.warn("received %s" % msg)
         dest, msg = self.splitMsg(msg)
         for chan in dest:
             logging.info("sending to %s" % chan)
             for botname in fleet.list():
                 if botname not in cfg.botnames: continue
                 bot = fleet.byname(botname)
                 if bot: bot.say(chan, msg)
                 else: logging.error("can't find %s bot in fleet" % botname)
     except Exception, ex:
         handle_exception()
Example #55
0
def fleet_disable(bot, ievent):
    """ disable a fleet bot. """
    if not ievent.rest:
        ievent.missing("list of fleet bots")
        return
    bots = ievent.rest.split()
    fleet = getfleet()
    for name in bots:
        bot = fleet.byname(name)
        if bot:
            bot.cfg['enable'] = 0
            bot.cfg.save()
            ievent.reply('disabled %s' % name)
            fleet.exit(name)
        else:
            ievent.reply("can't find %s bot in fleet" % name)
Example #56
0
def handle_nsdel(bot, ievent):
    """ remove a bot from nickserv. """
    if bot.jabber: return
    if len(ievent.args) != 1:
        ievent.missing('<fleetbot name>')
        return
    botname = ievent.args[0]
    fbot = getfleet().byname(botname)
    if not fbot:
        ievent.reply('fleet bot %s not found' % botname)
        return
    if not nsauth.has(fbot):
        ievent.reply('nickserv not configured on %s' % fbot.name)
        return
    nsauth.remove(fbot)
    ievent.reply('ok')
Example #57
0
def scan(skip=False):
    global teller, dostop
    if dostop: return
    teller += 1
    try:
        do = int(cfg.sleep)
    except ValueError:
        do = 5
    if do < 1: do = 5
    if teller % do != 0: return
    logging.info("running")
    fleet = getfleet()
    todo = sync()
    if not todo:
        logging.info("nothing todo")
        return
    for b in todo:
        uid = str(b.user_id)
        if not uid in state.data.ids:
            logging.warn("we don't follow id %s" % uid)
            continue
        for channel in state.data.ids[uid]:
            if dostop: return
            botname, chan = channel
            bot = fleet.byname(botname)
            if bot:
                if b.post_id:
                    url = ("http://stackoverflow.com/questions/%s" %
                           b.post_id) or "no url found"
                bot.say(
                    chan, "*%s* %s - *%s* - %s - %s - %s (%s)" %
                    (state.data.names[uid].upper(), b.action, b.description,
                     url, time.ctime(b.creation_date), b.detail
                     or "no detail", b.post_type))
                if b.action == "answered":
                    aa = getanswers(b.post_id)
                    if aa:
                        a = aa[-1]
                        try:
                            body = a['body']
                        except KeyError:
                            continue
                        (urls, c) = geturls(body)
                        if c: bot.say(chan, u"> " + c)
                        else: bot.say(chan, "can't find answers")
                        if urls: bot.say(chan, "urls: %s" % " -=- ".join(urls))
            else: logging.warn("no %s bot in fleet" % botname)