summaryrefslogtreecommitdiff
path: root/data/rbot/plugins
diff options
context:
space:
mode:
authorTom Gilbert <tom@linuxbrit.co.uk>2006-02-17 07:27:29 +0000
committerTom Gilbert <tom@linuxbrit.co.uk>2006-02-17 07:27:29 +0000
commit1043b1a026c7362ced11c6bb4cd706a2c0d5cec3 (patch)
tree0d1496bcbf01080ee796f802a24c2cf5b5941f55 /data/rbot/plugins
parent44688c76d937c2dade10aaa7bb7e70e508b33684 (diff)
Ralph Churchill submitted a weather forecast plugin :D
It uses the Yahoo Maps API, too. So you can say: rbot: forecast Tulsa, OK or rbot: forecast 74104
Diffstat (limited to 'data/rbot/plugins')
-rw-r--r--data/rbot/plugins/forecast.rb133
1 files changed, 133 insertions, 0 deletions
diff --git a/data/rbot/plugins/forecast.rb b/data/rbot/plugins/forecast.rb
new file mode 100644
index 00000000..63e3127c
--- /dev/null
+++ b/data/rbot/plugins/forecast.rb
@@ -0,0 +1,133 @@
+#
+# Forecast plugin for rbot
+# by MrChucho (mrchucho@mrchucho.net)
+# Copyright (C) 2006 Ralph M. Churchill
+#
+require 'soap/wsdlDriver'
+require 'open-uri'
+require 'rexml/document'
+require 'erb'
+
+
+class LatLong
+ include ERB::Util
+ # Determine the latitude and longitude of a location. City, State and/or ZIP
+ # are all valid.
+ # [+return+] latitude,longitude
+ def get_lat_long(loc)
+ loc = url_encode(loc)
+ url="http://api.local.yahoo.com/MapsService/V1/geocode?appid=mrchucho_rbot_weather&location=#{loc}"
+ lat,long = 0,0
+ begin
+ open(url) do |xmldoc|
+ results = (REXML::Document.new xmldoc).root
+ lat = results.elements["//Latitude/text()"].to_s
+ long = results.elements["//Longitude/text()"].to_s
+ end
+ rescue => err
+ raise err #?
+ end
+ return lat.to_f,long.to_f
+ end
+end
+
+class Forecast
+ WSDL_URI="http://www.nws.noaa.gov/forecasts/xml/SOAP_server/ndfdXMLserver.php?wsdl"
+ def initialize(lat,long)
+ @lat,@long=lat,long
+ # this extra step is for backward/forward compatibility
+ factory = SOAP::WSDLDriverFactory.new(WSDL_URI)
+ @forecaster=factory.respond_to?(:create_rpc_driver) ?
+ factory.create_rpc_driver : factory.create_driver
+ end
+ def forecast
+ return parse(retrieve),Time.new
+ end
+private
+ def retrieve
+ forecast = @forecaster.NDFDgenByDay(
+ @lat,@long,Time.now.strftime("%Y-%m-%d"),2,"24 hourly")
+ (REXML::Document.new(forecast)).root
+ end
+ def parse(xml)
+ msg = String.new
+ (1..2).each do |day|
+ d = (day==1) ? 'Today' : 'Tomorrow'
+ hi = xml.elements["//temperature[@type='maximum']/value[#{day}]/text()"]
+ lo = xml.elements["//temperature[@type='minimum']/value[#{day}]/text()"]
+ w = xml.elements["//weather/weather-conditions[#{day}]/@weather-summary"]
+ precip_am = xml.elements["//probability-of-precipitation/value[#{day*2-1}]/text()"]
+ precip_pm = xml.elements["//probability-of-precipitation/value[#{day*2}]/text()"]
+ msg += "#{d}: Hi #{hi} Lo #{lo}, #{w}. Precip: AM #{precip_am}% PM #{precip_pm}%\n"
+ end
+ msg
+ end
+end
+
+class ForecastPlugin < Plugin
+ USAGE='forecast <location> => show the 2-day forecast for a location. Location can be any combination of City, State, Country and ZIP'
+ def help(plugin,topic="")
+ USAGE
+ end
+ def usage(m,params={})
+ m.reply USAGE
+ end
+ def initialize
+ super
+ # this plugin only wants to store strings
+ class << @registry
+ def store(val)
+ val
+ end
+ def restore(val)
+ val
+ end
+ end
+ @forecast_cache = Hash.new
+ end
+
+ def forecast(m,params)
+ if params[:location] and params[:location].any?
+ loc = params[:location].join
+ @registry[m.sourcenick] = loc
+ get_forecast(m,loc)
+ else
+ if @registry.has_key?(m.sourcenick) then
+ loc = @registry[m.sourcenick]
+ get_forecast(m,loc)
+ else
+ m.reply "Please specifiy the City, State or ZIP"
+ end
+ end
+ end
+
+ def get_forecast(m,loc)
+ begin
+ if @forecast_cache.has_key?(loc) and
+ Time.new - @forecast_cache[loc][:date] < 3600
+ forecast = @forecast_cache[loc][:forecast]
+ forecast_date = @forecast_cache[loc][:date]
+ else
+ begin
+ l = LatLong.new
+ f = Forecast.new(*l.get_lat_long(loc))
+ forecast,forecast_date = f.forecast
+ rescue => err
+ m.reply err
+ end
+ end
+ if forecast
+ m.reply forecast
+ @forecast_cache[loc] = {}
+ @forecast_cache[loc][:forecast] = forecast
+ @forecast_cache[loc][:date] = forecast_date
+ else
+ m.reply "Couldn't find forecast for #{loc}"
+ end
+ rescue => e
+ m.reply "ERROR: #{e}"
+ end
+ end
+end
+plugin = ForecastPlugin.new
+plugin.map 'forecast *location', :defaults => {:location => false}