2008-10-21 14:00:00 +00:00
|
|
|
#
|
|
|
|
# Copyright (C) 2008 The Android Open Source Project
|
|
|
|
#
|
|
|
|
# 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.
|
|
|
|
|
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import subprocess
|
2009-04-11 01:53:46 +00:00
|
|
|
import tempfile
|
2008-10-21 14:00:00 +00:00
|
|
|
from error import GitError
|
2009-04-18 16:54:51 +00:00
|
|
|
from trace import REPO_TRACE, IsTrace, Trace
|
2008-10-21 14:00:00 +00:00
|
|
|
|
|
|
|
GIT = 'git'
|
|
|
|
MIN_GIT_VERSION = (1, 5, 4)
|
|
|
|
GIT_DIR = 'GIT_DIR'
|
|
|
|
|
|
|
|
LAST_GITDIR = None
|
|
|
|
LAST_CWD = None
|
|
|
|
|
2009-04-11 01:53:46 +00:00
|
|
|
_ssh_proxy_path = None
|
|
|
|
_ssh_sock_path = None
|
|
|
|
|
2010-04-27 21:35:27 +00:00
|
|
|
def ssh_sock(create=True):
|
2009-04-11 01:53:46 +00:00
|
|
|
global _ssh_sock_path
|
|
|
|
if _ssh_sock_path is None:
|
|
|
|
if not create:
|
|
|
|
return None
|
2009-04-21 15:05:27 +00:00
|
|
|
dir = '/tmp'
|
|
|
|
if not os.path.exists(dir):
|
|
|
|
dir = tempfile.gettempdir()
|
2009-04-11 01:53:46 +00:00
|
|
|
_ssh_sock_path = os.path.join(
|
2009-04-21 15:05:27 +00:00
|
|
|
tempfile.mkdtemp('', 'ssh-', dir),
|
2009-04-11 01:53:46 +00:00
|
|
|
'master-%r@%h:%p')
|
|
|
|
return _ssh_sock_path
|
|
|
|
|
|
|
|
def _ssh_proxy():
|
|
|
|
global _ssh_proxy_path
|
|
|
|
if _ssh_proxy_path is None:
|
|
|
|
_ssh_proxy_path = os.path.join(
|
|
|
|
os.path.dirname(__file__),
|
|
|
|
'git_ssh')
|
|
|
|
return _ssh_proxy_path
|
|
|
|
|
2008-10-21 14:00:00 +00:00
|
|
|
|
|
|
|
class _GitCall(object):
|
|
|
|
def version(self):
|
|
|
|
p = GitCommand(None, ['--version'], capture_stdout=True)
|
|
|
|
if p.Wait() == 0:
|
|
|
|
return p.stdout
|
|
|
|
return None
|
|
|
|
|
|
|
|
def __getattr__(self, name):
|
|
|
|
name = name.replace('_','-')
|
|
|
|
def fun(*cmdv):
|
|
|
|
command = [name]
|
|
|
|
command.extend(cmdv)
|
|
|
|
return GitCommand(None, command).Wait() == 0
|
|
|
|
return fun
|
|
|
|
git = _GitCall()
|
|
|
|
|
2009-06-12 16:32:50 +00:00
|
|
|
_git_version = None
|
|
|
|
|
|
|
|
def git_require(min_version, fail=False):
|
|
|
|
global _git_version
|
|
|
|
|
|
|
|
if _git_version is None:
|
|
|
|
ver_str = git.version()
|
|
|
|
if ver_str.startswith('git version '):
|
|
|
|
_git_version = tuple(
|
|
|
|
map(lambda x: int(x),
|
|
|
|
ver_str[len('git version '):].strip().split('.')[0:3]
|
|
|
|
))
|
|
|
|
else:
|
|
|
|
print >>sys.stderr, 'fatal: "%s" unsupported' % ver_str
|
|
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
if min_version <= _git_version:
|
|
|
|
return True
|
|
|
|
if fail:
|
|
|
|
need = '.'.join(map(lambda x: str(x), min_version))
|
|
|
|
print >>sys.stderr, 'fatal: git %s or later required' % need
|
|
|
|
sys.exit(1)
|
|
|
|
return False
|
|
|
|
|
2008-10-21 14:00:00 +00:00
|
|
|
class GitCommand(object):
|
|
|
|
def __init__(self,
|
|
|
|
project,
|
|
|
|
cmdv,
|
|
|
|
bare = False,
|
|
|
|
provide_stdin = False,
|
|
|
|
capture_stdout = False,
|
|
|
|
capture_stderr = False,
|
|
|
|
disable_editor = False,
|
2009-04-11 01:53:46 +00:00
|
|
|
ssh_proxy = False,
|
2008-10-21 14:00:00 +00:00
|
|
|
cwd = None,
|
|
|
|
gitdir = None):
|
|
|
|
env = dict(os.environ)
|
|
|
|
|
|
|
|
for e in [REPO_TRACE,
|
|
|
|
GIT_DIR,
|
|
|
|
'GIT_ALTERNATE_OBJECT_DIRECTORIES',
|
|
|
|
'GIT_OBJECT_DIRECTORY',
|
|
|
|
'GIT_WORK_TREE',
|
|
|
|
'GIT_GRAFT_FILE',
|
|
|
|
'GIT_INDEX_FILE']:
|
|
|
|
if e in env:
|
|
|
|
del env[e]
|
|
|
|
|
|
|
|
if disable_editor:
|
|
|
|
env['GIT_EDITOR'] = ':'
|
2009-04-11 01:53:46 +00:00
|
|
|
if ssh_proxy:
|
2010-04-27 21:35:27 +00:00
|
|
|
env['REPO_SSH_SOCK'] = ssh_sock()
|
2009-04-11 01:53:46 +00:00
|
|
|
env['GIT_SSH'] = _ssh_proxy()
|
2008-10-21 14:00:00 +00:00
|
|
|
|
|
|
|
if project:
|
|
|
|
if not cwd:
|
|
|
|
cwd = project.worktree
|
|
|
|
if not gitdir:
|
|
|
|
gitdir = project.gitdir
|
|
|
|
|
|
|
|
command = [GIT]
|
|
|
|
if bare:
|
|
|
|
if gitdir:
|
|
|
|
env[GIT_DIR] = gitdir
|
|
|
|
cwd = None
|
|
|
|
command.extend(cmdv)
|
|
|
|
|
|
|
|
if provide_stdin:
|
|
|
|
stdin = subprocess.PIPE
|
|
|
|
else:
|
|
|
|
stdin = None
|
|
|
|
|
|
|
|
if capture_stdout:
|
|
|
|
stdout = subprocess.PIPE
|
|
|
|
else:
|
|
|
|
stdout = None
|
|
|
|
|
|
|
|
if capture_stderr:
|
|
|
|
stderr = subprocess.PIPE
|
|
|
|
else:
|
|
|
|
stderr = None
|
|
|
|
|
2009-04-18 16:54:51 +00:00
|
|
|
if IsTrace():
|
2008-10-21 14:00:00 +00:00
|
|
|
global LAST_CWD
|
|
|
|
global LAST_GITDIR
|
|
|
|
|
|
|
|
dbg = ''
|
|
|
|
|
|
|
|
if cwd and LAST_CWD != cwd:
|
|
|
|
if LAST_GITDIR or LAST_CWD:
|
|
|
|
dbg += '\n'
|
|
|
|
dbg += ': cd %s\n' % cwd
|
|
|
|
LAST_CWD = cwd
|
|
|
|
|
|
|
|
if GIT_DIR in env and LAST_GITDIR != env[GIT_DIR]:
|
|
|
|
if LAST_GITDIR or LAST_CWD:
|
|
|
|
dbg += '\n'
|
|
|
|
dbg += ': export GIT_DIR=%s\n' % env[GIT_DIR]
|
|
|
|
LAST_GITDIR = env[GIT_DIR]
|
|
|
|
|
|
|
|
dbg += ': '
|
|
|
|
dbg += ' '.join(command)
|
|
|
|
if stdin == subprocess.PIPE:
|
|
|
|
dbg += ' 0<|'
|
|
|
|
if stdout == subprocess.PIPE:
|
|
|
|
dbg += ' 1>|'
|
|
|
|
if stderr == subprocess.PIPE:
|
|
|
|
dbg += ' 2>|'
|
2009-04-18 16:54:51 +00:00
|
|
|
Trace('%s', dbg)
|
2008-10-21 14:00:00 +00:00
|
|
|
|
|
|
|
try:
|
|
|
|
p = subprocess.Popen(command,
|
|
|
|
cwd = cwd,
|
|
|
|
env = env,
|
|
|
|
stdin = stdin,
|
|
|
|
stdout = stdout,
|
|
|
|
stderr = stderr)
|
|
|
|
except Exception, e:
|
|
|
|
raise GitError('%s: %s' % (command[1], e))
|
|
|
|
|
|
|
|
self.process = p
|
|
|
|
self.stdin = p.stdin
|
|
|
|
|
|
|
|
def Wait(self):
|
|
|
|
p = self.process
|
|
|
|
|
|
|
|
if p.stdin:
|
|
|
|
p.stdin.close()
|
|
|
|
self.stdin = None
|
|
|
|
|
|
|
|
if p.stdout:
|
|
|
|
self.stdout = p.stdout.read()
|
|
|
|
p.stdout.close()
|
|
|
|
else:
|
|
|
|
p.stdout = None
|
|
|
|
|
|
|
|
if p.stderr:
|
|
|
|
self.stderr = p.stderr.read()
|
|
|
|
p.stderr.close()
|
|
|
|
else:
|
|
|
|
p.stderr = None
|
|
|
|
|
|
|
|
return self.process.wait()
|