nifty-generators-improved 0.5

data/features/nifty_authentication.feature

@@ -0,0 +1,80 @@
+Feature: Nifty Authentication Generator
+  In order to authenticate users
+  As a rails developer
+  I want to generate some user authentication
+
+  Scenario: Generate default authentication
+    Given a new Rails app
+    When I insert "root :to => 'users#new'" into "config/routes.rb" after line 1
+    And I run "rails g nifty:authentication"
+    Then I should see the following files
+      | app/models/user.rb                          |
+      | app/controllers/users_controller.rb         |
+      | app/helpers/users_helper.rb                 |
+      | app/views/users/new.html.erb                |
+      | app/views/users/edit.html.erb               |
+      | app/controllers/sessions_controller.rb      |
+      | app/helpers/sessions_helper.rb              |
+      | app/views/sessions/new.html.erb             |
+      | lib/controller_authentication.rb            |
+      | test/fixtures/users.yml                     |
+      | test/unit/user_test.rb                      |
+      | test/functional/users_controller_test.rb    |
+      | test/functional/sessions_controller_test.rb |
+      | db/migrate                                  |
+    And I should see the following in file "config/routes.rb"
+      | resources :sessions                                          |
+      | resources :users                                             |
+      | match 'login' => 'sessions#new', :as => :login               |
+      | match 'logout' => 'sessions#destroy', :as => :logout         |
+      | match 'signup' => 'users#new', :as => :signup                |
+      | match 'user/edit' => 'users#edit', :as => :edit_current_user |
+    And I should see "include ControllerAuthentication" in file "app/controllers/application_controller.rb"
+    And I should see "gem "mocha", :group => :test" in file "Gemfile"
+    And I should see "gem "bcrypt-ruby", :require => "bcrypt"" in file "Gemfile"
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    Then I should successfully run "rake test"
+
+  Scenario: Generate named authentication
+    Given a new Rails app
+    When I insert "root :to => 'accounts#new'" into "config/routes.rb" after line 1
+    And I run "rails g nifty:authentication Account CurrentSession"
+    Then I should see the following files
+      | app/models/account.rb                               |
+      | app/controllers/accounts_controller.rb              |
+      | app/helpers/accounts_helper.rb                      |
+      | app/views/accounts/new.html.erb                     |
+      | app/views/accounts/edit.html.erb                    |
+      | app/controllers/current_sessions_controller.rb      |
+      | app/helpers/current_sessions_helper.rb              |
+      | app/views/current_sessions/new.html.erb             |
+      | test/fixtures/accounts.yml                          |
+      | test/unit/account_test.rb                           |
+      | test/functional/accounts_controller_test.rb         |
+      | test/functional/current_sessions_controller_test.rb |
+    And I should see the following in file "config/routes.rb"
+      | resources :current_sessions                                   |
+      | resources :accounts                                           |
+      | match 'login' => 'current_sessions#new', :as => :login        |
+      | match 'logout' => 'current_sessions#destroy', :as => :logout  |
+      | match 'signup' => 'accounts#new', :as => :signup              |
+      | match 'account/edit' => 'accounts#edit', :as => :edit_current_account |
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    Then I should successfully run "rake test"
+
+  Scenario: Generate named authentication with rspec
+    Given a new Rails app
+    When I insert "root :to => 'accounts#new'" into "config/routes.rb" after line 1
+    And I run "rails g nifty:authentication Account CurrentSession --rspec"
+    Then I should see the following files
+      | spec/models/account_spec.rb                          |
+      | spec/controllers/accounts_controller_spec.rb         |
+      | spec/controllers/current_sessions_controller_spec.rb |
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    And I add "gem 'rspec-rails', '>= 2.0.1'" to file "Gemfile"
+    And I run "rails g rspec:install"
+    And I replace "mock_with :rspec" with "mock_with :mocha" in file "spec/spec_helper.rb"
+    Then I should successfully run "rake spec"

data/features/nifty_config.feature

