Rakefile

require 'rubygems'
require 'aws'

s3 = AWS::S3.new
@bucket = s3.buckets['packetpig']

def upload(src, dst)
  print "#{src} -> s3://#{@bucket.name}/#{dst} (#{File.size(src)} bytes) ... "
  $stdout.flush
  remote = @bucket.objects[dst]
  remote.write(File.read(src))
  puts "done"
  $stdout.flush
end

task :all => [:snort, :pig, :lib, :packetpig, :scripts, :bootstrap, :srcs]

task :pig do
  Dir['pig/examples/*.pig'].each do |pig|
    upload(pig, pig)
  end

  ['pig/include-emr.pig'].each do |f|
    upload(f, f)
  end
end

task :bootstrap do
  upload('lib/bootstrap.sh', 'bootstrap.sh')
end

task :lib do
  libs = Dir['lib/*.jar'] - ['lib/packetpig.jar', 'lib/packetpig-with-dependencies.jar']
  libs.each do |lib|
    dst = File.basename(lib)
    upload(lib, dst)
  end
end

task :snort do
  Dir['lib/snort*'].each do |path|
    if File.directory?(path)
      fn = "#{path}.tar.gz"
      dir = File.basename(path)
      `tar czf #{fn} -C lib #{dir}`
      upload(fn, File.basename(fn))
    end
  end
end

task :packetpig do
  upload('lib/packetpig.jar', 'packetpig.jar')
end

task :scripts do
  `tar czf lib/scripts.tar.gz lib/scripts lib/tailer`
  upload('lib/scripts.tar.gz', 'scripts.tar.gz')
end

task :srcs do
  Dir['lib/src/*'].each do |fn|
    upload(fn, File.basename(fn))
  end
end