blob: 2248a7efac92ad721447078f2505531b9d7e2167 [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
Renaud Paquayd5cec5e2016-11-01 11:24:03 -070038import platform_utils
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -070039from trace import IsTrace, Trace
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070040
Shawn O. Pearced237b692009-04-17 18:49:50 -070041from git_refs import GitRefs, HEAD, R_HEADS, R_TAGS, R_PUB, R_M
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070042
David Pursehouse59bbb582013-05-17 10:49:33 +090043from pyversion import is_python3
Mike Frysinger40252c22016-08-15 21:23:44 -040044if is_python3():
45 import urllib.parse
46else:
47 import imp
48 import urlparse
49 urllib = imp.new_module('urllib')
50 urllib.parse = urlparse
David Pursehouse59bbb582013-05-17 10:49:33 +090051 # pylint:disable=W0622
Chirayu Desai217ea7d2013-03-01 19:14:38 +053052 input = raw_input
David Pursehouse59bbb582013-05-17 10:49:33 +090053 # pylint:enable=W0622
Chirayu Desai217ea7d2013-03-01 19:14:38 +053054
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070055
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -070056def _lwrite(path, content):
57 lock = '%s.lock' % path
58
Chirayu Desai303a82f2014-08-19 22:57:17 +053059 fd = open(lock, 'w')
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -070060 try:
61 fd.write(content)
62 finally:
63 fd.close()
64
65 try:
Renaud Paquayad1abcb2016-11-01 11:34:55 -070066 platform_utils.rename(lock, path)
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -070067 except OSError:
Renaud Paquay010fed72016-11-11 14:25:29 -080068 platform_utils.remove(lock)
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -070069 raise
70
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070071
Shawn O. Pearce48244782009-04-16 08:25:57 -070072def _error(fmt, *args):
73 msg = fmt % args
Sarah Owenscecd1d82012-11-01 22:59:27 -070074 print('error: %s' % msg, file=sys.stderr)
Shawn O. Pearce48244782009-04-16 08:25:57 -070075
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070076
David Pursehousef33929d2015-08-24 14:39:14 +090077def _warn(fmt, *args):
78 msg = fmt % args
79 print('warn: %s' % msg, file=sys.stderr)
80
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070081
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -070082def not_rev(r):
83 return '^' + r
84
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070085
Shawn O. Pearceb54a3922009-01-05 16:18:58 -080086def sq(r):
87 return "'" + r.replace("'", "'\''") + "'"
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -080088
Jonathan Nieder93719792015-03-17 11:29:58 -070089_project_hook_list = None
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -070090
91
Jonathan Nieder93719792015-03-17 11:29:58 -070092def _ProjectHooks():
93 """List the hooks present in the 'hooks' directory.
94
95 These hooks are project hooks and are copied to the '.git/hooks' directory
96 of all subprojects.
97
98 This function caches the list of hooks (based on the contents of the
99 'repo/hooks' directory) on the first call.
100
101 Returns:
102 A list of absolute paths to all of the files in the hooks directory.
103 """
104 global _project_hook_list
105 if _project_hook_list is None:
Renaud Paquay227ad2e2016-11-01 14:37:13 -0700106 d = platform_utils.realpath(os.path.abspath(os.path.dirname(__file__)))
Jonathan Nieder93719792015-03-17 11:29:58 -0700107 d = os.path.join(d, 'hooks')
108 _project_hook_list = [os.path.join(d, x) for x in os.listdir(d)]
109 return _project_hook_list
110
111
Shawn O. Pearce632768b2008-10-23 11:58:52 -0700112class DownloadedChange(object):
113 _commit_cache = None
114
115 def __init__(self, project, base, change_id, ps_id, commit):
116 self.project = project
117 self.base = base
118 self.change_id = change_id
119 self.ps_id = ps_id
120 self.commit = commit
121
122 @property
123 def commits(self):
124 if self._commit_cache is None:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700125 self._commit_cache = self.project.bare_git.rev_list('--abbrev=8',
126 '--abbrev-commit',
127 '--pretty=oneline',
128 '--reverse',
129 '--date-order',
130 not_rev(self.base),
131 self.commit,
132 '--')
Shawn O. Pearce632768b2008-10-23 11:58:52 -0700133 return self._commit_cache
134
135
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700136class ReviewableBranch(object):
137 _commit_cache = None
138
139 def __init__(self, project, branch, base):
140 self.project = project
141 self.branch = branch
142 self.base = base
143
144 @property
145 def name(self):
146 return self.branch.name
147
148 @property
149 def commits(self):
150 if self._commit_cache is None:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700151 self._commit_cache = self.project.bare_git.rev_list('--abbrev=8',
152 '--abbrev-commit',
153 '--pretty=oneline',
154 '--reverse',
155 '--date-order',
156 not_rev(self.base),
157 R_HEADS + self.name,
158 '--')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700159 return self._commit_cache
160
161 @property
Shawn O. Pearcec99883f2008-11-11 17:12:43 -0800162 def unabbrev_commits(self):
163 r = dict()
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700164 for commit in self.project.bare_git.rev_list(not_rev(self.base),
165 R_HEADS + self.name,
166 '--'):
Shawn O. Pearcec99883f2008-11-11 17:12:43 -0800167 r[commit[0:8]] = commit
168 return r
169
170 @property
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700171 def date(self):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700172 return self.project.bare_git.log('--pretty=format:%cd',
173 '-n', '1',
174 R_HEADS + self.name,
175 '--')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700176
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700177 def UploadForReview(self, people,
178 auto_topic=False,
Jonathan Niederc94d6eb2017-08-08 18:34:53 +0000179 draft=False,
Changcheng Xiao87984c62017-08-02 16:55:03 +0200180 private=False,
181 wip=False,
Łukasz Gardońbed59ce2017-08-08 10:18:11 +0200182 dest_branch=None,
Masaya Suzuki305a2d02017-11-13 10:48:34 -0800183 validate_certs=True,
184 push_options=None):
Shawn O. Pearcec99883f2008-11-11 17:12:43 -0800185 self.project.UploadForReview(self.name,
Shawn O. Pearcea5ece0e2010-07-15 16:52:42 -0700186 people,
Brian Harring435370c2012-07-28 15:37:04 -0700187 auto_topic=auto_topic,
Jonathan Niederc94d6eb2017-08-08 18:34:53 +0000188 draft=draft,
Changcheng Xiao87984c62017-08-02 16:55:03 +0200189 private=private,
190 wip=wip,
Łukasz Gardońbed59ce2017-08-08 10:18:11 +0200191 dest_branch=dest_branch,
Masaya Suzuki305a2d02017-11-13 10:48:34 -0800192 validate_certs=validate_certs,
193 push_options=push_options)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700194
Ficus Kirkpatrickbc7ef672009-05-04 12:45:11 -0700195 def GetPublishedRefs(self):
196 refs = {}
197 output = self.project.bare_git.ls_remote(
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700198 self.branch.remote.SshReviewUrl(self.project.UserEmail),
199 'refs/changes/*')
Ficus Kirkpatrickbc7ef672009-05-04 12:45:11 -0700200 for line in output.split('\n'):
201 try:
202 (sha, ref) = line.split()
203 refs[sha] = ref
204 except ValueError:
205 pass
206
207 return refs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700208
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700209
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700210class StatusColoring(Coloring):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700211
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700212 def __init__(self, config):
213 Coloring.__init__(self, config, 'status')
Anthony King7bdac712014-07-16 12:56:40 +0100214 self.project = self.printer('header', attr='bold')
215 self.branch = self.printer('header', attr='bold')
216 self.nobranch = self.printer('nobranch', fg='red')
217 self.important = self.printer('important', fg='red')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700218
Anthony King7bdac712014-07-16 12:56:40 +0100219 self.added = self.printer('added', fg='green')
220 self.changed = self.printer('changed', fg='red')
221 self.untracked = self.printer('untracked', fg='red')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700222
223
224class DiffColoring(Coloring):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700225
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700226 def __init__(self, config):
227 Coloring.__init__(self, config, 'diff')
Anthony King7bdac712014-07-16 12:56:40 +0100228 self.project = self.printer('header', attr='bold')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700229
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700230
Anthony King7bdac712014-07-16 12:56:40 +0100231class _Annotation(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700232
James W. Mills24c13082012-04-12 15:04:13 -0500233 def __init__(self, name, value, keep):
234 self.name = name
235 self.value = value
236 self.keep = keep
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700237
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700238
Anthony King7bdac712014-07-16 12:56:40 +0100239class _CopyFile(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700240
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -0800241 def __init__(self, src, dest, abssrc, absdest):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700242 self.src = src
243 self.dest = dest
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -0800244 self.abs_src = abssrc
245 self.abs_dest = absdest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700246
247 def _Copy(self):
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -0800248 src = self.abs_src
249 dest = self.abs_dest
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700250 # copy file if it does not exist or is out of date
251 if not os.path.exists(dest) or not filecmp.cmp(src, dest):
252 try:
253 # remove existing file first, since it might be read-only
254 if os.path.exists(dest):
Renaud Paquay010fed72016-11-11 14:25:29 -0800255 platform_utils.remove(dest)
Matthew Buckett2daf6672009-07-11 09:43:47 -0400256 else:
Mickaël Salaün2f6ab7f2012-09-30 00:37:55 +0200257 dest_dir = os.path.dirname(dest)
258 if not os.path.isdir(dest_dir):
259 os.makedirs(dest_dir)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700260 shutil.copy(src, dest)
261 # make the file read-only
262 mode = os.stat(dest)[stat.ST_MODE]
263 mode = mode & ~(stat.S_IWUSR | stat.S_IWGRP | stat.S_IWOTH)
264 os.chmod(dest, mode)
265 except IOError:
Shawn O. Pearce48244782009-04-16 08:25:57 -0700266 _error('Cannot copy file %s to %s', src, dest)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700267
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700268
Anthony King7bdac712014-07-16 12:56:40 +0100269class _LinkFile(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700270
Wink Saville4c426ef2015-06-03 08:05:17 -0700271 def __init__(self, git_worktree, src, dest, relsrc, absdest):
272 self.git_worktree = git_worktree
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500273 self.src = src
274 self.dest = dest
Colin Cross0184dcc2015-05-05 00:24:54 -0700275 self.src_rel_to_dest = relsrc
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500276 self.abs_dest = absdest
277
Wink Saville4c426ef2015-06-03 08:05:17 -0700278 def __linkIt(self, relSrc, absDest):
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500279 # link file if it does not exist or is out of date
Renaud Paquay227ad2e2016-11-01 14:37:13 -0700280 if not platform_utils.islink(absDest) or (platform_utils.readlink(absDest) != relSrc):
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500281 try:
282 # remove existing file first, since it might be read-only
Dan Willemsene1e0bd12015-11-18 16:49:38 -0800283 if os.path.lexists(absDest):
Renaud Paquay010fed72016-11-11 14:25:29 -0800284 platform_utils.remove(absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500285 else:
Wink Saville4c426ef2015-06-03 08:05:17 -0700286 dest_dir = os.path.dirname(absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500287 if not os.path.isdir(dest_dir):
288 os.makedirs(dest_dir)
Renaud Paquayd5cec5e2016-11-01 11:24:03 -0700289 platform_utils.symlink(relSrc, absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500290 except IOError:
Wink Saville4c426ef2015-06-03 08:05:17 -0700291 _error('Cannot link file %s to %s', relSrc, absDest)
292
293 def _Link(self):
294 """Link the self.rel_src_to_dest and self.abs_dest. Handles wild cards
295 on the src linking all of the files in the source in to the destination
296 directory.
297 """
298 # We use the absSrc to handle the situation where the current directory
299 # is not the root of the repo
300 absSrc = os.path.join(self.git_worktree, self.src)
301 if os.path.exists(absSrc):
302 # Entity exists so just a simple one to one link operation
303 self.__linkIt(self.src_rel_to_dest, self.abs_dest)
304 else:
305 # Entity doesn't exist assume there is a wild card
306 absDestDir = self.abs_dest
307 if os.path.exists(absDestDir) and not os.path.isdir(absDestDir):
308 _error('Link error: src with wildcard, %s must be a directory',
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700309 absDestDir)
Wink Saville4c426ef2015-06-03 08:05:17 -0700310 else:
311 absSrcFiles = glob.glob(absSrc)
312 for absSrcFile in absSrcFiles:
313 # Create a releative path from source dir to destination dir
314 absSrcDir = os.path.dirname(absSrcFile)
315 relSrcDir = os.path.relpath(absSrcDir, absDestDir)
316
317 # Get the source file name
318 srcFile = os.path.basename(absSrcFile)
319
320 # Now form the final full paths to srcFile. They will be
321 # absolute for the desintaiton and relative for the srouce.
322 absDest = os.path.join(absDestDir, srcFile)
323 relSrc = os.path.join(relSrcDir, srcFile)
324 self.__linkIt(relSrc, absDest)
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500325
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700326
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700327class RemoteSpec(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700328
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700329 def __init__(self,
330 name,
Anthony King7bdac712014-07-16 12:56:40 +0100331 url=None,
Steve Raed6480452016-08-10 15:00:00 -0700332 pushUrl=None,
Anthony King7bdac712014-07-16 12:56:40 +0100333 review=None,
Dan Willemsen96c2d652016-04-06 16:03:54 -0700334 revision=None,
David Rileye0684ad2017-04-05 00:02:59 -0700335 orig_name=None,
336 fetchUrl=None):
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700337 self.name = name
338 self.url = url
Steve Raed6480452016-08-10 15:00:00 -0700339 self.pushUrl = pushUrl
Shawn O. Pearced1f70d92009-05-19 14:58:02 -0700340 self.review = review
Anthony King36ea2fb2014-05-06 11:54:01 +0100341 self.revision = revision
Dan Willemsen96c2d652016-04-06 16:03:54 -0700342 self.orig_name = orig_name
David Rileye0684ad2017-04-05 00:02:59 -0700343 self.fetchUrl = fetchUrl
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700344
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700345
Doug Anderson37282b42011-03-04 11:54:18 -0800346class RepoHook(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700347
Doug Anderson37282b42011-03-04 11:54:18 -0800348 """A RepoHook contains information about a script to run as a hook.
349
350 Hooks are used to run a python script before running an upload (for instance,
351 to run presubmit checks). Eventually, we may have hooks for other actions.
352
353 This shouldn't be confused with files in the 'repo/hooks' directory. Those
354 files are copied into each '.git/hooks' folder for each project. Repo-level
355 hooks are associated instead with repo actions.
356
357 Hooks are always python. When a hook is run, we will load the hook into the
358 interpreter and execute its main() function.
359 """
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700360
Doug Anderson37282b42011-03-04 11:54:18 -0800361 def __init__(self,
362 hook_type,
363 hooks_project,
364 topdir,
Mike Frysinger40252c22016-08-15 21:23:44 -0400365 manifest_url,
Doug Anderson37282b42011-03-04 11:54:18 -0800366 abort_if_user_denies=False):
367 """RepoHook constructor.
368
369 Params:
370 hook_type: A string representing the type of hook. This is also used
371 to figure out the name of the file containing the hook. For
372 example: 'pre-upload'.
373 hooks_project: The project containing the repo hooks. If you have a
374 manifest, this is manifest.repo_hooks_project. OK if this is None,
375 which will make the hook a no-op.
376 topdir: Repo's top directory (the one containing the .repo directory).
377 Scripts will run with CWD as this directory. If you have a manifest,
378 this is manifest.topdir
Mike Frysinger40252c22016-08-15 21:23:44 -0400379 manifest_url: The URL to the manifest git repo.
Doug Anderson37282b42011-03-04 11:54:18 -0800380 abort_if_user_denies: If True, we'll throw a HookError() if the user
381 doesn't allow us to run the hook.
382 """
383 self._hook_type = hook_type
384 self._hooks_project = hooks_project
Mike Frysinger40252c22016-08-15 21:23:44 -0400385 self._manifest_url = manifest_url
Doug Anderson37282b42011-03-04 11:54:18 -0800386 self._topdir = topdir
387 self._abort_if_user_denies = abort_if_user_denies
388
389 # Store the full path to the script for convenience.
390 if self._hooks_project:
391 self._script_fullpath = os.path.join(self._hooks_project.worktree,
392 self._hook_type + '.py')
393 else:
394 self._script_fullpath = None
395
396 def _GetHash(self):
397 """Return a hash of the contents of the hooks directory.
398
399 We'll just use git to do this. This hash has the property that if anything
400 changes in the directory we will return a different has.
401
402 SECURITY CONSIDERATION:
403 This hash only represents the contents of files in the hook directory, not
404 any other files imported or called by hooks. Changes to imported files
405 can change the script behavior without affecting the hash.
406
407 Returns:
408 A string representing the hash. This will always be ASCII so that it can
409 be printed to the user easily.
410 """
411 assert self._hooks_project, "Must have hooks to calculate their hash."
412
413 # We will use the work_git object rather than just calling GetRevisionId().
414 # That gives us a hash of the latest checked in version of the files that
415 # the user will actually be executing. Specifically, GetRevisionId()
416 # doesn't appear to change even if a user checks out a different version
417 # of the hooks repo (via git checkout) nor if a user commits their own revs.
418 #
419 # NOTE: Local (non-committed) changes will not be factored into this hash.
420 # I think this is OK, since we're really only worried about warning the user
421 # about upstream changes.
422 return self._hooks_project.work_git.rev_parse('HEAD')
423
424 def _GetMustVerb(self):
425 """Return 'must' if the hook is required; 'should' if not."""
426 if self._abort_if_user_denies:
427 return 'must'
428 else:
429 return 'should'
430
431 def _CheckForHookApproval(self):
432 """Check to see whether this hook has been approved.
433
Mike Frysinger40252c22016-08-15 21:23:44 -0400434 We'll accept approval of manifest URLs if they're using secure transports.
435 This way the user can say they trust the manifest hoster. For insecure
436 hosts, we fall back to checking the hash of the hooks repo.
Doug Anderson37282b42011-03-04 11:54:18 -0800437
438 Note that we ask permission for each individual hook even though we use
439 the hash of all hooks when detecting changes. We'd like the user to be
440 able to approve / deny each hook individually. We only use the hash of all
441 hooks because there is no other easy way to detect changes to local imports.
442
443 Returns:
444 True if this hook is approved to run; False otherwise.
445
446 Raises:
447 HookError: Raised if the user doesn't approve and abort_if_user_denies
448 was passed to the consturctor.
449 """
Mike Frysinger40252c22016-08-15 21:23:44 -0400450 if self._ManifestUrlHasSecureScheme():
451 return self._CheckForHookApprovalManifest()
452 else:
453 return self._CheckForHookApprovalHash()
454
455 def _CheckForHookApprovalHelper(self, subkey, new_val, main_prompt,
456 changed_prompt):
457 """Check for approval for a particular attribute and hook.
458
459 Args:
460 subkey: The git config key under [repo.hooks.<hook_type>] to store the
461 last approved string.
462 new_val: The new value to compare against the last approved one.
463 main_prompt: Message to display to the user to ask for approval.
464 changed_prompt: Message explaining why we're re-asking for approval.
465
466 Returns:
467 True if this hook is approved to run; False otherwise.
468
469 Raises:
470 HookError: Raised if the user doesn't approve and abort_if_user_denies
471 was passed to the consturctor.
472 """
Doug Anderson37282b42011-03-04 11:54:18 -0800473 hooks_config = self._hooks_project.config
Mike Frysinger40252c22016-08-15 21:23:44 -0400474 git_approval_key = 'repo.hooks.%s.%s' % (self._hook_type, subkey)
Doug Anderson37282b42011-03-04 11:54:18 -0800475
Mike Frysinger40252c22016-08-15 21:23:44 -0400476 # Get the last value that the user approved for this hook; may be None.
477 old_val = hooks_config.GetString(git_approval_key)
Doug Anderson37282b42011-03-04 11:54:18 -0800478
Mike Frysinger40252c22016-08-15 21:23:44 -0400479 if old_val is not None:
Doug Anderson37282b42011-03-04 11:54:18 -0800480 # User previously approved hook and asked not to be prompted again.
Mike Frysinger40252c22016-08-15 21:23:44 -0400481 if new_val == old_val:
Doug Anderson37282b42011-03-04 11:54:18 -0800482 # Approval matched. We're done.
483 return True
484 else:
485 # Give the user a reason why we're prompting, since they last told
486 # us to "never ask again".
Mike Frysinger40252c22016-08-15 21:23:44 -0400487 prompt = 'WARNING: %s\n\n' % (changed_prompt,)
Doug Anderson37282b42011-03-04 11:54:18 -0800488 else:
489 prompt = ''
490
491 # Prompt the user if we're not on a tty; on a tty we'll assume "no".
492 if sys.stdout.isatty():
Mike Frysinger40252c22016-08-15 21:23:44 -0400493 prompt += main_prompt + ' (yes/always/NO)? '
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530494 response = input(prompt).lower()
David Pursehouse98ffba12012-11-14 11:18:00 +0900495 print()
Doug Anderson37282b42011-03-04 11:54:18 -0800496
497 # User is doing a one-time approval.
498 if response in ('y', 'yes'):
499 return True
Mike Frysinger40252c22016-08-15 21:23:44 -0400500 elif response == 'always':
501 hooks_config.SetString(git_approval_key, new_val)
Doug Anderson37282b42011-03-04 11:54:18 -0800502 return True
503
504 # For anything else, we'll assume no approval.
505 if self._abort_if_user_denies:
506 raise HookError('You must allow the %s hook or use --no-verify.' %
507 self._hook_type)
508
509 return False
510
Mike Frysinger40252c22016-08-15 21:23:44 -0400511 def _ManifestUrlHasSecureScheme(self):
512 """Check if the URI for the manifest is a secure transport."""
513 secure_schemes = ('file', 'https', 'ssh', 'persistent-https', 'sso', 'rpc')
514 parse_results = urllib.parse.urlparse(self._manifest_url)
515 return parse_results.scheme in secure_schemes
516
517 def _CheckForHookApprovalManifest(self):
518 """Check whether the user has approved this manifest host.
519
520 Returns:
521 True if this hook is approved to run; False otherwise.
522 """
523 return self._CheckForHookApprovalHelper(
524 'approvedmanifest',
525 self._manifest_url,
526 'Run hook scripts from %s' % (self._manifest_url,),
527 'Manifest URL has changed since %s was allowed.' % (self._hook_type,))
528
529 def _CheckForHookApprovalHash(self):
530 """Check whether the user has approved the hooks repo.
531
532 Returns:
533 True if this hook is approved to run; False otherwise.
534 """
535 prompt = ('Repo %s run the script:\n'
536 ' %s\n'
537 '\n'
Jonathan Nieder71e4cea2016-08-16 12:05:09 -0700538 'Do you want to allow this script to run')
Mike Frysinger40252c22016-08-15 21:23:44 -0400539 return self._CheckForHookApprovalHelper(
540 'approvedhash',
541 self._GetHash(),
Jonathan Nieder71e4cea2016-08-16 12:05:09 -0700542 prompt % (self._GetMustVerb(), self._script_fullpath),
Mike Frysinger40252c22016-08-15 21:23:44 -0400543 'Scripts have changed since %s was allowed.' % (self._hook_type,))
544
Doug Anderson37282b42011-03-04 11:54:18 -0800545 def _ExecuteHook(self, **kwargs):
546 """Actually execute the given hook.
547
548 This will run the hook's 'main' function in our python interpreter.
549
550 Args:
551 kwargs: Keyword arguments to pass to the hook. These are often specific
552 to the hook type. For instance, pre-upload hooks will contain
553 a project_list.
554 """
555 # Keep sys.path and CWD stashed away so that we can always restore them
556 # upon function exit.
557 orig_path = os.getcwd()
558 orig_syspath = sys.path
559
560 try:
561 # Always run hooks with CWD as topdir.
562 os.chdir(self._topdir)
563
564 # Put the hook dir as the first item of sys.path so hooks can do
565 # relative imports. We want to replace the repo dir as [0] so
566 # hooks can't import repo files.
567 sys.path = [os.path.dirname(self._script_fullpath)] + sys.path[1:]
568
569 # Exec, storing global context in the context dict. We catch exceptions
570 # and convert to a HookError w/ just the failing traceback.
Mike Frysinger4aa4b212016-03-04 15:03:00 -0500571 context = {'__file__': self._script_fullpath}
Doug Anderson37282b42011-03-04 11:54:18 -0800572 try:
Anthony King70f68902014-05-05 21:15:34 +0100573 exec(compile(open(self._script_fullpath).read(),
574 self._script_fullpath, 'exec'), context)
Doug Anderson37282b42011-03-04 11:54:18 -0800575 except Exception:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700576 raise HookError('%s\nFailed to import %s hook; see traceback above.' %
577 (traceback.format_exc(), self._hook_type))
Doug Anderson37282b42011-03-04 11:54:18 -0800578
579 # Running the script should have defined a main() function.
580 if 'main' not in context:
581 raise HookError('Missing main() in: "%s"' % self._script_fullpath)
582
Doug Anderson37282b42011-03-04 11:54:18 -0800583 # Add 'hook_should_take_kwargs' to the arguments to be passed to main.
584 # We don't actually want hooks to define their main with this argument--
585 # it's there to remind them that their hook should always take **kwargs.
586 # For instance, a pre-upload hook should be defined like:
587 # def main(project_list, **kwargs):
588 #
589 # This allows us to later expand the API without breaking old hooks.
590 kwargs = kwargs.copy()
591 kwargs['hook_should_take_kwargs'] = True
592
593 # Call the main function in the hook. If the hook should cause the
594 # build to fail, it will raise an Exception. We'll catch that convert
595 # to a HookError w/ just the failing traceback.
596 try:
597 context['main'](**kwargs)
598 except Exception:
599 raise HookError('%s\nFailed to run main() for %s hook; see traceback '
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700600 'above.' % (traceback.format_exc(),
601 self._hook_type))
Doug Anderson37282b42011-03-04 11:54:18 -0800602 finally:
603 # Restore sys.path and CWD.
604 sys.path = orig_syspath
605 os.chdir(orig_path)
606
607 def Run(self, user_allows_all_hooks, **kwargs):
608 """Run the hook.
609
610 If the hook doesn't exist (because there is no hooks project or because
611 this particular hook is not enabled), this is a no-op.
612
613 Args:
614 user_allows_all_hooks: If True, we will never prompt about running the
615 hook--we'll just assume it's OK to run it.
616 kwargs: Keyword arguments to pass to the hook. These are often specific
617 to the hook type. For instance, pre-upload hooks will contain
618 a project_list.
619
620 Raises:
621 HookError: If there was a problem finding the hook or the user declined
622 to run a required hook (from _CheckForHookApproval).
623 """
624 # No-op if there is no hooks project or if hook is disabled.
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700625 if ((not self._hooks_project) or (self._hook_type not in
626 self._hooks_project.enabled_repo_hooks)):
Doug Anderson37282b42011-03-04 11:54:18 -0800627 return
628
629 # Bail with a nice error if we can't find the hook.
630 if not os.path.isfile(self._script_fullpath):
631 raise HookError('Couldn\'t find repo hook: "%s"' % self._script_fullpath)
632
633 # Make sure the user is OK with running the hook.
634 if (not user_allows_all_hooks) and (not self._CheckForHookApproval()):
635 return
636
637 # Run the hook with the same version of python we're using.
638 self._ExecuteHook(**kwargs)
639
640
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700641class Project(object):
Kevin Degi384b3c52014-10-16 16:02:58 -0600642 # These objects can be shared between several working trees.
643 shareable_files = ['description', 'info']
644 shareable_dirs = ['hooks', 'objects', 'rr-cache', 'svn']
645 # These objects can only be used by a single working tree.
646 working_tree_files = ['config', 'packed-refs', 'shallow']
647 working_tree_dirs = ['logs', 'refs']
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700648
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700649 def __init__(self,
650 manifest,
651 name,
652 remote,
653 gitdir,
David James8d201162013-10-11 17:03:19 -0700654 objdir,
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700655 worktree,
656 relpath,
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -0700657 revisionExpr,
Mike Pontillod3153822012-02-28 11:53:24 -0800658 revisionId,
Anthony King7bdac712014-07-16 12:56:40 +0100659 rebase=True,
660 groups=None,
661 sync_c=False,
662 sync_s=False,
663 clone_depth=None,
664 upstream=None,
665 parent=None,
666 is_derived=False,
David Pursehouseb1553542014-09-04 21:28:09 +0900667 dest_branch=None,
Simran Basib9a1b732015-08-20 12:19:28 -0700668 optimized_fetch=False,
669 old_revision=None):
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800670 """Init a Project object.
671
672 Args:
673 manifest: The XmlManifest object.
674 name: The `name` attribute of manifest.xml's project element.
675 remote: RemoteSpec object specifying its remote's properties.
676 gitdir: Absolute path of git directory.
David James8d201162013-10-11 17:03:19 -0700677 objdir: Absolute path of directory to store git objects.
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800678 worktree: Absolute path of git working tree.
679 relpath: Relative path of git working tree to repo's top directory.
680 revisionExpr: The `revision` attribute of manifest.xml's project element.
681 revisionId: git commit id for checking out.
682 rebase: The `rebase` attribute of manifest.xml's project element.
683 groups: The `groups` attribute of manifest.xml's project element.
684 sync_c: The `sync-c` attribute of manifest.xml's project element.
685 sync_s: The `sync-s` attribute of manifest.xml's project element.
686 upstream: The `upstream` attribute of manifest.xml's project element.
687 parent: The parent Project object.
688 is_derived: False if the project was explicitly defined in the manifest;
689 True if the project is a discovered submodule.
Bryan Jacobsf609f912013-05-06 13:36:24 -0400690 dest_branch: The branch to which to push changes for review by default.
David Pursehouseb1553542014-09-04 21:28:09 +0900691 optimized_fetch: If True, when a project is set to a sha1 revision, only
692 fetch from the remote if the sha1 is not present locally.
Simran Basib9a1b732015-08-20 12:19:28 -0700693 old_revision: saved git commit id for open GITC projects.
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800694 """
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700695 self.manifest = manifest
696 self.name = name
697 self.remote = remote
Anthony Newnamdf14a702011-01-09 17:31:57 -0800698 self.gitdir = gitdir.replace('\\', '/')
David James8d201162013-10-11 17:03:19 -0700699 self.objdir = objdir.replace('\\', '/')
Shawn O. Pearce0ce6ca92011-01-10 13:26:01 -0800700 if worktree:
Renaud Paquayfef9f212016-11-01 18:28:01 -0700701 self.worktree = os.path.normpath(worktree).replace('\\', '/')
Shawn O. Pearce0ce6ca92011-01-10 13:26:01 -0800702 else:
703 self.worktree = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700704 self.relpath = relpath
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -0700705 self.revisionExpr = revisionExpr
706
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700707 if revisionId is None \
708 and revisionExpr \
709 and IsId(revisionExpr):
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -0700710 self.revisionId = revisionExpr
711 else:
712 self.revisionId = revisionId
713
Mike Pontillod3153822012-02-28 11:53:24 -0800714 self.rebase = rebase
Colin Cross5acde752012-03-28 20:15:45 -0700715 self.groups = groups
Anatol Pomazau79770d22012-04-20 14:41:59 -0700716 self.sync_c = sync_c
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800717 self.sync_s = sync_s
David Pursehouseede7f122012-11-27 22:25:30 +0900718 self.clone_depth = clone_depth
Brian Harring14a66742012-09-28 20:21:57 -0700719 self.upstream = upstream
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800720 self.parent = parent
721 self.is_derived = is_derived
David Pursehouseb1553542014-09-04 21:28:09 +0900722 self.optimized_fetch = optimized_fetch
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800723 self.subprojects = []
Mike Pontillod3153822012-02-28 11:53:24 -0800724
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700725 self.snapshots = {}
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700726 self.copyfiles = []
Jeff Hamiltone0df2322014-04-21 17:10:59 -0500727 self.linkfiles = []
James W. Mills24c13082012-04-12 15:04:13 -0500728 self.annotations = []
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700729 self.config = GitConfig.ForRepository(gitdir=self.gitdir,
730 defaults=self.manifest.globalConfig)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700731
Shawn O. Pearcee284ad12008-11-04 07:37:10 -0800732 if self.worktree:
David James8d201162013-10-11 17:03:19 -0700733 self.work_git = self._GitGetByExec(self, bare=False, gitdir=gitdir)
Shawn O. Pearcee284ad12008-11-04 07:37:10 -0800734 else:
735 self.work_git = None
David James8d201162013-10-11 17:03:19 -0700736 self.bare_git = self._GitGetByExec(self, bare=True, gitdir=gitdir)
Shawn O. Pearced237b692009-04-17 18:49:50 -0700737 self.bare_ref = GitRefs(gitdir)
David James8d201162013-10-11 17:03:19 -0700738 self.bare_objdir = self._GitGetByExec(self, bare=True, gitdir=objdir)
Bryan Jacobsf609f912013-05-06 13:36:24 -0400739 self.dest_branch = dest_branch
Simran Basib9a1b732015-08-20 12:19:28 -0700740 self.old_revision = old_revision
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700741
Doug Anderson37282b42011-03-04 11:54:18 -0800742 # This will be filled in if a project is later identified to be the
743 # project containing repo hooks.
744 self.enabled_repo_hooks = []
745
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700746 @property
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +0800747 def Derived(self):
748 return self.is_derived
749
750 @property
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700751 def Exists(self):
Kevin Degi384b3c52014-10-16 16:02:58 -0600752 return os.path.isdir(self.gitdir) and os.path.isdir(self.objdir)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700753
754 @property
755 def CurrentBranch(self):
756 """Obtain the name of the currently checked out branch.
757 The branch name omits the 'refs/heads/' prefix.
758 None is returned if the project is on a detached HEAD.
759 """
Shawn O. Pearce5b23f242009-04-17 18:43:33 -0700760 b = self.work_git.GetHead()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700761 if b.startswith(R_HEADS):
762 return b[len(R_HEADS):]
763 return None
764
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -0700765 def IsRebaseInProgress(self):
766 w = self.worktree
767 g = os.path.join(w, '.git')
768 return os.path.exists(os.path.join(g, 'rebase-apply')) \
769 or os.path.exists(os.path.join(g, 'rebase-merge')) \
770 or os.path.exists(os.path.join(w, '.dotest'))
Julius Gustavsson0cb1b3f2010-06-17 17:55:02 +0200771
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700772 def IsDirty(self, consider_untracked=True):
773 """Is the working directory modified in some way?
774 """
775 self.work_git.update_index('-q',
776 '--unmerged',
777 '--ignore-missing',
778 '--refresh')
David Pursehouse8f62fb72012-11-14 12:09:38 +0900779 if self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700780 return True
781 if self.work_git.DiffZ('diff-files'):
782 return True
783 if consider_untracked and self.work_git.LsOthers():
784 return True
785 return False
786
787 _userident_name = None
788 _userident_email = None
789
790 @property
791 def UserName(self):
792 """Obtain the user's personal name.
793 """
794 if self._userident_name is None:
795 self._LoadUserIdentity()
796 return self._userident_name
797
798 @property
799 def UserEmail(self):
800 """Obtain the user's email address. This is very likely
801 to be their Gerrit login.
802 """
803 if self._userident_email is None:
804 self._LoadUserIdentity()
805 return self._userident_email
806
807 def _LoadUserIdentity(self):
David Pursehousec1b86a22012-11-14 11:36:51 +0900808 u = self.bare_git.var('GIT_COMMITTER_IDENT')
809 m = re.compile("^(.*) <([^>]*)> ").match(u)
810 if m:
811 self._userident_name = m.group(1)
812 self._userident_email = m.group(2)
813 else:
814 self._userident_name = ''
815 self._userident_email = ''
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700816
817 def GetRemote(self, name):
818 """Get the configuration for a single remote.
819 """
820 return self.config.GetRemote(name)
821
822 def GetBranch(self, name):
823 """Get the configuration for a single branch.
824 """
825 return self.config.GetBranch(name)
826
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700827 def GetBranches(self):
828 """Get all existing local branches.
829 """
830 current = self.CurrentBranch
David Pursehouse8a68ff92012-09-24 12:15:13 +0900831 all_refs = self._allrefs
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700832 heads = {}
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700833
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530834 for name, ref_id in all_refs.items():
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700835 if name.startswith(R_HEADS):
836 name = name[len(R_HEADS):]
837 b = self.GetBranch(name)
838 b.current = name == current
839 b.published = None
David Pursehouse8a68ff92012-09-24 12:15:13 +0900840 b.revision = ref_id
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700841 heads[name] = b
842
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530843 for name, ref_id in all_refs.items():
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700844 if name.startswith(R_PUB):
845 name = name[len(R_PUB):]
846 b = heads.get(name)
847 if b:
David Pursehouse8a68ff92012-09-24 12:15:13 +0900848 b.published = ref_id
Shawn O. Pearce27b07322009-04-10 16:02:48 -0700849
850 return heads
851
Colin Cross5acde752012-03-28 20:15:45 -0700852 def MatchesGroups(self, manifest_groups):
853 """Returns true if the manifest groups specified at init should cause
854 this project to be synced.
855 Prefixing a manifest group with "-" inverts the meaning of a group.
Conley Owensbb1b5f52012-08-13 13:11:18 -0700856 All projects are implicitly labelled with "all".
Conley Owens971de8e2012-04-16 10:36:08 -0700857
858 labels are resolved in order. In the example case of
Conley Owensbb1b5f52012-08-13 13:11:18 -0700859 project_groups: "all,group1,group2"
Conley Owens971de8e2012-04-16 10:36:08 -0700860 manifest_groups: "-group1,group2"
861 the project will be matched.
David Holmer0a1c6a12012-11-14 19:19:00 -0500862
863 The special manifest group "default" will match any project that
864 does not have the special project group "notdefault"
Colin Cross5acde752012-03-28 20:15:45 -0700865 """
David Holmer0a1c6a12012-11-14 19:19:00 -0500866 expanded_manifest_groups = manifest_groups or ['default']
Conley Owensbb1b5f52012-08-13 13:11:18 -0700867 expanded_project_groups = ['all'] + (self.groups or [])
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700868 if 'notdefault' not in expanded_project_groups:
David Holmer0a1c6a12012-11-14 19:19:00 -0500869 expanded_project_groups += ['default']
Conley Owensbb1b5f52012-08-13 13:11:18 -0700870
Conley Owens971de8e2012-04-16 10:36:08 -0700871 matched = False
Conley Owensbb1b5f52012-08-13 13:11:18 -0700872 for group in expanded_manifest_groups:
873 if group.startswith('-') and group[1:] in expanded_project_groups:
Conley Owens971de8e2012-04-16 10:36:08 -0700874 matched = False
Conley Owensbb1b5f52012-08-13 13:11:18 -0700875 elif group in expanded_project_groups:
Conley Owens971de8e2012-04-16 10:36:08 -0700876 matched = True
Colin Cross5acde752012-03-28 20:15:45 -0700877
Conley Owens971de8e2012-04-16 10:36:08 -0700878 return matched
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700879
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700880# Status Display ##
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700881 def UncommitedFiles(self, get_all=True):
882 """Returns a list of strings, uncommitted files in the git tree.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700883
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700884 Args:
885 get_all: a boolean, if True - get information about all different
886 uncommitted files. If False - return as soon as any kind of
887 uncommitted files is detected.
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500888 """
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700889 details = []
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500890 self.work_git.update_index('-q',
891 '--unmerged',
892 '--ignore-missing',
893 '--refresh')
894 if self.IsRebaseInProgress():
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700895 details.append("rebase in progress")
896 if not get_all:
897 return details
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500898
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700899 changes = self.work_git.DiffZ('diff-index', '--cached', HEAD).keys()
900 if changes:
901 details.extend(changes)
902 if not get_all:
903 return details
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500904
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700905 changes = self.work_git.DiffZ('diff-files').keys()
906 if changes:
907 details.extend(changes)
908 if not get_all:
909 return details
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500910
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700911 changes = self.work_git.LsOthers()
912 if changes:
913 details.extend(changes)
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500914
Vadim Bendebury14e134d2014-10-05 15:40:30 -0700915 return details
916
917 def HasChanges(self):
918 """Returns true if there are uncommitted changes.
919 """
920 if self.UncommitedFiles(get_all=False):
921 return True
922 else:
923 return False
Anthony Newnamcc50bac2010-04-08 10:28:59 -0500924
Andrew Wheeler4d5bb682012-02-27 13:52:22 -0600925 def PrintWorkTreeStatus(self, output_redir=None, quiet=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700926 """Prints the status of the repository to stdout.
Terence Haddock4655e812011-03-31 12:33:34 +0200927
928 Args:
929 output: If specified, redirect the output to this object.
Andrew Wheeler4d5bb682012-02-27 13:52:22 -0600930 quiet: If True then only print the project name. Do not print
931 the modified files, branch name, etc.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700932 """
933 if not os.path.isdir(self.worktree):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700934 if output_redir is None:
Terence Haddock4655e812011-03-31 12:33:34 +0200935 output_redir = sys.stdout
Sarah Owenscecd1d82012-11-01 22:59:27 -0700936 print(file=output_redir)
937 print('project %s/' % self.relpath, file=output_redir)
938 print(' missing (run "repo sync")', file=output_redir)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700939 return
940
941 self.work_git.update_index('-q',
942 '--unmerged',
943 '--ignore-missing',
944 '--refresh')
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -0700945 rb = self.IsRebaseInProgress()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700946 di = self.work_git.DiffZ('diff-index', '-M', '--cached', HEAD)
947 df = self.work_git.DiffZ('diff-files')
948 do = self.work_git.LsOthers()
Ali Utku Selen76abcc12012-01-25 10:51:12 +0100949 if not rb and not di and not df and not do and not self.CurrentBranch:
Shawn O. Pearce161f4452009-04-10 17:41:44 -0700950 return 'CLEAN'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700951
952 out = StatusColoring(self.config)
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -0700953 if output_redir is not None:
Terence Haddock4655e812011-03-31 12:33:34 +0200954 out.redirect(output_redir)
Jakub Vrana0402cd82014-09-09 15:39:15 -0700955 out.project('project %-40s', self.relpath + '/ ')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700956
Andrew Wheeler4d5bb682012-02-27 13:52:22 -0600957 if quiet:
958 out.nl()
959 return 'DIRTY'
960
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700961 branch = self.CurrentBranch
962 if branch is None:
963 out.nobranch('(*** NO BRANCH ***)')
964 else:
965 out.branch('branch %s', branch)
966 out.nl()
967
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -0700968 if rb:
969 out.important('prior sync failed; rebase still in progress')
970 out.nl()
971
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700972 paths = list()
973 paths.extend(di.keys())
974 paths.extend(df.keys())
975 paths.extend(do)
976
Chirayu Desai217ea7d2013-03-01 19:14:38 +0530977 for p in sorted(set(paths)):
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900978 try:
979 i = di[p]
980 except KeyError:
981 i = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700982
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900983 try:
984 f = df[p]
985 except KeyError:
986 f = None
Julius Gustavsson0cb1b3f2010-06-17 17:55:02 +0200987
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900988 if i:
989 i_status = i.status.upper()
990 else:
991 i_status = '-'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700992
David Pursehouse5c6eeac2012-10-11 16:44:48 +0900993 if f:
994 f_status = f.status.lower()
995 else:
996 f_status = '-'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -0700997
998 if i and i.src_path:
Shawn O. Pearcefe086752009-03-03 13:49:48 -0800999 line = ' %s%s\t%s => %s (%s%%)' % (i_status, f_status,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001000 i.src_path, p, i.level)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001001 else:
1002 line = ' %s%s\t%s' % (i_status, f_status, p)
1003
1004 if i and not f:
1005 out.added('%s', line)
1006 elif (i and f) or (not i and f):
1007 out.changed('%s', line)
1008 elif not i and not f:
1009 out.untracked('%s', line)
1010 else:
1011 out.write('%s', line)
1012 out.nl()
Terence Haddock4655e812011-03-31 12:33:34 +02001013
Shawn O. Pearce161f4452009-04-10 17:41:44 -07001014 return 'DIRTY'
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001015
pelyad67872d2012-03-28 14:49:58 +03001016 def PrintWorkTreeDiff(self, absolute_paths=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001017 """Prints the status of the repository to stdout.
1018 """
1019 out = DiffColoring(self.config)
1020 cmd = ['diff']
1021 if out.is_on:
1022 cmd.append('--color')
1023 cmd.append(HEAD)
pelyad67872d2012-03-28 14:49:58 +03001024 if absolute_paths:
1025 cmd.append('--src-prefix=a/%s/' % self.relpath)
1026 cmd.append('--dst-prefix=b/%s/' % self.relpath)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001027 cmd.append('--')
1028 p = GitCommand(self,
1029 cmd,
Anthony King7bdac712014-07-16 12:56:40 +01001030 capture_stdout=True,
1031 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001032 has_diff = False
1033 for line in p.process.stdout:
1034 if not has_diff:
1035 out.nl()
1036 out.project('project %s/' % self.relpath)
1037 out.nl()
1038 has_diff = True
Sarah Owenscecd1d82012-11-01 22:59:27 -07001039 print(line[:-1])
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001040 p.Wait()
1041
1042
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001043# Publish / Upload ##
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001044
David Pursehouse8a68ff92012-09-24 12:15:13 +09001045 def WasPublished(self, branch, all_refs=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001046 """Was the branch published (uploaded) for code review?
1047 If so, returns the SHA-1 hash of the last published
1048 state for the branch.
1049 """
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001050 key = R_PUB + branch
David Pursehouse8a68ff92012-09-24 12:15:13 +09001051 if all_refs is None:
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001052 try:
1053 return self.bare_git.rev_parse(key)
1054 except GitError:
1055 return None
1056 else:
1057 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001058 return all_refs[key]
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001059 except KeyError:
1060 return None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001061
David Pursehouse8a68ff92012-09-24 12:15:13 +09001062 def CleanPublishedCache(self, all_refs=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001063 """Prunes any stale published refs.
1064 """
David Pursehouse8a68ff92012-09-24 12:15:13 +09001065 if all_refs is None:
1066 all_refs = self._allrefs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001067 heads = set()
1068 canrm = {}
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301069 for name, ref_id in all_refs.items():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001070 if name.startswith(R_HEADS):
1071 heads.add(name)
1072 elif name.startswith(R_PUB):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001073 canrm[name] = ref_id
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001074
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301075 for name, ref_id in canrm.items():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001076 n = name[len(R_PUB):]
1077 if R_HEADS + n not in heads:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001078 self.bare_git.DeleteRef(name, ref_id)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001079
Mandeep Singh Bainesd6c93a22011-05-26 10:34:11 -07001080 def GetUploadableBranches(self, selected_branch=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001081 """List any branches which can be uploaded for review.
1082 """
1083 heads = {}
1084 pubed = {}
1085
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301086 for name, ref_id in self._allrefs.items():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001087 if name.startswith(R_HEADS):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001088 heads[name[len(R_HEADS):]] = ref_id
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001089 elif name.startswith(R_PUB):
David Pursehouse8a68ff92012-09-24 12:15:13 +09001090 pubed[name[len(R_PUB):]] = ref_id
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001091
1092 ready = []
Chirayu Desai217ea7d2013-03-01 19:14:38 +05301093 for branch, ref_id in heads.items():
David Pursehouse8a68ff92012-09-24 12:15:13 +09001094 if branch in pubed and pubed[branch] == ref_id:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001095 continue
Mandeep Singh Bainesd6c93a22011-05-26 10:34:11 -07001096 if selected_branch and branch != selected_branch:
1097 continue
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001098
Shawn O. Pearce35f25962008-11-11 17:03:13 -08001099 rb = self.GetUploadableBranch(branch)
1100 if rb:
1101 ready.append(rb)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001102 return ready
1103
Shawn O. Pearce35f25962008-11-11 17:03:13 -08001104 def GetUploadableBranch(self, branch_name):
1105 """Get a single uploadable branch, or None.
1106 """
1107 branch = self.GetBranch(branch_name)
1108 base = branch.LocalMerge
1109 if branch.LocalMerge:
1110 rb = ReviewableBranch(self, branch, base)
1111 if rb.commits:
1112 return rb
1113 return None
1114
Shawn O. Pearcea5ece0e2010-07-15 16:52:42 -07001115 def UploadForReview(self, branch=None,
Anthony King7bdac712014-07-16 12:56:40 +01001116 people=([], []),
Brian Harring435370c2012-07-28 15:37:04 -07001117 auto_topic=False,
Jonathan Niederc94d6eb2017-08-08 18:34:53 +00001118 draft=False,
Changcheng Xiao87984c62017-08-02 16:55:03 +02001119 private=False,
1120 wip=False,
Łukasz Gardońbed59ce2017-08-08 10:18:11 +02001121 dest_branch=None,
Masaya Suzuki305a2d02017-11-13 10:48:34 -08001122 validate_certs=True,
1123 push_options=None):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001124 """Uploads the named branch for code review.
1125 """
1126 if branch is None:
1127 branch = self.CurrentBranch
1128 if branch is None:
1129 raise GitError('not currently on a branch')
1130
1131 branch = self.GetBranch(branch)
1132 if not branch.LocalMerge:
1133 raise GitError('branch %s does not track a remote' % branch.name)
1134 if not branch.remote.review:
1135 raise GitError('remote %s has no review url' % branch.remote.name)
1136
Bryan Jacobsf609f912013-05-06 13:36:24 -04001137 if dest_branch is None:
1138 dest_branch = self.dest_branch
1139 if dest_branch is None:
1140 dest_branch = branch.merge
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001141 if not dest_branch.startswith(R_HEADS):
1142 dest_branch = R_HEADS + dest_branch
1143
Shawn O. Pearce339ba9f2008-11-06 09:52:51 -08001144 if not branch.remote.projectname:
1145 branch.remote.projectname = self.name
1146 branch.remote.Save()
1147
Łukasz Gardońbed59ce2017-08-08 10:18:11 +02001148 url = branch.remote.ReviewUrl(self.UserEmail, validate_certs)
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001149 if url is None:
1150 raise UploadError('review not configured')
1151 cmd = ['push']
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001152
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001153 if url.startswith('ssh://'):
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001154 rp = ['gerrit receive-pack']
1155 for e in people[0]:
1156 rp.append('--reviewer=%s' % sq(e))
1157 for e in people[1]:
1158 rp.append('--cc=%s' % sq(e))
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001159 cmd.append('--receive-pack=%s' % " ".join(rp))
Shawn O. Pearcea5ece0e2010-07-15 16:52:42 -07001160
Masaya Suzuki305a2d02017-11-13 10:48:34 -08001161 for push_option in (push_options or []):
1162 cmd.append('-o')
1163 cmd.append(push_option)
1164
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001165 cmd.append(url)
Shawn O. Pearceb54a3922009-01-05 16:18:58 -08001166
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001167 if dest_branch.startswith(R_HEADS):
1168 dest_branch = dest_branch[len(R_HEADS):]
Brian Harring435370c2012-07-28 15:37:04 -07001169
Jonathan Niederc94d6eb2017-08-08 18:34:53 +00001170 upload_type = 'for'
1171 if draft:
1172 upload_type = 'drafts'
1173
1174 ref_spec = '%s:refs/%s/%s' % (R_HEADS + branch.name, upload_type,
1175 dest_branch)
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001176 if auto_topic:
1177 ref_spec = ref_spec + '/' + branch.name
Changcheng Xiao87984c62017-08-02 16:55:03 +02001178
Shawn Pearce45d21682013-02-28 00:35:51 -08001179 if not url.startswith('ssh://'):
1180 rp = ['r=%s' % p for p in people[0]] + \
1181 ['cc=%s' % p for p in people[1]]
Changcheng Xiao87984c62017-08-02 16:55:03 +02001182 if private:
1183 rp = rp + ['private']
1184 if wip:
1185 rp = rp + ['wip']
Shawn Pearce45d21682013-02-28 00:35:51 -08001186 if rp:
1187 ref_spec = ref_spec + '%' + ','.join(rp)
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001188 cmd.append(ref_spec)
1189
Anthony King7bdac712014-07-16 12:56:40 +01001190 if GitCommand(self, cmd, bare=True).Wait() != 0:
Shawn O. Pearcec9571422012-01-11 14:58:54 -08001191 raise UploadError('Upload failed')
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001192
1193 msg = "posted to %s for %s" % (branch.remote.review, dest_branch)
1194 self.bare_git.UpdateRef(R_PUB + branch.name,
1195 R_HEADS + branch.name,
Anthony King7bdac712014-07-16 12:56:40 +01001196 message=msg)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001197
1198
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001199# Sync ##
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001200
Julien Campergue335f5ef2013-10-16 11:02:35 +02001201 def _ExtractArchive(self, tarpath, path=None):
1202 """Extract the given tar on its current location
1203
1204 Args:
1205 - tarpath: The path to the actual tar file
1206
1207 """
1208 try:
1209 with tarfile.open(tarpath, 'r') as tar:
1210 tar.extractall(path=path)
1211 return True
1212 except (IOError, tarfile.TarError) as e:
David Pursehousef33929d2015-08-24 14:39:14 +09001213 _error("Cannot extract archive %s: %s", tarpath, str(e))
Julien Campergue335f5ef2013-10-16 11:02:35 +02001214 return False
1215
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -07001216 def Sync_NetworkHalf(self,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001217 quiet=False,
1218 is_new=None,
1219 current_branch_only=False,
1220 force_sync=False,
1221 clone_bundle=True,
1222 no_tags=False,
1223 archive=False,
1224 optimized_fetch=False,
Martin Kellye4e94d22017-03-21 16:05:12 -07001225 prune=False,
1226 submodules=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001227 """Perform only the network IO portion of the sync process.
1228 Local working directory/branch state is not affected.
1229 """
Julien Campergue335f5ef2013-10-16 11:02:35 +02001230 if archive and not isinstance(self, MetaProject):
1231 if self.remote.url.startswith(('http://', 'https://')):
David Pursehousef33929d2015-08-24 14:39:14 +09001232 _error("%s: Cannot fetch archives from http/https remotes.", self.name)
Julien Campergue335f5ef2013-10-16 11:02:35 +02001233 return False
1234
1235 name = self.relpath.replace('\\', '/')
1236 name = name.replace('/', '_')
1237 tarpath = '%s.tar' % name
1238 topdir = self.manifest.topdir
1239
1240 try:
1241 self._FetchArchive(tarpath, cwd=topdir)
1242 except GitError as e:
David Pursehousef33929d2015-08-24 14:39:14 +09001243 _error('%s', e)
Julien Campergue335f5ef2013-10-16 11:02:35 +02001244 return False
1245
1246 # From now on, we only need absolute tarpath
1247 tarpath = os.path.join(topdir, tarpath)
1248
1249 if not self._ExtractArchive(tarpath, path=topdir):
1250 return False
1251 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001252 platform_utils.remove(tarpath)
Julien Campergue335f5ef2013-10-16 11:02:35 +02001253 except OSError as e:
David Pursehousef33929d2015-08-24 14:39:14 +09001254 _warn("Cannot remove archive %s: %s", tarpath, str(e))
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001255 self._CopyAndLinkFiles()
Julien Campergue335f5ef2013-10-16 11:02:35 +02001256 return True
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07001257 if is_new is None:
1258 is_new = not self.Exists
Shawn O. Pearce88443382010-10-08 10:02:09 +02001259 if is_new:
Kevin Degiabaa7f32014-11-12 11:27:45 -07001260 self._InitGitDir(force_sync=force_sync)
Jimmie Westera0444582012-10-24 13:44:42 +02001261 else:
1262 self._UpdateHooks()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001263 self._InitRemote()
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001264
1265 if is_new:
1266 alt = os.path.join(self.gitdir, 'objects/info/alternates')
1267 try:
Renaud Paquay2a4be942016-11-01 13:48:15 -07001268 fd = open(alt)
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001269 try:
1270 alt_dir = fd.readline().rstrip()
1271 finally:
1272 fd.close()
1273 except IOError:
1274 alt_dir = None
1275 else:
1276 alt_dir = None
1277
Shawn O. Pearcee02ac0a2012-03-14 15:36:59 -07001278 if clone_bundle \
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001279 and alt_dir is None \
1280 and self._ApplyCloneBundle(initial=is_new, quiet=quiet):
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001281 is_new = False
1282
Shawn O. Pearce6ba6ba02012-05-24 09:46:50 -07001283 if not current_branch_only:
1284 if self.sync_c:
1285 current_branch_only = True
1286 elif not self.manifest._loaded:
1287 # Manifest cannot check defaults until it syncs.
1288 current_branch_only = False
1289 elif self.manifest.default.sync_c:
1290 current_branch_only = True
1291
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001292 if self.clone_depth:
1293 depth = self.clone_depth
1294 else:
1295 depth = self.manifest.manifestProject.config.GetString('repo.depth')
1296
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001297 need_to_fetch = not (optimized_fetch and
1298 (ID_RE.match(self.revisionExpr) and
Zac Livingstone4332262017-06-16 08:56:09 -06001299 self._CheckForImmutableRevision()))
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001300 if (need_to_fetch and
1301 not self._RemoteFetch(initial=is_new, quiet=quiet, alt_dir=alt_dir,
1302 current_branch_only=current_branch_only,
Martin Kellye4e94d22017-03-21 16:05:12 -07001303 no_tags=no_tags, prune=prune, depth=depth,
1304 submodules=submodules)):
Anthony King7bdac712014-07-16 12:56:40 +01001305 return False
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001306
1307 if self.worktree:
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001308 self._InitMRef()
1309 else:
1310 self._InitMirrorHead()
1311 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08001312 platform_utils.remove(os.path.join(self.gitdir, 'FETCH_HEAD'))
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08001313 except OSError:
1314 pass
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001315 return True
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08001316
1317 def PostRepoUpgrade(self):
1318 self._InitHooks()
1319
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001320 def _CopyAndLinkFiles(self):
Simran Basib9a1b732015-08-20 12:19:28 -07001321 if self.manifest.isGitcClient:
1322 return
David Pursehouse8a68ff92012-09-24 12:15:13 +09001323 for copyfile in self.copyfiles:
1324 copyfile._Copy()
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001325 for linkfile in self.linkfiles:
1326 linkfile._Link()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001327
Julien Camperguedd654222014-01-09 16:21:37 +01001328 def GetCommitRevisionId(self):
1329 """Get revisionId of a commit.
1330
1331 Use this method instead of GetRevisionId to get the id of the commit rather
1332 than the id of the current git object (for example, a tag)
1333
1334 """
1335 if not self.revisionExpr.startswith(R_TAGS):
1336 return self.GetRevisionId(self._allrefs)
1337
1338 try:
1339 return self.bare_git.rev_list(self.revisionExpr, '-1')[0]
1340 except GitError:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001341 raise ManifestInvalidRevisionError('revision %s in %s not found' %
1342 (self.revisionExpr, self.name))
Julien Camperguedd654222014-01-09 16:21:37 +01001343
David Pursehouse8a68ff92012-09-24 12:15:13 +09001344 def GetRevisionId(self, all_refs=None):
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001345 if self.revisionId:
1346 return self.revisionId
1347
1348 rem = self.GetRemote(self.remote.name)
1349 rev = rem.ToLocal(self.revisionExpr)
1350
David Pursehouse8a68ff92012-09-24 12:15:13 +09001351 if all_refs is not None and rev in all_refs:
1352 return all_refs[rev]
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001353
1354 try:
1355 return self.bare_git.rev_parse('--verify', '%s^0' % rev)
1356 except GitError:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001357 raise ManifestInvalidRevisionError('revision %s in %s not found' %
1358 (self.revisionExpr, self.name))
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001359
Martin Kellye4e94d22017-03-21 16:05:12 -07001360 def Sync_LocalHalf(self, syncbuf, force_sync=False, submodules=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001361 """Perform only the local IO portion of the sync process.
1362 Network access is not required.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001363 """
Martin Kellye4e94d22017-03-21 16:05:12 -07001364 self._InitWorkTree(force_sync=force_sync, submodules=submodules)
David Pursehouse8a68ff92012-09-24 12:15:13 +09001365 all_refs = self.bare_ref.all
1366 self.CleanPublishedCache(all_refs)
1367 revid = self.GetRevisionId(all_refs)
Skyler Kaufman835cd682011-03-08 12:14:41 -08001368
David Pursehouse1d947b32012-10-25 12:23:11 +09001369 def _doff():
1370 self._FastForward(revid)
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001371 self._CopyAndLinkFiles()
David Pursehouse1d947b32012-10-25 12:23:11 +09001372
Martin Kellye4e94d22017-03-21 16:05:12 -07001373 def _dosubmodules():
1374 self._SyncSubmodules(quiet=True)
1375
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001376 head = self.work_git.GetHead()
1377 if head.startswith(R_HEADS):
1378 branch = head[len(R_HEADS):]
1379 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001380 head = all_refs[head]
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001381 except KeyError:
1382 head = None
1383 else:
1384 branch = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001385
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001386 if branch is None or syncbuf.detach_head:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001387 # Currently on a detached HEAD. The user is assumed to
1388 # not have any local modifications worth worrying about.
1389 #
Shawn O. Pearce3d2cdd02009-04-18 15:26:10 -07001390 if self.IsRebaseInProgress():
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001391 syncbuf.fail(self, _PriorSyncFailedError())
1392 return
1393
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001394 if head == revid:
1395 # No changes; don't do anything further.
Florian Vallee7cf1b362012-06-07 17:11:42 +02001396 # Except if the head needs to be detached
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001397 #
Florian Vallee7cf1b362012-06-07 17:11:42 +02001398 if not syncbuf.detach_head:
Dan Willemsen029eaf32015-09-03 12:52:28 -07001399 # The copy/linkfile config may have changed.
1400 self._CopyAndLinkFiles()
Florian Vallee7cf1b362012-06-07 17:11:42 +02001401 return
1402 else:
1403 lost = self._revlist(not_rev(revid), HEAD)
1404 if lost:
1405 syncbuf.info(self, "discarding %d commits", len(lost))
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001406
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001407 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001408 self._Checkout(revid, quiet=True)
Martin Kellye4e94d22017-03-21 16:05:12 -07001409 if submodules:
1410 self._SyncSubmodules(quiet=True)
Sarah Owensa5be53f2012-09-09 15:37:57 -07001411 except GitError as e:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001412 syncbuf.fail(self, e)
1413 return
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001414 self._CopyAndLinkFiles()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001415 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001416
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001417 if head == revid:
1418 # No changes; don't do anything further.
1419 #
Dan Willemsen029eaf32015-09-03 12:52:28 -07001420 # The copy/linkfile config may have changed.
1421 self._CopyAndLinkFiles()
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07001422 return
1423
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001424 branch = self.GetBranch(branch)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001425
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001426 if not branch.LocalMerge:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001427 # The current branch has no tracking configuration.
Anatol Pomazau2a32f6a2011-08-30 10:52:33 -07001428 # Jump off it to a detached HEAD.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001429 #
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001430 syncbuf.info(self,
1431 "leaving %s; does not track upstream",
1432 branch.name)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001433 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001434 self._Checkout(revid, quiet=True)
Martin Kellye4e94d22017-03-21 16:05:12 -07001435 if submodules:
1436 self._SyncSubmodules(quiet=True)
Sarah Owensa5be53f2012-09-09 15:37:57 -07001437 except GitError as e:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001438 syncbuf.fail(self, e)
1439 return
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001440 self._CopyAndLinkFiles()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001441 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001442
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001443 upstream_gain = self._revlist(not_rev(HEAD), revid)
David Pursehouse8a68ff92012-09-24 12:15:13 +09001444 pub = self.WasPublished(branch.name, all_refs)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001445 if pub:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001446 not_merged = self._revlist(not_rev(revid), pub)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001447 if not_merged:
1448 if upstream_gain:
1449 # The user has published this branch and some of those
1450 # commits are not yet merged upstream. We do not want
1451 # to rewrite the published commits so we punt.
1452 #
Daniel Sandler4c50dee2010-03-02 15:38:03 -05001453 syncbuf.fail(self,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001454 "branch %s is published (but not merged) and is now "
1455 "%d commits behind" % (branch.name, len(upstream_gain)))
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001456 return
Shawn O. Pearce05f66b62009-04-21 08:26:32 -07001457 elif pub == head:
1458 # All published commits are merged, and thus we are a
1459 # strict subset. We can fast-forward safely.
Shawn O. Pearcea54c5272008-10-30 11:03:00 -07001460 #
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001461 syncbuf.later1(self, _doff)
Martin Kellye4e94d22017-03-21 16:05:12 -07001462 if submodules:
1463 syncbuf.later1(self, _dosubmodules)
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001464 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001465
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001466 # Examine the local commits not in the remote. Find the
1467 # last one attributed to this user, if any.
1468 #
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001469 local_changes = self._revlist(not_rev(revid), HEAD, format='%H %ce')
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001470 last_mine = None
1471 cnt_mine = 0
1472 for commit in local_changes:
Chirayu Desai0eb35cb2013-11-19 18:46:29 +05301473 commit_id, committer_email = commit.decode('utf-8').split(' ', 1)
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001474 if committer_email == self.UserEmail:
1475 last_mine = commit_id
1476 cnt_mine += 1
1477
Shawn O. Pearceda88ff42009-06-03 11:09:12 -07001478 if not upstream_gain and cnt_mine == len(local_changes):
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001479 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001480
1481 if self.IsDirty(consider_untracked=False):
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001482 syncbuf.fail(self, _DirtyError())
1483 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001484
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001485 # If the upstream switched on us, warn the user.
1486 #
1487 if branch.merge != self.revisionExpr:
1488 if branch.merge and self.revisionExpr:
1489 syncbuf.info(self,
1490 'manifest switched %s...%s',
1491 branch.merge,
1492 self.revisionExpr)
1493 elif branch.merge:
1494 syncbuf.info(self,
1495 'manifest no longer tracks %s',
1496 branch.merge)
1497
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001498 if cnt_mine < len(local_changes):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001499 # Upstream rebased. Not everything in HEAD
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001500 # was created by this user.
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001501 #
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001502 syncbuf.info(self,
1503 "discarding %d commits removed from upstream",
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001504 len(local_changes) - cnt_mine)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001505
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001506 branch.remote = self.GetRemote(self.remote.name)
Anatol Pomazaucd7c5de2012-03-20 13:45:00 -07001507 if not ID_RE.match(self.revisionExpr):
1508 # in case of manifest sync the revisionExpr might be a SHA1
1509 branch.merge = self.revisionExpr
Conley Owens04f2f0e2014-10-01 17:22:46 -07001510 if not branch.merge.startswith('refs/'):
1511 branch.merge = R_HEADS + branch.merge
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001512 branch.Save()
1513
Mike Pontillod3153822012-02-28 11:53:24 -08001514 if cnt_mine > 0 and self.rebase:
Martin Kellye4e94d22017-03-21 16:05:12 -07001515 def _docopyandlink():
1516 self._CopyAndLinkFiles()
1517
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001518 def _dorebase():
Anthony King7bdac712014-07-16 12:56:40 +01001519 self._Rebase(upstream='%s^1' % last_mine, onto=revid)
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001520 syncbuf.later2(self, _dorebase)
Martin Kellye4e94d22017-03-21 16:05:12 -07001521 if submodules:
1522 syncbuf.later2(self, _dosubmodules)
1523 syncbuf.later2(self, _docopyandlink)
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07001524 elif local_changes:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001525 try:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001526 self._ResetHard(revid)
Martin Kellye4e94d22017-03-21 16:05:12 -07001527 if submodules:
1528 self._SyncSubmodules(quiet=True)
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001529 self._CopyAndLinkFiles()
Sarah Owensa5be53f2012-09-09 15:37:57 -07001530 except GitError as e:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001531 syncbuf.fail(self, e)
1532 return
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001533 else:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07001534 syncbuf.later1(self, _doff)
Martin Kellye4e94d22017-03-21 16:05:12 -07001535 if submodules:
1536 syncbuf.later1(self, _dosubmodules)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001537
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -08001538 def AddCopyFile(self, src, dest, absdest):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001539 # dest should already be an absolute path, but src is project relative
1540 # make src an absolute path
Shawn O. Pearcec7a4eef2009-03-05 10:32:38 -08001541 abssrc = os.path.join(self.worktree, src)
1542 self.copyfiles.append(_CopyFile(src, dest, abssrc, absdest))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001543
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001544 def AddLinkFile(self, src, dest, absdest):
1545 # dest should already be an absolute path, but src is project relative
Colin Cross0184dcc2015-05-05 00:24:54 -07001546 # make src relative path to dest
1547 absdestdir = os.path.dirname(absdest)
1548 relsrc = os.path.relpath(os.path.join(self.worktree, src), absdestdir)
Wink Saville4c426ef2015-06-03 08:05:17 -07001549 self.linkfiles.append(_LinkFile(self.worktree, src, dest, relsrc, absdest))
Jeff Hamiltone0df2322014-04-21 17:10:59 -05001550
James W. Mills24c13082012-04-12 15:04:13 -05001551 def AddAnnotation(self, name, value, keep):
1552 self.annotations.append(_Annotation(name, value, keep))
1553
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001554 def DownloadPatchSet(self, change_id, patch_id):
1555 """Download a single patch set of a single change to FETCH_HEAD.
1556 """
1557 remote = self.GetRemote(self.remote.name)
1558
1559 cmd = ['fetch', remote.name]
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001560 cmd.append('refs/changes/%2.2d/%d/%d'
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001561 % (change_id % 100, change_id, patch_id))
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001562 if GitCommand(self, cmd, bare=True).Wait() != 0:
1563 return None
1564 return DownloadedChange(self,
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001565 self.GetRevisionId(),
Shawn O. Pearce632768b2008-10-23 11:58:52 -07001566 change_id,
1567 patch_id,
1568 self.bare_git.rev_parse('FETCH_HEAD'))
1569
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001570
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001571# Branch Management ##
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001572
Simran Basib9a1b732015-08-20 12:19:28 -07001573 def StartBranch(self, name, branch_merge=''):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001574 """Create a new branch off the manifest's revision.
1575 """
Simran Basib9a1b732015-08-20 12:19:28 -07001576 if not branch_merge:
1577 branch_merge = self.revisionExpr
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001578 head = self.work_git.GetHead()
1579 if head == (R_HEADS + name):
1580 return True
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001581
David Pursehouse8a68ff92012-09-24 12:15:13 +09001582 all_refs = self.bare_ref.all
Anthony King7bdac712014-07-16 12:56:40 +01001583 if R_HEADS + name in all_refs:
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001584 return GitCommand(self,
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001585 ['checkout', name, '--'],
Anthony King7bdac712014-07-16 12:56:40 +01001586 capture_stdout=True,
1587 capture_stderr=True).Wait() == 0
Shawn O. Pearce0a389e92009-04-10 16:21:18 -07001588
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001589 branch = self.GetBranch(name)
1590 branch.remote = self.GetRemote(self.remote.name)
Simran Basib9a1b732015-08-20 12:19:28 -07001591 branch.merge = branch_merge
1592 if not branch.merge.startswith('refs/') and not ID_RE.match(branch_merge):
1593 branch.merge = R_HEADS + branch_merge
David Pursehouse8a68ff92012-09-24 12:15:13 +09001594 revid = self.GetRevisionId(all_refs)
Shawn O. Pearce0a389e92009-04-10 16:21:18 -07001595
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001596 if head.startswith(R_HEADS):
1597 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001598 head = all_refs[head]
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001599 except KeyError:
1600 head = None
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001601 if revid and head and revid == head:
1602 ref = os.path.join(self.gitdir, R_HEADS + name)
1603 try:
1604 os.makedirs(os.path.dirname(ref))
1605 except OSError:
1606 pass
1607 _lwrite(ref, '%s\n' % revid)
1608 _lwrite(os.path.join(self.worktree, '.git', HEAD),
1609 'ref: %s%s\n' % (R_HEADS, name))
1610 branch.Save()
1611 return True
1612
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001613 if GitCommand(self,
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001614 ['checkout', '-b', branch.name, revid],
Anthony King7bdac712014-07-16 12:56:40 +01001615 capture_stdout=True,
1616 capture_stderr=True).Wait() == 0:
Shawn O. Pearceaccc56d2009-04-18 14:45:51 -07001617 branch.Save()
1618 return True
1619 return False
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001620
Wink Saville02d79452009-04-10 13:01:24 -07001621 def CheckoutBranch(self, name):
1622 """Checkout a local topic branch.
Doug Anderson3ba5f952011-04-07 12:51:04 -07001623
1624 Args:
1625 name: The name of the branch to checkout.
1626
1627 Returns:
1628 True if the checkout succeeded; False if it didn't; None if the branch
1629 didn't exist.
Wink Saville02d79452009-04-10 13:01:24 -07001630 """
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001631 rev = R_HEADS + name
1632 head = self.work_git.GetHead()
1633 if head == rev:
1634 # Already on the branch
1635 #
1636 return True
Wink Saville02d79452009-04-10 13:01:24 -07001637
David Pursehouse8a68ff92012-09-24 12:15:13 +09001638 all_refs = self.bare_ref.all
Wink Saville02d79452009-04-10 13:01:24 -07001639 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001640 revid = all_refs[rev]
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001641 except KeyError:
1642 # Branch does not exist in this project
1643 #
Doug Anderson3ba5f952011-04-07 12:51:04 -07001644 return None
Wink Saville02d79452009-04-10 13:01:24 -07001645
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001646 if head.startswith(R_HEADS):
1647 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09001648 head = all_refs[head]
Shawn O. Pearce89e717d2009-04-18 15:04:41 -07001649 except KeyError:
1650 head = None
1651
1652 if head == revid:
1653 # Same revision; just update HEAD to point to the new
1654 # target branch, but otherwise take no other action.
1655 #
1656 _lwrite(os.path.join(self.worktree, '.git', HEAD),
1657 'ref: %s%s\n' % (R_HEADS, name))
1658 return True
1659
1660 return GitCommand(self,
1661 ['checkout', name, '--'],
Anthony King7bdac712014-07-16 12:56:40 +01001662 capture_stdout=True,
1663 capture_stderr=True).Wait() == 0
Wink Saville02d79452009-04-10 13:01:24 -07001664
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001665 def AbandonBranch(self, name):
1666 """Destroy a local topic branch.
Doug Andersondafb1d62011-04-07 11:46:59 -07001667
1668 Args:
1669 name: The name of the branch to abandon.
1670
1671 Returns:
1672 True if the abandon succeeded; False if it didn't; None if the branch
1673 didn't exist.
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001674 """
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001675 rev = R_HEADS + name
David Pursehouse8a68ff92012-09-24 12:15:13 +09001676 all_refs = self.bare_ref.all
1677 if rev not in all_refs:
Doug Andersondafb1d62011-04-07 11:46:59 -07001678 # Doesn't exist
1679 return None
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001680
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001681 head = self.work_git.GetHead()
1682 if head == rev:
1683 # We can't destroy the branch while we are sitting
1684 # on it. Switch to a detached HEAD.
1685 #
David Pursehouse8a68ff92012-09-24 12:15:13 +09001686 head = all_refs[head]
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001687
David Pursehouse8a68ff92012-09-24 12:15:13 +09001688 revid = self.GetRevisionId(all_refs)
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001689 if head == revid:
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001690 _lwrite(os.path.join(self.worktree, '.git', HEAD),
1691 '%s\n' % revid)
1692 else:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001693 self._Checkout(revid, quiet=True)
Shawn O. Pearce552ac892009-04-18 15:15:24 -07001694
1695 return GitCommand(self,
1696 ['branch', '-D', name],
Anthony King7bdac712014-07-16 12:56:40 +01001697 capture_stdout=True,
1698 capture_stderr=True).Wait() == 0
Shawn O. Pearce9fa44db2008-11-03 11:24:59 -08001699
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001700 def PruneHeads(self):
1701 """Prune any topic branches already merged into upstream.
1702 """
1703 cb = self.CurrentBranch
1704 kill = []
Shawn O. Pearce3778f9d2009-03-02 12:30:50 -08001705 left = self._allrefs
1706 for name in left.keys():
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001707 if name.startswith(R_HEADS):
1708 name = name[len(R_HEADS):]
1709 if cb is None or name != cb:
1710 kill.append(name)
1711
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07001712 rev = self.GetRevisionId(left)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001713 if cb is not None \
1714 and not self._revlist(HEAD + '...' + rev) \
Anthony King7bdac712014-07-16 12:56:40 +01001715 and not self.IsDirty(consider_untracked=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001716 self.work_git.DetachHead(HEAD)
1717 kill.append(cb)
1718
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001719 if kill:
Shawn O. Pearce5b23f242009-04-17 18:43:33 -07001720 old = self.bare_git.GetHead()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001721
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001722 try:
1723 self.bare_git.DetachHead(rev)
1724
1725 b = ['branch', '-d']
1726 b.extend(kill)
1727 b = GitCommand(self, b, bare=True,
1728 capture_stdout=True,
1729 capture_stderr=True)
1730 b.Wait()
1731 finally:
Dan Willemsen1a799d12015-12-15 13:40:05 -08001732 if ID_RE.match(old):
1733 self.bare_git.DetachHead(old)
1734 else:
1735 self.bare_git.SetHead(old)
Shawn O. Pearce3778f9d2009-03-02 12:30:50 -08001736 left = self._allrefs
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001737
Shawn O. Pearce3778f9d2009-03-02 12:30:50 -08001738 for branch in kill:
1739 if (R_HEADS + branch) not in left:
1740 self.CleanPublishedCache()
1741 break
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001742
1743 if cb and cb not in kill:
1744 kill.append(cb)
Shawn O. Pearce7c6c64d2009-03-02 12:38:13 -08001745 kill.sort()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001746
1747 kept = []
1748 for branch in kill:
Anthony King7bdac712014-07-16 12:56:40 +01001749 if R_HEADS + branch in left:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001750 branch = self.GetBranch(branch)
1751 base = branch.LocalMerge
1752 if not base:
1753 base = rev
1754 kept.append(ReviewableBranch(self, branch, base))
1755 return kept
1756
1757
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001758# Submodule Management ##
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001759
1760 def GetRegisteredSubprojects(self):
1761 result = []
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001762
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001763 def rec(subprojects):
1764 if not subprojects:
1765 return
1766 result.extend(subprojects)
1767 for p in subprojects:
1768 rec(p.subprojects)
1769 rec(self.subprojects)
1770 return result
1771
1772 def _GetSubmodules(self):
1773 # Unfortunately we cannot call `git submodule status --recursive` here
1774 # because the working tree might not exist yet, and it cannot be used
1775 # without a working tree in its current implementation.
1776
1777 def get_submodules(gitdir, rev):
1778 # Parse .gitmodules for submodule sub_paths and sub_urls
1779 sub_paths, sub_urls = parse_gitmodules(gitdir, rev)
1780 if not sub_paths:
1781 return []
1782 # Run `git ls-tree` to read SHAs of submodule object, which happen to be
1783 # revision of submodule repository
1784 sub_revs = git_ls_tree(gitdir, rev, sub_paths)
1785 submodules = []
1786 for sub_path, sub_url in zip(sub_paths, sub_urls):
1787 try:
1788 sub_rev = sub_revs[sub_path]
1789 except KeyError:
1790 # Ignore non-exist submodules
1791 continue
1792 submodules.append((sub_rev, sub_path, sub_url))
1793 return submodules
1794
1795 re_path = re.compile(r'^submodule\.([^.]+)\.path=(.*)$')
1796 re_url = re.compile(r'^submodule\.([^.]+)\.url=(.*)$')
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001797
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001798 def parse_gitmodules(gitdir, rev):
1799 cmd = ['cat-file', 'blob', '%s:.gitmodules' % rev]
1800 try:
Anthony King7bdac712014-07-16 12:56:40 +01001801 p = GitCommand(None, cmd, capture_stdout=True, capture_stderr=True,
1802 bare=True, gitdir=gitdir)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001803 except GitError:
1804 return [], []
1805 if p.Wait() != 0:
1806 return [], []
1807
1808 gitmodules_lines = []
1809 fd, temp_gitmodules_path = tempfile.mkstemp()
1810 try:
1811 os.write(fd, p.stdout)
1812 os.close(fd)
1813 cmd = ['config', '--file', temp_gitmodules_path, '--list']
Anthony King7bdac712014-07-16 12:56:40 +01001814 p = GitCommand(None, cmd, capture_stdout=True, capture_stderr=True,
1815 bare=True, gitdir=gitdir)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001816 if p.Wait() != 0:
1817 return [], []
1818 gitmodules_lines = p.stdout.split('\n')
1819 except GitError:
1820 return [], []
1821 finally:
Renaud Paquay010fed72016-11-11 14:25:29 -08001822 platform_utils.remove(temp_gitmodules_path)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001823
1824 names = set()
1825 paths = {}
1826 urls = {}
1827 for line in gitmodules_lines:
1828 if not line:
1829 continue
1830 m = re_path.match(line)
1831 if m:
1832 names.add(m.group(1))
1833 paths[m.group(1)] = m.group(2)
1834 continue
1835 m = re_url.match(line)
1836 if m:
1837 names.add(m.group(1))
1838 urls[m.group(1)] = m.group(2)
1839 continue
1840 names = sorted(names)
1841 return ([paths.get(name, '') for name in names],
1842 [urls.get(name, '') for name in names])
1843
1844 def git_ls_tree(gitdir, rev, paths):
1845 cmd = ['ls-tree', rev, '--']
1846 cmd.extend(paths)
1847 try:
Anthony King7bdac712014-07-16 12:56:40 +01001848 p = GitCommand(None, cmd, capture_stdout=True, capture_stderr=True,
1849 bare=True, gitdir=gitdir)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001850 except GitError:
1851 return []
1852 if p.Wait() != 0:
1853 return []
1854 objects = {}
1855 for line in p.stdout.split('\n'):
1856 if not line.strip():
1857 continue
1858 object_rev, object_path = line.split()[2:4]
1859 objects[object_path] = object_rev
1860 return objects
1861
1862 try:
1863 rev = self.GetRevisionId()
1864 except GitError:
1865 return []
1866 return get_submodules(self.gitdir, rev)
1867
1868 def GetDerivedSubprojects(self):
1869 result = []
1870 if not self.Exists:
1871 # If git repo does not exist yet, querying its submodules will
1872 # mess up its states; so return here.
1873 return result
1874 for rev, path, url in self._GetSubmodules():
1875 name = self.manifest.GetSubprojectName(self, path)
David James8d201162013-10-11 17:03:19 -07001876 relpath, worktree, gitdir, objdir = \
1877 self.manifest.GetSubprojectPaths(self, name, path)
1878 project = self.manifest.paths.get(relpath)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001879 if project:
1880 result.extend(project.GetDerivedSubprojects())
1881 continue
David James8d201162013-10-11 17:03:19 -07001882
Shouheng Zhang02c0ee62017-12-08 09:54:58 +08001883 if url.startswith('..'):
1884 url = urllib.parse.urljoin("%s/" % self.remote.url, url)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001885 remote = RemoteSpec(self.remote.name,
Anthony King7bdac712014-07-16 12:56:40 +01001886 url=url,
Steve Raed6480452016-08-10 15:00:00 -07001887 pushUrl=self.remote.pushUrl,
Anthony King7bdac712014-07-16 12:56:40 +01001888 review=self.remote.review,
1889 revision=self.remote.revision)
1890 subproject = Project(manifest=self.manifest,
1891 name=name,
1892 remote=remote,
1893 gitdir=gitdir,
1894 objdir=objdir,
1895 worktree=worktree,
1896 relpath=relpath,
Aymen Bouaziz2598ed02016-06-24 14:34:08 +02001897 revisionExpr=rev,
Anthony King7bdac712014-07-16 12:56:40 +01001898 revisionId=rev,
1899 rebase=self.rebase,
1900 groups=self.groups,
1901 sync_c=self.sync_c,
1902 sync_s=self.sync_s,
1903 parent=self,
1904 is_derived=True)
Che-Liang Chioub2bd91c2012-01-11 11:28:42 +08001905 result.append(subproject)
1906 result.extend(subproject.GetDerivedSubprojects())
1907 return result
1908
1909
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07001910# Direct Git Commands ##
Zac Livingstone4332262017-06-16 08:56:09 -06001911 def _CheckForImmutableRevision(self):
Chris AtLee2fb64662014-01-16 21:32:33 -05001912 try:
1913 # if revision (sha or tag) is not present then following function
1914 # throws an error.
1915 self.bare_git.rev_parse('--verify', '%s^0' % self.revisionExpr)
1916 return True
1917 except GitError:
1918 # There is no such persistent revision. We have to fetch it.
1919 return False
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001920
Julien Campergue335f5ef2013-10-16 11:02:35 +02001921 def _FetchArchive(self, tarpath, cwd=None):
1922 cmd = ['archive', '-v', '-o', tarpath]
1923 cmd.append('--remote=%s' % self.remote.url)
1924 cmd.append('--prefix=%s/' % self.relpath)
1925 cmd.append(self.revisionExpr)
1926
1927 command = GitCommand(self, cmd, cwd=cwd,
1928 capture_stdout=True,
1929 capture_stderr=True)
1930
1931 if command.Wait() != 0:
1932 raise GitError('git archive %s: %s' % (self.name, command.stderr))
1933
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001934 def _RemoteFetch(self, name=None,
1935 current_branch_only=False,
Shawn O. Pearce16614f82010-10-29 12:05:43 -07001936 initial=False,
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001937 quiet=False,
Mitchel Humpherys597868b2012-10-29 10:18:34 -07001938 alt_dir=None,
David Pursehouse74cfd272015-10-14 10:50:15 +09001939 no_tags=False,
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001940 prune=False,
Martin Kellye4e94d22017-03-21 16:05:12 -07001941 depth=None,
1942 submodules=False):
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001943
1944 is_sha1 = False
1945 tag_name = None
David Pursehouse9bc422f2014-04-15 10:28:56 +09001946 # The depth should not be used when fetching to a mirror because
1947 # it will result in a shallow repository that cannot be cloned or
1948 # fetched from.
Aymen Bouaziz6c594462016-10-25 18:03:51 +02001949 # The repo project should also never be synced with partial depth.
1950 if self.manifest.IsMirror or self.relpath == '.repo/repo':
1951 depth = None
David Pursehouse9bc422f2014-04-15 10:28:56 +09001952
Shawn Pearce69e04d82014-01-29 12:48:54 -08001953 if depth:
1954 current_branch_only = True
1955
Nasser Grainawi909d58b2014-09-19 12:13:04 -06001956 if ID_RE.match(self.revisionExpr) is not None:
1957 is_sha1 = True
1958
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001959 if current_branch_only:
Nasser Grainawi909d58b2014-09-19 12:13:04 -06001960 if self.revisionExpr.startswith(R_TAGS):
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001961 # this is a tag and its sha1 value should never change
1962 tag_name = self.revisionExpr[len(R_TAGS):]
1963
1964 if is_sha1 or tag_name is not None:
Zac Livingstone4332262017-06-16 08:56:09 -06001965 if self._CheckForImmutableRevision():
1966 print('Skipped fetching project %s (already have persistent ref)'
1967 % self.name)
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001968 return True
Bertrand SIMONNET3000cda2014-11-25 16:19:29 -08001969 if is_sha1 and not depth:
1970 # When syncing a specific commit and --depth is not set:
1971 # * if upstream is explicitly specified and is not a sha1, fetch only
1972 # upstream as users expect only upstream to be fetch.
1973 # Note: The commit might not be in upstream in which case the sync
1974 # will fail.
1975 # * otherwise, fetch all branches to make sure we end up with the
1976 # specific commit.
Aymen Bouaziz037040f2016-06-28 12:27:23 +02001977 if self.upstream:
1978 current_branch_only = not ID_RE.match(self.upstream)
1979 else:
1980 current_branch_only = False
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07001981
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07001982 if not name:
1983 name = self.remote.name
Shawn O. Pearcefb231612009-04-10 18:53:46 -07001984
1985 ssh_proxy = False
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07001986 remote = self.GetRemote(name)
1987 if remote.PreConnectFetch():
Shawn O. Pearcefb231612009-04-10 18:53:46 -07001988 ssh_proxy = True
1989
Shawn O. Pearce88443382010-10-08 10:02:09 +02001990 if initial:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07001991 if alt_dir and 'objects' == os.path.basename(alt_dir):
1992 ref_dir = os.path.dirname(alt_dir)
Shawn O. Pearce88443382010-10-08 10:02:09 +02001993 packed_refs = os.path.join(self.gitdir, 'packed-refs')
1994 remote = self.GetRemote(name)
1995
David Pursehouse8a68ff92012-09-24 12:15:13 +09001996 all_refs = self.bare_ref.all
1997 ids = set(all_refs.values())
Shawn O. Pearce88443382010-10-08 10:02:09 +02001998 tmp = set()
1999
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302000 for r, ref_id in GitRefs(ref_dir).all.items():
David Pursehouse8a68ff92012-09-24 12:15:13 +09002001 if r not in all_refs:
Shawn O. Pearce88443382010-10-08 10:02:09 +02002002 if r.startswith(R_TAGS) or remote.WritesTo(r):
David Pursehouse8a68ff92012-09-24 12:15:13 +09002003 all_refs[r] = ref_id
2004 ids.add(ref_id)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002005 continue
2006
David Pursehouse8a68ff92012-09-24 12:15:13 +09002007 if ref_id in ids:
Shawn O. Pearce88443382010-10-08 10:02:09 +02002008 continue
2009
David Pursehouse8a68ff92012-09-24 12:15:13 +09002010 r = 'refs/_alt/%s' % ref_id
2011 all_refs[r] = ref_id
2012 ids.add(ref_id)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002013 tmp.add(r)
2014
heping3d7bbc92017-04-12 19:51:47 +08002015 tmp_packed_lines = []
2016 old_packed_lines = []
Shawn O. Pearce88443382010-10-08 10:02:09 +02002017
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302018 for r in sorted(all_refs):
David Pursehouse8a68ff92012-09-24 12:15:13 +09002019 line = '%s %s\n' % (all_refs[r], r)
heping3d7bbc92017-04-12 19:51:47 +08002020 tmp_packed_lines.append(line)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002021 if r not in tmp:
heping3d7bbc92017-04-12 19:51:47 +08002022 old_packed_lines.append(line)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002023
heping3d7bbc92017-04-12 19:51:47 +08002024 tmp_packed = ''.join(tmp_packed_lines)
2025 old_packed = ''.join(old_packed_lines)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002026 _lwrite(packed_refs, tmp_packed)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002027 else:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002028 alt_dir = None
Shawn O. Pearce88443382010-10-08 10:02:09 +02002029
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002030 cmd = ['fetch']
Doug Anderson30d45292011-05-04 15:01:04 -07002031
Conley Owensf97e8382015-01-21 11:12:46 -08002032 if depth:
Doug Anderson30d45292011-05-04 15:01:04 -07002033 cmd.append('--depth=%s' % depth)
Dan Willemseneeab6862015-08-03 13:11:53 -07002034 else:
2035 # If this repo has shallow objects, then we don't know which refs have
2036 # shallow objects or not. Tell git to unshallow all fetched refs. Don't
2037 # do this with projects that don't have shallow objects, since it is less
2038 # efficient.
2039 if os.path.exists(os.path.join(self.gitdir, 'shallow')):
2040 cmd.append('--depth=2147483647')
Doug Anderson30d45292011-05-04 15:01:04 -07002041
Shawn O. Pearce16614f82010-10-29 12:05:43 -07002042 if quiet:
2043 cmd.append('--quiet')
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002044 if not self.worktree:
2045 cmd.append('--update-head-ok')
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002046 cmd.append(name)
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07002047
Mitchel Humpherys26c45a72014-03-10 14:21:59 -07002048 # If using depth then we should not get all the tags since they may
2049 # be outside of the depth.
2050 if no_tags or depth:
2051 cmd.append('--no-tags')
2052 else:
2053 cmd.append('--tags')
2054
David Pursehouse74cfd272015-10-14 10:50:15 +09002055 if prune:
2056 cmd.append('--prune')
2057
Martin Kellye4e94d22017-03-21 16:05:12 -07002058 if submodules:
2059 cmd.append('--recurse-submodules=on-demand')
2060
Conley Owens80b87fe2014-05-09 17:13:44 -07002061 spec = []
Brian Harring14a66742012-09-28 20:21:57 -07002062 if not current_branch_only:
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07002063 # Fetch whole repo
Conley Owens80b87fe2014-05-09 17:13:44 -07002064 spec.append(str((u'+refs/heads/*:') + remote.ToLocal('refs/heads/*')))
Anatol Pomazau53d6f4d2011-08-25 17:21:47 -07002065 elif tag_name is not None:
Conley Owens80b87fe2014-05-09 17:13:44 -07002066 spec.append('tag')
2067 spec.append(tag_name)
Nasser Grainawi04e52d62014-09-30 13:34:52 -06002068
David Pursehouse403b64e2015-04-27 10:41:33 +09002069 if not self.manifest.IsMirror:
2070 branch = self.revisionExpr
Kevin Degi679bac42015-06-22 15:31:26 -06002071 if is_sha1 and depth and git_require((1, 8, 3)):
David Pursehouse403b64e2015-04-27 10:41:33 +09002072 # Shallow checkout of a specific commit, fetch from that commit and not
2073 # the heads only as the commit might be deeper in the history.
2074 spec.append(branch)
2075 else:
2076 if is_sha1:
2077 branch = self.upstream
2078 if branch is not None and branch.strip():
2079 if not branch.startswith('refs/'):
2080 branch = R_HEADS + branch
2081 spec.append(str((u'+%s:' % branch) + remote.ToLocal(branch)))
Conley Owens80b87fe2014-05-09 17:13:44 -07002082 cmd.extend(spec)
2083
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002084 ok = False
David Pursehouse8a68ff92012-09-24 12:15:13 +09002085 for _i in range(2):
John L. Villalovos9c76f672015-03-16 20:49:10 -07002086 gitcmd = GitCommand(self, cmd, bare=True, ssh_proxy=ssh_proxy)
John L. Villalovos126e2982015-01-29 21:58:12 -08002087 ret = gitcmd.Wait()
Brian Harring14a66742012-09-28 20:21:57 -07002088 if ret == 0:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002089 ok = True
2090 break
John L. Villalovos126e2982015-01-29 21:58:12 -08002091 # If needed, run the 'git remote prune' the first time through the loop
2092 elif (not _i and
2093 "error:" in gitcmd.stderr and
2094 "git remote prune" in gitcmd.stderr):
2095 prunecmd = GitCommand(self, ['remote', 'prune', name], bare=True,
John L. Villalovos9c76f672015-03-16 20:49:10 -07002096 ssh_proxy=ssh_proxy)
John L. Villalovose30f46b2015-02-25 14:27:02 -08002097 ret = prunecmd.Wait()
John L. Villalovose30f46b2015-02-25 14:27:02 -08002098 if ret:
John L. Villalovos126e2982015-01-29 21:58:12 -08002099 break
2100 continue
Brian Harring14a66742012-09-28 20:21:57 -07002101 elif current_branch_only and is_sha1 and ret == 128:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002102 # Exit code 128 means "couldn't find the ref you asked for"; if we're
2103 # in sha1 mode, we just tried sync'ing from the upstream field; it
2104 # doesn't exist, thus abort the optimization attempt and do a full sync.
Brian Harring14a66742012-09-28 20:21:57 -07002105 break
Colin Crossc4b301f2015-05-13 00:10:02 -07002106 elif ret < 0:
2107 # Git died with a signal, exit immediately
2108 break
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002109 time.sleep(random.randint(30, 45))
Shawn O. Pearce88443382010-10-08 10:02:09 +02002110
2111 if initial:
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002112 if alt_dir:
Shawn O. Pearce88443382010-10-08 10:02:09 +02002113 if old_packed != '':
2114 _lwrite(packed_refs, old_packed)
2115 else:
Renaud Paquay010fed72016-11-11 14:25:29 -08002116 platform_utils.remove(packed_refs)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002117 self.bare_git.pack_refs('--all', '--prune')
Brian Harring14a66742012-09-28 20:21:57 -07002118
Aymen Bouaziz6c594462016-10-25 18:03:51 +02002119 if is_sha1 and current_branch_only:
Brian Harring14a66742012-09-28 20:21:57 -07002120 # We just synced the upstream given branch; verify we
2121 # got what we wanted, else trigger a second run of all
2122 # refs.
Zac Livingstone4332262017-06-16 08:56:09 -06002123 if not self._CheckForImmutableRevision():
Aymen Bouaziz6c594462016-10-25 18:03:51 +02002124 if current_branch_only and depth:
2125 # Sync the current branch only with depth set to None
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002126 return self._RemoteFetch(name=name,
2127 current_branch_only=current_branch_only,
Aymen Bouaziz6c594462016-10-25 18:03:51 +02002128 initial=False, quiet=quiet, alt_dir=alt_dir,
2129 depth=None)
2130 else:
2131 # Avoid infinite recursion: sync all branches with depth set to None
2132 return self._RemoteFetch(name=name, current_branch_only=False,
2133 initial=False, quiet=quiet, alt_dir=alt_dir,
2134 depth=None)
Brian Harring14a66742012-09-28 20:21:57 -07002135
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002136 return ok
Shawn O. Pearce88443382010-10-08 10:02:09 +02002137
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002138 def _ApplyCloneBundle(self, initial=False, quiet=False):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002139 if initial and \
2140 (self.manifest.manifestProject.config.GetString('repo.depth') or
2141 self.clone_depth):
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002142 return False
2143
2144 remote = self.GetRemote(self.remote.name)
2145 bundle_url = remote.url + '/clone.bundle'
2146 bundle_url = GitConfig.ForUser().UrlInsteadOf(bundle_url)
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002147 if GetSchemeFromUrl(bundle_url) not in ('http', 'https',
2148 'persistent-http',
2149 'persistent-https'):
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002150 return False
2151
2152 bundle_dst = os.path.join(self.gitdir, 'clone.bundle')
2153 bundle_tmp = os.path.join(self.gitdir, 'clone.bundle.tmp')
2154
2155 exist_dst = os.path.exists(bundle_dst)
2156 exist_tmp = os.path.exists(bundle_tmp)
2157
2158 if not initial and not exist_dst and not exist_tmp:
2159 return False
2160
2161 if not exist_dst:
2162 exist_dst = self._FetchBundle(bundle_url, bundle_tmp, bundle_dst, quiet)
2163 if not exist_dst:
2164 return False
2165
2166 cmd = ['fetch']
2167 if quiet:
2168 cmd.append('--quiet')
2169 if not self.worktree:
2170 cmd.append('--update-head-ok')
2171 cmd.append(bundle_dst)
2172 for f in remote.fetch:
2173 cmd.append(str(f))
2174 cmd.append('refs/tags/*:refs/tags/*')
2175
2176 ok = GitCommand(self, cmd, bare=True).Wait() == 0
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002177 if os.path.exists(bundle_dst):
Renaud Paquay010fed72016-11-11 14:25:29 -08002178 platform_utils.remove(bundle_dst)
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002179 if os.path.exists(bundle_tmp):
Renaud Paquay010fed72016-11-11 14:25:29 -08002180 platform_utils.remove(bundle_tmp)
Shawn O. Pearce88443382010-10-08 10:02:09 +02002181 return ok
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002182
Shawn O. Pearcec325dc32011-10-03 08:30:24 -07002183 def _FetchBundle(self, srcUrl, tmpPath, dstPath, quiet):
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002184 if os.path.exists(dstPath):
Renaud Paquay010fed72016-11-11 14:25:29 -08002185 platform_utils.remove(dstPath)
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002186
Matt Gumbel2dc810c2012-08-30 09:39:36 -07002187 cmd = ['curl', '--fail', '--output', tmpPath, '--netrc', '--location']
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002188 if quiet:
2189 cmd += ['--silent']
2190 if os.path.exists(tmpPath):
2191 size = os.stat(tmpPath).st_size
2192 if size >= 1024:
2193 cmd += ['--continue-at', '%d' % (size,)]
2194 else:
Renaud Paquay010fed72016-11-11 14:25:29 -08002195 platform_utils.remove(tmpPath)
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002196 if 'http_proxy' in os.environ and 'darwin' == sys.platform:
2197 cmd += ['--proxy', os.environ['http_proxy']]
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002198 with GetUrlCookieFile(srcUrl, quiet) as (cookiefile, _proxy):
Dave Borowitz137d0132015-01-02 11:12:54 -08002199 if cookiefile:
Dave Borowitz4abf8e62015-01-02 11:39:04 -08002200 cmd += ['--cookie', cookiefile, '--cookie-jar', cookiefile]
Dave Borowitz137d0132015-01-02 11:12:54 -08002201 if srcUrl.startswith('persistent-'):
2202 srcUrl = srcUrl[len('persistent-'):]
2203 cmd += [srcUrl]
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002204
Dave Borowitz137d0132015-01-02 11:12:54 -08002205 if IsTrace():
2206 Trace('%s', ' '.join(cmd))
2207 try:
2208 proc = subprocess.Popen(cmd)
2209 except OSError:
2210 return False
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002211
Dave Borowitz137d0132015-01-02 11:12:54 -08002212 curlret = proc.wait()
Matt Gumbel2dc810c2012-08-30 09:39:36 -07002213
Dave Borowitz137d0132015-01-02 11:12:54 -08002214 if curlret == 22:
2215 # From curl man page:
2216 # 22: HTTP page not retrieved. The requested url was not found or
2217 # returned another error with the HTTP error code being 400 or above.
2218 # This return code only appears if -f, --fail is used.
2219 if not quiet:
2220 print("Server does not provide clone.bundle; ignoring.",
2221 file=sys.stderr)
2222 return False
Matt Gumbel2dc810c2012-08-30 09:39:36 -07002223
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002224 if os.path.exists(tmpPath):
Kris Giesingc8d882a2014-12-23 13:02:32 -08002225 if curlret == 0 and self._IsValidBundle(tmpPath, quiet):
Renaud Paquayad1abcb2016-11-01 11:34:55 -07002226 platform_utils.rename(tmpPath, dstPath)
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002227 return True
2228 else:
Renaud Paquay010fed72016-11-11 14:25:29 -08002229 platform_utils.remove(tmpPath)
Shawn O. Pearce5e7127d2012-08-02 14:57:37 -07002230 return False
2231 else:
2232 return False
Shawn O. Pearcef322b9a2011-09-19 14:50:58 -07002233
Kris Giesingc8d882a2014-12-23 13:02:32 -08002234 def _IsValidBundle(self, path, quiet):
Dave Borowitz91f3ba52013-06-03 12:15:23 -07002235 try:
2236 with open(path) as f:
2237 if f.read(16) == '# v2 git bundle\n':
2238 return True
2239 else:
Kris Giesingc8d882a2014-12-23 13:02:32 -08002240 if not quiet:
2241 print("Invalid clone.bundle file; ignoring.", file=sys.stderr)
Dave Borowitz91f3ba52013-06-03 12:15:23 -07002242 return False
2243 except OSError:
2244 return False
2245
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002246 def _Checkout(self, rev, quiet=False):
2247 cmd = ['checkout']
2248 if quiet:
2249 cmd.append('-q')
2250 cmd.append(rev)
2251 cmd.append('--')
2252 if GitCommand(self, cmd).Wait() != 0:
2253 if self._allrefs:
2254 raise GitError('%s checkout %s ' % (self.name, rev))
2255
Anthony King7bdac712014-07-16 12:56:40 +01002256 def _CherryPick(self, rev):
Pierre Tardye5a21222011-03-24 16:28:18 +01002257 cmd = ['cherry-pick']
2258 cmd.append(rev)
2259 cmd.append('--')
2260 if GitCommand(self, cmd).Wait() != 0:
2261 if self._allrefs:
2262 raise GitError('%s cherry-pick %s ' % (self.name, rev))
2263
Anthony King7bdac712014-07-16 12:56:40 +01002264 def _Revert(self, rev):
Erwan Mahea94f1622011-08-19 13:56:09 +02002265 cmd = ['revert']
2266 cmd.append('--no-edit')
2267 cmd.append(rev)
2268 cmd.append('--')
2269 if GitCommand(self, cmd).Wait() != 0:
2270 if self._allrefs:
2271 raise GitError('%s revert %s ' % (self.name, rev))
2272
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002273 def _ResetHard(self, rev, quiet=True):
2274 cmd = ['reset', '--hard']
2275 if quiet:
2276 cmd.append('-q')
2277 cmd.append(rev)
2278 if GitCommand(self, cmd).Wait() != 0:
2279 raise GitError('%s reset --hard %s ' % (self.name, rev))
2280
Martin Kellye4e94d22017-03-21 16:05:12 -07002281 def _SyncSubmodules(self, quiet=True):
2282 cmd = ['submodule', 'update', '--init', '--recursive']
2283 if quiet:
2284 cmd.append('-q')
2285 if GitCommand(self, cmd).Wait() != 0:
2286 raise GitError('%s submodule update --init --recursive %s ' % self.name)
2287
Anthony King7bdac712014-07-16 12:56:40 +01002288 def _Rebase(self, upstream, onto=None):
Shawn O. Pearce19a83d82009-04-16 08:14:26 -07002289 cmd = ['rebase']
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002290 if onto is not None:
2291 cmd.extend(['--onto', onto])
2292 cmd.append(upstream)
Shawn O. Pearce19a83d82009-04-16 08:14:26 -07002293 if GitCommand(self, cmd).Wait() != 0:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002294 raise GitError('%s rebase %s ' % (self.name, upstream))
2295
Pierre Tardy3d125942012-05-04 12:18:12 +02002296 def _FastForward(self, head, ffonly=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002297 cmd = ['merge', head]
Pierre Tardy3d125942012-05-04 12:18:12 +02002298 if ffonly:
2299 cmd.append("--ff-only")
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002300 if GitCommand(self, cmd).Wait() != 0:
2301 raise GitError('%s merge %s ' % (self.name, head))
2302
Kevin Degiabaa7f32014-11-12 11:27:45 -07002303 def _InitGitDir(self, mirror_git=None, force_sync=False):
Kevin Degi384b3c52014-10-16 16:02:58 -06002304 init_git_dir = not os.path.exists(self.gitdir)
2305 init_obj_dir = not os.path.exists(self.objdir)
Kevin Degib1a07b82015-07-27 13:33:43 -06002306 try:
2307 # Initialize the bare repository, which contains all of the objects.
2308 if init_obj_dir:
2309 os.makedirs(self.objdir)
2310 self.bare_objdir.init()
David James8d201162013-10-11 17:03:19 -07002311
Kevin Degib1a07b82015-07-27 13:33:43 -06002312 # If we have a separate directory to hold refs, initialize it as well.
2313 if self.objdir != self.gitdir:
2314 if init_git_dir:
2315 os.makedirs(self.gitdir)
2316
2317 if init_obj_dir or init_git_dir:
2318 self._ReferenceGitDir(self.objdir, self.gitdir, share_refs=False,
2319 copy_all=True)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002320 try:
2321 self._CheckDirReference(self.objdir, self.gitdir, share_refs=False)
2322 except GitError as e:
Kevin Degiabaa7f32014-11-12 11:27:45 -07002323 if force_sync:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002324 print("Retrying clone after deleting %s" %
2325 self.gitdir, file=sys.stderr)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002326 try:
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002327 platform_utils.rmtree(platform_utils.realpath(self.gitdir))
2328 if self.worktree and os.path.exists(platform_utils.realpath
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002329 (self.worktree)):
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002330 platform_utils.rmtree(platform_utils.realpath(self.worktree))
Kevin Degiabaa7f32014-11-12 11:27:45 -07002331 return self._InitGitDir(mirror_git=mirror_git, force_sync=False)
2332 except:
2333 raise e
2334 raise e
Kevin Degib1a07b82015-07-27 13:33:43 -06002335
Kevin Degi384b3c52014-10-16 16:02:58 -06002336 if init_git_dir:
Kevin Degib1a07b82015-07-27 13:33:43 -06002337 mp = self.manifest.manifestProject
2338 ref_dir = mp.config.GetString('repo.reference') or ''
Kevin Degi384b3c52014-10-16 16:02:58 -06002339
Kevin Degib1a07b82015-07-27 13:33:43 -06002340 if ref_dir or mirror_git:
2341 if not mirror_git:
2342 mirror_git = os.path.join(ref_dir, self.name + '.git')
2343 repo_git = os.path.join(ref_dir, '.repo', 'projects',
2344 self.relpath + '.git')
Shawn O. Pearce2816d4f2009-03-03 17:53:18 -08002345
Kevin Degib1a07b82015-07-27 13:33:43 -06002346 if os.path.exists(mirror_git):
2347 ref_dir = mirror_git
Shawn O. Pearce88443382010-10-08 10:02:09 +02002348
Kevin Degib1a07b82015-07-27 13:33:43 -06002349 elif os.path.exists(repo_git):
2350 ref_dir = repo_git
Shawn O. Pearce88443382010-10-08 10:02:09 +02002351
Kevin Degib1a07b82015-07-27 13:33:43 -06002352 else:
2353 ref_dir = None
Shawn O. Pearce88443382010-10-08 10:02:09 +02002354
Kevin Degib1a07b82015-07-27 13:33:43 -06002355 if ref_dir:
2356 _lwrite(os.path.join(self.gitdir, 'objects/info/alternates'),
2357 os.path.join(ref_dir, 'objects') + '\n')
Shawn O. Pearce88443382010-10-08 10:02:09 +02002358
Kevin Degib1a07b82015-07-27 13:33:43 -06002359 self._UpdateHooks()
2360
2361 m = self.manifest.manifestProject.config
2362 for key in ['user.name', 'user.email']:
2363 if m.Has(key, include_defaults=False):
2364 self.config.SetString(key, m.GetString(key))
David Pursehouse76a4a9d2016-08-16 12:11:12 +09002365 self.config.SetString('filter.lfs.smudge', 'git-lfs smudge --skip -- %f')
Kevin Degib1a07b82015-07-27 13:33:43 -06002366 if self.manifest.IsMirror:
2367 self.config.SetString('core.bare', 'true')
Shawn O. Pearce88443382010-10-08 10:02:09 +02002368 else:
Kevin Degib1a07b82015-07-27 13:33:43 -06002369 self.config.SetString('core.bare', None)
2370 except Exception:
2371 if init_obj_dir and os.path.exists(self.objdir):
Renaud Paquaya65adf72016-11-03 10:37:53 -07002372 platform_utils.rmtree(self.objdir)
Kevin Degib1a07b82015-07-27 13:33:43 -06002373 if init_git_dir and os.path.exists(self.gitdir):
Renaud Paquaya65adf72016-11-03 10:37:53 -07002374 platform_utils.rmtree(self.gitdir)
Kevin Degib1a07b82015-07-27 13:33:43 -06002375 raise
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002376
Jimmie Westera0444582012-10-24 13:44:42 +02002377 def _UpdateHooks(self):
2378 if os.path.exists(self.gitdir):
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002379 self._InitHooks()
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002380
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002381 def _InitHooks(self):
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002382 hooks = platform_utils.realpath(self._gitdir_path('hooks'))
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002383 if not os.path.exists(hooks):
2384 os.makedirs(hooks)
Jonathan Nieder93719792015-03-17 11:29:58 -07002385 for stock_hook in _ProjectHooks():
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002386 name = os.path.basename(stock_hook)
2387
Victor Boivie65e0f352011-04-18 11:23:29 +02002388 if name in ('commit-msg',) and not self.remote.review \
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002389 and self is not self.manifest.manifestProject:
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002390 # Don't install a Gerrit Code Review hook if this
2391 # project does not appear to use it for reviews.
2392 #
Victor Boivie65e0f352011-04-18 11:23:29 +02002393 # Since the manifest project is one of those, but also
2394 # managed through gerrit, it's excluded
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002395 continue
2396
2397 dst = os.path.join(hooks, name)
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002398 if platform_utils.islink(dst):
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002399 continue
2400 if os.path.exists(dst):
2401 if filecmp.cmp(stock_hook, dst, shallow=False):
Renaud Paquay010fed72016-11-11 14:25:29 -08002402 platform_utils.remove(dst)
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002403 else:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002404 _warn("%s: Not replacing locally modified %s hook",
2405 self.relpath, name)
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002406 continue
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002407 try:
Renaud Paquayd5cec5e2016-11-01 11:24:03 -07002408 platform_utils.symlink(
2409 os.path.relpath(stock_hook, os.path.dirname(dst)), dst)
Sarah Owensa5be53f2012-09-09 15:37:57 -07002410 except OSError as e:
Shawn O. Pearce9452e4e2009-08-22 18:17:46 -07002411 if e.errno == errno.EPERM:
Renaud Paquay788e9622017-01-27 11:41:12 -08002412 raise GitError(self._get_symlink_error_message())
Shawn O. Pearcec9ef7442008-11-03 10:32:09 -08002413 else:
2414 raise
2415
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002416 def _InitRemote(self):
Shawn O. Pearced1f70d92009-05-19 14:58:02 -07002417 if self.remote.url:
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002418 remote = self.GetRemote(self.remote.name)
Shawn O. Pearced1f70d92009-05-19 14:58:02 -07002419 remote.url = self.remote.url
Steve Raed6480452016-08-10 15:00:00 -07002420 remote.pushUrl = self.remote.pushUrl
Shawn O. Pearced1f70d92009-05-19 14:58:02 -07002421 remote.review = self.remote.review
2422 remote.projectname = self.name
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002423
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002424 if self.worktree:
2425 remote.ResetFetch(mirror=False)
2426 else:
2427 remote.ResetFetch(mirror=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002428 remote.Save()
2429
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002430 def _InitMRef(self):
2431 if self.manifest.branch:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002432 self._InitAnyMRef(R_M + self.manifest.branch)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002433
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002434 def _InitMirrorHead(self):
Shawn O. Pearcefe200ee2009-06-01 15:28:21 -07002435 self._InitAnyMRef(HEAD)
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002436
2437 def _InitAnyMRef(self, ref):
2438 cur = self.bare_ref.symref(ref)
2439
2440 if self.revisionId:
2441 if cur != '' or self.bare_ref.get(ref) != self.revisionId:
2442 msg = 'manifest set to %s' % self.revisionId
2443 dst = self.revisionId + '^0'
Anthony King7bdac712014-07-16 12:56:40 +01002444 self.bare_git.UpdateRef(ref, dst, message=msg, detach=True)
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002445 else:
2446 remote = self.GetRemote(self.remote.name)
2447 dst = remote.ToLocal(self.revisionExpr)
2448 if cur != dst:
2449 msg = 'manifest set to %s' % self.revisionExpr
2450 self.bare_git.symbolic_ref('-m', msg, ref, dst)
Shawn O. Pearcee284ad12008-11-04 07:37:10 -08002451
Kevin Degi384b3c52014-10-16 16:02:58 -06002452 def _CheckDirReference(self, srcdir, destdir, share_refs):
Dan Willemsenbdb866e2016-04-05 17:22:02 -07002453 symlink_files = self.shareable_files[:]
2454 symlink_dirs = self.shareable_dirs[:]
Kevin Degi384b3c52014-10-16 16:02:58 -06002455 if share_refs:
2456 symlink_files += self.working_tree_files
2457 symlink_dirs += self.working_tree_dirs
2458 to_symlink = symlink_files + symlink_dirs
2459 for name in set(to_symlink):
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002460 dst = platform_utils.realpath(os.path.join(destdir, name))
Kevin Degi384b3c52014-10-16 16:02:58 -06002461 if os.path.lexists(dst):
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002462 src = platform_utils.realpath(os.path.join(srcdir, name))
Kevin Degi384b3c52014-10-16 16:02:58 -06002463 # Fail if the links are pointing to the wrong place
2464 if src != dst:
Marc Herbertec287902016-10-27 12:58:26 -07002465 _error('%s is different in %s vs %s', name, destdir, srcdir)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002466 raise GitError('--force-sync not enabled; cannot overwrite a local '
Simon Ruggierf9b76832015-07-31 17:18:34 -04002467 'work tree. If you\'re comfortable with the '
2468 'possibility of losing the work tree\'s git metadata,'
2469 ' use `repo sync --force-sync {0}` to '
2470 'proceed.'.format(self.relpath))
Kevin Degi384b3c52014-10-16 16:02:58 -06002471
David James8d201162013-10-11 17:03:19 -07002472 def _ReferenceGitDir(self, gitdir, dotgit, share_refs, copy_all):
2473 """Update |dotgit| to reference |gitdir|, using symlinks where possible.
2474
2475 Args:
2476 gitdir: The bare git repository. Must already be initialized.
2477 dotgit: The repository you would like to initialize.
2478 share_refs: If true, |dotgit| will store its refs under |gitdir|.
2479 Only one work tree can store refs under a given |gitdir|.
2480 copy_all: If true, copy all remaining files from |gitdir| -> |dotgit|.
2481 This saves you the effort of initializing |dotgit| yourself.
2482 """
Dan Willemsenbdb866e2016-04-05 17:22:02 -07002483 symlink_files = self.shareable_files[:]
2484 symlink_dirs = self.shareable_dirs[:]
David James8d201162013-10-11 17:03:19 -07002485 if share_refs:
Kevin Degi384b3c52014-10-16 16:02:58 -06002486 symlink_files += self.working_tree_files
2487 symlink_dirs += self.working_tree_dirs
David James8d201162013-10-11 17:03:19 -07002488 to_symlink = symlink_files + symlink_dirs
2489
2490 to_copy = []
2491 if copy_all:
2492 to_copy = os.listdir(gitdir)
2493
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002494 dotgit = platform_utils.realpath(dotgit)
David James8d201162013-10-11 17:03:19 -07002495 for name in set(to_copy).union(to_symlink):
2496 try:
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002497 src = platform_utils.realpath(os.path.join(gitdir, name))
Dan Willemsen2a3e1522015-07-30 20:43:33 -07002498 dst = os.path.join(dotgit, name)
David James8d201162013-10-11 17:03:19 -07002499
Kevin Degi384b3c52014-10-16 16:02:58 -06002500 if os.path.lexists(dst):
2501 continue
David James8d201162013-10-11 17:03:19 -07002502
2503 # If the source dir doesn't exist, create an empty dir.
2504 if name in symlink_dirs and not os.path.lexists(src):
2505 os.makedirs(src)
2506
Cheuk Leung8ac0c962015-07-06 21:33:00 +02002507 if name in to_symlink:
Renaud Paquayd5cec5e2016-11-01 11:24:03 -07002508 platform_utils.symlink(
2509 os.path.relpath(src, os.path.dirname(dst)), dst)
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002510 elif copy_all and not platform_utils.islink(dst):
Cheuk Leung8ac0c962015-07-06 21:33:00 +02002511 if os.path.isdir(src):
2512 shutil.copytree(src, dst)
2513 elif os.path.isfile(src):
2514 shutil.copy(src, dst)
2515
Conley Owens80b87fe2014-05-09 17:13:44 -07002516 # If the source file doesn't exist, ensure the destination
2517 # file doesn't either.
2518 if name in symlink_files and not os.path.lexists(src):
2519 try:
Renaud Paquay010fed72016-11-11 14:25:29 -08002520 platform_utils.remove(dst)
Conley Owens80b87fe2014-05-09 17:13:44 -07002521 except OSError:
2522 pass
2523
David James8d201162013-10-11 17:03:19 -07002524 except OSError as e:
2525 if e.errno == errno.EPERM:
Renaud Paquay788e9622017-01-27 11:41:12 -08002526 raise DownloadError(self._get_symlink_error_message())
David James8d201162013-10-11 17:03:19 -07002527 else:
2528 raise
2529
Martin Kellye4e94d22017-03-21 16:05:12 -07002530 def _InitWorkTree(self, force_sync=False, submodules=False):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002531 dotgit = os.path.join(self.worktree, '.git')
Kevin Degi384b3c52014-10-16 16:02:58 -06002532 init_dotgit = not os.path.exists(dotgit)
Kevin Degib1a07b82015-07-27 13:33:43 -06002533 try:
2534 if init_dotgit:
2535 os.makedirs(dotgit)
2536 self._ReferenceGitDir(self.gitdir, dotgit, share_refs=True,
2537 copy_all=False)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002538
Kevin Degiabaa7f32014-11-12 11:27:45 -07002539 try:
2540 self._CheckDirReference(self.gitdir, dotgit, share_refs=True)
2541 except GitError as e:
2542 if force_sync:
2543 try:
Renaud Paquaya65adf72016-11-03 10:37:53 -07002544 platform_utils.rmtree(dotgit)
Martin Kellye4e94d22017-03-21 16:05:12 -07002545 return self._InitWorkTree(force_sync=False, submodules=submodules)
Kevin Degiabaa7f32014-11-12 11:27:45 -07002546 except:
2547 raise e
2548 raise e
Kevin Degi384b3c52014-10-16 16:02:58 -06002549
Kevin Degib1a07b82015-07-27 13:33:43 -06002550 if init_dotgit:
2551 _lwrite(os.path.join(dotgit, HEAD), '%s\n' % self.GetRevisionId())
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002552
Kevin Degib1a07b82015-07-27 13:33:43 -06002553 cmd = ['read-tree', '--reset', '-u']
2554 cmd.append('-v')
2555 cmd.append(HEAD)
2556 if GitCommand(self, cmd).Wait() != 0:
2557 raise GitError("cannot initialize work tree")
Victor Boivie0960b5b2010-11-26 13:42:13 +01002558
Martin Kellye4e94d22017-03-21 16:05:12 -07002559 if submodules:
2560 self._SyncSubmodules(quiet=True)
Kevin Degib1a07b82015-07-27 13:33:43 -06002561 self._CopyAndLinkFiles()
2562 except Exception:
2563 if init_dotgit:
Renaud Paquaya65adf72016-11-03 10:37:53 -07002564 platform_utils.rmtree(dotgit)
Kevin Degib1a07b82015-07-27 13:33:43 -06002565 raise
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002566
Renaud Paquay788e9622017-01-27 11:41:12 -08002567 def _get_symlink_error_message(self):
2568 if platform_utils.isWindows():
2569 return ('Unable to create symbolic link. Please re-run the command as '
2570 'Administrator, or see '
2571 'https://github.com/git-for-windows/git/wiki/Symbolic-Links '
2572 'for other options.')
2573 return 'filesystem must support symlinks'
2574
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002575 def _gitdir_path(self, path):
Renaud Paquay227ad2e2016-11-01 14:37:13 -07002576 return platform_utils.realpath(os.path.join(self.gitdir, path))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002577
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07002578 def _revlist(self, *args, **kw):
2579 a = []
2580 a.extend(args)
2581 a.append('--')
2582 return self.work_git.rev_list(*a, **kw)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002583
2584 @property
2585 def _allrefs(self):
Shawn O. Pearced237b692009-04-17 18:49:50 -07002586 return self.bare_ref.all
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002587
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002588 def _getLogs(self, rev1, rev2, oneline=False, color=True, pretty_format=None):
Julien Camperguedd654222014-01-09 16:21:37 +01002589 """Get logs between two revisions of this project."""
2590 comp = '..'
2591 if rev1:
2592 revs = [rev1]
2593 if rev2:
2594 revs.extend([comp, rev2])
2595 cmd = ['log', ''.join(revs)]
2596 out = DiffColoring(self.config)
2597 if out.is_on and color:
2598 cmd.append('--color')
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002599 if pretty_format is not None:
2600 cmd.append('--pretty=format:%s' % pretty_format)
Julien Camperguedd654222014-01-09 16:21:37 +01002601 if oneline:
2602 cmd.append('--oneline')
2603
2604 try:
2605 log = GitCommand(self, cmd, capture_stdout=True, capture_stderr=True)
2606 if log.Wait() == 0:
2607 return log.stdout
2608 except GitError:
2609 # worktree may not exist if groups changed for example. In that case,
2610 # try in gitdir instead.
2611 if not os.path.exists(self.worktree):
2612 return self.bare_git.log(*cmd[1:])
2613 else:
2614 raise
2615 return None
2616
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002617 def getAddedAndRemovedLogs(self, toProject, oneline=False, color=True,
2618 pretty_format=None):
Julien Camperguedd654222014-01-09 16:21:37 +01002619 """Get the list of logs from this revision to given revisionId"""
2620 logs = {}
2621 selfId = self.GetRevisionId(self._allrefs)
2622 toId = toProject.GetRevisionId(toProject._allrefs)
2623
Sebastian Schuberth7ecccf62016-03-29 14:11:20 +02002624 logs['added'] = self._getLogs(selfId, toId, oneline=oneline, color=color,
2625 pretty_format=pretty_format)
2626 logs['removed'] = self._getLogs(toId, selfId, oneline=oneline, color=color,
2627 pretty_format=pretty_format)
Julien Camperguedd654222014-01-09 16:21:37 +01002628 return logs
2629
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002630 class _GitGetByExec(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002631
David James8d201162013-10-11 17:03:19 -07002632 def __init__(self, project, bare, gitdir):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002633 self._project = project
2634 self._bare = bare
David James8d201162013-10-11 17:03:19 -07002635 self._gitdir = gitdir
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002636
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002637 def LsOthers(self):
2638 p = GitCommand(self._project,
2639 ['ls-files',
2640 '-z',
2641 '--others',
2642 '--exclude-standard'],
Anthony King7bdac712014-07-16 12:56:40 +01002643 bare=False,
David James8d201162013-10-11 17:03:19 -07002644 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002645 capture_stdout=True,
2646 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002647 if p.Wait() == 0:
2648 out = p.stdout
2649 if out:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002650 # Backslash is not anomalous
David Pursehouse1d947b32012-10-25 12:23:11 +09002651 return out[:-1].split('\0') # pylint: disable=W1401
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002652 return []
2653
2654 def DiffZ(self, name, *args):
2655 cmd = [name]
2656 cmd.append('-z')
2657 cmd.extend(args)
2658 p = GitCommand(self._project,
2659 cmd,
David James8d201162013-10-11 17:03:19 -07002660 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002661 bare=False,
2662 capture_stdout=True,
2663 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002664 try:
2665 out = p.process.stdout.read()
2666 r = {}
2667 if out:
David Pursehouse1d947b32012-10-25 12:23:11 +09002668 out = iter(out[:-1].split('\0')) # pylint: disable=W1401
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002669 while out:
Shawn O. Pearce02dbb6d2008-10-21 13:59:08 -07002670 try:
Anthony King2cd1f042014-05-05 21:24:05 +01002671 info = next(out)
2672 path = next(out)
Shawn O. Pearce02dbb6d2008-10-21 13:59:08 -07002673 except StopIteration:
2674 break
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002675
2676 class _Info(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002677
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002678 def __init__(self, path, omode, nmode, oid, nid, state):
2679 self.path = path
2680 self.src_path = None
2681 self.old_mode = omode
2682 self.new_mode = nmode
2683 self.old_id = oid
2684 self.new_id = nid
2685
2686 if len(state) == 1:
2687 self.status = state
2688 self.level = None
2689 else:
2690 self.status = state[:1]
2691 self.level = state[1:]
2692 while self.level.startswith('0'):
2693 self.level = self.level[1:]
2694
2695 info = info[1:].split(' ')
David Pursehouse8f62fb72012-11-14 12:09:38 +09002696 info = _Info(path, *info)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002697 if info.status in ('R', 'C'):
2698 info.src_path = info.path
Anthony King2cd1f042014-05-05 21:24:05 +01002699 info.path = next(out)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002700 r[info.path] = info
2701 return r
2702 finally:
2703 p.Wait()
2704
2705 def GetHead(self):
Shawn O. Pearce5b23f242009-04-17 18:43:33 -07002706 if self._bare:
2707 path = os.path.join(self._project.gitdir, HEAD)
2708 else:
2709 path = os.path.join(self._project.worktree, '.git', HEAD)
Conley Owens75ee0572012-11-15 17:33:11 -08002710 try:
Renaud Paquay2a4be942016-11-01 13:48:15 -07002711 fd = open(path)
Dan Sandler53e902a2014-03-09 13:20:02 -04002712 except IOError as e:
2713 raise NoManifestException(path, str(e))
Shawn O. Pearce76ca9f82009-04-18 14:48:03 -07002714 try:
Renaud Paquay2a4be942016-11-01 13:48:15 -07002715 line = fd.readline()
Shawn O. Pearce76ca9f82009-04-18 14:48:03 -07002716 finally:
2717 fd.close()
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302718 try:
2719 line = line.decode()
2720 except AttributeError:
2721 pass
Shawn O. Pearce5b23f242009-04-17 18:43:33 -07002722 if line.startswith('ref: '):
2723 return line[5:-1]
2724 return line[:-1]
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002725
2726 def SetHead(self, ref, message=None):
2727 cmdv = []
2728 if message is not None:
2729 cmdv.extend(['-m', message])
2730 cmdv.append(HEAD)
2731 cmdv.append(ref)
2732 self.symbolic_ref(*cmdv)
2733
2734 def DetachHead(self, new, message=None):
2735 cmdv = ['--no-deref']
2736 if message is not None:
2737 cmdv.extend(['-m', message])
2738 cmdv.append(HEAD)
2739 cmdv.append(new)
2740 self.update_ref(*cmdv)
2741
2742 def UpdateRef(self, name, new, old=None,
2743 message=None,
2744 detach=False):
2745 cmdv = []
2746 if message is not None:
2747 cmdv.extend(['-m', message])
2748 if detach:
2749 cmdv.append('--no-deref')
2750 cmdv.append(name)
2751 cmdv.append(new)
2752 if old is not None:
2753 cmdv.append(old)
2754 self.update_ref(*cmdv)
2755
2756 def DeleteRef(self, name, old=None):
2757 if not old:
2758 old = self.rev_parse(name)
2759 self.update_ref('-d', name, old)
Shawn O. Pearcefbcde472009-04-17 20:58:02 -07002760 self._project.bare_ref.deleted(name)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002761
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07002762 def rev_list(self, *args, **kw):
2763 if 'format' in kw:
2764 cmdv = ['log', '--pretty=format:%s' % kw['format']]
2765 else:
2766 cmdv = ['rev-list']
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002767 cmdv.extend(args)
2768 p = GitCommand(self._project,
2769 cmdv,
Anthony King7bdac712014-07-16 12:56:40 +01002770 bare=self._bare,
David James8d201162013-10-11 17:03:19 -07002771 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002772 capture_stdout=True,
2773 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002774 r = []
2775 for line in p.process.stdout:
Shawn O. Pearce8ad8a0e2009-05-29 18:28:25 -07002776 if line[-1] == '\n':
2777 line = line[:-1]
2778 r.append(line)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002779 if p.Wait() != 0:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002780 raise GitError('%s rev-list %s: %s' %
2781 (self._project.name, str(args), p.stderr))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002782 return r
2783
2784 def __getattr__(self, name):
Doug Anderson37282b42011-03-04 11:54:18 -08002785 """Allow arbitrary git commands using pythonic syntax.
2786
2787 This allows you to do things like:
2788 git_obj.rev_parse('HEAD')
2789
2790 Since we don't have a 'rev_parse' method defined, the __getattr__ will
2791 run. We'll replace the '_' with a '-' and try to run a git command.
Dave Borowitz091f8932012-10-23 17:01:04 -07002792 Any other positional arguments will be passed to the git command, and the
2793 following keyword arguments are supported:
2794 config: An optional dict of git config options to be passed with '-c'.
Doug Anderson37282b42011-03-04 11:54:18 -08002795
2796 Args:
2797 name: The name of the git command to call. Any '_' characters will
2798 be replaced with '-'.
2799
2800 Returns:
2801 A callable object that will try to call git with the named command.
2802 """
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002803 name = name.replace('_', '-')
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002804
Dave Borowitz091f8932012-10-23 17:01:04 -07002805 def runner(*args, **kwargs):
2806 cmdv = []
2807 config = kwargs.pop('config', None)
2808 for k in kwargs:
2809 raise TypeError('%s() got an unexpected keyword argument %r'
2810 % (name, k))
2811 if config is not None:
Dave Borowitzb42b4742012-10-31 12:27:27 -07002812 if not git_require((1, 7, 2)):
2813 raise ValueError('cannot set config on command line for %s()'
2814 % name)
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302815 for k, v in config.items():
Dave Borowitz091f8932012-10-23 17:01:04 -07002816 cmdv.append('-c')
2817 cmdv.append('%s=%s' % (k, v))
2818 cmdv.append(name)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002819 cmdv.extend(args)
2820 p = GitCommand(self._project,
2821 cmdv,
Anthony King7bdac712014-07-16 12:56:40 +01002822 bare=self._bare,
David James8d201162013-10-11 17:03:19 -07002823 gitdir=self._gitdir,
Anthony King7bdac712014-07-16 12:56:40 +01002824 capture_stdout=True,
2825 capture_stderr=True)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002826 if p.Wait() != 0:
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002827 raise GitError('%s %s: %s' %
2828 (self._project.name, name, p.stderr))
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002829 r = p.stdout
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302830 try:
Conley Owensedd01512013-09-26 12:59:58 -07002831 r = r.decode('utf-8')
Chirayu Desai217ea7d2013-03-01 19:14:38 +05302832 except AttributeError:
2833 pass
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002834 if r.endswith('\n') and r.index('\n') == len(r) - 1:
2835 return r[:-1]
2836 return r
2837 return runner
2838
2839
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002840class _PriorSyncFailedError(Exception):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002841
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002842 def __str__(self):
2843 return 'prior sync failed; rebase still in progress'
2844
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002845
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002846class _DirtyError(Exception):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002847
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002848 def __str__(self):
2849 return 'contains uncommitted changes'
2850
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002851
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002852class _InfoMessage(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002853
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002854 def __init__(self, project, text):
2855 self.project = project
2856 self.text = text
2857
2858 def Print(self, syncbuf):
2859 syncbuf.out.info('%s/: %s', self.project.relpath, self.text)
2860 syncbuf.out.nl()
2861
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002862
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002863class _Failure(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002864
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002865 def __init__(self, project, why):
2866 self.project = project
2867 self.why = why
2868
2869 def Print(self, syncbuf):
2870 syncbuf.out.fail('error: %s/: %s',
2871 self.project.relpath,
2872 str(self.why))
2873 syncbuf.out.nl()
2874
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002875
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002876class _Later(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002877
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002878 def __init__(self, project, action):
2879 self.project = project
2880 self.action = action
2881
2882 def Run(self, syncbuf):
2883 out = syncbuf.out
2884 out.project('project %s/', self.project.relpath)
2885 out.nl()
2886 try:
2887 self.action()
2888 out.nl()
2889 return True
David Pursehouse8a68ff92012-09-24 12:15:13 +09002890 except GitError:
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002891 out.nl()
2892 return False
2893
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002894
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002895class _SyncColoring(Coloring):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002896
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002897 def __init__(self, config):
2898 Coloring.__init__(self, config, 'reposync')
Anthony King7bdac712014-07-16 12:56:40 +01002899 self.project = self.printer('header', attr='bold')
2900 self.info = self.printer('info')
2901 self.fail = self.printer('fail', fg='red')
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002902
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002903
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002904class SyncBuffer(object):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002905
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002906 def __init__(self, config, detach_head=False):
2907 self._messages = []
2908 self._failures = []
2909 self._later_queue1 = []
2910 self._later_queue2 = []
2911
2912 self.out = _SyncColoring(config)
2913 self.out.redirect(sys.stderr)
2914
2915 self.detach_head = detach_head
2916 self.clean = True
David Rileye0684ad2017-04-05 00:02:59 -07002917 self.recent_clean = True
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002918
2919 def info(self, project, fmt, *args):
2920 self._messages.append(_InfoMessage(project, fmt % args))
2921
2922 def fail(self, project, err=None):
2923 self._failures.append(_Failure(project, err))
David Rileye0684ad2017-04-05 00:02:59 -07002924 self._MarkUnclean()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002925
2926 def later1(self, project, what):
2927 self._later_queue1.append(_Later(project, what))
2928
2929 def later2(self, project, what):
2930 self._later_queue2.append(_Later(project, what))
2931
2932 def Finish(self):
2933 self._PrintMessages()
2934 self._RunLater()
2935 self._PrintMessages()
2936 return self.clean
2937
David Rileye0684ad2017-04-05 00:02:59 -07002938 def Recently(self):
2939 recent_clean = self.recent_clean
2940 self.recent_clean = True
2941 return recent_clean
2942
2943 def _MarkUnclean(self):
2944 self.clean = False
2945 self.recent_clean = False
2946
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002947 def _RunLater(self):
2948 for q in ['_later_queue1', '_later_queue2']:
2949 if not self._RunQueue(q):
2950 return
2951
2952 def _RunQueue(self, queue):
2953 for m in getattr(self, queue):
2954 if not m.Run(self):
David Rileye0684ad2017-04-05 00:02:59 -07002955 self._MarkUnclean()
Shawn O. Pearce350cde42009-04-16 11:21:18 -07002956 return False
2957 setattr(self, queue, [])
2958 return True
2959
2960 def _PrintMessages(self):
2961 for m in self._messages:
2962 m.Print(self)
2963 for m in self._failures:
2964 m.Print(self)
2965
2966 self._messages = []
2967 self._failures = []
2968
2969
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002970class MetaProject(Project):
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002971
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002972 """A special project housed under .repo.
2973 """
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07002974
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002975 def __init__(self, manifest, name, gitdir, worktree):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002976 Project.__init__(self,
Anthony King7bdac712014-07-16 12:56:40 +01002977 manifest=manifest,
2978 name=name,
2979 gitdir=gitdir,
2980 objdir=gitdir,
2981 worktree=worktree,
2982 remote=RemoteSpec('origin'),
2983 relpath='.repo/%s' % name,
2984 revisionExpr='refs/heads/master',
2985 revisionId=None,
2986 groups=None)
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002987
2988 def PreSync(self):
2989 if self.Exists:
2990 cb = self.CurrentBranch
2991 if cb:
2992 base = self.GetBranch(cb).merge
2993 if base:
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07002994 self.revisionExpr = base
2995 self.revisionId = None
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07002996
Martin Kelly224a31a2017-07-10 14:46:25 -07002997 def MetaBranchSwitch(self, submodules=False):
Florian Vallee5d016502012-06-07 17:19:26 +02002998 """ Prepare MetaProject for manifest branch switch
2999 """
3000
3001 # detach and delete manifest branch, allowing a new
3002 # branch to take over
Anthony King7bdac712014-07-16 12:56:40 +01003003 syncbuf = SyncBuffer(self.config, detach_head=True)
Martin Kelly224a31a2017-07-10 14:46:25 -07003004 self.Sync_LocalHalf(syncbuf, submodules=submodules)
Florian Vallee5d016502012-06-07 17:19:26 +02003005 syncbuf.Finish()
3006
3007 return GitCommand(self,
Mark E. Hamilton30b0f4e2016-02-10 10:44:30 -07003008 ['update-ref', '-d', 'refs/heads/default'],
3009 capture_stdout=True,
3010 capture_stderr=True).Wait() == 0
Florian Vallee5d016502012-06-07 17:19:26 +02003011
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07003012 @property
Shawn O. Pearcef6906872009-04-18 10:49:00 -07003013 def LastFetch(self):
3014 try:
3015 fh = os.path.join(self.gitdir, 'FETCH_HEAD')
3016 return os.path.getmtime(fh)
3017 except OSError:
3018 return 0
3019
3020 @property
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07003021 def HasChanges(self):
3022 """Has the remote received new commits not yet checked out?
3023 """
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07003024 if not self.remote or not self.revisionExpr:
Shawn O. Pearce336f7bd2009-04-18 10:39:28 -07003025 return False
3026
David Pursehouse8a68ff92012-09-24 12:15:13 +09003027 all_refs = self.bare_ref.all
3028 revid = self.GetRevisionId(all_refs)
Shawn O. Pearce336f7bd2009-04-18 10:39:28 -07003029 head = self.work_git.GetHead()
3030 if head.startswith(R_HEADS):
3031 try:
David Pursehouse8a68ff92012-09-24 12:15:13 +09003032 head = all_refs[head]
Shawn O. Pearce336f7bd2009-04-18 10:39:28 -07003033 except KeyError:
3034 head = None
3035
3036 if revid == head:
3037 return False
Shawn O. Pearce3c8dea12009-05-29 18:38:17 -07003038 elif self._revlist(not_rev(HEAD), revid):
The Android Open Source Projectcf31fe92008-10-21 07:00:00 -07003039 return True
3040 return False