fog-internet-archive 0.0.1

data/lib/fog/internet_archive/requests/storage/put_bucket.rb

@@ -0,0 +1,70 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        # Create an S3 bucket
+        #
+        # @param bucket_name [String] name of bucket to create
+        # @option options [Hash] config arguments for bucket.  Defaults to {}.
+        # @option options LocationConstraint [Symbol] sets the location for the bucket
+        # @option options x-amz-acl [String] Permissions, must be in ['private', 'public-read', 'public-read-write', 'authenticated-read']
+        #
+        # @return [Excon::Response] response:
+        #   * status [Integer] 200
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUT.html
+        #
+        def put_bucket(bucket_name, options = {})
+          if location_constraint = options.delete('LocationConstraint')
+            data =
+<<-DATA
+  <CreateBucketConfiguration>
+    <LocationConstraint>#{location_constraint}</LocationConstraint>
+  </CreateBucketConfiguration>
+DATA
+          else
+            data = nil
+          end
+          request({
+            :expects    => 200,
+            :body       => data,
+            :headers    => options,
+            :idempotent => true,
+            :host       => "#{bucket_name}.#{@host}",
+            :method     => 'PUT'
+          })
+        end
+      end
+
+      class Mock # :nodoc:all
+        def put_bucket(bucket_name, options = {})
+          acl = options['x-amz-acl'] || 'private'
+          if !['private', 'public-read', 'public-read-write', 'authenticated-read'].include?(acl)
+            raise Excon::Errors::BadRequest.new('invalid x-amz-acl')
+          else
+            self.data[:acls][:bucket][bucket_name] = self.class.acls(acl)
+          end
+
+          response = Excon::Response.new
+          response.status = 200
+          bucket = {
+            :objects        => {},
+            'Name'          => bucket_name,
+            'CreationDate'  => Time.now,
+            'Owner'         => { 'DisplayName' => 'owner', 'ID' => 'some_id'},
+            'Payer'         => 'BucketOwner'
+          }
+          if options['LocationConstraint']
+            bucket['LocationConstraint'] = options['LocationConstraint']
+          else
+            bucket['LocationConstraint'] = nil
+          end
+          unless self.data[:buckets][bucket_name]
+            self.data[:buckets][bucket_name] = bucket
+          end
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_bucket_acl.rb

@@ -0,0 +1,69 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        require 'fog/internet_archive/requests/storage/acl_utils'
+
+        # Change access control list for an S3 bucket
+        #
+        # @param bucket_name [String] name of bucket to modify
+        # @param acl [Hash]
+        #   * Owner [Hash]:
+        #     * ID [String]: id of owner
+        #     * DisplayName [String]: display name of owner
+        #   * AccessControlList [Array]:
+        #     * Grantee [Hash]:
+        #       * DisplayName [String] Display name of grantee
+        #       * ID [String] Id of grantee
+        #       or
+        #       * EmailAddress [String] Email address of grantee
+        #       or
+        #       * URI [String] URI of group to grant access for
+        #     * Permission [String] Permission, in [FULL_CONTROL, WRITE, WRITE_ACP, READ, READ_ACP]
+        # * acl [String] Permissions, must be in ['private', 'public-read', 'public-read-write', 'authenticated-read']
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUTacl.html
+
+        def put_bucket_acl(bucket_name, acl)
+          data = ""
+          headers = {}
+
+          if acl.is_a?(Hash)
+            data = Fog::Storage::InternetArchive.hash_to_acl(acl)
+          else
+            if !['private', 'public-read', 'public-read-write', 'authenticated-read'].include?(acl)
+              raise Excon::Errors::BadRequest.new('invalid x-amz-acl')
+            end
+            headers['x-amz-acl'] = acl
+          end
+
+          headers['Content-MD5'] = Base64.encode64(Digest::MD5.digest(data)).strip
+          headers['Content-Type'] = 'application/json'
+          headers['Date'] = Fog::Time.now.to_date_header
+
+          request({
+            :body     => data,
+            :expects  => 200,
+            :headers  => headers,
+            :host     => "#{bucket_name}.#{@host}",
+            :method   => 'PUT',
+            :query    => {'acl' => nil}
+          })
+        end
+      end
+
+      class Mock
+        def put_bucket_acl(bucket_name, acl)
+          if acl.is_a?(Hash)
+            self.data[:acls][:bucket][bucket_name] = Fog::Storage::InternetArchive.hash_to_acl(acl)
+          else
+            if !['private', 'public-read', 'public-read-write', 'authenticated-read'].include?(acl)
+              raise Excon::Errors::BadRequest.new('invalid x-amz-acl')
+            end
+            self.data[:acls][:bucket][bucket_name] = acl
+          end
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_bucket_cors.rb

