1 files changed, 148 insertions, 0 deletions
diff --git a/data/rbot/plugins/youtube.rb b/data/rbot/plugins/youtube.rb
new file mode 100644
index 00000000..f82d9985
--- /dev/null
+++ b/data/rbot/plugins/youtube.rb
@@ -0,0 +1,148 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: YouTube plugin for rbot
+#
+# Author:: Giuseppe "Oblomov" Bilotta <giuseppe.bilotta@gmail.com>
+#
+# Copyright:: (C) 2008 Giuseppe Bilotta
+
+
+class YouTubePlugin < Plugin
+  YOUTUBE_SEARCH = "http://gdata.youtube.com/feeds/api/videos?vq=%{words}&orderby=relevance"
+
+  Config.register Config::IntegerValue.new('youtube.hits',
+    :default => 3,
+    :desc => "Number of hits to return from YouTube searches")
+  Config.register Config::IntegerValue.new('youtube.descs',
+    :default => 3,
+    :desc => "When set to n > 0, the bot will return the description of the first n videos found")
+
+  def youtube_filter(s)
+    loc = Utils.check_location(s, /youtube\.com/)
+    return nil unless loc
+    if s[:text].include? '<div id="vidTitle">'
+      video_info = @bot.filter(:"youtube.video", s)
+      return nil # TODO
+    elsif s[:text].include? '<!-- start search results -->'
+      vids = @bot.filter(:"youtube.search", s)[:videos]
+      if !vids.empty?
+        return nil # TODO
+      end
+    end
+    # otherwise, just grab the proper div
+    if defined? Hpricot
+      content = (Hpricot(s[:text])/"#mainContent").to_html.ircify_html
+    end
+    # suboptimal, but still better than the default HTML info extractor
+    content ||= /<div id="mainContent"[^>]*>/.match(s[:text]).post_match.ircify_html
+    return {:title => s[:text].ircify_html_title, :content => content}
+  end
+
+  def youtube_apisearch_filter(s)
+    vids = []
+    title = nil
+    begin
+      doc = REXML::Document.new(s[:text])
+      title = doc.elements["feed/title"].text
+      doc.elements.each("*/entry") { |e|
+        # TODO precomputing mg doesn't work on my REXML, despite what the doc
+        # says
+        #   mg = e.elements["media:group"]
+        #   :title => mg["media:title"].text
+        # fails because "media:title" is not an Integer. Bah
+        vid = {
+          :author => (e.elements["author/name"].text rescue nil),
+          :title =>  (e.elements["media:group/media:title"].text rescue nil),
+          :desc =>   (e.elements["media:group/media:description"].text rescue nil),
+          :cat => (e.elements["media:group/media:category"].text rescue nil),
+          :seconds => (e.elements["media:group/yt:duration/@seconds"].value.to_i rescue nil),
+          :url => (e.elements["media:group/media:player/@url"].value rescue nil),
+          :rating => (("%s/%s" % [e.elements["media:group/gd:rating/@average"].value, e.elements["media:group/gd:rating/@max"].value]) rescue nil),
+          :views => (e.elements["media:group/yt:statistics/@viewCount"].value rescue nil),
+          :faves => (e.elements["media:group/yt:statistics/@favoriteCount"].value rescue nil)
+        }
+        if vid[:desc]
+          vid[:desc].gsub!(/\s+/m, " ")
+        end
+        if secs = vid[:seconds]
+          mins, secs = secs.divmod 60
+          hours, mins = mins.divmod 60
+          if hours > 0
+            vid[:duration] = "%s:%s:%s" % [hours, mins, secs]
+          elsif mins > 0
+            vid[:duration] = "%s'%s\"" % [mins, secs]
+          else
+            vid[:duration] = "%ss" % [secs]
+          end
+        else
+          vid[:duration] = _("unknown duration")
+        end
+        vids << vid
+      }
+      debug vids
+    rescue => e
+      debug e
+    end
+    return {:title => title, :vids => vids}
+  end
+
+  def youtube_search_filter(s)
+    # TODO
+    # hits = s[:hits] || @bot.config['youtube.hits']
+    # scrap the videos
+    return []
+  end
+
+  def youtube_video_filter(s)
+    # TODO
+  end
+
+  def initialize
+    super
+    @bot.register_filter(:youtube, :htmlinfo) { |s| youtube_filter(s) }
+    @bot.register_filter(:apisearch, :youtube) { |s| youtube_apisearch_filter(s) }
+    @bot.register_filter(:search, :youtube) { |s| youtube_search_filter(s) }
+    @bot.register_filter(:video, :youtube) { |s| youtube_video_filter(s) }
+  end
+
+  def youtube(m, params)
+    what = params[:words].to_s
+    searchfor = CGI.escape what
+    url = YOUTUBE_SEARCH % {:words => searchfor}
+    resp, xml = @bot.httputil.get_response(url)
+    unless Net::HTTPSuccess === resp
+      m.reply(_("error looking for %{what} on youtube: %{e}") % {:what => what, :e => xml})
+      return
+    end
+    debug "filtering XML"
+    vids = @bot.filter(:"youtube.apisearch", DataStream.new(xml, params))[:vids][0, @bot.config['youtube.hits']]
+    debug vids
+    case vids.length
+    when 0
+      m.reply _("no videos found for %{what}") % {:what => what}
+      return
+    when 1
+      show = "%{title} (%{duration}) [%{desc}] @ %{url}" % vids.first
+      m.reply _("One video found for %{what}: %{show}") % {:what => what, :show => show}
+    else
+      idx = 0
+      shorts = vids.inject([]) { |list, el|
+        idx += 1
+        list << ("#{idx}. %{bold}%{title}%{bold} (%{duration}) @ %{url}" % {:bold => Bold}.merge(el))
+      }.join(" | ")
+      m.reply(_("Videos for %{what}: %{shorts}") % {:what =>what, :shorts => shorts},
+              :split_at => /\s+\|\s+/)
+      if (descs = @bot.config['youtube.descs']) > 0
+        vids[0, descs].each_with_index { |v, i|
+          m.reply("[#{i+1}] %{title} (%{duration}): %{desc}" % v, :overlong => :truncate)
+        }
+      end
+    end
+  end
+
+end
+
+plugin = YouTubePlugin.new
+
+plugin.map "youtube *words", :action => 'youtube', :threaded => true