@@ -0,0 +1,17 @@
+Feature: Nifty Config Generator
+  In order to have configure an app
+  As a rails developer
+  I want to generate a config file and manager
+
+  Scenario: Generate normal config
+    Given a new Rails app
+    When I run "rails g nifty:config"
+    Then I should see file "config/app_config.yml"
+    And I should see file "config/initializers/load_app_config.rb"
+
+  Scenario: Generate named config
+    Given a new Rails app
+    When I run "rails g nifty:config FooBar"
+    Then I should see "FOO_BAR_CONFIG" in file "config/initializers/load_foo_bar_config.rb"
+    And I should see "config/foo_bar_config.yml" in file "config/initializers/load_foo_bar_config.rb"
+    And I should see file "config/foo_bar_config.yml"

data/features/nifty_layout.feature

@@ -0,0 +1,20 @@
+Feature: Nifty Layout Generator
+  In order to have a layout
+  As a rails developer
+  I want to generate a simple layout
+
+  Scenario: Generate normal application layout
+    Given a new Rails app
+    When I run "rails g nifty:layout -f"
+    Then I should see "stylesheet_link_tag "application"" in file "app/views/layouts/application.html.erb"
+    Then I should see "h(page_title" in file "app/helpers/layout_helper.rb"
+    And I should see file "app/helpers/layout_helper.rb"
+    And I should see file "app/helpers/error_messages_helper.rb"
+    And I should see file "public/stylesheets/application.css"
+
+  Scenario: Generate named layout with haml and sass
+    Given a new Rails app
+    When I run "rails g nifty:layout FooBar --haml -f"
+    Then I should see "stylesheet_link_tag "foo_bar"" in file "app/views/layouts/foo_bar.html.haml"
+    And I should see file "public/stylesheets/sass/foo_bar.sass"
+    And I should see file "app/helpers/layout_helper.rb"

data/features/nifty_scaffold.feature

@@ -0,0 +1,80 @@
+Feature: Nifty Scaffold Generator
+  In order to manage a resource
+  As a rails developer
+  I want to generate a model, controller, and views for that resource
+
+  Scenario: Generate scaffold for simple resource
+    Given a new Rails app
+    When I run "rails g nifty:scaffold Project name:string"
+    Then I should see the following files
+      | app/models/project.rb                  |
+      | app/controllers/projects_controller.rb |
+      | app/helpers/projects_helper.rb         |
+      | app/views/projects/index.html.erb      |
+      | app/views/projects/show.html.erb       |
+      | app/views/projects/new.html.erb        |
+      | app/views/projects/edit.html.erb       |
+      | db/migrate                             |
+    And I should see "resources :projects" in file "config/routes.rb"
+    And I should see "gem "mocha", :group => :test" in file "Gemfile"
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    And I should successfully run "rails g nifty:scaffold Project -f"
+    Then I should successfully run "rake test"
+
+  Scenario: Generate scaffold with rspec tests
+    Given a new Rails app
+    When I run "rails g nifty:scaffold Project name:string --rspec"
+    Then I should see the following files
+      | spec/models/project_spec.rb                  |
+      | spec/controllers/projects_controller_spec.rb |
+    When I run "rails g nifty:scaffold Task project_id:integer"
+    Then I should see the following files
+      | spec/models/task_spec.rb                  |
+      | spec/controllers/tasks_controller_spec.rb |
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    And I add "gem 'rspec-rails', '>= 2.0.0.beta.19'" to file "Gemfile"
+    And I run "rails g rspec:install"
+    And I replace "mock_with :rspec" with "mock_with :mocha" in file "spec/spec_helper.rb"
+    Then I should successfully run "rake spec"
+
+  Scenario: Report error when invalid model name
+    Given a new Rails app
+    Then I should see "Usage:" when running "rails g nifty:scaffold name:string parent_id:integer"
+
+  Scenario: Generate scaffold for namespaced resource
+    Given a new Rails app
+    When I run "rails g nifty:scaffold Admin::User name:string"
+    Then I should see the following files
+      | app/models/user.rb                        |
+      | app/controllers/admin/users_controller.rb |
+      | app/helpers/admin/users_helper.rb         |
+      | app/views/admin/users/index.html.erb      |
+      | app/views/admin/users/show.html.erb       |
+      | app/views/admin/users/new.html.erb        |
+      | app/views/admin/users/edit.html.erb       |
+      | db/migrate                                |
+    And I should see "class User" in file "app/models/user.rb"
+    And I should not see "set_table_name :" in file "app/models/user.rb"
+    And I should see "namespace(:admin){ resources :users }" in file "config/routes.rb"
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    And I should successfully run "rails g nifty:scaffold Admin::User -f"
+    Then I should successfully run "rake test"
+
+  Scenario: Generate scaffold with a namespaced model
+    Given a new Rails app
+    When I run "rails g nifty:scaffold Admin::User name:string --namespace_model"
+    Then I should see "class Admin::User" in file "app/models/admin/user.rb"
+    And I should see "set_table_name :admin_users" in file "app/models/admin/user.rb"
+    When I run "rails g nifty:layout -f"
+    And I run "rake db:migrate"
+    And I should successfully run "rails g nifty:scaffold Admin::User -f --namespace_model"
+    Then I should successfully run "rake test"
+
+  Scenario: Given scaffold with a new and index action
+    Given a new Rails app
+    When I run "rails g nifty:scaffold Project name:string index new"
+    Then I should see "class Project" in file "app/models/project.rb"
+    And I should see "<%= form_for @project do |f| %>" in file "app/views/projects/new.html.erb"

