Skip to content

Instantly share code, notes, and snippets.

@CharlieSu
Created October 5, 2010 19:06
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save CharlieSu/1fa70497a56a7e6cdf91 to your computer and use it in GitHub Desktop.
class Chef
class Provider
class S3File < Chef::Provider::RemoteFile
def s3_uri?(source)
uri = URI.parse(source)
uri.absolute? and uri.scheme == "s3"
rescue URI::InvalidURIError
false
end
def action_create
Chef::Log.debug("Checking #{@new_resource} for changes")
if current_resource_matches_target_checksum?
Chef::Log.debug("File #{@new_resource} checksum matches target checksum (#{@new_resource.checksum}), not updating")
else
Chef::Log.debug("File #{@current_resource} checksum didn't match target checksum (#{@new_resource.checksum}), updating")
fetch_from_s3(@new_resource.source) do |raw_file|
if matches_current_checksum?(raw_file)
Chef::Log.debug "#{@new_resource}: Target and Source checksums are the same, taking no action"
else
backup_new_resource
Chef::Log.debug "copying remote file from origin #{raw_file.path} to destination #{@new_resource.path}"
FileUtils.cp raw_file.path, @new_resource.path
@new_resource.updated = true
end
end
end
enforce_ownership_and_permissions
@new_resource.updated
end
def fetch_from_s3(source)
require 'rubygems'
require 'aws/s3'
begin
uri = URI.parse(source)
AWS::S3::Base.establish_connection!(
:access_key_id => @new_resource.access_key_id,
:secret_access_key => @new_resource.secret_access_key
)
name = uri.path[1..-1]
bucket = uri.host
obj = AWS::S3::S3Object.find name, bucket
Chef::Log.debug("Downloading #{name} from S3 bucket #{bucket}")
file = Tempfile.new("chef-s3-file")
file.write obj.value
Chef::Log.debug("File #{name} is #{file.size} bytes on disk")
begin
yield file
ensure
file.close
end
rescue URI::InvalidURIError
Chef::Log.warn("Expected an S3 URL but found #{source}")
nil
end
end
end
end
end
class Chef
class Resource
class S3File < Chef::Resource::RemoteFile
def initialize(name, run_context=nil)
super
@resource_name = :s3_file
end
def provider
Chef::Provider::S3File
end
def access_key_id(args=nil)
set_or_return(
:access_key_id,
args,
:kind_of => String
)
end
def secret_access_key(args=nil)
set_or_return(
:secret_access_key,
args,
:kind_of => String
)
end
end
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment