2009-06-23 17:10:42 +00:00
|
|
|
require 'generators/base'
|
|
|
|
require 'generators/generated_attribute'
|
2009-06-23 14:19:23 +00:00
|
|
|
|
|
|
|
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
|
|
|
|
|
2009-06-26 08:40:02 +00:00
|
|
|
class << self
|
|
|
|
# Add a class collisions name to be checked on class initialization. You
|
|
|
|
# can supply a hash with a :prefix or :suffix to be tested.
|
|
|
|
#
|
|
|
|
# ==== Examples
|
|
|
|
#
|
|
|
|
# check_class_collision :suffix => "Observer"
|
|
|
|
#
|
|
|
|
# If the generator is invoked with class name Admin, it will check for
|
|
|
|
# the presence of "AdminObserver".
|
|
|
|
#
|
|
|
|
def check_class_collision(options={})
|
|
|
|
@class_collisions = options
|
|
|
|
end
|
|
|
|
|
|
|
|
# Returns the class collisions for this class and retreives one from
|
|
|
|
# superclass. The from_superclass method used below is from Thor.
|
|
|
|
#
|
|
|
|
def class_collisions #:nodoc:
|
2009-06-27 19:26:53 +00:00
|
|
|
@class_collisions ||= from_superclass(:class_collisions, nil)
|
2009-06-26 08:40:02 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def initialize(*args) #:nodoc:
|
2009-06-23 14:50:21 +00:00
|
|
|
super
|
2009-06-26 07:53:53 +00:00
|
|
|
assign_names!(self.name)
|
2009-06-23 16:12:37 +00:00
|
|
|
parse_attributes! if respond_to?(:attributes)
|
2009-06-26 08:40:02 +00:00
|
|
|
|
|
|
|
if self.class.class_collisions
|
|
|
|
value = add_prefix_and_suffix(class_name, self.class.class_collisions)
|
|
|
|
class_collisions(value)
|
|
|
|
end
|
2009-06-23 14:50:21 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
protected
|
|
|
|
|
2009-06-26 08:40:02 +00:00
|
|
|
def assign_names!(given_name) #:nodoc:
|
2009-06-26 07:53:53 +00:00
|
|
|
self.name, @class_path, @file_path, @class_nesting, @class_nesting_depth = extract_modules(given_name)
|
|
|
|
@class_name_without_nesting, @singular_name, @plural_name = inflect_names(self.name)
|
2009-06-23 14:50:21 +00:00
|
|
|
|
|
|
|
@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.
|
|
|
|
#
|
2009-06-26 08:40:02 +00:00
|
|
|
def parse_attributes! #:nodoc:
|
2009-06-27 11:03:07 +00:00
|
|
|
self.attributes = (attributes || {}).map do |name, type|
|
|
|
|
Rails::Generators::GeneratedAttribute.new(name, type)
|
2009-06-23 16:12:37 +00:00
|
|
|
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.
|
|
|
|
#
|
2009-06-26 08:40:02 +00:00
|
|
|
def extract_modules(name) #:nodoc:
|
2009-06-23 14:50:21 +00:00
|
|
|
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.
|
|
|
|
#
|
2009-06-26 08:40:02 +00:00
|
|
|
def inflect_names(name) #:nodoc:
|
2009-06-23 14:50:21 +00:00
|
|
|
camel = name.camelize
|
|
|
|
under = camel.underscore
|
|
|
|
plural = under.pluralize
|
|
|
|
[camel, under, plural]
|
|
|
|
end
|
|
|
|
|
2009-06-26 08:40:02 +00:00
|
|
|
# Receives a name and add suffix and prefix values frrm hash.
|
|
|
|
#
|
|
|
|
def add_prefix_and_suffix(name, hash) #:nodoc:
|
|
|
|
"#{hash[:prefix]}#{name}#{hash[:suffix]}"
|
|
|
|
end
|
|
|
|
|
2009-06-23 14:19:23 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|