data/features/step_definitions/common_steps.rb

@@ -0,0 +1,62 @@
+When /^I run "([^\"]*)"$/ do |command|
+  system("cd #{@current_directory} && #{command}").should be_true
+end
+
+When /^I add "([^\"]*)" to file "([^\"]*)"$/ do |content, short_path|
+  path = File.join(@current_directory, short_path)
+  File.should exist(path)
+  File.open(path, 'a') { |f| f.write(content + "\n") }
+end
+
+When /^I replace "([^\"]*)" with "([^\"]*)" in file "([^\"]*)"$/ do |old_content, new_content, short_path|
+  path = File.join(@current_directory, short_path)
+  File.should exist(path)
+  content = File.read(path).gsub(old_content, new_content)
+  File.open(path, 'w') { |f| f.write(content) }
+end
+
+When /^I insert "([^\"]*)" into "([^\"]*)" after line (\d+)$/ do |content, short_path, after_line|
+  path = File.join(@current_directory, short_path)
+  File.should exist(path)
+  lines = File.read(path).split("\n")
+  lines[after_line.to_i, 0] = content
+  File.open(path, 'w') { |f| f.write(lines.join("\n")) }
+end
+
+Then /^I should see file "([^\"]*)"$/ do |path|
+  File.should exist(File.join(@current_directory, path))
+end
+
+Then /^I should see "(.*)" in file "([^\"]*)"$/ do |content, short_path|
+  path = File.join(@current_directory, short_path)
+  File.should exist(path)
+  File.readlines(path).join.should include(content)
+end
+
+Then /^I should not see "(.*)" in file "([^\"]*)"$/ do |content, short_path|
+  path = File.join(@current_directory, short_path)
+  File.should exist(path)
+  File.readlines(path).join.should_not include(content)
+end
+
+Then /^I should see the following files$/ do |table|
+  table.raw.flatten.each do |path|
+    File.should exist(File.join(@current_directory, path))
+  end
+end
+
+Then /^I should see the following in file "([^\"]*)"$/ do |short_path, table|
+  path = File.join(@current_directory, short_path)
+  File.should exist(path)
+  table.raw.flatten.each do |content|
+    File.readlines(path).join.should include(content)
+  end
+end
+
+Then /^I should successfully run "([^\"]*)"$/ do |command|
+  system("cd #{@current_directory} && #{command}").should be_true
+end
+
+Then /^I should see "([^\"]*)" when running "([^\"]*)"$/ do |expected_response, command|
+  `cd #{@current_directory} && #{command}`.should include(expected_response)
+end

data/features/step_definitions/rails_setup_steps.rb

@@ -0,0 +1,6 @@
+Given /^a new Rails app$/ do
+  FileUtils.mkdir_p("tmp")
+  system("rails new tmp/rails_app").should be_true
+  system("ln -s ../../../lib/generators tmp/rails_app/lib/generators").should be_true
+  @current_directory = File.expand_path("tmp/rails_app")
+end

