bpokosh-nifty-generators 0.1.3

data/rails_generators/nifty_scaffold/nifty_scaffold_generator.rb

@@ -0,0 +1,205 @@
+class NiftyScaffoldGenerator < 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 spec_dir?
+          m.directory "spec/models"
+          m.template "model_spec.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 "model_test.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/#{action}.html.haml")
+            m.template "views/#{action}.html.haml", "app/views/#{plural_name}/#{action}.html.haml"
+          end
+        end
+      
+        if form_partial?
+          m.template "views/_form.html.haml", "app/views/#{plural_name}/_form.html.haml"
+        end
+      
+        m.route_resources plural_name
+        
+        if spec_dir?
+          m.directory "spec/controllers"
+          m.template "controller_spec.rb", "spec/controllers/#{plural_name}_controller_spec.rb"
+        else
+          m.directory "test/functional"
+          m.template "controller_test.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?
+      "= render :partial => 'form'"
+    else
+      read_template("views/_form.html.haml")
+    end
+  end
+  
+  def item_path(suffix = 'path')
+    if action? :show
+      "@#{singular_name}"
+    else
+      "#{plural_name}_#{suffix}"
+    end
+  end
+  
+  def item_path_for_spec(suffix = 'path')
+    if action? :show
+      "#{singular_name}_#{suffix}(assigns[:#{singular_name}])"
+    else
+      "#{plural_name}_#{suffix}"
+    end
+  end
+  
+  def item_path_for_test(suffix = 'path')
+    if action? :show
+      "#{singular_name}_#{suffix}(assigns(:#{singular_name}))"
+    else
+      "#{plural_name}_#{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 spec_dir?
+    File.exist? destination_path("spec")
+  end
+  
+protected
+
+  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 }
+  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/nifty_scaffold/templates/action_specs/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 to index action 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/nifty_scaffold/templates/action_specs/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(<%= plural_name %>_url)
+    <%= class_name %>.exists?(<%= singular_name %>.id).should be_false
+  end

data/rails_generators/nifty_scaffold/templates/action_specs/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/nifty_scaffold/templates/action_specs/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/nifty_scaffold/templates/action_specs/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/nifty_scaffold/templates/action_specs/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/nifty_scaffold/templates/action_specs/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 to show action 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/nifty_scaffold/templates/action_tests/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/nifty_scaffold/templates/action_tests/destroy.rb

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

data/rails_generators/nifty_scaffold/templates/action_tests/edit.rb

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

data/rails_generators/nifty_scaffold/templates/action_tests/index.rb

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

data/rails_generators/nifty_scaffold/templates/action_tests/new.rb

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

data/rails_generators/nifty_scaffold/templates/action_tests/show.rb

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

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

data/rails_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 <%= name.humanize.downcase %>."
+    redirect_to <%= plural_name %>_url
+  end

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

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

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

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

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

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

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

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

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

data/rails_generators/nifty_scaffold/templates/controller.rb

@@ -0,0 +1,14 @@
+class <%= plural_class_name %>Controller < ApplicationController
+  make_resourceful do
+    actions :all
+  end
+  
+  def current_objects
+    @current_objects = current_model.paginate(:all,
+          :order => ‘orders.created_at’,
+          :page => params[:page],
+          :conditions => search_conditions,
+          :joins => :user)
+  end
+  
+end

data/rails_generators/nifty_scaffold/templates/controller_spec.rb

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

data/rails_generators/nifty_scaffold/templates/controller_test.rb

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

data/rails_generators/nifty_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/nifty_scaffold/templates/helper.rb

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

data/rails_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 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/nifty_scaffold/templates/model.rb

@@ -0,0 +1,2 @@
+class <%= class_name %> < ActiveRecord::Base
+end