campbellhay-mongo 1.0.14

data/.gitignore

@@ -0,0 +1,3 @@
+Gemfile.lock
+*.gem
+

data/Gemfile

@@ -0,0 +1,2 @@
+source :rubygems
+gemspec

data/campbellhay-mongo.gemspec

@@ -0,0 +1,14 @@
+Gem::Specification.new do |s| 
+  s.name = 'campbellhay-mongo'
+  s.version = "1.0.14"
+  s.platform = Gem::Platform::RUBY
+  s.summary = "The CampbellHay Mongo toolbox"
+  s.description = "The CampbellHay Mongo toolbox"
+  s.files = `git ls-files`.split("\n").sort
+  s.require_path = './lib'
+  s.author = "Mickael Riga"
+  s.email = "mig@campbellhay.com"
+  s.homepage = "http://www.campbellhay.com"
+  s.add_dependency('mongo', '>=1.8.2')
+  s.add_development_dependency('bacon')
+end

data/gempush

@@ -0,0 +1,7 @@
+#!/bin/sh
+mv *.gem "/Users/mig/web_apps/CHA_gems/public/gems"
+cd "/Users/mig/web_apps/CHA_gems"
+gem generate_index -d './public'
+git add .
+git commit -am 'Update gems'
+git push heroku master

data/lib/mongo_crushyform.rb

