#!/usr/bin/env ruby

require 'rubygems'
require 'mechanize'

module WellsFargo
  class Session
    attr_accessor :agent, :page
    
    def initialize
      @agent = WWW::Mechanize.new
      @agent.user_agent_alias = 'Windows IE 6'
    end

    def extract_total(html)
      return $1.gsub(/,/, '').to_f if html =~ /<td.*?headers="cashTotal cashAvailBalance".*?([\d,]+\.\d+)/im
    end

    def login(username, password)
      @page = @agent.get('https://online.wellsfargo.com/')

      # find login form, fill it out and submit it
      form = page.forms.find {|f| f.name == 'Signon'}
      form.fields.find {|f| f.name == 'userid'}.value = username
      form.fields.find {|f| f.name == 'password'}.value = password
      @page = @agent.submit(form, form.buttons.first)
  
      # find the refresh url
      @page.body =~ /META.*?Refresh.*?URL=(.*?)"/
      nexturl = $1
  
      # wait a little while and then get the next page
      sleep 5
      @page = @agent.get nexturl  
    end
    
    def logout
      @page = @agent.get $1 if @page.body =~ /<a.*href="(.*?)".*>(Log|Sign) ?(Off|Out)/i
    end
  end
end