mongo_mapper_ext 0.2.2 → 0.2.3

data/lib/mongo_mapper_ext/rad/file_uploader.rb

@@ -0,0 +1,28 @@
+require 'carrierwave/processing/mini_magick'
+
+module Models
+  class FileUploader < CarrierWave::Uploader::Base
+    include CarrierWave::MiniMagick
+    
+    storage rad.config.fs.type(:file)
+  
+    # def sanitize_regexp
+    #   /[^[:word:]\.\-\+\s_]/i
+    # end
+  
+    def file_path
+      "#{rad.config.fs.prefix('/fs')}/system/#{model.class.model_name.underscore}/#{model.id}"
+    end
+  
+    def store_dir
+      "#{root}#{file_path}"
+    end
+  
+    def extension_white_list
+      [/.*/]
+    end
+  
+    def cache_dir; rad.config.fs.cache_path! end  
+    def root; rad.config.fs.path! end  
+  end
+end

data/lib/mongo_mapper_ext/rad.rb

@@ -0,0 +1,2 @@
+require 'rad'
+require 'mongo_mapper_ext/rad/file_uploader'

data/lib/mongo_mapper_ext/spec.rb

@@ -0,0 +1,103 @@
+require 'mongo_mapper_ext'
+
+# 
+# disabling :use_database, all tests will use the same :test database.
+# 
+module MongoMapper::Plugins::DbConfig::ClassMethods
+  alias_method :_use_database, :use_database
+  def use_database database_alias; end
+end
+
+
+# 
+# Disabling indexes in testing environment
+# 
+if Object.const_defined?(:Rad)
+  MongoMapper::Plugins::Indexes::ClassMethods.class_eval do        
+    alias_method :original_ensure_index, :ensure_index
+    def ensure_index_stub *args; end
+  end
+  
+  rad.after :mode, bang: false do
+    if rad.production?
+      MongoMapper::Plugins::Indexes::ClassMethods.send :alias_method, :ensure_index, :original_ensure_index
+    else
+      MongoMapper::Plugins::Indexes::ClassMethods.send :alias_method, :ensure_index, :ensure_index_stub
+    end
+  end  
+end
+
+
+# 
+# Clear db before each test
+# 
+rspec do
+  def self.with_mongo_mapper
+    before :all do
+      MongoMapper.db_config = {
+        'default' => {'name' => "test"}
+      }
+      
+      MongoMapper.database = 'test'      
+    end
+    
+    before do
+      MongoMapper.db_config.each do |db_alias, opt|      
+        db = MongoMapper.databases[db_alias]    
+        db.collection_names.each do |name|
+          next if name =~ /^system\./
+          db.collection(name).drop
+        end
+      end
+    end
+  end
+end
+
+
+# 
+# Silence logger
+# 
+MongoMapper.logger = Logger.new(nil)
+if defined? Paperclip
+  Paperclip.class_eval do
+    def self.logger
+      @logger ||= Logger.new(nil)
+    end
+  end
+end
+
+
+# 
+# Files
+# 
+rspec do
+  class << self
+    def with_files
+      path, cache_path = '/tmp/spec_fs', '/tmp/spec_fs_cache'
+      
+      before do
+        rad.config.merge!({fs: {path: path, cache_path: cache_path}}, override: true)
+        
+        Models::FileUploader.storage :file
+        
+        CarrierWave.configure do |config|          
+          config.storage = :file
+          config.enable_processing = false
+          
+          config.cache_dir = rad.config.fs.cache_path!
+          config.root = rad.config.fs.path!
+        end
+      end
+      
+      before do
+        path.to_dir.destroy
+        cache_path.to_dir.destroy
+      end
+      
+      after do
+        path.to_dir.destroy
+        cache_path.to_dir.destroy
+      end
+    end    
+  end
+end

data/lib/mongo_mapper_ext/{migrate.rake → tasks.rb}

@@ -1,23 +1,11 @@
-Rake::TaskManager.class_eval do
-  def remove_task(task_name)
-    @tasks.delete(task_name.to_s)
-  end
-end
-
-def remove_task(task_name)
-  Rake.application.remove_task(task_name)
-end
-
 namespace :db do
