publish

#!/usr/bin/env ruby

require 'aws-sdk-v1'
require 'digest/md5'

bucket = ENV['AWS_BUCKET'] || File.basename(Dir.pwd)

SITE = AWS::S3.new(region: ENV['AWS_REGION'] || 'eu-west-1')
              .buckets[bucket]

USAGE = <<EOF
usage: #{$0} file...

the following environment vars will be used:

  AWS_REGION
  AWS_ACCESS_KEY_ID
  AWS_SECRET_ACCESS_KEY

EOF

def upload(files=["."])
  files = files.map do |f|
    File.directory?(f) ? traverse(Dir.new(f)) : f
  end.flatten

  files.each do |f|
    obj = SITE.objects[f]
    local_md5 = md5sum(f).chomp
    remote_md5 = obj.etag.tr('"', '') if obj.exists?

    print "#{f}.. "

    if local_md5.to_s != remote_md5.to_s
      obj.write(Pathname.new(f))
      obj.acl = :public_read
      puts "ok"
    else
      puts "skipping"
    end
  end
end

def md5sum(file_name)
  Digest::MD5.hexdigest(File.read(file_name))
end

def traverse(dir)
  list = []
  dir.each do |entry|
    unless [".", "..", ".git"].include?(entry)
      path = File.join(dir.path, entry)
      if File.directory?(path)
        list += traverse(Dir.new(path))
      else
        list << path
      end
    end
  end
  list
end

if ARGV[0] == "-h"
  puts USAGE
  exit 0
end

begin
  upload(ARGV)
rescue Exception => e
  $stderr.puts "error: #{e.message}"
  exit 1
end