You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
#!/usr/bin/env python
|
|
# Copyright 2015 The Chromium Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Manages cached OAuth2 tokens used by other depot_tools scripts.
|
|
|
|
Usage:
|
|
depot-tools-auth login codereview.chromium.org
|
|
depot-tools-auth info codereview.chromium.org
|
|
depot-tools-auth logout codereview.chromium.org
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
|
|
import logging
|
|
import optparse
|
|
import sys
|
|
import os
|
|
|
|
import auth
|
|
import setup_color
|
|
import subcommand
|
|
|
|
__version__ = '1.0'
|
|
|
|
|
|
@subcommand.usage('<hostname>')
|
|
def CMDlogin(parser, args):
|
|
"""Performs interactive login and caches authentication token."""
|
|
# Forcefully relogin, revoking previous token.
|
|
hostname, authenticator = parser.parse_args(args)
|
|
authenticator.logout()
|
|
authenticator.login()
|
|
print_token_info(hostname, authenticator)
|
|
return 0
|
|
|
|
|
|
@subcommand.usage('<hostname>')
|
|
def CMDlogout(parser, args):
|
|
"""Revokes cached authentication token and removes it from disk."""
|
|
_, authenticator = parser.parse_args(args)
|
|
done = authenticator.logout()
|
|
print('Done.' if done else 'Already logged out.')
|
|
return 0
|
|
|
|
|
|
@subcommand.usage('<hostname>')
|
|
def CMDinfo(parser, args):
|
|
"""Shows email associated with a cached authentication token."""
|
|
# If no token is cached, AuthenticationError will be caught in 'main'.
|
|
hostname, authenticator = parser.parse_args(args)
|
|
print_token_info(hostname, authenticator)
|
|
return 0
|
|
|
|
|
|
def print_token_info(hostname, authenticator):
|
|
token_info = authenticator.get_token_info()
|
|
print('Logged in to %s as %s.' % (hostname, token_info['email']))
|
|
print('')
|
|
print('To login with a different email run:')
|
|
print(' depot-tools-auth login %s' % hostname)
|
|
print('To logout and purge the authentication token run:')
|
|
print(' depot-tools-auth logout %s' % hostname)
|
|
|
|
|
|
class OptionParser(optparse.OptionParser):
|
|
def __init__(self, *args, **kwargs):
|
|
optparse.OptionParser.__init__(
|
|
self, *args, prog='depot-tools-auth', version=__version__, **kwargs)
|
|
self.add_option(
|
|
'-v', '--verbose', action='count', default=0,
|
|
help='Use 2 times for more debugging info')
|
|
auth.add_auth_options(self, auth.make_auth_config(use_oauth2=True))
|
|
|
|
def parse_args(self, args=None, values=None):
|
|
"""Parses options and returns (hostname, auth.Authenticator object)."""
|
|
options, args = optparse.OptionParser.parse_args(self, args, values)
|
|
levels = [logging.WARNING, logging.INFO, logging.DEBUG]
|
|
logging.basicConfig(level=levels[min(options.verbose, len(levels) - 1)])
|
|
auth_config = auth.extract_auth_config_from_options(options)
|
|
if len(args) != 1:
|
|
self.error('Expecting single argument (hostname).')
|
|
if not auth_config.use_oauth2:
|
|
self.error('This command is only usable with OAuth2 authentication')
|
|
return args[0], auth.get_authenticator_for_host(args[0], auth_config)
|
|
|
|
|
|
def main(argv):
|
|
dispatcher = subcommand.CommandDispatcher(__name__)
|
|
try:
|
|
return dispatcher.execute(OptionParser(), argv)
|
|
except auth.AuthenticationError as e:
|
|
print(e, file=sys.stderr)
|
|
return 1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
setup_color.init()
|
|
try:
|
|
sys.exit(main(sys.argv[1:]))
|
|
except KeyboardInterrupt:
|
|
sys.stderr.write('interrupted\n')
|
|
sys.exit(1)
|