require 'fog/core/collection'
require 'fog/dtdream/models/storage/file'

module Fog
  module Storage
    class Dtdream
      class Files < Fog::Collection
        attribute :directory
        attribute :limit
        attribute :marker
        attribute :path
        attribute :prefix

        model Fog::Storage::Dtdream::File

        def all(options = {})
          requires :directory
          if directory.key != "" && directory.key != "." && directory.key != nil
            prefix = directory.key+"/"
          end
          files = service.list_objects({:prefix => prefix})["Contents"]
          data = Array.new
          i = 0
          files.each do |file|
            if file["Key"][0][-1] != "/"
              key = file["Key"][0]
              data[i] = {:key=>key}
              i = i + 1
            end
          end

          #load(data)
          data
          
        end

        alias_method :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
          if directory.key == ""
            object = key
          else
            object = directory.key+"/"+key
          end
          
          data = service.head_object(object).data
          contentLen = data[:headers]["Content-Length"].to_i
          if data[:status] != 200 || contentLen <= 0
            return nil
          end
          
          file_data = {
              :content_length => contentLen,
              :key            => key,
              :last_modified  => data[:headers]["Last-Modified"]
          }
          
          if block_given?
            pagesNum = (contentLen + Excon::CHUNK_SIZE - 1)/Excon::CHUNK_SIZE
            
            for i in 1..pagesNum
              _start = (i-1)*(Excon::CHUNK_SIZE)
              _end = i*(Excon::CHUNK_SIZE) - 1
              range = "#{_start}-#{_end}"
              chunk = service.get_object(object, range) && yield(chunk)
            end
            new(file_data)
          else
            data = service.get_object(object)
            file_data.merge!(:body => data[:body])
            new(file_data)
          end
          
        end

        def get_url(key)
          requires :directory
          if directory.key == ""
            object = key
          else
            object = directory.key+"/"+key
          end
          service.get_object_http_url_public(object, 3600)
        end

        def get_http_url(key, expires, options = {})
          requires :directory
          if directory.key == ""
            object = key
          else
            object = directory.key+"/"+key
          end
          service.get_object_http_url_public(object, expires, options)
        end

        def get_https_url(key, expires, options = {})
          requires :directory
          if directory.key == ""
            object = key
          else
            object = directory.key+"/"+key
          end
          service.get_object_https_url_public(object, expires, options)
        end

        def head(key, options = {})
          requires :directory
          if directory.key == ""
            object = key
          else
            object = directory.key+"/"+key
          end
          data = service.head_object(object).data
          file_data = data[:headers].merge({
            :key => key
          })
          new(file_data)
        rescue Fog::Storage::Dtdream::NotFound
          nil
        end

        def new(attributes = {})
          requires :directory
          super({ :directory => directory }.merge!(attributes))
        end
      end
    end
  end
end