samstarling/gradbot

require 'rest-client'
require 'json'
require 'set'
require 'uri'

class GoogleClient
  BASE_URL = "http://ajax.googleapis.com/ajax/services/search/web"
  
  def self.get_result_count term
    response = self.get_result "\"#{term}\""
    response['responseData']['cursor']['estimatedResultCount'].to_i
  end
  
  private
  
  def self.get_result term
    safe_term = URI.escape(term)
    JSON.parse(RestClient.get "#{BASE_URL}?v=1.0&q=#{safe_term}")
  end
end

class CelebTracker
  include Cinch::Plugin
  
  attr_reader :celeb
  
  @@filepath = ENV['celeb_file'] || '/home/samstarling/temp/celeb.marshal'
  
  match /^!celeb$/, use_prefix: false
  match /celeb ([\d]+)/, method: :list_leaderboard, use_prefix: true
  match /celeb ([^\d][\w\s\d]+)/, method: :register_celeb, use_prefix: true
  listen_to :nick
  
  def execute(m)
    load_hash
    m.reply "Type \"!celeb [celebrity name]\" to register a sighting or \"!celeb [number]\" to view [number] of leaderboard entries sorted by score"
  end

  def listen(m)
    # Updates entries for a user when they change their nick
    old_nick = m.user.last_nick
    new_nick = m.user.nick
    load_hash
    @celeb.each do |celeb, val|
      val[0].to_a.each do |viewed_by|
        if viewed_by == old_nick
          val[0].delete(old_nick)
          val[0].add(new_nick)
        end
      end
    end
    save_hash
  end

  def list_leaderboard(m, count)
    load_hash
    count=count.to_i
    arr = @celeb.sort_by {|key, value| value[1]}.reverse()
    loop_count = 0;
    arr.each do |celeb, val|
      break if (count <= loop_count)
      m.reply "#{titleise_name(celeb)} has been seen by \"#{human_readable_viewers(val[0])}\". #{titleise_name(celeb)} has a fame index of : #{val[1]}"
      loop_count+=1
    end
  end
  
  def register_celeb(m, celeb_name)
    celeb_name.downcase!
    load_hash
    @celeb[celeb_name.to_sym] ||= Array[Set[], -1]
    seen_by = human_readable_viewers(@celeb[celeb_name.to_sym][0].add(m.user.nick))
    fame = @celeb[celeb_name.to_sym][1]
    if fame == -1
      fame = @celeb[celeb_name.to_sym][1] = calc_fame(celeb_name)
    end
    m.reply "#{titleise_name(celeb_name)} has been seen by \"#{seen_by}\". #{titleise_name(celeb_name)} has a fame index of : #{fame}"
    save_hash
  end

  def calc_fame(celeb_name)
    number = GoogleClient.get_result_count celeb_name
    number / 1000
  end

  def human_readable_viewers(viewer_set)
    list = nil
    viewer_set.to_a.each do |viewer|
      if list === nil
        list = viewer
      else
        list = list + ", " + viewer
      end
    end
    return list
  end

  def titleise_name(name)
    split_name = name.to_s.split(" ")
    name = ""
    split_name.each do |word|
      name += word.capitalize + " "
    end
    return name.strip()
  end

  def load_hash
    if File.exists?(@@filepath)
      File.open(@@filepath) do |f|
        @celeb = Marshal.load(f)
      end
    else
      @celeb = Hash.new
      File.new(@@filepath, 'w')
      File.open(@@filepath, 'w+') do |f|
        Marshal.dump(@celeb, f)
      end
    end
  end

  def save_hash
    File.open(@@filepath, 'w+') do |f|
      Marshal.dump(@celeb, f)
    end
  end
end