mongo_db_gen 0.3.3 → 0.3.4

data/README

@@ -15,7 +15,22 @@ Setup use of Mongoid
 
 Create new model Person with name attribute
 
-  rails g mongoid:model person name:string
+  rails g mongoid:model person name:string                           
+  
+Create document Address with attributes defaulting to string  
+
+  rails g mongoid:model address name street                           
+
+TODO:
+  It should support inheritance
+
+  class Browser < Canvas
+    field :version, :type => Integer
+  end
+
+Fx like this:
+
+  rails g mongoid:model browser version:integer --inherit canvas                           
 
 Mongo Mapper
 ------------
@@ -33,6 +48,9 @@ Create new Mongo Mapper model Person with name attribute
 
   rails g mongoid:model person name:string
 
+Create Embedded document Address with attributes defaulting to string
+  rails g mongoid:model address name street --embedded
+
 Notice
 ======
 This is currently a work in progress. Please feel free to join in the effort and fork this project and fix whatever bugs etc.

data/README.rdoc

@@ -1,8 +1,18 @@
 = Mongo DB generators
 
-The Mongo DB Rails 3 generators includes the following generators:
+The Mongo DB Rails 3 generators includes generators for both Mongo Mapper and Mongoid.
+
+Add the following to your Gemfile in your Rails 3 project:
+
+gem 'mongo_ext'
+gem 'mongoid'
+gem "mongo_mapper", :git => 'http://github.com/merbjedi/mongomapper.git', :branch => 'rails3' 
+
+$ bundle install
   
 == Mongoid  
+
+Includes the following generators
   
 * mongoid:setup       - setup Rails app to use Mongoid
 * mongoid:model       - generates a Mongoid model
@@ -19,6 +29,8 @@ Create new model Person with name attribute
 
 == Mongo Mapper
 
+Includes the following generators
+
 * mongo_mapper:setup  - setup Rails app to use Mongo Mapper
 * mongo_mapper:model  - generates a Mongo Mapper model
 

data/lib/generators/mongo_mapper/model/model_generator.rb

@@ -1,9 +1,75 @@
+require 'rails/generators/base'
+
 module MongoMapper
   module Generators
-    class ModelGenerator < Rails::Generators::NamedBase
-      
-      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"  
+    class ModelGenerator < Rails::Generators::Base
+
+      argument :model_name, :type => :string, :required => true, :banner => 'ModelName'      
+      argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"
+
+      class_option :embedded, :type => :boolean, :aliases => "-E", :default => false,
+                              :desc => "Use EmbeddedDocument for model"
+
+      attr_accessor :model_attributes
+
+      def initialize(*args, &block)
+        super
 
+        @model_attributes = []
+
+        attributes.each do |arg|
+          if arg.include?(':')
+            @model_attributes << Rails::Generators::GeneratedAttribute.new(*arg.split(':'))
+          else
+            @model_attributes << Rails::Generators::GeneratedAttribute.new(arg, "string")                        
+          end
+        end
+
+        @model_attributes.uniq!
+
+        if @model_attributes.empty?
+          if model_exists?
+            model_columns_for_attributes.each do |column|
+              @model_attributes << Rails::Generators::GeneratedAttribute.new(column.name.to_s, column.type.to_s)
+            end
+          else
+            @model_attributes << Rails::Generators::GeneratedAttribute.new('name', 'string')
+          end
+        end
+      end
+
+      def create_model_file
+        template 'model.rb', "app/models/#{singular_name}.rb"        
+      end
+
+      no_tasks do 
+        def model_exists?
+          File.exist? destination_path("app/models/#{singular_name}.rb")
+        end        
+        
+        def singular_name
+          model_name.underscore
+        end
+
+        def plural_name
+          model_name.underscore.pluralize
+        end
+
+        def class_name
+          model_name.camelize
+        end
+
+        def plural_class_name
+          plural_name.camelize
+        end 
+        
+        def model_columns_for_attributes
+          class_name.constantize.columns.reject do |column|
+            column.name.to_s =~ /^(id|created_at|updated_at)$/
+          end
+        end        
+      end 
+      
       def self.source_root
         @source_root ||= File.expand_path('../templates', __FILE__)
       end
