Merge weather plugins

2 files changed, 127 insertions, 128 deletions

diff --git a/data/rbot/plugins/weather.rb b/data/rbot/plugins/weather.rb
index e6284e45..7a48ef23 100644
--- a/data/rbot/plugins/weather.rb
+++ b/data/rbot/plugins/weather.rb
@@ -1,11 +1,15 @@
-#
 # Weather plugin for rbot
-# by MrChucho (mrchucho@mrchucho.net)
+#
+# NOAA National Weather Service support by MrChucho (mrchucho@mrchucho.net)
 # Copyright (C) 2006 Ralph M. Churchill
 #
-require 'open-uri'
+# Weather Undeground support by Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
+# Copyright (C) 2006 Giuseppe Bilotta
+
+require 'uri'
 require 'rexml/document'
 
+# Wraps NOAA National Weather Service information
 class CurrentConditions
     def initialize(station)
         @station = station
@@ -58,36 +62,78 @@ private
     end
 end
 
-class MyWeatherPlugin < Plugin
+class WeatherPlugin < Plugin
   
   def help(plugin, topic="")
-    "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" 
+    case topic
+    when "nws"
+      "weather nws <station> => display the current conditions at the location specified by the NOAA National Weather Service station code <station> ( lookup your station code at http://www.nws.noaa.gov/data/current_obs/ )"
+    when "station", "wu"
+      "weather <location> => display the current conditions at the location specified, looking it up on the Weather Underground site; you can use 'station <code>' to look up data by station code ( lookup your station code at http://www.weatherunderground.com/ )" 
+    else
+      "weather information lookup. Looks up weather information for the last location you specified. See topics 'nws' and 'wu' for more information"
+    end
   end
   
   def initialize
     super
-    # this plugin only wants to store strings
-    class << @registry
-      def store(val)
-        val
+
+    @nws_cache = Hash.new
+
+    @wu_url="http://mobile.wunderground.com/cgi-bin/findweather/getForecast?brand=mobile&query=%s"
+    @wu_station_url="http://mobile.wunderground.com/global/stations/%s.html"
+  end
+  
+  def weather(m, params)
+    loc = nil
+    service = nil
+    if params[:where].empty?
+      if @registry.has_key?(m.sourcenick)
+        where = @registry[m.sourcenick]
+        debug "Loaded weather info #{where.inspect} for #{m.sourcenick}"
+        save = false
+      else
+        debug "No weather info for #{m.sourcenick}"
+        m.reply "I don't know where you are yet, #{m.sourcenick}. See 'help weather nws' or 'help weather wu' for additional help"
+        return
       end
-      def restore(val)
-        val
+    else
+      where = params[:where]
+      save = true
+      unless ['nws','station'].include?(where.first)
+        loc = where.to_s
+        service = :wu
       end
     end
-    @cc_cache = Hash.new
+    unless service
+      loc = where[1].to_s
+      service = where.first.to_sym
+    end
+    if loc.empty?
+      debug "No weather location found for #{m.sourcenick}"
+      m.reply "I don't know where you are yet, #{m.sourcenick}. See 'help weather nws' or 'help weather wu' for additional help"
+    end
+    case service
+    when :nws
+      nws_describe(m, loc)
+    when :station
+      wu_station(m, loc)
+    when :wu
+      wu_weather(m, loc)
+    end
+    @registry[m.sourcenick] = [service, loc] if save
   end
-  
-  def describe(m, where)
-    if @cc_cache.has_key?(where) then
-        met = @cc_cache[where]
+
+  def nws_describe(m, where)
+    if @nws_cache.has_key?(where) then
+        met = @nws_cache[where]
     else
         met = CurrentConditions.new(where)
     end
     if met
       begin
         m.reply met.update
-        @cc_cache[where] = met
+        @nws_cache[where] = met
       rescue => e
         m.reply e.message
       end
@@ -96,20 +142,73 @@ class MyWeatherPlugin < Plugin
     end
   end
 
-  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)
+  def wu_station(m, where)
+    begin
+      xml = @bot.httputil.get_cached(@wu_station_url % URI.escape(where))
+      case xml
+      when nil
+        m.reply "couldn't retrieve weather information, sorry"
+        return
+      when /<table border.*?>(.*?)<\/table>/m
+        data = $1
+        m.reply wu_weather_filter(data)
+      else
+        debug xml
+        m.reply "something went wrong with the data for #{where}..."
+      end
+    rescue => e
+      m.reply "retrieving info about '#{where}' failed (#{e})"
+    end
+  end
+
+  def wu_weather(m, where)
+    begin
+      xml = @bot.httputil.get_cached(@wu_url % URI.escape(where))
+      case xml
+      when nil
+        m.reply "couldn't retrieve weather information, sorry"
+        return
+      when /City Not Found/
+        m.reply "no such location found (#{where})"
+        return
+      when /<table border.*?>(.*?)<\/table>/m
+        data = $1
+        m.reply wu_weather_filter(data)
+      when /<a href="\/global\/stations\//
+        stations = xml.scan(/<a href="\/global\/stations\/(.*?)\.html">/)
+        m.reply "multiple stations available, use 'weather station <code>' where code is one of " + stations.join(", ")
       else
-        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."
+        debug xml
+        m.reply "something went wrong with the data from #{where}..."
       end
