corntrace-nifty-generators 0.4.0

data/lib/generators/nifty/scaffold/scaffold_generator.rb

@@ -0,0 +1,241 @@
+require 'generators/nifty'
+require 'rails/generators/migration'
+require 'rails/generators/generated_attribute'
+
+module Nifty
+  module Generators
+    class ScaffoldGenerator < Base
+      include Rails::Generators::Migration
+      no_tasks { attr_accessor :model_name, :model_attributes, :controller_actions }
+
+      argument :model_name, :type => :string, :required => true, :banner => 'ModelName'
+      argument :args_for_c_m, :type => :array, :default => [], :banner => 'controller_actions and model:attributes'
+
+      class_option :skip_model, :desc => 'Don\'t generate a model or migration file.', :type => :boolean
+      class_option :skip_migration, :desc => 'Dont generate migration file for model.', :type => :boolean
+      class_option :skip_timestamps, :desc => 'Don\'t add timestamps to migration file.', :type => :boolean
+      class_option :skip_controller, :desc => 'Don\'t generate controller, helper, or views.', :type => :boolean
+      class_option :invert, :desc => 'Generate all controller actions except these mentioned.', :type => :boolean
+      class_option :haml, :desc => 'Generate HAML views instead of ERB.', :type => :boolean
+
+      class_option :testunit, :desc => 'Use test/unit for test files.', :group => 'Test framework', :type => :boolean
+      class_option :rspec, :desc => 'Use RSpec for test files.', :group => 'Test framework', :type => :boolean
+      class_option :shoulda, :desc => 'Use shoulda for test files.', :group => 'Test framework', :type => :boolean
+
+      def initialize(*args, &block)
+        super
+
+        @controller_actions = []
+        @model_attributes = []
+
+        args_for_c_m.each do |arg|
+          if arg == '!'
+            options[:invert] = true
+          elsif arg.include?(':')
+            @model_attributes << Rails::Generators::GeneratedAttribute.new(*arg.split(':'))
+          else
+            @controller_actions << arg
+            @controller_actions << 'create' if arg == 'new'
+            @controller_actions << 'update' if arg == 'edit'
+          end
+        end
+
+        @controller_actions.uniq!
+        @model_attributes.uniq!
+
+        if options.invert? || @controller_actions.empty?
+          @controller_actions = all_actions - @controller_actions
+        end
+
+        if @model_attributes.empty?
+          options[:skip_model] = true # default to skipping model if no attributes passed
+          if model_exists?
+            model_columns_for_attributes.each do |column|
+              @model_attributes << Rails::Generators::GeneratedAttribute.new(column.name.to_s, column.type.to_s)
+            end
+          else
+            @model_attributes << Rails::Generators::GeneratedAttribute.new('name', 'string')
+          end
+        end
+      end
+
+      def create_model
+        unless options.skip_model?
+          template 'model.rb', "app/models/#{singular_name}.rb"
+          if options.rspec?
+            template "tests/rspec/model.rb", "spec/models/#{singular_name}_spec.rb"
+            template 'fixtures.yml', "spec/fixtures/#{plural_name}.yml"
+          else
+            template "tests/#{test_framework}/model.rb", "test/unit/#{singular_name}_test.rb"
+            template 'fixtures.yml', "test/fixtures/#{plural_name}.yml"
+          end
+        end
+      end
+
+      def create_migration
+        unless options.skip_model? || options.skip_migration?
+          migration_template 'migration.rb', "db/migrate/create_#{plural_name}.rb"
+        end
+      end
+
+      def create_controller
+        unless options.skip_controller?
+          template 'controller.rb', "app/controllers/#{plural_name}_controller.rb"
+
+          template 'helper.rb', "app/helpers/#{plural_name}_helper.rb"
+
+          controller_actions.each do |action|
+            if %w[index show new edit].include?(action) # Actions with templates
+              template "views/#{view_language}/#{action}.html.#{view_language}", "app/views/#{plural_name}/#{action}.html.#{view_language}"
+            end
+          end
+
+          if form_partial?
+            template "views/#{view_language}/_form.html.#{view_language}", "app/views/#{plural_name}/_form.html.#{view_language}"
+          end
+
+          route "resources #{plural_name.to_sym.inspect}"
+
+          if options.rspec?
+            template "tests/#{test_framework}/controller.rb", "spec/controllers/#{plural_name}_controller_spec.rb"
+          else
+            template "tests/#{test_framework}/controller.rb", "test/functional/#{plural_name}_controller_test.rb"
+          end
+        end
+      end
+
+      private
+      
+      def form_partial?
+        actions? :new, :edit
+      end
+
+      def all_actions
+        %w[index show new create edit update destroy]
+      end
+
+      def action?(name)
+        controller_actions.include? name.to_s
+      end
+
+      def actions?(*names)
+        names.all? { |name| action? name }
+      end
+
+      def singular_name
+        model_name.underscore
+      end
+
+      def plural_name
+        model_name.underscore.pluralize
+      end
+
+      def class_name
+        model_name.camelize
+      end
+
+      def plural_class_name
+        plural_name.camelize
+      end
+
+      def controller_methods(dir_name)
+        controller_actions.map do |action|
+          read_template("#{dir_name}/#{action}.rb")
+        end.join("  \n").strip
+      end
+
+      def render_form
+        if form_partial?
+          if options.haml?
+            "= render 'form'"
+          else
+            "<%= render 'form' %>"
+          end
+        else
+          read_template("views/#{view_language}/_form.html.#{view_language}")
+        end
+      end
+
+      def items_path(suffix = 'path')
+        if action? :index
+          "#{plural_name}_#{suffix}"
+        else
+          "root_#{suffix}"
+        end
+      end
+
+      def item_path(suffix = 'path')
+        if action? :show
+          "@#{singular_name}"
+        else
+          items_path(suffix)
+        end
+      end
+
+      def item_path_for_spec(suffix = 'path')
+        if action? :show
+          "#{singular_name}_#{suffix}(assigns[:#{singular_name}])"
+        else
+          items_path(suffix)
+        end
+      end
+
+      def item_path_for_test(suffix = 'path')
+        if action? :show
+          "#{singular_name}_#{suffix}(assigns(:#{singular_name}))"
+        else
+          items_path(suffix)
+        end
+      end
+
+      def model_columns_for_attributes
+        class_name.constantize.columns.reject do |column|
+          column.name.to_s =~ /^(id|created_at|updated_at)$/
+        end
+      end
+
+      def view_language
+        options.haml? ? 'haml' : 'erb'
+      end
+
+      def test_framework
+        return @test_framework if defined?(@test_framework)
+        if options.testunit?
+          return @test_framework = :testunit
+        elsif options.rspec?
+          return @test_framework = :rspec
+        elsif options.shoulda?
+          return @test_framework = :shoulda
+        else
+          return @test_framework = default_test_framework
+        end
+      end
+
+      def default_test_framework
+        File.exist?(destination_path("spec")) ? :rspec : :testunit
+      end
+
+      def model_exists?
+        File.exist? destination_path("app/models/#{singular_name}.rb")
+      end
+
+      def read_template(relative_path)
+        ERB.new(File.read(find_in_source_paths(relative_path)), nil, '-').result(binding)
+      end
+
+      def destination_path(path)
+        File.join(destination_root, path)
+      end
+
+      # FIXME: Should be proxied to ActiveRecord::Generators::Base
+      # Implement the required interface for Rails::Generators::Migration.
+      def self.next_migration_number(dirname) #:nodoc:
+        if ActiveRecord::Base.timestamped_migrations
+          Time.now.utc.strftime("%Y%m%d%H%M%S")
+        else
+          "%.3d" % (current_migration_number(dirname) + 1)
+        end
+      end
+    end
+  end
+end

