blob: ae771c42d51a4c0fa279d90509e18bfb6f8b23f9 [file] [log] [blame]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001# Copyright (C) 2008 The Android Open Source Project
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14
Sarah Owenscecd1d82012-11-01 22:59:27 -070015from __future__ import print_function
Shawn O. Pearce438ee1c2008-11-03 09:59:36 -080016import errno
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070017import filecmp
Wink Saville4c426ef2015-06-03 08:05:17 -070018import glob
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070019import os
Shawn O. Pearcec325dc32011-10-03 08:30:24 -070020import random
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070021import re
22import shutil
23import stat
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -070024import subprocess
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070025import sys
Julien Campergue335f5ef2013-10-16 11:02:35 +020026import tarfile
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +080027import tempfile
Shawn O. Pearcec325dc32011-10-03 08:30:24 -070028import time
Dave Borowitz137d0132015-01-02 11:12:54 -080029import traceback
Shawn O. Pearcedf5ee522011-10-11 14:05:21 -070030
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070031from color import Coloring
Dave Borowitzb42b4742012-10-31 12:27:27 -070032from git_command import GitCommand, git_require
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070033from git_config import GitConfig, IsId, GetSchemeFromUrl, GetUrlCookieFile, \
34 ID_RE
Kevin Degiabaa7f32014-11-12 11:27:45 -070035from error import GitError, HookError, UploadError, DownloadError
Shawn O. Pearce559b8462009-03-02 12:56:08 -080036from error import ManifestInvalidRevisionError
Conley Owens75ee0572012-11-15 17:33:11 -080037from error import NoManifestException
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -070038from trace import IsTrace, Trace
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070039
Shawn O. Pearced237b692009-04-17 18:49:50 -070040from git_refs import GitRefs, HEAD, R_HEADS, R_TAGS, R_PUB, R_M
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070041
David Pursehouse59bbb582013-05-17 10:49:33 +090042from pyversion import is_python3
Mike Frysinger40252c22016-08-15 21:23:44 -040043if is_python3():
44 import urllib.parse
45else:
46 import imp
47 import urlparse
48 urllib = imp.new_module('urllib')
49 urllib.parse = urlparse
David Pursehouse59bbb582013-05-17 10:49:33 +090050 # pylint:disable=W0622
Chirayu Desai217ea7d2013-03-01 19:14:38 +053051 input = raw_input
David Pursehouse59bbb582013-05-17 10:49:33 +090052 # pylint:enable=W0622
Chirayu Desai217ea7d2013-03-01 19:14:38 +053053
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070054
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -070055def _lwrite(path, content):
56 lock = '%s.lock' % path
57
Chirayu Desai303a82f2014-08-19 22:57:17 +053058 fd = open(lock, 'w')
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -070059 try:
60 fd.write(content)
61 finally:
62 fd.close()
63
64 try:
65 os.rename(lock, path)
66 except OSError:
67 os.remove(lock)
68 raise
69
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070070
Shawn O. Pearce48244782009-04-16 08:25:57 -070071def _error(fmt, *args):
72 msg = fmt % args
Sarah Owenscecd1d82012-11-01 22:59:27 -070073 print('error: %s' % msg, file=sys.stderr)
Shawn O. Pearce48244782009-04-16 08:25:57 -070074
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070075
David Pursehousef33929d2015-08-24 14:39:14 +090076def _warn(fmt, *args):
77 msg = fmt % args
78 print('warn: %s' % msg, file=sys.stderr)
79
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070080
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070081def not_rev(r):
82 return '^' + r
83
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070084
Shawn O. Pearceb54a3922009-01-05 16:18:58 -080085def sq(r):
86 return "'" + r.replace("'", "'\''") + "'"
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -080087
Jonathan Nieder93719792015-03-17 11:29:58 -070088_project_hook_list = None
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070089
90
Jonathan Nieder93719792015-03-17 11:29:58 -070091def _ProjectHooks():
92 """List the hooks present in the 'hooks' directory.
93
94 These hooks are project hooks and are copied to the '.git/hooks' directory
95 of all subprojects.
96
97 This function caches the list of hooks (based on the contents of the
98 'repo/hooks' directory) on the first call.
99
100 Returns:
101 A list of absolute paths to all of the files in the hooks directory.
102 """
103 global _project_hook_list
104 if _project_hook_list is None:
105 d = os.path.realpath(os.path.abspath(os.path.dirname(__file__)))
106 d = os.path.join(d, 'hooks')
107 _project_hook_list = [os.path.join(d, x) for x in os.listdir(d)]
108 return _project_hook_list
109
110
Shawn O. Pearce632768b2008-10-23 11:58:52 -0700111class DownloadedChange(object):
112 _commit_cache = None
113
114 def __init__(self, project, base, change_id, ps_id, commit):
115 self.project = project
116 self.base = base
117 self.change_id = change_id
118 self.ps_id = ps_id
119 self.commit = commit
120
121 @property
122 def commits(self):
123 if self._commit_cache is None:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700124 self._commit_cache = self.project.bare_git.rev_list('--abbrev=8',
125 '--abbrev-commit',
126 '--pretty=oneline',
127 '--reverse',
128 '--date-order',
129 not_rev(self.base),
130 self.commit,
131 '--')
Shawn O. Pearce632768b2008-10-23 11:58:52 -0700132 return self._commit_cache
133
134
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700135class ReviewableBranch(object):
136 _commit_cache = None
137
138 def __init__(self, project, branch, base):
139 self.project = project
140 self.branch = branch
141 self.base = base
142
143 @property
144 def name(self):
145 return self.branch.name
146
147 @property
148 def commits(self):
149 if self._commit_cache is None:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700150 self._commit_cache = self.project.bare_git.rev_list('--abbrev=8',
151 '--abbrev-commit',
152 '--pretty=oneline',
153 '--reverse',
154 '--date-order',
155 not_rev(self.base),
156 R_HEADS + self.name,
157 '--')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700158 return self._commit_cache
159
160 @property
Shawn O. Pearcec99883f2008-11-11 17:12:43 -0800161 def unabbrev_commits(self):
162 r = dict()
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700163 for commit in self.project.bare_git.rev_list(not_rev(self.base),
164 R_HEADS + self.name,
165 '--'):
Shawn O. Pearcec99883f2008-11-11 17:12:43 -0800166 r[commit[0:8]] = commit
167 return r
168
169 @property
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700170 def date(self):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700171 return self.project.bare_git.log('--pretty=format:%cd',
172 '-n', '1',
173 R_HEADS + self.name,
174 '--')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700175
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700176 def UploadForReview(self, people,
177 auto_topic=False,
178 draft=False,
179 dest_branch=None):
Shawn O. Pearcec99883f2008-11-11 17:12:43 -0800180 self.project.UploadForReview(self.name,
Shawn O. Pearcea5ece0e2010-07-15 16:52:42 -0700181 people,
Brian Harring435370c2012-07-28 15:37:04 -0700182 auto_topic=auto_topic,
Bryan Jacobsf609f912013-05-06 13:36:24 -0400183 draft=draft,
184 dest_branch=dest_branch)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700185
Ficus Kirkpatrickbc7ef672009-05-04 12:45:11 -0700186 def GetPublishedRefs(self):
187 refs = {}
188 output = self.project.bare_git.ls_remote(
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700189 self.branch.remote.SshReviewUrl(self.project.UserEmail),
190 'refs/changes/*')
Ficus Kirkpatrickbc7ef672009-05-04 12:45:11 -0700191 for line in output.split('\n'):
192 try:
193 (sha, ref) = line.split()
194 refs[sha] = ref
195 except ValueError:
196 pass
197
198 return refs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700199
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700200
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700201class StatusColoring(Coloring):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700202
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700203 def __init__(self, config):
204 Coloring.__init__(self, config, 'status')
Anthony King7bdac712014-07-16 12:56:40 +0100205 self.project = self.printer('header', attr='bold')
206 self.branch = self.printer('header', attr='bold')
207 self.nobranch = self.printer('nobranch', fg='red')
208 self.important = self.printer('important', fg='red')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700209
Anthony King7bdac712014-07-16 12:56:40 +0100210 self.added = self.printer('added', fg='green')
211 self.changed = self.printer('changed', fg='red')
212 self.untracked = self.printer('untracked', fg='red')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700213
214
215class DiffColoring(Coloring):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700216
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700217 def __init__(self, config):
218 Coloring.__init__(self, config, 'diff')
Anthony King7bdac712014-07-16 12:56:40 +0100219 self.project = self.printer('header', attr='bold')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700220
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700221
Anthony King7bdac712014-07-16 12:56:40 +0100222class _Annotation(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700223
James W. Mills24c13082012-04-12 15:04:13 -0500224 def __init__(self, name, value, keep):
225 self.name = name
226 self.value = value
227 self.keep = keep
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700228
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700229
Anthony King7bdac712014-07-16 12:56:40 +0100230class _CopyFile(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700231
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -0800232 def __init__(self, src, dest, abssrc, absdest):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700233 self.src = src
234 self.dest = dest
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -0800235 self.abs_src = abssrc
236 self.abs_dest = absdest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700237
238 def _Copy(self):
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -0800239 src = self.abs_src
240 dest = self.abs_dest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700241 # copy file if it does not exist or is out of date
242 if not os.path.exists(dest) or not filecmp.cmp(src, dest):
243 try:
244 # remove existing file first, since it might be read-only
245 if os.path.exists(dest):
246 os.remove(dest)
Matthew Buckett2daf6672009-07-11 09:43:47 -0400247 else:
Mickaël Salaün2f6ab7f2012-09-30 00:37:55 +0200248 dest_dir = os.path.dirname(dest)
249 if not os.path.isdir(dest_dir):
250 os.makedirs(dest_dir)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700251 shutil.copy(src, dest)
252 # make the file read-only
253 mode = os.stat(dest)[stat.ST_MODE]
254 mode = mode & ~(stat.S_IWUSR | stat.S_IWGRP | stat.S_IWOTH)
255 os.chmod(dest, mode)
256 except IOError:
Shawn O. Pearce48244782009-04-16 08:25:57 -0700257 _error('Cannot copy file %s to %s', src, dest)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700258
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700259
Anthony King7bdac712014-07-16 12:56:40 +0100260class _LinkFile(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700261
Wink Saville4c426ef2015-06-03 08:05:17 -0700262 def __init__(self, git_worktree, src, dest, relsrc, absdest):
263 self.git_worktree = git_worktree
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500264 self.src = src
265 self.dest = dest
Colin Cross0184dcc2015-05-05 00:24:54 -0700266 self.src_rel_to_dest = relsrc
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500267 self.abs_dest = absdest
268
Wink Saville4c426ef2015-06-03 08:05:17 -0700269 def __linkIt(self, relSrc, absDest):
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500270 # link file if it does not exist or is out of date
Wink Saville4c426ef2015-06-03 08:05:17 -0700271 if not os.path.islink(absDest) or (os.readlink(absDest) != relSrc):
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500272 try:
273 # remove existing file first, since it might be read-only
Dan Willemsene1e0bd12015-11-18 16:49:38 -0800274 if os.path.lexists(absDest):
Wink Saville4c426ef2015-06-03 08:05:17 -0700275 os.remove(absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500276 else:
Wink Saville4c426ef2015-06-03 08:05:17 -0700277 dest_dir = os.path.dirname(absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500278 if not os.path.isdir(dest_dir):
279 os.makedirs(dest_dir)
Wink Saville4c426ef2015-06-03 08:05:17 -0700280 os.symlink(relSrc, absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500281 except IOError:
Wink Saville4c426ef2015-06-03 08:05:17 -0700282 _error('Cannot link file %s to %s', relSrc, absDest)
283
284 def _Link(self):
285 """Link the self.rel_src_to_dest and self.abs_dest. Handles wild cards
286 on the src linking all of the files in the source in to the destination
287 directory.
288 """
289 # We use the absSrc to handle the situation where the current directory
290 # is not the root of the repo
291 absSrc = os.path.join(self.git_worktree, self.src)
292 if os.path.exists(absSrc):
293 # Entity exists so just a simple one to one link operation
294 self.__linkIt(self.src_rel_to_dest, self.abs_dest)
295 else:
296 # Entity doesn't exist assume there is a wild card
297 absDestDir = self.abs_dest
298 if os.path.exists(absDestDir) and not os.path.isdir(absDestDir):
299 _error('Link error: src with wildcard, %s must be a directory',
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700300 absDestDir)
Wink Saville4c426ef2015-06-03 08:05:17 -0700301 else:
302 absSrcFiles = glob.glob(absSrc)
303 for absSrcFile in absSrcFiles:
304 # Create a releative path from source dir to destination dir
305 absSrcDir = os.path.dirname(absSrcFile)
306 relSrcDir = os.path.relpath(absSrcDir, absDestDir)
307
308 # Get the source file name
309 srcFile = os.path.basename(absSrcFile)
310
311 # Now form the final full paths to srcFile. They will be
312 # absolute for the desintaiton and relative for the srouce.
313 absDest = os.path.join(absDestDir, srcFile)
314 relSrc = os.path.join(relSrcDir, srcFile)
315 self.__linkIt(relSrc, absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500316
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700317
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700318class RemoteSpec(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700319
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700320 def __init__(self,
321 name,
Anthony King7bdac712014-07-16 12:56:40 +0100322 url=None,
Steve Raed6480452016-08-10 15:00:00 -0700323 pushUrl=None,
Anthony King7bdac712014-07-16 12:56:40 +0100324 review=None,
Dan Willemsen96c2d652016-04-06 16:03:54 -0700325 revision=None,
326 orig_name=None):
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700327 self.name = name
328 self.url = url
Steve Raed6480452016-08-10 15:00:00 -0700329 self.pushUrl = pushUrl
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700330 self.review = review
Anthony King36ea2fb2014-05-06 11:54:01 +0100331 self.revision = revision
Dan Willemsen96c2d652016-04-06 16:03:54 -0700332 self.orig_name = orig_name
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700333
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700334
Doug Anderson37282b42011-03-04 11:54:18 -0800335class RepoHook(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700336
Doug Anderson37282b42011-03-04 11:54:18 -0800337 """A RepoHook contains information about a script to run as a hook.
338
339 Hooks are used to run a python script before running an upload (for instance,
340 to run presubmit checks). Eventually, we may have hooks for other actions.
341
342 This shouldn't be confused with files in the 'repo/hooks' directory. Those
343 files are copied into each '.git/hooks' folder for each project. Repo-level
344 hooks are associated instead with repo actions.
345
346 Hooks are always python. When a hook is run, we will load the hook into the
347 interpreter and execute its main() function.
348 """
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700349
Doug Anderson37282b42011-03-04 11:54:18 -0800350 def __init__(self,
351 hook_type,
352 hooks_project,
353 topdir,
Mike Frysinger40252c22016-08-15 21:23:44 -0400354 manifest_url,
Doug Anderson37282b42011-03-04 11:54:18 -0800355 abort_if_user_denies=False):
356 """RepoHook constructor.
357
358 Params:
359 hook_type: A string representing the type of hook. This is also used
360 to figure out the name of the file containing the hook. For
361 example: 'pre-upload'.
362 hooks_project: The project containing the repo hooks. If you have a
363 manifest, this is manifest.repo_hooks_project. OK if this is None,
364 which will make the hook a no-op.
365 topdir: Repo's top directory (the one containing the .repo directory).
366 Scripts will run with CWD as this directory. If you have a manifest,
367 this is manifest.topdir
Mike Frysinger40252c22016-08-15 21:23:44 -0400368 manifest_url: The URL to the manifest git repo.
Doug Anderson37282b42011-03-04 11:54:18 -0800369 abort_if_user_denies: If True, we'll throw a HookError() if the user
370 doesn't allow us to run the hook.
371 """
372 self._hook_type = hook_type
373 self._hooks_project = hooks_project
Mike Frysinger40252c22016-08-15 21:23:44 -0400374 self._manifest_url = manifest_url
Doug Anderson37282b42011-03-04 11:54:18 -0800375 self._topdir = topdir
376 self._abort_if_user_denies = abort_if_user_denies
377
378 # Store the full path to the script for convenience.
379 if self._hooks_project:
380 self._script_fullpath = os.path.join(self._hooks_project.worktree,
381 self._hook_type + '.py')
382 else:
383 self._script_fullpath = None
384
385 def _GetHash(self):
386 """Return a hash of the contents of the hooks directory.
387
388 We'll just use git to do this. This hash has the property that if anything
389 changes in the directory we will return a different has.
390
391 SECURITY CONSIDERATION:
392 This hash only represents the contents of files in the hook directory, not
393 any other files imported or called by hooks. Changes to imported files
394 can change the script behavior without affecting the hash.
395
396 Returns:
397 A string representing the hash. This will always be ASCII so that it can
398 be printed to the user easily.
399 """
400 assert self._hooks_project, "Must have hooks to calculate their hash."
401
402 # We will use the work_git object rather than just calling GetRevisionId().
403 # That gives us a hash of the latest checked in version of the files that
404 # the user will actually be executing. Specifically, GetRevisionId()
405 # doesn't appear to change even if a user checks out a different version
406 # of the hooks repo (via git checkout) nor if a user commits their own revs.
407 #
408 # NOTE: Local (non-committed) changes will not be factored into this hash.
409 # I think this is OK, since we're really only worried about warning the user
410 # about upstream changes.
411 return self._hooks_project.work_git.rev_parse('HEAD')
412
413 def _GetMustVerb(self):
414 """Return 'must' if the hook is required; 'should' if not."""
415 if self._abort_if_user_denies:
416 return 'must'
417 else:
418 return 'should'
419
420 def _CheckForHookApproval(self):
421 """Check to see whether this hook has been approved.
422
Mike Frysinger40252c22016-08-15 21:23:44 -0400423 We'll accept approval of manifest URLs if they're using secure transports.
424 This way the user can say they trust the manifest hoster. For insecure
425 hosts, we fall back to checking the hash of the hooks repo.
Doug Anderson37282b42011-03-04 11:54:18 -0800426
427 Note that we ask permission for each individual hook even though we use
428 the hash of all hooks when detecting changes. We'd like the user to be
429 able to approve / deny each hook individually. We only use the hash of all
430 hooks because there is no other easy way to detect changes to local imports.
431
432 Returns:
433 True if this hook is approved to run; False otherwise.
434
435 Raises:
436 HookError: Raised if the user doesn't approve and abort_if_user_denies
437 was passed to the consturctor.
438 """
Mike Frysinger40252c22016-08-15 21:23:44 -0400439 if self._ManifestUrlHasSecureScheme():
440 return self._CheckForHookApprovalManifest()
441 else:
442 return self._CheckForHookApprovalHash()
443
444 def _CheckForHookApprovalHelper(self, subkey, new_val, main_prompt,
445 changed_prompt):
446 """Check for approval for a particular attribute and hook.
447
448 Args:
449 subkey: The git config key under [repo.hooks.<hook_type>] to store the
450 last approved string.
451 new_val: The new value to compare against the last approved one.
452 main_prompt: Message to display to the user to ask for approval.
453 changed_prompt: Message explaining why we're re-asking for approval.
454
455 Returns:
456 True if this hook is approved to run; False otherwise.
457
458 Raises:
459 HookError: Raised if the user doesn't approve and abort_if_user_denies
460 was passed to the consturctor.
461 """
Doug Anderson37282b42011-03-04 11:54:18 -0800462 hooks_config = self._hooks_project.config
Mike Frysinger40252c22016-08-15 21:23:44 -0400463 git_approval_key = 'repo.hooks.%s.%s' % (self._hook_type, subkey)
Doug Anderson37282b42011-03-04 11:54:18 -0800464
Mike Frysinger40252c22016-08-15 21:23:44 -0400465 # Get the last value that the user approved for this hook; may be None.
466 old_val = hooks_config.GetString(git_approval_key)
Doug Anderson37282b42011-03-04 11:54:18 -0800467
Mike Frysinger40252c22016-08-15 21:23:44 -0400468 if old_val is not None:
Doug Anderson37282b42011-03-04 11:54:18 -0800469 # User previously approved hook and asked not to be prompted again.
Mike Frysinger40252c22016-08-15 21:23:44 -0400470 if new_val == old_val:
Doug Anderson37282b42011-03-04 11:54:18 -0800471 # Approval matched. We're done.
472 return True
473 else:
474 # Give the user a reason why we're prompting, since they last told
475 # us to "never ask again".
Mike Frysinger40252c22016-08-15 21:23:44 -0400476 prompt = 'WARNING: %s\n\n' % (changed_prompt,)
Doug Anderson37282b42011-03-04 11:54:18 -0800477 else:
478 prompt = ''
479
480 # Prompt the user if we're not on a tty; on a tty we'll assume "no".
481 if sys.stdout.isatty():
Mike Frysinger40252c22016-08-15 21:23:44 -0400482 prompt += main_prompt + ' (yes/always/NO)? '
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530483 response = input(prompt).lower()
David Pursehouse98ffba12012-11-14 11:18:00 +0900484 print()
Doug Anderson37282b42011-03-04 11:54:18 -0800485
486 # User is doing a one-time approval.
487 if response in ('y', 'yes'):
488 return True
Mike Frysinger40252c22016-08-15 21:23:44 -0400489 elif response == 'always':
490 hooks_config.SetString(git_approval_key, new_val)
Doug Anderson37282b42011-03-04 11:54:18 -0800491 return True
492
493 # For anything else, we'll assume no approval.
494 if self._abort_if_user_denies:
495 raise HookError('You must allow the %s hook or use --no-verify.' %
496 self._hook_type)
497
498 return False
499
Mike Frysinger40252c22016-08-15 21:23:44 -0400500 def _ManifestUrlHasSecureScheme(self):
501 """Check if the URI for the manifest is a secure transport."""
502 secure_schemes = ('file', 'https', 'ssh', 'persistent-https', 'sso', 'rpc')
503 parse_results = urllib.parse.urlparse(self._manifest_url)
504 return parse_results.scheme in secure_schemes
505
506 def _CheckForHookApprovalManifest(self):
507 """Check whether the user has approved this manifest host.
508
509 Returns:
510 True if this hook is approved to run; False otherwise.
511 """
512 return self._CheckForHookApprovalHelper(
513 'approvedmanifest',
514 self._manifest_url,
515 'Run hook scripts from %s' % (self._manifest_url,),
516 'Manifest URL has changed since %s was allowed.' % (self._hook_type,))
517
518 def _CheckForHookApprovalHash(self):
519 """Check whether the user has approved the hooks repo.
520
521 Returns:
522 True if this hook is approved to run; False otherwise.
523 """
524 prompt = ('Repo %s run the script:\n'
525 ' %s\n'
526 '\n'
Jonathan Nieder71e4cea2016-08-16 12:05:09 -0700527 'Do you want to allow this script to run')
Mike Frysinger40252c22016-08-15 21:23:44 -0400528 return self._CheckForHookApprovalHelper(
529 'approvedhash',
530 self._GetHash(),
Jonathan Nieder71e4cea2016-08-16 12:05:09 -0700531 prompt % (self._GetMustVerb(), self._script_fullpath),
Mike Frysinger40252c22016-08-15 21:23:44 -0400532 'Scripts have changed since %s was allowed.' % (self._hook_type,))
533
Doug Anderson37282b42011-03-04 11:54:18 -0800534 def _ExecuteHook(self, **kwargs):
535 """Actually execute the given hook.
536
537 This will run the hook's 'main' function in our python interpreter.
538
539 Args:
540 kwargs: Keyword arguments to pass to the hook. These are often specific
541 to the hook type. For instance, pre-upload hooks will contain
542 a project_list.
543 """
544 # Keep sys.path and CWD stashed away so that we can always restore them
545 # upon function exit.
546 orig_path = os.getcwd()
547 orig_syspath = sys.path
548
549 try:
550 # Always run hooks with CWD as topdir.
551 os.chdir(self._topdir)
552
553 # Put the hook dir as the first item of sys.path so hooks can do
554 # relative imports. We want to replace the repo dir as [0] so
555 # hooks can't import repo files.
556 sys.path = [os.path.dirname(self._script_fullpath)] + sys.path[1:]
557
558 # Exec, storing global context in the context dict. We catch exceptions
559 # and convert to a HookError w/ just the failing traceback.
Mike Frysinger4aa4b212016-03-04 15:03:00 -0500560 context = {'__file__': self._script_fullpath}
Doug Anderson37282b42011-03-04 11:54:18 -0800561 try:
Anthony King70f68902014-05-05 21:15:34 +0100562 exec(compile(open(self._script_fullpath).read(),
563 self._script_fullpath, 'exec'), context)
Doug Anderson37282b42011-03-04 11:54:18 -0800564 except Exception:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700565 raise HookError('%s\nFailed to import %s hook; see traceback above.' %
566 (traceback.format_exc(), self._hook_type))
Doug Anderson37282b42011-03-04 11:54:18 -0800567
568 # Running the script should have defined a main() function.
569 if 'main' not in context:
570 raise HookError('Missing main() in: "%s"' % self._script_fullpath)
571
Doug Anderson37282b42011-03-04 11:54:18 -0800572 # Add 'hook_should_take_kwargs' to the arguments to be passed to main.
573 # We don't actually want hooks to define their main with this argument--
574 # it's there to remind them that their hook should always take **kwargs.
575 # For instance, a pre-upload hook should be defined like:
576 # def main(project_list, **kwargs):
577 #
578 # This allows us to later expand the API without breaking old hooks.
579 kwargs = kwargs.copy()
580 kwargs['hook_should_take_kwargs'] = True
581
582 # Call the main function in the hook. If the hook should cause the
583 # build to fail, it will raise an Exception. We'll catch that convert
584 # to a HookError w/ just the failing traceback.
585 try:
586 context['main'](**kwargs)
587 except Exception:
588 raise HookError('%s\nFailed to run main() for %s hook; see traceback '
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700589 'above.' % (traceback.format_exc(),
590 self._hook_type))
Doug Anderson37282b42011-03-04 11:54:18 -0800591 finally:
592 # Restore sys.path and CWD.
593 sys.path = orig_syspath
594 os.chdir(orig_path)
595
596 def Run(self, user_allows_all_hooks, **kwargs):
597 """Run the hook.
598
599 If the hook doesn't exist (because there is no hooks project or because
600 this particular hook is not enabled), this is a no-op.
601
602 Args:
603 user_allows_all_hooks: If True, we will never prompt about running the
604 hook--we'll just assume it's OK to run it.
605 kwargs: Keyword arguments to pass to the hook. These are often specific
606 to the hook type. For instance, pre-upload hooks will contain
607 a project_list.
608
609 Raises:
610 HookError: If there was a problem finding the hook or the user declined
611 to run a required hook (from _CheckForHookApproval).
612 """
613 # No-op if there is no hooks project or if hook is disabled.
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700614 if ((not self._hooks_project) or (self._hook_type not in
615 self._hooks_project.enabled_repo_hooks)):
Doug Anderson37282b42011-03-04 11:54:18 -0800616 return
617
618 # Bail with a nice error if we can't find the hook.
619 if not os.path.isfile(self._script_fullpath):
620 raise HookError('Couldn\'t find repo hook: "%s"' % self._script_fullpath)
621
622 # Make sure the user is OK with running the hook.
623 if (not user_allows_all_hooks) and (not self._CheckForHookApproval()):
624 return
625
626 # Run the hook with the same version of python we're using.
627 self._ExecuteHook(**kwargs)
628
629
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700630class Project(object):
Kevin Degi384b3c52014-10-16 16:02:58 -0600631 # These objects can be shared between several working trees.
632 shareable_files = ['description', 'info']
633 shareable_dirs = ['hooks', 'objects', 'rr-cache', 'svn']
634 # These objects can only be used by a single working tree.
635 working_tree_files = ['config', 'packed-refs', 'shallow']
636 working_tree_dirs = ['logs', 'refs']
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700637
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700638 def __init__(self,
639 manifest,
640 name,
641 remote,
642 gitdir,
David James8d201162013-10-11 17:03:19 -0700643 objdir,
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700644 worktree,
645 relpath,
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -0700646 revisionExpr,
Mike Pontillod3153822012-02-28 11:53:24 -0800647 revisionId,
Anthony King7bdac712014-07-16 12:56:40 +0100648 rebase=True,
649 groups=None,
650 sync_c=False,
651 sync_s=False,
652 clone_depth=None,
653 upstream=None,
654 parent=None,
655 is_derived=False,
David Pursehouseb1553542014-09-04 21:28:09 +0900656 dest_branch=None,
Simran Basib9a1b732015-08-20 12:19:28 -0700657 optimized_fetch=False,
658 old_revision=None):
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800659 """Init a Project object.
660
661 Args:
662 manifest: The XmlManifest object.
663 name: The `name` attribute of manifest.xml's project element.
664 remote: RemoteSpec object specifying its remote's properties.
665 gitdir: Absolute path of git directory.
David James8d201162013-10-11 17:03:19 -0700666 objdir: Absolute path of directory to store git objects.
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800667 worktree: Absolute path of git working tree.
668 relpath: Relative path of git working tree to repo's top directory.
669 revisionExpr: The `revision` attribute of manifest.xml's project element.
670 revisionId: git commit id for checking out.
671 rebase: The `rebase` attribute of manifest.xml's project element.
672 groups: The `groups` attribute of manifest.xml's project element.
673 sync_c: The `sync-c` attribute of manifest.xml's project element.
674 sync_s: The `sync-s` attribute of manifest.xml's project element.
675 upstream: The `upstream` attribute of manifest.xml's project element.
676 parent: The parent Project object.
677 is_derived: False if the project was explicitly defined in the manifest;
678 True if the project is a discovered submodule.
Bryan Jacobsf609f912013-05-06 13:36:24 -0400679 dest_branch: The branch to which to push changes for review by default.
David Pursehouseb1553542014-09-04 21:28:09 +0900680 optimized_fetch: If True, when a project is set to a sha1 revision, only
681 fetch from the remote if the sha1 is not present locally.
Simran Basib9a1b732015-08-20 12:19:28 -0700682 old_revision: saved git commit id for open GITC projects.
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800683 """
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700684 self.manifest = manifest
685 self.name = name
686 self.remote = remote
Anthony Newnamdf14a702011-01-09 17:31:57 -0800687 self.gitdir = gitdir.replace('\\', '/')
David James8d201162013-10-11 17:03:19 -0700688 self.objdir = objdir.replace('\\', '/')
Shawn O. Pearce0ce6ca92011-01-10 13:26:01 -0800689 if worktree:
Mark E. Hamiltonf9fe3e12016-02-23 18:10:42 -0700690 self.worktree = os.path.normpath(worktree.replace('\\', '/'))
Shawn O. Pearce0ce6ca92011-01-10 13:26:01 -0800691 else:
692 self.worktree = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700693 self.relpath = relpath
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -0700694 self.revisionExpr = revisionExpr
695
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700696 if revisionId is None \
697 and revisionExpr \
698 and IsId(revisionExpr):
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -0700699 self.revisionId = revisionExpr
700 else:
701 self.revisionId = revisionId
702
Mike Pontillod3153822012-02-28 11:53:24 -0800703 self.rebase = rebase
Colin Cross5acde752012-03-28 20:15:45 -0700704 self.groups = groups
Anatol Pomazau79770d22012-04-20 14:41:59 -0700705 self.sync_c = sync_c
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800706 self.sync_s = sync_s
David Pursehouseede7f122012-11-27 22:25:30 +0900707 self.clone_depth = clone_depth
Brian Harring14a66742012-09-28 20:21:57 -0700708 self.upstream = upstream
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800709 self.parent = parent
710 self.is_derived = is_derived
David Pursehouseb1553542014-09-04 21:28:09 +0900711 self.optimized_fetch = optimized_fetch
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800712 self.subprojects = []
Mike Pontillod3153822012-02-28 11:53:24 -0800713
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700714 self.snapshots = {}
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700715 self.copyfiles = []
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500716 self.linkfiles = []
James W. Mills24c13082012-04-12 15:04:13 -0500717 self.annotations = []
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700718 self.config = GitConfig.ForRepository(gitdir=self.gitdir,
719 defaults=self.manifest.globalConfig)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700720
Shawn O. Pearcee284ad12008-11-04 07:37:10 -0800721 if self.worktree:
David James8d201162013-10-11 17:03:19 -0700722 self.work_git = self._GitGetByExec(self, bare=False, gitdir=gitdir)
Shawn O. Pearcee284ad12008-11-04 07:37:10 -0800723 else:
724 self.work_git = None
David James8d201162013-10-11 17:03:19 -0700725 self.bare_git = self._GitGetByExec(self, bare=True, gitdir=gitdir)
Shawn O. Pearced237b692009-04-17 18:49:50 -0700726 self.bare_ref = GitRefs(gitdir)
David James8d201162013-10-11 17:03:19 -0700727 self.bare_objdir = self._GitGetByExec(self, bare=True, gitdir=objdir)
Bryan Jacobsf609f912013-05-06 13:36:24 -0400728 self.dest_branch = dest_branch
Simran Basib9a1b732015-08-20 12:19:28 -0700729 self.old_revision = old_revision
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700730
Doug Anderson37282b42011-03-04 11:54:18 -0800731 # This will be filled in if a project is later identified to be the
732 # project containing repo hooks.
733 self.enabled_repo_hooks = []
734
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700735 @property
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800736 def Derived(self):
737 return self.is_derived
738
739 @property
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700740 def Exists(self):
Kevin Degi384b3c52014-10-16 16:02:58 -0600741 return os.path.isdir(self.gitdir) and os.path.isdir(self.objdir)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700742
743 @property
744 def CurrentBranch(self):
745 """Obtain the name of the currently checked out branch.
746 The branch name omits the 'refs/heads/' prefix.
747 None is returned if the project is on a detached HEAD.
748 """
Shawn O. Pearce5b23f242009-04-17 18:43:33 -0700749 b = self.work_git.GetHead()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700750 if b.startswith(R_HEADS):
751 return b[len(R_HEADS):]
752 return None
753
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -0700754 def IsRebaseInProgress(self):
755 w = self.worktree
756 g = os.path.join(w, '.git')
757 return os.path.exists(os.path.join(g, 'rebase-apply')) \
758 or os.path.exists(os.path.join(g, 'rebase-merge')) \
759 or os.path.exists(os.path.join(w, '.dotest'))
Julius Gustavsson0cb1b3f2010-06-17 17:55:02 +0200760
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700761 def IsDirty(self, consider_untracked=True):
762 """Is the working directory modified in some way?
763 """
764 self.work_git.update_index('-q',
765 '--unmerged',
766 '--ignore-missing',
767 '--refresh')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900768 if self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700769 return True
770 if self.work_git.DiffZ('diff-files'):
771 return True
772 if consider_untracked and self.work_git.LsOthers():
773 return True
774 return False
775
776 _userident_name = None
777 _userident_email = None
778
779 @property
780 def UserName(self):
781 """Obtain the user's personal name.
782 """
783 if self._userident_name is None:
784 self._LoadUserIdentity()
785 return self._userident_name
786
787 @property
788 def UserEmail(self):
789 """Obtain the user's email address. This is very likely
790 to be their Gerrit login.
791 """
792 if self._userident_email is None:
793 self._LoadUserIdentity()
794 return self._userident_email
795
796 def _LoadUserIdentity(self):
David Pursehousec1b86a22012-11-14 11:36:51 +0900797 u = self.bare_git.var('GIT_COMMITTER_IDENT')
798 m = re.compile("^(.*) <([^>]*)> ").match(u)
799 if m:
800 self._userident_name = m.group(1)
801 self._userident_email = m.group(2)
802 else:
803 self._userident_name = ''
804 self._userident_email = ''
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700805
806 def GetRemote(self, name):
807 """Get the configuration for a single remote.
808 """
809 return self.config.GetRemote(name)
810
811 def GetBranch(self, name):
812 """Get the configuration for a single branch.
813 """
814 return self.config.GetBranch(name)
815
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700816 def GetBranches(self):
817 """Get all existing local branches.
818 """
819 current = self.CurrentBranch
David Pursehouse8a68ff92012-09-24 12:15:13 +0900820 all_refs = self._allrefs
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700821 heads = {}
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700822
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530823 for name, ref_id in all_refs.items():
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700824 if name.startswith(R_HEADS):
825 name = name[len(R_HEADS):]
826 b = self.GetBranch(name)
827 b.current = name == current
828 b.published = None
David Pursehouse8a68ff92012-09-24 12:15:13 +0900829 b.revision = ref_id
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700830 heads[name] = b
831
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530832 for name, ref_id in all_refs.items():
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700833 if name.startswith(R_PUB):
834 name = name[len(R_PUB):]
835 b = heads.get(name)
836 if b:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900837 b.published = ref_id
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700838
839 return heads
840
Colin Cross5acde752012-03-28 20:15:45 -0700841 def MatchesGroups(self, manifest_groups):
842 """Returns true if the manifest groups specified at init should cause
843 this project to be synced.
844 Prefixing a manifest group with "-" inverts the meaning of a group.
Conley Owensbb1b5f52012-08-13 13:11:18 -0700845 All projects are implicitly labelled with "all".
Conley Owens971de8e2012-04-16 10:36:08 -0700846
847 labels are resolved in order. In the example case of
Conley Owensbb1b5f52012-08-13 13:11:18 -0700848 project_groups: "all,group1,group2"
Conley Owens971de8e2012-04-16 10:36:08 -0700849 manifest_groups: "-group1,group2"
850 the project will be matched.
David Holmer0a1c6a12012-11-14 19:19:00 -0500851
852 The special manifest group "default" will match any project that
853 does not have the special project group "notdefault"
Colin Cross5acde752012-03-28 20:15:45 -0700854 """
David Holmer0a1c6a12012-11-14 19:19:00 -0500855 expanded_manifest_groups = manifest_groups or ['default']
Conley Owensbb1b5f52012-08-13 13:11:18 -0700856 expanded_project_groups = ['all'] + (self.groups or [])
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700857 if 'notdefault' not in expanded_project_groups:
David Holmer0a1c6a12012-11-14 19:19:00 -0500858 expanded_project_groups += ['default']
Conley Owensbb1b5f52012-08-13 13:11:18 -0700859
Conley Owens971de8e2012-04-16 10:36:08 -0700860 matched = False
Conley Owensbb1b5f52012-08-13 13:11:18 -0700861 for group in expanded_manifest_groups:
862 if group.startswith('-') and group[1:] in expanded_project_groups:
Conley Owens971de8e2012-04-16 10:36:08 -0700863 matched = False
Conley Owensbb1b5f52012-08-13 13:11:18 -0700864 elif group in expanded_project_groups:
Conley Owens971de8e2012-04-16 10:36:08 -0700865 matched = True
Colin Cross5acde752012-03-28 20:15:45 -0700866
Conley Owens971de8e2012-04-16 10:36:08 -0700867 return matched
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700868
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700869# Status Display ##
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700870 def UncommitedFiles(self, get_all=True):
871 """Returns a list of strings, uncommitted files in the git tree.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700872
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700873 Args:
874 get_all: a boolean, if True - get information about all different
875 uncommitted files. If False - return as soon as any kind of
876 uncommitted files is detected.
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500877 """
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700878 details = []
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500879 self.work_git.update_index('-q',
880 '--unmerged',
881 '--ignore-missing',
882 '--refresh')
883 if self.IsRebaseInProgress():
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700884 details.append("rebase in progress")
885 if not get_all:
886 return details
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500887
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700888 changes = self.work_git.DiffZ('diff-index', '--cached', HEAD).keys()
889 if changes:
890 details.extend(changes)
891 if not get_all:
892 return details
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500893
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700894 changes = self.work_git.DiffZ('diff-files').keys()
895 if changes:
896 details.extend(changes)
897 if not get_all:
898 return details
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500899
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700900 changes = self.work_git.LsOthers()
901 if changes:
902 details.extend(changes)
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500903
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700904 return details
905
906 def HasChanges(self):
907 """Returns true if there are uncommitted changes.
908 """
909 if self.UncommitedFiles(get_all=False):
910 return True
911 else:
912 return False
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500913
Andrew Wheeler4d5bb682012-02-27 13:52:22 -0600914 def PrintWorkTreeStatus(self, output_redir=None, quiet=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700915 """Prints the status of the repository to stdout.
Terence Haddock4655e812011-03-31 12:33:34 +0200916
917 Args:
918 output: If specified, redirect the output to this object.
Andrew Wheeler4d5bb682012-02-27 13:52:22 -0600919 quiet: If True then only print the project name. Do not print
920 the modified files, branch name, etc.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700921 """
922 if not os.path.isdir(self.worktree):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700923 if output_redir is None:
Terence Haddock4655e812011-03-31 12:33:34 +0200924 output_redir = sys.stdout
Sarah Owenscecd1d82012-11-01 22:59:27 -0700925 print(file=output_redir)
926 print('project %s/' % self.relpath, file=output_redir)
927 print(' missing (run "repo sync")', file=output_redir)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700928 return
929
930 self.work_git.update_index('-q',
931 '--unmerged',
932 '--ignore-missing',
933 '--refresh')
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -0700934 rb = self.IsRebaseInProgress()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700935 di = self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD)
936 df = self.work_git.DiffZ('diff-files')
937 do = self.work_git.LsOthers()
Ali Utku Selen76abcc12012-01-25 10:51:12 +0100938 if not rb and not di and not df and not do and not self.CurrentBranch:
Shawn O. Pearce161f4452009-04-10 17:41:44 -0700939 return 'CLEAN'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700940
941 out = StatusColoring(self.config)
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700942 if output_redir is not None:
Terence Haddock4655e812011-03-31 12:33:34 +0200943 out.redirect(output_redir)
Jakub Vrana0402cd82014-09-09 15:39:15 -0700944 out.project('project %-40s', self.relpath + '/ ')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700945
Andrew Wheeler4d5bb682012-02-27 13:52:22 -0600946 if quiet:
947 out.nl()
948 return 'DIRTY'
949
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700950 branch = self.CurrentBranch
951 if branch is None:
952 out.nobranch('(*** NO BRANCH ***)')
953 else:
954 out.branch('branch %s', branch)
955 out.nl()
956
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -0700957 if rb:
958 out.important('prior sync failed; rebase still in progress')
959 out.nl()
960
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700961 paths = list()
962 paths.extend(di.keys())
963 paths.extend(df.keys())
964 paths.extend(do)
965
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530966 for p in sorted(set(paths)):
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900967 try:
968 i = di[p]
969 except KeyError:
970 i = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700971
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900972 try:
973 f = df[p]
974 except KeyError:
975 f = None
Julius Gustavsson0cb1b3f2010-06-17 17:55:02 +0200976
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900977 if i:
978 i_status = i.status.upper()
979 else:
980 i_status = '-'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700981
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900982 if f:
983 f_status = f.status.lower()
984 else:
985 f_status = '-'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700986
987 if i and i.src_path:
Shawn O. Pearcefe086752009-03-03 13:49:48 -0800988 line = ' %s%s\t%s => %s (%s%%)' % (i_status, f_status,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700989 i.src_path, p, i.level)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700990 else:
991 line = ' %s%s\t%s' % (i_status, f_status, p)
992
993 if i and not f:
994 out.added('%s', line)
995 elif (i and f) or (not i and f):
996 out.changed('%s', line)
997 elif not i and not f:
998 out.untracked('%s', line)
999 else:
1000 out.write('%s', line)
1001 out.nl()
Terence Haddock4655e812011-03-31 12:33:34 +02001002
Shawn O. Pearce161f4452009-04-10 17:41:44 -07001003 return 'DIRTY'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001004
pelyad67872d2012-03-28 14:49:58 +03001005 def PrintWorkTreeDiff(self, absolute_paths=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001006 """Prints the status of the repository to stdout.
1007 """
1008 out = DiffColoring(self.config)
1009 cmd = ['diff']
1010 if out.is_on:
1011 cmd.append('--color')
1012 cmd.append(HEAD)
pelyad67872d2012-03-28 14:49:58 +03001013 if absolute_paths:
1014 cmd.append('--src-prefix=a/%s/' % self.relpath)
1015 cmd.append('--dst-prefix=b/%s/' % self.relpath)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001016 cmd.append('--')
1017 p = GitCommand(self,
1018 cmd,
Anthony King7bdac712014-07-16 12:56:40 +01001019 capture_stdout=True,
1020 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001021 has_diff = False
1022 for line in p.process.stdout:
1023 if not has_diff:
1024 out.nl()
1025 out.project('project %s/' % self.relpath)
1026 out.nl()
1027 has_diff = True
Sarah Owenscecd1d82012-11-01 22:59:27 -07001028 print(line[:-1])
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001029 p.Wait()
1030
1031
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001032# Publish / Upload ##
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001033
David Pursehouse8a68ff92012-09-24 12:15:13 +09001034 def WasPublished(self, branch, all_refs=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001035 """Was the branch published (uploaded) for code review?
1036 If so, returns the SHA-1 hash of the last published
1037 state for the branch.
1038 """
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001039 key = R_PUB + branch
David Pursehouse8a68ff92012-09-24 12:15:13 +09001040 if all_refs is None:
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001041 try:
1042 return self.bare_git.rev_parse(key)
1043 except GitError:
1044 return None
1045 else:
1046 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001047 return all_refs[key]
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001048 except KeyError:
1049 return None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001050
David Pursehouse8a68ff92012-09-24 12:15:13 +09001051 def CleanPublishedCache(self, all_refs=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001052 """Prunes any stale published refs.
1053 """
David Pursehouse8a68ff92012-09-24 12:15:13 +09001054 if all_refs is None:
1055 all_refs = self._allrefs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001056 heads = set()
1057 canrm = {}
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301058 for name, ref_id in all_refs.items():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001059 if name.startswith(R_HEADS):
1060 heads.add(name)
1061 elif name.startswith(R_PUB):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001062 canrm[name] = ref_id
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001063
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301064 for name, ref_id in canrm.items():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001065 n = name[len(R_PUB):]
1066 if R_HEADS + n not in heads:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001067 self.bare_git.DeleteRef(name, ref_id)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001068
Mandeep Singh Bainesd6c93a22011-05-26 10:34:11 -07001069 def GetUploadableBranches(self, selected_branch=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001070 """List any branches which can be uploaded for review.
1071 """
1072 heads = {}
1073 pubed = {}
1074
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301075 for name, ref_id in self._allrefs.items():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001076 if name.startswith(R_HEADS):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001077 heads[name[len(R_HEADS):]] = ref_id
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001078 elif name.startswith(R_PUB):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001079 pubed[name[len(R_PUB):]] = ref_id
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001080
1081 ready = []
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301082 for branch, ref_id in heads.items():
David Pursehouse8a68ff92012-09-24 12:15:13 +09001083 if branch in pubed and pubed[branch] == ref_id:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001084 continue
Mandeep Singh Bainesd6c93a22011-05-26 10:34:11 -07001085 if selected_branch and branch != selected_branch:
1086 continue
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001087
Shawn O. Pearce35f25962008-11-11 17:03:13 -08001088 rb = self.GetUploadableBranch(branch)
1089 if rb:
1090 ready.append(rb)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001091 return ready
1092
Shawn O. Pearce35f25962008-11-11 17:03:13 -08001093 def GetUploadableBranch(self, branch_name):
1094 """Get a single uploadable branch, or None.
1095 """
1096 branch = self.GetBranch(branch_name)
1097 base = branch.LocalMerge
1098 if branch.LocalMerge:
1099 rb = ReviewableBranch(self, branch, base)
1100 if rb.commits:
1101 return rb
1102 return None
1103
Shawn O. Pearcea5ece0e2010-07-15 16:52:42 -07001104 def UploadForReview(self, branch=None,
Anthony King7bdac712014-07-16 12:56:40 +01001105 people=([], []),
Brian Harring435370c2012-07-28 15:37:04 -07001106 auto_topic=False,
Bryan Jacobsf609f912013-05-06 13:36:24 -04001107 draft=False,
1108 dest_branch=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001109 """Uploads the named branch for code review.
1110 """
1111 if branch is None:
1112 branch = self.CurrentBranch
1113 if branch is None:
1114 raise GitError('not currently on a branch')
1115
1116 branch = self.GetBranch(branch)
1117 if not branch.LocalMerge:
1118 raise GitError('branch %s does not track a remote' % branch.name)
1119 if not branch.remote.review:
1120 raise GitError('remote %s has no review url' % branch.remote.name)
1121
Bryan Jacobsf609f912013-05-06 13:36:24 -04001122 if dest_branch is None:
1123 dest_branch = self.dest_branch
1124 if dest_branch is None:
1125 dest_branch = branch.merge
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001126 if not dest_branch.startswith(R_HEADS):
1127 dest_branch = R_HEADS + dest_branch
1128
Shawn O. Pearce339ba9f2008-11-06 09:52:51 -08001129 if not branch.remote.projectname:
1130 branch.remote.projectname = self.name
1131 branch.remote.Save()
1132
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001133 url = branch.remote.ReviewUrl(self.UserEmail)
1134 if url is None:
1135 raise UploadError('review not configured')
1136 cmd = ['push']
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001137
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001138 if url.startswith('ssh://'):
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001139 rp = ['gerrit receive-pack']
1140 for e in people[0]:
1141 rp.append('--reviewer=%s' % sq(e))
1142 for e in people[1]:
1143 rp.append('--cc=%s' % sq(e))
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001144 cmd.append('--receive-pack=%s' % " ".join(rp))
Shawn O. Pearcea5ece0e2010-07-15 16:52:42 -07001145
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001146 cmd.append(url)
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001147
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001148 if dest_branch.startswith(R_HEADS):
1149 dest_branch = dest_branch[len(R_HEADS):]
Brian Harring435370c2012-07-28 15:37:04 -07001150
1151 upload_type = 'for'
1152 if draft:
1153 upload_type = 'drafts'
1154
1155 ref_spec = '%s:refs/%s/%s' % (R_HEADS + branch.name, upload_type,
1156 dest_branch)
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001157 if auto_topic:
1158 ref_spec = ref_spec + '/' + branch.name
Shawn Pearce45d21682013-02-28 00:35:51 -08001159 if not url.startswith('ssh://'):
1160 rp = ['r=%s' % p for p in people[0]] + \
1161 ['cc=%s' % p for p in people[1]]
1162 if rp:
1163 ref_spec = ref_spec + '%' + ','.join(rp)
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001164 cmd.append(ref_spec)
1165
Anthony King7bdac712014-07-16 12:56:40 +01001166 if GitCommand(self, cmd, bare=True).Wait() != 0:
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001167 raise UploadError('Upload failed')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001168
1169 msg = "posted to %s for %s" % (branch.remote.review, dest_branch)
1170 self.bare_git.UpdateRef(R_PUB + branch.name,
1171 R_HEADS + branch.name,
Anthony King7bdac712014-07-16 12:56:40 +01001172 message=msg)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001173
1174
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001175# Sync ##
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001176
Julien Campergue335f5ef2013-10-16 11:02:35 +02001177 def _ExtractArchive(self, tarpath, path=None):
1178 """Extract the given tar on its current location
1179
1180 Args:
1181 - tarpath: The path to the actual tar file
1182
1183 """
1184 try:
1185 with tarfile.open(tarpath, 'r') as tar:
1186 tar.extractall(path=path)
1187 return True
1188 except (IOError, tarfile.TarError) as e:
David Pursehousef33929d2015-08-24 14:39:14 +09001189 _error("Cannot extract archive %s: %s", tarpath, str(e))
Julien Campergue335f5ef2013-10-16 11:02:35 +02001190 return False
1191
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -07001192 def Sync_NetworkHalf(self,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001193 quiet=False,
1194 is_new=None,
1195 current_branch_only=False,
1196 force_sync=False,
1197 clone_bundle=True,
1198 no_tags=False,
1199 archive=False,
1200 optimized_fetch=False,
1201 prune=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001202 """Perform only the network IO portion of the sync process.
1203 Local working directory/branch state is not affected.
1204 """
Julien Campergue335f5ef2013-10-16 11:02:35 +02001205 if archive and not isinstance(self, MetaProject):
1206 if self.remote.url.startswith(('http://', 'https://')):
David Pursehousef33929d2015-08-24 14:39:14 +09001207 _error("%s: Cannot fetch archives from http/https remotes.", self.name)
Julien Campergue335f5ef2013-10-16 11:02:35 +02001208 return False
1209
1210 name = self.relpath.replace('\\', '/')
1211 name = name.replace('/', '_')
1212 tarpath = '%s.tar' % name
1213 topdir = self.manifest.topdir
1214
1215 try:
1216 self._FetchArchive(tarpath, cwd=topdir)
1217 except GitError as e:
David Pursehousef33929d2015-08-24 14:39:14 +09001218 _error('%s', e)
Julien Campergue335f5ef2013-10-16 11:02:35 +02001219 return False
1220
1221 # From now on, we only need absolute tarpath
1222 tarpath = os.path.join(topdir, tarpath)
1223
1224 if not self._ExtractArchive(tarpath, path=topdir):
1225 return False
1226 try:
1227 os.remove(tarpath)
1228 except OSError as e:
David Pursehousef33929d2015-08-24 14:39:14 +09001229 _warn("Cannot remove archive %s: %s", tarpath, str(e))
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001230 self._CopyAndLinkFiles()
Julien Campergue335f5ef2013-10-16 11:02:35 +02001231 return True
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07001232 if is_new is None:
1233 is_new = not self.Exists
Shawn O. Pearce88443382010-10-08 10:02:09 +02001234 if is_new:
Kevin Degiabaa7f32014-11-12 11:27:45 -07001235 self._InitGitDir(force_sync=force_sync)
Jimmie Westera0444582012-10-24 13:44:42 +02001236 else:
1237 self._UpdateHooks()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001238 self._InitRemote()
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001239
1240 if is_new:
1241 alt = os.path.join(self.gitdir, 'objects/info/alternates')
1242 try:
1243 fd = open(alt, 'rb')
1244 try:
1245 alt_dir = fd.readline().rstrip()
1246 finally:
1247 fd.close()
1248 except IOError:
1249 alt_dir = None
1250 else:
1251 alt_dir = None
1252
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -07001253 if clone_bundle \
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001254 and alt_dir is None \
1255 and self._ApplyCloneBundle(initial=is_new, quiet=quiet):
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001256 is_new = False
1257
Shawn O. Pearce6ba6ba02012-05-24 09:46:50 -07001258 if not current_branch_only:
1259 if self.sync_c:
1260 current_branch_only = True
1261 elif not self.manifest._loaded:
1262 # Manifest cannot check defaults until it syncs.
1263 current_branch_only = False
1264 elif self.manifest.default.sync_c:
1265 current_branch_only = True
1266
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001267 if self.clone_depth:
1268 depth = self.clone_depth
1269 else:
1270 depth = self.manifest.manifestProject.config.GetString('repo.depth')
1271
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001272 need_to_fetch = not (optimized_fetch and
1273 (ID_RE.match(self.revisionExpr) and
1274 self._CheckForSha1()))
1275 if (need_to_fetch and
1276 not self._RemoteFetch(initial=is_new, quiet=quiet, alt_dir=alt_dir,
1277 current_branch_only=current_branch_only,
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001278 no_tags=no_tags, prune=prune, depth=depth)):
Anthony King7bdac712014-07-16 12:56:40 +01001279 return False
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001280
1281 if self.worktree:
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001282 self._InitMRef()
1283 else:
1284 self._InitMirrorHead()
1285 try:
1286 os.remove(os.path.join(self.gitdir, 'FETCH_HEAD'))
1287 except OSError:
1288 pass
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001289 return True
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08001290
1291 def PostRepoUpgrade(self):
1292 self._InitHooks()
1293
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001294 def _CopyAndLinkFiles(self):
Simran Basib9a1b732015-08-20 12:19:28 -07001295 if self.manifest.isGitcClient:
1296 return
David Pursehouse8a68ff92012-09-24 12:15:13 +09001297 for copyfile in self.copyfiles:
1298 copyfile._Copy()
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001299 for linkfile in self.linkfiles:
1300 linkfile._Link()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001301
Julien Camperguedd654222014-01-09 16:21:37 +01001302 def GetCommitRevisionId(self):
1303 """Get revisionId of a commit.
1304
1305 Use this method instead of GetRevisionId to get the id of the commit rather
1306 than the id of the current git object (for example, a tag)
1307
1308 """
1309 if not self.revisionExpr.startswith(R_TAGS):
1310 return self.GetRevisionId(self._allrefs)
1311
1312 try:
1313 return self.bare_git.rev_list(self.revisionExpr, '-1')[0]
1314 except GitError:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001315 raise ManifestInvalidRevisionError('revision %s in %s not found' %
1316 (self.revisionExpr, self.name))
Julien Camperguedd654222014-01-09 16:21:37 +01001317
David Pursehouse8a68ff92012-09-24 12:15:13 +09001318 def GetRevisionId(self, all_refs=None):
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001319 if self.revisionId:
1320 return self.revisionId
1321
1322 rem = self.GetRemote(self.remote.name)
1323 rev = rem.ToLocal(self.revisionExpr)
1324
David Pursehouse8a68ff92012-09-24 12:15:13 +09001325 if all_refs is not None and rev in all_refs:
1326 return all_refs[rev]
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001327
1328 try:
1329 return self.bare_git.rev_parse('--verify', '%s^0' % rev)
1330 except GitError:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001331 raise ManifestInvalidRevisionError('revision %s in %s not found' %
1332 (self.revisionExpr, self.name))
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001333
Kevin Degiabaa7f32014-11-12 11:27:45 -07001334 def Sync_LocalHalf(self, syncbuf, force_sync=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001335 """Perform only the local IO portion of the sync process.
1336 Network access is not required.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001337 """
Kevin Degiabaa7f32014-11-12 11:27:45 -07001338 self._InitWorkTree(force_sync=force_sync)
David Pursehouse8a68ff92012-09-24 12:15:13 +09001339 all_refs = self.bare_ref.all
1340 self.CleanPublishedCache(all_refs)
1341 revid = self.GetRevisionId(all_refs)
Skyler Kaufman835cd682011-03-08 12:14:41 -08001342
David Pursehouse1d947b32012-10-25 12:23:11 +09001343 def _doff():
1344 self._FastForward(revid)
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001345 self._CopyAndLinkFiles()
David Pursehouse1d947b32012-10-25 12:23:11 +09001346
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001347 head = self.work_git.GetHead()
1348 if head.startswith(R_HEADS):
1349 branch = head[len(R_HEADS):]
1350 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001351 head = all_refs[head]
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001352 except KeyError:
1353 head = None
1354 else:
1355 branch = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001356
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001357 if branch is None or syncbuf.detach_head:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001358 # Currently on a detached HEAD. The user is assumed to
1359 # not have any local modifications worth worrying about.
1360 #
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -07001361 if self.IsRebaseInProgress():
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001362 syncbuf.fail(self, _PriorSyncFailedError())
1363 return
1364
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001365 if head == revid:
1366 # No changes; don't do anything further.
Florian Vallee7cf1b362012-06-07 17:11:42 +02001367 # Except if the head needs to be detached
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001368 #
Florian Vallee7cf1b362012-06-07 17:11:42 +02001369 if not syncbuf.detach_head:
Dan Willemsen029eaf32015-09-03 12:52:28 -07001370 # The copy/linkfile config may have changed.
1371 self._CopyAndLinkFiles()
Florian Vallee7cf1b362012-06-07 17:11:42 +02001372 return
1373 else:
1374 lost = self._revlist(not_rev(revid), HEAD)
1375 if lost:
1376 syncbuf.info(self, "discarding %d commits", len(lost))
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001377
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001378 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001379 self._Checkout(revid, quiet=True)
Sarah Owensa5be53f2012-09-09 15:37:57 -07001380 except GitError as e:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001381 syncbuf.fail(self, e)
1382 return
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001383 self._CopyAndLinkFiles()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001384 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001385
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001386 if head == revid:
1387 # No changes; don't do anything further.
1388 #
Dan Willemsen029eaf32015-09-03 12:52:28 -07001389 # The copy/linkfile config may have changed.
1390 self._CopyAndLinkFiles()
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001391 return
1392
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001393 branch = self.GetBranch(branch)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001394
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001395 if not branch.LocalMerge:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001396 # The current branch has no tracking configuration.
Anatol Pomazau2a32f6a2011-08-30 10:52:33 -07001397 # Jump off it to a detached HEAD.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001398 #
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001399 syncbuf.info(self,
1400 "leaving %s; does not track upstream",
1401 branch.name)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001402 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001403 self._Checkout(revid, quiet=True)
Sarah Owensa5be53f2012-09-09 15:37:57 -07001404 except GitError as e:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001405 syncbuf.fail(self, e)
1406 return
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001407 self._CopyAndLinkFiles()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001408 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001409
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001410 upstream_gain = self._revlist(not_rev(HEAD), revid)
David Pursehouse8a68ff92012-09-24 12:15:13 +09001411 pub = self.WasPublished(branch.name, all_refs)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001412 if pub:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001413 not_merged = self._revlist(not_rev(revid), pub)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001414 if not_merged:
1415 if upstream_gain:
1416 # The user has published this branch and some of those
1417 # commits are not yet merged upstream. We do not want
1418 # to rewrite the published commits so we punt.
1419 #
Daniel Sandler4c50dee2010-03-02 15:38:03 -05001420 syncbuf.fail(self,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001421 "branch %s is published (but not merged) and is now "
1422 "%d commits behind" % (branch.name, len(upstream_gain)))
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001423 return
Shawn O. Pearce05f66b62009-04-21 08:26:32 -07001424 elif pub == head:
1425 # All published commits are merged, and thus we are a
1426 # strict subset. We can fast-forward safely.
Shawn O. Pearcea54c5272008-10-30 11:03:00 -07001427 #
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001428 syncbuf.later1(self, _doff)
1429 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001430
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001431 # Examine the local commits not in the remote. Find the
1432 # last one attributed to this user, if any.
1433 #
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001434 local_changes = self._revlist(not_rev(revid), HEAD, format='%H %ce')
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001435 last_mine = None
1436 cnt_mine = 0
1437 for commit in local_changes:
Chirayu Desai0eb35cb2013-11-19 18:46:29 +05301438 commit_id, committer_email = commit.decode('utf-8').split(' ', 1)
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001439 if committer_email == self.UserEmail:
1440 last_mine = commit_id
1441 cnt_mine += 1
1442
Shawn O. Pearceda88ff42009-06-03 11:09:12 -07001443 if not upstream_gain and cnt_mine == len(local_changes):
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001444 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001445
1446 if self.IsDirty(consider_untracked=False):
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001447 syncbuf.fail(self, _DirtyError())
1448 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001449
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001450 # If the upstream switched on us, warn the user.
1451 #
1452 if branch.merge != self.revisionExpr:
1453 if branch.merge and self.revisionExpr:
1454 syncbuf.info(self,
1455 'manifest switched %s...%s',
1456 branch.merge,
1457 self.revisionExpr)
1458 elif branch.merge:
1459 syncbuf.info(self,
1460 'manifest no longer tracks %s',
1461 branch.merge)
1462
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001463 if cnt_mine < len(local_changes):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001464 # Upstream rebased. Not everything in HEAD
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001465 # was created by this user.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001466 #
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001467 syncbuf.info(self,
1468 "discarding %d commits removed from upstream",
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001469 len(local_changes) - cnt_mine)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001470
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001471 branch.remote = self.GetRemote(self.remote.name)
Anatol Pomazaucd7c5de2012-03-20 13:45:00 -07001472 if not ID_RE.match(self.revisionExpr):
1473 # in case of manifest sync the revisionExpr might be a SHA1
1474 branch.merge = self.revisionExpr
Conley Owens04f2f0e2014-10-01 17:22:46 -07001475 if not branch.merge.startswith('refs/'):
1476 branch.merge = R_HEADS + branch.merge
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001477 branch.Save()
1478
Mike Pontillod3153822012-02-28 11:53:24 -08001479 if cnt_mine > 0 and self.rebase:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001480 def _dorebase():
Anthony King7bdac712014-07-16 12:56:40 +01001481 self._Rebase(upstream='%s^1' % last_mine, onto=revid)
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001482 self._CopyAndLinkFiles()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001483 syncbuf.later2(self, _dorebase)
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001484 elif local_changes:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001485 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001486 self._ResetHard(revid)
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001487 self._CopyAndLinkFiles()
Sarah Owensa5be53f2012-09-09 15:37:57 -07001488 except GitError as e:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001489 syncbuf.fail(self, e)
1490 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001491 else:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001492 syncbuf.later1(self, _doff)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001493
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -08001494 def AddCopyFile(self, src, dest, absdest):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001495 # dest should already be an absolute path, but src is project relative
1496 # make src an absolute path
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -08001497 abssrc = os.path.join(self.worktree, src)
1498 self.copyfiles.append(_CopyFile(src, dest, abssrc, absdest))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001499
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001500 def AddLinkFile(self, src, dest, absdest):
1501 # dest should already be an absolute path, but src is project relative
Colin Cross0184dcc2015-05-05 00:24:54 -07001502 # make src relative path to dest
1503 absdestdir = os.path.dirname(absdest)
1504 relsrc = os.path.relpath(os.path.join(self.worktree, src), absdestdir)
Wink Saville4c426ef2015-06-03 08:05:17 -07001505 self.linkfiles.append(_LinkFile(self.worktree, src, dest, relsrc, absdest))
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001506
James W. Mills24c13082012-04-12 15:04:13 -05001507 def AddAnnotation(self, name, value, keep):
1508 self.annotations.append(_Annotation(name, value, keep))
1509
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001510 def DownloadPatchSet(self, change_id, patch_id):
1511 """Download a single patch set of a single change to FETCH_HEAD.
1512 """
1513 remote = self.GetRemote(self.remote.name)
1514
1515 cmd = ['fetch', remote.name]
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001516 cmd.append('refs/changes/%2.2d/%d/%d'
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001517 % (change_id % 100, change_id, patch_id))
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001518 if GitCommand(self, cmd, bare=True).Wait() != 0:
1519 return None
1520 return DownloadedChange(self,
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001521 self.GetRevisionId(),
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001522 change_id,
1523 patch_id,
1524 self.bare_git.rev_parse('FETCH_HEAD'))
1525
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001526
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001527# Branch Management ##
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001528
Simran Basib9a1b732015-08-20 12:19:28 -07001529 def StartBranch(self, name, branch_merge=''):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001530 """Create a new branch off the manifest's revision.
1531 """
Simran Basib9a1b732015-08-20 12:19:28 -07001532 if not branch_merge:
1533 branch_merge = self.revisionExpr
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001534 head = self.work_git.GetHead()
1535 if head == (R_HEADS + name):
1536 return True
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001537
David Pursehouse8a68ff92012-09-24 12:15:13 +09001538 all_refs = self.bare_ref.all
Anthony King7bdac712014-07-16 12:56:40 +01001539 if R_HEADS + name in all_refs:
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001540 return GitCommand(self,
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001541 ['checkout', name, '--'],
Anthony King7bdac712014-07-16 12:56:40 +01001542 capture_stdout=True,
1543 capture_stderr=True).Wait() == 0
Shawn O. Pearce0a389e92009-04-10 16:21:18 -07001544
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001545 branch = self.GetBranch(name)
1546 branch.remote = self.GetRemote(self.remote.name)
Simran Basib9a1b732015-08-20 12:19:28 -07001547 branch.merge = branch_merge
1548 if not branch.merge.startswith('refs/') and not ID_RE.match(branch_merge):
1549 branch.merge = R_HEADS + branch_merge
David Pursehouse8a68ff92012-09-24 12:15:13 +09001550 revid = self.GetRevisionId(all_refs)
Shawn O. Pearce0a389e92009-04-10 16:21:18 -07001551
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001552 if head.startswith(R_HEADS):
1553 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001554 head = all_refs[head]
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001555 except KeyError:
1556 head = None
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001557 if revid and head and revid == head:
1558 ref = os.path.join(self.gitdir, R_HEADS + name)
1559 try:
1560 os.makedirs(os.path.dirname(ref))
1561 except OSError:
1562 pass
1563 _lwrite(ref, '%s\n' % revid)
1564 _lwrite(os.path.join(self.worktree, '.git', HEAD),
1565 'ref: %s%s\n' % (R_HEADS, name))
1566 branch.Save()
1567 return True
1568
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001569 if GitCommand(self,
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001570 ['checkout', '-b', branch.name, revid],
Anthony King7bdac712014-07-16 12:56:40 +01001571 capture_stdout=True,
1572 capture_stderr=True).Wait() == 0:
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001573 branch.Save()
1574 return True
1575 return False
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001576
Wink Saville02d79452009-04-10 13:01:24 -07001577 def CheckoutBranch(self, name):
1578 """Checkout a local topic branch.
Doug Anderson3ba5f952011-04-07 12:51:04 -07001579
1580 Args:
1581 name: The name of the branch to checkout.
1582
1583 Returns:
1584 True if the checkout succeeded; False if it didn't; None if the branch
1585 didn't exist.
Wink Saville02d79452009-04-10 13:01:24 -07001586 """
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001587 rev = R_HEADS + name
1588 head = self.work_git.GetHead()
1589 if head == rev:
1590 # Already on the branch
1591 #
1592 return True
Wink Saville02d79452009-04-10 13:01:24 -07001593
David Pursehouse8a68ff92012-09-24 12:15:13 +09001594 all_refs = self.bare_ref.all
Wink Saville02d79452009-04-10 13:01:24 -07001595 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001596 revid = all_refs[rev]
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001597 except KeyError:
1598 # Branch does not exist in this project
1599 #
Doug Anderson3ba5f952011-04-07 12:51:04 -07001600 return None
Wink Saville02d79452009-04-10 13:01:24 -07001601
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001602 if head.startswith(R_HEADS):
1603 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001604 head = all_refs[head]
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001605 except KeyError:
1606 head = None
1607
1608 if head == revid:
1609 # Same revision; just update HEAD to point to the new
1610 # target branch, but otherwise take no other action.
1611 #
1612 _lwrite(os.path.join(self.worktree, '.git', HEAD),
1613 'ref: %s%s\n' % (R_HEADS, name))
1614 return True
1615
1616 return GitCommand(self,
1617 ['checkout', name, '--'],
Anthony King7bdac712014-07-16 12:56:40 +01001618 capture_stdout=True,
1619 capture_stderr=True).Wait() == 0
Wink Saville02d79452009-04-10 13:01:24 -07001620
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001621 def AbandonBranch(self, name):
1622 """Destroy a local topic branch.
Doug Andersondafb1d62011-04-07 11:46:59 -07001623
1624 Args:
1625 name: The name of the branch to abandon.
1626
1627 Returns:
1628 True if the abandon succeeded; False if it didn't; None if the branch
1629 didn't exist.
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001630 """
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001631 rev = R_HEADS + name
David Pursehouse8a68ff92012-09-24 12:15:13 +09001632 all_refs = self.bare_ref.all
1633 if rev not in all_refs:
Doug Andersondafb1d62011-04-07 11:46:59 -07001634 # Doesn't exist
1635 return None
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001636
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001637 head = self.work_git.GetHead()
1638 if head == rev:
1639 # We can't destroy the branch while we are sitting
1640 # on it. Switch to a detached HEAD.
1641 #
David Pursehouse8a68ff92012-09-24 12:15:13 +09001642 head = all_refs[head]
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001643
David Pursehouse8a68ff92012-09-24 12:15:13 +09001644 revid = self.GetRevisionId(all_refs)
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001645 if head == revid:
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001646 _lwrite(os.path.join(self.worktree, '.git', HEAD),
1647 '%s\n' % revid)
1648 else:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001649 self._Checkout(revid, quiet=True)
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001650
1651 return GitCommand(self,
1652 ['branch', '-D', name],
Anthony King7bdac712014-07-16 12:56:40 +01001653 capture_stdout=True,
1654 capture_stderr=True).Wait() == 0
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001655
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001656 def PruneHeads(self):
1657 """Prune any topic branches already merged into upstream.
1658 """
1659 cb = self.CurrentBranch
1660 kill = []
Shawn O. Pearce3778f9d2009-03-02 12:30:50 -08001661 left = self._allrefs
1662 for name in left.keys():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001663 if name.startswith(R_HEADS):
1664 name = name[len(R_HEADS):]
1665 if cb is None or name != cb:
1666 kill.append(name)
1667
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001668 rev = self.GetRevisionId(left)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001669 if cb is not None \
1670 and not self._revlist(HEAD + '...' + rev) \
Anthony King7bdac712014-07-16 12:56:40 +01001671 and not self.IsDirty(consider_untracked=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001672 self.work_git.DetachHead(HEAD)
1673 kill.append(cb)
1674
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001675 if kill:
Shawn O. Pearce5b23f242009-04-17 18:43:33 -07001676 old = self.bare_git.GetHead()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001677
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001678 try:
1679 self.bare_git.DetachHead(rev)
1680
1681 b = ['branch', '-d']
1682 b.extend(kill)
1683 b = GitCommand(self, b, bare=True,
1684 capture_stdout=True,
1685 capture_stderr=True)
1686 b.Wait()
1687 finally:
Dan Willemsen1a799d12015-12-15 13:40:05 -08001688 if ID_RE.match(old):
1689 self.bare_git.DetachHead(old)
1690 else:
1691 self.bare_git.SetHead(old)
Shawn O. Pearce3778f9d2009-03-02 12:30:50 -08001692 left = self._allrefs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001693
Shawn O. Pearce3778f9d2009-03-02 12:30:50 -08001694 for branch in kill:
1695 if (R_HEADS + branch) not in left:
1696 self.CleanPublishedCache()
1697 break
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001698
1699 if cb and cb not in kill:
1700 kill.append(cb)
Shawn O. Pearce7c6c64d2009-03-02 12:38:13 -08001701 kill.sort()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001702
1703 kept = []
1704 for branch in kill:
Anthony King7bdac712014-07-16 12:56:40 +01001705 if R_HEADS + branch in left:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001706 branch = self.GetBranch(branch)
1707 base = branch.LocalMerge
1708 if not base:
1709 base = rev
1710 kept.append(ReviewableBranch(self, branch, base))
1711 return kept
1712
1713
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001714# Submodule Management ##
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001715
1716 def GetRegisteredSubprojects(self):
1717 result = []
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001718
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001719 def rec(subprojects):
1720 if not subprojects:
1721 return
1722 result.extend(subprojects)
1723 for p in subprojects:
1724 rec(p.subprojects)
1725 rec(self.subprojects)
1726 return result
1727
1728 def _GetSubmodules(self):
1729 # Unfortunately we cannot call `git submodule status --recursive` here
1730 # because the working tree might not exist yet, and it cannot be used
1731 # without a working tree in its current implementation.
1732
1733 def get_submodules(gitdir, rev):
1734 # Parse .gitmodules for submodule sub_paths and sub_urls
1735 sub_paths, sub_urls = parse_gitmodules(gitdir, rev)
1736 if not sub_paths:
1737 return []
1738 # Run `git ls-tree` to read SHAs of submodule object, which happen to be
1739 # revision of submodule repository
1740 sub_revs = git_ls_tree(gitdir, rev, sub_paths)
1741 submodules = []
1742 for sub_path, sub_url in zip(sub_paths, sub_urls):
1743 try:
1744 sub_rev = sub_revs[sub_path]
1745 except KeyError:
1746 # Ignore non-exist submodules
1747 continue
1748 submodules.append((sub_rev, sub_path, sub_url))
1749 return submodules
1750
1751 re_path = re.compile(r'^submodule\.([^.]+)\.path=(.*)$')
1752 re_url = re.compile(r'^submodule\.([^.]+)\.url=(.*)$')
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001753
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001754 def parse_gitmodules(gitdir, rev):
1755 cmd = ['cat-file', 'blob', '%s:.gitmodules' % rev]
1756 try:
Anthony King7bdac712014-07-16 12:56:40 +01001757 p = GitCommand(None, cmd, capture_stdout=True, capture_stderr=True,
1758 bare=True, gitdir=gitdir)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001759 except GitError:
1760 return [], []
1761 if p.Wait() != 0:
1762 return [], []
1763
1764 gitmodules_lines = []
1765 fd, temp_gitmodules_path = tempfile.mkstemp()
1766 try:
1767 os.write(fd, p.stdout)
1768 os.close(fd)
1769 cmd = ['config', '--file', temp_gitmodules_path, '--list']
Anthony King7bdac712014-07-16 12:56:40 +01001770 p = GitCommand(None, cmd, capture_stdout=True, capture_stderr=True,
1771 bare=True, gitdir=gitdir)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001772 if p.Wait() != 0:
1773 return [], []
1774 gitmodules_lines = p.stdout.split('\n')
1775 except GitError:
1776 return [], []
1777 finally:
1778 os.remove(temp_gitmodules_path)
1779
1780 names = set()
1781 paths = {}
1782 urls = {}
1783 for line in gitmodules_lines:
1784 if not line:
1785 continue
1786 m = re_path.match(line)
1787 if m:
1788 names.add(m.group(1))
1789 paths[m.group(1)] = m.group(2)
1790 continue
1791 m = re_url.match(line)
1792 if m:
1793 names.add(m.group(1))
1794 urls[m.group(1)] = m.group(2)
1795 continue
1796 names = sorted(names)
1797 return ([paths.get(name, '') for name in names],
1798 [urls.get(name, '') for name in names])
1799
1800 def git_ls_tree(gitdir, rev, paths):
1801 cmd = ['ls-tree', rev, '--']
1802 cmd.extend(paths)
1803 try:
Anthony King7bdac712014-07-16 12:56:40 +01001804 p = GitCommand(None, cmd, capture_stdout=True, capture_stderr=True,
1805 bare=True, gitdir=gitdir)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001806 except GitError:
1807 return []
1808 if p.Wait() != 0:
1809 return []
1810 objects = {}
1811 for line in p.stdout.split('\n'):
1812 if not line.strip():
1813 continue
1814 object_rev, object_path = line.split()[2:4]
1815 objects[object_path] = object_rev
1816 return objects
1817
1818 try:
1819 rev = self.GetRevisionId()
1820 except GitError:
1821 return []
1822 return get_submodules(self.gitdir, rev)
1823
1824 def GetDerivedSubprojects(self):
1825 result = []
1826 if not self.Exists:
1827 # If git repo does not exist yet, querying its submodules will
1828 # mess up its states; so return here.
1829 return result
1830 for rev, path, url in self._GetSubmodules():
1831 name = self.manifest.GetSubprojectName(self, path)
David James8d201162013-10-11 17:03:19 -07001832 relpath, worktree, gitdir, objdir = \
1833 self.manifest.GetSubprojectPaths(self, name, path)
1834 project = self.manifest.paths.get(relpath)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001835 if project:
1836 result.extend(project.GetDerivedSubprojects())
1837 continue
David James8d201162013-10-11 17:03:19 -07001838
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001839 remote = RemoteSpec(self.remote.name,
Anthony King7bdac712014-07-16 12:56:40 +01001840 url=url,
Steve Raed6480452016-08-10 15:00:00 -07001841 pushUrl=self.remote.pushUrl,
Anthony King7bdac712014-07-16 12:56:40 +01001842 review=self.remote.review,
1843 revision=self.remote.revision)
1844 subproject = Project(manifest=self.manifest,
1845 name=name,
1846 remote=remote,
1847 gitdir=gitdir,
1848 objdir=objdir,
1849 worktree=worktree,
1850 relpath=relpath,
Aymen Bouaziz2598ed02016-06-24 14:34:08 +02001851 revisionExpr=rev,
Anthony King7bdac712014-07-16 12:56:40 +01001852 revisionId=rev,
1853 rebase=self.rebase,
1854 groups=self.groups,
1855 sync_c=self.sync_c,
1856 sync_s=self.sync_s,
1857 parent=self,
1858 is_derived=True)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001859 result.append(subproject)
1860 result.extend(subproject.GetDerivedSubprojects())
1861 return result
1862
1863
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001864# Direct Git Commands ##
Chris AtLee2fb64662014-01-16 21:32:33 -05001865 def _CheckForSha1(self):
1866 try:
1867 # if revision (sha or tag) is not present then following function
1868 # throws an error.
1869 self.bare_git.rev_parse('--verify', '%s^0' % self.revisionExpr)
1870 return True
1871 except GitError:
1872 # There is no such persistent revision. We have to fetch it.
1873 return False
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001874
Julien Campergue335f5ef2013-10-16 11:02:35 +02001875 def _FetchArchive(self, tarpath, cwd=None):
1876 cmd = ['archive', '-v', '-o', tarpath]
1877 cmd.append('--remote=%s' % self.remote.url)
1878 cmd.append('--prefix=%s/' % self.relpath)
1879 cmd.append(self.revisionExpr)
1880
1881 command = GitCommand(self, cmd, cwd=cwd,
1882 capture_stdout=True,
1883 capture_stderr=True)
1884
1885 if command.Wait() != 0:
1886 raise GitError('git archive %s: %s' % (self.name, command.stderr))
1887
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001888 def _RemoteFetch(self, name=None,
1889 current_branch_only=False,
Shawn O. Pearce16614f82010-10-29 12:05:43 -07001890 initial=False,
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001891 quiet=False,
Mitchel Humpherys597868b2012-10-29 10:18:34 -07001892 alt_dir=None,
David Pursehouse74cfd272015-10-14 10:50:15 +09001893 no_tags=False,
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001894 prune=False,
1895 depth=None):
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001896
1897 is_sha1 = False
1898 tag_name = None
David Pursehouse9bc422f2014-04-15 10:28:56 +09001899 # The depth should not be used when fetching to a mirror because
1900 # it will result in a shallow repository that cannot be cloned or
1901 # fetched from.
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001902 # The repo project should also never be synced with partial depth.
1903 if self.manifest.IsMirror or self.relpath == '.repo/repo':
1904 depth = None
David Pursehouse9bc422f2014-04-15 10:28:56 +09001905
Shawn Pearce69e04d82014-01-29 12:48:54 -08001906 if depth:
1907 current_branch_only = True
1908
Nasser Grainawi909d58b2014-09-19 12:13:04 -06001909 if ID_RE.match(self.revisionExpr) is not None:
1910 is_sha1 = True
1911
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001912 if current_branch_only:
Nasser Grainawi909d58b2014-09-19 12:13:04 -06001913 if self.revisionExpr.startswith(R_TAGS):
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001914 # this is a tag and its sha1 value should never change
1915 tag_name = self.revisionExpr[len(R_TAGS):]
1916
1917 if is_sha1 or tag_name is not None:
Chris AtLee2fb64662014-01-16 21:32:33 -05001918 if self._CheckForSha1():
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001919 return True
Bertrand SIMONNET3000cda2014-11-25 16:19:29 -08001920 if is_sha1 and not depth:
1921 # When syncing a specific commit and --depth is not set:
1922 # * if upstream is explicitly specified and is not a sha1, fetch only
1923 # upstream as users expect only upstream to be fetch.
1924 # Note: The commit might not be in upstream in which case the sync
1925 # will fail.
1926 # * otherwise, fetch all branches to make sure we end up with the
1927 # specific commit.
Aymen Bouaziz037040f2016-06-28 12:27:23 +02001928 if self.upstream:
1929 current_branch_only = not ID_RE.match(self.upstream)
1930 else:
1931 current_branch_only = False
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001932
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001933 if not name:
1934 name = self.remote.name
Shawn O. Pearcefb231612009-04-10 18:53:46 -07001935
1936 ssh_proxy = False
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07001937 remote = self.GetRemote(name)
1938 if remote.PreConnectFetch():
Shawn O. Pearcefb231612009-04-10 18:53:46 -07001939 ssh_proxy = True
1940
Shawn O. Pearce88443382010-10-08 10:02:09 +02001941 if initial:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001942 if alt_dir and 'objects' == os.path.basename(alt_dir):
1943 ref_dir = os.path.dirname(alt_dir)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001944 packed_refs = os.path.join(self.gitdir, 'packed-refs')
1945 remote = self.GetRemote(name)
1946
David Pursehouse8a68ff92012-09-24 12:15:13 +09001947 all_refs = self.bare_ref.all
1948 ids = set(all_refs.values())
Shawn O. Pearce88443382010-10-08 10:02:09 +02001949 tmp = set()
1950
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301951 for r, ref_id in GitRefs(ref_dir).all.items():
David Pursehouse8a68ff92012-09-24 12:15:13 +09001952 if r not in all_refs:
Shawn O. Pearce88443382010-10-08 10:02:09 +02001953 if r.startswith(R_TAGS) or remote.WritesTo(r):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001954 all_refs[r] = ref_id
1955 ids.add(ref_id)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001956 continue
1957
David Pursehouse8a68ff92012-09-24 12:15:13 +09001958 if ref_id in ids:
Shawn O. Pearce88443382010-10-08 10:02:09 +02001959 continue
1960
David Pursehouse8a68ff92012-09-24 12:15:13 +09001961 r = 'refs/_alt/%s' % ref_id
1962 all_refs[r] = ref_id
1963 ids.add(ref_id)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001964 tmp.add(r)
1965
heping3d7bbc92017-04-12 19:51:47 +08001966 tmp_packed_lines = []
1967 old_packed_lines = []
Shawn O. Pearce88443382010-10-08 10:02:09 +02001968
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301969 for r in sorted(all_refs):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001970 line = '%s %s\n' % (all_refs[r], r)
heping3d7bbc92017-04-12 19:51:47 +08001971 tmp_packed_lines.append(line)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001972 if r not in tmp:
heping3d7bbc92017-04-12 19:51:47 +08001973 old_packed_lines.append(line)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001974
heping3d7bbc92017-04-12 19:51:47 +08001975 tmp_packed = ''.join(tmp_packed_lines)
1976 old_packed = ''.join(old_packed_lines)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001977 _lwrite(packed_refs, tmp_packed)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001978 else:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001979 alt_dir = None
Shawn O. Pearce88443382010-10-08 10:02:09 +02001980
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001981 cmd = ['fetch']
Doug Anderson30d45292011-05-04 15:01:04 -07001982
Conley Owensf97e8382015-01-21 11:12:46 -08001983 if depth:
Doug Anderson30d45292011-05-04 15:01:04 -07001984 cmd.append('--depth=%s' % depth)
Dan Willemseneeab6862015-08-03 13:11:53 -07001985 else:
1986 # If this repo has shallow objects, then we don't know which refs have
1987 # shallow objects or not. Tell git to unshallow all fetched refs. Don't
1988 # do this with projects that don't have shallow objects, since it is less
1989 # efficient.
1990 if os.path.exists(os.path.join(self.gitdir, 'shallow')):
1991 cmd.append('--depth=2147483647')
Doug Anderson30d45292011-05-04 15:01:04 -07001992
Shawn O. Pearce16614f82010-10-29 12:05:43 -07001993 if quiet:
1994 cmd.append('--quiet')
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001995 if not self.worktree:
1996 cmd.append('--update-head-ok')
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001997 cmd.append(name)
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001998
Mitchel Humpherys26c45a72014-03-10 14:21:59 -07001999 # If using depth then we should not get all the tags since they may
2000 # be outside of the depth.
2001 if no_tags or depth:
2002 cmd.append('--no-tags')
2003 else:
2004 cmd.append('--tags')
2005
David Pursehouse74cfd272015-10-14 10:50:15 +09002006 if prune:
2007 cmd.append('--prune')
2008
Conley Owens80b87fe2014-05-09 17:13:44 -07002009 spec = []
Brian Harring14a66742012-09-28 20:21:57 -07002010 if not current_branch_only:
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07002011 # Fetch whole repo
Conley Owens80b87fe2014-05-09 17:13:44 -07002012 spec.append(str((u'+refs/heads/*:') + remote.ToLocal('refs/heads/*')))
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07002013 elif tag_name is not None:
Conley Owens80b87fe2014-05-09 17:13:44 -07002014 spec.append('tag')
2015 spec.append(tag_name)
Nasser Grainawi04e52d62014-09-30 13:34:52 -06002016
David Pursehouse403b64e2015-04-27 10:41:33 +09002017 if not self.manifest.IsMirror:
2018 branch = self.revisionExpr
Kevin Degi679bac42015-06-22 15:31:26 -06002019 if is_sha1 and depth and git_require((1, 8, 3)):
David Pursehouse403b64e2015-04-27 10:41:33 +09002020 # Shallow checkout of a specific commit, fetch from that commit and not
2021 # the heads only as the commit might be deeper in the history.
2022 spec.append(branch)
2023 else:
2024 if is_sha1:
2025 branch = self.upstream
2026 if branch is not None and branch.strip():
2027 if not branch.startswith('refs/'):
2028 branch = R_HEADS + branch
2029 spec.append(str((u'+%s:' % branch) + remote.ToLocal(branch)))
Conley Owens80b87fe2014-05-09 17:13:44 -07002030 cmd.extend(spec)
2031
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002032 ok = False
David Pursehouse8a68ff92012-09-24 12:15:13 +09002033 for _i in range(2):
John L. Villalovos9c76f672015-03-16 20:49:10 -07002034 gitcmd = GitCommand(self, cmd, bare=True, ssh_proxy=ssh_proxy)
John L. Villalovos126e2982015-01-29 21:58:12 -08002035 ret = gitcmd.Wait()
Brian Harring14a66742012-09-28 20:21:57 -07002036 if ret == 0:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002037 ok = True
2038 break
John L. Villalovos126e2982015-01-29 21:58:12 -08002039 # If needed, run the 'git remote prune' the first time through the loop
2040 elif (not _i and
2041 "error:" in gitcmd.stderr and
2042 "git remote prune" in gitcmd.stderr):
2043 prunecmd = GitCommand(self, ['remote', 'prune', name], bare=True,
John L. Villalovos9c76f672015-03-16 20:49:10 -07002044 ssh_proxy=ssh_proxy)
John L. Villalovose30f46b2015-02-25 14:27:02 -08002045 ret = prunecmd.Wait()
John L. Villalovose30f46b2015-02-25 14:27:02 -08002046 if ret:
John L. Villalovos126e2982015-01-29 21:58:12 -08002047 break
2048 continue
Brian Harring14a66742012-09-28 20:21:57 -07002049 elif current_branch_only and is_sha1 and ret == 128:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002050 # Exit code 128 means "couldn't find the ref you asked for"; if we're
2051 # in sha1 mode, we just tried sync'ing from the upstream field; it
2052 # doesn't exist, thus abort the optimization attempt and do a full sync.
Brian Harring14a66742012-09-28 20:21:57 -07002053 break
Colin Crossc4b301f2015-05-13 00:10:02 -07002054 elif ret < 0:
2055 # Git died with a signal, exit immediately
2056 break
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002057 time.sleep(random.randint(30, 45))
Shawn O. Pearce88443382010-10-08 10:02:09 +02002058
2059 if initial:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002060 if alt_dir:
Shawn O. Pearce88443382010-10-08 10:02:09 +02002061 if old_packed != '':
2062 _lwrite(packed_refs, old_packed)
2063 else:
2064 os.remove(packed_refs)
2065 self.bare_git.pack_refs('--all', '--prune')
Brian Harring14a66742012-09-28 20:21:57 -07002066
Aymen Bouaziz6c594462016-10-25 18:03:51 +02002067 if is_sha1 and current_branch_only:
Brian Harring14a66742012-09-28 20:21:57 -07002068 # We just synced the upstream given branch; verify we
2069 # got what we wanted, else trigger a second run of all
2070 # refs.
Chris AtLee2fb64662014-01-16 21:32:33 -05002071 if not self._CheckForSha1():
Aymen Bouaziz6c594462016-10-25 18:03:51 +02002072 if current_branch_only and depth:
2073 # Sync the current branch only with depth set to None
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002074 return self._RemoteFetch(name=name,
2075 current_branch_only=current_branch_only,
Aymen Bouaziz6c594462016-10-25 18:03:51 +02002076 initial=False, quiet=quiet, alt_dir=alt_dir,
2077 depth=None)
2078 else:
2079 # Avoid infinite recursion: sync all branches with depth set to None
2080 return self._RemoteFetch(name=name, current_branch_only=False,
2081 initial=False, quiet=quiet, alt_dir=alt_dir,
2082 depth=None)
Brian Harring14a66742012-09-28 20:21:57 -07002083
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002084 return ok
Shawn O. Pearce88443382010-10-08 10:02:09 +02002085
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002086 def _ApplyCloneBundle(self, initial=False, quiet=False):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002087 if initial and \
2088 (self.manifest.manifestProject.config.GetString('repo.depth') or
2089 self.clone_depth):
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002090 return False
2091
2092 remote = self.GetRemote(self.remote.name)
2093 bundle_url = remote.url + '/clone.bundle'
2094 bundle_url = GitConfig.ForUser().UrlInsteadOf(bundle_url)
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002095 if GetSchemeFromUrl(bundle_url) not in ('http', 'https',
2096 'persistent-http',
2097 'persistent-https'):
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002098 return False
2099
2100 bundle_dst = os.path.join(self.gitdir, 'clone.bundle')
2101 bundle_tmp = os.path.join(self.gitdir, 'clone.bundle.tmp')
2102
2103 exist_dst = os.path.exists(bundle_dst)
2104 exist_tmp = os.path.exists(bundle_tmp)
2105
2106 if not initial and not exist_dst and not exist_tmp:
2107 return False
2108
2109 if not exist_dst:
2110 exist_dst = self._FetchBundle(bundle_url, bundle_tmp, bundle_dst, quiet)
2111 if not exist_dst:
2112 return False
2113
2114 cmd = ['fetch']
2115 if quiet:
2116 cmd.append('--quiet')
2117 if not self.worktree:
2118 cmd.append('--update-head-ok')
2119 cmd.append(bundle_dst)
2120 for f in remote.fetch:
2121 cmd.append(str(f))
2122 cmd.append('refs/tags/*:refs/tags/*')
2123
2124 ok = GitCommand(self, cmd, bare=True).Wait() == 0
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002125 if os.path.exists(bundle_dst):
2126 os.remove(bundle_dst)
2127 if os.path.exists(bundle_tmp):
2128 os.remove(bundle_tmp)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002129 return ok
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002130
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002131 def _FetchBundle(self, srcUrl, tmpPath, dstPath, quiet):
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002132 if os.path.exists(dstPath):
2133 os.remove(dstPath)
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002134
Matt Gumbel2dc810c2012-08-30 09:39:36 -07002135 cmd = ['curl', '--fail', '--output', tmpPath, '--netrc', '--location']
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002136 if quiet:
2137 cmd += ['--silent']
2138 if os.path.exists(tmpPath):
2139 size = os.stat(tmpPath).st_size
2140 if size >= 1024:
2141 cmd += ['--continue-at', '%d' % (size,)]
2142 else:
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002143 os.remove(tmpPath)
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002144 if 'http_proxy' in os.environ and 'darwin' == sys.platform:
2145 cmd += ['--proxy', os.environ['http_proxy']]
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002146 with GetUrlCookieFile(srcUrl, quiet) as (cookiefile, _proxy):
Dave Borowitz137d0132015-01-02 11:12:54 -08002147 if cookiefile:
Dave Borowitz4abf8e62015-01-02 11:39:04 -08002148 cmd += ['--cookie', cookiefile, '--cookie-jar', cookiefile]
Dave Borowitz137d0132015-01-02 11:12:54 -08002149 if srcUrl.startswith('persistent-'):
2150 srcUrl = srcUrl[len('persistent-'):]
2151 cmd += [srcUrl]
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002152
Dave Borowitz137d0132015-01-02 11:12:54 -08002153 if IsTrace():
2154 Trace('%s', ' '.join(cmd))
2155 try:
2156 proc = subprocess.Popen(cmd)
2157 except OSError:
2158 return False
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002159
Dave Borowitz137d0132015-01-02 11:12:54 -08002160 curlret = proc.wait()
Matt Gumbel2dc810c2012-08-30 09:39:36 -07002161
Dave Borowitz137d0132015-01-02 11:12:54 -08002162 if curlret == 22:
2163 # From curl man page:
2164 # 22: HTTP page not retrieved. The requested url was not found or
2165 # returned another error with the HTTP error code being 400 or above.
2166 # This return code only appears if -f, --fail is used.
2167 if not quiet:
2168 print("Server does not provide clone.bundle; ignoring.",
2169 file=sys.stderr)
2170 return False
Matt Gumbel2dc810c2012-08-30 09:39:36 -07002171
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002172 if os.path.exists(tmpPath):
Kris Giesingc8d882a2014-12-23 13:02:32 -08002173 if curlret == 0 and self._IsValidBundle(tmpPath, quiet):
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002174 os.rename(tmpPath, dstPath)
2175 return True
2176 else:
2177 os.remove(tmpPath)
2178 return False
2179 else:
2180 return False
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002181
Kris Giesingc8d882a2014-12-23 13:02:32 -08002182 def _IsValidBundle(self, path, quiet):
Dave Borowitz91f3ba52013-06-03 12:15:23 -07002183 try:
2184 with open(path) as f:
2185 if f.read(16) == '# v2 git bundle\n':
2186 return True
2187 else:
Kris Giesingc8d882a2014-12-23 13:02:32 -08002188 if not quiet:
2189 print("Invalid clone.bundle file; ignoring.", file=sys.stderr)
Dave Borowitz91f3ba52013-06-03 12:15:23 -07002190 return False
2191 except OSError:
2192 return False
2193
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002194 def _Checkout(self, rev, quiet=False):
2195 cmd = ['checkout']
2196 if quiet:
2197 cmd.append('-q')
2198 cmd.append(rev)
2199 cmd.append('--')
2200 if GitCommand(self, cmd).Wait() != 0:
2201 if self._allrefs:
2202 raise GitError('%s checkout %s ' % (self.name, rev))
2203
Anthony King7bdac712014-07-16 12:56:40 +01002204 def _CherryPick(self, rev):
Pierre Tardye5a21222011-03-24 16:28:18 +01002205 cmd = ['cherry-pick']
2206 cmd.append(rev)
2207 cmd.append('--')
2208 if GitCommand(self, cmd).Wait() != 0:
2209 if self._allrefs:
2210 raise GitError('%s cherry-pick %s ' % (self.name, rev))
2211
Anthony King7bdac712014-07-16 12:56:40 +01002212 def _Revert(self, rev):
Erwan Mahea94f1622011-08-19 13:56:09 +02002213 cmd = ['revert']
2214 cmd.append('--no-edit')
2215 cmd.append(rev)
2216 cmd.append('--')
2217 if GitCommand(self, cmd).Wait() != 0:
2218 if self._allrefs:
2219 raise GitError('%s revert %s ' % (self.name, rev))
2220
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002221 def _ResetHard(self, rev, quiet=True):
2222 cmd = ['reset', '--hard']
2223 if quiet:
2224 cmd.append('-q')
2225 cmd.append(rev)
2226 if GitCommand(self, cmd).Wait() != 0:
2227 raise GitError('%s reset --hard %s ' % (self.name, rev))
2228
Anthony King7bdac712014-07-16 12:56:40 +01002229 def _Rebase(self, upstream, onto=None):
Shawn O. Pearce19a83d82009-04-16 08:14:26 -07002230 cmd = ['rebase']
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002231 if onto is not None:
2232 cmd.extend(['--onto', onto])
2233 cmd.append(upstream)
Shawn O. Pearce19a83d82009-04-16 08:14:26 -07002234 if GitCommand(self, cmd).Wait() != 0:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002235 raise GitError('%s rebase %s ' % (self.name, upstream))
2236
Pierre Tardy3d125942012-05-04 12:18:12 +02002237 def _FastForward(self, head, ffonly=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002238 cmd = ['merge', head]
Pierre Tardy3d125942012-05-04 12:18:12 +02002239 if ffonly:
2240 cmd.append("--ff-only")
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002241 if GitCommand(self, cmd).Wait() != 0:
2242 raise GitError('%s merge %s ' % (self.name, head))
2243
Kevin Degiabaa7f32014-11-12 11:27:45 -07002244 def _InitGitDir(self, mirror_git=None, force_sync=False):
Kevin Degi384b3c52014-10-16 16:02:58 -06002245 init_git_dir = not os.path.exists(self.gitdir)
2246 init_obj_dir = not os.path.exists(self.objdir)
Kevin Degib1a07b82015-07-27 13:33:43 -06002247 try:
2248 # Initialize the bare repository, which contains all of the objects.
2249 if init_obj_dir:
2250 os.makedirs(self.objdir)
2251 self.bare_objdir.init()
David James8d201162013-10-11 17:03:19 -07002252
Kevin Degib1a07b82015-07-27 13:33:43 -06002253 # If we have a separate directory to hold refs, initialize it as well.
2254 if self.objdir != self.gitdir:
2255 if init_git_dir:
2256 os.makedirs(self.gitdir)
2257
2258 if init_obj_dir or init_git_dir:
2259 self._ReferenceGitDir(self.objdir, self.gitdir, share_refs=False,
2260 copy_all=True)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002261 try:
2262 self._CheckDirReference(self.objdir, self.gitdir, share_refs=False)
2263 except GitError as e:
Kevin Degiabaa7f32014-11-12 11:27:45 -07002264 if force_sync:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002265 print("Retrying clone after deleting %s" %
2266 self.gitdir, file=sys.stderr)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002267 try:
2268 shutil.rmtree(os.path.realpath(self.gitdir))
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002269 if self.worktree and os.path.exists(os.path.realpath
2270 (self.worktree)):
Kevin Degiabaa7f32014-11-12 11:27:45 -07002271 shutil.rmtree(os.path.realpath(self.worktree))
2272 return self._InitGitDir(mirror_git=mirror_git, force_sync=False)
2273 except:
2274 raise e
2275 raise e
Kevin Degib1a07b82015-07-27 13:33:43 -06002276
Kevin Degi384b3c52014-10-16 16:02:58 -06002277 if init_git_dir:
Kevin Degib1a07b82015-07-27 13:33:43 -06002278 mp = self.manifest.manifestProject
2279 ref_dir = mp.config.GetString('repo.reference') or ''
Kevin Degi384b3c52014-10-16 16:02:58 -06002280
Kevin Degib1a07b82015-07-27 13:33:43 -06002281 if ref_dir or mirror_git:
2282 if not mirror_git:
2283 mirror_git = os.path.join(ref_dir, self.name + '.git')
2284 repo_git = os.path.join(ref_dir, '.repo', 'projects',
2285 self.relpath + '.git')
Shawn O. Pearce2816d4f2009-03-03 17:53:18 -08002286
Kevin Degib1a07b82015-07-27 13:33:43 -06002287 if os.path.exists(mirror_git):
2288 ref_dir = mirror_git
Shawn O. Pearce88443382010-10-08 10:02:09 +02002289
Kevin Degib1a07b82015-07-27 13:33:43 -06002290 elif os.path.exists(repo_git):
2291 ref_dir = repo_git
Shawn O. Pearce88443382010-10-08 10:02:09 +02002292
Kevin Degib1a07b82015-07-27 13:33:43 -06002293 else:
2294 ref_dir = None
Shawn O. Pearce88443382010-10-08 10:02:09 +02002295
Kevin Degib1a07b82015-07-27 13:33:43 -06002296 if ref_dir:
2297 _lwrite(os.path.join(self.gitdir, 'objects/info/alternates'),
2298 os.path.join(ref_dir, 'objects') + '\n')
Shawn O. Pearce88443382010-10-08 10:02:09 +02002299
Kevin Degib1a07b82015-07-27 13:33:43 -06002300 self._UpdateHooks()
2301
2302 m = self.manifest.manifestProject.config
2303 for key in ['user.name', 'user.email']:
2304 if m.Has(key, include_defaults=False):
2305 self.config.SetString(key, m.GetString(key))
David Pursehouse76a4a9d2016-08-16 12:11:12 +09002306 self.config.SetString('filter.lfs.smudge', 'git-lfs smudge --skip -- %f')
Kevin Degib1a07b82015-07-27 13:33:43 -06002307 if self.manifest.IsMirror:
2308 self.config.SetString('core.bare', 'true')
Shawn O. Pearce88443382010-10-08 10:02:09 +02002309 else:
Kevin Degib1a07b82015-07-27 13:33:43 -06002310 self.config.SetString('core.bare', None)
2311 except Exception:
2312 if init_obj_dir and os.path.exists(self.objdir):
2313 shutil.rmtree(self.objdir)
2314 if init_git_dir and os.path.exists(self.gitdir):
2315 shutil.rmtree(self.gitdir)
2316 raise
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002317
Jimmie Westera0444582012-10-24 13:44:42 +02002318 def _UpdateHooks(self):
2319 if os.path.exists(self.gitdir):
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002320 self._InitHooks()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002321
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002322 def _InitHooks(self):
Jesse Hall672cc492013-11-27 11:17:13 -08002323 hooks = os.path.realpath(self._gitdir_path('hooks'))
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002324 if not os.path.exists(hooks):
2325 os.makedirs(hooks)
Jonathan Nieder93719792015-03-17 11:29:58 -07002326 for stock_hook in _ProjectHooks():
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002327 name = os.path.basename(stock_hook)
2328
Victor Boivie65e0f352011-04-18 11:23:29 +02002329 if name in ('commit-msg',) and not self.remote.review \
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002330 and self is not self.manifest.manifestProject:
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002331 # Don't install a Gerrit Code Review hook if this
2332 # project does not appear to use it for reviews.
2333 #
Victor Boivie65e0f352011-04-18 11:23:29 +02002334 # Since the manifest project is one of those, but also
2335 # managed through gerrit, it's excluded
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002336 continue
2337
2338 dst = os.path.join(hooks, name)
2339 if os.path.islink(dst):
2340 continue
2341 if os.path.exists(dst):
2342 if filecmp.cmp(stock_hook, dst, shallow=False):
2343 os.remove(dst)
2344 else:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002345 _warn("%s: Not replacing locally modified %s hook",
2346 self.relpath, name)
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002347 continue
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002348 try:
Mickaël Salaünb9477bc2012-08-05 13:39:26 +02002349 os.symlink(os.path.relpath(stock_hook, os.path.dirname(dst)), dst)
Sarah Owensa5be53f2012-09-09 15:37:57 -07002350 except OSError as e:
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002351 if e.errno == errno.EPERM:
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002352 raise GitError('filesystem must support symlinks')
2353 else:
2354 raise
2355
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002356 def _InitRemote(self):
Shawn O. Pearced1f70d92009-05-19 14:58:02 -07002357 if self.remote.url:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002358 remote = self.GetRemote(self.remote.name)
Shawn O. Pearced1f70d92009-05-19 14:58:02 -07002359 remote.url = self.remote.url
Steve Raed6480452016-08-10 15:00:00 -07002360 remote.pushUrl = self.remote.pushUrl
Shawn O. Pearced1f70d92009-05-19 14:58:02 -07002361 remote.review = self.remote.review
2362 remote.projectname = self.name
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002363
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002364 if self.worktree:
2365 remote.ResetFetch(mirror=False)
2366 else:
2367 remote.ResetFetch(mirror=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002368 remote.Save()
2369
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002370 def _InitMRef(self):
2371 if self.manifest.branch:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002372 self._InitAnyMRef(R_M + self.manifest.branch)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002373
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002374 def _InitMirrorHead(self):
Shawn O. Pearcefe200ee2009-06-01 15:28:21 -07002375 self._InitAnyMRef(HEAD)
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002376
2377 def _InitAnyMRef(self, ref):
2378 cur = self.bare_ref.symref(ref)
2379
2380 if self.revisionId:
2381 if cur != '' or self.bare_ref.get(ref) != self.revisionId:
2382 msg = 'manifest set to %s' % self.revisionId
2383 dst = self.revisionId + '^0'
Anthony King7bdac712014-07-16 12:56:40 +01002384 self.bare_git.UpdateRef(ref, dst, message=msg, detach=True)
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002385 else:
2386 remote = self.GetRemote(self.remote.name)
2387 dst = remote.ToLocal(self.revisionExpr)
2388 if cur != dst:
2389 msg = 'manifest set to %s' % self.revisionExpr
2390 self.bare_git.symbolic_ref('-m', msg, ref, dst)
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002391
Kevin Degi384b3c52014-10-16 16:02:58 -06002392 def _CheckDirReference(self, srcdir, destdir, share_refs):
Dan Willemsenbdb866e2016-04-05 17:22:02 -07002393 symlink_files = self.shareable_files[:]
2394 symlink_dirs = self.shareable_dirs[:]
Kevin Degi384b3c52014-10-16 16:02:58 -06002395 if share_refs:
2396 symlink_files += self.working_tree_files
2397 symlink_dirs += self.working_tree_dirs
2398 to_symlink = symlink_files + symlink_dirs
2399 for name in set(to_symlink):
2400 dst = os.path.realpath(os.path.join(destdir, name))
2401 if os.path.lexists(dst):
2402 src = os.path.realpath(os.path.join(srcdir, name))
2403 # Fail if the links are pointing to the wrong place
2404 if src != dst:
Marc Herbertec287902016-10-27 12:58:26 -07002405 _error('%s is different in %s vs %s', name, destdir, srcdir)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002406 raise GitError('--force-sync not enabled; cannot overwrite a local '
Simon Ruggierf9b76832015-07-31 17:18:34 -04002407 'work tree. If you\'re comfortable with the '
2408 'possibility of losing the work tree\'s git metadata,'
2409 ' use `repo sync --force-sync {0}` to '
2410 'proceed.'.format(self.relpath))
Kevin Degi384b3c52014-10-16 16:02:58 -06002411
David James8d201162013-10-11 17:03:19 -07002412 def _ReferenceGitDir(self, gitdir, dotgit, share_refs, copy_all):
2413 """Update |dotgit| to reference |gitdir|, using symlinks where possible.
2414
2415 Args:
2416 gitdir: The bare git repository. Must already be initialized.
2417 dotgit: The repository you would like to initialize.
2418 share_refs: If true, |dotgit| will store its refs under |gitdir|.
2419 Only one work tree can store refs under a given |gitdir|.
2420 copy_all: If true, copy all remaining files from |gitdir| -> |dotgit|.
2421 This saves you the effort of initializing |dotgit| yourself.
2422 """
Dan Willemsenbdb866e2016-04-05 17:22:02 -07002423 symlink_files = self.shareable_files[:]
2424 symlink_dirs = self.shareable_dirs[:]
David James8d201162013-10-11 17:03:19 -07002425 if share_refs:
Kevin Degi384b3c52014-10-16 16:02:58 -06002426 symlink_files += self.working_tree_files
2427 symlink_dirs += self.working_tree_dirs
David James8d201162013-10-11 17:03:19 -07002428 to_symlink = symlink_files + symlink_dirs
2429
2430 to_copy = []
2431 if copy_all:
2432 to_copy = os.listdir(gitdir)
2433
Dan Willemsen2a3e1522015-07-30 20:43:33 -07002434 dotgit = os.path.realpath(dotgit)
David James8d201162013-10-11 17:03:19 -07002435 for name in set(to_copy).union(to_symlink):
2436 try:
2437 src = os.path.realpath(os.path.join(gitdir, name))
Dan Willemsen2a3e1522015-07-30 20:43:33 -07002438 dst = os.path.join(dotgit, name)
David James8d201162013-10-11 17:03:19 -07002439
Kevin Degi384b3c52014-10-16 16:02:58 -06002440 if os.path.lexists(dst):
2441 continue
David James8d201162013-10-11 17:03:19 -07002442
2443 # If the source dir doesn't exist, create an empty dir.
2444 if name in symlink_dirs and not os.path.lexists(src):
2445 os.makedirs(src)
2446
Cheuk Leung8ac0c962015-07-06 21:33:00 +02002447 if name in to_symlink:
2448 os.symlink(os.path.relpath(src, os.path.dirname(dst)), dst)
2449 elif copy_all and not os.path.islink(dst):
2450 if os.path.isdir(src):
2451 shutil.copytree(src, dst)
2452 elif os.path.isfile(src):
2453 shutil.copy(src, dst)
2454
Conley Owens80b87fe2014-05-09 17:13:44 -07002455 # If the source file doesn't exist, ensure the destination
2456 # file doesn't either.
2457 if name in symlink_files and not os.path.lexists(src):
2458 try:
2459 os.remove(dst)
2460 except OSError:
2461 pass
2462
David James8d201162013-10-11 17:03:19 -07002463 except OSError as e:
2464 if e.errno == errno.EPERM:
Kevin Degiabaa7f32014-11-12 11:27:45 -07002465 raise DownloadError('filesystem must support symlinks')
David James8d201162013-10-11 17:03:19 -07002466 else:
2467 raise
2468
Kevin Degiabaa7f32014-11-12 11:27:45 -07002469 def _InitWorkTree(self, force_sync=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002470 dotgit = os.path.join(self.worktree, '.git')
Kevin Degi384b3c52014-10-16 16:02:58 -06002471 init_dotgit = not os.path.exists(dotgit)
Kevin Degib1a07b82015-07-27 13:33:43 -06002472 try:
2473 if init_dotgit:
2474 os.makedirs(dotgit)
2475 self._ReferenceGitDir(self.gitdir, dotgit, share_refs=True,
2476 copy_all=False)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002477
Kevin Degiabaa7f32014-11-12 11:27:45 -07002478 try:
2479 self._CheckDirReference(self.gitdir, dotgit, share_refs=True)
2480 except GitError as e:
2481 if force_sync:
2482 try:
2483 shutil.rmtree(dotgit)
2484 return self._InitWorkTree(force_sync=False)
2485 except:
2486 raise e
2487 raise e
Kevin Degi384b3c52014-10-16 16:02:58 -06002488
Kevin Degib1a07b82015-07-27 13:33:43 -06002489 if init_dotgit:
2490 _lwrite(os.path.join(dotgit, HEAD), '%s\n' % self.GetRevisionId())
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002491
Kevin Degib1a07b82015-07-27 13:33:43 -06002492 cmd = ['read-tree', '--reset', '-u']
2493 cmd.append('-v')
2494 cmd.append(HEAD)
2495 if GitCommand(self, cmd).Wait() != 0:
2496 raise GitError("cannot initialize work tree")
Victor Boivie0960b5b2010-11-26 13:42:13 +01002497
Kevin Degib1a07b82015-07-27 13:33:43 -06002498 self._CopyAndLinkFiles()
2499 except Exception:
2500 if init_dotgit:
2501 shutil.rmtree(dotgit)
2502 raise
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002503
2504 def _gitdir_path(self, path):
David James8d201162013-10-11 17:03:19 -07002505 return os.path.realpath(os.path.join(self.gitdir, path))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002506
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07002507 def _revlist(self, *args, **kw):
2508 a = []
2509 a.extend(args)
2510 a.append('--')
2511 return self.work_git.rev_list(*a, **kw)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002512
2513 @property
2514 def _allrefs(self):
Shawn O. Pearced237b692009-04-17 18:49:50 -07002515 return self.bare_ref.all
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002516
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002517 def _getLogs(self, rev1, rev2, oneline=False, color=True, pretty_format=None):
Julien Camperguedd654222014-01-09 16:21:37 +01002518 """Get logs between two revisions of this project."""
2519 comp = '..'
2520 if rev1:
2521 revs = [rev1]
2522 if rev2:
2523 revs.extend([comp, rev2])
2524 cmd = ['log', ''.join(revs)]
2525 out = DiffColoring(self.config)
2526 if out.is_on and color:
2527 cmd.append('--color')
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002528 if pretty_format is not None:
2529 cmd.append('--pretty=format:%s' % pretty_format)
Julien Camperguedd654222014-01-09 16:21:37 +01002530 if oneline:
2531 cmd.append('--oneline')
2532
2533 try:
2534 log = GitCommand(self, cmd, capture_stdout=True, capture_stderr=True)
2535 if log.Wait() == 0:
2536 return log.stdout
2537 except GitError:
2538 # worktree may not exist if groups changed for example. In that case,
2539 # try in gitdir instead.
2540 if not os.path.exists(self.worktree):
2541 return self.bare_git.log(*cmd[1:])
2542 else:
2543 raise
2544 return None
2545
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002546 def getAddedAndRemovedLogs(self, toProject, oneline=False, color=True,
2547 pretty_format=None):
Julien Camperguedd654222014-01-09 16:21:37 +01002548 """Get the list of logs from this revision to given revisionId"""
2549 logs = {}
2550 selfId = self.GetRevisionId(self._allrefs)
2551 toId = toProject.GetRevisionId(toProject._allrefs)
2552
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002553 logs['added'] = self._getLogs(selfId, toId, oneline=oneline, color=color,
2554 pretty_format=pretty_format)
2555 logs['removed'] = self._getLogs(toId, selfId, oneline=oneline, color=color,
2556 pretty_format=pretty_format)
Julien Camperguedd654222014-01-09 16:21:37 +01002557 return logs
2558
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002559 class _GitGetByExec(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002560
David James8d201162013-10-11 17:03:19 -07002561 def __init__(self, project, bare, gitdir):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002562 self._project = project
2563 self._bare = bare
David James8d201162013-10-11 17:03:19 -07002564 self._gitdir = gitdir
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002565
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002566 def LsOthers(self):
2567 p = GitCommand(self._project,
2568 ['ls-files',
2569 '-z',
2570 '--others',
2571 '--exclude-standard'],
Anthony King7bdac712014-07-16 12:56:40 +01002572 bare=False,
David James8d201162013-10-11 17:03:19 -07002573 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002574 capture_stdout=True,
2575 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002576 if p.Wait() == 0:
2577 out = p.stdout
2578 if out:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002579 # Backslash is not anomalous
David Pursehouse1d947b32012-10-25 12:23:11 +09002580 return out[:-1].split('\0') # pylint: disable=W1401
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002581 return []
2582
2583 def DiffZ(self, name, *args):
2584 cmd = [name]
2585 cmd.append('-z')
2586 cmd.extend(args)
2587 p = GitCommand(self._project,
2588 cmd,
David James8d201162013-10-11 17:03:19 -07002589 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002590 bare=False,
2591 capture_stdout=True,
2592 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002593 try:
2594 out = p.process.stdout.read()
2595 r = {}
2596 if out:
David Pursehouse1d947b32012-10-25 12:23:11 +09002597 out = iter(out[:-1].split('\0')) # pylint: disable=W1401
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002598 while out:
Shawn O. Pearce02dbb6d2008-10-21 13:59:08 -07002599 try:
Anthony King2cd1f042014-05-05 21:24:05 +01002600 info = next(out)
2601 path = next(out)
Shawn O. Pearce02dbb6d2008-10-21 13:59:08 -07002602 except StopIteration:
2603 break
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002604
2605 class _Info(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002606
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002607 def __init__(self, path, omode, nmode, oid, nid, state):
2608 self.path = path
2609 self.src_path = None
2610 self.old_mode = omode
2611 self.new_mode = nmode
2612 self.old_id = oid
2613 self.new_id = nid
2614
2615 if len(state) == 1:
2616 self.status = state
2617 self.level = None
2618 else:
2619 self.status = state[:1]
2620 self.level = state[1:]
2621 while self.level.startswith('0'):
2622 self.level = self.level[1:]
2623
2624 info = info[1:].split(' ')
David Pursehouse8f62fb72012-11-14 12:09:38 +09002625 info = _Info(path, *info)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002626 if info.status in ('R', 'C'):
2627 info.src_path = info.path
Anthony King2cd1f042014-05-05 21:24:05 +01002628 info.path = next(out)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002629 r[info.path] = info
2630 return r
2631 finally:
2632 p.Wait()
2633
2634 def GetHead(self):
Shawn O. Pearce5b23f242009-04-17 18:43:33 -07002635 if self._bare:
2636 path = os.path.join(self._project.gitdir, HEAD)
2637 else:
2638 path = os.path.join(self._project.worktree, '.git', HEAD)
Conley Owens75ee0572012-11-15 17:33:11 -08002639 try:
2640 fd = open(path, 'rb')
Dan Sandler53e902a2014-03-09 13:20:02 -04002641 except IOError as e:
2642 raise NoManifestException(path, str(e))
Shawn O. Pearce76ca9f82009-04-18 14:48:03 -07002643 try:
2644 line = fd.read()
2645 finally:
2646 fd.close()
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302647 try:
2648 line = line.decode()
2649 except AttributeError:
2650 pass
Shawn O. Pearce5b23f242009-04-17 18:43:33 -07002651 if line.startswith('ref: '):
2652 return line[5:-1]
2653 return line[:-1]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002654
2655 def SetHead(self, ref, message=None):
2656 cmdv = []
2657 if message is not None:
2658 cmdv.extend(['-m', message])
2659 cmdv.append(HEAD)
2660 cmdv.append(ref)
2661 self.symbolic_ref(*cmdv)
2662
2663 def DetachHead(self, new, message=None):
2664 cmdv = ['--no-deref']
2665 if message is not None:
2666 cmdv.extend(['-m', message])
2667 cmdv.append(HEAD)
2668 cmdv.append(new)
2669 self.update_ref(*cmdv)
2670
2671 def UpdateRef(self, name, new, old=None,
2672 message=None,
2673 detach=False):
2674 cmdv = []
2675 if message is not None:
2676 cmdv.extend(['-m', message])
2677 if detach:
2678 cmdv.append('--no-deref')
2679 cmdv.append(name)
2680 cmdv.append(new)
2681 if old is not None:
2682 cmdv.append(old)
2683 self.update_ref(*cmdv)
2684
2685 def DeleteRef(self, name, old=None):
2686 if not old:
2687 old = self.rev_parse(name)
2688 self.update_ref('-d', name, old)
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07002689 self._project.bare_ref.deleted(name)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002690
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07002691 def rev_list(self, *args, **kw):
2692 if 'format' in kw:
2693 cmdv = ['log', '--pretty=format:%s' % kw['format']]
2694 else:
2695 cmdv = ['rev-list']
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002696 cmdv.extend(args)
2697 p = GitCommand(self._project,
2698 cmdv,
Anthony King7bdac712014-07-16 12:56:40 +01002699 bare=self._bare,
David James8d201162013-10-11 17:03:19 -07002700 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002701 capture_stdout=True,
2702 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002703 r = []
2704 for line in p.process.stdout:
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07002705 if line[-1] == '\n':
2706 line = line[:-1]
2707 r.append(line)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002708 if p.Wait() != 0:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002709 raise GitError('%s rev-list %s: %s' %
2710 (self._project.name, str(args), p.stderr))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002711 return r
2712
2713 def __getattr__(self, name):
Doug Anderson37282b42011-03-04 11:54:18 -08002714 """Allow arbitrary git commands using pythonic syntax.
2715
2716 This allows you to do things like:
2717 git_obj.rev_parse('HEAD')
2718
2719 Since we don't have a 'rev_parse' method defined, the __getattr__ will
2720 run. We'll replace the '_' with a '-' and try to run a git command.
Dave Borowitz091f8932012-10-23 17:01:04 -07002721 Any other positional arguments will be passed to the git command, and the
2722 following keyword arguments are supported:
2723 config: An optional dict of git config options to be passed with '-c'.
Doug Anderson37282b42011-03-04 11:54:18 -08002724
2725 Args:
2726 name: The name of the git command to call. Any '_' characters will
2727 be replaced with '-'.
2728
2729 Returns:
2730 A callable object that will try to call git with the named command.
2731 """
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002732 name = name.replace('_', '-')
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002733
Dave Borowitz091f8932012-10-23 17:01:04 -07002734 def runner(*args, **kwargs):
2735 cmdv = []
2736 config = kwargs.pop('config', None)
2737 for k in kwargs:
2738 raise TypeError('%s() got an unexpected keyword argument %r'
2739 % (name, k))
2740 if config is not None:
Dave Borowitzb42b4742012-10-31 12:27:27 -07002741 if not git_require((1, 7, 2)):
2742 raise ValueError('cannot set config on command line for %s()'
2743 % name)
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302744 for k, v in config.items():
Dave Borowitz091f8932012-10-23 17:01:04 -07002745 cmdv.append('-c')
2746 cmdv.append('%s=%s' % (k, v))
2747 cmdv.append(name)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002748 cmdv.extend(args)
2749 p = GitCommand(self._project,
2750 cmdv,
Anthony King7bdac712014-07-16 12:56:40 +01002751 bare=self._bare,
David James8d201162013-10-11 17:03:19 -07002752 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002753 capture_stdout=True,
2754 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002755 if p.Wait() != 0:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002756 raise GitError('%s %s: %s' %
2757 (self._project.name, name, p.stderr))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002758 r = p.stdout
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302759 try:
Conley Owensedd01512013-09-26 12:59:58 -07002760 r = r.decode('utf-8')
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302761 except AttributeError:
2762 pass
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002763 if r.endswith('\n') and r.index('\n') == len(r) - 1:
2764 return r[:-1]
2765 return r
2766 return runner
2767
2768
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002769class _PriorSyncFailedError(Exception):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002770
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002771 def __str__(self):
2772 return 'prior sync failed; rebase still in progress'
2773
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002774
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002775class _DirtyError(Exception):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002776
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002777 def __str__(self):
2778 return 'contains uncommitted changes'
2779
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002780
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002781class _InfoMessage(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002782
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002783 def __init__(self, project, text):
2784 self.project = project
2785 self.text = text
2786
2787 def Print(self, syncbuf):
2788 syncbuf.out.info('%s/: %s', self.project.relpath, self.text)
2789 syncbuf.out.nl()
2790
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002791
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002792class _Failure(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002793
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002794 def __init__(self, project, why):
2795 self.project = project
2796 self.why = why
2797
2798 def Print(self, syncbuf):
2799 syncbuf.out.fail('error: %s/: %s',
2800 self.project.relpath,
2801 str(self.why))
2802 syncbuf.out.nl()
2803
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002804
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002805class _Later(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002806
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002807 def __init__(self, project, action):
2808 self.project = project
2809 self.action = action
2810
2811 def Run(self, syncbuf):
2812 out = syncbuf.out
2813 out.project('project %s/', self.project.relpath)
2814 out.nl()
2815 try:
2816 self.action()
2817 out.nl()
2818 return True
David Pursehouse8a68ff92012-09-24 12:15:13 +09002819 except GitError:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002820 out.nl()
2821 return False
2822
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002823
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002824class _SyncColoring(Coloring):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002825
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002826 def __init__(self, config):
2827 Coloring.__init__(self, config, 'reposync')
Anthony King7bdac712014-07-16 12:56:40 +01002828 self.project = self.printer('header', attr='bold')
2829 self.info = self.printer('info')
2830 self.fail = self.printer('fail', fg='red')
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002831
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002832
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002833class SyncBuffer(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002834
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002835 def __init__(self, config, detach_head=False):
2836 self._messages = []
2837 self._failures = []
2838 self._later_queue1 = []
2839 self._later_queue2 = []
2840
2841 self.out = _SyncColoring(config)
2842 self.out.redirect(sys.stderr)
2843
2844 self.detach_head = detach_head
2845 self.clean = True
2846
2847 def info(self, project, fmt, *args):
2848 self._messages.append(_InfoMessage(project, fmt % args))
2849
2850 def fail(self, project, err=None):
2851 self._failures.append(_Failure(project, err))
2852 self.clean = False
2853
2854 def later1(self, project, what):
2855 self._later_queue1.append(_Later(project, what))
2856
2857 def later2(self, project, what):
2858 self._later_queue2.append(_Later(project, what))
2859
2860 def Finish(self):
2861 self._PrintMessages()
2862 self._RunLater()
2863 self._PrintMessages()
2864 return self.clean
2865
2866 def _RunLater(self):
2867 for q in ['_later_queue1', '_later_queue2']:
2868 if not self._RunQueue(q):
2869 return
2870
2871 def _RunQueue(self, queue):
2872 for m in getattr(self, queue):
2873 if not m.Run(self):
2874 self.clean = False
2875 return False
2876 setattr(self, queue, [])
2877 return True
2878
2879 def _PrintMessages(self):
2880 for m in self._messages:
2881 m.Print(self)
2882 for m in self._failures:
2883 m.Print(self)
2884
2885 self._messages = []
2886 self._failures = []
2887
2888
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002889class MetaProject(Project):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002890
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002891 """A special project housed under .repo.
2892 """
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002893
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002894 def __init__(self, manifest, name, gitdir, worktree):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002895 Project.__init__(self,
Anthony King7bdac712014-07-16 12:56:40 +01002896 manifest=manifest,
2897 name=name,
2898 gitdir=gitdir,
2899 objdir=gitdir,
2900 worktree=worktree,
2901 remote=RemoteSpec('origin'),
2902 relpath='.repo/%s' % name,
2903 revisionExpr='refs/heads/master',
2904 revisionId=None,
2905 groups=None)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002906
2907 def PreSync(self):
2908 if self.Exists:
2909 cb = self.CurrentBranch
2910 if cb:
2911 base = self.GetBranch(cb).merge
2912 if base:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002913 self.revisionExpr = base
2914 self.revisionId = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002915
Anthony King7bdac712014-07-16 12:56:40 +01002916 def MetaBranchSwitch(self):
Florian Vallee5d016502012-06-07 17:19:26 +02002917 """ Prepare MetaProject for manifest branch switch
2918 """
2919
2920 # detach and delete manifest branch, allowing a new
2921 # branch to take over
Anthony King7bdac712014-07-16 12:56:40 +01002922 syncbuf = SyncBuffer(self.config, detach_head=True)
Florian Vallee5d016502012-06-07 17:19:26 +02002923 self.Sync_LocalHalf(syncbuf)
2924 syncbuf.Finish()
2925
2926 return GitCommand(self,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002927 ['update-ref', '-d', 'refs/heads/default'],
2928 capture_stdout=True,
2929 capture_stderr=True).Wait() == 0
Florian Vallee5d016502012-06-07 17:19:26 +02002930
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002931 @property
Shawn O. Pearcef6906872009-04-18 10:49:00 -07002932 def LastFetch(self):
2933 try:
2934 fh = os.path.join(self.gitdir, 'FETCH_HEAD')
2935 return os.path.getmtime(fh)
2936 except OSError:
2937 return 0
2938
2939 @property
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002940 def HasChanges(self):
2941 """Has the remote received new commits not yet checked out?
2942 """
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002943 if not self.remote or not self.revisionExpr:
Shawn O. Pearce336f7bd2009-04-18 10:39:28 -07002944 return False
2945
David Pursehouse8a68ff92012-09-24 12:15:13 +09002946 all_refs = self.bare_ref.all
2947 revid = self.GetRevisionId(all_refs)
Shawn O. Pearce336f7bd2009-04-18 10:39:28 -07002948 head = self.work_git.GetHead()
2949 if head.startswith(R_HEADS):
2950 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09002951 head = all_refs[head]
Shawn O. Pearce336f7bd2009-04-18 10:39:28 -07002952 except KeyError:
2953 head = None
2954
2955 if revid == head:
2956 return False
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002957 elif self._revlist(not_rev(HEAD), revid):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002958 return True
2959 return False