grid_attachment 0.0.2

data/README.txt

@@ -0,0 +1,85 @@
+Summary
+
+   GridAttachment is a GridFS plugin for MongoDB ORMs.
+   Supports MongoMapper, Mongoid, MongoODM, and Mongomatic.
+
+   Support is built in for rack_grid and rack_grid_thumb to generate URLS and thumbnails:
+     http://github.com/dusty/rack_grid
+     http://github.com/dusty/rack_grid_thumb
+
+   You can pass in a File or a Hash as received by Sinatra on file uploads.
+
+Installation
+
+  # gem install grid_attachment
+
+Usage
+
+  require 'grid_attachment/mongo_mapper'
+  class Monkey
+    include MongoMapper::Document
+    plugin GridAttachment::MongoMapper
+
+    attachment :image, :prefix => :grid
+  end
+
+  require 'grid_attachment/mongo_odm'
+  class Monkey
+    include MongoODM::Document
+    include GridAttachment::MongoODM
+
+    attachment :image, :prefix => :grid
+  end
+
+  require 'grid_attachment/mongomatic'
+  class Monkey < Mongomatic::Base
+    include GridAttachment::Mongomatic
+
+    attachment :image, :prefix => :grid
+  end
+
+  require 'grid_attachment/mongoid'
+  class Monkey
+    include Mongoid::Document
+    include GridAttachment::Mongoid
+
+    attachment :image, :prefix => :grid
+  end
+
+  m = Monkey.new(:name => 'name')
+  m.save
+
+  # To add an attachment from the filesystem
+  m.image = File.open('/tmp/me.jpg')
+  m.save
+
+  # To remove an attachment
+  m.image = nil
+  m.save
+
+  # To get the attachment
+  m.image.read
+
+  # To get the URL for rack_grid
+  m.image_url                          # /grid/4e049e7c69c3b27d53000005/me.jpg
+
+  # To get the thumbail URL for rack_grid_thumb
+  m.image_thumb('50x50')               # /grid/4e049e7c69c3b27d53000005/me_50x50.jpg
+
+  # HTML form example
+  <form action = "/monkeys" method="post" enctype="multipart/form-data">
+    <input id="image" name="image" type="file" />
+  </form>
+
+  # Use the image hash provided in params with Sinatra
+  post '/monkeys' do
+    m = Monkey.new
+    m.image = params[:image]
+    m.save
+    # Or just Monkey.new(params).save
+  end
+
+
+Inspired By
+  - http://github.com/jnunemaker/joint
+

data/lib/grid_attachment/mongo_mapper.rb

