qoobaa-s3 0.0.3

data/lib/s3/connection.rb

@@ -0,0 +1,152 @@
+module S3
+  class Connection
+    attr_accessor :access_key_id, :secret_access_key, :use_ssl, :timeout, :debug
+    alias :use_ssl? :use_ssl
+
+    def initialize(options = {})
+      @access_key_id = options[:access_key_id]
+      @secret_access_key = options[:secret_access_key]
+      @use_ssl = options[:use_ssl] || false
+      @debug = options[:debug]
+      @timeout = options[:timeout]
+    end
+
+    def request(method, options)
+      host = options[:host] || HOST
+      path = options[:path] or raise ArgumentError.new("no path given")
+      body = options[:body]
+      params = options[:params]
+      headers = options[:headers]
+
+      if params
+        params = params.is_a?(String) ? params : self.class.parse_params(params)
+        path << "?#{params}"
+      end
+
+      path = URI.escape(path)
+      request = request_class(method).new(path)
+
+      headers = self.class.parse_headers(headers)
+      headers.each do |key, value|
+        request[key] = value
+      end
+
+      request.body = body
+
+      send_request(host, request)
+    end
+
+    def self.parse_params(params)
+      interesting_keys = [:max_keys, :prefix, :marker, :delimiter, :location]
+
+      result = []
+      params.each do |key, value|
+        if interesting_keys.include?(key)
+          parsed_key = key.to_s.gsub("_", "-")
+          case value
+          when nil
+            result << parsed_key
+          else
+            result << "#{parsed_key}=#{value}"
+          end
+        end
+      end
+      result.join("&")
+    end
+
+    def self.parse_headers(headers)
+      interesting_keys = [:content_type, :x_amz_acl, :range,
+                          :if_modified_since, :if_unmodified_since,
+                          :if_match, :if_none_match,
+                          :content_disposition, :content_encoding,
+                          :x_amz_copy_source, :x_amz_metadata_directive,
+                          :x_amz_copy_source_if_match,
+                          :x_amz_copy_source_if_none_match,
+                          :x_amz_copy_source_if_unmodified_since,
+                          :x_amz_copy_source_if_modified_since]
+
+      parsed_headers = {}
+      if headers
+        headers.each do |key, value|
+          if interesting_keys.include?(key)
+            parsed_key = key.to_s.gsub("_", "-")
+            parsed_value = value
+            case value
+            when Range
+              parsed_value = "bytes=#{value.first}-#{value.last}"
+            end
+            parsed_headers[parsed_key] = parsed_value
+          end
+        end
+      end
+      parsed_headers
+    end
+
+    private
+
+    def request_class(method)
+      case method
+      when :get
+        request_class = Net::HTTP::Get
+      when :put
+        request_class = Net::HTTP::Put
+      when :delete
+        request_class = Net::HTTP::Delete
+      end
+    end
+
+    def port
+      use_ssl ? 443 : 80
+    end
+
+    def http(host)
+      http = Net::HTTP.new(host, port)
+      http.set_debug_output(STDOUT) if @debug
+      http.use_ssl = @use_ssl
+      http.verify_mode = OpenSSL::SSL::VERIFY_NONE if @use_ssl
+      http.read_timeout = @timeout if @timeout
+      http
+    end
+
+    def send_request(host, request)
+      response = http(host).start do |http|
+        host = http.address
+
+        request['Date'] ||= Time.now.httpdate
+
+        if request.body
+          request["Content-Type"] ||= "application/octet-stream"
+          request["Content-MD5"] = Base64.encode64(Digest::MD5.digest(request.body)).chomp
+        end
+
+        request["Authorization"] = S3::Signature.generate(:host => host,
+                                                          :request => request,
+                                                          :access_key_id => access_key_id,
+                                                          :secret_access_key => secret_access_key)
+        http.request(request)
+      end
+
+      handle_response(response)
+    end
+
+    def handle_response(response)
+      case response.code.to_i
+      when 200...300
+        response
+      when 300...600
+        if response.body.nil? || response.body.empty?
+          raise S3::Error::ResponseError.new(nil, response)
+        else
+          xml = XmlSimple.xml_in(response.body)
+          message = xml["Message"].first
+          code = xml["Code"].first
+          raise S3::Error::ResponseError.exception(code).new(message, response)
+        end
+      else
+        raise(ConnectionError.new(response, "Unknown response code: #{response.code}"))
+      end
+      response
+    end
+
+  end
+end

