fog--fog/lib/fog/aws/models/s3/directories.rb

module Fog
  module AWS
    class S3

      def directories
        Fog::AWS::S3::Directories.new(:connection => self)
      end

      class Directories < Fog::Collection

        model Fog::AWS::S3::Directory

        def all
          if @loaded
            clear
          end
          @loaded = true
          data = connection.get_service.body
          data['Buckets'].each do |directory|
            self << new(directory)
          end
          self
        end

        def get(name, options = {})
          remap_attributes(options, {
            :delimiter  => 'delimiter',
            :marker     => 'marker',
            :max_keys   => 'max-keys',
            :prefix     => 'prefix'
          })
          data = connection.get_bucket(name, options).body
          directory = new(:name => data['Name'])
          options = {}
          for key, value in data
            if ['Delimiter', 'IsTruncated', 'Marker', 'MaxKeys', 'Prefix'].include?(key)
              options[key] = value
            end
          end
          directory.files.merge_attributes(options)
          directory.files.instance_variable_set(:@loaded, true)
          data['Contents'].each do |file|
            directory.files << directory.files.new(file)
          end
          directory
        rescue Excon::Errors::NotFound
          nil
        end

      end

    end
  end
end