Contents

module MongoidExt
  class FileList < EmbeddedHash
    attr_accessor :parent_document
    attr_accessor :list_name

    def put(id, io, metadata = {})
      mark_parent!

      if !parent_document.new_record?
        filename = id
        if io.respond_to?(:original_filename)
          filename = io.original_filename
        elsif io.respond_to?(:path) && io.path
          filename = ::File.basename(io.path)
        elsif io.kind_of?(String)
          io = StringIO.new(io)
        end

        get(id).put(filename, io, metadata)
      else
        (@_pending_files ||= {})[id] = [io, metadata]
      end
    end

    def files
      ids = self.keys
      ids.delete("_id")
      ids.map {|v| get(v) }
    end

    def each_file(&block)
      (self.keys-["_id"]).each do |key|
        file = self.get(key)
        yield key, file
      end
    end

    def get(id)
      mark_parent!

      if id.kind_of?(MongoidExt::File)
        self[id.id] = id
        return id
      end

      id = id.to_s.gsub(".", "_")
      file = self.fetch(id, nil)

      if file.nil?
        file = self[id] = MongoidExt::File.new
      elsif file.class == ::Hash || file.class == BSON::OrderedHash
        file = self[id] = MongoidExt::File.new(file)
      end

      file._root_document = parent_document
      file._list_name = self.list_name
      file
    end

    def sync_files
      if @_pending_files
        @_pending_files.each do |filename, data|
          put(filename, data[0], data[1])
        end
        @_pending_files = nil
      end
    end

    def delete(id)
      mark_parent!

      file = self.get(id)
      super(id)
      file.delete
    end

    def destroy_files
      each_file do |id, file|
        get(id).delete
      end
    end

    def serialize(v)
      v
    end

    def deserialize(v)
      doc = self.class.new
      v.each do |k,v|
        doc[k] = MongoidExt::File.new(v)
      end

      doc
    end

    def mark_parent!
      parent_document.send("#{list_name}_will_change!")
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
mongoid_ext-0.7.1	lib/mongoid_ext/file_list.rb