diff options
| -rw-r--r-- | bot.py | 95 | ||||
| -rw-r--r-- | bot.rb | 56 | ||||
| -rw-r--r-- | main.py | 41 |
3 files changed, 56 insertions, 136 deletions
@@ -1,95 +0,0 @@ -import time -import threading -import irctk -import random - -class Bot(threading.Thread): - host = 'tolkien.freenode.net' - port = 6697 - ssl = True - password = None - - def __init__(self, channel, threads, queue, name, rate, per): - threading.Thread.__init__(self) - self.threads = threads - self.queue = queue - self.name = name - self.channel = channel - # following for rate limiting: - self.rate = rate - self.per = per - self.last_check = time.time() - self.allowance = rate - - self.log('starting up') - self.ready = False - - self.client = irctk.Client(nickname=name, ident=name, realname=name, password=Bot.password) - self.client.delegate = self - self.client.connect(Bot.host, Bot.port, use_tls=Bot.ssl) - - def irc_raw(self, client, line): - try: - self.log('IRC: ' + line) - except: - self.log('failed to log') - #pass - - def log(self, line): - print "{} [{}] {}".format(time.strftime('%H:%M'), self.name, line) - - def irc_registered(self, client): - self.log('joining ' + self.channel) - channel = client.add_channel(self.channel) - channel.join() - - def irc_channel_join(self, client, nick, channel): - if str(nick) == str(client.get_nickname()): - self.log('im ready :)') - self.ready = True - - def can_send(self): - current = time.time() - time_passed = current - self.last_check; - self.last_check = current - self.allowance += time_passed * (self.rate / self.per) - - if self.allowance > self.rate: - self.allowance = self.rate - - if self.allowance > 1.0: - self.allowance -= 1.0 - return True - - return False - - def run(self): - self.log('run() called') - - while True: - waiting = True - while waiting: - for thread in self.threads: - if not thread.ready: - waiting = True - break - else: - waiting = False - - time.sleep(random.random()) - - # start flushing the queue - while self.can_send(): - try: - line = self.queue.popleft() - self.client.send_line( - 'PRIVMSG ' + self.channel + ' :' + line - ) - time.sleep(0.25) - except IndexError: - # end of queue - self.log('done working') - self.client.quit('bye bye') - return - - time.sleep(0.1) @@ -0,0 +1,56 @@ +require 'thread' +require 'cinch' + +workers = 5 +queue = Queue.new +throttle = 0.1 # lower the faster but easier to hit IRCd sendq +prev_item = nil + +@bots = Array.new(workers) do + Thread.new do + bot = Cinch::Bot.new do + configure do |c| + c.server = "irc.server.local" + c.port = 7001 + c.realname = 'robert' + c.password = "password" + c.channels = ["#test99"] + c.nicks = ['zz1', 'zz2', 'zz3', 'zz4', 'zz5', 'zz6'] + c.user = 'bot' + c.ssl.use = true + c.ssl.verify = false + end + + on :message, /^!spam (\d+)/ do |m, count| + if queue.empty? + count.to_i.times do |i| + queue << i + end + + prev_item = queue.pop + + # This kicks off the chain reaction + m.reply prev_item + end + end + + on :message do |m| + debug "#{m.bot.nick}: '#{m.message}' =/= '#{prev_item}' and queue empty: #{queue.empty?}'" + + if m.message == prev_item.to_s + if queue.empty? + prev_item = nil + else + sleep(throttle) + prev_item = queue.pop + m.reply prev_item + end + end + end + end + + bot.start + end +end + +@bots.each(&:join) diff --git a/main.py b/main.py deleted file mode 100644 index 2b8a902..0000000 --- a/main.py +++ /dev/null @@ -1,41 +0,0 @@ -import sys -import math -import collections -import zokket -from bot import Bot - -botname_prefix = 'audzx' -queue = collections.deque() -rate = 5.0 # messages -per = 8.0 # seconds -max_workers = 6 -channel = '#test123aszz' - -if len(sys.argv) == 2: - channel = sys.argv[1] - -if channel[0] != '#': - channel = '#' + channel - -linecount = 0 -for line in sys.stdin: - queue.append(line.strip()) - linecount += 1 - -workers = int(math.ceil(linecount / rate)) -if workers > max_workers: - workers = max_workers - -if workers == 0: - sys.exit(1) - -print 'starting {} worker threads'.format(workers) -threads = [] -for i in range(workers): - botname = botname_prefix + chr(97 + i) - threads.append(Bot(channel, threads, queue, botname, rate, per)) - -for bot in threads: - bot.start() - -zokket.DefaultRunloop.run() |