data/lib/s3/exceptions.rb

@@ -0,0 +1,94 @@
+module S3
+  module Error
+    class Exception < StandardError
+    end
+
+    # All responses with a code between 300 and 599 that contain an <Error></Error> body are wrapped in an
+    # ErrorResponse which contains an Error object. This Error class generates a custom exception with the name
+    # of the xml Error and its message. All such runtime generated exception classes descend from ResponseError
+    # and contain the ErrorResponse object so that all code that makes a request can rescue ResponseError and get
+    # access to the ErrorResponse.
+    class ResponseError < Exception
+      attr_reader :response
+      def initialize(message, response)
+        @response = response
+        super(message)
+      end
+
+      def self.exception(code)
+        S3::Error.const_get(code)
+      rescue NameError
+        ResponseError
+      end
+    end
+
+    #:stopdoc:
+
+    class AccessDenied < ResponseError; end
+    class AccountProblem < ResponseError; end
+    class AmbiguousGrantByEmailAddress < ResponseError; end
+    class BadDigest < ResponseError; end
+    class BucketAlreadyExists < ResponseError; end
+    class BucketAlreadyOwnedByYou < ResponseError; end
+    class BucketNotEmpty < ResponseError; end
+    class CredentialsNotSupported < ResponseError; end
+    class CrossLocationLoggingProhibited < ResponseError; end
+    class EntityTooSmall < ResponseError; end
+    class EntityTooLarge < ResponseError; end
+    class ExpiredToken < ResponseError; end
+    class IncompleteBody < ResponseError; end
+    class IncorrectNumberOfFilesInPostRequestPOST < ResponseError; end
+    class InlineDataTooLarge < ResponseError; end
+    class InternalError < ResponseError; end
+    class InvalidAccessKeyId < ResponseError; end
+    class InvalidAddressingHeader < ResponseError; end
+    class InvalidArgument < ResponseError; end
+    class InvalidBucketName < ResponseError; end
+    class InvalidDigest < ResponseError; end
+    class InvalidLocationConstraint < ResponseError; end
+    class InvalidPayer < ResponseError; end
+    class InvalidPolicyDocument < ResponseError; end
+    class InvalidRange < ResponseError; end
+    class InvalidSecurity < ResponseError; end
+    class InvalidSOAPRequest < ResponseError; end
+    class InvalidStorageClass < ResponseError; end
+    class InvalidTargetBucketForLogging < ResponseError; end
+    class InvalidToken < ResponseError; end
+    class InvalidURI < ResponseError; end
+    class KeyTooLong < ResponseError; end
+    class MalformedACLError < ResponseError; end
+    class MalformedACLError < ResponseError; end
+    class MalformedPOSTRequest < ResponseError; end
+    class MalformedXML < ResponseError; end
+    class MaxMessageLengthExceeded < ResponseError; end
+    class MaxPostPreDataLengthExceededErrorYour < ResponseError; end
+    class MetadataTooLarge < ResponseError; end
+    class MethodNotAllowed < ResponseError; end
+    class MissingAttachment < ResponseError; end
+    class MissingContentLength < ResponseError; end
+    class MissingRequestBodyError < ResponseError; end
+    class MissingSecurityElement < ResponseError; end
+    class MissingSecurityHeader < ResponseError; end
+    class NoLoggingStatusForKey < ResponseError; end
+    class NoSuchBucket < ResponseError; end
+    class NoSuchKey < ResponseError; end
+    class NotImplemented < ResponseError; end
+    class NotSignedUp < ResponseError; end
+    class OperationAborted < ResponseError; end
+    class PermanentRedirect < ResponseError; end
+    class PreconditionFailed < ResponseError; end
+    class Redirect < ResponseError; end
+    class RequestIsNotMultiPartContent < ResponseError; end
+    class RequestTimeout < ResponseError; end
+    class RequestTimeTooSkewed < ResponseError; end
+    class RequestTorrentOfBucketError < ResponseError; end
+    class SignatureDoesNotMatch < ResponseError; end
+    class SlowDown < ResponseError; end
+    class TemporaryRedirect < ResponseError; end
+    class TokenRefreshRequired < ResponseError; end
+    class TooManyBuckets < ResponseError; end
+    class UnexpectedContent < ResponseError; end
+    class UnresolvableGrantByEmailAddress < ResponseError; end
+    class UserKeyMustBeSpecified < ResponseError; end
+  end
+end