data/lib/generators/nifty/scaffold/templates/actions/create.rb

@@ -0,0 +1,9 @@
+  def create
+    @<%= singular_name %> = <%= class_name %>.new(params[:<%= singular_name %>])
+    if @<%= singular_name %>.save
+      flash[:notice] = "Successfully created <%= model_name.underscore.humanize.downcase %>."
+      redirect_to <%= item_path('url') %>
+    else
+      render :action => 'new'
+    end
+  end

data/lib/generators/nifty/scaffold/templates/actions/destroy.rb

@@ -0,0 +1,6 @@
+  def destroy
+    @<%= singular_name %> = <%= class_name %>.find(params[:id])
+    @<%= singular_name %>.destroy
+    flash[:notice] = "Successfully destroyed <%= model_name.underscore.humanize.downcase %>."
+    redirect_to <%= items_path('url') %>
+  end

data/lib/generators/nifty/scaffold/templates/actions/edit.rb

@@ -0,0 +1,3 @@
+  def edit
+    @<%= singular_name %> = <%= class_name %>.find(params[:id])
+  end

data/lib/generators/nifty/scaffold/templates/actions/index.rb

@@ -0,0 +1,3 @@
+  def index
+    @<%= plural_name %> = <%= class_name %>.all
+  end

data/lib/generators/nifty/scaffold/templates/actions/new.rb