@@ -11,12 +77,7 @@ module MongoMapper
       def self.banner
         "#{$0} mongo_mapper:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
       end
-
-      def create_model_file
-        template 'model.rb', File.join('app/models', class_path, "#{file_name}.rb")
-      end
-
-      # hook_for :test
+      
     end
   end
 end

data/lib/generators/mongo_mapper/model/templates/model.rb

@@ -0,0 +1,9 @@
+class <%= class_name %>   
+  <% if options[:embedded] %>
+  include MongoMapper::EmbeddedDocument  
+  <% else %>                   
+  include MongoMapper::Document  
+  <% end %>
+  
+  <%= model_attributes.map { |a| "key :#{a.name}, #{a.type.capitalize}" }.join("\n") %>
+end

data/lib/generators/mongo_mapper/setup/setup_generator.rb

@@ -1,12 +1,23 @@
-require 'generators/mongo_db_base'
+# require 'thor-ext'
+require 'rails/generators/base'
 
 module MongoMapper
   module Generators
     class SetupGenerator < Rails::Generators::Base            
-      include MongoDatabase::Generators::Base
+      # include ThorExtensions
 
       argument :database, :type => :string, :default => 'mongo_db_default'            
 
+      # def configure_gems
+      #   cleanup_gemfile
+      #   add_gems %w{mongo_ext mongo mongo_mapper}
+      # end
+
+      def create_files
+        template "mongo_mapper_db_config.rb" , "config/initializers/mongo_mapper_db_config.rb"
+        puts "Please ensure Gemfile contains: gem 'mongo_mapper'"
+      end
+
       def self.source_root
         @source_root ||= File.expand_path('../templates', __FILE__)
       end
@@ -15,13 +26,6 @@ module MongoMapper
         "#{$0} mongo_mapper:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
       end
 
-      def configure_gems
-        add_gems %w{mongo_ext mongo mongo_mapper}
-      end
-
-      def create_files
-        template "mongo_mapper_db_config.rb" , "config/initializers/mongo_mapper_db_config.rb" 
-      end
       
     end
   end

data/lib/generators/mongoid/model/model_generator.rb

@@ -1,8 +1,68 @@
+require 'rails/generators/base'
+
 module Mongoid
   module Generators
-    class ModelGenerator < Rails::Generators::NamedBase
+    class ModelGenerator < Rails::Generators::Base     
+      argument :model_name, :type => :string, :required => true, :banner => 'ModelName'            
       argument :attributes, :type => :array, :default => [], :banner => "field:type field:type"  
-  
+
+      class_option :inherit, :type => :string, :aliases => "-I",
+                              :desc => "Embed document by Inheriting"
+
+
+      attr_accessor :model_attributes
+    
+      def initialize(*args, &block)
+        super
+
+        @model_attributes = []
+
+        attributes.each do |arg|
+          if arg.include?(':')
+            @model_attributes << Rails::Generators::GeneratedAttribute.new(*arg.split(':'))
+          else
+            @model_attributes << Rails::Generators::GeneratedAttribute.new(arg, "string")            
+          end
+        end
+
+        @model_attributes.uniq!
+
+        if @model_attributes.empty?
+          if model_exists?
+            model_columns_for_attributes.each do |column|
+              @model_attributes << Rails::Generators::GeneratedAttribute.new(column.name.to_s, column.type.to_s)
+            end
+          else
+            @model_attributes << Rails::Generators::GeneratedAttribute.new('name', 'string')
+          end
+        end
+      end
+
+      def create_model_file
+        template 'model.rb', "app/models/#{singular_name}.rb"        
+      end
+
+      no_tasks do 
+        def model_exists?
+          File.exist? destination_path("app/models/#{singular_name}.rb")
+        end
+        def singular_name
+          model_name.underscore
+        end
+
+        def plural_name
+          model_name.underscore.pluralize
+        end
+
+        def class_name
+          model_name.camelize
+        end
+
+        def plural_class_name
+          plural_name.camelize
+        end
+      end   
+      
       def self.source_root
         @source_root ||= File.expand_path('../templates', __FILE__)
       end