data/features/support/env.rb

@@ -0,0 +1,6 @@
+require 'cucumber'
+require 'rspec'
+
+Before do
+  FileUtils.rm_rf "tmp/rails_app"
+end

data/features/support/matchers.rb

@@ -0,0 +1,7 @@
+RSpec::Matchers.define :exist do |path|
+  match do
+    File.exist?(path)
+  end
+  failure_message_for_should { "Expected #{path} to exist but no file found" }
+  failure_message_for_should_not { "Expected #{path} to not exist but file was found" }
+end

data/lib/generators/nifty.rb

@@ -0,0 +1,28 @@
+require 'rails/generators/base'
+
+module Nifty
+  module Generators
+    class Base < Rails::Generators::Base #:nodoc:
+      def self.source_root
+        @_nifty_source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'nifty', generator_name, 'templates'))
+      end
+
+      def self.banner
+        "rails generate nifty:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
+      end
+
+      private
+
+      def add_gem(name, options = {})
+        gemfile_content = File.read(destination_path("Gemfile"))
+        File.open(destination_path("Gemfile"), 'a') { |f| f.write("\n") } unless gemfile_content =~ /\n\Z/
+        gem name, options unless gemfile_content.include? name
+      end
+
+      def print_usage
+        self.class.help(Thor::Base.shell.new)
+        exit
+      end
+    end
+  end
+end

data/lib/generators/nifty/authentication/USAGE

@@ -0,0 +1,50 @@
+Description:
+    Generates a user model, users controller, and sessions controller. The
+    users controller handles the registration and the sessions controller
+    handles authentication. This is similar to restful_authentication, but
+    simpler.
+
+    IMPORTANT: This generator uses the "title" helper method which is generated
+    by the nifty_layout generator. You may want to run that generator first.
+
+Usage:
+    If you do not pass any arguments, the model name will default to "user", and
+    the authentication controller will default to "session". You can override
+    each of these respectively by passing one or two arguments. Either name can
+    be CamelCased or under_scored.
+
+    Make sure to setup the authlogic gem if you are using that option.
+
+      gem "authlogic" # in Gemfile
+
+Examples:
+    rails generate nifty:authentication
+
+        Creates user model, users_controller, and sessions_controller.
+
+    rails generate nifty:authentication account
+
+        Creates account model, accounts_controller, and sessions_controller.
+
+    rails generate nifty:authentication Account UserSession
+
+        Creates account model, accounts_controller, and user_sessions_controller.
+
+Methods:
+    There are several methods generated which you can use in your application.
+    Here's a common example of what you might add to your layout.
+
+      <% if logged_in? %>
+        Welcome <%= current_user.username %>! Not you?
+        <%= link_to "Log out", logout_path %>
+      <% else %>
+        <%= link_to "Sign up", signup_path %> or
+        <%= link_to "log in", login_path %>.
+      <% end %>
+
+    You can also restrict unregistered users from accessing a controller using
+    a before filter. For example.
+
+      before_filter :login_required, :except => [:index, :show]
+
+    See the generated file lib/authentication.rb for details.

data/lib/generators/nifty/authentication/authentication_generator.rb

