dust-generators 0.1.3 → 0.1.4

data/rails_generators/dust_scaffold/dust_scaffold_generator.rb

@@ -0,0 +1,232 @@
+class DustScaffoldGenerator < Rails::Generator::Base
+  attr_accessor :name, :attributes, :controller_actions
+  
+  def initialize(runtime_args, runtime_options = {})
+    super
+    usage if @args.empty?
+    
+    @name = @args.first
+    @controller_actions = []
+    @attributes = []
+    
+    @args[1..-1].each do |arg|
+      if arg == '!'
+        options[:invert] = true
+      elsif arg.include? ':'
+        @attributes << Rails::Generator::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!
+    @attributes.uniq!
+    
+    if options[:invert] || @controller_actions.empty?
+      @controller_actions = all_actions - @controller_actions
+    end
+    
+    if @attributes.empty?
+      options[:skip_model] = true # default to skipping model if no attributes passed
+      if model_exists?
+        model_columns_for_attributes.each do |column|
+          @attributes << Rails::Generator::GeneratedAttribute.new(column.name.to_s, column.type.to_s)
+        end
+      else
+        @attributes << Rails::Generator::GeneratedAttribute.new('name', 'string')
+      end
+    end
+  end
+  
+  def manifest
+    record do |m|
+      unless options[:skip_model]
+        m.directory "app/models"
+        m.template "model.rb", "app/models/#{singular_name}.rb"
+        unless options[:skip_migration]
+          m.migration_template "migration.rb", "db/migrate", :migration_file_name => "create_#{plural_name}"
+        end
+        
+        if rspec?
+          m.directory "spec/models"
+          m.template "tests/#{test_framework}/model.rb", "spec/models/#{singular_name}_spec.rb"
+          m.directory "spec/fixtures"
+          m.template "fixtures.yml", "spec/fixtures/#{plural_name}.yml"
+        else
+          m.directory "test/unit"
+          m.template "tests/#{test_framework}/model.rb", "test/unit/#{singular_name}_test.rb"
+          m.directory "test/fixtures"
+          m.template "fixtures.yml", "test/fixtures/#{plural_name}.yml"
+        end
+      end
+      
+      unless options[:skip_controller]
+        m.directory "app/controllers"
+        m.template "controller.rb", "app/controllers/#{plural_name}_controller.rb"
+        
+        m.directory "app/helpers"
+        m.template "helper.rb", "app/helpers/#{plural_name}_helper.rb"
+        
+        m.directory "app/views/#{plural_name}"
+        controller_actions.each do |action|
+          if File.exist? source_path("views/#{view_language}/#{action}.html.#{view_language}")
+            m.template "views/#{view_language}/#{action}.html.#{view_language}", "app/views/#{plural_name}/#{action}.html.#{view_language}"
+          end
+        end
+      
+        if form_partial?
+          m.template "views/#{view_language}/_form.html.#{view_language}", "app/views/#{plural_name}/_form.html.#{view_language}"
+        end
+      
+        m.route_resources plural_name
+        
+        if rspec?
+          m.directory "spec/controllers"
+          m.template "tests/#{test_framework}/controller.rb", "spec/controllers/#{plural_name}_controller_spec.rb"
+        else
+          m.directory "test/functional"
+          m.template "tests/#{test_framework}/controller.rb", "test/functional/#{plural_name}_controller_test.rb"
+        end
+      end
+    end
+  end
+  
+  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? { |n| action? n.to_s }
+  end
+  
+  def singular_name
+    name.underscore
+  end
+  
+  def plural_name
+    name.underscore.pluralize
+  end
+  
+  def class_name
+    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 :partial => 'form'"
+      else
+        "<%= render :partial => '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 rspec?
+    test_framework == :rspec
+  end
+  
+protected
+  
+  def view_language
+    options[:haml] ? 'haml' : 'erb'
+  end
+  
+  def test_framework
+    options[:test_framework] ||= default_test_framework
+  end
+  
+  def default_test_framework
+    File.exist?(destination_path("spec")) ? :rspec : :testunit
+  end
+  
+  def add_options!(opt)
+    opt.separator ''
+    opt.separator 'Options:'
+    opt.on("--skip-model", "Don't generate a model or migration file.") { |v| options[:skip_model] = v }
+    opt.on("--skip-migration", "Don't generate migration file for model.") { |v| options[:skip_migration] = v }
+    opt.on("--skip-timestamps", "Don't add timestamps to migration file.") { |v| options[:skip_timestamps] = v }
+    opt.on("--skip-controller", "Don't generate controller, helper, or views.") { |v| options[:skip_controller] = v }
+    opt.on("--invert", "Generate all controller actions except these mentioned.") { |v| options[:invert] = v }
+    opt.on("--haml", "Generate HAML views instead of ERB.") { |v| options[:haml] = v }
+    opt.on("--testunit", "Use test/unit for test files.") { options[:test_framework] = :testunit }
+    opt.on("--rspec", "Use RSpec for test files.") { options[:test_framework] = :rspec }
+    opt.on("--shoulda", "Use Shoulda for test files.") { options[:test_framework] = :shoulda }
+  end
+  
+  # is there a better way to do this? Perhaps with const_defined?
+  def model_exists?
+    File.exist? destination_path("app/models/#{singular_name}.rb")
+  end
+  
+  def read_template(relative_path)
+    ERB.new(File.read(source_path(relative_path)), nil, '-').result(binding)
+  end
+  
+  def banner
+    <<-EOS
+Creates a controller and optional model given the name, actions, and attributes.
+
+USAGE: #{$0} #{spec.name} ModelName [controller_actions and model:attributes] [options]
+EOS
+  end
+end

