git-googlesource-login

#!/usr/bin/python
# Copyright (C) 2012 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
Configures git to authenticate to *.googlesource.com.

Adds a cookie to ~/.git-credential-cache/cookies and sets
Git variable http.cookiefile to this path.
"""

import cookielib
import os
from subprocess import Popen, PIPE, check_call

def write_cookie(token):
  git_config = ['git', 'config', '--global', 'http.cookiefile']
  set_config = False

  fn = Popen(git_config, stdout=PIPE).communicate()[0]
  if fn == '':
    home = os.environ['HOME']
    fn = os.path.join(home, '.git-credential-cache', 'cookies')
    set_config = True
  elif fn[-1] == '\n':
    fn = fn[:-1]

  dir = os.path.dirname(fn)
  if os.path.exists(dir):
    os.chmod(dir, 0700)
  else:
    os.mkdir(dir, 0700)

  cj = cookielib.MozillaCookieJar(fn)
  if os.path.exists(cj.filename):
    cj.load()
  cj.set_cookie(cookielib.Cookie(
    version = 0,
    name = 'o',
    value = token,
    port = None,
    port_specified = False,
    domain = '.googlesource.com',
    domain_specified = True,
    domain_initial_dot = True,
    path = '/',
    path_specified = True,
    secure = True,
    expires = (1 << 31) - 1,
    discard = False,
    comment = None,
    comment_url = None,
    rest = None))
  cj.save()
  os.chmod(cj.filename, 0600)
  if set_config:
    check_call(git_config + [cj.filename])
  return cj.filename

def main():
  print 'Open https://www.googlesource.com/new-password ...'
  t = raw_input("Password (under 'Login for Git'): ")
  p = write_cookie(t)
  print "Saved to %s" % (p,)

if __name__ == '__main__':
  main()