Class: Heroku::Command::Auth

Inherits:
Base
  • Object
show all
Defined in:
lib/heroku/commands/auth.rb

Instance Attribute Summary collapse

Attributes inherited from Base

#args, #autodetected_app

Instance Method Summary collapse

Methods inherited from Base

#app_urls, #ask, #display, #error, #escape, #extract_app, #extract_app_in_dir, #extract_option, #format_date, #git_remotes, #git_url, #heroku, #initialize, #shell, #web_url

Methods included from Helpers

#home_directory, #running_on_a_mac?, #running_on_windows?

Constructor Details

This class inherits a constructor from Heroku::Command::Base

Instance Attribute Details

#credentials ⇒ Object

Returns the value of attribute credentials.



3
4
5
 
# File 'lib/heroku/commands/auth.rb', line 3

def credentials
  @credentials
end

Instance Method Details

#ask_for_credentials ⇒ Object 



59
60
61
62
63
64
65
66
67
68
69
 
# File 'lib/heroku/commands/auth.rb', line 59

def ask_for_credentials
	puts "Enter your Heroku credentials."

	print "Email: "
	user = ask

	print "Password: "
	password = running_on_windows? ? ask_for_password_on_windows : ask_for_password

	[ user, password ]
end

#ask_for_password ⇒ Object 



88
89
90
91
92
93
94
 
# File 'lib/heroku/commands/auth.rb', line 88

def ask_for_password
	echo_off
	password = ask
	puts
	echo_on
	return password
end

#ask_for_password_on_windows ⇒ Object 



71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/heroku/commands/auth.rb', line 71

def ask_for_password_on_windows
	require "Win32API"
	char = nil
	password = ''

	while char = Win32API.new("crtdll", "_getch", [ ], "L").Call do
		break if char == 10 || char == 13 # received carriage return or newline
		if char == 127 || char == 8 # backspace and delete
			password.slice!(-1, 1)
		else
			password << char.chr
		end
	end
	puts
	return password
end

#client ⇒ Object 



5
6
7
 
# File 'lib/heroku/commands/auth.rb', line 5

def client
	@client ||= init_heroku
end

#credentials_file ⇒ Object 



34
35
36
 
# File 'lib/heroku/commands/auth.rb', line 34

def credentials_file
	"#{home_directory}/.heroku/credentials"
end

#delete_credentials ⇒ Object 



133
134
135
 
# File 'lib/heroku/commands/auth.rb', line 133

def delete_credentials
	FileUtils.rm_f(credentials_file)
end

#echo_off ⇒ Object 



51
52
53
 
# File 'lib/heroku/commands/auth.rb', line 51

def echo_off
	system "stty -echo"
end

#echo_on ⇒ Object 



55
56
57
 
# File 'lib/heroku/commands/auth.rb', line 55

def echo_on
	system "stty echo"
end

#get_credentials ⇒ Object

:nodoc:



38
39
40
41
42
43
44
45
 
# File 'lib/heroku/commands/auth.rb', line 38

def get_credentials    # :nodoc:
	return if @credentials
	unless @credentials = read_credentials
		@credentials = ask_for_credentials
		save_credentials
	end
	@credentials
end

#host ⇒ Object 



15
16
17
 
# File 'lib/heroku/commands/auth.rb', line 15

def host
	ENV['HEROKU_HOST'] || 'heroku.com'
end

#init_heroku ⇒ Object 



9
10
11
12
13
 
# File 'lib/heroku/commands/auth.rb', line 9

def init_heroku
	client = Heroku::Client.new(user, password, host)
	client.on_warning { |msg| self.display("\n#{msg}\n\n") }
	client
end

#password ⇒ Object

:nodoc:



29
30
31
32
 
# File 'lib/heroku/commands/auth.rb', line 29

def password    # :nodoc:
	get_credentials
	@credentials[1]
end

#read_credentials ⇒ Object 



47
48
49
 
# File 'lib/heroku/commands/auth.rb', line 47

def read_credentials
	File.exists?(credentials_file) and File.read(credentials_file).split("\n")
end

#reauthorize ⇒ Object 



19
20
21
22
 
# File 'lib/heroku/commands/auth.rb', line 19

def reauthorize
	@credentials = ask_for_credentials
	write_credentials
end

#retry_login? ⇒ Boolean

Returns:

  • (Boolean) 


114
115
116
117
118
 
# File 'lib/heroku/commands/auth.rb', line 114

def retry_login?
	@login_attempts ||= 0
	@login_attempts += 1
	@login_attempts < 3
end

#save_credentials ⇒ Object 



96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
 
# File 'lib/heroku/commands/auth.rb', line 96

def save_credentials
	begin
		write_credentials
		Heroku::Command.run_internal('keys:add', args)
	rescue RestClient::Unauthorized => e
		delete_credentials
		raise e unless retry_login?

		display "\nAuthentication failed"
		@credentials = ask_for_credentials
		@client = init_heroku
		retry
	rescue Exception => e
		delete_credentials
		raise e
	end
end

#set_credentials_permissions ⇒ Object 



128
129
130
131
 
# File 'lib/heroku/commands/auth.rb', line 128

def set_credentials_permissions
	FileUtils.chmod 0700, File.dirname(credentials_file)
	FileUtils.chmod 0600, credentials_file
end

#user ⇒ Object

:nodoc:



24
25
26
27
 
# File 'lib/heroku/commands/auth.rb', line 24

def user    # :nodoc:
	get_credentials
	@credentials[0]
end

#write_credentials ⇒ Object 



120
121
122
123
124
125
126
 
# File 'lib/heroku/commands/auth.rb', line 120

def write_credentials
	FileUtils.mkdir_p(File.dirname(credentials_file))
	File.open(credentials_file, 'w') do |f|
		f.puts self.credentials
	end
	set_credentials_permissions
end