mongoid_ext 0.6.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/Gemfile

@@ -0,0 +1,20 @@
+source :gemcutter
+
+gem 'mongoid', '~> 2'
+gem 'uuidtools', '>= 2.1.1'
+gem 'i18n'
+gem 'tzinfo'
+gem 'differ', '>= 0.1.2'
+
+group :development do
+  gem "yard", "~> 0.6.0"
+  gem "bundler", "~> 1.0.0"
+  gem "jeweler", "~> 1.5.2"
+
+  gem "shoulda", "~> 2.11.3"
+  gem "jnunemaker-matchy", "~> 0.4"
+  gem 'shoulda', '~> 2.11.3'
+  gem 'mocha', '~> 0.9.4'
+  gem 'timecop'
+end
+

data/Gemfile.lock

@@ -0,0 +1,50 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    activemodel (3.0.6)
+      activesupport (= 3.0.6)
+      builder (~> 2.1.2)
+      i18n (~> 0.5.0)
+    activesupport (3.0.6)
+    bson (1.3.0)
+    builder (2.1.2)
+    differ (0.1.2)
+    git (1.2.5)
+    i18n (0.5.0)
+    jeweler (1.5.2)
+      bundler (~> 1.0.0)
+      git (>= 1.2.5)
+      rake
+    jnunemaker-matchy (0.4.0)
+    mocha (0.9.12)
+    mongo (1.3.0)
+      bson (>= 1.3.0)
+    mongoid (2.0.1)
+      activemodel (~> 3.0)
+      mongo (~> 1.3)
+      tzinfo (~> 0.3.22)
+      will_paginate (~> 3.0.pre)
+    rake (0.8.7)
+    shoulda (2.11.3)
+    timecop (0.3.5)
+    tzinfo (0.3.26)
+    uuidtools (2.1.2)
+    will_paginate (3.0.pre2)
+    yard (0.6.7)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.0.0)
+  differ (>= 0.1.2)
+  i18n
+  jeweler (~> 1.5.2)
+  jnunemaker-matchy (~> 0.4)
+  mocha (~> 0.9.4)
+  mongoid (~> 2.0.0)
+  shoulda (~> 2.11.3)
+  timecop
+  tzinfo
+  uuidtools (>= 2.1.1)
+  yard (~> 0.6.0)

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 David A. Cuadrado
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,17 @@
+= mongoid_ext
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 David A. Cuadrado. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,44 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "mongoid_ext"
+    gem.summary = %Q{mongoid plugins}
+    gem.description = %Q{mongoid plugins}
+    gem.email = "krawek@gmail.com"
+    gem.homepage = "http://github.com/dcu/mongoid_ext"
+    gem.authors = ["David A. Cuadrado"]
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :default => :test
+
+require 'yard'
+YARD::Rake::YardocTask.new
+

data/VERSION

@@ -0,0 +1 @@
+0.6.1

data/bin/mongoid_console

@@ -0,0 +1,85 @@
+#!/usr/bin/env ruby
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+
+begin
+  require 'mongoid_ext'
+  require 'irb'
+rescue LoadError
+  require 'rubygems'
+  retry
+end
+
+IRB.setup(nil)
+irb = IRB::Irb.new
+
+IRB.conf[:MAIN_CONTEXT] = irb.context
+
+irb.context.evaluate("require 'irb/completion'", 0)
+irb.context.evaluate(%@
+include Mongoid
+
+Mongoid.configure do |config|
+  config.master = Mongo::Connection.new.db("test")
+end
+$db = Mongoid.config.database
+
+@, 0)
+
+puts %@
+Welcome to the Mongoid Console!
+
+Example 1: Using the driver
+  things = $db.collection("things")
+  things.insert("name" => "Raw Thing")
+  things.insert("name" => "Another Thing", "date" => Time.now)
+
+  cursor = things.find("name" => "Raw Thing")
+  puts cursor.next_object.inspect
+
+Example 2: Defining documents
+  class Person
+    include Mongoid::Document
+    field :first_name
+    field :middle_initial
+    field :last_name
+  end
+
+Example 3: Associations
+  class Person
+    include Mongoid::Document
+    references_one :policy
+    references_many :prescriptions
+    references_many :preferences, :stored_as => :array, :inverse_of => :people
+  end
+
+  class Policy
+    include Mongoid::Document
+    referenced_in :person
+  end
+
+  class Prescription
+    include Mongoid::Document
+    referenced_in :person
+  end
+
+  class Preference
+    include Mongoid::Document
+    references_many :people, :stored_as => :array, :inverse_of => :preferences
+  end
+
+  person = Person.create
+  policy = Policy.create
+  prescription = Prescription.create
+
+  person.policy = policy
+  person.prescriptions = [prescription]
+@
+
+trap("SIGINT") do
+  irb.signal_handle
+end
+
+catch(:IRB_EXIT) do
+  irb.eval_input
+end
+

data/lib/mongoid_ext.rb