@@ -0,0 +1,238 @@
+require 'mime/types'
+require 'uri'
+
+module GridAttachment
+  module MongoMapper
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+
+      ##
+      # Declare an attachment for the object
+      #
+      # eg: attachment :image
+      def attachment(name,options={})
+        prefix = options[:prefix] ||= :grid
+
+        ##
+        # Callbacks to handle the attachment saving and deleting
+        after_save     :create_attachments
+        after_save     :delete_attachments
+        after_destroy  :queue_delete_attachments
+        after_destroy  :delete_attachments
+
+        ##
+        # Fields for the attachment.
+        #
+        # Only the _id is really needed, the others are helpful cached
+        # so you don't need to hit GridFS
+        key "#{name}_id".to_sym,   BSON::ObjectId
+        key "#{name}_name".to_sym, String
+        key "#{name}_size".to_sym, Integer
+        key "#{name}_type".to_sym, String
+
+        ##
+        # Add this name to the attachment_types
+        attachment_types.push(name).uniq!
+
+        ##
+        # Return the Grid object.
+        # eg: image.filename, image.read
+        define_method(name) do
+          grid.get(read_attribute("#{name}_id")) if read_attribute("#{name}_id")
+        end
+
+        ##
+        # Create a method to set the attachment
+        # eg: object.image = File.open('/tmp/somefile.jpg')
+        define_method("#{name}=") do |file|
+          # delete the old file if it exists
+          unless read_attribute("#{name}_id").blank?
+            send(:delete_attachment, name, read_attribute("#{name}_id"))
+          end
+          case
+          when file.is_a?(Hash) && file[:tempfile]
+            send(:create_attachment, name, file)
+          when file.respond_to?(:read)
+            send(:create_attachment, name, file)
+          end
+        end
+
+        ##
+        # Create a method to set the attachment for binary string.
+        # eg: object.set_image(binary_string, "generated_filename.png")
+        define_method("set_#{name}") do |binary, filename|
+          if !binary.nil?
+            send(:create_attachment_raw, name, binary, filename)
+          else
+            send(:delete_attachment, name, read_attribute("#{name}_id"))
+          end
+        end
+
+        ##
+        # Unset the attachment, queue for removal
+        define_method("unset_#{name}") do
+          send(:delete_attachment, name, read_attribute("#{name}_id"))
+        end
+
+        ##
+        # Return the relative URL to the attachment for use with Rack::Grid
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile.png
+        define_method("#{name}_url") do
+          _id   = read_attribute("#{name}_id")
+          _name = read_attribute("#{name}_name")
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+
+        ##
+        # Return the relative URL to the thumbnail for use with Rack::GridThumb
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile_50x.png
+        define_method("#{name}_thumb") do |thumb|
+          _id   = read_attribute("#{name}_id")
+          _name = read_attribute("#{name}_name")
+          _ext  = File.extname(_name)
+          _base = File.basename(_name,_ext)
+          _name = "#{_base}_#{thumb}#{_ext}"
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+      end
+
+      ##
+      # Accessor to Grid
+      def grid
+        @grid ||= Mongo::Grid.new(::MongoMapper.database)
+      end
+
+      ##
+      # All the attachments types for this class
+      def attachment_types
+        @attachment_types ||= []
+      end
+
+    end
+
+    module InstanceMethods
+
+      ##
+      # Accessor to Grid
+      def grid
+        self.class.grid
+      end
+
+      private
+      ##
+      # Holds queue of attachments to create
+      def create_attachment_queue
+        @create_attachment_queue ||= {}
+      end
+
+      ##
+      # Holds queue of attachments to delete
+      def delete_attachment_queue
+        @delete_attachment_queue ||= {}
+      end
+
+      ##
+      # Attachments we need to add after save.
+      def create_attachment(name,file)
+        case
+        when file.is_a?(Hash)
+          filename = file[:filename]
+          size     = File.size(file[:tempfile])
+          mime     = file[:type]
+          unless mime
+            type = MIME::Types.type_for(filename).first
+            mime = type ? type.content_type : "application/octet-stream"
+          end
+        when file.respond_to?(:read)
+          filename = case
+          when file.respond_to?(:original_filename) && file.original_filename
+            file.original_filename
+          when file.respond_to?(:tempfile)
+            File.basename(file.tempfile.path)
+          else
+            File.basename(file.path)
+          end
+          size = File.size(file.respond_to?(:tempfile) ? file.tempfile : file)
+          type = MIME::Types.type_for(filename).first
+          mime = type ? type.content_type : "application/octet-stream"
+        else
+          return
+        end
+        write_attribute("#{name}_id", BSON::ObjectId.new)
+        write_attribute("#{name}_name", filename)
+        write_attribute("#{name}_size", size)
+        write_attribute("#{name}_type", mime)
+        create_attachment_queue[name] = file
+      end
+
+      ##
+      # Attachments we need to add after save.
+      # For binary String data.
+      def create_attachment_raw(name, binary, filename)
+        type = MIME::Types.type_for(filename).first
+        mime = type ? type.content_type : "application/octet-stream"
+        write_attribute("#{name}_id", BSON::ObjectId.new)
+        write_attribute("#{name}_name", filename)
+        write_attribute("#{name}_size", binary.size)
+        write_attribute("#{name}_type", mime)
+        create_attachment_queue[name] = binary
+      end
+
+      ##
+      # Save an attachment to Grid
+      def create_grid_attachment(name,file)
+        data = case
+        when file.is_a?(Hash)
+          file[:tempfile].read
+        else
+          file.respond_to?(:read) ? file.read : file
+        end
+        grid.put(
+          data,
+          :filename => read_attribute("#{name}_name"),
+          :content_type => read_attribute("#{name}_type"),
+          :_id => read_attribute("#{name}_id")
+        )
+        create_attachment_queue.delete(name)
+      end
+
+      ##
+      # Attachments we need to remove after save
+      def delete_attachment(name,id)
+        delete_attachment_queue[name] = id if id.is_a?(BSON::ObjectId)
+        write_attribute("#{name}_id", nil)
+        write_attribute("#{name}_name", nil)
+        write_attribute("#{name}_size", nil)
+        write_attribute("#{name}_type", nil)
+      end
+
+      ##
+      # Delete an attachment from Grid
+      def delete_grid_attachment(name,id)
+        grid.delete(id) if id.is_a?(BSON::ObjectId)
+        delete_attachment_queue.delete(name)
+      end
+
+      ##
+      # Create attachments marked for creation
+      def create_attachments
+        create_attachment_queue.each {|k,v| create_grid_attachment(k,v)}
+      end
+
+      ##
+      # Delete attachments marked for deletion
+      def delete_attachments
+        delete_attachment_queue.each {|k,v| delete_grid_attachment(k,v)}
+      end
+
+      ##
+      # Queues all attachments for deletion
+      def queue_delete_attachments
+        self.class.attachment_types.each do |name|
+          delete_attachment(name, read_attribute("#{name}_id"))
+        end
+      end
+    end
+  end
+end

