Module: Aptible::CLI::Helpers::Token

Included in:
Agent, App, Database, Environment
Defined in:
lib/aptible/cli/helpers/token.rb

Constant Summary collapse

TOKEN_ENV_VAR =
'APTIBLE_ACCESS_TOKEN'.freeze

Instance Method Summary collapse

Instance Method Details

#current_token_hashObject



33
34
35
36
37
38
39
40
# File 'lib/aptible/cli/helpers/token.rb', line 33

def current_token_hash
  # NOTE: older versions of the CLI did not properly create the
  # token_file with mode 600, which is why we update it when reading.
  File.chmod(0o600, token_file)
  JSON.parse(File.read(token_file))
rescue
  {}
end

#fetch_tokenObject

Raises:

  • (Thor::Error)


9
10
11
12
13
14
15
# File 'lib/aptible/cli/helpers/token.rb', line 9

def fetch_token
  @token ||= ENV[TOKEN_ENV_VAR] ||
             current_token_hash[Aptible::Auth.configuration.root_url]
  return @token if @token
  raise Thor::Error, 'Could not read token: please run aptible login ' \
                     "or set #{TOKEN_ENV_VAR}"
end

#save_token(token) ⇒ Object



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# File 'lib/aptible/cli/helpers/token.rb', line 17

def save_token(token)
  hash = current_token_hash.merge(
    Aptible::Auth.configuration.root_url => token
  )

  FileUtils.mkdir_p(File.dirname(token_file))

  File.open(token_file, 'w', 0o600) do |file|
    file.puts hash.to_json
  end
rescue StandardError => e
  m = "Could not write token to #{token_file}: #{e}. " \
      'Check filesystem permissions.'
  raise Thor::Error, m
end

#token_fileObject



42
43
44
# File 'lib/aptible/cli/helpers/token.rb', line 42

def token_file
  File.join ENV['HOME'], '.aptible', 'tokens.json'
end