1
0
Fork 0
mirror of https://github.com/fog/fog.git synced 2022-11-09 13:51:43 -05:00
fog--fog/lib/fog/aws/s3.rb

175 lines
5.9 KiB
Ruby
Raw Normal View History

2009-05-19 02:06:49 -04:00
module Fog
module AWS
class S3
if Fog.mocking?
def self.reset_data
@data = { :buckets => {} }
end
def self.data
@data
end
end
2009-08-10 11:30:28 -04:00
def self.reload
load "fog/aws/models/s3/bucket.rb"
load "fog/aws/models/s3/buckets.rb"
load "fog/aws/models/s3/object.rb"
load "fog/aws/models/s3/objects.rb"
load "fog/aws/models/s3/owner.rb"
2009-08-17 18:58:36 -04:00
load "fog/aws/parsers/s3/copy_object.rb"
load "fog/aws/parsers/s3/get_bucket.rb"
load "fog/aws/parsers/s3/get_bucket_location.rb"
load "fog/aws/parsers/s3/get_request_payment.rb"
load "fog/aws/parsers/s3/get_service.rb"
2009-08-10 11:30:28 -04:00
load "fog/aws/requests/s3/copy_object.rb"
load "fog/aws/requests/s3/delete_bucket.rb"
load "fog/aws/requests/s3/delete_object.rb"
load "fog/aws/requests/s3/get_bucket.rb"
load "fog/aws/requests/s3/get_bucket_location.rb"
load "fog/aws/requests/s3/get_object.rb"
load "fog/aws/requests/s3/get_request_payment.rb"
load "fog/aws/requests/s3/get_service.rb"
load "fog/aws/requests/s3/head_object.rb"
load "fog/aws/requests/s3/put_bucket.rb"
load "fog/aws/requests/s3/put_object.rb"
load "fog/aws/requests/s3/put_request_payment.rb"
2009-08-10 11:30:28 -04:00
if Fog.mocking?
reset_data
end
2009-08-09 01:40:42 -04:00
end
2009-05-19 02:06:49 -04:00
# Initialize connection to S3
#
# ==== Notes
# options parameter must include values for :aws_access_key_id and
# :aws_secret_access_key in order to create a connection
#
# ==== Examples
2009-10-04 18:30:44 -04:00
# s3 = S3.new(
# :aws_access_key_id => your_aws_access_key_id,
# :aws_secret_access_key => your_aws_secret_access_key
# )
2009-05-19 02:06:49 -04:00
#
# ==== Parameters
# * options<~Hash> - config arguments for connection. Defaults to {}.
2009-05-19 02:06:49 -04:00
#
# ==== Returns
# * S3 object with connection to aws.
2009-05-19 02:06:49 -04:00
def initialize(options={})
@aws_access_key_id = options[:aws_access_key_id]
@aws_secret_access_key = options[:aws_secret_access_key]
@hmac = HMAC::SHA1.new(@aws_secret_access_key)
@host = options[:host] || 's3.amazonaws.com'
@port = options[:port] || 443
@scheme = options[:scheme] || 'https'
@connection = Fog::Connection.new("#{@scheme}://#{@host}:#{@port}")
2009-05-19 02:06:49 -04:00
end
private
def parse_data(data)
metadata = {
:body => nil,
:headers => {}
}
2009-09-09 23:44:28 -04:00
if data.is_a?(String)
metadata[:body] = data
metadata[:headers]['Content-Length'] = metadata[:body].size.to_s
else
filename = File.basename(data.path)
unless (mime_types = MIME::Types.of(filename)).empty?
metadata[:headers]['Content-Type'] = mime_types.first.content_type
end
metadata[:body] = data.read
metadata[:headers]['Content-Length'] = File.size(data.path).to_s
end
2009-09-09 23:44:28 -04:00
# metadata[:headers]['Content-MD5'] = Base64.encode64(Digest::MD5.digest(metadata[:body])).strip
metadata
end
def request(params)
2009-06-23 23:55:57 -04:00
params[:headers]['Date'] = Time.now.utc.strftime("%a, %d %b %Y %H:%M:%S +0000")
params[:headers]['Authorization'] = "AWS #{@aws_access_key_id}:#{signature(params)}"
response = @connection.request({
:block => params[:block],
:body => params[:body],
:expects => params[:expects],
:headers => params[:headers],
:host => params[:host],
:method => params[:method],
:parser => params[:parser],
:path => params[:path],
:query => params[:query]
})
response
end
def url(params, expires)
params[:headers]['Date'] = expires.to_i
query = [params[:query]].compact
query << "AWSAccessKeyId=#{@aws_access_key_id}"
query << "Signature=#{CGI.escape(signature(params))}"
query << "Expires=#{params[:headers]['Date']}"
"http://#{params[:host]}/#{params[:path]}?#{query.join('&')}"
end
def signature(params)
string_to_sign =
<<-DATA
#{params[:method]}
#{params[:headers]['Content-MD5']}
#{params[:headers]['Content-Type']}
#{params[:headers]['Date']}
DATA
amz_headers, canonical_amz_headers = {}, ''
for key, value in params[:headers]
if key[0..5] == 'x-amz-'
amz_headers[key] = value
2009-06-22 13:06:49 -04:00
end
end
amz_headers = amz_headers.sort {|x, y| x[0] <=> y[0]}
for pair in amz_headers
canonical_amz_headers << "#{pair[0]}:#{pair[1]}\n"
end
string_to_sign << "#{canonical_amz_headers}"
2009-07-22 22:28:53 -04:00
subdomain = params[:host].split(".#{@host}").first
unless subdomain =~ /^(?:[a-z]|\d(?!\d{0,2}(?:\.\d{1,3}){3}$))(?:[a-z0-9]|\.(?![\.\-])|\-(?![\.])){1,61}[a-z0-9]$/
puts("[WARN] fog: the specified s3 bucket name(#{subdomain}) is not a valid dns name. See: http://docs.amazonwebservices.com/AmazonS3/latest/dev/index.html?Introduction.html")
params[:host] = params[:host].split("#{subdomain}.")[-1]
if params[:path]
params[:path] = "#{subdomain}/#{params[:path]}"
else
params[:path] = "#{subdomain}"
end
subdomain = nil
end
canonical_resource = "/"
unless subdomain.nil? || subdomain == @host
canonical_resource << "#{CGI.escape(subdomain).downcase}/"
end
canonical_resource << "#{params[:path]}"
2009-08-17 21:34:44 -04:00
if ['acl', 'location', 'logging', 'requestPayment', 'torrent'].include?(params[:query])
canonical_resource << "?#{params[:query]}"
end
string_to_sign << "#{canonical_resource}"
hmac = @hmac.update(string_to_sign)
signature = Base64.encode64(hmac.digest).chomp!
2009-05-19 02:06:49 -04:00
end
2009-06-05 19:51:17 -04:00
2009-05-19 02:06:49 -04:00
end
end
2009-08-17 18:11:53 -04:00
end
Fog::AWS::S3.reload