@@ -0,0 +1,3 @@
+  def new
+    @<%= singular_name %> = <%= class_name %>.new
+  end

data/lib/generators/nifty/scaffold/templates/actions/show.rb

@@ -0,0 +1,3 @@
+  def show
+    @<%= singular_name %> = <%= class_name %>.find(params[:id])
+  end

data/lib/generators/nifty/scaffold/templates/actions/update.rb

@@ -0,0 +1,9 @@
+  def update
+    @<%= singular_name %> = <%= class_name %>.find(params[:id])
+    if @<%= singular_name %>.update_attributes(params[:<%= singular_name %>])
+      flash[:notice] = "Successfully updated <%= model_name.underscore.humanize.downcase %>."
+      redirect_to <%= item_path('url') %>
+    else
+      render :action => 'edit'
+    end
+  end

data/lib/generators/nifty/scaffold/templates/controller.rb

@@ -0,0 +1,3 @@
+class <%= plural_class_name %>Controller < ApplicationController
+  <%= controller_methods :actions %>
+end

data/lib/generators/nifty/scaffold/templates/fixtures.yml

@@ -0,0 +1,9 @@
+one:
+<%- for attribute in model_attributes -%>
+  <%= attribute.name %>: <%= attribute.default %>
+<%- end -%>
+
+two:
+<%- for attribute in model_attributes -%>
+  <%= attribute.name %>: <%= attribute.default %>
+<%- end -%>

data/lib/generators/nifty/scaffold/templates/helper.rb

@@ -0,0 +1,2 @@
+module <%= plural_class_name %>Helper
+end

data/lib/generators/nifty/scaffold/templates/migration.rb

@@ -0,0 +1,16 @@
+class Create<%= plural_class_name %> < ActiveRecord::Migration
+  def self.up
+    create_table :<%= plural_name %> do |t|
+    <%- for attribute in model_attributes -%>
+      t.<%= attribute.type %> :<%= attribute.name %>
+    <%- end -%>
+    <%- unless options[:skip_timestamps] -%>
+      t.timestamps
+    <%- end -%>
+    end
+  end
+  
+  def self.down
+    drop_table :<%= plural_name %>
+  end
+end

data/lib/generators/nifty/scaffold/templates/model.rb

@@ -0,0 +1,3 @@
+class <%= class_name %> < ActiveRecord::Base
+  attr_accessible <%= model_attributes.map { |a| ":#{a.name}" }.join(", ") %>
+end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/create.rb

@@ -0,0 +1,11 @@
+  it "create action should render new template when model is invalid" do
+    <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+    post :create
+    response.should render_template(:new)
+  end
+  
+  it "create action should redirect when model is valid" do
+    <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+    post :create
+    response.should redirect_to(<%= item_path_for_spec('url') %>)
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/destroy.rb

@@ -0,0 +1,6 @@
+  it "destroy action should destroy model and redirect to index action" do
+    <%= singular_name %> = <%= class_name %>.first
+    delete :destroy, :id => <%= singular_name %>
+    response.should redirect_to(<%= items_path('url') %>)
+    <%= class_name %>.exists?(<%= singular_name %>.id).should be_false
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/edit.rb

@@ -0,0 +1,4 @@
+  it "edit action should render edit template" do
+    get :edit, :id => <%= class_name %>.first
+    response.should render_template(:edit)
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/index.rb

@@ -0,0 +1,4 @@
+  it "index action should render index template" do
+    get :index
+    response.should render_template(:index)
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/new.rb

