mirror of
https://gerrit.googlesource.com/git-repo
synced 2024-12-21 07:16:21 +00:00
02dbb6d120
If we run out of entries next() will throw StopIteration. Signed-off-by: Shawn O. Pearce <sop@google.com>
1065 lines
29 KiB
Python
1065 lines
29 KiB
Python
# 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 filecmp
|
|
import os
|
|
import re
|
|
import shutil
|
|
import stat
|
|
import sys
|
|
import urllib2
|
|
|
|
from color import Coloring
|
|
from git_command import GitCommand
|
|
from git_config import GitConfig, IsId
|
|
from gerrit_upload import UploadBundle
|
|
from error import GitError, ImportError, UploadError
|
|
from remote import Remote
|
|
from codereview import proto_client
|
|
|
|
HEAD = 'HEAD'
|
|
R_HEADS = 'refs/heads/'
|
|
R_TAGS = 'refs/tags/'
|
|
R_PUB = 'refs/published/'
|
|
R_M = 'refs/remotes/m/'
|
|
|
|
def _warn(fmt, *args):
|
|
msg = fmt % args
|
|
print >>sys.stderr, 'warn: %s' % msg
|
|
|
|
def _info(fmt, *args):
|
|
msg = fmt % args
|
|
print >>sys.stderr, 'info: %s' % msg
|
|
|
|
def not_rev(r):
|
|
return '^' + r
|
|
|
|
class ReviewableBranch(object):
|
|
_commit_cache = None
|
|
|
|
def __init__(self, project, branch, base):
|
|
self.project = project
|
|
self.branch = branch
|
|
self.base = base
|
|
|
|
@property
|
|
def name(self):
|
|
return self.branch.name
|
|
|
|
@property
|
|
def commits(self):
|
|
if self._commit_cache is None:
|
|
self._commit_cache = self.project.bare_git.rev_list(
|
|
'--abbrev=8',
|
|
'--abbrev-commit',
|
|
'--pretty=oneline',
|
|
'--reverse',
|
|
'--date-order',
|
|
not_rev(self.base),
|
|
R_HEADS + self.name,
|
|
'--')
|
|
return self._commit_cache
|
|
|
|
@property
|
|
def date(self):
|
|
return self.project.bare_git.log(
|
|
'--pretty=format:%cd',
|
|
'-n', '1',
|
|
R_HEADS + self.name,
|
|
'--')
|
|
|
|
def UploadForReview(self):
|
|
self.project.UploadForReview(self.name)
|
|
|
|
@property
|
|
def tip_url(self):
|
|
me = self.project.GetBranch(self.name)
|
|
commit = self.project.bare_git.rev_parse(R_HEADS + self.name)
|
|
return 'http://%s/r/%s' % (me.remote.review, commit[0:12])
|
|
|
|
|
|
class StatusColoring(Coloring):
|
|
def __init__(self, config):
|
|
Coloring.__init__(self, config, 'status')
|
|
self.project = self.printer('header', attr = 'bold')
|
|
self.branch = self.printer('header', attr = 'bold')
|
|
self.nobranch = self.printer('nobranch', fg = 'red')
|
|
|
|
self.added = self.printer('added', fg = 'green')
|
|
self.changed = self.printer('changed', fg = 'red')
|
|
self.untracked = self.printer('untracked', fg = 'red')
|
|
|
|
|
|
class DiffColoring(Coloring):
|
|
def __init__(self, config):
|
|
Coloring.__init__(self, config, 'diff')
|
|
self.project = self.printer('header', attr = 'bold')
|
|
|
|
|
|
class _CopyFile:
|
|
def __init__(self, src, dest):
|
|
self.src = src
|
|
self.dest = dest
|
|
|
|
def _Copy(self):
|
|
src = self.src
|
|
dest = self.dest
|
|
# copy file if it does not exist or is out of date
|
|
if not os.path.exists(dest) or not filecmp.cmp(src, dest):
|
|
try:
|
|
# remove existing file first, since it might be read-only
|
|
if os.path.exists(dest):
|
|
os.remove(dest)
|
|
shutil.copy(src, dest)
|
|
# make the file read-only
|
|
mode = os.stat(dest)[stat.ST_MODE]
|
|
mode = mode & ~(stat.S_IWUSR | stat.S_IWGRP | stat.S_IWOTH)
|
|
os.chmod(dest, mode)
|
|
except IOError:
|
|
print >>sys.stderr, \
|
|
'error: Cannot copy file %s to %s' \
|
|
% (src, dest)
|
|
|
|
|
|
class Project(object):
|
|
def __init__(self,
|
|
manifest,
|
|
name,
|
|
remote,
|
|
gitdir,
|
|
worktree,
|
|
relpath,
|
|
revision):
|
|
self.manifest = manifest
|
|
self.name = name
|
|
self.remote = remote
|
|
self.gitdir = gitdir
|
|
self.worktree = worktree
|
|
self.relpath = relpath
|
|
self.revision = revision
|
|
self.snapshots = {}
|
|
self.extraRemotes = {}
|
|
self.copyfiles = []
|
|
self.config = GitConfig.ForRepository(
|
|
gitdir = self.gitdir,
|
|
defaults = self.manifest.globalConfig)
|
|
|
|
self.work_git = self._GitGetByExec(self, bare=False)
|
|
self.bare_git = self._GitGetByExec(self, bare=True)
|
|
|
|
@property
|
|
def Exists(self):
|
|
return os.path.isdir(self.gitdir)
|
|
|
|
@property
|
|
def CurrentBranch(self):
|
|
"""Obtain the name of the currently checked out branch.
|
|
The branch name omits the 'refs/heads/' prefix.
|
|
None is returned if the project is on a detached HEAD.
|
|
"""
|
|
try:
|
|
b = self.work_git.GetHead()
|
|
except GitError:
|
|
return None
|
|
if b.startswith(R_HEADS):
|
|
return b[len(R_HEADS):]
|
|
return None
|
|
|
|
def IsDirty(self, consider_untracked=True):
|
|
"""Is the working directory modified in some way?
|
|
"""
|
|
self.work_git.update_index('-q',
|
|
'--unmerged',
|
|
'--ignore-missing',
|
|
'--refresh')
|
|
if self.work_git.DiffZ('diff-index','-M','--cached',HEAD):
|
|
return True
|
|
if self.work_git.DiffZ('diff-files'):
|
|
return True
|
|
if consider_untracked and self.work_git.LsOthers():
|
|
return True
|
|
return False
|
|
|
|
_userident_name = None
|
|
_userident_email = None
|
|
|
|
@property
|
|
def UserName(self):
|
|
"""Obtain the user's personal name.
|
|
"""
|
|
if self._userident_name is None:
|
|
self._LoadUserIdentity()
|
|
return self._userident_name
|
|
|
|
@property
|
|
def UserEmail(self):
|
|
"""Obtain the user's email address. This is very likely
|
|
to be their Gerrit login.
|
|
"""
|
|
if self._userident_email is None:
|
|
self._LoadUserIdentity()
|
|
return self._userident_email
|
|
|
|
def _LoadUserIdentity(self):
|
|
u = self.bare_git.var('GIT_COMMITTER_IDENT')
|
|
m = re.compile("^(.*) <([^>]*)> ").match(u)
|
|
if m:
|
|
self._userident_name = m.group(1)
|
|
self._userident_email = m.group(2)
|
|
else:
|
|
self._userident_name = ''
|
|
self._userident_email = ''
|
|
|
|
def GetRemote(self, name):
|
|
"""Get the configuration for a single remote.
|
|
"""
|
|
return self.config.GetRemote(name)
|
|
|
|
def GetBranch(self, name):
|
|
"""Get the configuration for a single branch.
|
|
"""
|
|
return self.config.GetBranch(name)
|
|
|
|
|
|
## Status Display ##
|
|
|
|
def PrintWorkTreeStatus(self):
|
|
"""Prints the status of the repository to stdout.
|
|
"""
|
|
if not os.path.isdir(self.worktree):
|
|
print ''
|
|
print 'project %s/' % self.relpath
|
|
print ' missing (run "repo sync")'
|
|
return
|
|
|
|
self.work_git.update_index('-q',
|
|
'--unmerged',
|
|
'--ignore-missing',
|
|
'--refresh')
|
|
di = self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD)
|
|
df = self.work_git.DiffZ('diff-files')
|
|
do = self.work_git.LsOthers()
|
|
if not di and not df and not do:
|
|
return
|
|
|
|
out = StatusColoring(self.config)
|
|
out.project('project %-40s', self.relpath + '/')
|
|
|
|
branch = self.CurrentBranch
|
|
if branch is None:
|
|
out.nobranch('(*** NO BRANCH ***)')
|
|
else:
|
|
out.branch('branch %s', branch)
|
|
out.nl()
|
|
|
|
paths = list()
|
|
paths.extend(di.keys())
|
|
paths.extend(df.keys())
|
|
paths.extend(do)
|
|
|
|
paths = list(set(paths))
|
|
paths.sort()
|
|
|
|
for p in paths:
|
|
try: i = di[p]
|
|
except KeyError: i = None
|
|
|
|
try: f = df[p]
|
|
except KeyError: f = None
|
|
|
|
if i: i_status = i.status.upper()
|
|
else: i_status = '-'
|
|
|
|
if f: f_status = f.status.lower()
|
|
else: f_status = '-'
|
|
|
|
if i and i.src_path:
|
|
line = ' %s%s\t%s => (%s%%)' % (i_status, f_status,
|
|
i.src_path, p, i.level)
|
|
else:
|
|
line = ' %s%s\t%s' % (i_status, f_status, p)
|
|
|
|
if i and not f:
|
|
out.added('%s', line)
|
|
elif (i and f) or (not i and f):
|
|
out.changed('%s', line)
|
|
elif not i and not f:
|
|
out.untracked('%s', line)
|
|
else:
|
|
out.write('%s', line)
|
|
out.nl()
|
|
|
|
def PrintWorkTreeDiff(self):
|
|
"""Prints the status of the repository to stdout.
|
|
"""
|
|
out = DiffColoring(self.config)
|
|
cmd = ['diff']
|
|
if out.is_on:
|
|
cmd.append('--color')
|
|
cmd.append(HEAD)
|
|
cmd.append('--')
|
|
p = GitCommand(self,
|
|
cmd,
|
|
capture_stdout = True,
|
|
capture_stderr = True)
|
|
has_diff = False
|
|
for line in p.process.stdout:
|
|
if not has_diff:
|
|
out.nl()
|
|
out.project('project %s/' % self.relpath)
|
|
out.nl()
|
|
has_diff = True
|
|
print line[:-1]
|
|
p.Wait()
|
|
|
|
|
|
## Publish / Upload ##
|
|
|
|
def WasPublished(self, branch):
|
|
"""Was the branch published (uploaded) for code review?
|
|
If so, returns the SHA-1 hash of the last published
|
|
state for the branch.
|
|
"""
|
|
try:
|
|
return self.bare_git.rev_parse(R_PUB + branch)
|
|
except GitError:
|
|
return None
|
|
|
|
def CleanPublishedCache(self):
|
|
"""Prunes any stale published refs.
|
|
"""
|
|
heads = set()
|
|
canrm = {}
|
|
for name, id in self._allrefs.iteritems():
|
|
if name.startswith(R_HEADS):
|
|
heads.add(name)
|
|
elif name.startswith(R_PUB):
|
|
canrm[name] = id
|
|
|
|
for name, id in canrm.iteritems():
|
|
n = name[len(R_PUB):]
|
|
if R_HEADS + n not in heads:
|
|
self.bare_git.DeleteRef(name, id)
|
|
|
|
def GetUploadableBranches(self):
|
|
"""List any branches which can be uploaded for review.
|
|
"""
|
|
heads = {}
|
|
pubed = {}
|
|
|
|
for name, id in self._allrefs.iteritems():
|
|
if name.startswith(R_HEADS):
|
|
heads[name[len(R_HEADS):]] = id
|
|
elif name.startswith(R_PUB):
|
|
pubed[name[len(R_PUB):]] = id
|
|
|
|
ready = []
|
|
for branch, id in heads.iteritems():
|
|
if branch in pubed and pubed[branch] == id:
|
|
continue
|
|
|
|
branch = self.GetBranch(branch)
|
|
base = branch.LocalMerge
|
|
if branch.LocalMerge:
|
|
rb = ReviewableBranch(self, branch, base)
|
|
if rb.commits:
|
|
ready.append(rb)
|
|
return ready
|
|
|
|
def UploadForReview(self, branch=None):
|
|
"""Uploads the named branch for code review.
|
|
"""
|
|
if branch is None:
|
|
branch = self.CurrentBranch
|
|
if branch is None:
|
|
raise GitError('not currently on a branch')
|
|
|
|
branch = self.GetBranch(branch)
|
|
if not branch.LocalMerge:
|
|
raise GitError('branch %s does not track a remote' % branch.name)
|
|
if not branch.remote.review:
|
|
raise GitError('remote %s has no review url' % branch.remote.name)
|
|
|
|
dest_branch = branch.merge
|
|
if not dest_branch.startswith(R_HEADS):
|
|
dest_branch = R_HEADS + dest_branch
|
|
|
|
base_list = []
|
|
for name, id in self._allrefs.iteritems():
|
|
if branch.remote.WritesTo(name):
|
|
base_list.append(not_rev(name))
|
|
if not base_list:
|
|
raise GitError('no base refs, cannot upload %s' % branch.name)
|
|
|
|
print >>sys.stderr, ''
|
|
_info("Uploading %s to %s:", branch.name, self.name)
|
|
try:
|
|
UploadBundle(project = self,
|
|
server = branch.remote.review,
|
|
email = self.UserEmail,
|
|
dest_project = self.name,
|
|
dest_branch = dest_branch,
|
|
src_branch = R_HEADS + branch.name,
|
|
bases = base_list)
|
|
except proto_client.ClientLoginError:
|
|
raise UploadError('Login failure')
|
|
except urllib2.HTTPError, e:
|
|
raise UploadError('HTTP error %d' % e.code)
|
|
|
|
msg = "posted to %s for %s" % (branch.remote.review, dest_branch)
|
|
self.bare_git.UpdateRef(R_PUB + branch.name,
|
|
R_HEADS + branch.name,
|
|
message = msg)
|
|
|
|
|
|
## Sync ##
|
|
|
|
def Sync_NetworkHalf(self):
|
|
"""Perform only the network IO portion of the sync process.
|
|
Local working directory/branch state is not affected.
|
|
"""
|
|
if not self.Exists:
|
|
print >>sys.stderr
|
|
print >>sys.stderr, 'Initializing project %s ...' % self.name
|
|
self._InitGitDir()
|
|
self._InitRemote()
|
|
for r in self.extraRemotes.values():
|
|
if not self._RemoteFetch(r.name):
|
|
return False
|
|
if not self._SnapshotDownload():
|
|
return False
|
|
if not self._RemoteFetch():
|
|
return False
|
|
self._InitMRef()
|
|
return True
|
|
|
|
def _CopyFiles(self):
|
|
for file in self.copyfiles:
|
|
file._Copy()
|
|
|
|
def Sync_LocalHalf(self):
|
|
"""Perform only the local IO portion of the sync process.
|
|
Network access is not required.
|
|
|
|
Return:
|
|
True: the sync was successful
|
|
False: the sync requires user input
|
|
"""
|
|
self._InitWorkTree()
|
|
self.CleanPublishedCache()
|
|
|
|
rem = self.GetRemote(self.remote.name)
|
|
rev = rem.ToLocal(self.revision)
|
|
branch = self.CurrentBranch
|
|
|
|
if branch is None:
|
|
# Currently on a detached HEAD. The user is assumed to
|
|
# not have any local modifications worth worrying about.
|
|
#
|
|
lost = self._revlist(not_rev(rev), HEAD)
|
|
if lost:
|
|
_info("[%s] Discarding %d commits", self.name, len(lost))
|
|
try:
|
|
self._Checkout(rev, quiet=True)
|
|
except GitError:
|
|
return False
|
|
self._CopyFiles()
|
|
return True
|
|
|
|
branch = self.GetBranch(branch)
|
|
merge = branch.LocalMerge
|
|
|
|
if not merge:
|
|
# The current branch has no tracking configuration.
|
|
# Jump off it to a deatched HEAD.
|
|
#
|
|
_info("[%s] Leaving %s"
|
|
" (does not track any upstream)",
|
|
self.name,
|
|
branch.name)
|
|
try:
|
|
self._Checkout(rev, quiet=True)
|
|
except GitError:
|
|
return False
|
|
self._CopyFiles()
|
|
return True
|
|
|
|
upstream_gain = self._revlist(not_rev(HEAD), rev)
|
|
pub = self.WasPublished(branch.name)
|
|
if pub:
|
|
not_merged = self._revlist(not_rev(rev), pub)
|
|
if not_merged:
|
|
if upstream_gain:
|
|
# The user has published this branch and some of those
|
|
# commits are not yet merged upstream. We do not want
|
|
# to rewrite the published commits so we punt.
|
|
#
|
|
_info("[%s] Branch %s is published,"
|
|
" but is now %d commits behind.",
|
|
self.name, branch.name, len(upstream_gain))
|
|
_info("[%s] Consider merging or rebasing the"
|
|
" unpublished commits.", self.name)
|
|
return True
|
|
|
|
if merge == rev:
|
|
try:
|
|
old_merge = self.bare_git.rev_parse('%s@{1}' % merge)
|
|
except GitError:
|
|
old_merge = merge
|
|
if old_merge == '0000000000000000000000000000000000000000' \
|
|
or old_merge == '':
|
|
old_merge = merge
|
|
else:
|
|
# The upstream switched on us. Time to cross our fingers
|
|
# and pray that the old upstream also wasn't in the habit
|
|
# of rebasing itself.
|
|
#
|
|
_info("[%s] Manifest switched from %s to %s",
|
|
self.name, merge, rev)
|
|
old_merge = merge
|
|
|
|
if rev == old_merge:
|
|
upstream_lost = []
|
|
else:
|
|
upstream_lost = self._revlist(not_rev(rev), old_merge)
|
|
|
|
if not upstream_lost and not upstream_gain:
|
|
# Trivially no changes caused by the upstream.
|
|
#
|
|
return True
|
|
|
|
if self.IsDirty(consider_untracked=False):
|
|
_warn('[%s] commit (or discard) uncommitted changes'
|
|
' before sync', self.name)
|
|
return False
|
|
|
|
if upstream_lost:
|
|
# Upstream rebased. Not everything in HEAD
|
|
# may have been caused by the user.
|
|
#
|
|
_info("[%s] Discarding %d commits removed from upstream",
|
|
self.name, len(upstream_lost))
|
|
|
|
branch.remote = rem
|
|
branch.merge = self.revision
|
|
branch.Save()
|
|
|
|
my_changes = self._revlist(not_rev(old_merge), HEAD)
|
|
if my_changes:
|
|
try:
|
|
self._Rebase(upstream = old_merge, onto = rev)
|
|
except GitError:
|
|
return False
|
|
elif upstream_lost:
|
|
try:
|
|
self._ResetHard(rev)
|
|
except GitError:
|
|
return False
|
|
else:
|
|
try:
|
|
self._FastForward(rev)
|
|
except GitError:
|
|
return False
|
|
|
|
self._CopyFiles()
|
|
return True
|
|
|
|
def _SnapshotDownload(self):
|
|
if self.snapshots:
|
|
have = set(self._allrefs.keys())
|
|
need = []
|
|
|
|
for tag, sn in self.snapshots.iteritems():
|
|
if tag not in have:
|
|
need.append(sn)
|
|
|
|
if need:
|
|
print >>sys.stderr, """
|
|
*** Downloading source(s) from a mirror site. ***
|
|
*** If the network hangs, kill and restart repo. ***
|
|
"""
|
|
for sn in need:
|
|
try:
|
|
sn.Import()
|
|
except ImportError, e:
|
|
print >>sys.stderr, \
|
|
'error: Cannot import %s: %s' \
|
|
% (self.name, e)
|
|
return False
|
|
cmd = ['repack', '-a', '-d', '-f', '-l']
|
|
if GitCommand(self, cmd, bare = True).Wait() != 0:
|
|
return False
|
|
return True
|
|
|
|
def AddCopyFile(self, src, dest):
|
|
# dest should already be an absolute path, but src is project relative
|
|
# make src an absolute path
|
|
src = os.path.join(self.worktree, src)
|
|
self.copyfiles.append(_CopyFile(src, dest))
|
|
|
|
|
|
## Branch Management ##
|
|
|
|
def StartBranch(self, name):
|
|
"""Create a new branch off the manifest's revision.
|
|
"""
|
|
branch = self.GetBranch(name)
|
|
branch.remote = self.GetRemote(self.remote.name)
|
|
branch.merge = self.revision
|
|
|
|
rev = branch.LocalMerge
|
|
cmd = ['checkout', '-b', branch.name, rev]
|
|
if GitCommand(self, cmd).Wait() == 0:
|
|
branch.Save()
|
|
else:
|
|
raise GitError('%s checkout %s ' % (self.name, rev))
|
|
|
|
def PruneHeads(self):
|
|
"""Prune any topic branches already merged into upstream.
|
|
"""
|
|
cb = self.CurrentBranch
|
|
kill = []
|
|
for name in self._allrefs.keys():
|
|
if name.startswith(R_HEADS):
|
|
name = name[len(R_HEADS):]
|
|
if cb is None or name != cb:
|
|
kill.append(name)
|
|
|
|
rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
|
|
if cb is not None \
|
|
and not self._revlist(HEAD + '...' + rev) \
|
|
and not self.IsDirty(consider_untracked = False):
|
|
self.work_git.DetachHead(HEAD)
|
|
kill.append(cb)
|
|
|
|
deleted = set()
|
|
if kill:
|
|
try:
|
|
old = self.bare_git.GetHead()
|
|
except GitError:
|
|
old = 'refs/heads/please_never_use_this_as_a_branch_name'
|
|
|
|
rm_re = re.compile(r"^Deleted branch (.*)\.$")
|
|
try:
|
|
self.bare_git.DetachHead(rev)
|
|
|
|
b = ['branch', '-d']
|
|
b.extend(kill)
|
|
b = GitCommand(self, b, bare=True,
|
|
capture_stdout=True,
|
|
capture_stderr=True)
|
|
b.Wait()
|
|
finally:
|
|
self.bare_git.SetHead(old)
|
|
|
|
for line in b.stdout.split("\n"):
|
|
m = rm_re.match(line)
|
|
if m:
|
|
deleted.add(m.group(1))
|
|
|
|
if deleted:
|
|
self.CleanPublishedCache()
|
|
|
|
if cb and cb not in kill:
|
|
kill.append(cb)
|
|
kill.sort()
|
|
|
|
kept = []
|
|
for branch in kill:
|
|
if branch not in deleted:
|
|
branch = self.GetBranch(branch)
|
|
base = branch.LocalMerge
|
|
if not base:
|
|
base = rev
|
|
kept.append(ReviewableBranch(self, branch, base))
|
|
return kept
|
|
|
|
|
|
## Direct Git Commands ##
|
|
|
|
def _RemoteFetch(self, name=None):
|
|
if not name:
|
|
name = self.remote.name
|
|
|
|
hide_errors = False
|
|
if self.extraRemotes or self.snapshots:
|
|
hide_errors = True
|
|
|
|
proc = GitCommand(self,
|
|
['fetch', name],
|
|
bare = True,
|
|
capture_stderr = hide_errors)
|
|
if hide_errors:
|
|
err = proc.process.stderr.fileno()
|
|
buf = ''
|
|
while True:
|
|
b = os.read(err, 256)
|
|
if b:
|
|
buf += b
|
|
while buf:
|
|
r = buf.find('remote: error: unable to find ')
|
|
if r >= 0:
|
|
lf = buf.find('\n')
|
|
if lf < 0:
|
|
break
|
|
buf = buf[lf + 1:]
|
|
continue
|
|
|
|
cr = buf.find('\r')
|
|
if cr < 0:
|
|
break
|
|
os.write(2, buf[0:cr + 1])
|
|
buf = buf[cr + 1:]
|
|
if not b:
|
|
if buf:
|
|
os.write(2, buf)
|
|
break
|
|
return proc.Wait() == 0
|
|
|
|
def _Checkout(self, rev, quiet=False):
|
|
cmd = ['checkout']
|
|
if quiet:
|
|
cmd.append('-q')
|
|
cmd.append(rev)
|
|
cmd.append('--')
|
|
if GitCommand(self, cmd).Wait() != 0:
|
|
if self._allrefs:
|
|
raise GitError('%s checkout %s ' % (self.name, rev))
|
|
|
|
def _ResetHard(self, rev, quiet=True):
|
|
cmd = ['reset', '--hard']
|
|
if quiet:
|
|
cmd.append('-q')
|
|
cmd.append(rev)
|
|
if GitCommand(self, cmd).Wait() != 0:
|
|
raise GitError('%s reset --hard %s ' % (self.name, rev))
|
|
|
|
def _Rebase(self, upstream, onto = None):
|
|
cmd = ['rebase', '-i']
|
|
if onto is not None:
|
|
cmd.extend(['--onto', onto])
|
|
cmd.append(upstream)
|
|
if GitCommand(self, cmd, disable_editor=True).Wait() != 0:
|
|
raise GitError('%s rebase %s ' % (self.name, upstream))
|
|
|
|
def _FastForward(self, head):
|
|
cmd = ['merge', head]
|
|
if GitCommand(self, cmd).Wait() != 0:
|
|
raise GitError('%s merge %s ' % (self.name, head))
|
|
|
|
def _InitGitDir(self):
|
|
if not os.path.exists(self.gitdir):
|
|
os.makedirs(self.gitdir)
|
|
self.bare_git.init()
|
|
self.config.SetString('core.bare', None)
|
|
|
|
hooks = self._gitdir_path('hooks')
|
|
for old_hook in os.listdir(hooks):
|
|
os.remove(os.path.join(hooks, old_hook))
|
|
|
|
# TODO(sop) install custom repo hooks
|
|
|
|
m = self.manifest.manifestProject.config
|
|
for key in ['user.name', 'user.email']:
|
|
if m.Has(key, include_defaults = False):
|
|
self.config.SetString(key, m.GetString(key))
|
|
|
|
def _InitRemote(self):
|
|
if self.remote.fetchUrl:
|
|
remote = self.GetRemote(self.remote.name)
|
|
|
|
url = self.remote.fetchUrl
|
|
while url.endswith('/'):
|
|
url = url[:-1]
|
|
url += '/%s.git' % self.name
|
|
remote.url = url
|
|
remote.review = self.remote.reviewUrl
|
|
|
|
remote.ResetFetch()
|
|
remote.Save()
|
|
|
|
for r in self.extraRemotes.values():
|
|
remote = self.GetRemote(r.name)
|
|
remote.url = r.fetchUrl
|
|
remote.review = r.reviewUrl
|
|
remote.ResetFetch()
|
|
remote.Save()
|
|
|
|
def _InitMRef(self):
|
|
if self.manifest.branch:
|
|
msg = 'manifest set to %s' % self.revision
|
|
ref = R_M + self.manifest.branch
|
|
|
|
if IsId(self.revision):
|
|
dst = self.revision + '^0',
|
|
self.bare_git.UpdateRef(ref, dst, message = msg, detach = True)
|
|
else:
|
|
remote = self.GetRemote(self.remote.name)
|
|
dst = remote.ToLocal(self.revision)
|
|
self.bare_git.symbolic_ref('-m', msg, ref, dst)
|
|
|
|
def _InitWorkTree(self):
|
|
dotgit = os.path.join(self.worktree, '.git')
|
|
if not os.path.exists(dotgit):
|
|
os.makedirs(dotgit)
|
|
|
|
topdir = os.path.commonprefix([self.gitdir, dotgit])
|
|
if topdir.endswith('/'):
|
|
topdir = topdir[:-1]
|
|
else:
|
|
topdir = os.path.dirname(topdir)
|
|
|
|
tmpdir = dotgit
|
|
relgit = ''
|
|
while topdir != tmpdir:
|
|
relgit += '../'
|
|
tmpdir = os.path.dirname(tmpdir)
|
|
relgit += self.gitdir[len(topdir) + 1:]
|
|
|
|
for name in ['config',
|
|
'description',
|
|
'hooks',
|
|
'info',
|
|
'logs',
|
|
'objects',
|
|
'packed-refs',
|
|
'refs',
|
|
'rr-cache',
|
|
'svn']:
|
|
os.symlink(os.path.join(relgit, name),
|
|
os.path.join(dotgit, name))
|
|
|
|
rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
|
|
rev = self.bare_git.rev_parse('%s^0' % rev)
|
|
|
|
f = open(os.path.join(dotgit, HEAD), 'wb')
|
|
f.write("%s\n" % rev)
|
|
f.close()
|
|
|
|
cmd = ['read-tree', '--reset', '-u']
|
|
cmd.append('-v')
|
|
cmd.append('HEAD')
|
|
if GitCommand(self, cmd).Wait() != 0:
|
|
raise GitError("cannot initialize work tree")
|
|
|
|
def _gitdir_path(self, path):
|
|
return os.path.join(self.gitdir, path)
|
|
|
|
def _revlist(self, *args):
|
|
cmd = []
|
|
cmd.extend(args)
|
|
cmd.append('--')
|
|
return self.work_git.rev_list(*args)
|
|
|
|
@property
|
|
def _allrefs(self):
|
|
return self.bare_git.ListRefs()
|
|
|
|
class _GitGetByExec(object):
|
|
def __init__(self, project, bare):
|
|
self._project = project
|
|
self._bare = bare
|
|
|
|
def ListRefs(self, *args):
|
|
cmdv = ['for-each-ref', '--format=%(objectname) %(refname)']
|
|
cmdv.extend(args)
|
|
p = GitCommand(self._project,
|
|
cmdv,
|
|
bare = self._bare,
|
|
capture_stdout = True,
|
|
capture_stderr = True)
|
|
r = {}
|
|
for line in p.process.stdout:
|
|
id, name = line[:-1].split(' ', 2)
|
|
r[name] = id
|
|
if p.Wait() != 0:
|
|
raise GitError('%s for-each-ref %s: %s' % (
|
|
self._project.name,
|
|
str(args),
|
|
p.stderr))
|
|
return r
|
|
|
|
def LsOthers(self):
|
|
p = GitCommand(self._project,
|
|
['ls-files',
|
|
'-z',
|
|
'--others',
|
|
'--exclude-standard'],
|
|
bare = False,
|
|
capture_stdout = True,
|
|
capture_stderr = True)
|
|
if p.Wait() == 0:
|
|
out = p.stdout
|
|
if out:
|
|
return out[:-1].split("\0")
|
|
return []
|
|
|
|
def DiffZ(self, name, *args):
|
|
cmd = [name]
|
|
cmd.append('-z')
|
|
cmd.extend(args)
|
|
p = GitCommand(self._project,
|
|
cmd,
|
|
bare = False,
|
|
capture_stdout = True,
|
|
capture_stderr = True)
|
|
try:
|
|
out = p.process.stdout.read()
|
|
r = {}
|
|
if out:
|
|
out = iter(out[:-1].split('\0'))
|
|
while out:
|
|
try:
|
|
info = out.next()
|
|
path = out.next()
|
|
except StopIteration:
|
|
break
|
|
|
|
class _Info(object):
|
|
def __init__(self, path, omode, nmode, oid, nid, state):
|
|
self.path = path
|
|
self.src_path = None
|
|
self.old_mode = omode
|
|
self.new_mode = nmode
|
|
self.old_id = oid
|
|
self.new_id = nid
|
|
|
|
if len(state) == 1:
|
|
self.status = state
|
|
self.level = None
|
|
else:
|
|
self.status = state[:1]
|
|
self.level = state[1:]
|
|
while self.level.startswith('0'):
|
|
self.level = self.level[1:]
|
|
|
|
info = info[1:].split(' ')
|
|
info =_Info(path, *info)
|
|
if info.status in ('R', 'C'):
|
|
info.src_path = info.path
|
|
info.path = out.next()
|
|
r[info.path] = info
|
|
return r
|
|
finally:
|
|
p.Wait()
|
|
|
|
def GetHead(self):
|
|
return self.symbolic_ref(HEAD)
|
|
|
|
def SetHead(self, ref, message=None):
|
|
cmdv = []
|
|
if message is not None:
|
|
cmdv.extend(['-m', message])
|
|
cmdv.append(HEAD)
|
|
cmdv.append(ref)
|
|
self.symbolic_ref(*cmdv)
|
|
|
|
def DetachHead(self, new, message=None):
|
|
cmdv = ['--no-deref']
|
|
if message is not None:
|
|
cmdv.extend(['-m', message])
|
|
cmdv.append(HEAD)
|
|
cmdv.append(new)
|
|
self.update_ref(*cmdv)
|
|
|
|
def UpdateRef(self, name, new, old=None,
|
|
message=None,
|
|
detach=False):
|
|
cmdv = []
|
|
if message is not None:
|
|
cmdv.extend(['-m', message])
|
|
if detach:
|
|
cmdv.append('--no-deref')
|
|
cmdv.append(name)
|
|
cmdv.append(new)
|
|
if old is not None:
|
|
cmdv.append(old)
|
|
self.update_ref(*cmdv)
|
|
|
|
def DeleteRef(self, name, old=None):
|
|
if not old:
|
|
old = self.rev_parse(name)
|
|
self.update_ref('-d', name, old)
|
|
|
|
def rev_list(self, *args):
|
|
cmdv = ['rev-list']
|
|
cmdv.extend(args)
|
|
p = GitCommand(self._project,
|
|
cmdv,
|
|
bare = self._bare,
|
|
capture_stdout = True,
|
|
capture_stderr = True)
|
|
r = []
|
|
for line in p.process.stdout:
|
|
r.append(line[:-1])
|
|
if p.Wait() != 0:
|
|
raise GitError('%s rev-list %s: %s' % (
|
|
self._project.name,
|
|
str(args),
|
|
p.stderr))
|
|
return r
|
|
|
|
def __getattr__(self, name):
|
|
name = name.replace('_', '-')
|
|
def runner(*args):
|
|
cmdv = [name]
|
|
cmdv.extend(args)
|
|
p = GitCommand(self._project,
|
|
cmdv,
|
|
bare = self._bare,
|
|
capture_stdout = True,
|
|
capture_stderr = True)
|
|
if p.Wait() != 0:
|
|
raise GitError('%s %s: %s' % (
|
|
self._project.name,
|
|
name,
|
|
p.stderr))
|
|
r = p.stdout
|
|
if r.endswith('\n') and r.index('\n') == len(r) - 1:
|
|
return r[:-1]
|
|
return r
|
|
return runner
|
|
|
|
|
|
class MetaProject(Project):
|
|
"""A special project housed under .repo.
|
|
"""
|
|
def __init__(self, manifest, name, gitdir, worktree):
|
|
repodir = manifest.repodir
|
|
Project.__init__(self,
|
|
manifest = manifest,
|
|
name = name,
|
|
gitdir = gitdir,
|
|
worktree = worktree,
|
|
remote = Remote('origin'),
|
|
relpath = '.repo/%s' % name,
|
|
revision = 'refs/heads/master')
|
|
|
|
def PreSync(self):
|
|
if self.Exists:
|
|
cb = self.CurrentBranch
|
|
if cb:
|
|
base = self.GetBranch(cb).merge
|
|
if base:
|
|
self.revision = base
|
|
|
|
@property
|
|
def HasChanges(self):
|
|
"""Has the remote received new commits not yet checked out?
|
|
"""
|
|
rev = self.GetRemote(self.remote.name).ToLocal(self.revision)
|
|
if self._revlist(not_rev(HEAD), rev):
|
|
return True
|
|
return False
|