-  
-  remove_task 'db:migrate'
   desc "Migrate Database"
-  task :migrate => :environment do
+  task migrate: :environment do
     ::Migration = MongoMapper::Migration
-    Dir["#{Rails.root}/lib/db/**/*.rb"].each{|f| require f.sub(/\.rb$/, '')}
+    Dir["#{rad.config.runtime_dir!}/db/**/*.rb"].each{|f| require f.sub(/\.rb$/, '')}
     
     database_alias = ENV['d'] || ENV['database']
-    database_alias = 'accounts' if database_alias.blank?
+    database_alias = 'default' if database_alias.blank?
     
     version = ENV['v'] || ENV['version']
     if version.blank?

data/lib/mongo_mapper_ext.rb

@@ -1,32 +1,49 @@
 require 'mongo_mapper_ext/gems'
 
 require 'mongo_mapper'
+require 'carrierwave'
 
-[
-  'hacks/fixes',
-  'hacks/time_measuring',
-  'migration',
-  'mongo_mapper',
-  'view_helpers',
-  'db_config',
-  'micelaneous',
+%w(
+  micelaneous
+).each{|file| require "mongo_db_ext/#{file}"}
+
+%w(
+  hacks/fixes
+  hacks/active_model
   
-  'plugins/default_scope',
-  # 'plugins/db_config',  
-  'plugins/attributes_cache',
-  'plugins/micelaneous',  
-].each do |file|
-  require "mongo_mapper_ext/#{file}"
-end
+  migration
+  mongo_mapper
+  view_helpers  
+  logging
+  
+  plugins/db_config
+  plugins/custom_scope
+  plugins/attribute_cache
+  plugins/carrierwave
+  plugins/attribute_convertors
+  plugins/micelaneous
+  plugins/belongs_to_with_counter_cache
+).each{|file| require "mongo_mapper_ext/#{file}"}
 
-module CommonPluginsAddition
-  def self.included(model)
-    model.plugin MongoMapper::Plugins::DefaultScope
-    # model.plugin MongoMapper::Plugins::DbConfig
-    model.plugin MongoMapper::Plugins::AttributesCache
-    model.plugin MongoMapper::Plugins::Micelaneous
-    
-    model.attr_protected :id, :_id, :_type, :created_at, :updated_at
+# 
+# Default plugins and settings
+# 
+module MongoMapper::Plugins
+  [CustomScope, AttributeCache, AttributeConvertors, BelongsToWithCounterCache, Micelaneous, CarrierWave, DbConfig].each do |plugin|
+    ::MongoMapper::Document.send :include, plugin
   end
 end
-MongoMapper::Document.append_inclusions(CommonPluginsAddition)
+  
+# 
+# Attribute protection
+# 
+MongoMapper::Document.included do
+  attr_protected :id, :_id, :_type, :created_at, :updated_at
+end
+
+
+# 
+# Locales
+# 
+dir = File.expand_path("#{__FILE__}/../..")
+I18n.load_path += Dir["#{dir}/locales/**/*.{rb,yml}"]

data/readme.md

@@ -1,3 +1,75 @@
-# Extensions for MongoMapper.
+# Extensions for MongoMapper
 