@@ -0,0 +1,4 @@
+  it "new action should render new template" do
+    get :new
+    response.should render_template(:new)
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/show.rb

@@ -0,0 +1,4 @@
+  it "show action should render show template" do
+    get :show, :id => <%= class_name %>.first
+    response.should render_template(:show)
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/actions/update.rb

@@ -0,0 +1,11 @@
+  it "update action should render edit template when model is invalid" do
+    <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+    put :update, :id => <%= class_name %>.first
+    response.should render_template(:edit)
+  end
+  
+  it "update action should redirect when model is valid" do
+    <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+    put :update, :id => <%= class_name %>.first
+    response.should redirect_to(<%= item_path_for_spec('url') %>)
+  end

data/lib/generators/nifty/scaffold/templates/tests/rspec/controller.rb

@@ -0,0 +1,8 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+ 
+describe <%= plural_class_name %>Controller do
+  fixtures :all
+  integrate_views
+  
+  <%= controller_methods 'tests/rspec/actions' %>
+end

data/lib/generators/nifty/scaffold/templates/tests/rspec/model.rb

@@ -0,0 +1,7 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe <%= class_name %> do
+  it "should be valid" do
+    <%= class_name %>.new.should be_valid
+  end
+end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/create.rb

@@ -0,0 +1,13 @@
+  context "create action" do
+    should "render new template when model is invalid" do
+      <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+      post :create
+      assert_template 'new'
+    end
+    
+    should "redirect when model is valid" do
+      <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+      post :create
+      assert_redirected_to <%= item_path_for_test('url') %>
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/destroy.rb

@@ -0,0 +1,8 @@
+  context "destroy action" do
+    should "destroy model and redirect to index action" do
+      <%= singular_name %> = <%= class_name %>.first
+      delete :destroy, :id => <%= singular_name %>
+      assert_redirected_to <%= items_path('url') %>
+      assert !<%= class_name %>.exists?(<%= singular_name %>.id)
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/edit.rb

@@ -0,0 +1,6 @@
+  context "edit action" do
+    should "render edit template" do
+      get :edit, :id => <%= class_name %>.first
+      assert_template 'edit'
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/index.rb

@@ -0,0 +1,6 @@
+  context "index action" do
+    should "render index template" do
+      get :index
+      assert_template 'index'
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/new.rb

@@ -0,0 +1,6 @@
+  context "new action" do
+    should "render new template" do
+      get :new
+      assert_template 'new'
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/show.rb

@@ -0,0 +1,6 @@
+  context "show action" do
+    should "render show template" do
+      get :show, :id => <%= class_name %>.first
+      assert_template 'show'
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/actions/update.rb

@@ -0,0 +1,13 @@
+  context "update action" do
+    should "render edit template when model is invalid" do
+      <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+      put :update, :id => <%= class_name %>.first
+      assert_template 'edit'
+    end
+  
+    should "redirect when model is valid" do
+      <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+      put :update, :id => <%= class_name %>.first
+      assert_redirected_to <%= item_path_for_test('url') %>
+    end
+  end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/controller.rb

@@ -0,0 +1,5 @@
+require 'test_helper'
+
+class <%= plural_class_name %>ControllerTest < ActionController::TestCase
+  <%= controller_methods 'tests/shoulda/actions' %>
+end

data/lib/generators/nifty/scaffold/templates/tests/shoulda/model.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class <%= class_name %>Test < ActiveSupport::TestCase
+  should "be valid" do
+    assert <%= class_name %>.new.valid?
+  end
+end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/create.rb

@@ -0,0 +1,11 @@
+  def test_create_invalid
+    <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+    post :create
+    assert_template 'new'
+  end
+  
+  def test_create_valid
+    <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+    post :create
+    assert_redirected_to <%= item_path_for_test('url') %>
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/destroy.rb

@@ -0,0 +1,6 @@
+  def test_destroy
+    <%= singular_name %> = <%= class_name %>.first
+    delete :destroy, :id => <%= singular_name %>
+    assert_redirected_to <%= items_path('url') %>
+    assert !<%= class_name %>.exists?(<%= singular_name %>.id)
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/edit.rb

@@ -0,0 +1,4 @@
+  def test_edit
+    get :edit, :id => <%= class_name %>.first
+    assert_template 'edit'
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/index.rb