@@ -10,12 +70,7 @@ module Mongoid
       def self.banner
         "#{$0} mongoid:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
       end
-  
-      def create_model_file
-        template 'model.rb', File.join('app/models', class_path, "#{file_name}.rb")
-      end
-  
-      # hook_for :test
+      
     end
   end
 end

data/lib/generators/mongoid/model/templates/model.rb

@@ -0,0 +1,5 @@
+class <%= class_name.camelize %> <%= "< #{options[:inherit]}" if options[:inherit] %>
+  include Mongoid::Document
+
+  <%= model_attributes.map { |a| "field :#{a.name}, :type => #{a.type.capitalize}" }.join("\n") %>    
+end

data/lib/generators/mongoid/setup/setup_generator.rb

@@ -1,12 +1,25 @@
-require 'generators/mongo_db_base'
+# require 'thor-ext'
+require 'rails/generators/base'
 
 module Mongoid
   module Generators
     class SetupGenerator < Rails::Generators::Base            
-      include MongoDatabase::Generators::Base
+      # include ThorExtensions
 
       argument :database, :type => :string, :default => 'mongo_db_default'            
 
+      # def configure_gems
+      #   cleanup_gemfile
+      #   add_gems %w{mongo_ext mongo mongoid}
+      # end
+
+      def create_files
+        template "mongoid.rb" , "config/initializers/mongoid.rb" 
+        template 'database.mongo.yml', "config/database.mongo.yml"   
+
+        puts "Please ensure Gemfile contains: gem 'mongoid'"
+      end
+
       def self.source_root
         @source_root ||= File.expand_path('../templates', __FILE__)
       end
@@ -15,15 +28,6 @@ module Mongoid
         "#{$0} mongoid:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
       end
 
-      def configure_gems
-        add_gems %w{mongo_ext mongo mongoid}
-      end
-
-      def create_files
-        template "mongoid.rb" , "config/initializers/mongoid.rb" 
-        template 'database.mongo.yml', "config/database.mongo.yml"
-      end
-
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: mongo_db_gen
 version: !ruby/object:Gem::Version 
-  version: 0.3.3
+  version: 0.3.4
 platform: ruby
 authors: 
 - Kristian Mandrup
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-02-07 00:00:00 +01:00
+date: 2010-02-21 00:00:00 -05:00
 default_executable: mongomap_model
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -50,11 +50,12 @@ files:
 - bin/mongoid_setup
 - bin/mongomapper_model
 - bin/mongomapper_setup
-- lib/generators/mongo_db_base.rb
 - lib/generators/mongo_mapper/model/model_generator.rb
+- lib/generators/mongo_mapper/model/templates/model.rb
 - lib/generators/mongo_mapper/setup/setup_generator.rb
 - lib/generators/mongo_mapper/setup/templates/mongo_mapper_db_config.rb
 - lib/generators/mongoid/model/model_generator.rb
+- lib/generators/mongoid/model/templates/model.rb
 - lib/generators/mongoid/setup/setup_generator.rb
 - lib/generators/mongoid/setup/templates/database.mongo.yml
 - lib/generators/mongoid/setup/templates/mongoid.rb

data/lib/generators/mongo_db_base.rb

@@ -1,15 +0,0 @@
-require 'thor-ext'
-
-module MongoDatabase
-  module Generators
-    module Base
-      include ThorExtensions
-          
-      def configure_gems
-        cleanup_gemfile        
-        add_gems(%w{mongo mongo-ext mongoid})        
-      end
-    end
-  end
-end
-