@@ -0,0 +1,47 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        require 'fog/internet_archive/requests/storage/cors_utils'
+
+        # Sets the cors configuration for your bucket. If the configuration exists, Amazon S3 replaces it.
+        #
+        # @param bucket_name [String] name of bucket to modify
+        # @param cors [Hash]
+        #   * CORSConfiguration [Array]:
+        #     * ID [String]: A unique identifier for the rule.
+        #     * AllowedMethod [String]: An HTTP method that you want to allow the origin to execute.
+        #     * AllowedOrigin [String]: An origin that you want to allow cross-domain requests from.
+        #     * AllowedHeader [String]: Specifies which headers are allowed in a pre-flight OPTIONS request via the Access-Control-Request-Headers header.
+        #     * MaxAgeSeconds [String]: The time in seconds that your browser is to cache the preflight response for the specified resource.
+        #     * ExposeHeader [String]: One or more headers in the response that you want customers to be able to access from their applications.
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUTcors.html
+
+        def put_bucket_cors(bucket_name, cors)
+          data = Fog::Storage::InternetArchive.hash_to_cors(cors)
+
+          headers = {}
+          headers['Content-MD5'] = Base64.encode64(Digest::MD5.digest(data)).strip
+          headers['Content-Type'] = 'application/json'
+          headers['Date'] = Fog::Time.now.to_date_header
+
+          request({
+            :body     => data,
+            :expects  => 200,
+            :headers  => headers,
+            :host     => "#{bucket_name}.#{@host}",
+            :method   => 'PUT',
+            :query    => {'cors' => nil}
+          })
+        end
+      end
+
+      class Mock
+        def put_bucket_cors(bucket_name, cors)
+          self.data[:cors][:bucket][bucket_name] = Fog::Storage::InternetArchive.hash_to_cors(cors)
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_bucket_lifecycle.rb

@@ -0,0 +1,76 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        # Change lifecycle configuration for an S3 bucket
+        #
+        # @param bucket_name [String] name of bucket to set lifecycle configuration for
+        # * lifecycle [Hash]:
+        #   * Rules [Array] object expire rules
+        #     * ID [String] Unique identifier for the rule
+        #     * Prefix [String] Prefix identifying one or more objects to which the rule applies
+        #     * Enabled [Boolean] if rule is currently being applied
+        #     * Expiration [Hash] Container for the object expiration rule.
+        #       * Days [Integer] lifetime, in days, of the objects that are subject to the rule
+        #       * Date [Date] Indicates when the specific rule take effect.
+        #         The date value must conform to the ISO 8601 format. The time is always midnight UTC.
+        #     * Transition [Hash] Container for the transition rule that describes when objects transition
+        #       to the Glacier storage class
+        #       * Days [Integer] lifetime, in days, of the objects that are subject to the rule
+        #       * Date [Date] Indicates when the specific rule take effect.
+        #         The date value must conform to the ISO 8601 format. The time is always midnight UTC.
+        #       * StorageClass [String] Indicates the Amazon S3 storage class to which you want the object
+        #         to transition to.
+        #
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUTlifecycle.html
+        #
+        def put_bucket_lifecycle(bucket_name, lifecycle)
+          builder = Nokogiri::XML::Builder.new do
+            LifecycleConfiguration {
+              lifecycle['Rules'].each do |rule|
+                Rule {
+                  ID rule['ID']
+                  Prefix rule['Prefix']
+                  Status rule['Enabled'] ? 'Enabled' : 'Disabled'
+                  unless (rule['Expiration'] or rule['Transition'])
+                    Expiration { Days rule['Days'] }
+                  else
+                    if rule['Expiration']
+                      if rule['Expiration']['Days']
+                        Expiration { Days rule['Expiration']['Days'] }
+                      elsif rule['Expiration']['Date']
+                        Expiration { Date rule['Expiration']['Date'].is_a?(Time) ? rule['Expiration']['Date'].utc.iso8601 : Time.parse(rule['Expiration']['Date']).utc.iso8601 }
+                      end
+                    end
+                    if rule['Transition']
+                      Transition {
+                        if rule['Transition']['Days']
+                          Days rule['Transition']['Days']
+                        elsif rule['Transition']['Date']
+                          Date rule['Transition']['Date'].is_a?(Time) ? time.utc.iso8601 : Time.parse(time).utc.iso8601
+                        end
+                        StorageClass rule['Transition']['StorageClass'].nil? ? 'GLACIER' : rule['Transition']['StorageClass']
+                      }
+                    end
+                  end
+                }
+              end
+            }
+          end
+          body = builder.to_xml
+          body.gsub! /<([^<>]+)\/>/, '<\1></\1>'
+          request({
+                    :body     => body,
+                    :expects  => 200,
+                    :headers  => {'Content-MD5' => Base64.encode64(Digest::MD5.digest(body)).chomp!,
+                      'Content-Type' => 'application/xml'},
+                    :host     => "#{bucket_name}.#{@host}",
+                    :method   => 'PUT',
+                    :query    => {'lifecycle' => nil}
+                  })
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_bucket_logging.rb

