popularity 0.1.1 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8f43df47453436c7ba42f3d70d5196f939f518a5
-  data.tar.gz: 8261963d61f74d92a8b06767d936a843d2d04c45
+  metadata.gz: 3c17e6a773a83fc813ffd30d5c14c2f72691c9e5
+  data.tar.gz: cd24795291873364d6383a473d94d191b21e83c9
 SHA512:
-  metadata.gz: 5d867060540e84600a65bf8f33202314cc245466743a9b1acf64630c91964fe620624d56b435c6558bba88c9fb12ce54da34a201d5cc5be22c1d6ff45d97a95d
-  data.tar.gz: 5e98e6ab95d4b38dc6e957523107d5e867a9712e517a43abb58c9f3d98ab992b48272f7d27c007ea091432a56789e1a65b3910278e85f3645b8240c92cf38bfa
+  metadata.gz: 473ced1189eab299c44f58b6fe32025c89e37c88b37311894adb3696ef9511ac282fbc8518d070a593692084ec281d1130106fbaed0459078a27e29598924f61
+  data.tar.gz: 01bfbee1959f78f8887da8dd1070ab31389ae7ac5537a2efc25d9cb3e4d2bca0abd2e59e4f46d86cd7f012e439978e4ce9317706711a17b8cd17be43f3dbea8f

data/.travis.yml

@@ -0,0 +1,5 @@
+language: ruby
+
+rvm:
+  - "2.0.0"
+  - "2.1.1"

data/Gemfile

@@ -3,9 +3,10 @@ source "http://rubygems.org"
 gem 'open_uri_redirections', '~> 0'
 gem 'json', '~> 1.8'
 gem 'unirest', '~> 1'
+gem 'pry'
 
 group :development, :test do
-  gem "pry", '~> 0.10'
+  # gem "pry", '~> 0.10'
   gem "shoulda", ">= 0"
   gem "rdoc", "~> 3.12"
   gem "bundler", "~> 1.0"

data/Gemfile.lock

@@ -116,7 +116,7 @@ DEPENDENCIES
   jeweler (~> 2.0)
   json (~> 1.8)
   open_uri_redirections (~> 0)
-  pry (~> 0.10)
+  pry
   rdoc (~> 3.12)
   rspec (~> 3.2)
   shoulda

data/README.md

@@ -50,7 +50,7 @@ Sometimes multiple URLs refer to the same thing. Maybe your URLs changed, but yo
 JSON: 
 
 ```ruby
-> search.to_json 
+> search.as_json 
 #=> spits out by network, by url, and total stats
 ```
 

data/Rakefile

@@ -25,20 +25,17 @@ Jeweler::Tasks.new do |gem|
 end
 Jeweler::RubygemsDotOrgTasks.new
 
-require 'rake/testtask'
-Rake::TestTask.new(:test) do |test|
-  test.libs << 'lib' << 'test'
-  test.pattern = 'test/**/test_*.rb'
-  test.verbose = true
-end
-
 desc "Code coverage detail"
 task :simplecov do
   ENV['COVERAGE'] = "true"
-  Rake::Task['test'].execute
+  Rake::Task['spec'].execute
 end
 
-task :default => :test
+task :default => [:spec]
+desc 'run Rspec specs'
+task :spec do
+  sh 'rspec spec'
+end
 
 require 'rdoc/task'
 Rake::RDocTask.new do |rdoc|

data/VERSION

@@ -1 +1 @@
-0.1.1
+0.2.1

data/lib/popularity.rb

@@ -7,7 +7,7 @@ module Popularity
   TYPES = []
 end
 