+    rescue => e
+      m.reply "retrieving info about '#{where}' failed (#{e})"
+    end
+  end
+
+  def wu_weather_filter(stuff)
+    txt = stuff
+    txt.gsub!(/[\n\s]+/,' ')
+    data = Hash.new
+    txt.gsub!(/&nbsp;/, ' ')
+    txt.gsub!(/&#176;/, ' ') # degree sign
+    txt.gsub!(/<\/?b>/,'')
+    txt.gsub!(/<\/?span[^<>]*?>/,'')
+    txt.gsub!(/<img\s*[^<>]*?>/,'')
+    txt.gsub!(/<br\s?\/?>/,'')
+
+    result = Array.new
+    if txt.match(/<\/a>\s*Updated:\s*(.*?)\s*Observed at\s*(.*?)\s*<\/td>/)
+      result << ("Weather info for %s (updated on %s)" % [$2, $1])
     end
+    txt.scan(/<tr>\s*<td>\s*(.*?)\s*<\/td>\s*<td>\s*(.*?)\s*<\/td>\s*<\/tr>/) { |k, v|
+      unless v.empty? or v == "-" or k =="Raw METAR"
+        result << ("%s: %s" % [k, v])
+      end
+    }
+    return result.join('; ')
   end
 end
 
-plugin = MyWeatherPlugin.new
-plugin.map 'weather :where', :defaults => {:where => false}
+plugin = WeatherPlugin.new
+plugin.map 'weather *where', :defaults => {:where => false}
diff --git a/data/rbot/plugins/wuweather.rb b/data/rbot/plugins/wuweather.rb
deleted file mode 100644
index d02e9a63..00000000
--- a/data/rbot/plugins/wuweather.rb
+++ /dev/null
@@ -1,100 +0,0 @@
-# Weather plugin for rbot

-# Copyright (C) 2006 Giuseppe Bilotta

-#

-# Author: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>

-

-require 'uri'

-

-class WuWeatherPlugin < Plugin

-  

-  def help(plugin, topic="")

-    "weather <location> => display the current conditions at the location specified; you can use 'station <code>' to look up data by station code ( lookup your station code at http://www.weatherunderground.com/ )" 

-  end

-  

-  def initialize

-    super

-    @url="http://mobile.wunderground.com/cgi-bin/findweather/getForecast?brand=mobile&query=%s"

-    @station_url="http://mobile.wunderground.com/global/stations/%s.html"

-  end

-

-  def station(m, params)

-    where = params[:where]

-    begin

-      unless where

-        m.reply "I don't know where you are, #{m.sourcenick} (and you can't set it yet)"

-      end

-      xml = @bot.httputil.get_cached(@station_url % URI.escape(where))

-      case xml

-      when nil

-        m.reply "couldn't retrieve weather information, sorry"

-        return

-      when /<table border.*?>(.*?)<\/table>/m

-        data = $1

-        m.reply weather_filter(data)

-      else

-        debug xml

-        m.reply "something went wrong with the data for #{where}..."

-      end

-    rescue => e

-      m.reply "retrieving info about '#{where}' failed (#{e})"

-    end

-  end

-

-  def weather(m, params)

-    where = params[:where].to_s

-    begin

-      if where.empty?

-        m.reply "I don't know where you are, #{m.sourcenick} (and you can't set it yet)"

-      end

-      xml = @bot.httputil.get_cached(@url % URI.escape(where))

-      case xml

-      when nil

-        m.reply "couldn't retrieve weather information, sorry"

-        return

-      when /City Not Found/

-        m.reply "no such location found (#{where})"

-        return

-      when /<table border.*?>(.*?)<\/table>/m

-        data = $1

-        m.reply weather_filter(data)

-      when /<a href="\/global\/stations\//

-        stations = xml.scan(/<a href="\/global\/stations\/(.*?)\.html">/)

-        m.reply "multiple stations available, use 'weather station <code>' where code is one of " + stations.join(", ")

-      else

-        debug xml

-        m.reply "something went wrong with the data from #{where}..."

-      end

-    rescue => e

-      m.reply "retrieving info about '#{where}' failed (#{e})"

-    end

-  end

-

-  def weather_filter(stuff)

-    txt = stuff

-    txt.gsub!(/[\n\s]+/,' ')

-    data = Hash.new

-    txt.gsub!(/&nbsp;/, ' ')

-    txt.gsub!(/&#176;/, ' ') # degree sign

-    txt.gsub!(/<\/?b>/,'')

-    txt.gsub!(/<\/?span[^<>]*?>/,'')

-    txt.gsub!(/<img\s*[^<>]*?>/,'')

-    txt.gsub!(/<br\s?\/?>/,'')

-

-    result = Array.new

-    if txt.match(/<\/a>\s*Updated:\s*(.*?)\s*Observed at\s*(.*?)\s*<\/td>/)

-      result << ("Weather info for %s (updated on %s)" % [$2, $1])

-    end

-    txt.scan(/<tr>\s*<td>\s*(.*?)\s*<\/td>\s*<td>\s*(.*?)\s*<\/td>\s*<\/tr>/) { |k, v|

-      unless v.empty? or v == "-" or k =="Raw METAR"

-        result << ("%s: %s" % [k, v])

-      end

-    }

-    return result.join('; ')

-  end

-

-end

-

-plugin = WuWeatherPlugin.new

-plugin.map 'weather station :where', :action => 'station', :defaults => {:where => false}

-plugin.map 'weather *where', :defaults => {:where => false}

-