##########################################################################
# This is now a proper Github project: http://github.com/klondike/wesabe
##########################################################################
 
wesabe = Wesabe.new 'username', 'password'
 
# login to trigger automatic updates, so your account info will be fresh
wesabe.trigger_updates
 
# fetch your account data (you may want to wait a minute after triggering updates)
wesabe.fetch_accounts
 
# the floating point balance of account #1
wesabe[1][:balance] 
 
# the integer balance in cents of account #1
wesabe[1][:balance_in_cents]
 
# name of account #1
wesabe[1][:name]

##########################################################################
# This is now a proper Github project: http://github.com/klondike/wesabe
##########################################################################
 
require 'net/https'
require 'rexml/document'
require 'yaml'
 
CERT_FILENAME = File.dirname(__FILE__) + "/cacert.pem"
 
class Wesabe
  attr_accessor :username, :password, :accounts, :account_xml, :http
  
  def initialize(username, password)
    self.username = username
    self.password = password
    initialize_http    
  end
  
  def initialize_http
    unless File.exist?(CERT_FILENAME)
      puts "Downloading certificate..."
      cert = Net::HTTP.get('curl.haxx.se', '/ca/cacert.pem')
      File.open(CERT_FILENAME, "w") { |f| f << cert }
      puts "Certificate downloaded."
    end
    self.http = Net::HTTP.new("www.wesabe.com", 443)
    http.use_ssl = true
    http.verify_mode = OpenSSL::SSL::VERIFY_PEER
    http.ca_file = CERT_FILENAME
  end
  
  def fetch_accounts
    self.account_xml = get_accounts_xml
    self.accounts = parse_accounts_list account_xml
  end
  
  def trigger_updates
    http.start do |wesabe|
      request = Net::HTTP::Post.new('/user/login')
      request.set_form_data({'username' => username, 'password' => password})
      response = http.request(request)
  
      case response
      when Net::HTTPOK
        STDERR.puts "Bad username and password"
        exit(-1)
      when Net::HTTPRedirection
        puts "Logged in successfully"
        response.body
      else
        STDERR.puts "Unexpected response: #{response.inspect}"
        exit(-1)
      end
    end
  end
  
  def get_accounts_xml
    http.start do |wesabe|
      request = Net::HTTP::Get.new('/accounts.xml')
      request.basic_auth(username, password)
      response = http.request(request)
  
      case response
      when Net::HTTPFound
        STDERR.puts "Incorrect username or password."
        exit(-1)
      when Net::HTTPOK
        response.body
      else
        STDERR.puts "Unexpected response: #{response.inspect}"
        exit(-1)
      end
    end
  end
  
  def parse_accounts_list(xml)
    accounts = {}
    doc = REXML::Document.new(xml)
    doc.root.each_element('//account') do |account|
      id = account.elements["id"].text
      name = account.elements["name"].text
      balance = account.elements["current-balance"].text.to_f
      balance_in_cents = (balance * 100).to_i
      accounts[id] = {
        :name => name, 
        :balance => balance, 
        :balance_in_cents => balance_in_cents
      }
    end
    accounts
  end
  
  def [](id)
    accounts[id.to_s]
  end
end