-Gem.find_files("popularity/networks/*.rb").each { |path| 
+Gem.find_files("popularity/networks/*.rb").each { |path|
   require path
   file_name   = path.split('/').last.split('.').first
   class_name  = file_name.gsub(/_[a-z]|^[a-z]/, &:upcase).gsub('_', '')
@@ -18,8 +18,6 @@ Gem.find_files("popularity/networks/*.rb").each { |path|
 
 module Popularity
   def self.search(*urls)
-  	response = {}
-
     MultiSearch.new(:urls => urls)
   end
 
@@ -43,20 +41,20 @@ module Popularity
           add_search_result(result)
         end
       end
-    end 
+    end
 
     def results
       searches.collect(&:results).reduce(:+)
     end
 
-    def to_json(options = {})      
+    def as_json(options = {})
       json = {}
       self.searches.collect do |search|
-        json[search.url] = search.to_json
+        json[search.url] = search.as_json
       end
 
       self.sources.collect do |source|
-        json[source.to_s] = self.send(source.to_sym).to_json
+        json[source.to_s] = self.send(source.to_sym).as_json
       end
 
       json["total"] = total
@@ -65,22 +63,17 @@ module Popularity
     end
 
     def total
-      total = 0
-      self.searches.each do |a|
-        total += a.total
-      end
-
-      total
+      self.searches.collect(&:total).compact.reduce(:+)
     end
 
     protected
 
     def add_search_result(result)
-      container = self.instance_variable_get("@#{result.name}") 
+      container = self.instance_variable_get("@#{result.name}")
 
       unless container
-        @sources ||= [] 
-        @sources << result.name.to_sym 
+        @sources ||= []
+        @sources << result.name.to_sym
         container = Popularity::ResultsContainer.new
         self.instance_variable_set "@#{result.name}", container
         self.define_singleton_method(result.name.to_sym) { container }
@@ -90,4 +83,4 @@ module Popularity
     end
 
   end
-end
+end

data/lib/popularity/crawler.rb

@@ -2,11 +2,27 @@ require 'open-uri'
 require 'open_uri_redirections'
 require 'json'
 require 'unirest'
+require 'pry'
 
 module Popularity
   class Crawler
     attr_reader :url
 
+    def self.stats(*args)
+      @property_names ||= []
+      args.each do |name|
+        @property_names << name
+      end
+    end
+
+    def self.property_names
+      @property_names
+    end
+
+    def total
+      self.class.property_names.uniq.collect { |n| self.send(n.to_sym) }.select { |t| t.class == Fixnum }.compact.reduce(:+)
+    end
+
     def initialize(url)
       @url = url
     end
@@ -22,13 +38,13 @@ module Popularity
     def has_response?
       response #fetch it
 
-      return false if response.nil? 
+      return false if response.nil?
       return false if response.empty?
 
       true
     end
 
-    def valid? 
+    def valid?
       true # to be overridden in subclasses
     end
 
@@ -48,8 +64,15 @@ module Popularity
       self.class.to_s.split('::').last.gsub(/(.)([A-Z])/,'\1_\2').downcase
     end
 
-    def to_json(options = {})
-      as_json(options)
+    def as_json(options = {})
+      json = {}
+
+      self.class.property_names.each do |name|
+        json[name.to_s] = self.send(name.to_sym)
+      end
+
+      json["total"] = total
+      json
     end
 
     def fetch_async(&block)
@@ -58,13 +81,13 @@ module Popularity
       Unirest.get(request_url) do |response|
         @async_done = true
         @response = response.raw_body
-        block.call(response.code, response.raw_body) if block_given? 
+        block.call(response.code, response.raw_body) if block_given?
       end
     end
 
     def fetch
       return false unless valid?
-        
+
       begin
         response = Unirest.get(request_url)
         @response = response.raw_body
@@ -73,4 +96,4 @@ module Popularity
       end
     end
   end
-end
+end

data/lib/popularity/networks/facebook.rb

@@ -1,5 +1,7 @@
 module Popularity
 	class Facebook < Crawler
+    stats :shares, :comments
+
     def shares
       response_json['shares'].to_f.to_i
     end
@@ -8,19 +10,10 @@ module Popularity
       response_json['comments'].to_f.to_i
     end
 
-    def as_json(options = {})
-      { "shares" => shares,
-        "comments" => comments }
-    end
-
-    def total
-       shares + comments
-    end
-
     protected
 
-		def request_url
-			"http://graph.facebook.com/?id=#{@url}"
-		end
-  end
+    def request_url
+     "http://graph.facebook.com/?id=#{@url}"
+   end
+ end
 end

data/lib/popularity/networks/github.rb

@@ -1,14 +1,8 @@
 module Popularity
   class Github < Crawler
-    def stars
-      response_json.size
-    end
+    stats :stars
 
-    def as_json(options = {})      
-      { "stars" => stars }
-    end
-
-    def total
+    def stars
       response_json.size
     end
 
@@ -25,4 +19,4 @@ module Popularity
       "https://api.github.com/repos/#{owner}/#{repo}/stargazers"
     end
   end
-end
+end

data/lib/popularity/networks/google_plus.rb

@@ -1,22 +1,16 @@
 module Popularity
   class GooglePlus < Crawler
+    stats :plus_ones
+
     def plus_ones
-      matches = response.scan(/window.__SSR = {c\: (\d+.\d+E?\d+)/) 
+      matches = response.scan(/window.__SSR = {c\: (\d+.\d+E?\d+)/)
       matches.flatten.first.to_f.to_i
     end
 
-    def as_json(options = {})
-      {"plus_ones" => plus_ones}
-    end
-
-    def total
-      plus_ones
-    end
-
     protected
 
     def request_url
       "https://plusone.google.com/_/+1/fastbutton?url=#{URI::encode(@url)}"
     end
   end
-end
+end

data/lib/popularity/networks/medium.rb

@@ -1,17 +1,11 @@
 module Popularity
   class Medium < Crawler
+    stats :recommends
+
     def recommends
       response_json["payload"]["value"]["count"]
     end
 
-    def as_json(options = {})      
-      {"recommends" => recommends}
-    end
-
-    def total
-      recommends
-    end
-    
     def valid?
       host == 'medium.com'
     end
@@ -30,4 +24,4 @@ module Popularity
       JSON.parse(response.sub("])}while(1);</x>", ""))
     end
   end
-end
+end

data/lib/popularity/networks/pinterest.rb

@@ -1,21 +1,15 @@
 module Popularity
   class Pinterest < Crawler
+    stats :pins
+
     def pins
       JSON.parse(response.gsub('receiveCount(','').gsub(')',''))['count'].to_f.to_i
     end
 
-    def as_json(options = {})      
-      {"pins" => pins}
-    end
-
-    def total
-      pins
-    end
-
     protected
 
     def request_url
       "http://api.pinterest.com/v1/urls/count.json?url=#{@url}"
     end
   end
-end
+end

data/lib/popularity/networks/reddit_comment.rb

@@ -1,20 +1,14 @@
 module Popularity
   class RedditComment < Crawler
-    def score
-      response_json[1]["data"]["children"][0]["data"]["score"] 
-    end
-
-    def as_json(options = {})
-      {"score" => score}
-    end
+    stats :score
 
-    def total
-      score
+    def score
+      response_json[1]["data"]["children"][0]["data"]["score"]
     end
 
     def valid?
       return false unless host == 'reddit.com'
-      
+
       path = URI.parse(@url).path
       path.split('/').delete_if { |a| a.empty? }.size == 6
     end

data/lib/popularity/networks/reddit_post.rb

@@ -1,8 +1,10 @@
 module Popularity
   class RedditPost < Crawler
+    stats :score, :comments
+
     def score
       begin
-        response_json[0]["data"]["children"][0]["data"]["score"] 
+        response_json[0]["data"]["children"][0]["data"]["score"]
       rescue
         0
       end
@@ -10,26 +12,15 @@ module Popularity
 
     def comments
       begin
-        response_json[0]["data"]["children"][0]["data"]["num_comments"] 
+        response_json[0]["data"]["children"][0]["data"]["num_comments"]
       rescue
         0
       end
     end
 
-    def as_json(options = {})      
-      {
-        "comments" => comments,
-        "score" => score
-      }
-    end
-
-    def total
-      comments + score
-    end
-
     def valid?
       return false unless host == 'reddit.com'
-      
+
       path = URI.parse(@url).path
       path.split('/').delete_if { |a| a.empty? }.size < 6
     end
@@ -46,11 +37,14 @@ module Popularity
   end
 
   class RedditResult < RedditPost
-    # A stubbed out version of a RedditPost so RedditShare can 
+    # A stubbed out version of a RedditPost so RedditShare can
     # stub in the json response it already has
 
+    stats :score, :comments
+
     def initialize(url, r)
       super(url)
+      @url = url
       @response = r
 
       self
@@ -60,6 +54,10 @@ module Popularity
       true
     end
 
+    def reddit_url
+      @url
+    end
+
     def valid?
       URI.parse(@url).host
     end