lib/pleiades/railtie.rb in pleiades-0.1.1 vs lib/pleiades/railtie.rb in pleiades-0.1.2

@@ -1,18 +1,10 @@
 module Pleiades
   class Railtie < ::Rails::Railtie
     generators do
-
-      generator_names = %w(
-                            install
-                            setup
-                            command
-                          )
-
-      generator_path = './generators/pleiades'.freeze
-      generator_names
-        .map  { |f| "#{generator_path}/#{f}/#{f}_generator.rb" }
+      Dir
+        .glob("#{File.expand_path('generators/pleiades', __dir__)}/*")
+        .map  { |f| "#{f}/#{File.basename(f)}_generator.rb" }
         .each { |f| require_relative f }
-
     end
   end
 end