Contents

require 'popularity/crawler'
require 'popularity/search'
require 'popularity/results_container'

module Popularity
  TYPES = []
end

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('_', '')

  # Auto register the types in /networks
  Popularity::TYPES << Object.const_get("Popularity::#{class_name}")
}

module Popularity
  def self.search(*urls)
  	response = {}

    MultiSearch.new(:urls => urls)
  end

  def self.select_types(url)
    # github.com stats only valid for github urls, etc
    selected_types = Popularity::TYPES.collect { |n|
      network = n.new(@url)
      network if network.valid?
    }.compact
  end

  class MultiSearch
    attr_accessor :searches
    attr_accessor :sources

    def initialize(options = {})
      @searches = options[:urls].collect { |url| Search.new(url) }

      @searches.each do |search|
        search.results.each do |result|
          add_search_result(result)
        end
      end
    end 

    def results
      searches.collect(&:results).reduce(:+)
    end

    def to_json(options = {})      
      json = {}
      self.searches.collect do |search|
        json[search.url] = search.to_json
      end

      self.sources.collect do |source|
        json[source.to_s] = self.send(source.to_sym).to_json
      end

      json["total"] = total

      json
    end

    def total
      total = 0
      self.searches.each do |a|
        total += a.total
      end

      total
    end

    protected

    def add_search_result(result)
      container = self.instance_variable_get("@#{result.name}") 

      unless container
        @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 }
      end

      container.add_result(result)
    end

  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
popularity-0.1.1	lib/popularity.rb
popularity-0.1.0	lib/popularity.rb