@@ -0,0 +1,221 @@
+# encoding: utf-8
+
+require 'mongo_mutation'
+
+module MongoCrushyform
+  
+  module ClassMethods
+    
+    def crushyform_types
+      @crushyform_types ||= {
+        :none => proc{''},
+        :string => proc do |m,c,o|
+          if o[:autocompleted]
+            values = o[:autocomplete_options] || m.class.collection.distinct(c)
+            js = <<-EOJS
+            <script type="text/javascript" charset="utf-8">
+              $(function(){
+                $( "##{m.field_id_for(c)}" ).autocomplete({source: ["#{values.join('","')}"]});
+              });
+            </script>
+            EOJS
+          end
+          tag = "<input type='%s' name='%s' value=\"%s\" id='%s' class='%s' %s />%s\n" % [o[:input_type]||'text', o[:input_name], o[:input_value], m.field_id_for(c), o[:input_class], o[:required]&&'required', o[:required]]
+          "#{tag}#{js}"
+        end,
+        :slug => proc do |m,c,o|
+          crushyform_types[:string].call(m,c,o)
+        end,
+        :boolean => proc do |m,c,o|
+          crushid = m.field_id_for(c)
+          s = ['checked', nil]
+          s.reverse! unless o[:input_value]
+          out = "<span class='%s'>"
+          out += "<input type='radio' name='%s' value='true' id='%s' %s /> <label for='%s'>Yes</label> "
+          out += "<input type='radio' name='%s' value='false' id='%s-no' %s /> <label for='%s-no'>No</label>"
+          out += "</span>\n"
+          out % [o[:input_class], o[:input_name], crushid, s[0], crushid, o[:input_name], crushid, s[1], crushid]
+        end,
+        :text => proc do |m,c,o|
+          "<textarea name='%s' id='%s' class='%s' %s>%s</textarea>%s\n" % [o[:input_name], m.field_id_for(c), o[:input_class], o[:required]&&'required', o[:input_value], o[:required]]
+        end,
+        :date => proc do |m,c,o|
+          o[:input_value] = o[:input_value].strftime("%Y-%m-%d") if o[:input_value].respond_to?(:strftime)
+          o[:required] = "%s Format: yyyy-mm-dd" % [o[:required]]
+          crushyform_types[:string].call(m,c,o)
+        end,
+        :time => proc do |m,c,o|
+          o[:input_value] = o[:input_value].strftime("%T") if o[:input_value].respond_to?(:strftime)
+          o[:required] = "%s Format: hh:mm:ss" % [o[:required]]
+          crushyform_types[:string].call(m,c,o)
+        end,
+        :datetime => proc do |m,c,o|
+          o[:input_value] = o[:input_value].strftime("%Y-%m-%d %T") if o[:input_value].respond_to?(:strftime)
+          o[:required] = "%s Format: yyyy-mm-dd hh:mm:ss" % [o[:required]]
+          crushyform_types[:string].call(m,c,o)
+        end,
+        :parent => proc do |m,c,o|
+          parent_class = o[:parent_class].nil? ? Kernel.const_get(c.sub(/^id_/, '')) : m.resolve_class(o[:parent_class])
+          option_list = parent_class.to_dropdown(o[:input_value])
+          "<select name='%s' id='%s' class='%s'>%s</select>\n" % [o[:input_name], m.field_id_for(c), o[:input_class], option_list]
+        end,
+        :children => proc do |m,c,o|
+          children_class = o[:children_class].nil? ? Kernel.const_get(c.sub(/^ids_/, '')) : m.resolve_class(o[:children_class])
+          opts = o.update({
+            :multiple=>true,
+            :select_options=>children_class.dropdown_cache
+          })
+          @crushyform_types[:select].call(m,c,opts)
+        end,
+        :attachment => proc do |m,c,o|
+				  deleter = "<input type='checkbox' name='#{o[:input_name]}' class='deleter' value='nil' /> Delete this file<br />" unless m.doc[c].nil?
+          "%s%s<input type='file' name='%s' id='%s' class='%s' />%s\n" % [m.to_thumb(c), deleter, o[:input_name], m.field_id_for(c), o[:input_class], o[:required]]
+        end,
+        :select => proc do |m,c,o|
+          out = "<select name='%s%s' id='%s' class='%s' %s title='-- Select --'>\n" % [o[:input_name], ('[]' if o[:multiple]), m.field_id_for(c), o[:input_class], ('multiple' if o[:multiple])]
+          o[:select_options] = m.__send__(o[:select_options]) unless o[:select_options].kind_of?(Array)
+          select_options = o[:select_options].dup
+          if (o[:multiple] && !o[:input_value].nil? && o[:input_value].size>1)
+            o[:input_value].reverse.each do |v|
+              elem = select_options.find{|x| x==v||(x||[])[1]==v }
+              select_options.unshift(select_options.delete(elem)) unless elem.nil?
+            end
+          end
+          if select_options.kind_of?(Array)
+            select_options.each do |op|
+              key,val = op.kind_of?(Array) ? [op[0],op[1]] : [op,op]
+              if key==:optgroup
+                out << "<optgroup label='%s'>\n" % [val]
+              elsif key==:closegroup
+                out << "</optgroup>\n"
+              else
+                selected = 'selected' if (val==o[:input_value] || (o[:input_value]||[]).include?(val))
+                out << "<option value='%s' %s>%s</option>\n" % [val,selected,key]
+              end
+            end
+          end
+          out << "</select>%s\n" % [o[:required]]
+        end,
+				:string_list => proc do |m,c,o|
+				  if o[:autocompleted]
+				    values = o[:autocomplete_options] || m.class.collection.distinct(c)
+				    js = <<-EOJS
+            <script type="text/javascript" charset="utf-8">
+              $(function(){
+                $( "##{m.field_id_for(c)}" )
+                .bind( "keydown", function( event ) {
+                	if ( event.keyCode === $.ui.keyCode.TAB &&
+                	$( this ).data( "autocomplete" ).menu.active ) {
+                		event.preventDefault();
+                	}
+                })
+                .autocomplete({
+                	minLength: 0,
+                	source: function( request, response ) {
+                		response($.ui.autocomplete.filter(["#{values.join('","')}"], request.term.split(/,\s*/).pop()));
+                	},
+                	focus: function() { return false; },
+                	select: function( event, ui ) {
+                		var terms = this.value.split(/,\s*/);
+                		terms.pop();
+                		terms.push(ui.item.value);
+                		terms.push("");
+                		this.value = terms.join( ", " );
+                		return false;
+                	}
+                });
+              });
+            </script>
+            EOJS
+            o[:autocompleted] = false # reset so that it does not autocomplete for :string type below
+			    end
+				  tag = @crushyform_types[:string].call(m,c,o.update({:input_value=>(o[:input_value]||[]).join(',')}))
+				  "#{tag}#{js}"
+				end,
+				:permalink => proc do |instance, column_name, options|
+          values = "<option value=''>Or Browse the list</option>\n"
+          tag = @crushyform_types[:string].call(instance, column_name, options)
+          return tag if options[:permalink_classes].nil?
+          options[:permalink_classes].each do |sym|
+            c = Kernel.const_get sym
+            entries = c.find
+            unless entries.count==0
+              values << "<optgroup label='#{c.human_name}'>\n"
+              entries.each do |e|
+                values << "<option value='#{e.permalink}' #{'selected' if e.permalink==options[:input_value]}>#{e.to_label}</option>\n"
+              end
+              values << "</optgroup>\n"
+            end
+          end
+          "#{tag}<br />\n<select name='__permalink' class='permalink-dropdown'>\n#{values}</select>\n"
+        end
+      }
+    end
+    
+    # What represents a required field
+    # Can be overriden
+    def crushyfield_required; "<span class='crushyfield-required'> *</span>"; end
+    # Stolen from ERB
+    def html_escape(s)
+      s.to_s.gsub(/&/, "&amp;").gsub(/\"/, "&quot;").gsub(/>/, "&gt;").gsub(/</, "&lt;")
+    end
+    # Cache dropdown options for children classes to use  
+    # Meant to be reseted each time an entry is created, updated or destroyed  
+    # So it is only rebuild once required after the list has changed  
+    # Maintaining an array and not rebuilding it all might be faster  
+    # But it will not happen much so that it is fairly acceptable  
+    def to_dropdown(selection=nil, nil_name='** UNDEFINED **')
+      dropdown_cache.inject("<option value=''>#{nil_name}</option>\n") do |out, row|
+        selected = 'selected' if row[1]==selection
+        "%s%s%s%s" % [out, row[2], selected, row[3]]
+      end
+    end
+    def dropdown_cache
+      @dropdown_cache ||= self.find({},:fields=>['_id',label_column]).inject([]) do |out,row|
+        out.push([row.to_label, row.id.to_s, "<option value='#{row.id}' ", ">#{row.to_label}</option>\n"])
+      end
+    end
+    def reset_dropdown_cache; @dropdown_cache = nil; end
+    
+  end
+  
+  module InstanceMethods
+    def crushyform(columns=model.schema.keys, action=nil, meth='POST')
+      columns.delete('_id')
+      fields = columns.inject(""){|out,c|out+crushyfield(c)}
+      enctype = fields.match(/type='file'/) ? "enctype='multipart/form-data'" : ''
+      action.nil? ? fields : "<form action='%s' method='%s' %s>%s</form>\n" % [action, meth, enctype, fields]
+    end
+    # crushyfield is crushyinput but with label+error
+    def crushyfield(col, o={})
+      return '' if (o[:type]==:none || model.schema[col][:type]==:none)
+      default_field_name = col[/^id_/] ? Kernel.const_get(col.sub(/^id_/, '')).human_name : col.tr('_', ' ').capitalize
+      field_name = o[:name] || model.schema[col][:name] || default_field_name
+      error_list = errors_on(col).map{|e|" - #{e}"} if !errors_on(col).nil?
+      invisibility = "style='display: none;'" if (o[:invisible]==true || model.schema[col][:invisible]==true)
+      "<p class='crushyfield %s' %s><label for='%s'>%s</label><span class='crushyfield-error-list'>%s</span><br />\n%s</p>\n" % [error_list&&'crushyfield-error', invisibility, field_id_for(col), field_name, error_list, crushyinput(col, o)]
+    end
+    def crushyinput(col, o={})
+      o = model.schema[col].dup.update(o)
+      o[:input_name] ||= "model[#{col}]"
+      o[:input_value] = o[:input_value].nil? ? self[col] : o[:input_value]
+      o[:input_value] = model.html_escape(o[:input_value]) if (o[:input_value].is_a?(String) && o[:html_escape]!=false)
+      o[:required] = o[:required]==true ? model.crushyfield_required : o[:required]
+      crushyform_type = model.crushyform_types[o[:type]] || model.crushyform_types[:string]
+      crushyform_type.call(self,col,o)
+    end
+    # Provide a thumbnail for the column
+    def to_thumb(c)
+      current = @doc[c]
+      if current.respond_to?(:[])
+        "<img src='/gridfs/#{@doc[c]['stash_thumb_gif']}' width='100' onerror=\"this.style.display='none'\" />\n"
+      end
+    end
+    # Reset dropdowns on hooks
+    def after_save; model.reset_dropdown_cache; super; end
+    def after_delete; model.reset_dropdown_cache; super; end
+		# Fix types
+		def fix_type_string_list(k,v); @doc[k] = v.to_s.strip.split(/\s*,\s*/).compact if v.is_a?(String); end
+  end
+  
+end

data/lib/mongo_mutation.rb

@@ -0,0 +1,272 @@
+# encoding: utf-8
+
+require 'mongo'
+
+module MongoMutation
+  
+	def self.included(weak)
+		weak.extend(MutateClass)
+		weak.db = DB if defined?(DB)
+		weak.schema = BSON::OrderedHash.new
+		weak.relationships = BSON::OrderedHash.new
+	end
+
+  module MutateClass
+	  attr_accessor :db, :schema, :relationships
+		attr_writer :label_column, :slug_column, :sorting_order
+
+    LABEL_COLUMNS = ['title', 'label', 'fullname', 'full_name', 'surname', 'lastname', 'last_name', 'name', 'firstname', 'first_name', 'login', 'caption', 'reference', 'file_name', 'body', '_id']
+		def label_column; @label_column ||= LABEL_COLUMNS.find{|c| @schema.keys.include?(c)||c=='_id'}; end
+		def slug_column; @slug_column ||= (@schema.find{|k,v| v[:type]==:slug}||[])[0]; end 
+	  def foreign_key_name(plural=false); "id#{'s' if plural}_"+self.name; end
+		def human_name; self.name.gsub(/([A-Z])/, ' \1')[1..-1]; end
+		def human_plural_name; human_name+'s'; end
+		def collection; db[self.name]; end
+		def ref(id)
+		  if id.is_a?(String)&&BSON::ObjectId.legal?(id)
+			  id = BSON::ObjectId.from_string(id)
+			elsif !id.is_a?(BSON::ObjectId)
+			  id = ''
+			end
+			{'_id'=>id}
+		end
+		def find(selector={},opts={})
+		  selector.update(opts.delete(:selector)||{})
+      opts = {:sort=>self.sorting_order}.update(opts)
+      collection.find(selector,opts).extend(CursorMutation)
+    end
+    def find_one(spec_or_object_id=nil,opts={})
+      spec_or_object_id.nil? ? spec_or_object_id = opts.delete(:selector) : spec_or_object_id.update(opts.delete(:selector)||{})
+      opts = {:sort=>self.sorting_order}.update(opts)
+      item = collection.find_one(spec_or_object_id,opts)
+      item.nil? ? nil : self.new(item)
+    end
+    def count(opts={}); collection.count(opts); end
+
+    def sorting_order
+			@sorting_order ||= if @schema.key?('position')&&!@schema['position'][:scope].nil?
+				[[@schema['position'][:scope], :asc], ['position', :asc]]
+			elsif @schema.key?('position')
+				[['position', :asc],['_id', :asc]]
+			else
+				['_id', :asc]
+			end
+		end
+
+		def sort(id_list)
+		  id_list.each_with_index do |id, position|
+			  collection.update(ref(id), {'$set' => {'position'=>position}})
+			end
+		end
+
+		# CRUD
+		def get(id, opts={}); doc = collection.find_one(ref(id), opts); doc.nil? ? nil : self.new(doc); end
+		def delete(id); collection.remove(ref(id)); end
+
+		def is_unique(doc={})
+		  return unless collection.count==0
+		  doc = {'_id'=>BSON::ObjectId('000000000000000000000000')}.update(doc)
+		  d = self.new
+			d.doc.update(doc)
+			d.save 
+		end
+
+		private
+		def slot(name,opts={})
+		  @schema[name] = {:type=>:string}.update(opts)
+		  define_method(name) { @doc[name] }
+      define_method("#{name}=") { |x| @doc[name] = x }
+		end
+		def image_slot(name='image',opts={})
+		  slot name, {:type=>:attachment}.update(opts)
+			slot "#{name}_tooltip"
+			slot "#{name}_alternative_text"
+		end
+		def has_many(k,opts={}); @relationships[k] = opts; end
+	end
+
+  attr_accessor :doc, :old_doc, :errors, :is_new
+	def initialize(document=nil); @errors={}; @doc = document || default_doc; end
+	def default_doc
+	  @is_new = true
+	  out = {}
+		model.schema.each { |k,v| out.store(k,v[:default].is_a?(Proc) ? v[:default].call : v[:default]) }
+		out
+	end
+	def model; self.class; end
+	def id; @doc['_id']; end
+	def [](field); @doc[field]; end
+	def []=(field,val); @doc[field] = val; end
+	def to_label;  @doc[model.label_column].to_s.tr("\n\r", ' '); end
+	ACCENTS_FROM = "ÀÁÂÃÄÅàáâãäåĀāĂ㥹ÇçĆćĈĉĊċČčÐðĎďĐđÈÉÊËèéêëĒēĔĕĖėĘęĚěĜĝĞğĠġĢģĤĥĦħÌÍÎÏìíîïĨĩĪīĬĭĮįİıĴĵĶķĸĹĺĻļĽľĿŀŁłÑñŃńŅņŇňʼnŊŋÒÓÔÕÖØòóôõöøŌōŎŏŐőŔŕŖŗŘřŚśŜŝŞşŠšſŢţŤťŦŧÙÚÛÜùúûüŨũŪūŬŭŮůŰűŲųŴŵÝýÿŶŷŸŹźŻżŽž"
+  ACCENTS_TO   = "AAAAAAaaaaaaAaAaAaCcCcCcCcCcDdDdDdEEEEeeeeEeEeEeEeEeGgGgGgGgHhHhIIIIiiiiIiIiIiIiIiJjKkkLlLlLlLlLlNnNnNnNnnNnOOOOOOooooooOoOoOoRrRrRrSsSsSsSssTtTtTtUUUUuuuuUuUuUuUuUuUuWwYyyYyYZzZzZz"
+	def auto_slug
+	  s = self.to_label.tr(ACCENTS_FROM,ACCENTS_TO).tr(' .,;:?!/\'"()[]{}<>','-').gsub(/&/, 'and')
+    defined?(::Rack::Utils) ? ::Rack::Utils.escape(s) : s
+  end
+  def to_slug; @doc[model.column_slug]||self.auto_slug; end
+	def to_param; "#{@doc['_id']}-#{to_label.scan(/\w+/).join('-')}"; end
+  def field_id_for(col); "%s-%s-%s" % [id||'new',model.name,col]; end
+
+	# relationships
+	def resolve_class(k); k.kind_of?(Class) ? k : Kernel.const_get(k); end
+	def parent(k, opts={})
+	  if k.kind_of?(String)
+	    key = k
+	    klass = resolve_class(model.schema[k][:parent_class]) 
+    else
+      klass = resolve_class(k)
+      key = klass.foreign_key_name
+    end
+	  klass.get(@doc[key], opts)
+	end
+	def slot_children(k, opts={})
+	  if k.kind_of?(String)
+	    key = k
+	    klass = resolve_class(model.schema[k][:children_class]) 
+    else
+      klass = resolve_class(k)
+      key = klass.foreign_key_name(true)
+    end
+    ids = (@doc[key]||[]).map{|i| BSON::ObjectId.from_string(i) }
+    selector = {'_id'=>{'$in'=>ids}}
+    sort_proc = proc{ |a,b| ids.index(a['_id'])<=>ids.index(b['_id']) }
+    klass.find(selector, opts).to_a.sort(&sort_proc)
+  end
+  def first_slot_child(k, opts={})
+    if k.kind_of?(String)
+	    key = k
+	    klass = resolve_class(model.schema[k][:children_class]) 
+    else
+      klass = resolve_class(k)
+      key = klass.foreign_key_name(true)
+    end
+    klass.get((@doc[key]||[])[0], opts)
+  end
+	def children(k,opts={})
+	  k = resolve_class(k)
+	  slot_name = opts.delete(:slot_name) || model.foreign_key_name
+	  k.find({slot_name=>@doc['_id'].to_s}, opts)
+	end
+	def first_child(k,opts={})
+	  k = resolve_class(k)
+	  slot_name = opts.delete(:slot_name) || model.foreign_key_name
+	  d = k.find_one({slot_name=>@doc['_id'].to_s}, opts)
+	end
+	def children_count(k,sel={})
+	  k = resolve_class(k)
+	  slot_name = sel.delete(:slot_name) || model.foreign_key_name
+	  k.collection.count(:query => {slot_name=>@doc['_id'].to_s}.update(sel))
+  end
+
+	# CRUD
+	def delete
+	  before_delete
+	  model.delete(@doc['_id'])
+		after_delete
+	end
+
+	# saving and hooks
+	def new?; @is_new ||= !@doc.key?('_id'); end
+	def update_doc(fields); @old_doc = @doc.dup; @doc.update(fields); @is_new = false; self; end
+	# Getter and setter in one
+	def errors_on(col,message=nil)
+	  message.nil? ? @errors[col] : @errors[col] = (@errors[col]||[]) << message
+  end
+	def before_delete; @old_doc = @doc.dup; end
+	alias before_destroy before_delete
+	def after_delete
+	  model.relationships.each do |k,v|
+		  Kernel.const_get(k).find({model.foreign_key_name=>@old_doc['_id'].to_s}).each{|m| m.delete} unless v[:independent]
+		end
+	end
+	alias after_destroy after_delete
+	def valid?
+		before_validation
+		validate
+		after_validation
+		@errors.empty?
+	end
+	def before_validation
+	  @errors = {}
+	  @doc.each do |k,v|
+	    next unless model.schema.key?(k)
+			type = k=='_id' ? :primary_key : model.schema[k][:type]
+	    fix_method = "fix_type_#{type}"
+	    if v==''
+	      default = model.schema[k][:default]
+	      @doc[k] = default.is_a?(Proc) ? default.call : default
+      else
+	      self.__send__(fix_method, k, v) if self.respond_to?(fix_method)
+      end
+    end
+	end
+	def validate; end
+	def after_validation; end
+	def fix_type_integer(k,v); @doc[k] = v.to_i; end
+	def fix_type_boolean(k,v); @doc[k] = (v=='true'||v==true) ? true : false; end
+	def fix_type_slug(k,v); @doc[k] = self.auto_slug if v.to_s==''; end
+	def fix_type_date(k,v)
+	  if v.is_a?(String)
+	    if v[/\d\d\d\d-\d\d-\d\d/]
+	      @doc[k] =  ::Time.utc(*v.split('-'))
+      else
+	      default = model.schema[k][:default]
+        @doc[k] = default.is_a?(Proc) ? default.call : default
+      end
+	  end
+	end
+	def fix_type_datetime(k,v)
+	  if v.is_a?(String)
+	    if v[/\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d/]
+	      @doc[k] =  ::Time.utc(*v.split(/[-:\s]/))
+      else
+	      default = model.schema[k][:default]
+        @doc[k] = default.is_a?(Proc) ? default.call : default
+      end
+	  end
+	end
+
+	def save
+	  return nil unless valid?
+		before_save
+		if new?
+		  before_create
+		  id = model.collection.insert(@doc)
+			@doc['_id'] = id
+			after_create
+		else
+		  before_update
+		  id = model.collection.update({'_id'=>@doc['_id']}, @doc)
+			after_update
+		end
+		after_save
+		id.nil? ? nil : self
+	end
+	def before_save; end
+	def before_create; end
+	def before_update; end
+	def after_save; end
+	def after_create; @is_new = false; end
+	def after_update; end
+
+  # ==========
+  # = Cursor =
+  # ==========
+  module CursorMutation
+    def self.extended(into)
+      into.set_mutant_class
+    end
+    def set_mutant_class
+      @mutant_class = Kernel.const_get(collection.name)
+    end
+    def next
+      n = super
+      n.nil? ? nil : @mutant_class.new(n)
+    end
+    # legacy
+    def each_mutant(&b); each(&b); end
+    def each_mutant_with_index(&b); each_with_index(&b); end
+  end
+
+end

data/lib/mongo_stash.rb

@@ -0,0 +1,155 @@
+# encoding: utf-8
+
+require 'mongo_mutation'
+
+module MongoStash
+  
+	def self.included(base)
+	  MongoStash.classes << base
+	  base.extend(ClassMethods)
+		base.gridfs = GRID
+	end
+
+  module ClassMethods
+	  attr_accessor :gridfs
+	  def all_after_stash
+  	  self.collection.find.each do |i|
+  		  self.schema.each do |k,v|
+  			  obj = self.new(i)
+  			  obj.after_stash(k) if v[:type]==:attachment&&obj[k].to_s!=''
+  			end
+  		end
+  	end
+	end
+	
+	def build_image_tag(col='image', style='original', html_attributes={})
+	  return '' if @doc[col].nil?||@doc[col][style].nil?
+	  title_field, alt_field = col+'_tooltip', col+'_alternative_text'
+		title = @doc[title_field] if model.schema.keys.include?(title_field)
+		alt = @doc[alt_field] if model.schema.keys.include?(alt_field)
+		html_attributes = {:src => "/gridfs/#{@doc[col][style]}", :title => title, :alt => alt}.update(html_attributes)
+		html_attributes = html_attributes.map do |k,v|
+		  %{#{k}="#{model.html_escape(v.to_s)}"}
+		end.join(' ')
+	  "<img #{html_attributes} />"
+	end
+
+	def fix_type_attachment(k,v)
+	  if v=='nil'
+      delete_files_for(k) unless new?
+			@doc[k] = nil
+	  elsif v.is_a?(Hash)&&v.key?(:tempfile)
+		  delete_files_for(k) unless new?
+			@temp_attachments ||= {}
+			@temp_attachments[k] = v
+      attachment_id = model.gridfs.put(v[:tempfile], {:filename=>v[:filename], :content_type=>v[:type]})
+			@doc[k] = {'original'=>attachment_id}
+		end
+	end
+
+	def delete_files_for(col)
+	  obj = (@old_doc||@doc)[col]
+		if obj.respond_to?(:each)
+		  obj.each do |k,v|
+			  model.gridfs.delete(v)
+			end
+		end
+	end
+
+	def after_delete
+	  super
+    model.schema.each do |k,v|
+		  delete_files_for(k) if v[:type]==:attachment
+		end
+	end
+
+	def after_save
+	  super
+		unless @temp_attachments.nil?
+			@temp_attachments.each do |k,v|
+				after_stash(k)
+			end
+		end
+	end
+
+	def after_stash(col); end
+
+	def convert(col, convert_steps, style)
+	  return if @doc[col].nil?
+	  if @temp_attachments.nil? || @temp_attachments[col].nil?
+	    f = model.gridfs.get(@doc[col]['original'])
+	    return unless f.content_type[/^image\//]
+	    src = Tempfile.new('MongoStash_src')
+	    src.binmode
+	    src.write(f.read(4096)) until f.eof?
+	    src.close
+	    @temp_attachments ||= {}
+	    @temp_attachments[col] ||= {}
+	    @temp_attachments[col][:tempfile] = src
+	    @temp_attachments[col][:type] = f.content_type
+		else
+		  return unless @temp_attachments[col][:type][/^image\//]
+		  src = @temp_attachments[col][:tempfile]
+		end
+	  model.gridfs.delete(@doc[col][style]) unless @doc[col][style].nil?
+		ext = style[/[a-zA-Z]+$/].insert(0,'.')
+    content_type = Rack::Mime.mime_type(ext) 
+    unless content_type[/^image\//]
+      ext = '.jpg'
+      content_type = 'image/jpeg'
+    end
+		dest = Tempfile.new(['MongoStash_dest', ext])
+		dest.binmode
+		dest.close
+	  system "convert \"#{src.path}\" #{convert_steps} \"#{dest.path}\""
+		filename = "#{model.name}/#{self.id}/#{style}"
+		attachment_id = model.gridfs.put(dest.open, {:filename=>filename, :content_type=>content_type})
+    @doc[col] = @doc[col].update({style=>attachment_id})
+		model.collection.update({'_id'=>@doc['_id']}, @doc)
+		#src.close!
+		dest.close!
+	end
+
+	class << self
+	  attr_accessor :classes
+		MongoStash.classes = []
+		
+		def all_after_stash
+		  MongoStash.classes.each do |m|
+			  m.all_after_stash
+			end
+		end
+		
+		def fix_dots_in_keys(c, for_real=false)
+      puts "\n#{c}" unless for_real
+      selection = c.schema.select{|k,v| v[:type]==:attachment }
+      img_keys = selection.respond_to?(:keys) ? selection.keys : selection.map{|a|a[0]}
+      c.find({}, {:fields=>img_keys}).each do |e|
+        old_hash = e.doc.select{|k,v| img_keys.include?(k) }
+        fixed_hash = Marshal.load(Marshal.dump(old_hash))
+        img_keys.each do |k|
+          (fixed_hash[k]||{}).keys.each do |style|
+            fixed_hash[k][style.tr('.','_')] = fixed_hash[k].delete(style)
+          end
+        end
+        next if old_hash==fixed_hash
+        
+        if for_real
+          c.collection.update({'_id'=>e.id}, {'$set'=>fixed_hash})
+        else
+          puts old_hash.inspect
+          puts fixed_hash.inspect
+        end
+        
+      end
+    end
+    
+    def all_fix_dots_in_keys(for_real=false)
+      MongoStash.classes.each do |c|
+        MongoStash.fix_dots_in_keys(c, for_real)
+      end
+    end
+    
+	end
+
+end

data/test/spec_mutation.rb

@@ -0,0 +1,161 @@
+Encoding.default_internal = Encoding.default_external = Encoding::UTF_8 if RUBY_VERSION >= '1.9.0'
+
+require 'rubygems'
+require 'bacon'
+$:.unshift './lib'
+require 'mongo_mutation'
+
+MONGO = ::Mongo::MongoClient.new
+class NoDB; include MongoMutation; end
+DB  = MONGO['test-mongo-mutation']
+
+class Naked; include MongoMutation; end
+
+class Person
+  include MongoMutation
+  slot "name" # no options
+  slot "surname"
+  slot "age", :type=>:integer, :default=>18
+  image_slot "portrait"
+  slot 'subscribed_on', :type=>:date, :default=>proc{Date.today}
+end
+
+class Address
+  include MongoMutation
+  slot "body"
+end
+
+class Article
+  include MongoMutation
+  slot 'title'
+  slot 'content', :type=>:text
+  image_slot
+end
+
+describe "MongoMutation" do
+  
+  describe ".db" do 
+    it "Should be set to DB constant by default" do
+      NoDB.db.should==nil
+      Naked.db.should==DB
+    end
+  end
+  
+  shared "Empty BSON" do
+    it "Should be set to an empty BSON ordered hash by default" do
+      @bson.class.should==::BSON::OrderedHash
+      @bson.empty?.should==true
+    end
+  end
+  
+  describe ".schema" do
+    before { @bson = Naked.schema }
+    behaves_like "Empty BSON"
+  end
+  
+  describe ".relationships" do
+    before { @bson = Naked.relationships }
+    behaves_like "Empty BSON"
+  end
+  
+  shared "Basic slot" do
+    it "Adds the declared slot to the schema" do
+      @klass.schema.key?(@key).should==true
+    end
+    it "Defines getters and setters" do
+      @klass.new.respond_to?(@key).should==true
+      @klass.new.respond_to?(@key+'=').should==true
+    end
+  end
+  
+  describe ".slot" do
+    before { @klass = Person; @key = 'age' }
+    behaves_like "Basic slot"
+    it "Keeps the options in schema" do
+      Person.schema['age'][:default].should==18
+      Person.schema['age'][:type].should==:integer
+    end
+    it "Sets :type option to :string by default if not provided" do
+      Person.schema['name'][:type].should==:string
+    end
+  end
+  
+  shared "Correctly typed" do
+    it "Has the correct type" do
+      @klass.schema[@key][:type].should==@type
+    end
+  end
+  
+  shared "Image slot" do
+    describe "Reference slot" do
+      before { @type = :attachment }
+      behaves_like "Basic slot"
+      behaves_like "Correctly typed"
+    end
+    describe "Tooltip slot" do
+      before { @key = @key+'_tooltip'; @type = :string }
+      behaves_like "Basic slot"
+      behaves_like "Correctly typed"
+    end
+    describe "Alt text slot" do
+      before { @key = @key+'_alternative_text'; @type = :string }
+      behaves_like "Basic slot"
+      behaves_like "Correctly typed"
+    end
+  end
+  
+  describe ".image_slot" do
+    describe "Name provided" do
+      before { @klass = Person; @key = 'portrait' }
+      behaves_like "Image slot"
+    end
+    describe "Name not provided" do
+      before { @klass = Article; @key = 'image' }
+      behaves_like "Image slot"
+    end
+  end
+  
+  shared "Has error recipient" do
+    it "Has an empty recipent for errors" do
+      @inst.errors.should=={}
+    end
+  end
+  
+  describe "#initialize" do
+    describe "With fields" do
+      before { @inst = Person.new({'age'=>42, 'name'=>'Bozo', 'car'=>'Jaguar'}) }
+      it "Should be flagged as new if the doc has no _id yet" do
+        @inst.new?.should==true
+      end
+      it "Should not be flagged as new if the doc has an _id" do
+        @inst['_id'] = 'abc'
+        @inst.new?.should==false
+      end
+      it "Should pass all keys to the doc" do
+        @inst['name'].should=='Bozo'
+        @inst['car'].should=='Jaguar'
+      end
+      behaves_like "Has error recipient"
+    end
+    describe "Fresh new one" do
+      before { @inst = Person.new }
+      it "Should be flagged as new" do
+        @inst.new?.should==true
+      end
+      it "Has default values correctly set" do
+        @inst['age'].should==18 # direct value
+        @inst['subscribed_on'].class.should==Date # with proc
+      end
+      behaves_like "Has error recipient"
+    end
+  end
+  
+  describe "#model" do
+    it "Is a shortcut for self.class" do
+      Naked.new.model.should==Naked
+    end
+  end
+
+end
+
+MONGO.drop_database('test-mongo-mutation')

metadata

@@ -0,0 +1,84 @@
+--- !ruby/object:Gem::Specification
+name: campbellhay-mongo
+version: !ruby/object:Gem::Version
+  version: 1.0.14
+  prerelease: 
+platform: ruby
+authors:
+- Mickael Riga
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mongo
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.8.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.8.2
+- !ruby/object:Gem::Dependency
+  name: bacon
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: The CampbellHay Mongo toolbox
+email: mig@campbellhay.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- campbellhay-mongo.gemspec
+- gempush
+- lib/mongo_crushyform.rb
+- lib/mongo_mutation.rb
+- lib/mongo_stash.rb
+- test/spec_mutation.rb
+homepage: http://www.campbellhay.com
+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: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: The CampbellHay Mongo toolbox
+test_files: []