blob: b53071380511f20b62de3e28bc7274048d15d086 (
plain) (
tree)
|
|
require 'generators/actions'
module Rails
module Generators
class Error < Thor::Error
end
class Base < Thor::Group
include Rails::Generators::Actions
include Thor::Actions
# Automatically sets the source root based on the class name.
#
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.
#
def self.namespace(name=nil)
if name
super
else
@namespace ||= "#{base_name}:#{generator_name}"
end
end
protected
# Use Rails default banner.
#
def self.banner
"#{$0} #{generator_name} #{self.arguments.map(&:usage).join(' ')} [options]"
end
# Sets the base_name. Overwriten by test unit generators.
#
def self.base_name
'rails'
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
klass_name.gsub!(/Generator$/, '')
klass_name.underscore
end
end
# Small macro to add ruby as an option to the generator with proper
# default value plus an instance helper method called shebang.
#
def self.add_shebang_option!
require 'rbconfig'
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"
no_tasks do
define_method :shebang do
"#!#{options[:ruby] || "/usr/bin/env ruby"}"
end
end
end
# Small macro to add test_framework option and invoke it.
#
def self.add_test_framework_option!
# TODO Reduce the example name
class_option :test_framework, :type => :string, :aliases => "-t", :default => "test_unit",
:desc => "Test framework to be invoked by this generator"
define_method :invoke_test_framework do
return unless options[:test_framework]
name = "#{options[:test_framework]}:#{self.class.generator_name}"
begin
invoke name
rescue Thor::UndefinedTaskError # TODO Ensure this message is called.
say "Could not find and/or invoke #{name}."
end
end
end
end
end
end
|