-[add readme, for now go to specs]
+## Simultaneous use of multiple databases
+
+    MongoMapper.db_config = {
+      'default' => {'name' => "default_test"}
+      'global' => {'name' => 'global_test'},      
+    }
+
+    # Comment will be connected to :default database
+    module Comment
+      belongs_to :user
+    end
+
+    # User will be connected to :global database
+    module User
+      use_database :global
+      
+      has_many :comments
+    end
+    
+## Migrations
+
+    # Works with multiple databases, support versions
+    Migration.define :default, 1 do |m|
+      m.up{Sample.create name: 'name'}
+      m.down{Sample.destroy_all}
+    end
+    
+    # Tell it the version of database You need, and it's smart enough to figure out all needed :up or :down
+    Migration.update(:global, 1)
+    
+
+## Custom Scope
+    :with_scope, :default_scope, :with_exclusive_scope, see spec for details.
+    
+## Counter Cache
+
+    class Comment
+      belongs_to :post, counter_cache: true
+    end
+    
+    Post.comments_count
+    
+## Attribute Converters
+
+For editing complex objects in forms:
+
+    class Post
+      key :tags, Array, as_string: :line
+    end
+    
+    @post.tags_as_string = "personal, article"
+    @post.tags                                  # => ['personal', 'article']
+    @post.tags_as_string                        # => "personal, article"
+    
+## Handy upserts
+
+    @post.upsert! :$inc => {comments_count: 1}
+    
+## CarrierWave integration
+
+File attachments (stored on File System, S3, MongoDB-GridFS)
+
+    class User
+      mount_uploader :avatar, AvatarUploader
+    end
+    
+## more
+
+Attribute Cache, Spec Helpers (auto-clean db before each test).
+
+# License
+
+Copyright (c) Alexey Petrushin [http://4ire.net](http://4ire.net), released under the MIT license.

data/spec/attribute_convertors_spec.rb

@@ -0,0 +1,71 @@
+require 'spec_helper'
+
+describe "Attribute Convertors" do
+  with_mongo_mapper
+  
+  after(:all){remove_constants %w(AsStringSample)}    
+    
+  before do
+    @convertors = MongoMapper::Plugins::AttributeConvertors::ClassMethods::ATTRIBUTE_CONVERTORS
+    @convertors[:test_convertor] = {
+      from_string: -> s {"from_string: #{s}"},
+      to_string:   -> v {"to_string: #{v}"}
+    }
+  end
+  
+  it ":line convertor" do      
+    v = ['a', 'b']
+    str_v = 'a, b'
+    @convertors[:line][:from_string].call(str_v).should == v
+    @convertors[:line][:to_string].call(v).should == str_v
+  end
+  
+  it ":yaml convertor" do
+    v = {'a' => 'b'}
+    str_v = v.to_yaml.strip
+    
+    @convertors[:yaml][:from_string].call(str_v).should == v
+    @convertors[:yaml][:to_string].call(v).should == str_v
+  end
+  
+  it ":json convertor" do
+    v = {'a' => 'b'}
+    str_v = v.to_json.strip
+    @convertors[:json][:from_string].call(str_v).should == v
+    @convertors[:json][:to_string].call(v).should == str_v
+  end
+  
+  it ":key extension" do
+    class ::AsStringSample
+      include MongoMapper::Document
+      include MongoMapper::Plugins::Micelaneous
+      
+      key :key, String, as_string: :test_convertor
+      key :protected_key, String, as_string: :test_convertor, protected: true
+    end
+  
+    o = AsStringSample.new
+    
+    # get
+    o.key_as_string.should == 'to_string: '
+    o.key = 'value'
+    o.clear_cache
+    o.key_as_string.should == 'to_string: value'
+    
+    # set
+    o.key_as_string = ''
+    o.key.should == 'from_string: '
+    o.key_as_string = 'value'
+    o.key.should == 'from_string: value'
+    
+    # mass assignment
+    o.key = ''
+    o.update_attributes key_as_string: 'value'
+    o.key.should == 'from_string: value'
+    
+    # protection
+    o.protected_key = ''
+    o.update_attributes protected_key_as_string: 'value'
+    o.protected_key.should == ''
+  end
+end

data/spec/carrierwave_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+describe "MongoMapper & CarrierWave" do  
+  with_mongo_mapper
+  with_tmp_spec_dir
+  
+  before :all do
+    class PlaneUploader < CarrierWave::Uploader::Base
+      storage :file
+      
+      class << self
+        attr_accessor :store_dir
+      end      
+      
+      def store_dir
+        self.class.store_dir
+      end
+    end
+    
+    class Plane
+      include MongoMapper::Document
+  
+      mount_uploader :image, PlaneUploader
+    end        
+  end
+  after(:all){remove_constants :Plane, :PlaneUploader}
+  
+  before do
+    PlaneUploader.store_dir = "#{spec_dir}/data"
+  end    
+  
+  it "basic" do
+    plane = Plane.new
+    File.open "#{spec_dir}/plane.jpg" do |f|
+      plane.image = f
+      plane.save!
+    end
+    
+    plane.image.current_path.should == '/tmp/carrierwave_spec/data/plane.jpg'
+    File.should exist("#{spec_dir}/data/plane.jpg")
+  end
+end

data/spec/{scope_spec.rb → custom_scope_spec.rb}

@@ -1,29 +1,23 @@
 require 'spec_helper'
 
-require "mongo_mapper_ext/plugins/default_scope"
-
 describe "MongoMapper Default Scope" do
-  before :all do
-    MongoMapper.database = 'test'
-  end
+  with_mongo_mapper
   
-  before :each do 
+  before do 
     class ::ScopeSample
       include MongoMapper::Document
-      plugin MongoMapper::Plugins::DefaultScope
+      include MongoMapper::Plugins::CustomScope
       
       key :name, String
       key :_type, String
     end
     
     ScopeSample.delete_all
-    ScopeSample.create! :name => 'a'
-    ScopeSample.create! :name => 'b'
+    ScopeSample.create! name: 'a'
+    ScopeSample.create! name: 'b'
   end
   
-  after :each do 
-    Object.send :remove_const, :ScopeSample if Object.const_defined? :ScopeSample
-  end
+  after(:all){remove_constants :ScopeSample}
     
   it "should not affect objects without default_scope" do
     ScopeSample.count.should == 2
@@ -31,7 +25,7 @@ describe "MongoMapper Default Scope" do
   end
   
   it "default_scope" do
-    ScopeSample.send :default_scope, :name => 'a'
+    ScopeSample.send :default_scope, name: 'a'
     ScopeSample.count.should == 1
     ScopeSample.all.count.should == 1
     ScopeSample.first.name.should == 'a'
@@ -39,7 +33,7 @@ describe "MongoMapper Default Scope" do
   
   it "default_scope as block" do
     ScopeSample.send :default_scope do
-      {:name => 'a'}
+      {name: 'a'}
     end
     ScopeSample.count.should == 1
     ScopeSample.all.count.should == 1
@@ -47,12 +41,12 @@ describe "MongoMapper Default Scope" do
   end
   
   it "default_scope should be inherited" do    
-    ScopeSample.send :default_scope, :name => 'a'
+    ScopeSample.send :default_scope, name: 'a'
     class ScopeSampleAncestor < ScopeSample
     end
     
-    ScopeSampleAncestor.create! :name => 'a'
-    ScopeSampleAncestor.create! :name => 'b'
+    ScopeSampleAncestor.create! name: 'a'
+    ScopeSampleAncestor.create! name: 'b'
 
     ScopeSampleAncestor.count.should == 1
     ScopeSampleAncestor.all.count.should == 1
@@ -60,13 +54,13 @@ describe "MongoMapper Default Scope" do
   end
   
   it "ancestors should be able to override default_scope" do
-    ScopeSample.send :default_scope, :name => 'a'
+    ScopeSample.send :default_scope, name: 'a'
     class ScopeSampleAncestor2 < ScopeSample
-      default_scope :name => 'b'
+      default_scope name: 'b'
     end
     
-    ScopeSampleAncestor2.create! :name => 'a'
-    ScopeSampleAncestor2.create! :name => 'b'
+    ScopeSampleAncestor2.create! name: 'a'
+    ScopeSampleAncestor2.create! name: 'b'
         
     ScopeSampleAncestor2.count.should == 1
     ScopeSampleAncestor2.all.count.should == 1
@@ -74,13 +68,13 @@ describe "MongoMapper Default Scope" do
   end
   
   it "shouldn't allow to nest with_exclusive_scope" do
-    lambda{
+    -> {
       ScopeSample.with_exclusive_scope do
         ScopeSample.with_exclusive_scope{}
       end
     }.should raise_error(AssertionError)
     
-    lambda{
+    -> {
       ScopeSample.with_exclusive_scope do
         ScopeSample.with_scope{}
       end
@@ -88,10 +82,10 @@ describe "MongoMapper Default Scope" do
   end
       
   it "with_exclusive_scope should clear other scopes" do
-    ScopeSample.send :default_scope, :name => 'a'
+    ScopeSample.send :default_scope, name: 'a'
     
-    ScopeSample.with_scope :name => 'a' do
-      ScopeSample.with_exclusive_scope :name => 'b' do        
+    ScopeSample.with_scope name: 'a' do
+      ScopeSample.with_exclusive_scope name: 'b' do        
         ScopeSample.count.should == 1
         ScopeSample.first.name.should == 'b'
       end
@@ -99,7 +93,7 @@ describe "MongoMapper Default Scope" do
   end
   
   it "with_scope" do
-    ScopeSample.with_scope :name => 'a' do
+    ScopeSample.with_scope name: 'a' do
       ScopeSample.count.should == 1
       ScopeSample.first.name.should == 'a'
     end
@@ -109,13 +103,13 @@ describe "MongoMapper Default Scope" do
     ScopeSample.class_eval do
       key :name2, String
       key :name3, String
-      default_scope :name => 'a'
+      default_scope name: 'a'
     end
-    ScopeSample.create! :name => 'a', :name2 => 'a2', :name3 => 'a3'
-    ScopeSample.create! :name => 'b', :name2 => 'b2', :name3 => 'b3'
+    ScopeSample.create! name: 'a', name2: 'a2', name3: 'a3'
+    ScopeSample.create! name: 'b', name2: 'b2', name3: 'b3'
     
-    ScopeSample.with_scope :name2 => 'a2' do
-      ScopeSample.with_scope :name3 => 'a3' do
+    ScopeSample.with_scope name2: 'a2' do
+      ScopeSample.with_scope name3: 'a3' do
         ScopeSample.count.should == 1
         ScopeSample.first.name2.should == 'a2'
       end

data/spec/micelaneous_plugin_spec.rb

@@ -1,40 +1,27 @@
 require 'spec_helper'
 
-require "mongo_mapper_ext/micelaneous"
-require "mongo_mapper_ext/plugins/micelaneous"
-
 describe "MongoMapper Default Scope" do
+  with_mongo_mapper
   
   before :all do
     class ::Post
       include MongoMapper::Document      
-      plugin MongoMapper::Plugins::Micelaneous
+      include MongoMapper::Plugins::Micelaneous
       
-      key :comments_count, Integer, :default => 0
+      key :comments_count, Integer, default: 0
       has_many :comments
     end
     
     class ::Comment
       include MongoMapper::Document
-      plugin MongoMapper::Plugins::Micelaneous
+      include MongoMapper::Plugins::Micelaneous
       
       key :post_id
-      belongs_to :post, :counter_cache => true
-    end
-    
-    MongoMapper.database = 'test'
-  end
-  
-  after :all do
-    %w{Post Comment}.each do |obj_name|
-      Object.send :remove_const, obj_name if Object.const_defined? obj_name
+      belongs_to :post, counter_cache: true
     end
-  end
+  end  
+  after(:all){remove_constants %w{Post Comment Post2 Namespace}}
   
-  before :each do 
-    [Post, Comment].every.delete_all
-  end
-    
   it "should increase count of comments" do
     post = Post.create!
     comment = post.comments.create!
@@ -53,4 +40,31 @@ describe "MongoMapper Default Scope" do
     post.reload
     post.comments_count.should == 0
   end
+  
+  describe "model_name" do
+    it "basic" do
+      Post.model_name.should == "Post"
+      Post.model_name "SuperPost"
+      Post.model_name.should == "SuperPost"
+    end
+    
+    it "by default should be initialized from class alias" do
+      class ::Post2
+        include MongoMapper::Document      
+        include MongoMapper::Plugins::Micelaneous
+
+        self.alias 'PostAlias'
+      end
+      
+      module ::Namespace
+        class Post
+          include MongoMapper::Document      
+          include MongoMapper::Plugins::Micelaneous
+        end
+      end
+      
+      Post2.model_name.should == 'PostAlias'
+      Namespace::Post.model_name.should == 'Post' # not the Namespace::Post
+    end
+  end
 end