Chris@909: require 'rails_generator/base'
Chris@909: require 'rails_generator/generators/components/model/model_generator'
Chris@909: 
Chris@909: class RedminePluginModelGenerator < ModelGenerator
Chris@909:   attr_accessor :plugin_path, :plugin_name, :plugin_pretty_name
Chris@909:   
Chris@909:   def initialize(runtime_args, runtime_options = {})
Chris@909:     runtime_args = runtime_args.dup
Chris@909:     usage if runtime_args.empty?
Chris@909:     @plugin_name = "redmine_" + runtime_args.shift.underscore
Chris@909:     @plugin_pretty_name = plugin_name.titleize
Chris@909:     @plugin_path = "vendor/plugins/#{plugin_name}"
Chris@909:     super(runtime_args, runtime_options)
Chris@909:   end
Chris@909:   
Chris@909:   def destination_root
Chris@909:     File.join(Rails.root, plugin_path)
Chris@909:   end
Chris@909:   
Chris@909:   def manifest
Chris@909:     record do |m|
Chris@909:       # Check for class naming collisions.
Chris@909:       m.class_collisions class_path, class_name, "#{class_name}Test"
Chris@909: 
Chris@909:       # Model, test, and fixture directories.
Chris@909:       m.directory File.join('app/models', class_path)
Chris@909:       m.directory File.join('test/unit', class_path)
Chris@909:       m.directory File.join('test/fixtures', class_path)
Chris@909: 
Chris@909:       # Model class, unit test, and fixtures.
Chris@909:       m.template 'model.rb.erb',      File.join('app/models', class_path, "#{file_name}.rb")
Chris@909:       m.template 'unit_test.rb.erb',  File.join('test/unit', class_path, "#{file_name}_test.rb")
Chris@909: 
Chris@909:       unless options[:skip_fixture] 
Chris@909:        	m.template 'fixtures.yml',  File.join('test/fixtures', "#{table_name}.yml")
Chris@909:       end
Chris@909: 
Chris@909:       unless options[:skip_migration]
Chris@909:         m.migration_template 'migration.rb.erb', 'db/migrate', :assigns => {
Chris@909:           :migration_name => "Create#{class_name.pluralize.gsub(/::/, '')}"
Chris@909:         }, :migration_file_name => "create_#{file_path.gsub(/\//, '_').pluralize}"
Chris@909:       end
Chris@909:     end
Chris@909:   end
Chris@909: end