X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=data%2Frbot%2Fplugins%2Fgeoip.rb;h=615511728df524469571276977a16965ac1579f7;hb=52c9564c96dbf95e5c883e41940527f2394ba55a;hp=15372a8ba7f4fa1d75f8f64e6ac83a271922d510;hpb=9125a2e6d1ef8af8d380da3766c704f9028dec69;p=user%2Fhenk%2Fcode%2Fruby%2Frbot.git

diff --git a/data/rbot/plugins/geoip.rb b/data/rbot/plugins/geoip.rb
index 15372a8b..61551172 100755
--- a/data/rbot/plugins/geoip.rb
+++ b/data/rbot/plugins/geoip.rb
@@ -9,23 +9,61 @@
 #
 # Resolves the geographic locations of users (network-wide) and IP addresses
 
-module GeoIP
-  GEO_IP = "http://www.geoiptool.com/en/?IP="
+module ::GeoIP
+  class InvalidHostError < 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
+    :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
   }
 
+  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)
-    res = {}
-    raw = Irc::Utils.bot.httputil.get_response(GEO_IP+hostname)
-    raw = raw.decompress_body(raw.raw_body)
+    raise InvalidHostError unless valid_host?(hostname)
 
-    REGEX.each { |key, regex| res[key] = Iconv.conv('utf-8', 'ISO-8859-1', raw.scan(regex).to_s) }
+    yaml = Irc::Utils.bot.httputil.get(GEO_IP_PRIMARY+hostname)
 
-    return res
+    if yaml
+      return YAML::load(yaml)
+    else
+      res = {}
+      raw = Irc::Utils.bot.httputil.get_response(GEO_IP_SECONDARY+hostname)
+      raw = raw.decompress_body(raw.raw_body)
+
+      REGEX.each { |key, regex| res[key] = Iconv.conv('utf-8', 'ISO-8859-1', raw.scan(regex).to_s) }
+
+      return res
+    end
+  end
+end
+
+class Stack
+  def initialize
+    @hash = {}
+  end
+
+  def [](nick)
+    @hash[nick] = [] unless @hash[nick]
+    @hash[nick]
+  end
+
+  def has_nick?(nick)
+    @hash.has_key?(nick)
+  end
+
+  def clear(nick)
+    @hash.delete(nick)
   end
 end
 
@@ -34,17 +72,28 @@ class GeoIpPlugin < Plugin
     "geoip [<user|hostname|ip>] => returns the geographic location of whichever has been given -- note: user can be anyone on the network"
   end
 
+  def initialize
+    super
+
+    @stack = Stack.new
+  end
+
   def whois(m)
+    nick = m.whois[:nick].downcase
+
     # need to see if the whois reply was invoked by this plugin
-    return unless m.whois[:nick] == @nick
+    return unless @stack.has_nick?(nick)
 
     if m.target
-      @bot.say @source, host2output(m.target.host, m.target.nick)
+      msg = host2output(m.target.host, m.target.nick)
     else
-      @bot.say @source, "no such user on "+@bot.server.hostname.split(".")[-2]
+      msg = "no such user on "+@bot.server.hostname.split(".")[-2]
+    end
+    @stack[nick].each do |source|
+      @bot.say source, msg
     end
 
-    @nick, @source = nil
+    @stack.clear(nick)
   end
 
   def geoip(m, params)
@@ -63,24 +112,29 @@ class GeoIpPlugin < Plugin
       end
 
       # input is a host name or an IP
-      if params[:input] =~ /[a-z0-9\-]+(?:\.[a-z0-9\-]+)*\.[a-z]{2,3}/i ||
-         params[:input] =~ Resolv::IPv4::Regex
-        m.reply host2output(params[:input])
+      if GeoIP::valid_host?(params[:input])
+         m.reply host2output(params[:input])
 
       # assume input is a nick
-      else
-        @source = m.replyto
-        @nick   = params[:input]
+      elsif params[:input] !~ /\./
+        nick = params[:input].downcase
 
-        @bot.whois(@nick)
+        @stack[nick] << m.replyto
+        @bot.whois(nick)
+      else
+        m.reply "invalid input"
       end
     end
   end
 
   def host2output(host, nick=nil)
-    geo = GeoIP::resolve(host)
+    return "127.0.0.1 could not be res.. wait, what?" if host == "127.0.0.1"
+
+    begin
+      geo = GeoIP::resolve(host)
 
-    if geo[:country].empty?
+      raise if geo[:country].empty?
+    rescue GeoIP::InvalidHostError, RuntimeError
       return _("#{nick ? "#{nick}'s location" : host} could not be resolved")
     end
 
@@ -91,7 +145,7 @@ class GeoIpPlugin < Plugin
 
     res << " %{city}," % {
       :city => geo[:city]
-    } unless geo[:city].empty?
+    } unless geo[:city].to_s.empty?
 
     res << " %{country}" % {
       :country => geo[:country]
@@ -99,11 +153,11 @@ class GeoIpPlugin < Plugin
 
     res << " (%{region})" % {
       :region  => geo[:region]
-    } unless geo[:region].empty? || geo[:region] == geo[:city]
+    } unless geo[:region].to_s.empty? || geo[:region] == geo[:city]
 
     return res
   end
 end
 
 plugin = GeoIpPlugin.new
-plugin.map "geoip [:input]", :action => 'geoip'
+plugin.map "geoip [:input]", :action => 'geoip', :thread => true