lib/generators/scaffold_view/install_generator.rb in actionscaffold-0.2.2.pre vs lib/generators/scaffold_view/install_generator.rb in actionscaffold-0.2.4.pre

@@ -1,19 +1,25 @@
 #
-#  controller_generator.rb
+#  install_generator.rb
 #
 #  Created by Marko Tunjic on 15/07/16.
 #  Copyright © 2016 Marko Tunjic. All rights reserved.
 #
 module ScaffoldView
    module Generators
     class InstallGenerator < Rails::Generators::Base
+
+      # TODO: add opt args
+      ui = "bootstrap"
       desc "This generator override default scaffold generator for views."
-      source_root File.expand_path("../templates", __FILE__)
+      source_root File.expand_path("../templates/#{ui}", __FILE__)
+
       def copy_template_file
-        copy_file "_form.html.erb",
-          "lib/templates/erb/scaffold/_form.html.erb"
+        %w(index show _form).each do |template|
+          copy_file "#{template}.html.erb",
+            "lib/templates/erb/scaffold/#{template}.html.erb"
+        end
       end
+
     end
   end
 end
-