simple-s3 0.0.8 → 0.0.10

data/bin/simple-s3

@@ -1,3 +1,3 @@
 #!/usr/bin/env ruby
 require File.dirname(__FILE__) + '/../lib/simple-s3'
-SimpleS3.upload!
+SimpleS3.run!

data/lib/simple-s3/simple-s3.rb

@@ -9,12 +9,30 @@ class SimpleS3
     @config
   end
 
+  def self.default_exclude_files
+    ['simple-s3.yml', '.git', '.rvmrc', 'Gemfile', 'Gemfile.lock', '.gitignore', '.gitmodules']
+  end
+
+  def self.s3_access_key
+    self.config['s3_access_key']
+  end
+
+  def self.s3_secret_key
+    self.config['s3_secret_key']
+  end
+
+  def self.run!
+    self.upload!
+    self.invalidate!
+  end
+
   def self.upload!
     @config = config
     bucket      = @config['bucket'].to_s
     raise 'Simple-S3: Bucket not defined' if bucket.length == 0
 
     exclude     = @config['exclude_files'] || []
+    exclude |= self.default_exclude_files
     inc         = @config['include_files'] || './**/*'
     metadata    = @config['metadata'] || {}
     metadata[:access] ||= 'public-read'
@@ -32,8 +50,8 @@ class SimpleS3
     raise 'Simple-S3: No files found' if files.length == 0
 
     AWS::S3::Base.establish_connection!(
-      :access_key_id     => @config['access_key'],
-      :secret_access_key => @config['secret_key']
+      :access_key_id     => self.s3_access_key
+      :secret_access_key => self.s3_secret_key
     )
 
     files.each do |file|
@@ -50,4 +68,54 @@ class SimpleS3
     puts "Simple-S3: Upload Completed!"
   end
 
+  def invalidate!
+    @config = config
+    distribution   = @config['cloudfront_distribution_id']
+    return if distribution.nil?
+
+    aws_account    = self.s3_access_key
+    aws_secret     = self.s3_secret_key
+
+    path = ['/','/*','/**/*']
+
+    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
+    uri = URI.parse("https://cloudfront.amazonaws.com/2012-07-01/distribution/#{distribution}/invalidation")
+
+    path_xml = ''
+    path.each do |p|
+      path_xml += '<Path>'+p+'</Path>'
+    end
+
+    req = Net::HTTP::Post.new(uri.path)
+    req.initialize_http_header({
+      'x-amz-date' => date,
+      'Content-Type' => 'text/xml',
+      'Authorization' => "AWS %s:%s" % [aws_account, digest]
+    })
+
+    xml = '<InvalidationBatch xmlns="http://cloudfront.amazonaws.com/doc/2012-07-01/">' +
+          ' <Paths>' +
+          '  <Quantity>' + path.length.to_s + '</Quantity>' + 
+          '  <Items>' +
+          path_xml + 
+          '  </Items>' +
+          ' </Paths>' +
+          ' <CallerReference>SOMETHING_SPECIAL_' + Time.now.utc.to_i.to_s + '</CallerReference>' +
+          '</InvalidationBatch>'
+    req.body = xml
+
+    http = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = true
+    http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+    res = http.request(req)
+    
+    # it was successful if response code was a 201
+    if res.code == '201'
+      puts 'Simple-S3: Invalidation Successful'
+    else
+      raise res.body.to_s
+    end
+  end
+
 end

data/lib/simple-s3/version.rb

@@ -1,3 +1,3 @@
 class SimpleS3
-  VERSION = "0.0.8"
+  VERSION = "0.0.10"
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 8
-  version: 0.0.8
+  - 10
+  version: 0.0.10
 platform: ruby
 authors: 
 - Matias Niemela