Class: GitHubPr::FileToken

Inherits:
Token
  • Object
show all
Defined in:
lib/github-pr/token.rb

Instance Method Summary collapse

Methods inherited from Token

create

Constructor Details

#initialize(token_file_path) ⇒ FileToken

Returns a new instance of FileToken.



42
43
44
 
# File 'lib/github-pr/token.rb', line 42

def initialize(token_file_path)
	@token_file_path = token_file_path
end

Instance Method Details

#create_dir_if_needed ⇒ Object 



78
79
80
81
82
83
 
# File 'lib/github-pr/token.rb', line 78

def create_dir_if_needed
	token_dir = File.dirname(@token_file_path)
	unless File.directory?(token_dir)
		FileUtils.mkdir_p(token_dir)
	end
end

#get(regenerate: false) ⇒ Object 



93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/github-pr/token.rb', line 93

def get(regenerate: false)
			create_dir_if_needed()

			if regenerate
puts 'Token has been reveked'
FileUtils.rm(@token_file_path)
			end

			token = token_from_file()
			if token
return token
			end
			puts 'Token is nil regenerating'

			token = token_from_github()
  File.open(@token_file_path, 'w+') {|file| file.write token}
  return token
end

#get_username_and_pass ⇒ Object 



85
86
87
88
89
90
91
 
# File 'lib/github-pr/token.rb', line 85

def get_username_and_pass
	cli = HighLine.new

	github_username = cli.ask("GitHub Username: ")
	github_password = cli.ask("GitHub Password(Only one time): ") { |q| q.echo = "*" }
	return {:login => github_username, :password => github_password}
end

#retry_logic(count, initial_note) ⇒ Object 



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/github-pr/token.rb', line 50

def retry_logic(count, initial_note)
	note = initial_note
	count.times do |i|
    	begin
    		yield(note)
    		break
    	rescue Exception => e 
    		if i == 8
    			raise 'Please remove some personal tokens from your github account, you have too many tokens'
    		end
    		if e.errors[0][:code] != "already_exists" || e.errors[0][:field] != "description"
    			raise e 
    		else
    			note = initial_note + (i + 2).to_s
    		end
    	end
    end
end

#token_from_file ⇒ Object 



46
47
48
 
# File 'lib/github-pr/token.rb', line 46

def token_from_file
	File.exist?(@token_file_path) ? IO.read(@token_file_path) : nil
end

#token_from_github ⇒ Object 



69
70
71
72
73
74
75
76
 
# File 'lib/github-pr/token.rb', line 69

def token_from_github
	client = Octokit::Client.new(get_username_and_pass())
    # trying 9 different notes, if all is failed, 
    # then you have to remove some of them
    retry_logic(9, 'CLI uses') do |note|
    	return client.create_authorization(:scopes => ["repo", "user"], :note => note)[:token]
    end
end