fog-zj 0.1.28

data/lib/fog/storage/openstack/models/directory.rb

@@ -0,0 +1,50 @@
+require 'fog/openstack/models/model'
+require 'fog/storage/openstack/models/files'
+
+module Fog
+  module Storage
+    class OpenStack
+      class Directory < Fog::OpenStack::Model
+        identity  :key, :aliases => 'name'
+
+        attribute :bytes, :aliases => 'X-Container-Bytes-Used'
+        attribute :count, :aliases => 'X-Container-Object-Count'
+
+        attr_writer :public
+
+        def destroy
+          requires :key
+          service.delete_container(key)
+          true
+        rescue Excon::Errors::NotFound
+          false
+        end
+
+        def files
+          @files ||= begin
+            Fog::Storage::OpenStack::Files.new(
+              :directory => self,
+              :service   => service
+            )
+          end
+        end
+
+        def public_url
+          requires :key
+
+          @public_url ||= begin
+            service.public_url(key)
+          rescue Fog::Storage::OpenStack::NotFound => err
+            nil
+          end
+        end
+
+        def save
+          requires :key
+          service.put_container(key, :public => @public)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/storage/openstack/models/file.rb

@@ -0,0 +1,195 @@
+require 'fog/openstack/models/model'
+
+module Fog
+  module Storage
+    class OpenStack
+      class File < Fog::OpenStack::Model
+        identity  :key,             :aliases => 'name'
+
+        attribute :access_control_allow_origin, :aliases => ['Access-Control-Allow-Origin']
+        attribute :cache_control,   :aliases => ['Cache-Control']
+        attribute :content_length,  :aliases => ['bytes', 'Content-Length'], :type => :integer
+        attribute :content_type,    :aliases => ['content_type', 'Content-Type']
+        attribute :content_disposition, :aliases => ['content_disposition', 'Content-Disposition']
+        attribute :etag,            :aliases => ['hash', 'Etag']
+        attribute :last_modified,   :aliases => ['last_modified', 'Last-Modified'], :type => :time
+        attribute :metadata
+        attribute :origin,          :aliases => ['Origin']
+        # @!attribute [rw] delete_at
+        # A Unix Epoch Timestamp, in integer form, representing the time when this object will be automatically deleted.
+        # @return [Integer] the unix epoch timestamp of when this object will be automatically deleted
+        # @see http://docs.openstack.org/developer/swift/overview_expiring_objects.html
+        attribute :delete_at, :aliases => ['X-Delete-At']
+
+        # @!attribute [rw] delete_after
+        # A number of seconds representing how long from now this object will be automatically deleted.
+        # @return [Integer] the number of seconds until this object will be automatically deleted
+        # @see http://docs.openstack.org/developer/swift/overview_expiring_objects.html
+        attribute :delete_after, :aliases => ['X-Delete-After']
+
+        # @!attribute [rw] content_encoding
+        # When you create an object or update its metadata, you can optionally set the Content-Encoding metadata.
+        # This metadata enables you to indicate that the object content is compressed without losing the identity of the
+        # underlying media type (Content-Type) of the file, such as a video.
+        # @see http://docs.openstack.org/developer/swift/api/use_content-encoding_metadata.html#use-content-encoding-metadata
+        attribute :content_encoding, :aliases => 'Content-Encoding'
+
+        def initialize(new_attributes = {})
+          super
+          @dirty = if last_modified then false else true end
+        end
+
+        def body
+          attributes[:body] ||= if last_modified
+                                  collection.get(identity).try(:body) || ''
+                                else
+                                  ''
+                                end
+        end
+
+        def body=(new_body)
+          attributes[:body] = new_body
+          @dirty = true
+        end
+
+        attr_reader :directory
+
+        def copy(target_directory_key, target_file_key, options = {})
+          requires :directory, :key
+          options['Content-Type'] ||= content_type if content_type
+          options['Access-Control-Allow-Origin'] ||= access_control_allow_origin if access_control_allow_origin
+          options['Origin'] ||= origin if origin
+          options['Content-Encoding'] ||= content_encoding if content_encoding
+          service.copy_object(directory.key, key, target_directory_key, target_file_key, options)
+          target_directory = service.directories.new(:key => target_directory_key)
+          target_directory.files.get(target_file_key)
+        end
+
+        def destroy
+          requires :directory, :key
+          service.delete_object(directory.key, key)
+          @dirty = true
+          true
+        end
+
+        def metadata
+          attributes[:metadata] ||= headers_to_metadata
+        end
+
+        def owner=(new_owner)
+          if new_owner
+            attributes[:owner] = {
+              :display_name => new_owner['DisplayName'],
+              :id           => new_owner['ID']
+            }
+          end
+        end
+
+        def public=(new_public)
+          new_public
+        end
+
+        # Get a url for file.
+        #
+        #     required attributes: key
+        #
+        # @param expires [String] number of seconds (since 1970-01-01 00:00) before url expires
+        # @param options [Hash]
+        # @return [String] url
+        #
+        def url(expires, options = {})
+          requires :directory, :key
+          service.create_temp_url(directory.key, key, expires, "GET", options)
+        end
+
+        def public_url
+          requires :key
+          collection.get_url(key)
+        end
+
+        def save(options = {})
+          requires :directory, :key
+          options['Cache-Control'] = cache_control if cache_control
+          options['Content-Type'] = content_type if content_type
+          options['Content-Disposition'] = content_disposition if content_disposition
+          options['Access-Control-Allow-Origin'] = access_control_allow_origin if access_control_allow_origin
+          options['Origin'] = origin if origin
+          options['X-Delete-At'] = delete_at if delete_at
+          options['X-Delete-After'] = delete_after if delete_after
+          options['Content-Encoding'] = content_encoding if content_encoding
+          options.merge!(metadata_to_headers)
+
+          if not @dirty
+            data = service.post_object(directory.key, key, options)
+          else
+            requires :body
+            data = service.put_object(directory.key, key, body, options)
+            self.content_length = Fog::Storage.get_body_size(body)
+            self.content_type ||= Fog::Storage.get_content_type(body)
+          end
+          update_attributes_from(data)
+          refresh_metadata
+
+          true
+        end
+
+        private
+
+        attr_writer :directory
+
+        def refresh_metadata
+          metadata.reject! { |_k, v| v.nil? }
+        end
+
+        def headers_to_metadata
+          key_map = key_mapping
+          Hash[metadata_attributes.map { |k, v| [key_map[k], v] }]
+        end
+
+        def key_mapping
+          key_map = metadata_attributes
+          key_map.each_pair { |k, _v| key_map[k] = header_to_key(k) }
+        end
+
+        def header_to_key(opt)
+          opt.gsub(metadata_prefix, '').split('-').map { |k| k[0, 1].downcase + k[1..-1] }.join('_').to_sym
+        end
+
+        def metadata_to_headers
+          header_map = header_mapping
+          Hash[metadata.map { |k, v| [header_map[k], v] }]
+        end
+
+        def header_mapping
+          header_map = metadata.dup
+          header_map.each_pair { |k, _v| header_map[k] = key_to_header(k) }
+        end
+
+        def key_to_header(key)
+          metadata_prefix + key.to_s.split(/[-_]/).map(&:capitalize).join('-')
+        end
+
+        def metadata_attributes
+          if last_modified
+            headers = service.head_object(directory.key, key).headers
+            headers.reject! { |k, _v| !metadata_attribute?(k) }
+          else
+            {}
+          end
+        end
+
+        def metadata_attribute?(key)
+          key.to_s =~ /^#{metadata_prefix}/
+        end
+
+        def metadata_prefix
+          "X-Object-Meta-"
+        end
+
+        def update_attributes_from(data)
+          merge_attributes(data.headers.reject { |key, _value| ['Content-Length', 'Content-Type'].include?(key) })
+        end
+      end
+    end
+  end
+end