@@ -0,0 +1,71 @@
+$:.unshift File.dirname(__FILE__)
+
+if RUBY_VERSION =~ /^1\.8/
+  $KCODE = 'u'
+end
+
+require 'mongoid'
+require 'uuidtools'
+require 'differ'
+require 'active_support/inflector'
+
+begin
+  require 'magic'
+rescue LoadError
+  $stderr.puts "disabling `magic` support. use 'gem install magic' to enable it"
+end
+
+require 'mongoid_ext/patches'
+
+# types
+require 'mongoid_ext/types/open_struct'
+require 'mongoid_ext/types/timestamp'
+require 'mongoid_ext/types/translation'
+require 'mongoid_ext/types/embedded_hash'
+
+# storage
+require 'mongoid_ext/file_list'
+require 'mongoid_ext/file'
+require 'mongoid_ext/storage'
+require 'mongoid_ext/file_server'
+
+# update
+require 'mongoid_ext/update'
+
+# filter
+require 'mongoid_ext/filter'
+require 'mongoid_ext/filter/parser'
+require 'mongoid_ext/filter/result_set'
+
+# slug
+require 'mongoid_ext/slugizer'
+
+# tags
+require 'mongoid_ext/tags'
+
+require 'mongoid_ext/versioning'
+require 'mongoid_ext/voteable'
+require 'mongoid_ext/paranoia'
+
+require 'mongoid_ext/random'
+require 'mongoid_ext/mongo_mapper'
+require 'mongoid_ext/document_ext'
+require 'mongoid_ext/criteria_ext'
+require 'mongoid_ext/modifiers'
+
+module MongoidExt
+  def self.init
+    load_jsfiles(::File.dirname(__FILE__)+"/mongoid_ext/js")
+  end
+
+  def self.load_jsfiles(path)
+    Dir.glob(::File.join(path, "*.js")) do |js_path|
+      code = ::File.read(js_path)
+      name = ::File.basename(js_path, ".js")
+
+      # HACK: looks like ruby driver doesn't support this
+      Mongoid.master.eval("db.system.js.save({_id: '#{name}', value: #{code}})")
+    end
+  end
+end
+

data/lib/mongoid_ext/criteria_ext.rb

@@ -0,0 +1,15 @@
+module MongoidExt
+  module CriteriaExt
+    extend ActiveSupport::Concern
+
+    included do
+    end
+
+    module ClassMethods
+    end
+
+    module InstanceMethods
+    end
+  end
+end
+Mongoid::Criteria.send(:include, MongoidExt::CriteriaExt)

data/lib/mongoid_ext/document_ext.rb

@@ -0,0 +1,29 @@
+module MongoidExt
+  module DocumentExt
+    extend ActiveSupport::Concern
+
+    included do
+    end
+
+    module ClassMethods
+      def find!(*args)
+        find(*args) || raise(Mongoid::Errors::DocumentNotFound.new(self, args))
+      end
+
+    end
+
+    module InstanceMethods
+      def raw_save(opts = {})
+        return true if !changed? && !opts.delete(:force)
+
+        if (opts.delete(:validate) != false || valid?)
+          self.collection.save(raw_attributes, opts)
+          true
+        else
+          false
+        end
+      end
+    end
+  end
+end
+Mongoid::Document.send(:include, MongoidExt::DocumentExt)

data/lib/mongoid_ext/file.rb

@@ -0,0 +1,86 @@
+module MongoidExt
+  class File < EmbeddedHash
+    attr_accessor :_root_document
+
+    field :name, :type => String
+    field :extension, :type => String
+    field :content_type, :type => String
+
+    alias :filename :name
+
+    def put(filename, io, options = {})
+      options[:_id] = grid_filename
+
+      options[:metadata] ||= {}
+      options[:metadata][:collection] = _root_document.collection.name
+
+      self["name"] = filename
+      if filename =~ /\.([\w]{2,4})$/
+        self["extension"] = $1
+      end
+
+      if io.kind_of?(String)
+        io = StringIO.new(io)
+      end
+
+      if defined?(Magic) && Magic.respond_to?(:guess_string_mime_type)
+        data = io.read(256) # be nice with memory usage
+        self["content_type"] = options[:content_type] = Magic.guess_string_mime_type(data.to_s)
+        self["extension"] ||= options[:content_type].to_s.split("/").last.split("-").last
+
+        if io.respond_to?(:rewind)
+          io.rewind
+        else
+          io.seek(0)
+        end
+      end
+
+      options[:filename] = grid_filename
+      gridfs.delete(grid_filename)
+      gridfs.put(io, options)
+    end
+
+    def get
+      @io ||= gridfs.get(grid_filename)
+    end
+
+    def reset
+      @io = nil
+    end
+
+    def grid_filename
+      @grid_filename ||= "#{_root_document.collection.name}/#{self.id}"
+    end
+
+    def mime_type
+      self.content_type || get.content_type
+    end
+
+    def size
+      get.file_length
+    end
+
+    def read(size = nil)
+      self.get.read(size)
+    end
+
+    def delete
+      @io = nil
+      gridfs.delete(grid_filename)
+    end
+
+    def method_missing(name, *args, &block)
+      f = self.get rescue nil
+      if f && f.respond_to?(name)
+        f.send(name, *args, &block)
+      else
+        super(name, *args, &block)
+      end
+    end
+
+    protected
+    def gridfs
+      _root_document.class.gridfs
+    end
+  end
+end