diff options
| -rwxr-xr-x | data/rbot/plugins/geoip.rb | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/data/rbot/plugins/geoip.rb b/data/rbot/plugins/geoip.rb new file mode 100755 index 00000000..2fde9594 --- /dev/null +++ b/data/rbot/plugins/geoip.rb @@ -0,0 +1,83 @@ +#-- vim:sw=2:et +#++ +# +# :title: Geo IP Plugin +# +# Author:: Raine Virta <rane@kapsi.fi> +# Copyright:: (C) 2008 Raine Virta +# License:: GPL v2 +# +# Resolves the geographic locations of users and IP addresses + +module GeoIP + GEO_IP = "http://www.geoiptool.com/en/?IP=" + + def self.resolve(hostname) + raw = Irc::Utils.bot.httputil.get(GEO_IP+hostname, :cache => true) + + { + :country => raw.scan(%r{Country:.*?<a href=".*?" target="_blank"> (.*?)</a>}m).to_s, + :region => raw.scan(%r{Region:.*?<a href=".*?" target="_blank">(.*?)</a>}m).to_s, + :city => raw.scan(%r{City:.*?<td align="left" class="arial_bold">(.*?)</td>}m).to_s + } + end +end + +class GeoIpPlugin < Plugin + def help(plugin, topic="") + "geoip [<user|hostname|ip>] => returns the geographic location of whichever has been given" + end + + def geoip(m, params) + if params.empty? + m.reply host2output(m.source.host, m.source.nick) + else + if m.replyto.class == Channel + # check if there is an user with nick same as input given + user = m.replyto.users.find { |user| user.nick == params[:input] } + + if user + m.reply host2output(user.host, user.nick) + return + end + end + + if params[:input] =~ /[a-z0-9\-]+(?:\.[a-z0-9\-]+)*\.[a-z]{2,3}/i || + params[:input] =~ Resolv::IPv4::Regex + m.reply host2output(params[:input]) + else + m.reply "invalid input" + end + end + end + + def host2output(host, nick=nil) + geo = GeoIP::resolve(host) + + if geo[:country].empty? + return _("#{host} could not be resolved") + end + + res = _("%{thing} is #{nick ? "from" : "located in"}") % { + :thing => (nick ? nick : Resolv::getaddress(host)), + :country => geo[:country] + } + + res << " %{city}," % { + :city => geo[:city] + } unless geo[:city].empty? + + res << " %{country}" % { + :country => geo[:country] + } + + res << " (%{region})" % { + :region => geo[:region] + } unless geo[:region].empty? || geo[:region] == geo[:city] + + return res + end +end + +plugin = GeoIpPlugin.new +plugin.map "geoip [:input]", :action => 'geoip' |