data/lib/fog/storage/openstack/models/files.rb

@@ -0,0 +1,95 @@
+require 'fog/openstack/models/collection'
+require 'fog/storage/openstack/models/file'
+
+module Fog
+  module Storage
+    class OpenStack
+      class Files < Fog::OpenStack::Collection
+        attribute :directory
+        attribute :limit
+        attribute :marker
+        attribute :path
+        attribute :prefix
+
+        model Fog::Storage::OpenStack::File
+
+        def all(options = {})
+          requires :directory
+          options = {
+            'limit'  => limit,
+            'marker' => marker,
+            'path'   => path,
+            'prefix' => prefix
+          }.merge!(options)
+          merge_attributes(options)
+          parent = directory.collection.get(
+            directory.key,
+            options
+          )
+          if parent
+            # TODO: change to load_response?
+            load(parent.files.map(&:attributes))
+          end
+        end
+
+        alias each_file_this_page each
+        def each
+          if !block_given?
+            self
+          else
+            subset = dup.all
+
+            subset.each_file_this_page { |f| yield f }
+            while subset.length == (subset.limit || 10000)
+              subset = subset.all(:marker => subset.last.key)
+              subset.each_file_this_page { |f| yield f }
+            end
+
+            self
+          end
+        end
+
+        def get(key, &block)
+          requires :directory
+          data = service.get_object(directory.key, key, &block)
+          file_data = data.headers.merge(:body => data.body,
+                                         :key  => key)
+          new(file_data)
+        rescue Fog::Storage::OpenStack::NotFound
+          nil
+        end
+
+        def get_url(key)
+          requires :directory
+          if directory.public_url
+            "#{directory.public_url}/#{Fog::OpenStack.escape(key, '/')}"
+          end
+        end
+
+        def get_http_url(key, expires, options = {})
+          requires :directory
+          service.get_object_http_url(directory.key, key, expires, options)
+        end
+
+        def get_https_url(key, expires, options = {})
+          requires :directory
+          service.get_object_https_url(directory.key, key, expires, options)
+        end
+
+        def head(key, _options = {})
+          requires :directory
+          data = service.head_object(directory.key, key)
+          file_data = data.headers.merge(:key => key)
+          new(file_data)
+        rescue Fog::Storage::OpenStack::NotFound
+          nil
+        end
+
+        def new(attributes = {})
+          requires :directory
+          super({:directory => directory}.merge!(attributes))
+        end
+      end
+    end
+  end
+end

