se.rb

require 'serel'
require './bounty'


class SiteDoesNotExistError < StandardError
end

class Site
  def initialize(domain, se)
    @domain = domain
    @se = se
  end
  
  # http://kylecronin.me/blog/2012/4/22/a-clever-ruby-equality-trick.html
  def ==(other)
    other == @domain
  end
  
  def hash()
    @domain.hash
  end
  
  def to_s()
    @domain
  end
  
  def bounties()
    Serel::Base.config(@domain, @se.apikey)
    bounties = Serel::Question.featured.pagesize(99).request
        
    number = bounties.length
    rep    = bounties.map { |bounty| bounty.bounty_amount }.reduce(:+)
      
    rep = 0 if number == 0
      
    [@domain, number, rep]
  end
end


class SE  
  attr_accessor :apikey
  
  def initialize(apikey)
    @apikey = apikey
    @se_sites = []
    update_sites()
  end
  
  def site(url)
    @se_sites[validate(url)]
  end
  
  def validate(url)
    domain = extract_domain(url)
    if @se_sites.include?(domain)
      domain
    elsif @se_sites.include?(domain[5..-1])
      domain[5..-1]
    elsif @se_sites.include?(domain[6..-1])
      domain[6..-1]
    else
      raise SiteDoesNotExistError
    end
  end
  
  def extract_domain(url)
    m = /^(?:https?:\/\/)?([\w\.]*)/.match(url)
    raise SiteDoesNotExistError if not m
    m[1]
  end
  
  def update_sites()
    Serel::Base.config('', @apikey)
    @se_sites = Hash[Serel::Site.all.
      select { |site|   site.site_type == 'main_site' }.
      map    { |site|   site.site_url }.
      map    { |url|    extract_domain(url) }.
      map    { |domain| [domain, Site.new(domain, self)] }
    ]
  end

  def info()
    Serel::Base.config(:apple, @apikey)
    Serel::User.get.quota_remaining
  end
end