rbot 0.9.14 → 0.9.15

data/data/rbot/plugins/geoip.rb

@@ -9,40 +9,80 @@
 #
 # Resolves the geographic locations of users (network-wide) and IP addresses
 
-module GeoIP
+module ::GeoIP
   class InvalidHostError < RuntimeError; end
+  class BadAPIError < RuntimeError; end
 
-  GEO_IP_PRIMARY   = "http://lakka.kapsi.fi:40086/lookup.yaml?host="
-  GEO_IP_SECONDARY = "http://www.geoiptool.com/en/?IP="
   HOST_NAME_REGEX  = /^[a-z0-9\-]+(?:\.[a-z0-9\-]+)*\.[a-z]{2,4}/i
 
-  REGEX  = {
-    :country => %r{Country:.*?<a href=".*?" target="_blank"> (.*?)</a>}m,
-    :region  => %r{Region:.*?<a href=".*?" target="_blank">(.*?)</a>}m,
-    :city    => %r{City:.*?<td align="left" class="arial_bold">(.*?)</td>}m
-  }
-
   def self.valid_host?(hostname)
     hostname =~ HOST_NAME_REGEX ||
     hostname =~ Resolv::IPv4::Regex && (hostname.split(".").map { |e| e.to_i }.max <= 255)
   end
 
-  def self.resolve(hostname)
-    raise InvalidHostError unless valid_host?(hostname)
+  def self.geoiptool(ip)
+    url = "http://www.geoiptool.com/en/?IP="
+    regexes  = {
+      :country => %r{Country:.*?<a href=".*?" target="_blank"> (.*?)</a>}m,
+      :region  => %r{Region:.*?<a href=".*?" target="_blank">(.*?)</a>}m,
+      :city    => %r{City:.*?<td align="left" class="arial_bold">(.*?)</td>}m,
+      :lat     => %r{Latitude:.*?<td align="left" class="arial_bold">(.*?)</td>}m,
+      :lon     => %r{Longitude:.*?<td align="left" class="arial_bold">(.*?)</td>}m
+    }
+    res = {}
+    raw = Irc::Utils.bot.httputil.get_response(url+ip)
+    raw = raw.decompress_body(raw.raw_body)
+
+    regexes.each { |key, regex| res[key] = Iconv.conv('utf-8', 'ISO-8859-1', raw.scan(regex).to_s) }
+
+    return res
+  end
 
-    yaml = Irc::Utils.bot.httputil.get(GEO_IP_PRIMARY+hostname)
+  def self.kapsi(ip)
+    url = "http://lakka.kapsi.fi:40086/lookup.yaml?host="
+    yaml = Irc::Utils.bot.httputil.get(url+ip)
+    return YAML::load(yaml)
+  end
 
-    if yaml
-      return YAML::load(yaml)
+  def self.blogama(ip)
+    url = "http://ipinfodb.com/ip_query.php?ip="
+    debug "Requesting #{url+ip}"
+
+    xml = Irc::Utils.bot.httputil.get(url+ip)
+
+    if xml
+      obj = REXML::Document.new(xml)
+      debug "Found #{obj}"
+      newobj = {
+        :country => obj.elements["Response"].elements["CountryName"].text,
+        :city => obj.elements["Response"].elements["City"].text,
+        :region => obj.elements["Response"].elements["RegionName"].text,
+      }
+      debug "Returning #{newobj}"
+      return newobj
     else
-      res = {}
-      raw = Irc::Utils.bot.httputil.get_response(GEO_IP_SECONDARY+hostname)
-      raw = raw.decompress_body(raw.raw_body)
+      raise InvalidHostError
+    end
+  end
 
-      REGEX.each { |key, regex| res[key] = Iconv.conv('utf-8', 'ISO-8859-1', raw.scan(regex).to_s) }
+  def self.resolve(hostname, api)
+    raise InvalidHostError unless valid_host?(hostname)
 
-      return res
+    begin
+      ip = Resolv.getaddress(hostname)
+    rescue Resolv::ResolvError
+      raise InvalidHostError
     end
+
+    jump_table = {
+        "blogama" => Proc.new { |ip| blogama(ip) },
+        "kapsi" => Proc.new { |ip| kapsi(ip) },
+        "geoiptool" => Proc.new { |ip| geoiptool(ip) },
+    }
+
+    raise BadAPIError unless jump_table.key?(api)
+
+    return jump_table[api].call(ip)
   end
 end
 
