cf-s3-invalidator 0.1.0 → 0.2.0

data/.gitignore

@@ -0,0 +1,2 @@
+*.gem
+Gemfile.lock

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify the gem's dependencies in the .gemspec file
+gemspec

data/README.md

@@ -0,0 +1,13 @@
+# Invalidator for AWS S3-based Cloudfront distributions
+
+If your Amazon Web Services Cloudfront distribution is based on AWS S3, then
+this library may be useful to you.
+
+This library fetches all the object keys of a particular S3 bucket and then
+calls on them the invalidation REST API of Cloudfront.
+
+## Usage
+
+`gem install cf-s3-invalidator`
+
+`cf-s3-inv <AWS-KEY> <AWS-SECRET> <S3-BUCKET-NAME> <CLOUDFRONT-DISTRIBUTION-ID>`

data/bin/cf-s3-inv

@@ -1,21 +1,43 @@
 #!/usr/bin/env ruby
-require 'rubygems'
-require 'aws-sdk'
-require 'cloudfront_invalidator'
+require 'optparse'
+require File.dirname(__FILE__)+'/../lib/cloudfront_invalidator'
+require File.dirname(__FILE__)+'/../lib/s3_loader'
 
-key = ARGV[0]
-secret = ARGV[1]
-s3_bucket = ARGV[2]
-cloudfront_distribution_id = ARGV[3]
-
-s3 = AWS::S3.new(
-  :access_key_id => key,
-  :secret_access_key => secret)
+options = {}
+optparse = OptionParser.new do |opts|
+  opts.banner = "Usage: #{File.basename(__FILE__)} [options]"
+  opts.separator ""
+  opts.separator "Invalidate an S3-based Cloudfront distribution"
+  opts.separator ""
+  opts.separator "Options:"
+  opts.on("-k", "--key AWS KEY",
+          "Amazon Web Services API key that has access to your S3 and Cloudfront") do |val|
+    options[:key] = val
+  end
+  opts.on("-s", "--secret AWS SECRET",
+          "Amazon Web Services API secret key") do |val|
+    options[:secret] = val
+  end
+  opts.on("-b", "--bucket BUCKET NAME",
+          "S3 bucket name") do |val|
+    options[:bucket] = val
+  end
+  opts.on("-d", "--distribution CLOUDFRONT ID",
+          "Cloudfront distribution id") do |val|
+    options[:distribution] = val
+  end
+end
 
-s3_objects = []
-s3.buckets[s3_bucket].objects.each do |object|
-  s3_objects << "/#{object.key}"
+ARGV_ORIGINAL = ARGV.dup
+optparse.parse!
+if ARGV_ORIGINAL.length == 0
+  puts optparse.help
+  exit
 end
 
-invalidator = CloudfrontInvalidator.new(key, secret, cloudfront_distribution_id)
-invalidator.invalidate(s3_objects)
+s3_object_keys = S3Loader.new(
+  options[:key], options[:secret]).list_keys(options[:bucket])
+
+invalidator = CloudfrontInvalidator.new(
+  options[:key], options[:secret], options[:distribution])
+invalidator.invalidate(s3_object_keys)

data/cf-s3-invalidator.gemspec

@@ -0,0 +1,20 @@
+Gem::Specification.new do |s|
+  s.name          = 'cf-s3-invalidator'
+  s.version       = '0.2.0'
+  s.date          = '2012-07-03'
+  s.summary       = 'A tool for invalidating AWS S3-based Cloudfront distributions'
+  s.description   =
+    'This lib fetches the names of all objects on an Amazon Web
+    Services S3 bucket. Then it calls the Cloudfront invalidation REST API on the
+    objects.
+
+    The Cloudfront distribution must be based on the S3 bucket.'
+  s.authors       = ['Lauri Lehmijoki']
+  s.email         = 'lauri.lehmijoki@iki.fi'
+  s.files         = `git ls-files`.split("\n")
+  s.require_paths = ['lib']
+  s.executables   << 'cf-s3-inv'
+  s.homepage      = 'https://github.com/laurilehmijoki/cf-s3-invalidator'
+
+  s.add_dependency 'aws-sdk'
+end

data/lib/cloudfront_invalidator.rb

@@ -3,7 +3,7 @@ require 'digest/sha1'
 require 'net/https'
 require 'base64'
 
-# Adapted from: 
+# Adapted from:
 # Confabulus @ http://blog.confabulus.com/2011/05/13/cloudfront-invalidation-from-ruby
 class CloudfrontInvalidator
   def initialize(aws_account, aws_secret, distribution)
@@ -13,6 +13,7 @@ class CloudfrontInvalidator
   end
 
   def invalidate(items)
+    items = to_cf_keys(items)
     date = Time.now.strftime("%a, %d %b %Y %H:%M:%S %Z")
     digest = Base64.encode64(
       OpenSSL::HMAC.digest(OpenSSL::Digest::Digest.new('sha1'), @aws_secret, date)).strip
@@ -44,7 +45,7 @@ class CloudfrontInvalidator
 
   def print_operation_result(http_response, items)
     success = http_response.code == '201'
-    puts "Invalidating items"
+    puts "Invalidating Cloudfront items"
     items.each do |item|
       puts "  #{item}"
     end
@@ -52,10 +53,14 @@ class CloudfrontInvalidator
       puts "succeeded"
     else
       puts "FAILED, reason:"
-      puts res.body
+      puts http_response.body
     end
   end
 
+  def to_cf_keys(s3_keys)
+    s3_keys.map { |s3_key| "/#{s3_key}" }
+  end
+
   def to_xml(items)
     items.map{ |item| "<Path>#{item}</Path>" }
   end

data/lib/s3_loader.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'aws-sdk'
+
+class S3Loader
+  def initialize(key, secret)
+    @s3 = AWS::S3.new(
+      :access_key_id => key,
+      :secret_access_key => secret)
+  end
+
+  def list_keys(bucket_name)
+    s3_objects = []
+    @s3.buckets[bucket_name].objects.each do |object|
+      s3_objects << object.key
+    end
+    s3_objects
+  end
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: cf-s3-invalidator
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 23
   prerelease: 
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Lauri Lehmijoki
@@ -45,8 +45,13 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
-- lib/cloudfront_invalidator.rb
+- .gitignore
+- Gemfile
+- README.md
 - bin/cf-s3-inv
+- cf-s3-invalidator.gemspec
+- lib/cloudfront_invalidator.rb
+- lib/s3_loader.rb
 homepage: https://github.com/laurilehmijoki/cf-s3-invalidator
 licenses: []