tweetdeck

#!/usr/bin/env ruby

require 'rubygems'
require 'sequel'
require 'optparse'

class Account
  def initialize(name='')
    pref_dir = File.expand_path('~/Library/Preferences')
    db_file = Dir.glob("#{pref_dir}/TweetDeck*/*/*#{name}.db").first
    @db = Sequel.sqlite(db_file)
  end
  
  def groups
    @db[:columns].filter(:cType => 0).select(:cName).map { |r| r[:cName].split(': ')[1] }
  end
  
  def group(name)
    group = @db[:columns].filter(:cType => 0, :cName => "Group: #{name}").first
    friend_ids = @db[:groups].filter(:gCID => group[:cID]).select(:gUserID)
    @db[:friends].filter(:fUserID => friend_ids).select(:fScreenName).map { |r| r[:fScreenName] }
  end
  
  def ungrouped
    ungrouped = friends
    groups.each do |name|
      ungrouped -= group(name)
    end
    ungrouped
  end
  
  def friends
    @db[:friends].select(:fScreenName).map { |r| r[:fScreenName] }
  end
end

def usage
  puts "Usage:
#{$0} groups
#{$0} group group-name
#{$0} ungrouped
#{$0} friends"
  exit
end

command = ARGV.shift
usage unless %w{groups group ungrouped friends}.include?(command)

account = Account.new
usage unless account.method(command).arity == ARGV.length

puts account.method(command).call(*ARGV).sort.join("\n")