提交 a748bb79 编写于 作者: J José Valim

Get base_name from class_name.

上级 0ed602b3
......@@ -10,7 +10,7 @@ end
if ARGV.size == 0
puts "Please select a generator. Options: foo, bar"
return
else ARGV.size == 1
elsif ARGV.size == 1
ARGV << "--help"
end
......@@ -22,7 +22,7 @@ end
name = ARGV.shift
if klass = Thor::Util.find_by_namespace("rails:#{name}")
if klass = Thor::Util.find_by_namespace("rails:generators:#{name}")
klass.start
elsif klass = Thor::Util.find_by_namespace(name)
klass.start
......
......@@ -15,13 +15,15 @@ def self.source_root
@source_root ||= File.expand_path(File.join(File.dirname(__FILE__), base_name, generator_name, 'templates'))
end
# Convenience method to get the namespace from the class name.
# Convenience method to get the namespace from the class name. It's the
# same as Thor default except that the Generator at the end of the class
# is removed.
#
def self.namespace(name=nil)
def self.namespace(name=nil) #:nodoc:
if name
super
else
@namespace ||= "#{base_name}:#{generator_name}"
@namespace ||= "#{base_name}:generators:#{generator_name}"
end
end
......@@ -33,18 +35,15 @@ def self.banner
"#{$0} #{generator_name} #{self.arguments.map(&:usage).join(' ')} [options]"
end
# Sets the base_name. Overwriten by test unit generators.
# Sets the base_name taking into account the current class namespace.
#
def self.base_name
'rails'
def self.base_name #:nodoc:
@base_name ||= self.name.split('::').first.underscore
end
# Removes the namespaces and get the generator name. For example,
# Rails::Generators::MetalGenerator will return "metal" as generator name.
#
# The name is used to set the namespace (in this case "rails:metal")
# and to set the source root ("rails/metal/templates").
#
def self.generator_name
@generator_name ||= begin
klass_name = self.name.split('::').last
......@@ -61,7 +60,7 @@ def self.add_shebang_option!
default = File.join(Config::CONFIG['bindir'], Config::CONFIG['ruby_install_name'])
class_option :ruby, :type => :string, :aliases => "-r", :default => default,
:desc => "Path to the Ruby binary of your choice"
:desc => "Path to the Ruby binary of your choice", :banner => "PATH"
no_tasks do
define_method :shebang do
......@@ -78,7 +77,7 @@ def self.add_test_framework_option!
define_method :invoke_test_framework do
return unless options[:test_framework]
name = "#{options[:test_framework]}:#{self.class.generator_name}"
name = "#{options[:test_framework]}:generators:#{self.class.generator_name}"
begin
invoke name
......
......@@ -3,10 +3,6 @@
module TestUnit
module Generators
class Base < Rails::Generators::NamedBase
protected
def self.base_name
'test_unit'
end
end
end
end
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册