@@ -66,6 +106,10 @@ class Stack
 end
 
 class GeoIpPlugin < Plugin
+  Config.register Config::ArrayValue.new('geoip.sources',
+      :default => [ "blogama", "kapsi", "geoiptool" ],
+      :desc => "Which API to use for lookups. Supported values: blogama, kapsi, geoiptool")
+
   def help(plugin, topic="")
     "geoip [<user|hostname|ip>] => returns the geographic location of whichever has been given -- note: user can be anyone on the network"
   end
@@ -128,12 +172,23 @@ class GeoIpPlugin < Plugin
   def host2output(host, nick=nil)
     return "127.0.0.1 could not be res.. wait, what?" if host == "127.0.0.1"
 
+    geo = {:country => ""}
     begin
-      geo = GeoIP::resolve(host)
-
-      raise if geo[:country].empty?
+      apis = @bot.config['geoip.sources']
+      apis.compact.each { |api|
+        geo = GeoIP::resolve(host, api)
+        if geo[:country] != ""
+          break
+        end
+      }
     rescue GeoIP::InvalidHostError, RuntimeError
-      return _("#{nick ? "#{nick}'s location" : host} could not be resolved")
+      if nick
+        return _("#{nick}'s location could not be resolved")
+      else
+        return _("#{host} could not be resolved")
+      end
+    rescue GeoIP::BadAPIError
+      return _("The owner configured me to use an API that doesn't exist, bug them!")
     end
 
     res = _("%{thing} is #{nick ? "from" : "located in"}") % {
@@ -141,18 +196,18 @@ class GeoIpPlugin < Plugin
       :country => geo[:country]
     }
 
-    res << " %{city}," % {
+    res << " %{city}" % {
       :city => geo[:city]
     } unless geo[:city].to_s.empty?
 
+    res << " %{region}," % {
+      :region  => geo[:region]
+    } unless geo[:region].to_s.empty? || geo[:region] == geo[:city]
+
     res << " %{country}" % {
       :country => geo[:country]
     }
 
-    res << " (%{region})" % {
-      :region  => geo[:region]
-    } unless geo[:region].to_s.empty? || geo[:region] == geo[:city]
-
     return res
   end
 end

data/data/rbot/plugins/googlefight.rb

@@ -0,0 +1,64 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: Googlefight plugin for rbot
+#
+# Author:: Raine Virta <rane@kapsi.fi
+# Copyright:: (C) 2009 Raine Virta
+# License:: GPL v2
+
+class GoogleFightPlugin < Plugin
+  def help(plugin, topic)
+    "googlefight <keyword 1> <keyword 2> [... <keyword n+1>] => battles given keywords based on amount of google search results and announces the winner!"
+  end
+
+  def fight(m, params)
+    keywords = parse_keywords(params)
+    return if keywords.nil?
+
+    keywords.map! do |k|
+      [k, google_count(k)]
+    end
+
+    m.reply output(keywords)
+  end
+
+  def output(result)
+    result = result.sort_by { |e| e[1] }.reverse
+    str = result.map do |kw|
+      "%{keyword} (%{count})" % {
+        :keyword => Bold+kw[0]+Bold,
+        :count   => kw[1].to_s.gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,")
+      }
+    end.join(" vs. ")
+
+    unless result[0][1].zero?
+      str << _(" -- %{keyword} wins!") % {
+        :keyword => Bold+result[0][0]+Bold
+      }
+    else
+      str << _(" -- no winner here!")
+    end
+  end
+
+  def parse_keywords(params)
+    str = params[:keywords].join(" ")
+
+    # foo "foo bar" bar
+    # no separators so assume they're all separate keywords
+    if str.match(/(?:"[\w\s]+"|\w+)(?: (?:"[\w\s]+"|\w+))+/)
+      str.scan(/"[^"]+"|\S+/).flatten
+    end
+  end
+
+  def google_count(query)
+    url  = 'http://www.google.com/search?hl=en&safe=off&btnG=Search&q=' << CGI.escape(query)
+    html = Net::HTTP.get(URI.parse((url)))
+    res  = html.scan(%r{Results <b>1</b> - <b>\d+</b> of(?: about)? <b>([\d,]+)</b>})
+    res.to_s.tr(",", "").to_i
+  end
+end
+
+plugin = GoogleFightPlugin.new
+plugin.map "googlefight *keywords", :action => "fight",
+  :requirements => { :keywords => /^[\w\s"]+? (?:(?:(?:\||vs\.) )?[\w\s"]+?)+/ }

data/data/rbot/plugins/greet.rb

@@ -0,0 +1,45 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: Greet Plugin
+#
+# Author:: Raine Virta <rane@kapsi.fi>
+# Copyright:: (C) 2009 Raine Virta
+# License:: GPL v2
+#
+# Description:: Greet people when they join a channel
+
+class GreetPlugin < Plugin
+  Config.register Config::ArrayValue.new('greet.channels',
+    :desc => _("Greet people on these channels."),
+    :default => [])
+
+  Config.register Config::ArrayValue.new('greet.messages',
+    :desc => _("By default, greetings are fetched from lang files. You can use this to specify custom messages, use %s to represent a nick."),
+    :default => [])
+
+  Config.register Config::BooleanValue.new('greet.delay',
+    :desc => _("Greet with delay so that the greeting seems human-like."),
+    :default => false)
+
+
+  def join(m)
+    return unless @bot.config['greet.channels'].include?(m.channel.to_s)
+
+    greeting = if @bot.config['greet.messages'].empty?
+      @bot.lang.get("hello_X")
+    else
+      @bot.config['greet.messages'].pick_one
+    end
+
+    msg = Proc.new { @bot.say m.channel, greeting % m.sourcenick }
+
+    if @bot.config['greet.delay']
+      @bot.timer.add_once(2+rand(3)) { msg.call }
+    else
+      msg.call
+    end
+  end
+end
+
+plugin = GreetPlugin.new

data/data/rbot/plugins/grouphug.rb

@@ -10,11 +10,13 @@
 # License:: GPL v2
 
 class GrouphugPlugin < Plugin
+  START = '<div id="main"'
   REG  = Regexp.new('<div class="content">\s*<p>(.*?)</p>\s+</div>', Regexp::MULTILINE)
   REGPOST = Regexp.new('title>(.*?) \| Group Hug')
   def initialize
     super
     @confessions = Array.new
+    @bot.register_filter(:grouphug, :htmlinfo) { |s| grouphug_filter(s) }
   end
 
   def help( plugin, topic="" )
@@ -45,6 +47,15 @@ class GrouphugPlugin < Plugin
     end
   end
 
+  def get_confessions(html)
+    return [] unless html
+    start = html.index(START)
+    res = html[start, html.length - start].scan(REG)
+    return [] unless res
+    return res.map { |quote|
+      quote[0].ircify_html
+    }
+  end
 
   def confess(m, params)
     opts = { :cache => false }
@@ -55,26 +66,43 @@ class GrouphugPlugin < Plugin
         path = "confessions/#{params[:num]}"
         opts.delete(:cache)
         data = @bot.httputil.get("http://grouphug.us/#{path}", opts)
-
-        res = data.scan(REG)
-        confession = res[0][0].ircify_html
-        confession = "no confession ##{params[:num]} found" if confession.empty? and params[:num]
-        m.reply confession
+        confessions = get_confessions(data)
+        if confessions.length > 1
+          warn "more than one confession found!"
+          warn confessions
+        end
+        confessions << "no confession ##{params[:num]} found" if confessions.empty?
+        m.reply confessions.first
       else # Cache random confessions
         if @confessions.empty?
           data = @bot.httputil.get("http://grouphug.us/#{path}", opts)
-          res = data.scan(REG)
-          res.each do |quote|
-            @confessions << quote[0].ircify_html
-          end
+          @confessions.replace get_confessions(data)
         end
-        confession = @confessions.pop
-        m.reply confession
+        @confessions << "no confessions found!" if @confessions.empty?
+        m.reply @confessions.pop
       end
-    rescue
+    rescue Exception => e
+      error e
       m.reply "failed to connect to grouphug.us"
     end
   end
+
+  def grouphug_filter(s)
+    # check if we like the location of the page
+    loc = Utils.check_location(s, %r{http://(?:.*\.)?grouphug\.us})
+    return unless loc
+    # check if there are any conefssions
+    confessions = get_confessions(s[:text])
+    return if confessions.empty?
+    title = s[:text].ircify_html_title
+    # return the first confession
+    return {
+          :title => title,
+          :content => confessions.first,
+          :grouphug_confessions => confessions
+    }
+  end
+
 end
 
 

data/data/rbot/plugins/imdb.rb

@@ -12,7 +12,7 @@
 # License:: MIT license
 
 class Imdb
-  IMDB = "http://us.imdb.com"
+  IMDB = "http://www.imdb.com"
   TITLE_OR_NAME_MATCH = /<a href="(\/(?:title|name)\/(?:tt|nm)[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/
   TITLE_MATCH = /<a href="(\/title\/tt[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/
   NAME_MATCH = /<a href="(\/name\/nm[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/
@@ -99,7 +99,7 @@ class Imdb
   end
 
   def grab_info(info, body)
-    /<div class="info">\s+<h5>#{info}:<\/h5>\s+(.*?)<\/div>/mi.match(body)[1] rescue nil
+    /<div (?:id="\S+-info" )?class="info">\s+<h5>#{info}:<\/h5>\s+(.*?)<\/div>/mi.match(body)[1] rescue nil
   end
 
   def fix_article(org_tit)
@@ -165,7 +165,7 @@ class Imdb
       end
 
       ratings = "no votes"
-      m = /<b>([0-9.]+)\/10<\/b>\n?\r?\s+<small>\(<a href="ratings">([0-9,]+) votes?<\/a>\)<\/small>/.match(resp.body)
+      m = /<b>([0-9.]+)\/10<\/b>\n?\r?\s+[^<]+<a href="ratings"[^>]+>([0-9,]+) votes?<\/a>/.match(resp.body)
       if m
         ratings = "#{m[1]}/10 (#{m[2]} voters)"
       end
@@ -469,7 +469,7 @@ class ImdbPlugin < Plugin
   end
 
   # Find the movies with a participation of :who in the year :year
-  # TODO: allow year to be either a year or a decade ('[in the] 1960s') 
+  # TODO: allow year to be either a year or a decade ('[in the] 1960s')
   #
   def movies(m, params)
     who = params[:who].to_s

data/data/rbot/plugins/insult.rb

@@ -134,7 +134,7 @@ class InsultPlugin < Plugin
 ]
 
 ##
-# Amounts 
+# Amounts
 ##
 @@amt = [
 "accumulation",
@@ -206,7 +206,7 @@ class InsultPlugin < Plugin
 "cold sores",
 "anal warts",
 ]
-  
+
   def help(plugin, topic="")
     return "[msg]insult me|<person> => insult you or <person>. msginsult insults in private"
   end

data/data/rbot/plugins/karma.rb

@@ -14,16 +14,17 @@ class KarmaPlugin < Plugin
     @registry.set_default(0)
 
     # import if old file format found
-    if(File.exist?("#{@bot.botclass}/karma.rbot"))
+    oldkarma = @bot.path 'karma.rbot'
+    if File.exist? oldkarma
       log "importing old karma data"
-      IO.foreach("#{@bot.botclass}/karma.rbot") do |line|
+      IO.foreach(oldkarma) do |line|
         if(line =~ /^(\S+)<=>([\d-]+)$/)
           item = $1
           karma = $2.to_i
           @registry[item] = karma
         end
       end
-      File.delete("#{@bot.botclass}/karma.rbot")
+      File.delete oldkarma
     end
   end
 
@@ -54,7 +55,7 @@ class KarmaPlugin < Plugin
     @registry[thing] = params[:val].to_i
     karma(m, params)
   end
-  
+
   def help(plugin, topic="")
     "karma module: Listens to everyone's chat. <thing>++/<thing>-- => increase/decrease karma for <thing>, karma for <thing>? => show karma for <thing>, karmastats => show stats. Karma is a community rating system - only in-channel messages can affect karma and you cannot adjust your own."
   end
@@ -86,7 +87,7 @@ class KarmaPlugin < Plugin
 
       if op && arg
         ac[arg] ||= 0
-        ac[arg] += (op == '--' ? -1 : 1)
+        ac[arg] += (op == '--' ? -1 : 1) unless arg.downcase == m.sourcenick.downcase
         op = arg = nil
       end
     end