blob: 17be1e428715fce07fb05f43cb1244483b721ba0 [file] [log] [blame]
#!/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.
depot-tools-auth login
depot-tools-auth info
depot-tools-auth logout
from __future__ import print_function
import logging
import optparse
import sys
import os
import auth
import setup_color
import subcommand
__version__ = '1.0'
def CMDlogin(parser, args):
"""Performs interactive login and caches authentication token."""
# Forcefully relogin, revoking previous token.
hostname, authenticator = parser.parse_args(args)
print_token_info(hostname, authenticator)
return 0
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
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('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):
self, *args, prog='depot-tools-auth', version=__version__, **kwargs)
'-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__)
return dispatcher.execute(OptionParser(), argv)
except auth.AuthenticationError as e:
print(e, file=sys.stderr)
return 1
if __name__ == '__main__':
except KeyboardInterrupt: