ryanb-nifty-generators 0.0.3

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2008 Ryan Bates
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Manifest

@@ -0,0 +1,30 @@
+LICENSE
+Manifest
+rails_generators/nifty_layout/nifty_layout_generator.rb
+rails_generators/nifty_layout/templates/helper.rb
+rails_generators/nifty_layout/templates/layout.html.erb
+rails_generators/nifty_layout/templates/stylesheet.css
+rails_generators/nifty_layout/USAGE
+rails_generators/nifty_scaffold/nifty_scaffold_generator.rb
+rails_generators/nifty_scaffold/templates/actions/create.rb
+rails_generators/nifty_scaffold/templates/actions/destroy.rb
+rails_generators/nifty_scaffold/templates/actions/edit.rb
+rails_generators/nifty_scaffold/templates/actions/index.rb
+rails_generators/nifty_scaffold/templates/actions/new.rb
+rails_generators/nifty_scaffold/templates/actions/show.rb
+rails_generators/nifty_scaffold/templates/actions/update.rb
+rails_generators/nifty_scaffold/templates/controller.rb
+rails_generators/nifty_scaffold/templates/helper.rb
+rails_generators/nifty_scaffold/templates/migration.rb
+rails_generators/nifty_scaffold/templates/model.rb
+rails_generators/nifty_scaffold/templates/views/_form.html.erb
+rails_generators/nifty_scaffold/templates/views/edit.html.erb
+rails_generators/nifty_scaffold/templates/views/index.html.erb
+rails_generators/nifty_scaffold/templates/views/new.html.erb
+rails_generators/nifty_scaffold/templates/views/show.html.erb
+rails_generators/nifty_scaffold/USAGE
+README
+test/test_helper.rb
+test/test_nifty_layout_generator.rb
+test/test_nifty_scaffold_generator.rb
+TODO

data/README

@@ -0,0 +1,49 @@
+= nifty-generator
+
+* http://github.com/ryanb/nifty-generators/
+
+== DESCRIPTION:
+
+A collection of Ryan's generator scripts for Rails.
+
+== FEATURES/PROBLEMS:
+
+* Still in very early development
+
+== SYNOPSIS:
+
+  script/generate nifty_layout
+
+== REQUIREMENTS:
+
+* Rails
+* Similar opinions as me
+
+== INSTALL:
+
+* gem install ryanb-nifty-generators --source=http://gems.github.com
+
+== LICENSE:
+
+(The MIT License)
+
+Copyright (c) 2008 Ryan Bates
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+'Software'), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/TODO

@@ -0,0 +1,4 @@
+
+Nifty Scaffold Generator
+  - generate tests/specs
+  + support custom actions

data/nifty-generators.gemspec

@@ -0,0 +1,52 @@
+
+# Gem::Specification for Nifty-generators-0.0.3
+# Originally generated by Echoe
+
+Gem::Specification.new do |s|
+  s.name = %q{nifty-generators}
+  s.version = "0.0.3"
+
+  s.specification_version = 2 if s.respond_to? :specification_version=
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Ryan Bates"]
+  s.date = %q{2008-05-07}
+  s.description = %q{A collection of Ryan's generator scripts for Rails.}
+  s.email = %q{ryan@railscasts.com}
+  s.extra_rdoc_files = ["LICENSE", "README", "TODO"]
+  s.files = ["LICENSE", "Manifest", "rails_generators/nifty_layout/nifty_layout_generator.rb", "rails_generators/nifty_layout/templates/helper.rb", "rails_generators/nifty_layout/templates/layout.html.erb", "rails_generators/nifty_layout/templates/stylesheet.css", "rails_generators/nifty_layout/USAGE", "rails_generators/nifty_scaffold/nifty_scaffold_generator.rb", "rails_generators/nifty_scaffold/templates/actions/create.rb", "rails_generators/nifty_scaffold/templates/actions/destroy.rb", "rails_generators/nifty_scaffold/templates/actions/edit.rb", "rails_generators/nifty_scaffold/templates/actions/index.rb", "rails_generators/nifty_scaffold/templates/actions/new.rb", "rails_generators/nifty_scaffold/templates/actions/show.rb", "rails_generators/nifty_scaffold/templates/actions/update.rb", "rails_generators/nifty_scaffold/templates/controller.rb", "rails_generators/nifty_scaffold/templates/migration.rb", "rails_generators/nifty_scaffold/templates/model.rb", "rails_generators/nifty_scaffold/templates/views/_form.html.erb", "rails_generators/nifty_scaffold/templates/views/edit.html.erb", "rails_generators/nifty_scaffold/templates/views/index.html.erb", "rails_generators/nifty_scaffold/templates/views/new.html.erb", "rails_generators/nifty_scaffold/templates/views/show.html.erb", "rails_generators/nifty_scaffold/USAGE", "README", "test/test_helper.rb", "test/test_nifty_layout_generator.rb", "test/test_nifty_scaffold_generator.rb", "TODO", "nifty-generators.gemspec"]
+  s.has_rdoc = true
+  s.homepage = %q{http://github.com/ryanb/nifty-generators}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Nifty-generators", "--main", "README"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{nifty-generators}
+  s.rubygems_version = %q{1.0.0}
+  s.summary = %q{A collection of Ryan's generator scripts for Rails.}
+  s.test_files = ["test/test_helper.rb", "test/test_nifty_layout_generator.rb", "test/test_nifty_scaffold_generator.rb"]
+end
+
+
+# # Original Rakefile source (requires the Echoe gem):
+# 
+# require 'rubygems'
+# require 'rake'
+#  
+# begin
+#   require 'echoe'
+# 
+#   Echoe.new('nifty-generators', '0.0.3') do |p|
+#     p.summary        = "A collection of Ryan's generator scripts for Rails."
+#     p.description    = "A collection of Ryan's generator scripts for Rails."
+#     p.url            = "http://github.com/ryanb/nifty-generators"
+#     p.author         = 'Ryan Bates'
+#     p.email          = "ryan@railscasts.com"
+#     p.ignore_pattern = ["script/*", "*.gemspec"]
+#   end
+# 
+# rescue LoadError => boom
+#   puts "You are missing a dependency required for meta-operations on this gem."
+#   puts "#{boom.to_s.capitalize}."
+# end
+# 
+# desc "Build the manifest and gemspec files."
+# task :build => [:build_manifest, :build_gemspec]

data/rails_generators/nifty_layout/USAGE

@@ -0,0 +1,21 @@
+Description:
+  The nifty_layout generator creates a basic stylesheet and layout
+	which will give some structure to a starting Rails app.
+
+  The generator takes one argument which will be the name of the
+	layout and stylesheet files. If no argument is passed then it defaults
+	to "application".
+
+Usage:
+  ./script/generate nifty_layout
+
+    This will create two files:
+      Layout:     app/views/layouts/application.html.erb
+      Stylesheet: public/stylesheets/application.css
+
+
+  ./script/generate nifty_layout admin
+
+    This will create two files:
+      Layout:     app/views/layouts/admin.html.erb
+      Stylesheet: public/stylesheets/admin.css

data/rails_generators/nifty_layout/nifty_layout_generator.rb

@@ -0,0 +1,31 @@
+class NiftyLayoutGenerator < Rails::Generator::Base
+  def initialize(runtime_args, runtime_options = {})
+    super
+    @name = @args.first || 'application'
+  end
+  
+  def manifest
+    record do |m|
+      m.directory 'app/views/layouts'
+      m.directory 'public/stylesheets'
+      m.directory 'app/helpers'
+
+      m.template "layout.html.erb", "app/views/layouts/#{file_name}.html.erb"
+      m.file     "stylesheet.css",  "public/stylesheets/#{file_name}.css"
+      m.file     "helper.rb",       "app/helpers/layout_helper.rb"
+    end
+  end
+  
+  def file_name
+    @name.underscore
+  end
+
+  protected
+    def banner
+      <<-EOS
+Creates layout and stylesheet files.
+
+USAGE: #{$0} #{spec.name} application
+EOS
+    end
+end

data/rails_generators/nifty_layout/templates/helper.rb

@@ -0,0 +1,23 @@
+# These helper methods can be called in your template to set variables to be used in the layout
+# This module should be included in all views globally,
+# to do so you may need to add this line to your ApplicationController
+#   helper :layout
+module LayoutHelper
+  def title(page_title, show_title = true)
+    @content_for_title = page_title.to_s
+    @show_title = show_title
+  end
+  
+  def show_title?
+    @show_title
+  end
+  
+  def stylesheet(*args)
+    content_for(:head) { stylesheet_link_tag(*args.map(&:to_s)) }
+  end
+  
+  def javascript(*args)
+    args = args.map { |arg| arg == :defaults ? arg : arg.to_s }
+    content_for(:head) { javascript_include_tag(*args) }
+  end
+end

data/rails_generators/nifty_layout/templates/layout.html.erb

@@ -0,0 +1,23 @@
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN"
+  "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
+<html>
+  <head>
+    <title><%%= h(yield(:title) || "Untitled") %></title>
+    <%%= stylesheet_link_tag '<%= file_name %>' %>
+    <%%= javascript_include_tag :defaults %>
+    <%%= yield(:head) %>
+  </head>
+  <body>
+    <div id="container">
+      <%%- flash.each do |name, msg| -%>
+        <%%= content_tag :div, msg, :id => "flash_#{name}" %>
+      <%%- end -%>
+      
+      <%%- if show_title? -%>
+        <h1><%%=h yield(:title) %></h1>
+      <%%- end -%>
+      
+      <%%= yield %>
+    </div>
+  </body>
+</html>

data/rails_generators/nifty_layout/templates/stylesheet.css

@@ -0,0 +1,82 @@
+body {
+  background-color: #FFF;
+  font-family: Verdana, Helvetica, Arial;
+  font-size: 14px;
+}
+
+a img {
+  border: none;
+}
+
+a {
+  color: #0000FF;
+}
+
+body {
+  background-color: #4B7399;
+}
+
+#container {
+  width: 75%;
+  margin: 0 auto;
+  background-color: #FFF;
+  padding: 20px 40px;
+  border: solid 1px black;
+  margin-top: 20px;
+}
+
+#flash_notice, #flash_error {
+  padding: 5px 8px;
+  margin: 10px 0;
+}
+
+#flash_notice {
+  background-color: #CFC;
+  border: solid 1px #6C6;
+}
+
+#flash_error {
+  background-color: #FCC;
+  border: solid 1px #C66;
+}
+
+.clear {
+  clear: both;
+}
+
+
+
+.fieldWithErrors {
+  display: inline;
+}
+
+#errorExplanation {
+  width: 400px;
+  border: 2px solid #CF0000;
+  padding: 0px;
+  padding-bottom: 12px;
+  margin-bottom: 20px;
+  background-color: #f0f0f0;
+}
+
+#errorExplanation h2 {
+  text-align: left;
+  font-weight: bold;
+  padding: 5px 5px 5px 15px;
+  font-size: 12px;
+  margin: 0px;
+  background-color: #c00;
+  color: #fff;
+}
+#errorExplanation p {
+  color: #333;
+  margin-bottom: 0;
+  padding: 8px;
+}
+#errorExplanation ul {
+  margin: 2px 24px;
+}
+#errorExplanation ul li {
+  font-size: 12px;
+  list-style: disc;
+}

data/rails_generators/nifty_scaffold/USAGE

@@ -0,0 +1,5 @@
+Description:
+  
+  
+Usage:
+  

data/rails_generators/nifty_scaffold/nifty_scaffold_generator.rb

@@ -0,0 +1,129 @@
+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.include? ':'
+        @attributes << Rails::Generator::GeneratedAttribute.new(*arg.split(":"))
+      else
+        @controller_actions << arg
+      end
+    end
+    
+    if @controller_actions.empty?
+      @controller_actions = %w[index show new create edit update destroy]
+    end
+    
+    if @attributes.empty?
+      if model_exists?
+        class_name.constantize.columns.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|
+      m.directory "app/controllers"
+      m.directory "app/helpers"
+      m.directory "app/models"
+      m.directory "db/migrate"
+      m.directory "app/views/#{plural_name}"
+      m.template "controller.rb", "app/controllers/#{plural_name}_controller.rb"
+      m.template "helper.rb", "app/helpers/#{plural_name}_helper.rb"
+      
+      unless model_exists?
+        m.template "model.rb", "app/models/#{singular_name}.rb"
+        m.migration_template "migration.rb", "db/migrate", :migration_file_name => "create_#{plural_name}"
+      end
+      
+      controller_actions.each do |action|
+        if File.exist? source_path("views/#{action}.html.erb")
+          m.template "views/#{action}.html.erb", "app/views/#{plural_name}/#{action}.html.erb"
+        end
+      end
+      
+      if form_partial?
+        m.template "views/_form.html.erb", "app/views/#{plural_name}/_form.html.erb"
+      end
+      
+      m.route_resources plural_name
+    end
+  end
+  
+  def form_partial?
+    controller_actions.include?('new') && controller_actions.include?('edit')
+  end
+  
+  def singular_name
+    name.underscore
+  end
+  
+  def plural_name
+    name.pluralize
+  end
+  
+  def class_name
+    name.camelize
+  end
+  
+  def plural_class_name
+    plural_name.camelize
+  end
+  
+  def controller_methods
+    controller_actions.map do |action|
+      controller_method(action)
+    end.join
+  end
+  
+  def controller_method(name)
+    read_template("actions/#{name}.rb")
+  end
+  
+  def render_form
+    if form_partial?
+      "<%= render :partial => 'form' %>"
+    else
+      read_template("views/_form.html.erb")
+    end
+  end
+  
+  def item_path
+    if controller_actions.include? 'show'
+      "@#{singular_name}"
+    else
+      "#{plural_name}_path"
+    end
+  end
+  
+protected
+
+  # 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 model given the name and attributes.
+
+USAGE: #{$0} #{spec.name} name
+EOS
+  end
+end

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

@@ -0,0 +1,8 @@
+  def create
+    @<%= singular_name %> = <%= class_name %>.new(params[:<%= singular_name %>])
+    if @<%= singular_name %>.save
+      redirect_to <%= item_path %>
+    else
+      render :action => 'new'
+    end
+  end

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

@@ -0,0 +1,5 @@
+  def destroy
+    @<%= singular_name %> = <%= class_name %>.find(params[:id])
+    @<%= singular_name %>.destroy
+    redirect_to <%= plural_name %>_path
+  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,8 @@
+  def update
+    @<%= singular_name %> = <%= class_name %>.find(params[:id])
+    if @<%= singular_name %>.update_attributes(params[:<%= singular_name %>])
+      redirect_to <%= item_path %>
+    else
+      render :action => 'edit'
+    end
+  end

data/rails_generators/nifty_scaffold/templates/controller.rb

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

data/rails_generators/nifty_scaffold/templates/migration.rb

@@ -0,0 +1,14 @@
+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 -%>
+      t.timestamps
+    end
+  end
+  
+  def self.down
+    drop_table :<%= plural_name %>
+  end
+end

data/rails_generators/nifty_scaffold/templates/model.rb

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

data/rails_generators/nifty_scaffold/templates/views/_form.html.erb

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

data/rails_generators/nifty_scaffold/templates/views/edit.html.erb

@@ -0,0 +1,8 @@
+<%% title "Edit <%= singular_name.titleize %>" %>
+
+<%= render_form %>
+
+<p>
+  <%%= link_to "Show", @<%= singular_name %> %> |
+  <%%= link_to "View All", <%= plural_name %>_path %>
+</p>

data/rails_generators/nifty_scaffold/templates/views/index.html.erb

@@ -0,0 +1,22 @@
+<%% title "<%= plural_name.titleize %>" %>
+
+<table>
+  <tr>
+  <%- for attribute in attributes -%>
+    <th><%= attribute.column.human_name %></th>
+  <%- end -%>
+  </tr>
+
+  <%% for <%= singular_name %> in @<%= plural_name %> %>
+    <tr>
+      <%- for attribute in attributes -%>
+        <td><%%=h <%= singular_name %>.<%= attribute.name %> %></td>
+      <%- end -%>
+      <td><%%= link_to "Show", <%= singular_name %> %></td>
+      <td><%%= link_to "Edit", edit_<%= singular_name %>_path(<%= singular_name %>) %></td>
+      <td><%%= link_to "Destroy", <%= singular_name %>, :confirm => 'Are you sure?', :method => :delete %></td>
+    </tr>
+  <%% end %>
+</table>
+
+<p><%%= link_to "New <%= singular_name %>", new_<%= singular_name %>_path %></p>

data/rails_generators/nifty_scaffold/templates/views/new.html.erb

@@ -0,0 +1,5 @@
+<%% title "New <%= singular_name.titleize %>" %>
+
+<%= render_form %>
+
+<p><%%= link_to "Back to List", <%= plural_name %>_path %></p>

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

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

data/test/test_helper.rb

@@ -0,0 +1,116 @@
+require 'test/unit'
+
+# Must set before requiring generator libs.
+TMP_ROOT = File.dirname(__FILE__) + "/tmp" unless defined?(TMP_ROOT)
+PROJECT_NAME = "myproject" unless defined?(PROJECT_NAME)
+app_root = File.join(TMP_ROOT, PROJECT_NAME)
+if defined?(APP_ROOT)
+  APP_ROOT.replace(app_root)
+else
+  APP_ROOT = app_root
+end
+if defined?(RAILS_ROOT)
+  RAILS_ROOT.replace(app_root)
+else
+  RAILS_ROOT = app_root
+end
+
+require 'rubygems'
+require 'rubigen'  # gem install rubigen
+require 'rubigen/helpers/generator_test_helper'
+require 'rails_generator'
+require 'shoulda' # gem install Shoulda
+require 'mocha'
+
+module NiftyGenerators
+  module TestHelper
+    include RubiGen::GeneratorTestHelper
+  
+    def setup
+      bare_setup
+    end
+
+    def teardown
+      bare_teardown
+    end
+  
+    protected
+  
+    def run_rails_generator(generator, *args)
+      options = args.shift! if args.last.kind_of? Hash
+      options ||= {}
+      run_generator(generator.to_s, args, generator_sources, options.reverse_merge(:quiet => true))
+    end
+  
+    def generator_sources
+      [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__), "..", "rails_generators"))]
+    end
+  end
+  
+  module ShouldaAdditions
+    def rails_generator(*args)
+      setup do
+        run_rails_generator(*args)
+      end
+    end
+    
+    def should_generate_file(file, &block)
+      should "generate file #{file}" do
+        yield("foo") if block_given?
+        assert_generated_file(file)
+      end
+    end
+    
+    def should_not_generate_file(file)
+      should "not generate file #{file}" do
+        assert !File.exists?("#{APP_ROOT}/#{file}"),"The file '#{file}' should not exist"
+      end
+    end
+  end
+end
+
+class Thoughtbot::Shoulda::Context
+  include NiftyGenerators::ShouldaAdditions
+end
+
+# Mock out what we need from AR::Base.
+module ActiveRecord
+  class Base
+    class << self
+      attr_accessor :pluralize_table_names, :columns
+
+      def add_column(name, type = :string)
+        returning ActiveRecord::ConnectionAdapters::Column.new(name, nil) do |column|
+          column.type = type
+          @columns ||= []
+          @columns << column
+        end
+      end
+    end
+    self.pluralize_table_names = true
+  end
+
+  module ConnectionAdapters
+    class Column
+      attr_accessor :name, :default, :type, :limit, :null, :sql_type, :precision, :scale
+
+      def initialize(name, default, sql_type = nil)
+        @name=name
+        @default=default
+        @type=@sql_type=sql_type
+      end
+
+      def human_name
+        @name.humanize
+      end
+    end
+  end
+end
+
+# And what we need from ActionView
+module ActionView
+  module Helpers
+    module ActiveRecordHelper; end
+    class InstanceTag; end
+  end
+end

data/test/test_nifty_layout_generator.rb

@@ -0,0 +1,36 @@
+require File.join(File.dirname(__FILE__), "test_helper.rb")
+
+class TestNiftyLayoutGenerator < Test::Unit::TestCase
+  include NiftyGenerators::TestHelper
+  
+  # Some generator-related assertions:
+  #   assert_generated_file(name, &block) # block passed the file contents
+  #   assert_directory_exists(name)
+  #   assert_generated_class(name, &block)
+  #   assert_generated_module(name, &block)
+  #   assert_generated_test_for(name, &block)
+  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
+  #   assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
+  #
+  # Other helper methods are:
+  #   app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
+  #   bare_setup - place this in setup method to create the APP_ROOT folder for each test
+  #   bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
+  
+  context "generator without name" do
+    rails_generator :nifty_layout
+    should_generate_file 'app/views/layouts/application.html.erb'
+  end
+  
+  context "generator with name" do
+    rails_generator :nifty_layout, "foobar"
+    should_generate_file 'app/views/layouts/foobar.html.erb'
+    should_generate_file 'public/stylesheets/foobar.css'
+    should_generate_file 'app/helpers/layout_helper.rb'
+  end
+  
+  context "generator with CamelCase name" do
+    rails_generator :nifty_layout, "FooBar"
+    should_generate_file 'app/views/layouts/foo_bar.html.erb'
+  end
+end

data/test/test_nifty_scaffold_generator.rb