@@ -0,0 +1,154 @@
+require 'generators/nifty'
+require 'rails/generators/migration'
+
+module Nifty
+  module Generators
+    class AuthenticationGenerator < Base
+      include Rails::Generators::Migration
+
+      argument :user_name, :type => :string, :default => 'user', :banner => 'user_name'
+      argument :session_name, :type => :string, :default => '[[DEFAULT]]', :banner => 'sessions_controller_name'
+
+      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
+
+      class_option :haml, :desc => 'Generate HAML views instead of ERB.', :type => :boolean
+      class_option :authlogic, :desc => 'Use Authlogic for authentication.', :type => :boolean
+
+      def add_gems
+        add_gem "bcrypt-ruby", :require => "bcrypt"
+        add_gem "mocha", :group => :test
+      end
+
+      def create_model_files
+        template 'user.rb', "app/models/#{user_singular_name}.rb"
+        template 'authlogic_session.rb', "app/models/#{user_singular_name}_session.rb" if options.authlogic?
+      end
+
+      def create_controller_files
+        template 'users_controller.rb', "app/controllers/#{user_plural_name}_controller.rb"
+        template 'sessions_controller.rb', "app/controllers/#{session_plural_name}_controller.rb"
+      end
+
+      def create_helper_files
+        template 'users_helper.rb', "app/helpers/#{user_plural_name}_helper.rb"
+        template 'sessions_helper.rb', "app/helpers/#{session_plural_name}_helper.rb"
+      end
+
+      def create_view_files
+        template "views/#{view_language}/signup.html.#{view_language}", "app/views/#{user_plural_name}/new.html.#{view_language}"
+        template "views/#{view_language}/edit.html.#{view_language}", "app/views/#{user_plural_name}/edit.html.#{view_language}"
+        template "views/#{view_language}/_form.html.#{view_language}", "app/views/#{user_plural_name}/_form.html.#{view_language}"
+        template "views/#{view_language}/login.html.#{view_language}", "app/views/#{session_plural_name}/new.html.#{view_language}"
+      end
+
+      def create_lib_files
+        template 'controller_authentication.rb', 'lib/controller_authentication.rb'
+      end
+
+      def create_routes
+        route "resources #{user_plural_name.to_sym.inspect}"
+        route "resources #{session_plural_name.to_sym.inspect}"
+        route "match 'login' => '#{session_plural_name}#new', :as => :login"
+        route "match 'logout' => '#{session_plural_name}#destroy', :as => :logout"
+        route "match 'signup' => '#{user_plural_name}#new', :as => :signup"
+        route "match '#{user_singular_name}/edit' => '#{user_plural_name}#edit', :as => :edit_current_#{user_singular_name}"
+      end
+
+      def create_migration
+        migration_template 'migration.rb', "db/migrate/create_#{user_plural_name}.rb"
+      end
+
+      def load_and_include_authentication
+        inject_into_class "config/application.rb", "Application", "    config.autoload_paths << \"\#{config.root}/lib\""
+        inject_into_class "app/controllers/application_controller.rb", "ApplicationController", "  include ControllerAuthentication\n"
+      end
+
+      def create_test_files
+        if test_framework == :rspec
+          template 'fixtures.yml', "spec/fixtures/#{user_plural_name}.yml"
+          template 'tests/rspec/user.rb', "spec/models/#{user_singular_name}_spec.rb"
+          template 'tests/rspec/users_controller.rb', "spec/controllers/#{user_plural_name}_controller_spec.rb"
+          template 'tests/rspec/sessions_controller.rb', "spec/controllers/#{session_plural_name}_controller_spec.rb"
+        else
+          template 'fixtures.yml', "test/fixtures/#{user_plural_name}.yml"
+          template "tests/#{test_framework}/user.rb", "test/unit/#{user_singular_name}_test.rb"
+          template "tests/#{test_framework}/users_controller.rb", "test/functional/#{user_plural_name}_controller_test.rb"
+          template "tests/#{test_framework}/sessions_controller.rb", "test/functional/#{session_plural_name}_controller_test.rb"
+        end
+      end
+
+      private
+
+      def session_name
+        @_session_name ||= @session_name == '[[DEFAULT]]' ?
+          (options.authlogic? ? user_name + '_session' : 'session') :
+          @session_name
+      end
+
+      def user_singular_name
+        user_name.underscore
+      end
+
+      def user_plural_name
+        user_singular_name.pluralize
+      end
+
+      def user_class_name
+        user_name.camelize
+      end
+
+      def user_plural_class_name
+        user_plural_name.camelize
+      end
+
+      def session_singular_name
+        session_name.underscore
+      end
+
+      def session_plural_name
+        session_singular_name.pluralize
+      end
+
+      def session_class_name
+        session_name.camelize
+      end
+
+      def session_plural_class_name
+        session_plural_name.camelize
+      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 = File.exist?(destination_path('spec')) ? :rspec : :testunit
+        end
+      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