require 'graybook/importer/page_scraper'
require 'cgi'

##
# imports contacts for MSN/Hotmail
class Graybook::Importer::Hotmail < Graybook::Importer::PageScraper

  DOMAINS = { "compaq.net"        => "https://msnia.login.live.com/ppsecure/post.srf",
              "hotmail.co.jp"     => "https://login.live.com/ppsecure/post.srf",
              "hotmail.co.uk"     => "https://login.live.com/ppsecure/post.srf",
              "hotmail.com"       => "https://login.live.com/ppsecure/post.srf",
              "hotmail.de"        => "https://login.live.com/ppsecure/post.srf",
              "hotmail.fr"        => "https://login.live.com/ppsecure/post.srf",
              "hotmail.it"        => "https://login.live.com/ppsecure/post.srf",
              "live.com"          => "https://login.live.com/ppsecure/post.srf",
              "messengeruser.com" => "https://login.live.com/ppsecure/post.srf",
              "msn.com"           => "https://msnia.login.live.com/ppsecure/post.srf",
              "passport.com"      => "https://login.live.com/ppsecure/post.srf",
              "webtv.net"         => "https://login.live.com/ppsecure/post.srf" }
              
  ##
  # Matches this importer to an user's name/address

  def =~(options)
    return false unless options && options[:username]
    domain = username_domain(options[:username].downcase)
    !domain.empty? && DOMAINS.keys.include?( domain ) ? true : false
  end
   
  ##
  # Login procedure
  # 1. Go to login form
  # 2. Set login and passwd
  # 3. Set PwdPad to IfYouAreReadingThisYouHaveTooMuchFreeTime minus however many characters are in passwd (so if passwd
  #    was 8 chars, you'd chop 8 chars of the end of IfYouAreReadingThisYouHaveTooMuchFreeTime - giving you IfYouAreReadingThisYouHaveTooMuch)
  # 4. Set the action to the appropriate URL for the username's domain
  # 5. Get the query string to append to the new action
  # 5. Submit the form and parse the url from the resulting page's javascript
  # 6. Go to that url

  def login
    page = agent.get('http://login.live.com/login.srf?id=2')
    form = page.forms.first
    form.login   = options[:username]
    form.passwd  = options[:password]
    form.PwdPad  = ( "IfYouAreReadingThisYouHaveTooMuchFreeTime"[0..(-1 - options[:password].to_s.size )])
    query_string = page.body.scan(/g_QS="([^"]+)/).first.first rescue nil
    form.action  = login_url + "?#{query_string.to_s}"
    page = agent.submit(form)
    
    # Check for login success
    if page.body =~ /The e-mail address or password is incorrect/ ||
      page.body =~ /Sign in failed\./
      return Problem.new("Username and password were not accepted. Please check them and try again.")
    end
    
    @first_page = agent.get( page.body.scan(/http\:\/\/[^"]+/).first )
  end
  
  ##
  # prepare this importer

  def prepare
    login
  end
  
  ##
  # Scrape contacts for Hotmail
  # Seems like a POST to directly fetch CSV contacts from options.aspx?subsection=26&n=
  # Seems like Hotmail addresses are now hosted on Windows Live.

  def scrape_contacts
    unless agent.cookies.find{|c| c.name == 'MSPPre' && c.value == options[:username]}
      return Problem.new("Username and password were not accepted. Please check them and try again.")
    end

    page = agent.get('http://mail.live.com/')
    
    if page.iframes.detect { |f| f.src =~ /\/mail\/TodayLight.aspx/ }
      page = agent.get( page.iframes.first.src )
      
      if button = page.forms.first.buttons.detect { |b| b.name == 'TakeMeToInbox' }
        page = agent.submit( page.forms.first, button )
      end
    end
    
    page = page.link_with(:text => 'Contact list').click
    
    contacts = parse_contacts(page.body)
    while link = page.link_with(:text => 'Next page')
      page = link.click
      contacts += parse_contacts(page.body)
    end
    
    contacts
  end
  
  def parse_contacts(source)
    source.scan(/ICc.*\:\[.*?,.*?,\['se','ct'\],'(.*?)',.*?,.*?,'(.*?)',.*\]/).collect do |name, email|
      { :name => (name =~ /\\x26\\x2364\\x3b/ ? nil : name), :email => email.gsub(/\\x40/, '@') }
    end
  end
  
  ##
  # lookup for the login service that should be used based on the user's
  # address

  def login_url
    DOMAINS[username_domain] || DOMAINS['hotmail.com']
  end
  

  ##
  # normalizes the host for the page that is currently being "viewed" by the
  # Mechanize agent

  def current_host
    return nil unless agent && agent.current_page
    uri = agent.current_page.uri
    "#{uri.scheme}://#{uri.host}"
  end
  
  ##
  # determines the domain for the user

  def username_domain(username = nil)
    username ||= options[:username] if options
    return unless username
    username.to_s.split('@').last
  end
  
  Graybook.register(:hotmail, self)
end