rails--rails/railties/lib/generators/named_base.rb

75 lines
2.3 KiB
Ruby
Raw Normal View History

2009-06-23 17:10:42 +00:00
require 'generators/base'
require 'generators/generated_attribute'
module Rails
module Generators
class NamedBase < Base
2009-06-23 14:50:21 +00:00
argument :name, :type => :string
attr_reader :class_name, :singular_name, :plural_name, :table_name,
:class_path, :file_path, :class_nesting, :class_nesting_depth
2009-06-23 16:12:37 +00:00
2009-06-23 14:50:21 +00:00
alias :file_name :singular_name
def initialize(*args)
super
assign_names!
2009-06-23 16:12:37 +00:00
parse_attributes! if respond_to?(:attributes)
2009-06-23 14:50:21 +00:00
end
protected
def assign_names!
base_name, @class_path, @file_path, @class_nesting, @class_nesting_depth = extract_modules(name)
@class_name_without_nesting, @singular_name, @plural_name = inflect_names(base_name)
@table_name = if !defined?(ActiveRecord::Base) || ActiveRecord::Base.pluralize_table_names
plural_name
else
singular_name
end
@table_name.gsub! '/', '_'
if @class_nesting.empty?
@class_name = @class_name_without_nesting
else
@table_name = @class_nesting.underscore << "_" << @table_name
@class_name = "#{@class_nesting}::#{@class_name_without_nesting}"
end
end
2009-06-23 16:12:37 +00:00
# Convert attributes hash into an array with GeneratedAttribute objects.
#
def parse_attributes!
attributes.map! do |name, type|
Rails::Generator::GeneratedAttribute.new(name, type)
end
end
2009-06-23 14:50:21 +00:00
# Extract modules from filesystem-style or ruby-style path. Both
# good/fun/stuff and Good::Fun::Stuff produce the same results.
#
def extract_modules(name)
modules = name.include?('/') ? name.split('/') : name.split('::')
name = modules.pop
path = modules.map { |m| m.underscore }
file_path = (path + [name.underscore]).join('/')
nesting = modules.map { |m| m.camelize }.join('::')
[name, path, file_path, nesting, modules.size]
end
# Receives name and return camelized, underscored and pluralized names.
#
def inflect_names(name)
camel = name.camelize
under = camel.underscore
plural = under.pluralize
[camel, under, plural]
end
end
end
end