rbot 0.9.9

data/data/rbot/plugins/nslookup.rb

@@ -0,0 +1,43 @@
+class DnsPlugin < Plugin
+  require 'resolv'
+  def gethostname(address)
+    Resolv.getname(address)
+  end
+  def getaddresses(name)
+    Resolv.getaddresses(name)
+  end
+
+  def help(plugin, topic="")
+    "dns <hostname|ip> => show local resolution results for hostname or ip address"
+  end
+  
+  def name_to_ip(m, params)
+    Thread.new do
+      begin
+        a = getaddresses(params[:host])
+        if a.length > 0
+          m.reply m.params + ": " + a.join(", ")
+        else
+          m.reply "#{params[:host]}: not found"
+        end
+      rescue StandardError => err
+        m.reply "#{params[:host]}: not found"
+      end
+    end
+  end
+  
+  def ip_to_name(m, params)
+    Thread.new do
+       begin
+         a = gethostname(params[:ip])
+         m.reply m.params + ": " + a if a
+       rescue StandardError => err
+         m.reply "#{params[:ip]}: not found (does not reverse resolve)"
+       end
+     end
+  end
+end
+plugin = DnsPlugin.new
+plugin.map 'dns :ip', :action => 'ip_to_name', 
+                      :requirements => {:ip => /^\d+\.\d+\.\d+\.\d+$/}
+plugin.map 'dns :host', :action => 'name_to_ip'

data/data/rbot/plugins/opme.rb

@@ -0,0 +1,19 @@
+class OpMePlugin < Plugin
+
+  def help(plugin, topic="")
+    return "opme <channel> => grant user ops in <channel>"
+  end
+
+  def privmsg(m)
+    if(m.params)
+      channel = m.params
+    else
+      channel = m.channel
+    end
+    target = m.sourcenick
+    @bot.sendq("MODE #{channel} +o #{target}")
+    m.okay
+  end
+end
+plugin = OpMePlugin.new
+plugin.register("opme")

data/data/rbot/plugins/quakeauth.rb

@@ -0,0 +1,51 @@
+# automatically auths with Q on quakenet servers
+
+class QPlugin < Plugin
+  
+  def help(plugin, topic="")
+    case topic
+    when ""
+      return "quath plugin: handles Q auths. topics set, identify"
+    when "set"
+      return "nickserv set <user> <passwd>: set the Q user and password and use it to identify in future"
+    when "identify"
+      return "quath identify: identify with Q (if user and auth are set)"
+    end
+  end
+  
+  def initialize
+    super
+    # this plugin only wants to store strings!
+    class << @registry
+      def store(val)
+        val
+      end
+      def restore(val)
+        val
+      end
+    end
+  end
+
+  def set(m, params)
+    @registry['quakenet.user'] = params[:user]
+    @registry['quakenet.auth'] = params[:passwd]
+    m.okay
+  end
+  
+  def connect
+    identify(nil, nil)
+  end
+  def identify(m, params)
+    if @registry.has_key?('quakenet.user') && @registry.has_key?('quakenet.auth')
+      debug "authing with Q using  #{@registry['quakenet.user']} #{@registry['quakenet.auth']}"
+      @bot.sendmsg "PRIVMSG", "Q@CServe.quakenet.org", "auth #{@registry['quakenet.user']} #{@registry['quakenet.auth']}"                                                    
+      m.okay if m
+    else
+      m.reply "not configured, try 'qauth set :nick :passwd'" if m
+    end
+  end
+
+end
+plugin = QPlugin.new
+plugin.map 'qauth set :nick :passwd', :action => "set"
+plugin.map 'quath identify', :action => "identify"

data/data/rbot/plugins/quotes.rb