data/lib/s3/object.rb

@@ -0,0 +1,153 @@
+module S3
+  class Object
+    extend Forwardable
+
+    attr_accessor :content_type, :content_disposition, :content_encoding
+    attr_reader :last_modified, :etag, :size, :bucket, :key, :acl
+    attr_writer :content
+
+    def_instance_delegators :bucket, :name, :service, :bucket_request, :vhost?, :host, :path_prefix
+    def_instance_delegators :service, :protocol, :port
+
+    def full_key
+      [name, key].join("/")
+    end
+
+    def key=(key)
+      raise ArgumentError.new("Invalid key name: #{key}") unless key_valid?(key)
+      @key ||= key
+    end
+
+    def acl=(acl)
+      @acl = acl.to_s.gsub("_", "-")
+    end
+
+    def retrieve
+      response = object_request(:get, :headers => { :range => 0..0 })
+      parse_headers(response)
+      self
+    end
+
+    def exists?
+      retrieve
+      true
+    rescue Error::NoSuchKey
+      false
+    end
+
+    def content(reload = false)
+      if reload or @content.nil?
+        response = object_request(:get)
+        parse_headers(response)
+        self.content = response.body
+      end
+      @content
+    end
+
+    def save
+      body = content.is_a?(IO) ? content.read : content
+      response = object_request(:put, :body => body, :headers => dump_headers)
+      parse_headers(response)
+      true
+    end
+
+    def copy(options = {})
+      key = options[:key] || self.key
+      bucket = options[:bucket] || self.bucket
+
+      headers = {}
+      headers[:x_amz_acl] = options[:acl] || acl || "public-read"
+      headers[:content_type] = options[:content_type] || content_type || "application/octet-stream"
+      headers[:content_encoding] = options[:content_encoding] if options[:content_encoding]
+      headers[:content_disposition] = options[:content_disposition] if options[:content_disposition]
+      headers[:x_amz_copy_source] = full_key
+      headers[:x_amz_metadata_directive] = "REPLACE"
+      headers[:x_amz_copy_source_if_match] = options[:if_match] if options[:if_match]
+      headers[:x_amz_copy_source_if_none_match] = options[:if_none_match] if options[:if_none_match]
+      headers[:x_amz_copy_source_if_unmodified_since] = options[:if_modified_since] if options[:if_modified_since]
+      headers[:x_amz_copy_source_if_modified_since] = options[:if_unmodified_since] if options[:if_unmodified_since]
+
+      response = bucket.send(:bucket_request, :put, :path => key, :headers => headers)
+      self.class.parse_copied(:object => self, :bucket => bucket, :key => key, :body => response.body, :headers => headers)
+    end
+
+    def destroy
+      object_request(:delete)
+      true
+    end
+
+    def url
+      "#{protocol}#{host}/#{path_prefix}#{key}"
+    end
+
+    def cname_url
+      "#{protocol}#{name}/#{key}" if bucket.vhost?
+    end
+
+    def inspect
+      "#<#{self.class}:/#{name}/#{key}>"
+    end
+
+    def initialize(bucket, key, options = {})
+      self.bucket = bucket
+      self.key = key
+      self.last_modified = options[:last_modified]
+      self.etag = options[:etag]
+      self.size = options[:size]
+    end
+
+    private
+
+    attr_writer :last_modified, :etag, :size, :original_key, :bucket
+
+    def object_request(method, options = {})
+      bucket_request(method, options.merge(:path => key))
+    end
+
+    def last_modified=(last_modified)
+      @last_modified = Time.parse(last_modified) if last_modified
+    end
+
+    def etag=(etag)
+      @etag = etag[1..-2] if etag
+    end
+
+    def dump_headers
+      headers = {}
+      headers[:x_amz_acl] = @acl || "public-read"
+      headers[:content_type] = @content_type || "application/octet-stream"
+      headers[:content_encoding] = @content_encoding if @content_encoding
+      headers[:content_disposition] = @content_disposition if @content_disposition
+      headers
+    end
+
+    def key_valid?(key)
+      key !~ /\/\//
+    end
+
+    def parse_headers(response)
+      self.etag = response["etag"]
+      self.content_type = response["content-type"]
+      self.content_disposition = response["content-disposition"]
+      self.content_encoding = response["content-encoding"]
+      self.last_modified = response["last-modified"]
+      self.size = response["content-length"]
+      if response["content-range"]
+        self.size = response["content-range"].sub(/[^\/]+\//, "").to_i
+      end
+    end
+
+    def self.parse_copied(options)
+      xml = XmlSimple.xml_in(options[:body])
+      etag = xml["ETag"].first
+      last_modified = xml["LastModified"].first
+      size = options[:object].size
+      object = Object.new(options[:bucket], options[:key], :etag => etag, :last_modified => last_modified, :size => size)
+      object.acl = options[:headers][:x_amz_acl]
+      object.content_type = options[:headers][:content_type]
+      object.content_encoding = options[:headers][:content_encoding]
+      object.content_disposition = options[:headers][:content_disposition]
+      object
+    end
+  end
+end

data/lib/s3/roxy/moxie.rb

@@ -0,0 +1,58 @@
+# Copyright (c) 2008 Ryan Daigle
+
+# Permission is hereby granted, free of charge, to any person
+# obtaining a copy of this software and associated documentation files
+# (the "Software"), to deal in the Software without restriction,
+# including without limitation the rights to use, copy, modify, merge,
+# publish, distribute, sublicense, and/or sell copies of the Software,
+# and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+module S3
+  module Roxy
+    module Moxie
+      # Set up this class to proxy on the given name
+      def proxy(name, options = {}, &block)
+
+        # Make sure args are OK
+        original_method = method_defined?(name) ? instance_method(name) : nil
+        raise "Cannot proxy an existing method, \"#{name}\", and also have a :to option.  Please use one or the other." if
+          original_method and options[:to]
+
+        # If we're proxying an existing method, we need to store
+        # the original method and move it out of the way so
+        # we can take over
+        if original_method
+          new_method = "proxied_#{name}"
+          alias_method new_method, "#{name}"
+          options[:to] = original_method
+        end
+
+        # Thanks to Jerry for this simplification of my original class_eval approach
+        # http://ryandaigle.com/articles/2008/11/10/implement-ruby-proxy-objects-with-roxy/comments/8059#comment-8059
+        if !original_method or original_method.arity == 0
+          define_method name do
+            @proxy_for ||= {}
+            @proxy_for[name] ||= Proxy.new(self, options, nil, &block)
+          end
+        else
+          define_method name do |*args|
+            Proxy.new(self, options, args, &block)
+          end
+        end
+      end
+    end
+  end
+end