onewheel_filmplot.rb in lita-onewheel-filmplot-0.1.0

- old
+ new

@@ -1,7 +1,8 @@
 require 'rest-client'
 require 'nokogiri'
+require 'cgi'
 
 module Lita
   module Handlers
     class OnewheelFilmplot < Handler
       config :api_key
@@ -15,14 +16,37 @@
       route /^plotline\s+(.*)$/i,
             :get_plot,
             command: true
 
       def get_plot(response)
-        movie_slug = response.matches[0][0].gsub ' ', '_'
+
+        title = response.matches[0][0]
+        url_root = 'https://www.rottentomatoes.com'
+
+        if title.match? '\s'
+          # search
+
+          search_term = CGI.escape(title)
+          search_url = "https://www.rottentomatoes.com/search/?search=#{search_term}"
+          Lita.logger.debug("Search mode!  #{search_url}")
+          r = RestClient.get(search_url)
+          if match = r.match(/RT.PrivateApiV2FrontendHost, '.*', ({.*})/)
+            movie_json = JSON.parse match[1]
+            slug = movie_json['movies'][0]['url']
+            get_url = "#{url_root}#{slug}"
+            Lita.logger.debug("Searching for #{get_url}")
+          end
+        else
+          movie_slug = title.gsub ' ', '_'
+          get_url = "#{url_root}/m/#{movie_slug}"
+          Lita.logger.debug("Getting #{get_url}")
+        end
+
         begin
-          r = RestClient.get("https://www.rottentomatoes.com/m/#{movie_slug}")
+          r = RestClient.get(get_url)
           noko_doc = Nokogiri::HTML(r)
           node = noko_doc.css('div#movieSynopsis')
+          Lita.logger.debug("Replying with #{node.text.strip}")
           response.reply node.text.strip
         rescue RestClient::ResourceNotFound => e
           response.reply "#{movie_slug} not found."
         end
       end