data/lib/grid_attachment/mongo_odm.rb

@@ -0,0 +1,242 @@
+require 'mime/types'
+require 'uri'
+
+module GridAttachment
+  module MongoODM
+
+    def self.included(base)
+      base.send(:include, InstanceMethods)
+      base.send(:extend, ClassMethods)
+    end
+
+    module ClassMethods
+
+      ##
+      # Declare an attachment for the object
+      #
+      # eg: attachment :image
+      def attachment(name,options={})
+        prefix = options[:prefix] ||= :grid
+
+        ##
+        # Callbacks to handle the attachment saving and deleting
+        after_save     :create_attachments
+        after_save     :delete_attachments
+        after_destroy  :queue_delete_attachments
+        after_destroy  :delete_attachments
+
+        ##
+        # Fields for the attachment.
+        #
+        # Only the _id is really needed, the others are helpful cached
+        # so you don't need to hit GridFS
+        field "#{name}_id".to_sym,   BSON::ObjectId
+        field "#{name}_name".to_sym, String
+        field "#{name}_size".to_sym, Integer
+        field "#{name}_type".to_sym, String
+
+        ##
+        # Add this name to the attachment_types
+        attachment_types.push(name).uniq!
+
+        ##
+        # Return the Grid object.
+        # eg: image.filename, image.read
+        define_method(name) do
+          grid.get(read_attribute("#{name}_id")) if read_attribute("#{name}_id")
+        end
+
+        ##
+        # Create a method to set the attachment
+        # eg: object.image = File.open('/tmp/somefile.jpg')
+        define_method("#{name}=") do |file|
+          # delete the old file if it exists
+          unless read_attribute("#{name}_id").blank?
+            send(:delete_attachment, name, read_attribute("#{name}_id"))
+          end
+          case
+          when file.is_a?(Hash) && file[:tempfile]
+            send(:create_attachment, name, file)
+          when file.respond_to?(:read)
+            send(:create_attachment, name, file)
+          end
+        end
+
+        ##
+        # Create a method to set the attachment for binary string.
+        # eg: object.set_image(binary_string, "generated_filename.png")
+        define_method("set_#{name}") do |binary, filename|
+          if !binary.nil?
+            send(:create_attachment_raw, name, binary, filename)
+          else
+            send(:delete_attachment, name, read_attribute("#{name}_id"))
+          end
+        end
+
+        ##
+        # Unset the attachment, queue for removal
+        define_method("unset_#{name}") do
+          send(:delete_attachment, name, read_attribute("#{name}_id"))
+        end
+
+        ##
+        # Return the relative URL to the attachment for use with Rack::Grid
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile.png
+        define_method("#{name}_url") do
+          _id   = read_attribute("#{name}_id")
+          _name = read_attribute("#{name}_name")
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+
+        ##
+        # Return the relative URL to the thumbnail for use with Rack::GridThumb
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile_50x.png
+        define_method("#{name}_thumb") do |thumb|
+          _id   = read_attribute("#{name}_id")
+          _name = read_attribute("#{name}_name")
+          _ext  = File.extname(_name)
+          _base = File.basename(_name,_ext)
+          _name = "#{_base}_#{thumb}#{_ext}"
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+      end
+
+      ##
+      # Accessor to Grid
+      def grid
+        @grid ||= Mongo::Grid.new(::MongoODM.database)
+      end
+
+      ##
+      # All the attachments types for this class
+      def attachment_types
+        @attachment_types ||= []
+      end
+
+    end
+
+    module InstanceMethods
+
+      ##
+      # Accessor to Grid
+      def grid
+        self.class.grid
+      end
+
+      private
+      ##
+      # Holds queue of attachments to create
+      def create_attachment_queue
+        @create_attachment_queue ||= {}
+      end
+
+      ##
+      # Holds queue of attachments to delete
+      def delete_attachment_queue
+        @delete_attachment_queue ||= {}
+      end
+
+      ##
+      # Attachments we need to add after save.
+      def create_attachment(name,file)
+        case
+        when file.is_a?(Hash)
+          filename = file[:filename]
+          size     = File.size(file[:tempfile])
+          mime     = file[:type]
+          unless mime
+            type = MIME::Types.type_for(filename).first
+            mime = type ? type.content_type : "application/octet-stream"
+          end
+        when file.respond_to?(:read)
+          filename = case
+          when file.respond_to?(:original_filename) && file.original_filename
+            file.original_filename
+          when file.respond_to?(:tempfile)
+            File.basename(file.tempfile.path)
+          else
+            File.basename(file.path)
+          end
+          size = File.size(file.respond_to?(:tempfile) ? file.tempfile : file)
+          type = MIME::Types.type_for(filename).first
+          mime = type ? type.content_type : "application/octet-stream"
+        else
+          return
+        end
+        write_attribute("#{name}_id", BSON::ObjectId.new)
+        write_attribute("#{name}_name", filename)
+        write_attribute("#{name}_size", size)
+        write_attribute("#{name}_type", mime)
+        create_attachment_queue[name] = file
+      end
+
+      ##
+      # Attachments we need to add after save.
+      # For binary String data.
+      def create_attachment_raw(name, binary, filename)
+        type = MIME::Types.type_for(filename).first
+        mime = type ? type.content_type : "application/octet-stream"
+        write_attribute("#{name}_id", BSON::ObjectId.new)
+        write_attribute("#{name}_name", filename)
+        write_attribute("#{name}_size", binary.size)
+        write_attribute("#{name}_type", mime)
+        create_attachment_queue[name] = binary
+      end
+
+      ##
+      # Save an attachment to Grid
+      def create_grid_attachment(name,file)
+        data = case
+        when file.is_a?(Hash)
+          file[:tempfile].read
+        else
+          file.respond_to?(:read) ? file.read : file
+        end
+        grid.put(
+          data,
+          :filename => read_attribute("#{name}_name"),
+          :content_type => read_attribute("#{name}_type"),
+          :_id => read_attribute("#{name}_id")
+        )
+        create_attachment_queue.delete(name)
+      end
+
+      ##
+      # Attachments we need to remove after save
+      def delete_attachment(name,id)
+        delete_attachment_queue[name] = id if id.is_a?(BSON::ObjectId)
+        write_attribute("#{name}_id", nil)
+        write_attribute("#{name}_name", nil)
+        write_attribute("#{name}_size", nil)
+        write_attribute("#{name}_type", nil)
+      end
+
+      ##
+      # Delete an attachment from Grid
+      def delete_grid_attachment(name,id)
+        grid.delete(id) if id.is_a?(BSON::ObjectId)
+        delete_attachment_queue.delete(name)
+      end
+
+      ##
+      # Create attachments marked for creation
+      def create_attachments
+        create_attachment_queue.each {|k,v| create_grid_attachment(k,v)}
+      end
+
+      ##
+      # Delete attachments marked for deletion
+      def delete_attachments
+        delete_attachment_queue.each {|k,v| delete_grid_attachment(k,v)}
+      end
+
+      ##
+      # Queues all attachments for deletion
+      def queue_delete_attachments
+        self.class.attachment_types.each do |name|
+          delete_attachment(name, read_attribute("#{name}_id"))
+        end
+      end
+    end
+  end
+end