@@ -0,0 +1,257 @@
+require File.join(File.dirname(__FILE__), "test_helper.rb")
+
+class TestNiftyScaffoldGenerator < Test::Unit::TestCase
+  include NiftyGenerators::TestHelper
+  
+  # Some generator-related assertions:
+  #   assert_generated_file(name, &block) # block passed the file contents
+  #   assert_directory_exists(name)
+  #   assert_generated_class(name, &block)
+  #   assert_generated_module(name, &block)
+  #   assert_generated_test_for(name, &block)
+  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
+  #   assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
+  #
+  # Other helper methods are:
+  #   app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
+  #   bare_setup - place this in setup method to create the APP_ROOT folder for each test
+  #   bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
+  
+  context "routed" do
+    setup do
+      Dir.mkdir("#{RAILS_ROOT}/config") unless File.exists?("#{RAILS_ROOT}/config")
+      File.open("#{RAILS_ROOT}/config/routes.rb", 'w') do |f|
+        f.puts "ActionController::Routing::Routes.draw do |map|\n\nend"
+      end
+    end
+    
+    teardown do
+      FileUtils.rm_rf "#{RAILS_ROOT}/config"
+    end
+    
+    context "generator without name" do
+      should "raise usage error" do
+        assert_raise Rails::Generator::UsageError do
+          run_rails_generator :nifty_scaffold
+        end
+      end
+    end
+  
+    context "generator with no options" do
+      rails_generator :nifty_scaffold, "line_item"
+    
+      should "generate model with class as camelcase name" do
+        assert_generated_file "app/models/line_item.rb" do |body|
+          assert_match "class LineItem < ActiveRecord::Base", body
+        end
+      end
+    
+      should "generate migration with default name column" do
+        file = Dir.glob("#{RAILS_ROOT}/db/migrate/*.rb").first
+        assert file, "migration file doesn't exist"
+        assert_match(/[0-9]+_create_line_items.rb$/, file)
+        assert_generated_file "db/migrate/#{File.basename(file)}" do |body|
+          assert_match "class CreateLineItems", body
+          assert_match "t.string :name", body
+          assert_match "t.timestamp", body
+        end
+      end
+    
+      should_generate_file "app/helpers/line_items_helper.rb"
+    
+      should "generate controller with class as camelcase name pluralized and all actions" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "class LineItemsController < ApplicationController", body
+          %w[index show new create edit update destroy].each do |action|
+            assert_match "def #{action}", body
+          end
+        end
+      end
+    
+      %w[index show new edit].each do |action|
+        should_generate_file "app/views/line_items/#{action}.html.erb"
+      end
+    
+      should "have name attribute" do
+        assert_generated_file "app/views/line_items/_form.html.erb" do |body|
+          assert_match "<%= f.text_field :name %>", body
+        end
+      end
+      
+      should "add map.resources line to routes" do
+        assert_generated_file "config/routes.rb" do |body|
+          assert_match "map.resources :line_items", body
+        end
+      end
+    end
+  
+    context "generator with index action" do
+      rails_generator :nifty_scaffold, "line_item", "index"
+    
+      should_generate_file "app/views/line_items/index.html.erb"
+    
+      should "generate controller with index action" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "def index", body
+          assert_match "@line_items = LineItem.find(:all)", body
+          assert_no_match(/    def index/, body)
+        end
+      end
+    end
+  
+    context "generator with show action" do
+      rails_generator :nifty_scaffold, "line_item", "show"
+    
+      should_generate_file "app/views/line_items/show.html.erb"
+    
+      should "generate controller with show action" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "def show", body
+          assert_match "@line_item = LineItem.find(params[:id])", body
+        end
+      end
+    end
+  
+    context "generator with new and create actions" do
+      rails_generator :nifty_scaffold, "line_item", "new", "create"
+    
+      should_not_generate_file "app/views/line_items/create.html.erb"
+      should_not_generate_file "app/views/line_items/_form.html.erb"
+    
+      should "render form in 'new' template" do
+        assert_generated_file "app/views/line_items/new.html.erb" do |body|
+          assert_match "<% form_for @line_item do |f| %>", body
+        end
+      end
+    
+      should "generate controller with actions" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "def new", body
+          assert_match "@line_item = LineItem.new\n", body
+          assert_match "def create", body
+          assert_match "@line_item = LineItem.new(params[:line_item])", body
+          assert_match "if @line_item.save", body
+          assert_match "redirect_to line_items_path", body
+          assert_match "render :action => 'new'", body
+        end
+      end
+    end
+  
+    context "generator with edit and update actions" do
+      rails_generator :nifty_scaffold, "line_item", "edit", "update"
+    
+      should_not_generate_file "app/views/line_items/update.html.erb"
+      should_not_generate_file "app/views/line_items/_form.html.erb"
+    
+      should "render form in 'edit' template" do
+        assert_generated_file "app/views/line_items/edit.html.erb" do |body|
+          assert_match "<% form_for @line_item do |f| %>", body
+        end
+      end
+    
+      should "generate controller with actions" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "def edit", body
+          assert_match "@line_item = LineItem.find(params[:id])", body
+          assert_match "def update", body
+          assert_match "if @line_item.update_attributes(params[:line_item])", body
+          assert_match "redirect_to line_items_path", body
+          assert_match "render :action => 'edit'", body
+        end
+      end
+    end
+  
+    context "generator with edit and update actions" do
+      rails_generator :nifty_scaffold, "line_item", "destroy"
+    
+      should_not_generate_file "app/views/line_items/destroy.html.erb"
+    
+      should "generate controller with action" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "def destroy", body
+          assert_match "@line_item = LineItem.find(params[:id])", body
+          assert_match "@line_item.destroy", body
+          assert_match "redirect_to line_items_path", body
+        end
+      end
+    end
+  
+    context "generator with new and edit actions" do
+      rails_generator :nifty_scaffold, "line_item", "new", "edit"
+    
+      should_generate_file "app/views/line_items/_form.html.erb"
+    
+      should "render the form partial in views" do
+        %w[new edit].each do |action|
+          assert_generated_file "app/views/line_items/#{action}.html.erb" do |body|
+            assert_match "<%= render :partial => 'form' %>", body
+          end
+        end
+      end
+    end
+  
+    context "generator with attributes and actions" do
+      rails_generator :nifty_scaffold, "line_item", "name:string", "new", "price:float", "index", "available:boolean"
+    
+      should "render a form field for each attribute in 'new' template" do
+        assert_generated_file "app/views/line_items/new.html.erb" do |body|
+          assert_match "<%= f.text_field :name %>", body
+          assert_match "<%= f.text_field :price %>", body
+          assert_match "<%= f.check_box :available %>", body
+        end
+      end
+    end
+  
+    context "generator with show, create, and update actions" do
+      rails_generator :nifty_scaffold, "line_item", "show", "create", "update"
+    
+      should "redirect to line item show page, not index" do
+        assert_generated_file "app/controllers/line_items_controller.rb" do |body|
+          assert_match "redirect_to @line_item", body
+          assert_no_match(/redirect_to line_items_path/, body)
+        end
+      end
+    end
+  
+    context "existing model" do
+      setup do
+        Dir.mkdir("#{RAILS_ROOT}/app") unless File.exists?("#{RAILS_ROOT}/app")
+        Dir.mkdir("#{RAILS_ROOT}/app/models") unless File.exists?("#{RAILS_ROOT}/app/models")
+        File.open("#{RAILS_ROOT}/app/models/recipe.rb", 'w') do |f|
+          f.puts "raise 'should not be loaded'"
+        end
+      end
+    
+      teardown do
+        FileUtils.rm_rf "#{RAILS_ROOT}/app"
+      end
+    
+      context "generator with no options" do
+        rails_generator :nifty_scaffold, "recipe"
+    
+        should "use model columns for attributes" do
+          assert_generated_file "app/views/recipes/_form.html.erb" do |body|
+            assert_match "<%= f.text_field :foo %>", body
+            assert_match "<%= f.text_field :bar %>", body
+          end
+        end
+      end
+    
+      context "generator with attribute specified" do
+        rails_generator :nifty_scaffold, "recipe", "zippo:string"
+    
+        should "use specified attribute" do
+          assert_generated_file "app/views/recipes/_form.html.erb" do |body|
+            assert_match "<%= f.text_field :zippo %>", body
+          end
+        end
+      end
+    end
+  end
+end
+
+# just an example model we can use
+class Recipe < ActiveRecord::Base
+  add_column :foo, :string
+  add_column :bar, :string
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification 
+name: ryanb-nifty-generators
+version: !ruby/object:Gem::Version 
+  version: 0.0.3
+platform: ruby
+authors: 
+- Ryan Bates
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-05-07 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: A collection of Ryan's generator scripts for Rails.
+email: ryan@railscasts.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README
+- TODO
+files: 
+- LICENSE
+- Manifest
+- rails_generators/nifty_layout/nifty_layout_generator.rb
+- rails_generators/nifty_layout/templates/helper.rb
+- rails_generators/nifty_layout/templates/layout.html.erb
+- rails_generators/nifty_layout/templates/stylesheet.css
+- rails_generators/nifty_layout/USAGE
+- rails_generators/nifty_scaffold/nifty_scaffold_generator.rb
+- rails_generators/nifty_scaffold/templates/actions/create.rb
+- rails_generators/nifty_scaffold/templates/actions/destroy.rb
+- rails_generators/nifty_scaffold/templates/actions/edit.rb
+- rails_generators/nifty_scaffold/templates/actions/index.rb
+- rails_generators/nifty_scaffold/templates/actions/new.rb
+- rails_generators/nifty_scaffold/templates/actions/show.rb
+- rails_generators/nifty_scaffold/templates/actions/update.rb
+- rails_generators/nifty_scaffold/templates/controller.rb
+- rails_generators/nifty_scaffold/templates/migration.rb
+- rails_generators/nifty_scaffold/templates/model.rb
+- rails_generators/nifty_scaffold/templates/views/_form.html.erb
+- rails_generators/nifty_scaffold/templates/views/edit.html.erb
+- rails_generators/nifty_scaffold/templates/views/index.html.erb
+- rails_generators/nifty_scaffold/templates/views/new.html.erb
+- rails_generators/nifty_scaffold/templates/views/show.html.erb
+- rails_generators/nifty_scaffold/USAGE
+- README
+- test/test_helper.rb
+- test/test_nifty_layout_generator.rb
+- test/test_nifty_scaffold_generator.rb
+- TODO
+- nifty-generators.gemspec
+has_rdoc: true
+homepage: http://github.com/ryanb/nifty-generators
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Nifty-generators
+- --main
+- README
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: nifty-generators
+rubygems_version: 1.0.1
+signing_key: 
+specification_version: 2
+summary: A collection of Ryan's generator scripts for Rails.
+test_files: 
+- test/test_helper.rb
+- test/test_nifty_layout_generator.rb
+- test/test_nifty_scaffold_generator.rb