data/rails_generators/dust_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 <%= name.underscore.humanize.downcase %>."
+      redirect_to <%= item_path('url') %>
+    else
+      render :action => 'new'
+    end
+  end

data/rails_generators/dust_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 <%= name.underscore.humanize.downcase %>."
+    redirect_to <%= items_path('url') %>
+  end

data/rails_generators/dust_scaffold/templates/actions/edit.rb

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

data/rails_generators/dust_scaffold/templates/actions/index.rb

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

data/rails_generators/dust_scaffold/templates/actions/new.rb

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

data/rails_generators/dust_scaffold/templates/actions/show.rb

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

data/rails_generators/dust_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 <%= name.underscore.humanize.downcase %>."
+      redirect_to <%= item_path('url') %>
+    else
+      render :action => 'edit'
+    end
+  end

data/rails_generators/dust_scaffold/templates/controller.rb

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

data/rails_generators/dust_scaffold/templates/fixtures.yml

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

data/rails_generators/dust_scaffold/templates/helper.rb

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

data/rails_generators/dust_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 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/rails_generators/dust_scaffold/templates/model.rb

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

data/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_scaffold/templates/tests/testunit/actions/index.rb

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

data/rails_generators/dust_scaffold/templates/tests/testunit/actions/new.rb

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

data/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_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/rails_generators/dust_scaffold/templates/views/erb/_form.html.erb

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

data/rails_generators/dust_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/rails_generators/dust_scaffold/templates/views/erb/index.html.erb

@@ -0,0 +1,29 @@
+<%% title "<%= plural_name.titleize %>" %>
+
+<table>
+  <tr>
+  <%- for attribute in attributes -%>
+    <th><%= attribute.column.human_name.titleize %></th>
+  <%- end -%>
+  </tr>
+  <%% for <%= singular_name %> in @<%= plural_name %> %>
+    <tr>
+    <%- for attribute in attributes -%>
+      <td><%%=h <%= 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/rails_generators/dust_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 -%>

data/rails_generators/dust_scaffold/templates/views/erb/show.html.erb

@@ -0,0 +1,20 @@
+<%% title "<%= singular_name.titleize %>" %>
+
+<%- for attribute in attributes -%>
+<p>
+  <strong><%= attribute.column.human_name.titleize %>:</strong>
+  <%%=h @<%= singular_name %>.<%= attribute.name %> %>
+</p>
+<%- end -%>
+
+<p>
+<%- if action? :edit -%>
+  <%%= link_to "Edit", edit_<%= singular_name %>_path(@<%= singular_name %>) %> |
+<%- end -%>
+<%- if action? :destroy -%>
+  <%%= link_to "Destroy", @<%= singular_name %>, :confirm => 'Are you sure?', :method => :delete %> |
+<%- end -%>
+<%- if action? :index -%>
+  <%%= link_to "View All", <%= plural_name %>_path %>
+<%- end -%>
+</p>