1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
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
|
