#!/usr/bin/ruby
require 'flickraw'
require 'open-uri'
require 'RMagick'
 
#
# Obtain the the photos from Flickr.
#
def get_photos( username, count = 9 )
  begin
    user = flickr.people.findByUsername( :username => username )
    photo_list = flickr.photos.getContactsPublicPhotos( :user_id => user.nsid, :count => 9 )
  rescue Exception => ex
    puts ex
    return []
  end
 
  return photo_list.slice(0, 9)
end
 
#
# Construct the URL of the photo, given a photo object as returned by flickraw.
#
def photo_url ( photo )
  return "http://farm#{photo.farm}.static.flickr.com/#{photo.server}/#{photo.id}_#{photo.secret}.jpg"
end
 
#
# Download the specified photos. It will use a caching mechanism in order to save 
# bandwidth.
#
def download_photos( photo_list )
  files = []
 
  Dir.mkdir 'images' unless File.exists?( 'images' )
 
  photo_list.each do |p|
    url = photo_url p 
 
    fileName = 'images/' + File.basename( url )
 
    if !File.exists?( fileName )
      puts url.to_s + " => " + fileName
      open url do |remote|
        open(fileName, 'wb') { |local| local << remote.read }
      end
      else
      puts "Skipping #{url} because it already exists."
    end
    files << fileName
  end
  return files
end
 
#
# Create the collage, using Magick::ImageList#montage
# method.
#
def create_image( args = { } )
  # Set some default values
  props = { 
    :width => 1024,
    :height => 768,
    :files => nil,
    :username => 'unknown-user',
    :output => '',
    :tile_h => 3,
    :tile_v => 3
  }
 
  args.each { |k,v|
    k = k.to_sym
    raise ArgumentError, "unknown property #{k}" unless props.key?(k)
    props[k] = v
  }
 
  props[:output] = props[:username] + '.jpg' unless props[:output].length > 0
 
  # Construct the image list
  image_list = Magick::ImageList.new
  props[:files].each do |img|
    image = Magick::Image.read( img ).first
 
    # Add a nice, white border around each image
    image.border!(30, 30, "#ffffff")
    image.background_color = "none"
    image_list << image
  end
 
  # Compute the placement of the images. Not a very interesting part...
  padding, image_dim = Hash.new, Hash.new
  padding[:x], padding[:y] = props[:width] * 0.05, props[:height] * 0.05
  image_dim[:width], image_dim[:height] = (props[:width] - 6*padding[:x]) / props[:tile_h], (props[:height] - 6*padding[:y]) / props[:tile_v]
 
  # Construct the montage.
  image_list = image_list.montage do
    self.tile = "#{props[:tile_h]}x#{props[:tile_v]}"
    self.background_color = '#000000'
    self.geometry = "#{image_dim[:width]}x#{image_dim[:height]}+#{padding[:x]}+#{padding[:y]}"
    self.shadow = true
  end
 
  # Flatten everything and write it to a file.
  image = image_list.flatten_images
  image.resize!( props[:width], props[:height] )
  image.write( props[:output] )
end
 
# Check the number of parameters
if ARGV.length <= 0
  puts "Not enough arguments."
  exit
end
 
# Main body
username = ARGV.first
photos = get_photos( username )
image_files = download_photos( photos )
create_image( :files => image_files, :username => username, :width => 400, :height => 300 )