@@ -0,0 +1,79 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        # Change logging status for an S3 bucket
+        #
+        # @param bucket_name [String] name of bucket to modify
+        # @param logging_status [Hash]:
+        #   * Owner [Hash]:
+        #     * ID [String]: id of owner
+        #     * DisplayName [String]: display name of owner
+        #   * AccessControlList [Array]:
+        #     * Grantee [Hash]:
+        #       * DisplayName [String] Display name of grantee
+        #       * ID [String] Id of grantee
+        #       or
+        #       * EmailAddress [String] Email address of grantee
+        #       or
+        #       * URI [String] URI of group to grant access for
+        #     * Permission [String] Permission, in [FULL_CONTROL, WRITE, WRITE_ACP, READ, READ_ACP]
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUTlogging.html
+
+        def put_bucket_logging(bucket_name, logging_status)
+          if logging_status['LoggingEnabled'].empty?
+            data =
+<<-DATA
+<BucketLoggingStatus xmlns="http://doc.s3.amazonaws.com/2006-03-01" />
+DATA
+          else
+            data =
+<<-DATA
+<BucketLoggingStatus xmlns="http://doc.s3.amazonaws.com/2006-03-01">
+  <LoggingEnabled>
+    <TargetBucket>#{logging_status['LoggingEnabled']['TargetBucket']}</TargetBucket>
+    <TargetPrefix>#{logging_status['LoggingEnabled']['TargetBucket']}</TargetPrefix>
+    <TargetGrants>
+DATA
+
+            acl['AccessControlList'].each do |grant|
+              data << "      <Grant>"
+              type = case grant['Grantee'].keys.sort
+              when ['DisplayName', 'ID']
+                'CanonicalUser'
+              when ['EmailAddress']
+                'AmazonCustomerByEmail'
+              when ['URI']
+                'Group'
+              end
+              data << "        <Grantee xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"#{type}\">"
+              for key, value in grant['Grantee']
+                data << "          <#{key}>#{value}</#{key}>"
+              end
+              data << "        </Grantee>"
+              data << "        <Permission>#{grant['Permission']}</Permission>"
+              data << "      </Grant>"
+            end
+
+            data <<
+<<-DATA
+    </TargetGrants>
+  </LoggingEnabled>
+</BucketLoggingStatus>
+DATA
+          end
+
+          request({
+            :body     => data,
+            :expects  => 200,
+            :headers  => {},
+            :host     => "#{bucket_name}.#{@host}",
+            :method   => 'PUT',
+            :query    => {'logging' => nil}
+          })
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_bucket_policy.rb

@@ -0,0 +1,25 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        # Change bucket policy for an S3 bucket
+        #
+        # @param bucket_name [String] name of bucket to modify
+        # @param policy [Hash] policy document
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUTpolicy.html
+
+        def put_bucket_policy(bucket_name, policy)
+          request({
+            :body     => Fog::JSON.encode(policy),
+            :expects  => 204,
+            :headers  => {},
+            :host     => "#{bucket_name}.#{@host}",
+            :method   => 'PUT',
+            :query    => {'policy' => nil}
+          })
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_bucket_website.rb

@@ -0,0 +1,59 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        # Change website configuration for an S3 bucket
+        #
+        # @param bucket_name [String] name of bucket to modify
+        # @param suffix [String] suffix to append to requests for the bucket
+        # @param options [Hash]
+        # @option options key [String] key to use for 4XX class errors
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTBucketPUTwebsite.html
+
+        def put_bucket_website(bucket_name, suffix, options = {})
+          data =
+<<-DATA
+<WebsiteConfiguration xmlns="http://s3.amazonaws.com/doc/2006-03-01/">
+    <IndexDocument>
+        <Suffix>#{suffix}</Suffix>
+    </IndexDocument>
+DATA
+
+          if options[:key]
+            data <<
+<<-DATA
+<ErrorDocument>
+    <Key>#{options[:key]}</Key>
+</ErrorDocument>
+DATA
+          end
+
+          data << '</WebsiteConfiguration>'
+          request({
+            :body     => data,
+            :expects  => 200,
+            :headers  => {},
+            :host     => "#{bucket_name}.#{@host}",
+            :method   => 'PUT',
+            :query    => {'website' => nil}
+          })
+        end
+      end
+
+      class Mock # :nodoc:all
+        def put_bucket_website(bucket_name, suffix, options = {})
+          response = Excon::Response.new
+          if self.data[:buckets][bucket_name]
+            response.status = 200
+          else
+            response.status = 404
+            raise(Excon::Errors.status_error({:expects => 200}, response))
+          end
+
+          response
+        end
+      end
+    end
+  end
+end

