Rakefile

require 'bundler'
require 'kindle_highlights'
require 'fileutils'
require 'json'
require 'cgi'
require 'mail'
require 'htmlentities'

Mail.defaults do
  delivery_method :smtp,
    address:              ENV['MAILGUN_SMTP_SERVER'] || "smtp.mailgun.org",
    port:                 ENV['MAILGUN_SMTP_PORT'] || 587,
    user_name:            ENV['MAILGUN_SMTP_LOGIN'],
    password:             ENV['MAILGUN_SMTP_PASSWORD'],
    to:                   ENV['TO'] || "youremail@domain.com",
    authentication:       'plain',
    enable_starttls_auto: true
end

class Kindle
  def initialize(path = 'data.json')
    @path = path
  end

  def simpleTest

    kindle = KindleHighlights::Client.new(email_address: ENV["AMAZON_USER"] || "youremail@domain.com", password: ENV["AMAZON_PASS"] || "youramazonpassword")

    # begin
    #   puts kindle.books.sample
    # rescue Exception => e
    #   puts e
    # end

    begin
      puts kindle.books.sample
    rescue Exception => e
      puts e
    end

    return kindle
  end

  def update(kindle)
    html = HTMLEntities.new
    # kindle = KindleHighlights::Client.new(email_address: ENV["AMAZON_USER"] || "youremail@domain.com", password: ENV["AMAZON_PASS"] || "youramazonpassword")
    @highlights = []

    kindle.books.each do |key, title|
      kindle.highlights_for(key).each do |highlight|
        highlight["book"] = html.decode(title)
        highlight["highlight"] = html.decode(highlight["highlight"])
        @highlights << highlight
      end
    end
  end

  def save
    File.open(@path, "w+") do |fp|
      fp << @highlights.to_json
    end
  end

  def highlights
    @highlights ||= JSON.load(open(@path))
  end

  def random_highlight
    highlights[rand(highlights.length)]
  end
end

task :download do
  data = Kindle.new
  data.update(data.simpleTest)
  data.save
end

task :print do
  data = Kindle.new
  highlight = data.random_highlight
  puts "\"#{highlight["highlight"]}\""
  puts
  puts " -- #{highlight["book"]}, #{highlight["howLongAgo"]}"
  puts
end


task :email do
  data = Kindle.new
  highlight = data.random_highlight

  puts "OK, sent email"

  mail = Mail.new do
    from    'Kindle Highlights <kindle@highlights.mailgun.com>'
    to      ENV['TO']
    subject "#{Time.now.strftime("%b %d")}: #{highlight["book"]}"
    html_part do
      content_type 'text/html; charset=UTF-8'

      body "<p><i>#{highlight["highlight"]}</i><p><br><p>&mdash; #{highlight["book"]}, #{highlight["howLongAgo"]}</p>"
    end
  end

  mail.deliver
end

task default: [:download, :email]