data/lib/fog/storage/openstack/requests/copy_object.rb

@@ -0,0 +1,23 @@
+module Fog
+  module Storage
+    class OpenStack
+      class Real
+        # Copy object
+        #
+        # ==== Parameters
+        # * source_container_name<~String> - Name of source bucket
+        # * source_object_name<~String> - Name of source object
+        # * target_container_name<~String> - Name of bucket to create copy in
+        # * target_object_name<~String> - Name for new copy of object
+        # * options<~Hash> - Additional headers
+        def copy_object(source_container_name, source_object_name, target_container_name, target_object_name, options = {})
+          headers = {'X-Copy-From' => "/#{source_container_name}/#{source_object_name}"}.merge(options)
+          request(:expects => 201,
+                  :headers => headers,
+                  :method  => 'PUT',
+                  :path    => "#{Fog::OpenStack.escape(target_container_name)}/#{Fog::OpenStack.escape(target_object_name)}")
+        end
+      end
+    end
+  end
+end

data/lib/fog/storage/openstack/requests/delete_container.rb

@@ -0,0 +1,20 @@
+module Fog
+  module Storage
+    class OpenStack
+      class Real
+        # Delete an existing container
+        #
+        # ==== Parameters
+        # * name<~String> - Name of container to delete
+        #
+        def delete_container(name)
+          request(
+            :expects => 204,
+            :method  => 'DELETE',
+            :path    => Fog::OpenStack.escape(name)
+          )
+        end
+      end
+    end
+  end
+end

data/lib/fog/storage/openstack/requests/delete_multiple_objects.rb

@@ -0,0 +1,65 @@
+module Fog
+  module Storage
+    class OpenStack
+      class Real
+        # Deletes multiple objects or containers with a single request.
+        #
+        # To delete objects from a single container, +container+ may be provided
+        # and +object_names+ should be an Array of object names within the container.
+        #
+        # To delete objects from multiple containers or delete containers,
+        # +container+ should be +nil+ and all +object_names+ should be prefixed with a container name.
+        #
+        # Containers must be empty when deleted. +object_names+ are processed in the order given,
+        # so objects within a container should be listed first to empty the container.
+        #
+        # Up to 10,000 objects may be deleted in a single request.
+        # The server will respond with +200 OK+ for all requests.
+        # +response.body+ must be inspected for actual results.
+        #
+        # @example Delete objects from a container
+        #   object_names = ['object', 'another/object']
+        #   conn.delete_multiple_objects('my_container', object_names)
+        #
+        # @example Delete objects from multiple containers
+        #   object_names = ['container_a/object', 'container_b/object']
+        #   conn.delete_multiple_objects(nil, object_names)
+        #
+        # @example Delete a container and all it's objects
+        #   object_names = ['my_container/object_a', 'my_container/object_b', 'my_container']
+        #   conn.delete_multiple_objects(nil, object_names)
+        #
+        # @param container [String,nil] Name of container.
+        # @param object_names [Array<String>] Object names to be deleted.
+        # @param options [Hash] Additional request headers.
+        #
+        # @return [Excon::Response]
+        #   * body [Hash] - Results of the operation.
+        #     * "Number Not Found" [Integer] - Number of missing objects or containers.
+        #     * "Response Status" [String] - Response code for the subrequest of the last failed operation.
+        #     * "Errors" [Array<object_name, response_status>]
+        #       * object_name [String] - Object that generated an error when the delete was attempted.
+        #       * response_status [String] - Response status from the subrequest for object_name.
+        #     * "Number Deleted" [Integer] - Number of objects or containers deleted.
+        #     * "Response Body" [String] - Response body for "Response Status".
+        def delete_multiple_objects(container, object_names, options = {})
+          body = object_names.map do |name|
+            object_name = container ? "#{container}/#{name}" : name
+            URI.encode(object_name)
+          end.join("\n")
+
+          response = request({
+                               :expects => 200,
+                               :method  => 'DELETE',
+                               :headers => options.merge('Content-Type' => 'text/plain',
+                                                         'Accept'       => 'application/json'),
+                               :body    => body,
+                               :query   => {'bulk-delete' => true}
+                             }, false)
+          response.body = Fog::JSON.decode(response.body)
+          response
+        end
+      end
+    end
+  end
+end