@@ -0,0 +1,4 @@
+  def test_index
+    get :index
+    assert_template 'index'
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/new.rb

@@ -0,0 +1,4 @@
+  def test_new
+    get :new
+    assert_template 'new'
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/show.rb

@@ -0,0 +1,4 @@
+  def test_show
+    get :show, :id => <%= class_name %>.first
+    assert_template 'show'
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/actions/update.rb

@@ -0,0 +1,11 @@
+  def test_update_invalid
+    <%= class_name %>.any_instance.stubs(:valid?).returns(false)
+    put :update, :id => <%= class_name %>.first
+    assert_template 'edit'
+  end
+  
+  def test_update_valid
+    <%= class_name %>.any_instance.stubs(:valid?).returns(true)
+    put :update, :id => <%= class_name %>.first
+    assert_redirected_to <%= item_path_for_test('url') %>
+  end

data/lib/generators/nifty/scaffold/templates/tests/testunit/controller.rb

@@ -0,0 +1,5 @@
+require 'test_helper'
+
+class <%= plural_class_name %>ControllerTest < ActionController::TestCase
+  <%= controller_methods 'tests/testunit/actions' %>
+end

data/lib/generators/nifty/scaffold/templates/tests/testunit/model.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+class <%= class_name %>Test < ActiveSupport::TestCase
+  def test_should_be_valid
+    assert <%= class_name %>.new.valid?
+  end
+end

data/lib/generators/nifty/scaffold/templates/views/erb/_form.html.erb

@@ -0,0 +1,10 @@
+<%%= form_for @<%= singular_name %> do |f| %>
+  <%%= f.error_messages %>
+<%- for attribute in model_attributes -%>
+  <p>
+    <%%= f.label :<%= attribute.name %> %><br />
+    <%%= f.<%= attribute.field_type %> :<%= attribute.name %> %>
+  </p>
+<%- end -%>
+  <p><%%= f.submit %></p>
+<%% end %>

data/lib/generators/nifty/scaffold/templates/views/erb/edit.html.erb

@@ -0,0 +1,14 @@
+<%% title "Edit <%= singular_name.titleize %>" %>
+
+<%= render_form %>
+
+<%- if actions? :show, :index -%>
+<p>
+<%- if action? :show -%>
+  <%%= link_to "Show", @<%= singular_name %> %> |
+<%- end -%>
+<%- if action? :index -%>
+  <%%= link_to "View All", <%= plural_name %>_path %>
+<%- end -%>
+</p>
+<%- end -%>

data/lib/generators/nifty/scaffold/templates/views/erb/index.html.erb

@@ -0,0 +1,29 @@
+<%% title "<%= plural_name.titleize %>" %>
+
+<table>
+  <tr>
+  <%- for attribute in model_attributes -%>
+    <th><%= attribute.human_name.titleize %></th>
+  <%- end -%>
+  </tr>
+  <%% for <%= singular_name %> in @<%= plural_name %> %>
+    <tr>
+    <%- for attribute in model_attributes -%>
+      <td><%%= <%= singular_name %>.<%= attribute.name %> %></td>
+    <%- end -%>
+    <%- if action? :show -%>
+      <td><%%= link_to "Show", <%= singular_name %> %></td>
+    <%- end -%>
+    <%- if action? :edit -%>
+      <td><%%= link_to "Edit", edit_<%= singular_name %>_path(<%= singular_name %>) %></td>
+    <%- end -%>
+    <%- if action? :destroy -%>
+      <td><%%= link_to "Destroy", <%= singular_name %>, :confirm => 'Are you sure?', :method => :delete %></td>
+    <%- end -%>
+    </tr>
+  <%% end %>
+</table>
+
+<%- if action? :new -%>
+<p><%%= link_to "New <%= singular_name.titleize %>", new_<%= singular_name %>_path %></p>
+<%- end -%>

data/lib/generators/nifty/scaffold/templates/views/erb/new.html.erb

@@ -0,0 +1,7 @@
+<%% title "New <%= singular_name.titleize %>" %>
+
+<%= render_form %>
+
+<%- if action? :index -%>
+<p><%%= link_to "Back to List", <%= plural_name %>_path %></p>
+<%- end -%>