data/lib/fog/internet_archive/requests/storage/put_object.rb

@@ -0,0 +1,94 @@
+module Fog
+  module Storage
+    class InternetArchive
+      class Real
+        # Create an object in an S3 bucket
+        #
+        # @param bucket_name [String] Name of bucket to create object in
+        # @param object_name [String] Name of object to create
+        # @param data [File||String] File or String to create object from
+        # @param options [Hash]
+        # @option options Cache-Control [String] Caching behaviour
+        # @option options Content-Disposition [String] Presentational information for the object
+        # @option options Content-Encoding [String] Encoding of object data
+        # @option options Content-Length [String] Size of object in bytes (defaults to object.read.length)
+        # @option options Content-MD5 [String] Base64 encoded 128-bit MD5 digest of message
+        # @option options Content-Type [String] Standard MIME type describing contents (defaults to MIME::Types.of.first)
+        # @option options Expires [String] Cache expiry
+        # @option options x-amz-acl [String] Permissions, must be in ['private', 'public-read', 'public-read-write', 'authenticated-read']
+        # @option options x-amz-storage-class [String] Default is 'STANDARD', set to 'REDUCED_REDUNDANCY' for non-critical, reproducable data
+        # @option options x-amz-meta-#{name} Headers to be returned with object, note total size of request without body must be less than 8 KB.
+        #
+        # @return [Excon::Response] response:
+        #   * headers [Hash]:
+        #     * ETag [String] etag of new object
+        #
+        # @see http://docs.amazonwebservices.com/AmazonS3/latest/API/RESTObjectPUT.html
+
+        def put_object(bucket_name, object_name, data, options = {})
+          data = Fog::Storage.parse_data(data)
+          headers = data[:headers].merge!(options)
+          request({
+            :body       => data[:body],
+            :expects    => 200,
+            :headers    => headers,
+            :host       => "#{bucket_name}.#{@host}",
+            :idempotent => true,
+            :method     => 'PUT',
+            :path       => CGI.escape(object_name)
+          })
+        end
+      end
+
+      class Mock # :nodoc:all
+        def put_object(bucket_name, object_name, data, options = {})
+          acl = options['x-amz-acl'] || 'private'
+          if !['private', 'public-read', 'public-read-write', 'authenticated-read'].include?(acl)
+            raise Excon::Errors::BadRequest.new('invalid x-amz-acl')
+          else
+            self.data[:acls][:object][bucket_name] ||= {}
+            self.data[:acls][:object][bucket_name][object_name] = self.class.acls(acl)
+          end
+
+          data = Fog::Storage.parse_data(data)
+          unless data[:body].is_a?(String)
+            data[:body] = data[:body].read
+          end
+          response = Excon::Response.new
+          if (bucket = self.data[:buckets][bucket_name])
+            response.status = 200
+            object = {
+              :body             => data[:body],
+              'Content-Type'    => options['Content-Type'] || data[:headers]['Content-Type'],
+              'ETag'            => Digest::MD5.hexdigest(data[:body]),
+              'Key'             => object_name,
+              'Last-Modified'   => Fog::Time.now.to_date_header,
+              'Content-Length'  => options['Content-Length'] || data[:headers]['Content-Length'],
+            }
+
+            for key, value in options
+              case key
+              when 'Cache-Control', 'Content-Disposition', 'Content-Encoding', 'Content-MD5', 'Expires', /^x-amz-meta-/, /^x-archive-/
+                object[key] = value
+              end
+            end
+
+            bucket[:objects][object_name] = [object]
+
+            response.headers = {
+              'Content-Length'   => object['Content-Length'],
+              'Content-Type'     => object['Content-Type'],
+              'ETag'             => object['ETag'],
+              'Last-Modified'    => object['Last-Modified'],
+            }
+
+          else
+            response.status = 404
+            raise(Excon::Errors.status_error({:expects => 200}, response))
+          end
+          response
+        end
+      end
+    end
+  end
+end