Make TemplateRunner#generate accept any number of arguments

eladmeidar · Dec 3, 2008 · ef0b1cb · ef0b1cb
1 parent 7c0dfa3
commit ef0b1cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/railties/lib/rails_generator/generators/applications/app/template_runner.rb b/railties/lib/rails_generator/generators/applications/app/template_runner.rb
@@ -233,11 +233,11 @@ def initializer(filename, data = nil, &block)
     #
     #   generate(:authenticated, "user session")
     #
-    def generate(what, args = nil)
+    def generate(what, *args)
       puts "generating #{what}"
-      args = args.join(" ") if args.class == Array
+      argument = args.map(&:to_s).flatten.join(" ")
 
-      in_root { `#{root}/script/generate #{what} #{args}` }
+      in_root { `#{root}/script/generate #{what} #{argument}` }
     end
 
     # Executes a command