module KindleHighlights
  class Client
    attr_reader :books
        
    def initialize(email_address, password)
      @email_address   = email_address
      @password        = password
      
      setup_mechanize_agent
      load_books_from_kindle_account
    end
  
    def highlights_for(asin)
      highlights = @mechanize_agent.get("https://kindle.amazon.com/kcw/highlights?asin=#{asin}&cursor=0&count=1000")
      json = JSON.parse(highlights.body)
      json["items"]
    end
    
    private
   
    def load_books_from_kindle_account
      @books = Hash.new
      signin_page = @mechanize_agent.get(KINDLE_LOGIN_PAGE)
      
      signin_form = signin_page.form(SIGNIN_FORM_IDENTIFIER)
      signin_form.email = @email_address
      signin_form.password = @password
      
      kindle_logged_in_page = @mechanize_agent.submit(signin_form)
      highlights_page = @mechanize_agent.click(kindle_logged_in_page.link_with(:text => /Your Books/))

      loop do
        books = highlights_page.search(".//td[@class='titleAndAuthor']")
        books.each do |book|
          asin = book.search("a").first.attributes["href"].value.split("/").last
          title = book.search("a").first.inner_html
          @books[asin] = title
        end
        break if highlights_page.link_with(text: /Next/).nil?
        highlights_page = @mechanize_agent.click(highlights_page.link_with(text: /Next/))
      end
    end
    
    def setup_mechanize_agent
      @mechanize_agent = Mechanize.new
      @mechanize_agent.user_agent_alias = 'Windows Mozilla'
      @mechanize_agent.agent.http.verify_mode = OpenSSL::SSL::VERIFY_NONE
    end
  end
end