require 'flickraw'
require 'net/http'
require 'cgi'

module FlickrAirlift
  
  UPLOADABLE_FORMATS = [".jpg", ".jpeg", ".gif", ".png", ".mov", ".avi"]
  
  def self.download
    begin
      establish_session

      # Prompt
      puts "Exactly who's photos would you like to archive?:"
      scraped_user = STDIN.gets
      scraped_user = scraped_user.strip

      # Find
      user_id       = flickr.people.findByUsername(:username => scraped_user).id
      photos        = flickr.photos.search(:user_id => user_id)
      photo_count   = photos.total
      page_count    = photos.pages

      # Downloading
      puts "#{scraped_user} has #{photo_count} pictures"
      puts "* Creating folder named '#{scraped_user}'"
      Dir.mkdir(scraped_user) unless File.directory?(scraped_user)

      (1..page_count.to_i).each do |page_number|
        puts "* PAGE #{page_number} of #{page_count}"
        flickr.photos.search(:user_id => user_id, :page => page_number).each_with_index do |photo, i|
          photo_id     = photo.id
          info         = flickr.photos.getInfo(:photo_id => photo_id)
          download_url = flickr.photos.getSizes(:photo_id => photo_id).find{|size| size.label == "Original" || size.label == "Large" || size.label == "Medium"}.source

          puts "** Downloading #{i+1}: #{photo.title} from #{download_url}"
          File.open(File.join(scraped_user, "#{photo_id}.jpg"), 'w') do |file|
            file.puts Net::HTTP.get_response(URI.parse(download_url)).body
          end

        end
      end

      rescue FlickRaw::FailedResponse => e
        puts e.msg
      end
  end

  def self.upload(relative_url)
    establish_session("write")
    
    image_file_names = Dir.entries(".").find_all{ |file_name|  UPLOADABLE_FORMATS.any?{ |extension| file_name.downcase.include?(extension)} }
    uploaded_ids = []
    puts "Uploading #{image_file_names.length} files:"
    sleep 1
    image_file_names.each_with_index do |file_name, index|
      puts "  Uploading (#{index+1} of #{image_file_names.length}): #{file_name}"
      uploaded_ids << flickr.upload_photo(File.join(relative_url, file_name), :title => file_name.split(".").first)
    end
    puts "...DONE!"
    edit_url = "http://www.flickr.com/photos/upload/edit/?ids=#{uploaded_ids.join(',')}"

    system("which open") ? system("open '#{edit_url}'") : puts(" Go to #{edit_url} to edit your photos more")
  end

  def self.establish_session(permission = "read")
    FlickRaw.api_key        = "d4d152785af1b0ea68a5a2d173c75707"
    FlickRaw.shared_secret  = "b9da0b4f99507dd0"
    frob                    = flickr.auth.getFrob
    auth_url                = FlickRaw.auth_url :frob => frob, :perms => permission

    puts " "
    if system("which open")
      puts "opening your browser..."
      sleep 1
      puts "Come back and press Enter when you are finished"
      sleep 2
      system("open '#{auth_url}'")
    else
      puts "Open this url in your process to complete the authication process:"
      puts auth_url
      puts "Press Enter when you are finished."
    end
    STDIN.getc

    # Authentication
    auth  = flickr.auth.getToken :frob => frob
    login = flickr.test.login
    puts "You are now authenticated as #{login.username} with token #{auth.token}"
  end

end