lita-onewheel-filmplot 0.0.2 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 63fdf561eee692ec695744a1a6375546c0b1f3e1a7261432a7df6c786cc32d61
-  data.tar.gz: c69022b2f335234ba66257d15a2d16dea18b43d43ca489d7c510da6f6391ffdf
+  metadata.gz: 015d87a547242abace2df04bc11022d8868cc56226bab10022be706bc8935265
+  data.tar.gz: 34999c17449f39b45c5eb4fde6ab1a278786f1443f4eded6d1b209e2da8c3067
 SHA512:
-  metadata.gz: dc3fd466097e604724a6d8594483cf4ae79f66be59785e1bf2621ae29d3fb201052aa74015fc8c5cee9b5b3c8e3de5414ad5c0b68d0ffcbee6d12c9a968e8ce3
-  data.tar.gz: 513019dcc5152680be0d8cfa7c1a0eb0061b34ee766e472ee3ca286650a38321a3c8cfc85b1956049da88fbf63b31c7f8ea6d22af85a4283e152d2672ec561ea
+  metadata.gz: c9b12ae6a0f005f998b5a79c3a952d5b73a16e310ea0a3e6c6a53c6a31202173a2a40026d862e6badb2dfc3913ecb371e702a3941b0d37c76aa7912c592dd0f0
+  data.tar.gz: 384e83cc3d05214d546647b80a5fe42e1538bbdcd7ee143c3ca06c0c3ff19668a6666291838e1d281af9233384fdddc7c5e13c414676eb2fb24026d7d9969717

data/lib/lita/handlers/onewheel_filmplot.rb

@@ -1,5 +1,6 @@
 require 'rest-client'
 require 'nokogiri'
+require 'cgi'
 
 module Lita
   module Handlers
@@ -17,11 +18,34 @@ module Lita
             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."

data/lita-onewheel-filmplot.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = 'lita-onewheel-filmplot'
-  spec.version       = '0.0.2'
+  spec.version       = '0.1.0'
   spec.authors       = ['Andrew Kreps']
   spec.email         = ['andrew.kreps@gmail.com']
   spec.description   = 'Film plot retrieval bot.'

data/spec/lita/handlers/onewheel_filmplot_spec.rb

@@ -4,22 +4,22 @@ describe Lita::Handlers::OnewheelFilmplot, lita_handler: true do
   it { is_expected.to route_command('filmplot x') }
   it { is_expected.to route_command('plotline x') }
 
-  def mock(file)
+  def mock(file, param)
     mock = File.open("spec/fixtures/#{file}.html").read
-    allow(RestClient).to receive(:get) { mock }
+    allow(RestClient).to receive(:get).with(param) { mock }
   end
 
   it 'should get timecop' do
-    mock('timecop')
+    mock('timecop', 'https://www.rottentomatoes.com/m/timecop')
     send_command('filmplot timecop')
     expect(replies.last).to eq "Based on a comic book story, this futuristic film follows the time-travel exploits of policeman Max Walker (Jean Claude Van Damme). In 1994, Walker's wife Melissa (Mia Sara) is about to tell him that she is expecting their first child when they are attacked by a group of criminals. Walker is shot and beaten and lies helplessly on his lawn while he sees their home and his wife blown up by the killers. Ten years later, Walker remains an employee of the Time Enforcement Commission, a federal agency which was set up in 1994 after the U.S. government learned that time travel technology is feasible. The commission's role is to prevent time travel to protect U.S. economic interests. Walker learns that the corrupt Senator McComb (Ron Silver), who helped establish the agency, is exploiting it for personal gain, trying to establish a monopoly on time travel so that he can enrich himself in the stock market. Walker travels back in time to stop McComb from murdering his former partner. At the same time, Walker hopes to rescue his wife, and he learns that the attack on his home was ordered by McComb to stop Walker from foiling his plans. ~ Michael Betzold, Rovi"
   end
 
-  # it 'should find infinity war by the wrong name' do
-  #   mock('search')
-  #   mock('infinity')
-  #
-  # end
+  it 'should find infinity war by the wrong name' do
+    # mock('search', 'https://www.rottentomatoes.com/search/?search=infinity%20war')
+    # mock('infinity', 'https://www.rottentomatoes.com/m/avengers_infinity_war')
+    send_command('filmplot infinity war')
+  end
 
   it '404s' do
     allow(RestClient).to receive(:get).and_raise(RestClient::ResourceNotFound)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lita-onewheel-filmplot
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
 platform: ruby
 authors:
 - Andrew Kreps