Class: Infinum::Omniauth::Generators::ConfigGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Defined in:
lib/generators/infinum/omniauth/config_generator.rb

Constant Summary collapse

CONFIGURATION = 
{}
INITIALIZER_LOCATION = 
"config/initializers/omniauth.rb"
OMNIAUTH_CONTROLLER_LOCATION = 
"app/controllers/omniauth_controller.rb"

Instance Method Summary collapse

Instance Method Details

#add_omniauth_user_creation ⇒ Object

create_user_model



78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 78

def add_omniauth_user_creation
	method = 	"\n"\
				"	def self.new_from_omniauth(omniauth)\n"\
						"		user = self.find_by_uid(omniauth['uid']) || self.find_by_email(omniauth[:extra][:email]) || self.new(:uid => omniauth['uid'])\n"\
						"		user.uid = omniauth['uid']\n"\
						"		user.first_name = omniauth['extra']['first_name']\n"\
						"		user.last_name = omniauth['extra']['last_name']\n"\
						"		user.email = omniauth['extra']['email']\n"\
						"		user.avatar_url = omniauth['extra']['avatar_url']\n"\
						"		user.save\n"\
						"		user\n"\
						"	end"
			append_to_file 	"app/models/#{user_model_name.downcase}.rb", method,
					after: Regexp.new("^.*" + Regexp.escape(" < ActiveRecord::Base") + ".*$", Regexp::IGNORECASE)
end

#ask_user_for_configuration ⇒ Object

copy_configuration



40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 40

def ask_user_for_configuration
	return nil if CONFIGURATION['no_config']
	# User model name
	CONFIGURATION['user_model_name'] = ask "Define user model name:"
	CONFIGURATION['user_model_name'] = "User" if check_option CONFIGURATION['user_model_name']
	# App ID
	CONFIGURATION['app_id'] = ask "App id:"
	CONFIGURATION['app_id'] = "APP_ID" if check_option CONFIGURATION['app_id']
	# App secret
	CONFIGURATION['app_secret'] = ask "App secret:"
	CONFIGURATION['app_secret'] = "APP_SECRET" if check_option CONFIGURATION['app_secret']
end

#check_installation ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 16

def check_installation
	installation_files = [INITIALIZER_LOCATION, OMNIAUTH_CONTROLLER_LOCATION]
	valid_install = true
	installation_files.each do |file_location|
		valid_install = valid_install && File.exists?(file_location)
	end
	if !valid_install
		say "###############", :red
		say "### WARNING ###", :red
		say "###############", :red
		say "None or partial installation was detected!"
		say "Please run the install generator before"
		say "running this generator:"
		say "  rails generate omniauth_infinum:install"
	end
end

#copy_configuration ⇒ Object

check_installation



33
34
35
36
37
38
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 33

def copy_configuration
	# This is just to make the configuration writeable
	options.each do |key, value|
		CONFIGURATION[key] = value
	end
end

#create_user_model ⇒ Object

explane_to_user



53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 53

def create_user_model
	if File.exists? "models/#{user_model_name}"
		# Generate migration for existing model
		say "Generating migration files for #{user_model_name} model", :green
		command = 	"cd #{Rails.root.to_s};"\
					"rails g migration add_auth_to_#{user_model_name.downcase}"\
					" first_name:string last_name:string email:string avatar_url:string uid:string"
		system command
		# Warn the user
		say ""
		say "############", :red
		say "### NOTE ###", :red
		say "############", :red
		say "Unnecessary columns should be deleted by hand"
		say "from the newly generated migration file!"
		say ""
	else
		# Generate model
		say "Generating #{user_model_name} model", :green
		command = 	"cd #{Rails.root.to_s};"\
					"rails g model #{user_model_name} first_name:string last_name:string email:string avatar_url:string uid:string"
		system command
	end
end

#rename_user_model_in_controllers ⇒ Object

add_omniauth_user_creation



94
95
96
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 94

def rename_user_model_in_controllers
	gsub_file OMNIAUTH_CONTROLLER_LOCATION, "USER_MODEL", user_model_name
end

#set_app_id_and_secret ⇒ Object

rename_user_model_in_controllers



98
99
100
101
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 98

def set_app_id_and_secret
	gsub_file INITIALIZER_LOCATION, "APP_ID", app_id
	gsub_file INITIALIZER_LOCATION, "APP_SECRET", app_secret
end

#set_user_model_for_authentification ⇒ Object

set_app_id_and_secret



103
104
105
 
# File 'lib/generators/infinum/omniauth/config_generator.rb', line 103

def set_user_model_for_authentification
	gsub_file "app/controllers/application_controller.rb", "USER_MODEL", user_model_name
end