#!/usr/bin/ruby

require 'rubygems'
require 'rubytter'
require 'yaml'
require 'pit'
require 'pp'

def toCsv(friends)
  open('friends.csv', 'w') do |f|
    friends.each do |user|
      f.puts "#{user.screen_name},#{user.id}, #{user.name}, #{user.description.to_s.gsub(/\r\n/,'')}"
    end
  end

end

def toYaml(friends)
  friends.to_yaml
  open('friends.yaml', 'w') do |f|
    YAML.dump(friends, f)
  end
end

config = Pit.get("twitter")
username = config['username']
password = config['password']

client = Rubytter.new(username, password)

friends = Array.new
last = nil
begin
  last = client.friends(username, :cursor => last ? last.next_cursor : -1)
  friends.concat(last.users)
rescue Timeout::Error, StandardError
  puts "print Timeout::Error, StandardError"
  break
end until last.next_cursor == 0
toYaml(friends)
toCsv(friends)