require 'generators/erb'

module Erb
  module Generators
    class ScaffoldGenerator < Base
      include Rails::Generators::ScaffoldBase

      argument :attributes, :type => :hash, :default => {}, :banner => "field:type field:type"

      class_option :singleton, :type => :boolean, :desc => "Supply to skip index action"
      class_option :layout, :type => :boolean

      def create_root_folder
        empty_directory File.join("app/views", controller_file_path)
      end

      def copy_index_file
        return if options[:singleton]
        copy_view :index
      end

      def copy_edit_file
        copy_view :edit
      end

      def copy_show_file
        copy_view :show
      end

      def copy_new_file
        copy_view :new
      end

      def copy_layout_file
        return unless options[:layout]
        template "layout.html.erb",
                 File.join("app/views/layouts", controller_class_path, "#{controller_file_name}.html.erb")
      end

      protected

        def copy_view(view)
          template "#{view}.html.erb", File.join("app/views", controller_file_path, "#{view}.html.erb")
        end

    end
  end
end