@@ -0,0 +1,321 @@
+Quote = Struct.new("Quote", "num", "date", "source", "quote")
+
+class QuotePlugin < Plugin
+  def initialize
+    super
+    @lists = Hash.new
+    Dir["#{@bot.botclass}/quotes/*"].each {|f|
+      channel = File.basename(f)
+      @lists[channel] = Array.new if(!@lists.has_key?(channel))
+      IO.foreach(f) {|line|
+        if(line =~ /^(\d+) \| ([^|]+) \| (\S+) \| (.*)$/)
+          num = $1.to_i
+          @lists[channel][num] = Quote.new(num, $2, $3, $4)
+        end
+      }
+    }
+  end
+  def save
+    Dir.mkdir("#{@bot.botclass}/quotes") if(!FileTest.directory?("#{@bot.botclass}/quotes"))
+    @lists.each {|channel, quotes|
+      File.open("#{@bot.botclass}/quotes/#{channel}", "w") {|file|
+        quotes.compact.each {|q| 
+          file.puts "#{q.num} | #{q.date} | #{q.source} | #{q.quote}"
+        }
+      }
+    }
+  end
+  def addquote(source, channel, quote)
+    @lists[channel] = Array.new if(!@lists.has_key?(channel))
+    num = @lists[channel].length 
+    @lists[channel][num] = Quote.new(num, Time.new, source, quote)
+    return num
+  end
+  def getquote(source, channel, num=nil)
+    return nil unless(@lists.has_key?(channel))
+    return nil unless(@lists[channel].length > 0)
+    if(num)
+      if(@lists[channel][num])
+        return @lists[channel][num], @lists[channel].length - 1
+      end
+    else
+      # random quote
+      return @lists[channel].compact[rand(@lists[channel].nitems)],
+                                       @lists[channel].length - 1
+    end
+  end
+  def delquote(channel, num)
+    return false unless(@lists.has_key?(channel))
+    return false unless(@lists[channel].length > 0)
+    if(@lists[channel][num])
+      @lists[channel][num] = nil
+      return true
+    end
+    return false
+  end
+  def countquote(source, channel=nil, regexp=nil)
+    unless(channel)
+      total=0
+      @lists.each_value {|l|
+        total += l.compact.length
+      }
+      return total
+    end
+    return 0 unless(@lists.has_key?(channel))
+    return 0 unless(@lists[channel].length > 0)
+    if(regexp)
+      matches = @lists[channel].compact.find_all {|a| a.quote =~ /#{regexp}/i }
+    else
+      matches = @lists[channel].compact
+    end
+    return matches.length
+  end
+  def searchquote(source, channel, regexp)
+    return nil unless(@lists.has_key?(channel))
+    return nil unless(@lists[channel].length > 0)
+    matches = @lists[channel].compact.find_all {|a| a.quote =~ /#{regexp}/i }
+    if(matches.length > 0)
+       return matches[rand(matches.length)], @lists[channel].length - 1
+    else
+      return nil
+    end
+  end
+  def help(plugin, topic="")
+    case topic
+      when "addquote"
+        return "addquote [<channel>] <quote> => Add quote <quote> for channel <channel>. You only need to supply <channel> if you are addressing #{@bot.nick} privately. Responds to !addquote without addressing if so configured"
+      when "delquote"
+        return "delquote [<channel>] <num> => delete quote from <channel> with number <num>. You only need to supply <channel> if you are addressing #{@bot.nick} privately. Responds to !delquote without addressing if so configured"
+      when "getquote"
+        return "getquote [<channel>] [<num>] => get quote from <channel> with number <num>. You only need to supply <channel> if you are addressing #{@bot.nick} privately. Without <num>, a random quote will be returned. Responds to !getquote without addressing if so configured"
+      when "searchquote"
+        return "searchquote [<channel>] <regexp> => search for quote from <channel> that matches <regexp>. You only need to supply <channel> if you are addressing #{@bot.nick} privately. Responds to !searchquote without addressing if so configured"
+      when "topicquote"
+        return "topicquote [<channel>] [<num>] => set topic to quote from <channel> with number <num>. You only need to supply <channel> if you are addressing #{@bot.nick} privately. Without <num>, a random quote will be set. Responds to !topicquote without addressing if so configured"
+      when "countquote"
+        return "countquote [<channel>] <regexp> => count quotes from <channel> that match <regexp>. You only need to supply <channel> if you are addressing #{@bot.nick} privately. Responds to !countquote without addressing if so configured"
+      when "whoquote"
+        return "whoquote [<channel>] <num> => show who added quote <num>. You only need to supply <channel> if you are addressing #{@bot.nick} privately"
+      when "whenquote"
+        return "whenquote [<channel>] <num> => show when quote <num> was added. You only need to supply <channel> if you are addressing #{@bot.nick} privately"
+      else
+        return "Quote module (Quote storage and retrieval) topics: addquote, getquote, searchquote, topicquote, countquote, whoquote, whenquote"
+    end
+  end
+  def listen(m)
+    return unless(m.kind_of? PrivMessage)
+
+    command = m.message.dup
+    if(m.address? && m.private?)
+      case command
+        when (/^addquote\s+(#\S+)\s+(.*)/)
+          channel = $1
+          quote = $2
+          if(@bot.auth.allow?("addquote", m.source, m.replyto))
+            if(channel =~ /^#/)
+              num = addquote(m.source, channel, quote)
+              m.reply "added the quote (##{num})"
+            end
+          end
+        when (/^getquote\s+(#\S+)$/)
+          channel = $1
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, channel)
+            if(quote)
+              m.reply "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^getquote\s+(#\S+)\s+(\d+)$/)
+          channel = $1
+          num = $2.to_i
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, channel, num)
+            if(quote)
+              m.reply "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^whoquote\s+(#\S+)\s+(\d+)$/)
+          channel = $1
+          num = $2.to_i
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, channel, num)
+            if(quote)
+              m.reply "quote #{quote.num} added by #{quote.source}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^whenquote\s+(#\S+)\s+(\d+)$/)
+          channel = $1
+          num = $2.to_i
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, channel, num)
+            if(quote)
+              m.reply "quote #{quote.num} added on #{quote.date}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^topicquote\s+(#\S+)$/)
+          channel = $1
+          if(@bot.auth.allow?("topicquote", m.source, m.replyto))
+            quote, total = getquote(m.source, channel)
+            if(quote)
+              @bot.topic channel, "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^topicquote\s+(#\S+)\s+(\d+)$/)
+          channel = $1
+          num = $2.to_i
+          if(@bot.auth.allow?("topicquote", m.source, m.replyto))
+            quote, total = getquote(m.source, channel, num)
+            if(quote)
+              @bot.topic channel, "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^delquote\s+(#\S+)\s+(\d+)$/)
+          channel = $1
+          num = $2.to_i
+          if(@bot.auth.allow?("delquote", m.source, m.replyto))
+            if(delquote(channel, num))
+              m.okay
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^searchquote\s+(#\S+)\s+(.*)$/)
+          channel = $1
+          reg = $2
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = searchquote(m.source, channel, reg)
+            if(quote)
+              m.reply "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^countquote$/)
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            total = countquote(m.source)
+            m.reply "#{total} quotes"
+          end
+        when (/^countquote\s+(#\S+)\s*(.*)$/)
+          channel = $1
+          reg = $2
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            total = countquote(m.source, channel, reg)
+            if(reg.length > 0)
+              m.reply "#{total} quotes match: #{reg}"
+            else
+              m.reply "#{total} quotes"
+            end
+          end
+      end
+    elsif (m.address? || (@bot.config["QUOTE_LISTEN"] && command.gsub!(/^!/, "")))
+      case command
+        when (/^addquote\s+(.+)/)
+          if(@bot.auth.allow?("addquote", m.source, m.replyto))
+            num = addquote(m.source, m.target, $1)
+            m.reply "added the quote (##{num})"
+          end
+        when (/^getquote$/)
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, m.target)
+            if(quote)
+              m.reply "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "no quotes found!"
+            end
+          end
+        when (/^getquote\s+(\d+)$/)
+          num = $1.to_i
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, m.target, num)
+            if(quote)
+              m.reply "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^whenquote\s+(\d+)$/)
+          num = $1.to_i
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, m.target, num)
+            if(quote)
+              m.reply "quote #{quote.num} added on #{quote.date}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^whoquote\s+(\d+)$/)
+          num = $1.to_i
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = getquote(m.source, m.target, num)
+            if(quote)
+              m.reply "quote #{quote.num} added by #{quote.source}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^topicquote$/)
+          if(@bot.auth.allow?("topicquote", m.source, m.replyto))
+            quote, total = getquote(m.source, m.target)
+            if(quote)
+              @bot.topic m.target, "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "no quotes found!"
+            end
+          end
+        when (/^topicquote\s+(\d+)$/)
+          num = $1.to_i
+          if(@bot.auth.allow?("topicquote", m.source, m.replyto))
+            quote, total = getquote(m.source, m.target, num)
+            if(quote)
+              @bot.topic m.target, "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^delquote\s+(\d+)$/)
+          num = $1.to_i
+          if(@bot.auth.allow?("delquote", m.source, m.replyto))
+            if(delquote(m.target, num))
+              m.okay
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^searchquote\s+(.*)$/)
+          reg = $1
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            quote, total = searchquote(m.source, m.target, reg)
+            if(quote)
+              m.reply "[#{quote.num}] #{quote.quote}"
+            else
+              m.reply "quote not found!"
+            end
+          end
+        when (/^countquote(?:\s+(.*))?$/)
+          reg = $1
+          if(@bot.auth.allow?("getquote", m.source, m.replyto))
+            total = countquote(m.source, m.target, reg)
+            if(reg && reg.length > 0)
+              m.reply "#{total} quotes match: #{reg}"
+            else
+              m.reply "#{total} quotes"
+            end
+          end
+      end
+    end
+  end
+end
+plugin = QuotePlugin.new
+plugin.register("quotes")

data/data/rbot/plugins/remind.rb

@@ -0,0 +1,228 @@
+require 'rbot/utils'
+
+class RemindPlugin < Plugin
+  # read a time in string format, turn it into "seconds from now".
+  # example formats handled are "5 minutes", "2 days", "five hours",
+  # "11:30", "15:45:11", "one day", etc.
+  #
+  # Throws:: RunTimeError "invalid time string" on parse failure
+  def timestr_offset(timestr)
+    case timestr
+      when (/^(\S+)\s+(\S+)$/)
+        mult = $1
+        unit = $2
+        if(mult =~ /^([\d.]+)$/)
+          num = $1.to_f
+          raise "invalid time string" unless num
+        else
+          case mult
+            when(/^(one|an|a)$/)
+              num = 1
+            when(/^two$/)
+              num = 2
+            when(/^three$/)
+              num = 3
+            when(/^four$/)
+              num = 4
+            when(/^five$/)
+              num = 5
+            when(/^six$/)
+              num = 6
+            when(/^seven$/)
+              num = 7
+            when(/^eight$/)
+              num = 8
+            when(/^nine$/)
+              num = 9
+            when(/^ten$/)
+              num = 10
+            when(/^fifteen$/)
+              num = 15
+            when(/^twenty$/)
+              num = 20
+            when(/^thirty$/)
+              num = 30
+            when(/^sixty$/)
+              num = 60
+            else
+              raise "invalid time string"
+          end
+        end
+        case unit
+          when (/^(s|sec(ond)?s?)$/)
+            return num
+          when (/^(m|min(ute)?s?)$/)
+            return num * 60
+          when (/^(h|h(ou)?rs?)$/)
+            return num * 60 * 60
+          when (/^(d|days?)$/)
+            return num * 60 * 60 * 24
+          else
+            raise "invalid time string"
+        end
+      when (/^(\d+):(\d+):(\d+)$/)
+        hour = $1.to_i
+        min = $2.to_i
+        sec = $3.to_i
+        now = Time.now
+        later = Time.mktime(now.year, now.month, now.day, hour, min, sec)
+        return later - now
+      when (/^(\d+):(\d+)$/)
+        hour = $1.to_i
+        min = $2.to_i
+        now = Time.now
+        later = Time.mktime(now.year, now.month, now.day, hour, min, now.sec)
+        return later - now
+      when (/^(\d+):(\d+)(am|pm)$/)
+        hour = $1.to_i
+        min = $2.to_i
+        ampm = $3
+        if ampm == "pm"
+          hour += 12
+        end
+        now = Time.now
+        later = Time.mktime(now.year, now.month, now.day, hour, min, now.sec)
+        return later - now
+      when (/^(\S+)$/)
+        num = 1
+        unit = $1
+        case unit
+          when (/^(s|sec(ond)?s?)$/)
+            return num
+          when (/^(m|min(ute)?s?)$/)
+            return num * 60
+          when (/^(h|h(ou)?rs?)$/)
+            return num * 60 * 60
+          when (/^(d|days?)$/)
+            return num * 60 * 60 * 24
+          else
+            raise "invalid time string"
+        end
+      else
+        raise "invalid time string"
+    end
+  end
+
+  def initialize
+    super
+    @reminders = Hash.new
+  end
+  def cleanup
+    @reminders.each_value {|v|
+      v.each_value {|vv|
+        @bot.timer.remove(vv)
+      }
+    }
+    @reminders.clear
+  end
+  def help(plugin, topic="")
+    "reminder plugin: remind <who> [about] <message> in <time>, remind <who> [about] <message> every <time>, remind <who> [about] <message> at <time>, remind <who> no more [about] <message>, remind <who> no more. Generally <who> should be 'me', but you can remind others (nick or channel) if you have remind_others auth"
+  end
+  def add_reminder(who, subject, timestr, repeat=false)
+    begin
+      period = timestr_offset(timestr)
+    rescue RuntimeError
+      return "couldn't parse that time string (#{timestr}) :("
+    end
+    if(period <= 0)
+      return "that time is in the past! (#{timestr})"
+    end
+    if(period < 30 && repeat)
+      return "repeats of less than 30 seconds are forbidden"
+    end
+    if(!@reminders.has_key?(who))
+      @reminders[who] = Hash.new
+    elsif(@reminders[who].has_key?(subject))
+      del_reminder(who, subject)
+    end
+
+    if(repeat)
+      @reminders[who][subject] = @bot.timer.add(period) {
+        time = Time.now + period
+        tstr = time.strftime("%H:%M:%S")
+        @bot.say who, "repeat reminder (next at #{tstr}): #{subject}"
+      }
+    else
+      @reminders[who][subject] = @bot.timer.add_once(period) {
+        time = Time.now + period
+        tstr = time.strftime("%H:%M:%S")
+        @bot.say who, "reminder (#{tstr}): #{subject}"
+      }
+    end
+    return false
+  end
+  def del_reminder(who, subject=nil)
+    if(subject)
+      if(@reminders.has_key?(who) && @reminders[who].has_key?(subject))
+        @bot.timer.remove(@reminders[who][subject])
+        @reminders[who].delete(subject)
+        return true
+      else
+        return false
+      end
+    else
+      if(@reminders.has_key?(who))
+        @reminders[who].each_value {|v|
+          @bot.timer.remove(v)
+        }
+        @reminders.delete(who)
+        return true
+      else
+        return false
+      end
+    end
+  end
+  def remind(m, params)
+    who = params.has_key?(:who) ? params[:who] : m.sourcenick
+    string = params[:string].to_s
+    puts "in remind, string is: #{string}"
+    if(string =~ /^(.*)\s+in\s+(.*)$/)
+      subject = $1
+      period = $2
+      if(err = add_reminder(who, subject, period))
+        m.reply "incorrect usage: " + err
+        return
+      end
+    elsif(string =~ /^(.*)\s+every\s+(.*)$/)
+      subject = $1
+      period = $2
+      if(err = add_reminder(who, subject, period, true))
+        m.reply "incorrect usage: " + err
+        return
+      end
+    elsif(string =~ /^(.*)\s+at\s+(.*)$/)
+      subject = $1
+      time = $2
+      if(err = add_reminder(who, subject, time))
+        m.reply "incorrect usage: " + err
+        return
+      end
+    else
+      usage(m)
+      return
+    end
+    m.okay
+  end
+  def no_more(m, params)
+    who = params.has_key?(:who) ? params[:who] : m.sourcenick
+    deleted = params.has_key?(:string) ? 
+              del_reminder(who, params[:string].to_s) : del_reminder(who)
+    if deleted
+      m.okay
+    else 
+      m.reply "but I wasn't going to :/"
+    end
+  end
+end
+plugin = RemindPlugin.new
+plugin.map 'remind me no more', :action => 'no_more'
+plugin.map 'remind me no more about *string', :action => 'no_more'
+plugin.map 'remind me no more *string', :action => 'no_more'
+plugin.map 'remind me about *string'
+plugin.map 'remind me *string'
+plugin.map 'remind :who no more', :auth => 'remind_other', :action => 'no_more'
+plugin.map 'remind :who no more about *string', :auth => 'remind_other', :action => 'no_more'
+plugin.map 'remind :who no more *string', :auth => 'remind_other', :action => 'no_more'
+plugin.map 'remind :who about *string', :auth => 'remind_other'
+plugin.map 'remind :who *string', :auth => 'remind_other'
+