data/lib/grid_attachment/mongoid.rb

@@ -0,0 +1,238 @@
+require 'mime/types'
+require 'uri'
+
+module GridAttachment
+  module Mongoid
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+
+      ##
+      # Declare an attachment for the object
+      #
+      # eg: attachment :image
+      def attachment(name,options={})
+        prefix = options[:prefix] ||= :grid
+
+        ##
+        # Callbacks to handle the attachment saving and deleting
+        after_save     :create_attachments
+        after_save     :delete_attachments
+        after_destroy  :queue_delete_attachments
+        after_destroy  :delete_attachments
+
+        ##
+        # Fields for the attachment.
+        #
+        # Only the _id is really needed, the others are helpful cached
+        # so you don't need to hit GridFS
+        field "#{name}_id".to_sym,   :type => BSON::ObjectId
+        field "#{name}_name".to_sym, :type => String
+        field "#{name}_size".to_sym, :type => Integer
+        field "#{name}_type".to_sym, :type => String
+
+        ##
+        # Add this name to the attachment_types
+        attachment_types.push(name).uniq!
+
+        ##
+        # Return the Grid object.
+        # eg: image.filename, image.read
+        define_method(name) do
+          grid.get(read_attribute("#{name}_id")) if read_attribute("#{name}_id")
+        end
+
+        ##
+        # Create a method to set the attachment
+        # eg: object.image = File.open('/tmp/somefile.jpg')
+        define_method("#{name}=") do |file|
+          # delete the old file if it exists
+          unless read_attribute("#{name}_id").blank?
+            send(:delete_attachment, name, read_attribute("#{name}_id"))
+          end
+          case
+          when file.is_a?(Hash) && file[:tempfile]
+            send(:create_attachment, name, file)
+          when file.respond_to?(:read)
+            send(:create_attachment, name, file)
+          end
+        end
+
+        ##
+        # Create a method to set the attachment for binary string.
+        # eg: object.set_image(binary_string, "generated_filename.png")
+        define_method("set_#{name}") do |binary, filename|
+          if !binary.nil?
+            send(:create_attachment_raw, name, binary, filename)
+          else
+            send(:delete_attachment, name, read_attribute("#{name}_id"))
+          end
+        end
+
+        ##
+        # Unset the attachment, queue for removal
+        define_method("unset_#{name}") do
+          send(:delete_attachment, name, read_attribute("#{name}_id"))
+        end
+
+        ##
+        # Return the relative URL to the attachment for use with Rack::Grid
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile.png
+        define_method("#{name}_url") do
+          _id   = read_attribute("#{name}_id")
+          _name = read_attribute("#{name}_name")
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+
+        ##
+        # Return the relative URL to the thumbnail for use with Rack::GridThumb
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile_50x.png
+        define_method("#{name}_thumb") do |thumb|
+          _id   = read_attribute("#{name}_id")
+          _name = read_attribute("#{name}_name")
+          _ext  = File.extname(_name)
+          _base = File.basename(_name,_ext)
+          _name = "#{_base}_#{thumb}#{_ext}"
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+      end
+
+      ##
+      # Accessor to Grid
+      def grid
+        @grid ||= Mongo::Grid.new(::Mongoid.database)
+      end
+
+      ##
+      # All the attachments types for this class
+      def attachment_types
+        @attachment_types ||= []
+      end
+
+    end
+
+    module InstanceMethods
+
+      ##
+      # Accessor to Grid
+      def grid
+        self.class.grid
+      end
+
+      private
+      ##
+      # Holds queue of attachments to create
+      def create_attachment_queue
+        @create_attachment_queue ||= {}
+      end
+
+      ##
+      # Holds queue of attachments to delete
+      def delete_attachment_queue
+        @delete_attachment_queue ||= {}
+      end
+
+      ##
+      # Attachments we need to add after save.
+      def create_attachment(name,file)
+        case
+        when file.is_a?(Hash)
+          filename = file[:filename]
+          size     = File.size(file[:tempfile])
+          mime     = file[:type]
+          unless mime
+            type = MIME::Types.type_for(filename).first
+            mime = type ? type.content_type : "application/octet-stream"
+          end
+        when file.respond_to?(:read)
+          filename = case
+          when file.respond_to?(:original_filename) && file.original_filename
+            file.original_filename
+          when file.respond_to?(:tempfile)
+            File.basename(file.tempfile.path)
+          else
+            File.basename(file.path)
+          end
+          size = File.size(file.respond_to?(:tempfile) ? file.tempfile : file)
+          type = MIME::Types.type_for(filename).first
+          mime = type ? type.content_type : "application/octet-stream"
+        else
+          return
+        end
+        write_attribute("#{name}_id", BSON::ObjectId.new)
+        write_attribute("#{name}_name", filename)
+        write_attribute("#{name}_size", size)
+        write_attribute("#{name}_type", mime)
+        create_attachment_queue[name] = file
+      end
+
+      ##
+      # Attachments we need to add after save.
+      # For binary String data.
+      def create_attachment_raw(name, binary, filename)
+        type = MIME::Types.type_for(filename).first
+        mime = type ? type.content_type : "application/octet-stream"
+        write_attribute("#{name}_id", BSON::ObjectId.new)
+        write_attribute("#{name}_name", filename)
+        write_attribute("#{name}_size", binary.size)
+        write_attribute("#{name}_type", mime)
+        create_attachment_queue[name] = binary
+      end
+
+      ##
+      # Save an attachment to Grid
+      def create_grid_attachment(name,file)
+        data = case
+        when file.is_a?(Hash)
+          file[:tempfile].read
+        else
+          file.respond_to?(:read) ? file.read : file
+        end
+        grid.put(
+          data,
+          :filename => read_attribute("#{name}_name"),
+          :content_type => read_attribute("#{name}_type"),
+          :_id => read_attribute("#{name}_id")
+        )
+        create_attachment_queue.delete(name)
+      end
+
+      ##
+      # Attachments we need to remove after save
+      def delete_attachment(name,id)
+        delete_attachment_queue[name] = id if id.is_a?(BSON::ObjectId)
+        write_attribute("#{name}_id", nil)
+        write_attribute("#{name}_name", nil)
+        write_attribute("#{name}_size", nil)
+        write_attribute("#{name}_type", nil)
+      end
+
+      ##
+      # Delete an attachment from Grid
+      def delete_grid_attachment(name,id)
+        grid.delete(id) if id.is_a?(BSON::ObjectId)
+        delete_attachment_queue.delete(name)
+      end
+
+      ##
+      # Create attachments marked for creation
+      def create_attachments
+        create_attachment_queue.each {|k,v| create_grid_attachment(k,v)}
+      end
+
+      ##
+      # Delete attachments marked for deletion
+      def delete_attachments
+        delete_attachment_queue.each {|k,v| delete_grid_attachment(k,v)}
+      end
+
+      ##
+      # Queues all attachments for deletion
+      def queue_delete_attachments
+        self.class.attachment_types.each do |name|
+          delete_attachment(name, read_attribute("#{name}_id"))
+        end
+      end
+    end
+  end
+end

