Remove unused code from rss

[user/henk/code/ruby/rbot.git] / data / rbot / plugins / weather.rb

diff --git a/data/rbot/plugins/weather.rb b/data/rbot/plugins/weather.rb
index 3e4134e4b4c70c88531337ff316976f1f29e35aa..e6284e458e49a0bb84ce97580ec260708509d077 100644 (file)
--- a/data/rbot/plugins/weather.rb
+++ b/data/rbot/plugins/weather.rb
@@ -1,7 +1,67 @@
-class WeatherPlugin < Plugin
+#
+# Weather plugin for rbot
+# by MrChucho (mrchucho@mrchucho.net)
+# Copyright (C) 2006 Ralph M. Churchill
+#
+require 'open-uri'
+require 'rexml/document'
+
+class CurrentConditions
+    def initialize(station)
+        @station = station
+        @url = "http://www.nws.noaa.gov/data/current_obs/#{@station.upcase}.xml"
+        @etag = String.new
+        @mtime = Time.mktime(0)
+        @current_conditions = String.new
+        @iscached = false
+    end
+    def update
+        begin
+            open(@url,"If-Modified-Since" => @mtime.rfc2822) do |feed|
+            # open(@url,"If-None-Match"=>@etag) do |feed|
+                @etag = feed.meta['etag']
+                @mtime = feed.last_modified
+                cc_doc = (REXML::Document.new feed).root
+                @iscached = false
+                @current_conditions = parse(cc_doc)
+            end
+        rescue OpenURI::HTTPError => e
+            case e
+            when /304/:
+                @iscached = true
+            when /404/:
+                raise "Data for #{@station} not found"
+            else
+                raise "Error retrieving data: #{e}"
+            end
+        end
+        @current_conditions # +" Cached? "+ ((@iscached) ? "Y" : "N")
+    end
+    def parse(cc_doc)
+        cc = Hash.new
+        cc_doc.elements.each do |c|
+            cc[c.name.to_sym] = c.text
+        end
+        "At #{cc[:observation_time_rfc822]}, the wind was #{cc[:wind_string]} at #{cc[:location]} (#{cc[:station_id]}). The temperature was #{cc[:temperature_string]}#{heat_index_or_wind_chill(cc)}, and the pressure was #{cc[:pressure_string]}. The relative humidity was #{cc[:relative_humidity]}%. Current conditions are #{cc[:weather]} with #{cc[:visibility_mi]}mi visibility."
+    end
+private
+    def heat_index_or_wind_chill(cc)
+        hi = cc[:heat_index_string]
+        wc = cc[:windchill_string]
+        if hi != 'NA' then
+            " with a heat index of #{hi}"
+        elsif wc != 'NA' then
+            " with a windchill of #{wc}"
+        else
+            ""
+        end
+    end
+end
+
+class MyWeatherPlugin < Plugin
   
   def help(plugin, topic="")
-    "weather <ICAO> => display the current weather at the location specified by the ICAO code [Lookup your ICAO code at http://www.nws.noaa.gov/oso/siteloc.shtml] - this will also store the ICAO against your nick, so you can later just say \"weather\", weather => display the current weather at the location you last asked for"
+    "weather <STATION> => display the current conditions at the location specified by the STATION code [Lookup your STATION code at http://www.nws.noaa.gov/data/current_obs/ - this will also store the STATION against your nick, so you can later just say \"weather\", weather => display the current weather at the location you last asked for" 
   end
   
   def initialize
@@ -15,41 +75,41 @@ class WeatherPlugin < Plugin
         val
       end
     end
-    @metar_cache = Hash.new
+    @cc_cache = Hash.new
   end
   
   def describe(m, where)
-    if @metar_cache.has_key?(where) &&
-       Time.now - @metar_cache[where].date < 3600
-      met = @metar_cache[where]
+    if @cc_cache.has_key?(where) then
+        met = @cc_cache[where]
     else
-      met = Utils.get_metar(where)
+        met = CurrentConditions.new(where)
     end
-    
     if met
-      m.reply met.pretty_print
-      @metar_cache[where] = met
+      begin
+        m.reply met.update
+        @cc_cache[where] = met
+      rescue => e
+        m.reply e.message
+      end
     else
       m.reply "couldn't find weather data for #{where}"
     end
   end
-  
-  def privmsg(m)
-    case m.params
-    when nil
+
+  def weather(m, params)
+    if params[:where]
+      @registry[m.sourcenick] = params[:where]
+      describe(m,params[:where])
+    else
       if @registry.has_key?(m.sourcenick)
         where = @registry[m.sourcenick]
         describe(m,where)
       else
-        m.reply "I don't know where #{m.sourcenick} is!"
+        m.reply "I don't know where you are yet! Lookup your station at http://www.nws.noaa.gov/data/current_obs/ and tell me 'weather <station>', then I'll know."
       end
-    when (/^(\S{4})$/)
-      where = $1
-      @registry[m.sourcenick] = where
-      describe(m,where)
     end
   end
-  
 end
-plugin = WeatherPlugin.new
-plugin.register("weather")
+
+plugin = MyWeatherPlugin.new
+plugin.map 'weather :where', :defaults => {:where => false}