data/lib/grid_attachment/mongomatic.rb

@@ -0,0 +1,238 @@
+require 'mime/types'
+require 'uri'
+
+class GridmaticObserver < Mongomatic::Observer
+  def after_insert_or_update(instance, opts)
+    instance.send(:create_attachments)
+    instance.send(:delete_attachments)
+  end
+  def before_remove(instance,opts)
+    instance.send(:queue_delete_attachments)
+    instance.send(:delete_attachments)
+  end
+end
+
+module GridAttachment
+  module Mongomatic
+
+    def self.included(base)
+      base.send(:include, Mongomatic::Observable)
+      base.send(:observer, :GridmaticObserver)
+      base.send(:include, InstanceMethods)
+      base.send(:extend, ClassMethods)
+    end
+
+    module ClassMethods
+
+      ##
+      # Declare an attachment for the object
+      #
+      # eg: attachment :image
+      def attachment(name,options={})
+        prefix = options[:prefix] ||= :grid
+
+        ##
+        # Add this name to the attachment_types
+        attachment_types.push(name).uniq!
+
+        ##
+        # Return the Grid object.
+        # eg: image.filename, image.read
+        define_method(name) do
+          grid.get(self["#{name}_id"]) if self["#{name}_id"]
+        end
+
+        ##
+        # Create a method to set the attachment
+        # eg: object.image = File.open('/tmp/somefile.jpg')
+        define_method("#{name}=") do |file|
+          # delete the old file if it exists
+          unless self["#{name}_id"].blank?
+            send(:delete_attachment, name, self["#{name}_id"])
+          end
+          case
+          when file.is_a?(Hash) && file[:tempfile]
+            send(:create_attachment, name, file)
+          when file.respond_to?(:read)
+            send(:create_attachment, name, file)
+          end
+        end
+
+        ##
+        # Create a method to set the attachment for binary string.
+        # eg: object.set_image(binary_string, "generated_filename.png")
+        define_method("set_#{name}") do |binary, filename|
+          if !binary.nil?
+            send(:create_attachment_raw, name, binary, filename)
+          else
+            send(:delete_attachment, name, self["#{name}_id"])
+          end
+        end
+
+        ##
+        # Unset the attachment, queue for removal
+        define_method("unset_#{name}") do
+          send(:delete_attachment, name, self["#{name}_id"])
+        end
+
+        ##
+        # Return the relative URL to the attachment for use with Rack::Grid
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile.png
+        define_method("#{name}_url") do
+          _id   = self["#{name}_id"]
+          _name = self["#{name}_name"]
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+
+        ##
+        # Return the relative URL to the thumbnail for use with Rack::GridThumb
+        # eg: /grid/4ba69fde8c8f369a6e000003/somefile_50x.png
+        define_method("#{name}_thumb") do |thumb|
+          _id   = self["#{name}_id"]
+          _name = self["#{name}_name"]
+          _ext  = File.extname(_name)
+          _base = File.basename(_name,_ext)
+          _name = "#{_base}_#{thumb}#{_ext}"
+          URI.escape(["/#{prefix}", _id, _name].join('/')) if _id && _name
+        end
+      end
+
+      ##
+      # Accessor to Grid
+      def grid
+        @grid ||= Mongo::Grid.new(::Mongomatic.db)
+      end
+
+      ##
+      # All the attachments types for this class
+      def attachment_types
+        @attachment_types ||= []
+      end
+
+    end
+
+    module InstanceMethods
+
+      ##
+      # Accessor to Grid
+      def grid
+        self.class.grid
+      end
+
+      private
+      ##
+      # Holds queue of attachments to create
+      def create_attachment_queue
+        @create_attachment_queue ||= {}
+      end
+
+      ##
+      # Holds queue of attachments to delete
+      def delete_attachment_queue
+        @delete_attachment_queue ||= {}
+      end
+
+      ##
+      # Attachments we need to add after save.
+      def create_attachment(name,file)
+        case
+        when file.is_a?(Hash)
+          filename = file[:filename]
+          size     = File.size(file[:tempfile])
+          mime     = file[:type]
+          unless mime
+            type = MIME::Types.type_for(filename).first
+            mime = type ? type.content_type : "application/octet-stream"
+          end
+        when file.respond_to?(:read)
+          filename = case
+          when file.respond_to?(:original_filename) && file.original_filename
+            file.original_filename
+          when file.respond_to?(:tempfile)
+            File.basename(file.tempfile.path)
+          else
+            File.basename(file.path)
+          end
+          size = File.size(file.respond_to?(:tempfile) ? file.tempfile : file)
+          type = MIME::Types.type_for(filename).first
+          mime = type ? type.content_type : "application/octet-stream"
+        else
+          return
+        end
+        self["#{name}_id"]   = BSON::ObjectId.new
+        self["#{name}_name"] = filename
+        self["#{name}_size"] = size
+        self["#{name}_type"] = mime
+        create_attachment_queue[name] = file
+      end
+
+      ##
+      # Attachments we need to add after save.
+      # For binary String data.
+      def create_attachment_raw(name, binary, filename)
+        type = MIME::Types.type_for(filename).first
+        mime = type ? type.content_type : "application/octet-stream"
+        self["#{name}_id"]   = BSON::ObjectId.new
+        self["#{name}_name"] = filename
+        self["#{name}_size"] = binary.size
+        self["#{name}_type"] =  mime
+        create_attachment_queue[name] = binary
+      end
+
+      ##
+      # Save an attachment to Grid
+      def create_grid_attachment(name,file)
+        data = case
+        when file.is_a?(Hash)
+          file[:tempfile].read
+        else
+          file.respond_to?(:read) ? file.read : file
+        end
+        grid.put(
+          data,
+          :filename => self["#{name}_name"],
+          :content_type => self["#{name}_type"],
+          :_id => self["#{name}_id"]
+        )
+        create_attachment_queue.delete(name)
+      end
+
+      ##
+      # Attachments we need to remove after save
+      def delete_attachment(name,id)
+        delete_attachment_queue[name] = id if id.is_a?(BSON::ObjectId)
+        self["#{name}_id"]   = nil
+        self["#{name}_name"] = nil
+        self["#{name}_size"] = nil
+        self["#{name}_type"] = nil
+      end
+
+      ##
+      # Delete an attachment from Grid
+      def delete_grid_attachment(name,id)
+        grid.delete(id) if id.is_a?(BSON::ObjectId)
+        delete_attachment_queue.delete(name)
+      end
+
+      ##
+      # Create attachments marked for creation
+      def create_attachments
+        create_attachment_queue.each {|k,v| create_grid_attachment(k,v)}
+      end
+
+      ##
+      # Delete attachments marked for deletion
+      def delete_attachments
+        delete_attachment_queue.each {|k,v| delete_grid_attachment(k,v)}
+      end
+
+      ##
+      # Queues all attachments for deletion
+      def queue_delete_attachments
+        self.class.attachment_types.each do |name|
+          delete_attachment(name, self["#{name}_id"])
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: grid_attachment
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Dusty Doris
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-08-15 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mime-types
+  requirement: &70132069007560 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70132069007560
+description: Plugin for various Mongo ODMs to attach files via GridFS
+email: github@dusty.name
+executables: []
+extensions: []
+extra_rdoc_files:
+- README.txt
+files:
+- README.txt
+- lib/grid_attachment/mongo_mapper.rb
+- lib/grid_attachment/mongo_odm.rb
+- lib/grid_attachment/mongomatic.rb
+- lib/grid_attachment/mongoid.rb
+- test/test_mongo_mapper.rb
+- test/test_mongo_odm.rb
+- test/test_mongomatic.rb
+- test/test_mongoid.rb
+homepage: http://github.com/dusty/grid_attachment
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: none
+rubygems_version: 1.8.7
+signing_key: 
+specification_version: 3
+summary: Plugin for various Mongo ODMs to attach files via GridFS
+test_files: []