blob: 5045f706d5549b1da43717c54871419cb7517438 [file] [log] [blame]
steveblock@chromium.org93567042012-02-15 01:02:26 +00001# Copyright (c) 2012 The Chromium Authors. All rights reserved.
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +00002# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00004
maruel@chromium.orgd5800f12009-11-12 20:03:43 +00005"""Gclient-specific SCM-specific operations."""
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00006
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00007from __future__ import print_function
8
borenet@google.comb2256212014-05-07 20:57:28 +00009import errno
maruel@chromium.org754960e2009-09-21 12:31:05 +000010import logging
maruel@chromium.org5f3eee32009-09-17 00:34:30 +000011import os
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000012import posixpath
maruel@chromium.org5f3eee32009-09-17 00:34:30 +000013import re
borenet@google.com88d10082014-03-21 17:24:48 +000014import shlex
maruel@chromium.org90541732011-04-01 17:54:18 +000015import sys
ilevy@chromium.org3534aa52013-07-20 01:58:08 +000016import tempfile
zty@chromium.org6279e8a2014-02-13 01:45:25 +000017import traceback
hinoka@google.com2f2ca142014-01-07 03:59:18 +000018import urlparse
maruel@chromium.org5f3eee32009-09-17 00:34:30 +000019
hinoka@google.com2f2ca142014-01-07 03:59:18 +000020import download_from_google_storage
maruel@chromium.org5f3eee32009-09-17 00:34:30 +000021import gclient_utils
szager@chromium.org848fd492014-04-09 19:06:44 +000022import git_cache
maruel@chromium.org31cb48a2011-04-04 18:01:36 +000023import scm
borenet@google.comb2256212014-05-07 20:57:28 +000024import shutil
maruel@chromium.org31cb48a2011-04-04 18:01:36 +000025import subprocess2
maruel@chromium.org5f3eee32009-09-17 00:34:30 +000026
27
szager@chromium.org71cbb502013-04-19 23:30:15 +000028THIS_FILE_PATH = os.path.abspath(__file__)
29
hinoka@google.com2f2ca142014-01-07 03:59:18 +000030GSUTIL_DEFAULT_PATH = os.path.join(
31 os.path.dirname(os.path.abspath(__file__)),
32 'third_party', 'gsutil', 'gsutil')
33
szager@chromium.org848fd492014-04-09 19:06:44 +000034CHROMIUM_SRC_URL = 'https://chromium.googlesource.com/chromium/src.git'
haitao.feng@intel.com306080c2012-05-04 13:11:29 +000035class DiffFiltererWrapper(object):
36 """Simple base class which tracks which file is being diffed and
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000037 replaces instances of its file name in the original and
msb@chromium.orgd6504212010-01-13 17:34:31 +000038 working copy lines of the svn/git diff output."""
haitao.feng@intel.com306080c2012-05-04 13:11:29 +000039 index_string = None
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000040 original_prefix = "--- "
41 working_prefix = "+++ "
42
szager@chromium.orgfe0d1902014-04-08 20:50:44 +000043 def __init__(self, relpath, print_func):
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000044 # Note that we always use '/' as the path separator to be
45 # consistent with svn's cygwin-style output on Windows
46 self._relpath = relpath.replace("\\", "/")
haitao.feng@intel.com306080c2012-05-04 13:11:29 +000047 self._current_file = None
szager@chromium.orgfe0d1902014-04-08 20:50:44 +000048 self._print_func = print_func
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000049
maruel@chromium.org6e29d572010-06-04 17:32:20 +000050 def SetCurrentFile(self, current_file):
51 self._current_file = current_file
haitao.feng@intel.com306080c2012-05-04 13:11:29 +000052
iannucci@chromium.org3830a672013-02-19 20:15:14 +000053 @property
54 def _replacement_file(self):
haitao.feng@intel.com306080c2012-05-04 13:11:29 +000055 return posixpath.join(self._relpath, self._current_file)
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000056
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +000057 def _Replace(self, line):
58 return line.replace(self._current_file, self._replacement_file)
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000059
60 def Filter(self, line):
61 if (line.startswith(self.index_string)):
62 self.SetCurrentFile(line[len(self.index_string):])
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +000063 line = self._Replace(line)
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000064 else:
65 if (line.startswith(self.original_prefix) or
66 line.startswith(self.working_prefix)):
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +000067 line = self._Replace(line)
szager@chromium.orgfe0d1902014-04-08 20:50:44 +000068 self._print_func(line)
maruel@chromium.orgee4071d2009-12-22 22:25:37 +000069
70
haitao.feng@intel.com306080c2012-05-04 13:11:29 +000071class SvnDiffFilterer(DiffFiltererWrapper):
72 index_string = "Index: "
73
74
75class GitDiffFilterer(DiffFiltererWrapper):
76 index_string = "diff --git "
77
78 def SetCurrentFile(self, current_file):
79 # Get filename by parsing "a/<filename> b/<filename>"
80 self._current_file = current_file[:(len(current_file)/2)][2:]
81
82 def _Replace(self, line):
83 return re.sub("[a|b]/" + self._current_file, self._replacement_file, line)
84
85
maruel@chromium.org5f3eee32009-09-17 00:34:30 +000086### SCM abstraction layer
87
msb@chromium.orgcb5442b2009-09-22 16:51:24 +000088# Factory Method for SCM wrapper creation
89
maruel@chromium.org9eda4112010-06-11 18:56:10 +000090def GetScmName(url):
91 if url:
92 url, _ = gclient_utils.SplitUrlRevision(url)
93 if (url.startswith('git://') or url.startswith('ssh://') or
igorgatis@gmail.com4e075672011-11-21 16:35:08 +000094 url.startswith('git+http://') or url.startswith('git+https://') or
borenet@google.coma2aec972014-04-22 21:35:15 +000095 url.endswith('.git') or url.startswith('sso://') or
96 'googlesource' in url):
maruel@chromium.org9eda4112010-06-11 18:56:10 +000097 return 'git'
maruel@chromium.orgb74dca22010-06-11 20:10:40 +000098 elif (url.startswith('http://') or url.startswith('https://') or
maruel@chromium.org54a07a22010-06-14 19:07:39 +000099 url.startswith('svn://') or url.startswith('svn+ssh://')):
maruel@chromium.org9eda4112010-06-11 18:56:10 +0000100 return 'svn'
101 return None
102
103
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000104def CreateSCM(url, root_dir=None, relpath=None, out_fh=None, out_cb=None):
maruel@chromium.org9eda4112010-06-11 18:56:10 +0000105 SCM_MAP = {
msb@chromium.orgcb5442b2009-09-22 16:51:24 +0000106 'svn' : SVNWrapper,
msb@chromium.orge28e4982009-09-25 20:51:45 +0000107 'git' : GitWrapper,
msb@chromium.orgcb5442b2009-09-22 16:51:24 +0000108 }
msb@chromium.orge28e4982009-09-25 20:51:45 +0000109
maruel@chromium.org9eda4112010-06-11 18:56:10 +0000110 scm_name = GetScmName(url)
111 if not scm_name in SCM_MAP:
112 raise gclient_utils.Error('No SCM found for url %s' % url)
mukai@chromium.org9e3e82c2012-04-18 12:55:43 +0000113 scm_class = SCM_MAP[scm_name]
114 if not scm_class.BinaryExists():
115 raise gclient_utils.Error('%s command not found' % scm_name)
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000116 return scm_class(url, root_dir, relpath, out_fh, out_cb)
msb@chromium.orgcb5442b2009-09-22 16:51:24 +0000117
118
119# SCMWrapper base class
120
maruel@chromium.org5f3eee32009-09-17 00:34:30 +0000121class SCMWrapper(object):
122 """Add necessary glue between all the supported SCM.
123
msb@chromium.orgd6504212010-01-13 17:34:31 +0000124 This is the abstraction layer to bind to different SCM.
125 """
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000126
127 def __init__(self, url=None, root_dir=None, relpath=None, out_fh=None,
128 out_cb=None):
maruel@chromium.org5f3eee32009-09-17 00:34:30 +0000129 self.url = url
maruel@chromium.org5e73b0c2009-09-18 19:47:48 +0000130 self._root_dir = root_dir
131 if self._root_dir:
132 self._root_dir = self._root_dir.replace('/', os.sep)
133 self.relpath = relpath
134 if self.relpath:
135 self.relpath = self.relpath.replace('/', os.sep)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000136 if self.relpath and self._root_dir:
137 self.checkout_path = os.path.join(self._root_dir, self.relpath)
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000138 if out_fh is None:
139 out_fh = sys.stdout
140 self.out_fh = out_fh
141 self.out_cb = out_cb
142
143 def Print(self, *args, **kwargs):
144 kwargs.setdefault('file', self.out_fh)
145 if kwargs.pop('timestamp', True):
146 self.out_fh.write('[%s] ' % gclient_utils.Elapsed())
147 print(*args, **kwargs)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +0000148
maruel@chromium.org5f3eee32009-09-17 00:34:30 +0000149 def RunCommand(self, command, options, args, file_list=None):
phajdan.jr@chromium.org6e043f72011-05-02 07:24:32 +0000150 commands = ['cleanup', 'update', 'updatesingle', 'revert',
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000151 'revinfo', 'status', 'diff', 'pack', 'runhooks']
maruel@chromium.org5f3eee32009-09-17 00:34:30 +0000152
153 if not command in commands:
154 raise gclient_utils.Error('Unknown command %s' % command)
155
msb@chromium.orgcb5442b2009-09-22 16:51:24 +0000156 if not command in dir(self):
maruel@chromium.orgee4071d2009-12-22 22:25:37 +0000157 raise gclient_utils.Error('Command %s not implemented in %s wrapper' % (
maruel@chromium.org9eda4112010-06-11 18:56:10 +0000158 command, self.__class__.__name__))
msb@chromium.orgcb5442b2009-09-22 16:51:24 +0000159
160 return getattr(self, command)(options, args, file_list)
161
borenet@google.com4e9be262014-04-08 19:40:30 +0000162 def GetActualRemoteURL(self, options):
borenet@google.com88d10082014-03-21 17:24:48 +0000163 """Attempt to determine the remote URL for this SCMWrapper."""
borenet@google.com4e9be262014-04-08 19:40:30 +0000164 # Git
borenet@google.combda475e2014-03-24 19:04:45 +0000165 if os.path.exists(os.path.join(self.checkout_path, '.git')):
szager@chromium.orgce2fccb2014-05-02 00:57:10 +0000166 actual_remote_url = shlex.split(scm.GIT.Capture(
borenet@google.com88d10082014-03-21 17:24:48 +0000167 ['config', '--local', '--get-regexp', r'remote.*.url'],
borenet@google.comc3e09d22014-04-10 13:58:18 +0000168 cwd=self.checkout_path))[1]
borenet@google.com4e9be262014-04-08 19:40:30 +0000169
170 # If a cache_dir is used, obtain the actual remote URL from the cache.
171 if getattr(self, 'cache_dir', None):
borenet@google.com46d09f62014-04-10 18:50:23 +0000172 url, _ = gclient_utils.SplitUrlRevision(self.url)
173 mirror = git_cache.Mirror(url)
szager@chromium.org848fd492014-04-09 19:06:44 +0000174 if (mirror.exists() and mirror.mirror_path.replace('\\', '/') ==
borenet@google.com4e9be262014-04-08 19:40:30 +0000175 actual_remote_url.replace('\\', '/')):
szager@chromium.orgce2fccb2014-05-02 00:57:10 +0000176 actual_remote_url = shlex.split(scm.GIT.Capture(
borenet@google.com4e9be262014-04-08 19:40:30 +0000177 ['config', '--local', '--get-regexp', r'remote.*.url'],
szager@chromium.org848fd492014-04-09 19:06:44 +0000178 cwd=mirror.mirror_path))[1]
borenet@google.com4e9be262014-04-08 19:40:30 +0000179 return actual_remote_url
180
181 # Svn
borenet@google.combda475e2014-03-24 19:04:45 +0000182 if os.path.exists(os.path.join(self.checkout_path, '.svn')):
borenet@google.com88d10082014-03-21 17:24:48 +0000183 return scm.SVN.CaptureLocalInfo([], self.checkout_path)['URL']
borenet@google.com88d10082014-03-21 17:24:48 +0000184 return None
185
borenet@google.com4e9be262014-04-08 19:40:30 +0000186 def DoesRemoteURLMatch(self, options):
borenet@google.com88d10082014-03-21 17:24:48 +0000187 """Determine whether the remote URL of this checkout is the expected URL."""
188 if not os.path.exists(self.checkout_path):
189 # A checkout which doesn't exist can't be broken.
190 return True
191
borenet@google.com4e9be262014-04-08 19:40:30 +0000192 actual_remote_url = self.GetActualRemoteURL(options)
borenet@google.com88d10082014-03-21 17:24:48 +0000193 if actual_remote_url:
borenet@google.com8156c9f2014-04-01 16:41:36 +0000194 return (gclient_utils.SplitUrlRevision(actual_remote_url)[0].rstrip('/')
195 == gclient_utils.SplitUrlRevision(self.url)[0].rstrip('/'))
borenet@google.com88d10082014-03-21 17:24:48 +0000196 else:
197 # This may occur if the self.checkout_path exists but does not contain a
198 # valid git or svn checkout.
199 return False
200
borenet@google.comb09097a2014-04-09 19:09:08 +0000201 def _DeleteOrMove(self, force):
202 """Delete the checkout directory or move it out of the way.
203
204 Args:
205 force: bool; if True, delete the directory. Otherwise, just move it.
206 """
borenet@google.comb2256212014-05-07 20:57:28 +0000207 if force and os.environ.get('CHROME_HEADLESS') == '1':
208 self.Print('_____ Conflicting directory found in %s. Removing.'
209 % self.checkout_path)
210 gclient_utils.AddWarning('Conflicting directory %s deleted.'
211 % self.checkout_path)
212 gclient_utils.rmtree(self.checkout_path)
213 else:
214 bad_scm_dir = os.path.join(self._root_dir, '_bad_scm',
215 os.path.dirname(self.relpath))
216
217 try:
218 os.makedirs(bad_scm_dir)
219 except OSError as e:
220 if e.errno != errno.EEXIST:
221 raise
222
223 dest_path = tempfile.mkdtemp(
224 prefix=os.path.basename(self.relpath),
225 dir=bad_scm_dir)
226 self.Print('_____ Conflicting directory found in %s. Moving to %s.'
227 % (self.checkout_path, dest_path))
228 gclient_utils.AddWarning('Conflicting directory %s moved to %s.'
229 % (self.checkout_path, dest_path))
230 shutil.move(self.checkout_path, dest_path)
borenet@google.comb09097a2014-04-09 19:09:08 +0000231
msb@chromium.orgcb5442b2009-09-22 16:51:24 +0000232
maruel@chromium.org55e724e2010-03-11 19:36:49 +0000233class GitWrapper(SCMWrapper):
msb@chromium.orge28e4982009-09-25 20:51:45 +0000234 """Wrapper for Git"""
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +0000235 name = 'git'
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000236 remote = 'origin'
msb@chromium.orge28e4982009-09-25 20:51:45 +0000237
iannucci@chromium.org53456aa2013-07-03 19:38:34 +0000238 cache_dir = None
iannucci@chromium.org53456aa2013-07-03 19:38:34 +0000239
szager@chromium.org848fd492014-04-09 19:06:44 +0000240 def __init__(self, url=None, *args):
igorgatis@gmail.com4e075672011-11-21 16:35:08 +0000241 """Removes 'git+' fake prefix from git URL."""
242 if url.startswith('git+http://') or url.startswith('git+https://'):
243 url = url[4:]
szager@chromium.org848fd492014-04-09 19:06:44 +0000244 SCMWrapper.__init__(self, url, *args)
245 filter_kwargs = { 'time_throttle': 1, 'out_fh': self.out_fh }
246 if self.out_cb:
247 filter_kwargs['predicate'] = self.out_cb
248 self.filter = gclient_utils.GitFilter(**filter_kwargs)
igorgatis@gmail.com4e075672011-11-21 16:35:08 +0000249
mukai@chromium.org9e3e82c2012-04-18 12:55:43 +0000250 @staticmethod
251 def BinaryExists():
252 """Returns true if the command exists."""
253 try:
254 # We assume git is newer than 1.7. See: crbug.com/114483
255 result, version = scm.GIT.AssertVersion('1.7')
256 if not result:
257 raise gclient_utils.Error('Git version is older than 1.7: %s' % version)
258 return result
259 except OSError:
260 return False
261
xusydoc@chromium.org885a9602013-05-31 09:54:40 +0000262 def GetCheckoutRoot(self):
263 return scm.GIT.GetCheckoutRoot(self.checkout_path)
264
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000265 def GetRevisionDate(self, _revision):
floitsch@google.comeaab7842011-04-28 09:07:58 +0000266 """Returns the given revision's date in ISO-8601 format (which contains the
267 time zone)."""
268 # TODO(floitsch): get the time-stamp of the given revision and not just the
269 # time-stamp of the currently checked out revision.
270 return self._Capture(['log', '-n', '1', '--format=%ai'])
271
maruel@chromium.org6e29d572010-06-04 17:32:20 +0000272 @staticmethod
273 def cleanup(options, args, file_list):
msb@chromium.orgd8a63782010-01-25 17:47:05 +0000274 """'Cleanup' the repo.
275
276 There's no real git equivalent for the svn cleanup command, do a no-op.
277 """
msb@chromium.orge28e4982009-09-25 20:51:45 +0000278
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000279 def diff(self, options, _args, _file_list):
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000280 merge_base = self._Capture(['merge-base', 'HEAD', self.remote])
maruel@chromium.org37e89872010-09-07 16:11:33 +0000281 self._Run(['diff', merge_base], options)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000282
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000283 def pack(self, _options, _args, _file_list):
maruel@chromium.orgee4071d2009-12-22 22:25:37 +0000284 """Generates a patch file which can be applied to the root of the
msb@chromium.orgd6504212010-01-13 17:34:31 +0000285 repository.
286
287 The patch file is generated from a diff of the merge base of HEAD and
288 its upstream branch.
289 """
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000290 merge_base = self._Capture(['merge-base', 'HEAD', self.remote])
maruel@chromium.org17d01792010-09-01 18:07:10 +0000291 gclient_utils.CheckCallAndFilter(
maruel@chromium.org8469bf92010-09-03 19:03:15 +0000292 ['git', 'diff', merge_base],
293 cwd=self.checkout_path,
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000294 filter_fn=GitDiffFilterer(self.relpath).Filter, print_func=self.Print)
maruel@chromium.orgee4071d2009-12-22 22:25:37 +0000295
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +0000296 def _FetchAndReset(self, revision, file_list, options):
297 """Equivalent to git fetch; git reset."""
298 quiet = []
299 if not options.verbose:
300 quiet = ['--quiet']
301 self._UpdateBranchHeads(options, fetch=False)
302
szager@chromium.orgff113292014-03-25 06:02:08 +0000303 cfg = gclient_utils.DefaultIndexPackConfig(self.url)
szager@chromium.orgfc616382014-03-18 20:32:04 +0000304 fetch_cmd = cfg + ['fetch', self.remote, '--prune']
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +0000305 self._Run(fetch_cmd + quiet, options, retry=True)
306 self._Run(['reset', '--hard', revision] + quiet, options)
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +0000307 if file_list is not None:
308 files = self._Capture(['ls-files']).splitlines()
309 file_list.extend([os.path.join(self.checkout_path, f) for f in files])
310
szager@chromium.org8a139702014-06-20 15:55:01 +0000311 def _DisableHooks(self):
312 hook_dir = os.path.join(self.checkout_path, '.git', 'hooks')
313 if not os.path.isdir(hook_dir):
314 return
315 for f in os.listdir(hook_dir):
316 if not f.endswith('.sample') and not f.endswith('.disabled'):
317 os.rename(os.path.join(hook_dir, f),
318 os.path.join(hook_dir, f + '.disabled'))
319
msb@chromium.orge28e4982009-09-25 20:51:45 +0000320 def update(self, options, args, file_list):
321 """Runs git to update or transparently checkout the working copy.
322
323 All updated files will be appended to file_list.
324
325 Raises:
326 Error: if can't get URL for relative path.
327 """
msb@chromium.orge28e4982009-09-25 20:51:45 +0000328 if args:
329 raise gclient_utils.Error("Unsupported argument(s): %s" % ",".join(args))
330
nasser@codeaurora.orgece406f2010-02-23 17:29:15 +0000331 self._CheckMinVersion("1.6.6")
msb@chromium.org923a0372009-12-11 20:42:43 +0000332
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000333 # If a dependency is not pinned, track the default remote branch.
334 default_rev = 'refs/remotes/%s/master' % self.remote
nasser@codeaurora.org7080e942010-03-15 15:06:16 +0000335 url, deps_revision = gclient_utils.SplitUrlRevision(self.url)
msb@chromium.orgac915bb2009-11-13 17:03:01 +0000336 rev_str = ""
nasser@codeaurora.org7080e942010-03-15 15:06:16 +0000337 revision = deps_revision
cmp@chromium.orgeb2756d2011-09-20 20:17:51 +0000338 managed = True
msb@chromium.orge28e4982009-09-25 20:51:45 +0000339 if options.revision:
msb@chromium.orgac915bb2009-11-13 17:03:01 +0000340 # Override the revision number.
341 revision = str(options.revision)
cmp@chromium.orgeb2756d2011-09-20 20:17:51 +0000342 if revision == 'unmanaged':
romain.pokrzywka@gmail.com483a0ba2014-05-30 00:06:07 +0000343 # Check again for a revision in case an initial ref was specified
344 # in the url, for example bla.git@refs/heads/custombranch
345 revision = deps_revision
cmp@chromium.orgeb2756d2011-09-20 20:17:51 +0000346 managed = False
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000347 if not revision:
348 revision = default_rev
msb@chromium.orge28e4982009-09-25 20:51:45 +0000349
szager@chromium.org8a139702014-06-20 15:55:01 +0000350 if managed:
351 self._DisableHooks()
352
floitsch@google.comeaab7842011-04-28 09:07:58 +0000353 if gclient_utils.IsDateRevision(revision):
354 # Date-revisions only work on git-repositories if the reflog hasn't
355 # expired yet. Use rev-list to get the corresponding revision.
356 # git rev-list -n 1 --before='time-stamp' branchname
357 if options.transitive:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000358 self.Print('Warning: --transitive only works for SVN repositories.')
floitsch@google.comeaab7842011-04-28 09:07:58 +0000359 revision = default_rev
360
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000361 rev_str = ' at %s' % revision
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000362 files = [] if file_list is not None else None
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000363
364 printed_path = False
365 verbose = []
msb@chromium.orgb1a22bf2009-11-07 02:33:50 +0000366 if options.verbose:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000367 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000368 verbose = ['--verbose']
369 printed_path = True
370
ilevy@chromium.orga41249c2013-07-03 00:09:12 +0000371 if revision.startswith('refs/'):
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000372 rev_type = "branch"
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000373 elif revision.startswith(self.remote + '/'):
agable@chromium.org69995f42014-04-25 21:43:29 +0000374 # Rewrite remote refs to their local equivalents.
375 revision = 'refs/remotes/' + revision
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000376 rev_type = "branch"
377 else:
378 # hash is also a tag, only make a distinction at checkout
379 rev_type = "hash"
380
szager@chromium.orgb0a13a22014-06-18 00:52:25 +0000381 mirror = self._GetMirror(url, options)
382 if mirror:
383 url = mirror.mirror_path
384
szager@chromium.org6c2b49d2014-02-26 23:57:38 +0000385 if (not os.path.exists(self.checkout_path) or
386 (os.path.isdir(self.checkout_path) and
387 not os.path.exists(os.path.join(self.checkout_path, '.git')))):
szager@chromium.orgb0a13a22014-06-18 00:52:25 +0000388 if mirror:
389 self._UpdateMirror(mirror, options)
borenet@google.com90fe58b2014-05-01 18:22:00 +0000390 try:
391 self._Clone(revision, url, options)
392 except subprocess2.CalledProcessError:
393 self._DeleteOrMove(options.force)
394 self._Clone(revision, url, options)
jochen@chromium.org048da082014-05-06 08:32:40 +0000395 if deps_revision and deps_revision.startswith('branch-heads/'):
396 deps_branch = deps_revision.replace('branch-heads/', '')
397 self._Capture(['branch', deps_branch, deps_revision])
398 self._Capture(['checkout', '--quiet', deps_branch])
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000399 if file_list is not None:
400 files = self._Capture(['ls-files']).splitlines()
401 file_list.extend([os.path.join(self.checkout_path, f) for f in files])
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000402 if not verbose:
403 # Make the output a little prettier. It's nice to have some whitespace
404 # between projects when cloning.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000405 self.Print('')
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +0000406 return self._Capture(['rev-parse', '--verify', 'HEAD'])
msb@chromium.orge28e4982009-09-25 20:51:45 +0000407
szager@chromium.org3dc5cb72014-06-17 15:06:05 +0000408 if not managed:
409 self._UpdateBranchHeads(options, fetch=False)
410 self.Print('________ unmanaged solution; skipping %s' % self.relpath)
411 return self._Capture(['rev-parse', '--verify', 'HEAD'])
412
szager@chromium.orgb0a13a22014-06-18 00:52:25 +0000413 if mirror:
414 self._UpdateMirror(mirror, options)
415
szager@chromium.org6c2b49d2014-02-26 23:57:38 +0000416 # See if the url has changed (the unittests use git://foo for the url, let
417 # that through).
418 current_url = self._Capture(['config', 'remote.%s.url' % self.remote])
419 return_early = False
420 # TODO(maruel): Delete url != 'git://foo' since it's just to make the
421 # unit test pass. (and update the comment above)
422 # Skip url auto-correction if remote.origin.gclient-auto-fix-url is set.
423 # This allows devs to use experimental repos which have a different url
424 # but whose branch(s) are the same as official repos.
borenet@google.comb09097a2014-04-09 19:09:08 +0000425 if (current_url.rstrip('/') != url.rstrip('/') and
szager@chromium.org6c2b49d2014-02-26 23:57:38 +0000426 url != 'git://foo' and
427 subprocess2.capture(
428 ['git', 'config', 'remote.%s.gclient-auto-fix-url' % self.remote],
429 cwd=self.checkout_path).strip() != 'False'):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000430 self.Print('_____ switching %s to a new upstream' % self.relpath)
szager@chromium.org6c2b49d2014-02-26 23:57:38 +0000431 # Make sure it's clean
432 self._CheckClean(rev_str)
433 # Switch over to the new upstream
434 self._Run(['remote', 'set-url', self.remote, url], options)
435 self._FetchAndReset(revision, file_list, options)
436 return_early = True
437
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +0000438 if return_early:
439 return self._Capture(['rev-parse', '--verify', 'HEAD'])
440
msb@chromium.org5bde4852009-12-14 16:47:12 +0000441 cur_branch = self._GetCurrentBranch()
442
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000443 # Cases:
msb@chromium.org786fb682010-06-02 15:16:23 +0000444 # 0) HEAD is detached. Probably from our initial clone.
445 # - make sure HEAD is contained by a named ref, then update.
446 # Cases 1-4. HEAD is a branch.
447 # 1) current branch is not tracking a remote branch (could be git-svn)
448 # - try to rebase onto the new hash or branch
449 # 2) current branch is tracking a remote branch with local committed
450 # changes, but the DEPS file switched to point to a hash
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000451 # - rebase those changes on top of the hash
msb@chromium.org786fb682010-06-02 15:16:23 +0000452 # 3) current branch is tracking a remote branch w/or w/out changes,
453 # no switch
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000454 # - see if we can FF, if not, prompt the user for rebase, merge, or stop
msb@chromium.org786fb682010-06-02 15:16:23 +0000455 # 4) current branch is tracking a remote branch, switches to a different
456 # remote branch
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000457 # - exit
458
maruel@chromium.org81e012c2010-04-29 16:07:24 +0000459 # GetUpstreamBranch returns something like 'refs/remotes/origin/master' for
460 # a tracking branch
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000461 # or 'master' if not a tracking branch (it's based on a specific rev/hash)
462 # or it returns None if it couldn't find an upstream
msb@chromium.org786fb682010-06-02 15:16:23 +0000463 if cur_branch is None:
464 upstream_branch = None
465 current_type = "detached"
466 logging.debug("Detached HEAD")
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000467 else:
msb@chromium.org786fb682010-06-02 15:16:23 +0000468 upstream_branch = scm.GIT.GetUpstreamBranch(self.checkout_path)
469 if not upstream_branch or not upstream_branch.startswith('refs/remotes'):
470 current_type = "hash"
471 logging.debug("Current branch is not tracking an upstream (remote)"
472 " branch.")
473 elif upstream_branch.startswith('refs/remotes'):
474 current_type = "branch"
475 else:
476 raise gclient_utils.Error('Invalid Upstream: %s' % upstream_branch)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000477
ilevy@chromium.orga41249c2013-07-03 00:09:12 +0000478 if not scm.GIT.IsValidRevision(self.checkout_path, revision, sha_only=True):
bauerb@chromium.orgcbd20a42012-06-27 13:49:27 +0000479 # Update the remotes first so we have all the refs.
ilevy@chromium.orga41249c2013-07-03 00:09:12 +0000480 remote_output = scm.GIT.Capture(['remote'] + verbose + ['update'],
bauerb@chromium.orgcbd20a42012-06-27 13:49:27 +0000481 cwd=self.checkout_path)
bauerb@chromium.orgcbd20a42012-06-27 13:49:27 +0000482 if verbose:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000483 self.Print(remote_output)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000484
mmoss@chromium.orge409df62013-04-16 17:28:57 +0000485 self._UpdateBranchHeads(options, fetch=True)
486
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000487 # This is a big hammer, debatable if it should even be here...
davemoore@chromium.org793796d2010-02-19 17:27:41 +0000488 if options.force or options.reset:
iannucci@chromium.orgd4fffee2013-06-28 00:35:26 +0000489 target = 'HEAD'
490 if options.upstream and upstream_branch:
491 target = upstream_branch
492 self._Run(['reset', '--hard', target], options)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000493
msb@chromium.org786fb682010-06-02 15:16:23 +0000494 if current_type == 'detached':
495 # case 0
496 self._CheckClean(rev_str)
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000497 self._CheckDetachedHead(rev_str, options)
szager@chromium.org848fd492014-04-09 19:06:44 +0000498 if self._Capture(['rev-list', '-n', '1', 'HEAD']) == revision:
499 self.Print('Up-to-date; skipping checkout.')
500 else:
vadimsh@chromium.org2b7d3ed2014-06-20 18:15:37 +0000501 # 'git checkout' may need to overwrite existing untracked files. Allow
502 # it only when nuclear options are enabled.
503 if options.force and options.delete_unversioned_trees:
504 self._Capture(['checkout', '--force', '--quiet', '%s' % revision])
505 else:
506 self._Capture(['checkout', '--quiet', '%s' % revision])
msb@chromium.org786fb682010-06-02 15:16:23 +0000507 if not printed_path:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000508 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
msb@chromium.org786fb682010-06-02 15:16:23 +0000509 elif current_type == 'hash':
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000510 # case 1
maruel@chromium.org55e724e2010-03-11 19:36:49 +0000511 if scm.GIT.IsGitSvn(self.checkout_path) and upstream_branch is not None:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000512 # Our git-svn branch (upstream_branch) is our upstream
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000513 self._AttemptRebase(upstream_branch, files, options,
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000514 newbase=revision, printed_path=printed_path,
515 merge=options.merge)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000516 printed_path = True
517 else:
518 # Can't find a merge-base since we don't know our upstream. That makes
519 # this command VERY likely to produce a rebase failure. For now we
520 # assume origin is our upstream since that's what the old behavior was.
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000521 upstream_branch = self.remote
nasser@codeaurora.org7080e942010-03-15 15:06:16 +0000522 if options.revision or deps_revision:
nasser@codeaurora.org3b29de12010-03-08 18:34:28 +0000523 upstream_branch = revision
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000524 self._AttemptRebase(upstream_branch, files, options,
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000525 printed_path=printed_path, merge=options.merge)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000526 printed_path = True
maruel@chromium.org55e724e2010-03-11 19:36:49 +0000527 elif rev_type == 'hash':
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000528 # case 2
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000529 self._AttemptRebase(upstream_branch, files, options,
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000530 newbase=revision, printed_path=printed_path,
531 merge=options.merge)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000532 printed_path = True
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000533 elif revision.replace('heads', 'remotes/' + self.remote) != upstream_branch:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000534 # case 4
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000535 new_base = revision.replace('heads', 'remotes/' + self.remote)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000536 if not printed_path:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000537 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000538 switch_error = ("Switching upstream branch from %s to %s\n"
539 % (upstream_branch, new_base) +
540 "Please merge or rebase manually:\n" +
541 "cd %s; git rebase %s\n" % (self.checkout_path, new_base) +
542 "OR git checkout -b <some new branch> %s" % new_base)
543 raise gclient_utils.Error(switch_error)
544 else:
545 # case 3 - the default case
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000546 if files is not None:
547 files = self._Capture(['diff', upstream_branch, '--name-only']).split()
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000548 if verbose:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000549 self.Print('Trying fast-forward merge to branch : %s' % upstream_branch)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000550 try:
bauerb@chromium.org2aad1b22011-07-22 12:00:41 +0000551 merge_args = ['merge']
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000552 if options.merge:
553 merge_args.append('--ff')
554 else:
bauerb@chromium.org2aad1b22011-07-22 12:00:41 +0000555 merge_args.append('--ff-only')
556 merge_args.append(upstream_branch)
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000557 merge_output = self._Capture(merge_args)
bratell@opera.com18fa4542013-05-21 13:30:46 +0000558 except subprocess2.CalledProcessError as e:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000559 if re.match('fatal: Not possible to fast-forward, aborting.', e.stderr):
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000560 files = []
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000561 if not printed_path:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000562 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000563 printed_path = True
564 while True:
565 try:
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000566 action = self._AskForData(
567 'Cannot %s, attempt to rebase? '
568 '(y)es / (q)uit / (s)kip : ' %
569 ('merge' if options.merge else 'fast-forward merge'),
570 options)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000571 except ValueError:
maruel@chromium.org90541732011-04-01 17:54:18 +0000572 raise gclient_utils.Error('Invalid Character')
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000573 if re.match(r'yes|y', action, re.I):
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000574 self._AttemptRebase(upstream_branch, files, options,
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000575 printed_path=printed_path, merge=False)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000576 printed_path = True
577 break
578 elif re.match(r'quit|q', action, re.I):
579 raise gclient_utils.Error("Can't fast-forward, please merge or "
580 "rebase manually.\n"
581 "cd %s && git " % self.checkout_path
582 + "rebase %s" % upstream_branch)
583 elif re.match(r'skip|s', action, re.I):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000584 self.Print('Skipping %s' % self.relpath)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000585 return
586 else:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000587 self.Print('Input not recognized')
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000588 elif re.match("error: Your local changes to '.*' would be "
589 "overwritten by merge. Aborting.\nPlease, commit your "
590 "changes or stash them before you can merge.\n",
591 e.stderr):
592 if not printed_path:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000593 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000594 printed_path = True
595 raise gclient_utils.Error(e.stderr)
596 else:
597 # Some other problem happened with the merge
598 logging.error("Error during fast-forward merge in %s!" % self.relpath)
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000599 self.Print(e.stderr)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000600 raise
601 else:
602 # Fast-forward merge was successful
603 if not re.match('Already up-to-date.', merge_output) or verbose:
604 if not printed_path:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000605 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000606 printed_path = True
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000607 self.Print(merge_output.strip())
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000608 if not verbose:
609 # Make the output a little prettier. It's nice to have some
610 # whitespace between projects when syncing.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000611 self.Print('')
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000612
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000613 if file_list is not None:
614 file_list.extend([os.path.join(self.checkout_path, f) for f in files])
msb@chromium.org5bde4852009-12-14 16:47:12 +0000615
616 # If the rebase generated a conflict, abort and ask user to fix
msb@chromium.org786fb682010-06-02 15:16:23 +0000617 if self._IsRebasing():
msb@chromium.org5bde4852009-12-14 16:47:12 +0000618 raise gclient_utils.Error('\n____ %s%s\n'
619 '\nConflict while rebasing this branch.\n'
620 'Fix the conflict and run gclient again.\n'
621 'See man git-rebase for details.\n'
622 % (self.relpath, rev_str))
623
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000624 if verbose:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000625 self.Print('Checked out revision %s' % self.revinfo(options, (), None),
626 timestamp=False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000627
steveblock@chromium.org98e69452012-02-16 16:36:43 +0000628 # If --reset and --delete_unversioned_trees are specified, remove any
629 # untracked directories.
630 if options.reset and options.delete_unversioned_trees:
631 # GIT.CaptureStatus() uses 'dit diff' to compare to a specific SHA1 (the
632 # merge-base by default), so doesn't include untracked files. So we use
633 # 'git ls-files --directory --others --exclude-standard' here directly.
634 paths = scm.GIT.Capture(
635 ['ls-files', '--directory', '--others', '--exclude-standard'],
636 self.checkout_path)
637 for path in (p for p in paths.splitlines() if p.endswith('/')):
638 full_path = os.path.join(self.checkout_path, path)
639 if not os.path.islink(full_path):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000640 self.Print('_____ removing unversioned directory %s' % path)
digit@chromium.orgdc112ac2013-04-24 13:00:19 +0000641 gclient_utils.rmtree(full_path)
steveblock@chromium.org98e69452012-02-16 16:36:43 +0000642
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +0000643 return self._Capture(['rev-parse', '--verify', 'HEAD'])
644
steveblock@chromium.org98e69452012-02-16 16:36:43 +0000645
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000646 def revert(self, options, _args, file_list):
msb@chromium.orge28e4982009-09-25 20:51:45 +0000647 """Reverts local modifications.
648
649 All reverted files will be appended to file_list.
650 """
maruel@chromium.org8469bf92010-09-03 19:03:15 +0000651 if not os.path.isdir(self.checkout_path):
msb@chromium.org260c6532009-10-28 03:22:35 +0000652 # revert won't work if the directory doesn't exist. It needs to
653 # checkout instead.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000654 self.Print('_____ %s is missing, synching instead' % self.relpath)
msb@chromium.org260c6532009-10-28 03:22:35 +0000655 # Don't reuse the args.
656 return self.update(options, [], file_list)
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000657
658 default_rev = "refs/heads/master"
iannucci@chromium.orgd4fffee2013-06-28 00:35:26 +0000659 if options.upstream:
660 if self._GetCurrentBranch():
661 upstream_branch = scm.GIT.GetUpstreamBranch(self.checkout_path)
662 default_rev = upstream_branch or default_rev
maruel@chromium.org6e29d572010-06-04 17:32:20 +0000663 _, deps_revision = gclient_utils.SplitUrlRevision(self.url)
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000664 if not deps_revision:
665 deps_revision = default_rev
666 if deps_revision.startswith('refs/heads/'):
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000667 deps_revision = deps_revision.replace('refs/heads/', self.remote + '/')
iannucci@chromium.org9a8b0662014-02-20 00:56:15 +0000668 deps_revision = self.GetUsableRev(deps_revision, options)
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000669
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000670 if file_list is not None:
671 files = self._Capture(['diff', deps_revision, '--name-only']).split()
672
maruel@chromium.org37e89872010-09-07 16:11:33 +0000673 self._Run(['reset', '--hard', deps_revision], options)
lliabraa@chromium.orgade83db2012-09-27 14:06:49 +0000674 self._Run(['clean', '-f', '-d'], options)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000675
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000676 if file_list is not None:
677 file_list.extend([os.path.join(self.checkout_path, f) for f in files])
678
679 def revinfo(self, _options, _args, _file_list):
maruel@chromium.org6cafa132010-09-07 14:17:26 +0000680 """Returns revision"""
681 return self._Capture(['rev-parse', 'HEAD'])
msb@chromium.org0f282062009-11-06 20:14:02 +0000682
msb@chromium.orge28e4982009-09-25 20:51:45 +0000683 def runhooks(self, options, args, file_list):
684 self.status(options, args, file_list)
685
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000686 def status(self, options, _args, file_list):
msb@chromium.orge28e4982009-09-25 20:51:45 +0000687 """Display status information."""
688 if not os.path.isdir(self.checkout_path):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000689 self.Print('________ couldn\'t run status in %s:\n'
690 'The directory does not exist.' % self.checkout_path)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000691 else:
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000692 merge_base = self._Capture(['merge-base', 'HEAD', self.remote])
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000693 self._Run(['diff', '--name-status', merge_base], options,
694 stdout=self.out_fh)
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000695 if file_list is not None:
696 files = self._Capture(['diff', '--name-only', merge_base]).split()
697 file_list.extend([os.path.join(self.checkout_path, f) for f in files])
msb@chromium.orge28e4982009-09-25 20:51:45 +0000698
dbeam@chromium.orge5d1e612011-12-19 19:49:19 +0000699 def GetUsableRev(self, rev, options):
700 """Finds a useful revision for this repository.
701
702 If SCM is git-svn and the head revision is less than |rev|, git svn fetch
703 will be called on the source."""
704 sha1 = None
iannucci@chromium.org3830a672013-02-19 20:15:14 +0000705 if not os.path.isdir(self.checkout_path):
706 raise gclient_utils.Error(
707 ( 'We could not find a valid hash for safesync_url response "%s".\n'
708 'Safesync URLs with a git checkout currently require the repo to\n'
709 'be cloned without a safesync_url before adding the safesync_url.\n'
710 'For more info, see: '
711 'http://code.google.com/p/chromium/wiki/UsingNewGit'
712 '#Initial_checkout' ) % rev)
713 elif rev.isdigit() and len(rev) < 7:
714 # Handles an SVN rev. As an optimization, only verify an SVN revision as
715 # [0-9]{1,6} for now to avoid making a network request.
szager@chromium.org6c2b49d2014-02-26 23:57:38 +0000716 if scm.GIT.IsGitSvn(cwd=self.checkout_path):
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000717 local_head = scm.GIT.GetGitSvnHeadRev(cwd=self.checkout_path)
718 if not local_head or local_head < int(rev):
dbeam@chromium.org2a75fdb2012-02-15 01:32:57 +0000719 try:
720 logging.debug('Looking for git-svn configuration optimizations.')
721 if scm.GIT.Capture(['config', '--get', 'svn-remote.svn.fetch'],
722 cwd=self.checkout_path):
723 scm.GIT.Capture(['fetch'], cwd=self.checkout_path)
724 except subprocess2.CalledProcessError:
725 logging.debug('git config --get svn-remote.svn.fetch failed, '
726 'ignoring possible optimization.')
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000727 if options.verbose:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000728 self.Print('Running git svn fetch. This might take a while.\n')
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000729 scm.GIT.Capture(['svn', 'fetch'], cwd=self.checkout_path)
szager@google.com312a6a42012-10-11 21:19:42 +0000730 try:
szager@chromium.orgc51def32012-10-15 18:50:37 +0000731 sha1 = scm.GIT.GetBlessedSha1ForSvnRev(
732 cwd=self.checkout_path, rev=rev)
szager@google.com312a6a42012-10-11 21:19:42 +0000733 except gclient_utils.Error, e:
734 sha1 = e.message
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000735 self.Print('Warning: Could not find a git revision with accurate\n'
szager@google.com312a6a42012-10-11 21:19:42 +0000736 '.DEPS.git that maps to SVN revision %s. Sync-ing to\n'
737 'the closest sane git revision, which is:\n'
738 ' %s\n' % (rev, e.message))
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000739 if not sha1:
740 raise gclient_utils.Error(
741 ( 'It appears that either your git-svn remote is incorrectly\n'
742 'configured or the revision in your safesync_url is\n'
743 'higher than git-svn remote\'s HEAD as we couldn\'t find a\n'
744 'corresponding git hash for SVN rev %s.' ) % rev)
iannucci@chromium.org3830a672013-02-19 20:15:14 +0000745 else:
746 if scm.GIT.IsValidRevision(cwd=self.checkout_path, rev=rev):
747 sha1 = rev
748 else:
749 # May exist in origin, but we don't have it yet, so fetch and look
750 # again.
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000751 scm.GIT.Capture(['fetch', self.remote], cwd=self.checkout_path)
iannucci@chromium.org3830a672013-02-19 20:15:14 +0000752 if scm.GIT.IsValidRevision(cwd=self.checkout_path, rev=rev):
753 sha1 = rev
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000754
dbeam@chromium.orge5d1e612011-12-19 19:49:19 +0000755 if not sha1:
756 raise gclient_utils.Error(
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000757 ( 'We could not find a valid hash for safesync_url response "%s".\n'
758 'Safesync URLs with a git checkout currently require a git-svn\n'
759 'remote or a safesync_url that provides git sha1s. Please add a\n'
760 'git-svn remote or change your safesync_url. For more info, see:\n'
dbeam@chromium.orge5d1e612011-12-19 19:49:19 +0000761 'http://code.google.com/p/chromium/wiki/UsingNewGit'
dbeam@chromium.org051c88b2011-12-22 00:23:03 +0000762 '#Initial_checkout' ) % rev)
763
dbeam@chromium.orge5d1e612011-12-19 19:49:19 +0000764 return sha1
765
msb@chromium.orge6f78352010-01-13 17:05:33 +0000766 def FullUrlForRelativeUrl(self, url):
767 # Strip from last '/'
768 # Equivalent to unix basename
769 base_url = self.url
770 return base_url[:base_url.rfind('/')] + url
771
szager@chromium.orgb0a13a22014-06-18 00:52:25 +0000772 def _GetMirror(self, url, options):
773 """Get a git_cache.Mirror object for the argument url."""
774 if not git_cache.Mirror.GetCachePath():
775 return None
hinoka@google.comb1b54572014-04-16 22:29:23 +0000776 mirror_kwargs = {
777 'print_func': self.filter,
778 'refs': []
779 }
hinoka@google.com356cb5f2014-04-25 00:02:21 +0000780 # TODO(hinoka): This currently just fails because lkcr/lkgr are branches
781 # not tags. This also adds 20 seconds to every bot_update
782 # run, so I'm commenting this out until lkcr/lkgr become
783 # tags. (2014/4/24)
784 # if url == CHROMIUM_SRC_URL or url + '.git' == CHROMIUM_SRC_URL:
785 # mirror_kwargs['refs'].extend(['refs/tags/lkgr', 'refs/tags/lkcr'])
hinoka@google.comb1b54572014-04-16 22:29:23 +0000786 if hasattr(options, 'with_branch_heads') and options.with_branch_heads:
787 mirror_kwargs['refs'].append('refs/branch-heads/*')
szager@chromium.orgb0a13a22014-06-18 00:52:25 +0000788 return git_cache.Mirror(url, **mirror_kwargs)
789
790 @staticmethod
791 def _UpdateMirror(mirror, options):
792 """Update a git mirror by fetching the latest commits from the remote."""
hinoka@chromium.org46b87412014-05-15 00:42:05 +0000793 if options.shallow:
794 # HACK(hinoka): These repositories should be super shallow.
szager@chromium.orgb0a13a22014-06-18 00:52:25 +0000795 if 'flash' in mirror.url:
hinoka@chromium.org46b87412014-05-15 00:42:05 +0000796 depth = 10
797 else:
798 depth = 10000
799 else:
800 depth = None
hinoka@chromium.org8a10f6d2014-06-23 18:38:57 +0000801 mirror.populate(verbose=options.verbose, bootstrap=True, depth=depth,
802 ignore_lock=options.ignore_locks)
szager@chromium.org848fd492014-04-09 19:06:44 +0000803 mirror.unlock()
iannucci@chromium.org53456aa2013-07-03 19:38:34 +0000804
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000805 def _Clone(self, revision, url, options):
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000806 """Clone a git repository from the given URL.
807
msb@chromium.org786fb682010-06-02 15:16:23 +0000808 Once we've cloned the repo, we checkout a working branch if the specified
809 revision is a branch head. If it is a tag or a specific commit, then we
810 leave HEAD detached as it makes future updates simpler -- in this case the
811 user should first create a new branch or switch to an existing branch before
812 making changes in the repo."""
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000813 if not options.verbose:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000814 # git clone doesn't seem to insert a newline properly before printing
815 # to stdout
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000816 self.Print('')
szager@chromium.orgf1d73eb2014-04-21 17:07:04 +0000817 cfg = gclient_utils.DefaultIndexPackConfig(url)
szager@chromium.org8a139702014-06-20 15:55:01 +0000818 clone_cmd = cfg + ['clone', '--no-checkout', '--progress']
iannucci@chromium.org53456aa2013-07-03 19:38:34 +0000819 if self.cache_dir:
820 clone_cmd.append('--shared')
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000821 if options.verbose:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000822 clone_cmd.append('--verbose')
ilevy@chromium.org3534aa52013-07-20 01:58:08 +0000823 clone_cmd.append(url)
nsylvain@chromium.org328c3c72011-06-01 20:50:27 +0000824 # If the parent directory does not exist, Git clone on Windows will not
825 # create it, so we need to do it manually.
826 parent_dir = os.path.dirname(self.checkout_path)
ilevy@chromium.org3534aa52013-07-20 01:58:08 +0000827 gclient_utils.safe_makedirs(parent_dir)
828 tmp_dir = tempfile.mkdtemp(
829 prefix='_gclient_%s_' % os.path.basename(self.checkout_path),
830 dir=parent_dir)
831 try:
832 clone_cmd.append(tmp_dir)
agable@chromium.orgfd5b6382013-10-25 20:54:34 +0000833 self._Run(clone_cmd, options, cwd=self._root_dir, retry=True)
ilevy@chromium.org3534aa52013-07-20 01:58:08 +0000834 gclient_utils.safe_makedirs(self.checkout_path)
cyrille@nnamrak.orgef509e42013-09-20 13:19:08 +0000835 gclient_utils.safe_rename(os.path.join(tmp_dir, '.git'),
836 os.path.join(self.checkout_path, '.git'))
zty@chromium.org6279e8a2014-02-13 01:45:25 +0000837 except:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000838 traceback.print_exc(file=self.out_fh)
zty@chromium.org6279e8a2014-02-13 01:45:25 +0000839 raise
ilevy@chromium.org3534aa52013-07-20 01:58:08 +0000840 finally:
841 if os.listdir(tmp_dir):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000842 self.Print('_____ removing non-empty tmp dir %s' % tmp_dir)
ilevy@chromium.org3534aa52013-07-20 01:58:08 +0000843 gclient_utils.rmtree(tmp_dir)
mmoss@chromium.org1a6bec02014-06-02 21:53:29 +0000844 self._UpdateBranchHeads(options, fetch=True)
romain.pokrzywka@gmail.com483a0ba2014-05-30 00:06:07 +0000845 self._Run(['checkout', '--quiet', revision.replace('refs/heads/', '')],
846 options)
847 if self._GetCurrentBranch() is None:
msb@chromium.org786fb682010-06-02 15:16:23 +0000848 # Squelch git's very verbose detached HEAD warning and use our own
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000849 self.Print(
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000850 ('Checked out %s to a detached HEAD. Before making any commits\n'
851 'in this repo, you should use \'git checkout <branch>\' to switch to\n'
maruel@chromium.org1a60dca2013-11-26 14:06:26 +0000852 'an existing branch or use \'git checkout %s -b <branch>\' to\n'
853 'create a new branch for your work.') % (revision, self.remote))
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000854
szager@chromium.org6cd41b62014-04-21 23:55:22 +0000855 def _AskForData(self, prompt, options):
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000856 if options.jobs > 1:
szager@chromium.org6cd41b62014-04-21 23:55:22 +0000857 self.Print(prompt)
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000858 raise gclient_utils.Error("Background task requires input. Rerun "
859 "gclient with --jobs=1 so that\n"
860 "interaction is possible.")
861 try:
862 return raw_input(prompt)
863 except KeyboardInterrupt:
864 # Hide the exception.
865 sys.exit(1)
866
867
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000868 def _AttemptRebase(self, upstream, files, options, newbase=None,
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000869 branch=None, printed_path=False, merge=False):
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000870 """Attempt to rebase onto either upstream or, if specified, newbase."""
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000871 if files is not None:
872 files.extend(self._Capture(['diff', upstream, '--name-only']).split())
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000873 revision = upstream
874 if newbase:
875 revision = newbase
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000876 action = 'merge' if merge else 'rebase'
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000877 if not printed_path:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000878 self.Print('_____ %s : Attempting %s onto %s...' % (
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000879 self.relpath, action, revision))
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000880 printed_path = True
881 else:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000882 self.Print('Attempting %s onto %s...' % (action, revision))
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000883
884 if merge:
885 merge_output = self._Capture(['merge', revision])
886 if options.verbose:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000887 self.Print(merge_output)
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000888 return
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000889
890 # Build the rebase command here using the args
891 # git rebase [options] [--onto <newbase>] <upstream> [<branch>]
892 rebase_cmd = ['rebase']
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000893 if options.verbose:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000894 rebase_cmd.append('--verbose')
895 if newbase:
896 rebase_cmd.extend(['--onto', newbase])
897 rebase_cmd.append(upstream)
898 if branch:
899 rebase_cmd.append(branch)
900
901 try:
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000902 rebase_output = scm.GIT.Capture(rebase_cmd, cwd=self.checkout_path)
maruel@chromium.orgbffad372011-09-08 17:54:22 +0000903 except subprocess2.CalledProcessError, e:
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000904 if (re.match(r'cannot rebase: you have unstaged changes', e.stderr) or
905 re.match(r'cannot rebase: your index contains uncommitted changes',
906 e.stderr)):
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000907 while True:
bauerb@chromium.org30c46d62014-01-23 12:11:56 +0000908 rebase_action = self._AskForData(
maruel@chromium.org90541732011-04-01 17:54:18 +0000909 'Cannot rebase because of unstaged changes.\n'
910 '\'git reset --hard HEAD\' ?\n'
911 'WARNING: destroys any uncommitted work in your current branch!'
bratell@opera.com18fa4542013-05-21 13:30:46 +0000912 ' (y)es / (q)uit / (s)how : ', options)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000913 if re.match(r'yes|y', rebase_action, re.I):
maruel@chromium.org37e89872010-09-07 16:11:33 +0000914 self._Run(['reset', '--hard', 'HEAD'], options)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000915 # Should this be recursive?
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000916 rebase_output = scm.GIT.Capture(rebase_cmd, cwd=self.checkout_path)
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000917 break
918 elif re.match(r'quit|q', rebase_action, re.I):
919 raise gclient_utils.Error("Please merge or rebase manually\n"
920 "cd %s && git " % self.checkout_path
921 + "%s" % ' '.join(rebase_cmd))
922 elif re.match(r'show|s', rebase_action, re.I):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000923 self.Print('%s' % e.stderr.strip())
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000924 continue
925 else:
926 gclient_utils.Error("Input not recognized")
927 continue
928 elif re.search(r'^CONFLICT', e.stdout, re.M):
929 raise gclient_utils.Error("Conflict while rebasing this branch.\n"
930 "Fix the conflict and run gclient again.\n"
931 "See 'man git-rebase' for details.\n")
932 else:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000933 self.Print(e.stdout.strip())
934 self.Print('Rebase produced error output:\n%s' % e.stderr.strip())
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000935 raise gclient_utils.Error("Unrecognized error, please merge or rebase "
936 "manually.\ncd %s && git " %
937 self.checkout_path
938 + "%s" % ' '.join(rebase_cmd))
939
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000940 self.Print(rebase_output.strip())
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +0000941 if not options.verbose:
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000942 # Make the output a little prettier. It's nice to have some
943 # whitespace between projects when syncing.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +0000944 self.Print('')
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000945
maruel@chromium.org6e29d572010-06-04 17:32:20 +0000946 @staticmethod
947 def _CheckMinVersion(min_version):
maruel@chromium.orgd0f854a2010-03-11 19:35:53 +0000948 (ok, current_version) = scm.GIT.AssertVersion(min_version)
949 if not ok:
950 raise gclient_utils.Error('git version %s < minimum required %s' %
951 (current_version, min_version))
msb@chromium.org923a0372009-12-11 20:42:43 +0000952
msb@chromium.org786fb682010-06-02 15:16:23 +0000953 def _IsRebasing(self):
954 # Check for any of REBASE-i/REBASE-m/REBASE/AM. Unfortunately git doesn't
955 # have a plumbing command to determine whether a rebase is in progress, so
956 # for now emualate (more-or-less) git-rebase.sh / git-completion.bash
957 g = os.path.join(self.checkout_path, '.git')
958 return (
959 os.path.isdir(os.path.join(g, "rebase-merge")) or
960 os.path.isdir(os.path.join(g, "rebase-apply")))
961
962 def _CheckClean(self, rev_str):
963 # Make sure the tree is clean; see git-rebase.sh for reference
964 try:
965 scm.GIT.Capture(['update-index', '--ignore-submodules', '--refresh'],
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000966 cwd=self.checkout_path)
maruel@chromium.orgbffad372011-09-08 17:54:22 +0000967 except subprocess2.CalledProcessError:
maruel@chromium.org6e29d572010-06-04 17:32:20 +0000968 raise gclient_utils.Error('\n____ %s%s\n'
969 '\tYou have unstaged changes.\n'
970 '\tPlease commit, stash, or reset.\n'
971 % (self.relpath, rev_str))
msb@chromium.org786fb682010-06-02 15:16:23 +0000972 try:
973 scm.GIT.Capture(['diff-index', '--cached', '--name-status', '-r',
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000974 '--ignore-submodules', 'HEAD', '--'],
975 cwd=self.checkout_path)
maruel@chromium.orgbffad372011-09-08 17:54:22 +0000976 except subprocess2.CalledProcessError:
maruel@chromium.org6e29d572010-06-04 17:32:20 +0000977 raise gclient_utils.Error('\n____ %s%s\n'
978 '\tYour index contains uncommitted changes\n'
979 '\tPlease commit, stash, or reset.\n'
980 % (self.relpath, rev_str))
msb@chromium.org786fb682010-06-02 15:16:23 +0000981
iannucci@chromium.org396e1a62013-07-03 19:41:04 +0000982 def _CheckDetachedHead(self, rev_str, _options):
msb@chromium.org786fb682010-06-02 15:16:23 +0000983 # HEAD is detached. Make sure it is safe to move away from (i.e., it is
984 # reference by a commit). If not, error out -- most likely a rebase is
985 # in progress, try to detect so we can give a better error.
986 try:
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000987 scm.GIT.Capture(['name-rev', '--no-undefined', 'HEAD'],
988 cwd=self.checkout_path)
maruel@chromium.orgbffad372011-09-08 17:54:22 +0000989 except subprocess2.CalledProcessError:
msb@chromium.org786fb682010-06-02 15:16:23 +0000990 # Commit is not contained by any rev. See if the user is rebasing:
991 if self._IsRebasing():
992 # Punt to the user
993 raise gclient_utils.Error('\n____ %s%s\n'
994 '\tAlready in a conflict, i.e. (no branch).\n'
995 '\tFix the conflict and run gclient again.\n'
996 '\tOr to abort run:\n\t\tgit-rebase --abort\n'
997 '\tSee man git-rebase for details.\n'
998 % (self.relpath, rev_str))
999 # Let's just save off the commit so we can proceed.
maruel@chromium.org6cafa132010-09-07 14:17:26 +00001000 name = ('saved-by-gclient-' +
1001 self._Capture(['rev-parse', '--short', 'HEAD']))
mmoss@chromium.org77bd7362013-09-25 23:46:14 +00001002 self._Capture(['branch', '-f', name])
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001003 self.Print('_____ found an unreferenced commit and saved it as \'%s\'' %
maruel@chromium.orgf5d37bf2010-09-02 00:50:34 +00001004 name)
msb@chromium.org786fb682010-06-02 15:16:23 +00001005
msb@chromium.org5bde4852009-12-14 16:47:12 +00001006 def _GetCurrentBranch(self):
msb@chromium.org786fb682010-06-02 15:16:23 +00001007 # Returns name of current branch or None for detached HEAD
maruel@chromium.org6cafa132010-09-07 14:17:26 +00001008 branch = self._Capture(['rev-parse', '--abbrev-ref=strict', 'HEAD'])
msb@chromium.org786fb682010-06-02 15:16:23 +00001009 if branch == 'HEAD':
msb@chromium.org5bde4852009-12-14 16:47:12 +00001010 return None
1011 return branch
1012
borenet@google.comc3e09d22014-04-10 13:58:18 +00001013 def _Capture(self, args, **kwargs):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001014 kwargs.setdefault('cwd', self.checkout_path)
1015 kwargs.setdefault('stderr', subprocess2.PIPE)
szager@chromium.org6d8115d2014-04-23 20:59:23 +00001016 env = scm.GIT.ApplyEnvVars(kwargs)
1017 return subprocess2.check_output(['git'] + args, env=env, **kwargs).strip()
maruel@chromium.org6cafa132010-09-07 14:17:26 +00001018
mmoss@chromium.orge409df62013-04-16 17:28:57 +00001019 def _UpdateBranchHeads(self, options, fetch=False):
1020 """Adds, and optionally fetches, "branch-heads" refspecs if requested."""
1021 if hasattr(options, 'with_branch_heads') and options.with_branch_heads:
maruel@chromium.org1a60dca2013-11-26 14:06:26 +00001022 config_cmd = ['config', 'remote.%s.fetch' % self.remote,
szager@chromium.orgf2d7d6b2013-10-17 20:41:43 +00001023 '+refs/branch-heads/*:refs/remotes/branch-heads/*',
1024 '^\\+refs/branch-heads/\\*:.*$']
1025 self._Run(config_cmd, options)
1026 if fetch:
szager@chromium.orgff113292014-03-25 06:02:08 +00001027 cfg = gclient_utils.DefaultIndexPackConfig(self.url)
szager@chromium.orgfc616382014-03-18 20:32:04 +00001028 fetch_cmd = cfg + ['fetch', self.remote]
szager@chromium.orgf2d7d6b2013-10-17 20:41:43 +00001029 if options.verbose:
1030 fetch_cmd.append('--verbose')
1031 self._Run(fetch_cmd, options, retry=True)
mmoss@chromium.orge409df62013-04-16 17:28:57 +00001032
agable@chromium.orgfd5b6382013-10-25 20:54:34 +00001033 def _Run(self, args, options, **kwargs):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001034 cwd = kwargs.setdefault('cwd', self.checkout_path)
1035 kwargs.setdefault('stdout', self.out_fh)
szager@chromium.org848fd492014-04-09 19:06:44 +00001036 kwargs['filter_fn'] = self.filter
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001037 kwargs.setdefault('print_stdout', False)
szager@chromium.org6d8115d2014-04-23 20:59:23 +00001038 env = scm.GIT.ApplyEnvVars(kwargs)
agable@chromium.org772efaf2014-04-01 02:35:44 +00001039 cmd = ['git'] + args
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001040 header = "running '%s' in '%s'" % (' '.join(cmd), cwd)
szager@chromium.org848fd492014-04-09 19:06:44 +00001041 self.filter(header)
szager@chromium.org6d8115d2014-04-23 20:59:23 +00001042 return gclient_utils.CheckCallAndFilter(cmd, env=env, **kwargs)
msb@chromium.orge28e4982009-09-25 20:51:45 +00001043
1044
maruel@chromium.org55e724e2010-03-11 19:36:49 +00001045class SVNWrapper(SCMWrapper):
msb@chromium.orgcb5442b2009-09-22 16:51:24 +00001046 """ Wrapper for SVN """
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +00001047 name = 'svn'
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001048
mukai@chromium.org9e3e82c2012-04-18 12:55:43 +00001049 @staticmethod
1050 def BinaryExists():
1051 """Returns true if the command exists."""
1052 try:
1053 result, version = scm.SVN.AssertVersion('1.4')
1054 if not result:
1055 raise gclient_utils.Error('SVN version is older than 1.4: %s' % version)
1056 return result
1057 except OSError:
1058 return False
1059
xusydoc@chromium.org885a9602013-05-31 09:54:40 +00001060 def GetCheckoutRoot(self):
1061 return scm.SVN.GetCheckoutRoot(self.checkout_path)
1062
floitsch@google.comeaab7842011-04-28 09:07:58 +00001063 def GetRevisionDate(self, revision):
1064 """Returns the given revision's date in ISO-8601 format (which contains the
1065 time zone)."""
maruel@chromium.orgd579fcf2011-12-13 20:36:03 +00001066 date = scm.SVN.Capture(
1067 ['propget', '--revprop', 'svn:date', '-r', revision],
1068 os.path.join(self.checkout_path, '.'))
floitsch@google.comeaab7842011-04-28 09:07:58 +00001069 return date.strip()
1070
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001071 def cleanup(self, options, args, _file_list):
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001072 """Cleanup working copy."""
maruel@chromium.org669600d2010-09-01 19:06:31 +00001073 self._Run(['cleanup'] + args, options)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001074
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001075 def diff(self, options, args, _file_list):
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001076 # NOTE: This function does not currently modify file_list.
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001077 if not os.path.isdir(self.checkout_path):
1078 raise gclient_utils.Error('Directory %s is not present.' %
1079 self.checkout_path)
maruel@chromium.org669600d2010-09-01 19:06:31 +00001080 self._Run(['diff'] + args, options)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001081
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001082 def pack(self, _options, args, _file_list):
maruel@chromium.orgee4071d2009-12-22 22:25:37 +00001083 """Generates a patch file which can be applied to the root of the
1084 repository."""
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001085 if not os.path.isdir(self.checkout_path):
1086 raise gclient_utils.Error('Directory %s is not present.' %
1087 self.checkout_path)
1088 gclient_utils.CheckCallAndFilter(
1089 ['svn', 'diff', '-x', '--ignore-eol-style'] + args,
1090 cwd=self.checkout_path,
1091 print_stdout=False,
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001092 filter_fn=SvnDiffFilterer(self.relpath).Filter, print_func=self.Print)
maruel@chromium.orgee4071d2009-12-22 22:25:37 +00001093
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001094 def update(self, options, args, file_list):
msb@chromium.orgd6504212010-01-13 17:34:31 +00001095 """Runs svn to update or transparently checkout the working copy.
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001096
1097 All updated files will be appended to file_list.
1098
1099 Raises:
1100 Error: if can't get URL for relative path.
1101 """
borenet@google.comb09097a2014-04-09 19:09:08 +00001102 # Only update if hg is not controlling the directory.
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001103 hg_path = os.path.join(self.checkout_path, '.hg')
1104 if os.path.exists(hg_path):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001105 self.Print('________ found .hg directory; skipping %s' % self.relpath)
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001106 return
1107
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001108 if args:
1109 raise gclient_utils.Error("Unsupported argument(s): %s" % ",".join(args))
1110
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001111 # revision is the revision to match. It is None if no revision is specified,
1112 # i.e. the 'deps ain't pinned'.
msb@chromium.orgac915bb2009-11-13 17:03:01 +00001113 url, revision = gclient_utils.SplitUrlRevision(self.url)
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +00001114 # Keep the original unpinned url for reference in case the repo is switched.
1115 base_url = url
cmp@chromium.orgeb2756d2011-09-20 20:17:51 +00001116 managed = True
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001117 if options.revision:
1118 # Override the revision number.
msb@chromium.orgac915bb2009-11-13 17:03:01 +00001119 revision = str(options.revision)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001120 if revision:
cmp@chromium.orgeb2756d2011-09-20 20:17:51 +00001121 if revision != 'unmanaged':
1122 forced_revision = True
1123 # Reconstruct the url.
1124 url = '%s@%s' % (url, revision)
1125 rev_str = ' at %s' % revision
1126 else:
1127 managed = False
1128 revision = None
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001129 else:
1130 forced_revision = False
1131 rev_str = ''
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001132
thakis@chromium.orgd0b0a5b2014-03-10 19:30:21 +00001133 exists = os.path.exists(self.checkout_path)
borenet@google.com7ff04292014-03-10 12:57:25 +00001134 if exists and managed:
borenet@google.comb09097a2014-04-09 19:09:08 +00001135 # Git is only okay if it's a git-svn checkout of the right repo.
1136 if scm.GIT.IsGitSvn(self.checkout_path):
1137 remote_url = scm.GIT.Capture(['config', '--local', '--get',
1138 'svn-remote.svn.url'],
1139 cwd=self.checkout_path).rstrip()
1140 if remote_url.rstrip('/') == base_url.rstrip('/'):
borenet@google.comb2256212014-05-07 20:57:28 +00001141 self.Print('\n_____ %s looks like a git-svn checkout. Skipping.'
1142 % self.relpath)
borenet@google.comb09097a2014-04-09 19:09:08 +00001143 return # TODO(borenet): Get the svn revision number?
1144
1145 # Get the existing scm url and the revision number of the current checkout.
1146 if exists and managed:
davidjames@chromium.org13349e22012-11-15 17:11:28 +00001147 try:
1148 from_info = scm.SVN.CaptureLocalInfo(
1149 [], os.path.join(self.checkout_path, '.'))
1150 except (gclient_utils.Error, subprocess2.CalledProcessError):
borenet@google.comb09097a2014-04-09 19:09:08 +00001151 self._DeleteOrMove(options.force)
1152 exists = False
davidjames@chromium.org13349e22012-11-15 17:11:28 +00001153
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001154 BASE_URLS = {
1155 '/chrome/trunk/src': 'gs://chromium-svn-checkout/chrome/',
1156 '/blink/trunk': 'gs://chromium-svn-checkout/blink/',
1157 }
hinoka@google.comca35be32014-01-17 01:48:18 +00001158 WHITELISTED_ROOTS = [
1159 'svn://svn.chromium.org',
1160 'svn://svn-mirror.golo.chromium.org',
1161 ]
davidjames@chromium.org13349e22012-11-15 17:11:28 +00001162 if not exists:
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001163 try:
1164 # Split out the revision number since it's not useful for us.
1165 base_path = urlparse.urlparse(url).path.split('@')[0]
hinoka@google.comca35be32014-01-17 01:48:18 +00001166 # Check to see if we're on a whitelisted root. We do this because
1167 # only some svn servers have matching UUIDs.
1168 local_parsed = urlparse.urlparse(url)
1169 local_root = '%s://%s' % (local_parsed.scheme, local_parsed.netloc)
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001170 if ('CHROME_HEADLESS' in os.environ
1171 and sys.platform == 'linux2' # TODO(hinoka): Enable for win/mac.
hinoka@google.comca35be32014-01-17 01:48:18 +00001172 and base_path in BASE_URLS
1173 and local_root in WHITELISTED_ROOTS):
1174
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001175 # Use a tarball for initial sync if we are on a bot.
1176 # Get an unauthenticated gsutil instance.
1177 gsutil = download_from_google_storage.Gsutil(
1178 GSUTIL_DEFAULT_PATH, boto_path=os.devnull)
1179
1180 gs_path = BASE_URLS[base_path]
1181 _, out, _ = gsutil.check_call('ls', gs_path)
1182 # So that we can get the most recent revision.
1183 sorted_items = sorted(out.splitlines())
1184 latest_checkout = sorted_items[-1]
1185
1186 tempdir = tempfile.mkdtemp()
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001187 self.Print('Downloading %s...' % latest_checkout)
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001188 code, out, err = gsutil.check_call('cp', latest_checkout, tempdir)
1189 if code:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001190 self.Print('%s\n%s' % (out, err))
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001191 raise Exception()
1192 filename = latest_checkout.split('/')[-1]
1193 tarball = os.path.join(tempdir, filename)
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001194 self.Print('Unpacking into %s...' % self.checkout_path)
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001195 gclient_utils.safe_makedirs(self.checkout_path)
1196 # TODO(hinoka): Use 7z for windows.
1197 cmd = ['tar', '--extract', '--ungzip',
1198 '--directory', self.checkout_path,
1199 '--file', tarball]
1200 gclient_utils.CheckCallAndFilter(
1201 cmd, stdout=sys.stdout, print_stdout=True)
1202
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001203 self.Print('Deleting temp file')
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001204 gclient_utils.rmtree(tempdir)
1205
1206 # Rewrite the repository root to match.
1207 tarball_url = scm.SVN.CaptureLocalInfo(
1208 ['.'], self.checkout_path)['Repository Root']
1209 tarball_parsed = urlparse.urlparse(tarball_url)
1210 tarball_root = '%s://%s' % (tarball_parsed.scheme,
1211 tarball_parsed.netloc)
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001212
1213 if tarball_root != local_root:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001214 self.Print('Switching repository root to %s' % local_root)
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001215 self._Run(['switch', '--relocate', tarball_root,
1216 local_root, self.checkout_path],
1217 options)
1218 except Exception as e:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001219 self.Print('We tried to get a source tarball but failed.')
1220 self.Print('Resuming normal operations.')
1221 self.Print(str(e))
hinoka@google.com2f2ca142014-01-07 03:59:18 +00001222
maruel@chromium.org6c48a302011-10-20 23:44:20 +00001223 gclient_utils.safe_makedirs(os.path.dirname(self.checkout_path))
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001224 # We need to checkout.
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001225 command = ['checkout', url, self.checkout_path]
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001226 command = self._AddAdditionalUpdateFlags(command, options, revision)
maruel@chromium.org669600d2010-09-01 19:06:31 +00001227 self._RunAndGetFileList(command, options, file_list, self._root_dir)
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +00001228 return self.Svnversion()
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001229
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001230 if not managed:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001231 self.Print(('________ unmanaged solution; skipping %s' % self.relpath))
borenet@google.comb09097a2014-04-09 19:09:08 +00001232 if os.path.exists(os.path.join(self.checkout_path, '.svn')):
1233 return self.Svnversion()
1234 return
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001235
maruel@chromium.org49fcb0c2011-09-23 14:34:38 +00001236 if 'URL' not in from_info:
1237 raise gclient_utils.Error(
1238 ('gclient is confused. Couldn\'t get the url for %s.\n'
1239 'Try using @unmanaged.\n%s') % (
1240 self.checkout_path, from_info))
1241
stip@chromium.org3031d732014-04-21 22:18:02 +00001242 # Look for locked directories.
1243 dir_info = scm.SVN.CaptureStatus(
1244 None, os.path.join(self.checkout_path, '.'))
1245 if any(d[0][2] == 'L' for d in dir_info):
1246 try:
1247 self._Run(['cleanup', self.checkout_path], options)
1248 except subprocess2.CalledProcessError, e:
1249 # Get the status again, svn cleanup may have cleaned up at least
1250 # something.
1251 dir_info = scm.SVN.CaptureStatus(
1252 None, os.path.join(self.checkout_path, '.'))
1253
phajdan.jr@chromium.orgd558c4b2011-09-22 18:56:24 +00001254 # Try to fix the failures by removing troublesome files.
1255 for d in dir_info:
1256 if d[0][2] == 'L':
1257 if d[0][0] == '!' and options.force:
kustermann@google.com1580d952013-08-19 07:31:40 +00001258 # We don't pass any files/directories to CaptureStatus and set
1259 # cwd=self.checkout_path, so we should get relative paths here.
1260 assert not os.path.isabs(d[1])
1261 path_to_remove = os.path.normpath(
1262 os.path.join(self.checkout_path, d[1]))
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001263 self.Print('Removing troublesome path %s' % path_to_remove)
kustermann@google.com1580d952013-08-19 07:31:40 +00001264 gclient_utils.rmtree(path_to_remove)
phajdan.jr@chromium.orgd558c4b2011-09-22 18:56:24 +00001265 else:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001266 self.Print(
1267 'Not removing troublesome path %s automatically.' % d[1])
phajdan.jr@chromium.orgd558c4b2011-09-22 18:56:24 +00001268 if d[0][0] == '!':
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001269 self.Print('You can pass --force to enable automatic removal.')
phajdan.jr@chromium.orgd558c4b2011-09-22 18:56:24 +00001270 raise e
maruel@chromium.orge407c9a2010-08-09 19:11:37 +00001271
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001272 # Retrieve the current HEAD version because svn is slow at null updates.
1273 if options.manually_grab_svn_rev and not revision:
maruel@chromium.orgd579fcf2011-12-13 20:36:03 +00001274 from_info_live = scm.SVN.CaptureRemoteInfo(from_info['URL'])
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001275 revision = str(from_info_live['Revision'])
1276 rev_str = ' at %s' % revision
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001277
borenet@google.comb09097a2014-04-09 19:09:08 +00001278 if from_info['URL'].rstrip('/') != base_url.rstrip('/'):
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001279 # The repository url changed, need to switch.
1280 try:
1281 to_info = scm.SVN.CaptureRemoteInfo(url)
1282 except (gclient_utils.Error, subprocess2.CalledProcessError):
1283 # The url is invalid or the server is not accessible, it's safer to bail
1284 # out right now.
1285 raise gclient_utils.Error('This url is unreachable: %s' % url)
1286 can_switch = ((from_info['Repository Root'] != to_info['Repository Root'])
1287 and (from_info['UUID'] == to_info['UUID']))
1288 if can_switch:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001289 self.Print('_____ relocating %s to a new checkout' % self.relpath)
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001290 # We have different roots, so check if we can switch --relocate.
1291 # Subversion only permits this if the repository UUIDs match.
1292 # Perform the switch --relocate, then rewrite the from_url
1293 # to reflect where we "are now." (This is the same way that
1294 # Subversion itself handles the metadata when switch --relocate
1295 # is used.) This makes the checks below for whether we
1296 # can update to a revision or have to switch to a different
1297 # branch work as expected.
1298 # TODO(maruel): TEST ME !
1299 command = ['switch', '--relocate',
1300 from_info['Repository Root'],
1301 to_info['Repository Root'],
1302 self.relpath]
1303 self._Run(command, options, cwd=self._root_dir)
1304 from_info['URL'] = from_info['URL'].replace(
1305 from_info['Repository Root'],
1306 to_info['Repository Root'])
1307 else:
1308 if not options.force and not options.reset:
1309 # Look for local modifications but ignore unversioned files.
1310 for status in scm.SVN.CaptureStatus(None, self.checkout_path):
1311 if status[0][0] != '?':
1312 raise gclient_utils.Error(
1313 ('Can\'t switch the checkout to %s; UUID don\'t match and '
1314 'there is local changes in %s. Delete the directory and '
1315 'try again.') % (url, self.checkout_path))
1316 # Ok delete it.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001317 self.Print('_____ switching %s to a new checkout' % self.relpath)
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001318 gclient_utils.rmtree(self.checkout_path)
1319 # We need to checkout.
1320 command = ['checkout', url, self.checkout_path]
1321 command = self._AddAdditionalUpdateFlags(command, options, revision)
1322 self._RunAndGetFileList(command, options, file_list, self._root_dir)
1323 return self.Svnversion()
1324
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001325 # If the provided url has a revision number that matches the revision
1326 # number of the existing directory, then we don't need to bother updating.
maruel@chromium.org2e0c6852009-09-24 00:02:07 +00001327 if not options.force and str(from_info['Revision']) == revision:
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001328 if options.verbose or not forced_revision:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001329 self.Print('_____ %s%s' % (self.relpath, rev_str), timestamp=False)
steveblock@chromium.org98e69452012-02-16 16:36:43 +00001330 else:
1331 command = ['update', self.checkout_path]
1332 command = self._AddAdditionalUpdateFlags(command, options, revision)
1333 self._RunAndGetFileList(command, options, file_list, self._root_dir)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001334
steveblock@chromium.org98e69452012-02-16 16:36:43 +00001335 # If --reset and --delete_unversioned_trees are specified, remove any
1336 # untracked files and directories.
1337 if options.reset and options.delete_unversioned_trees:
1338 for status in scm.SVN.CaptureStatus(None, self.checkout_path):
1339 full_path = os.path.join(self.checkout_path, status[1])
1340 if (status[0][0] == '?'
1341 and os.path.isdir(full_path)
1342 and not os.path.islink(full_path)):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001343 self.Print('_____ removing unversioned directory %s' % status[1])
digit@chromium.orgdc112ac2013-04-24 13:00:19 +00001344 gclient_utils.rmtree(full_path)
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +00001345 return self.Svnversion()
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001346
tony@chromium.org4b5b1772010-04-08 01:52:56 +00001347 def updatesingle(self, options, args, file_list):
tony@chromium.org4b5b1772010-04-08 01:52:56 +00001348 filename = args.pop()
tony@chromium.org57564662010-04-14 02:35:12 +00001349 if scm.SVN.AssertVersion("1.5")[0]:
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001350 if not os.path.exists(os.path.join(self.checkout_path, '.svn')):
tony@chromium.org57564662010-04-14 02:35:12 +00001351 # Create an empty checkout and then update the one file we want. Future
1352 # operations will only apply to the one file we checked out.
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001353 command = ["checkout", "--depth", "empty", self.url, self.checkout_path]
maruel@chromium.org669600d2010-09-01 19:06:31 +00001354 self._Run(command, options, cwd=self._root_dir)
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001355 if os.path.exists(os.path.join(self.checkout_path, filename)):
1356 os.remove(os.path.join(self.checkout_path, filename))
tony@chromium.org57564662010-04-14 02:35:12 +00001357 command = ["update", filename]
maruel@chromium.org669600d2010-09-01 19:06:31 +00001358 self._RunAndGetFileList(command, options, file_list)
tony@chromium.org57564662010-04-14 02:35:12 +00001359 # After the initial checkout, we can use update as if it were any other
1360 # dep.
1361 self.update(options, args, file_list)
1362 else:
1363 # If the installed version of SVN doesn't support --depth, fallback to
1364 # just exporting the file. This has the downside that revision
1365 # information is not stored next to the file, so we will have to
1366 # re-export the file every time we sync.
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001367 if not os.path.exists(self.checkout_path):
maruel@chromium.org6c48a302011-10-20 23:44:20 +00001368 gclient_utils.safe_makedirs(self.checkout_path)
tony@chromium.org57564662010-04-14 02:35:12 +00001369 command = ["export", os.path.join(self.url, filename),
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001370 os.path.join(self.checkout_path, filename)]
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001371 command = self._AddAdditionalUpdateFlags(command, options,
1372 options.revision)
maruel@chromium.org669600d2010-09-01 19:06:31 +00001373 self._Run(command, options, cwd=self._root_dir)
tony@chromium.org4b5b1772010-04-08 01:52:56 +00001374
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001375 def revert(self, options, _args, file_list):
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001376 """Reverts local modifications. Subversion specific.
1377
1378 All reverted files will be appended to file_list, even if Subversion
1379 doesn't know about them.
1380 """
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001381 if not os.path.isdir(self.checkout_path):
maruel@chromium.orgc0cc0872011-10-12 17:02:41 +00001382 if os.path.exists(self.checkout_path):
1383 gclient_utils.rmtree(self.checkout_path)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001384 # svn revert won't work if the directory doesn't exist. It needs to
1385 # checkout instead.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001386 self.Print('_____ %s is missing, synching instead' % self.relpath)
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001387 # Don't reuse the args.
1388 return self.update(options, [], file_list)
1389
maruel@chromium.orgc0cc0872011-10-12 17:02:41 +00001390 if not os.path.isdir(os.path.join(self.checkout_path, '.svn')):
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +00001391 if os.path.isdir(os.path.join(self.checkout_path, '.git')):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001392 self.Print('________ found .git directory; skipping %s' % self.relpath)
mmoss@chromium.org50fd47f2014-02-13 01:03:19 +00001393 return
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001394 if os.path.isdir(os.path.join(self.checkout_path, '.hg')):
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001395 self.Print('________ found .hg directory; skipping %s' % self.relpath)
szager@chromium.org6c2b49d2014-02-26 23:57:38 +00001396 return
maruel@chromium.orgc0cc0872011-10-12 17:02:41 +00001397 if not options.force:
1398 raise gclient_utils.Error('Invalid checkout path, aborting')
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001399 self.Print(
maruel@chromium.orgc0cc0872011-10-12 17:02:41 +00001400 '\n_____ %s is not a valid svn checkout, synching instead' %
1401 self.relpath)
1402 gclient_utils.rmtree(self.checkout_path)
1403 # Don't reuse the args.
1404 return self.update(options, [], file_list)
1405
maruel@chromium.org07ab60e2011-02-08 21:54:00 +00001406 def printcb(file_status):
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001407 if file_list is not None:
1408 file_list.append(file_status[1])
maruel@chromium.orgaa3dd472009-09-21 19:02:48 +00001409 if logging.getLogger().isEnabledFor(logging.INFO):
maruel@chromium.org07ab60e2011-02-08 21:54:00 +00001410 logging.info('%s%s' % (file_status[0], file_status[1]))
maruel@chromium.orgaa3dd472009-09-21 19:02:48 +00001411 else:
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001412 self.Print(os.path.join(self.checkout_path, file_status[1]))
maruel@chromium.org07ab60e2011-02-08 21:54:00 +00001413 scm.SVN.Revert(self.checkout_path, callback=printcb)
maruel@chromium.orgaa3dd472009-09-21 19:02:48 +00001414
maruel@chromium.org8b322b32011-11-01 19:05:50 +00001415 # Revert() may delete the directory altogether.
1416 if not os.path.isdir(self.checkout_path):
1417 # Don't reuse the args.
1418 return self.update(options, [], file_list)
1419
maruel@chromium.org810a50b2009-10-05 23:03:18 +00001420 try:
1421 # svn revert is so broken we don't even use it. Using
1422 # "svn up --revision BASE" achieve the same effect.
maruel@chromium.org07ab60e2011-02-08 21:54:00 +00001423 # file_list will contain duplicates.
maruel@chromium.org669600d2010-09-01 19:06:31 +00001424 self._RunAndGetFileList(['update', '--revision', 'BASE'], options,
1425 file_list)
maruel@chromium.org810a50b2009-10-05 23:03:18 +00001426 except OSError, e:
maruel@chromium.org07ab60e2011-02-08 21:54:00 +00001427 # Maybe the directory disapeared meanwhile. Do not throw an exception.
maruel@chromium.org810a50b2009-10-05 23:03:18 +00001428 logging.error('Failed to update:\n%s' % str(e))
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001429
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001430 def revinfo(self, _options, _args, _file_list):
msb@chromium.org0f282062009-11-06 20:14:02 +00001431 """Display revision"""
maruel@chromium.org54019f32010-09-09 13:50:11 +00001432 try:
1433 return scm.SVN.CaptureRevision(self.checkout_path)
maruel@chromium.org31cb48a2011-04-04 18:01:36 +00001434 except (gclient_utils.Error, subprocess2.CalledProcessError):
maruel@chromium.org54019f32010-09-09 13:50:11 +00001435 return None
msb@chromium.org0f282062009-11-06 20:14:02 +00001436
msb@chromium.orgcb5442b2009-09-22 16:51:24 +00001437 def runhooks(self, options, args, file_list):
1438 self.status(options, args, file_list)
1439
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001440 def status(self, options, args, file_list):
1441 """Display status information."""
maruel@chromium.org669600d2010-09-01 19:06:31 +00001442 command = ['status'] + args
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001443 if not os.path.isdir(self.checkout_path):
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001444 # svn status won't work if the directory doesn't exist.
szager@chromium.orgfe0d1902014-04-08 20:50:44 +00001445 self.Print(('\n________ couldn\'t run \'%s\' in \'%s\':\n'
maruel@chromium.org77e4eca2010-09-21 13:23:07 +00001446 'The directory does not exist.') %
1447 (' '.join(command), self.checkout_path))
maruel@chromium.org5f3eee32009-09-17 00:34:30 +00001448 # There's no file list to retrieve.
1449 else:
maruel@chromium.org669600d2010-09-01 19:06:31 +00001450 self._RunAndGetFileList(command, options, file_list)
msb@chromium.orge6f78352010-01-13 17:05:33 +00001451
iannucci@chromium.org396e1a62013-07-03 19:41:04 +00001452 def GetUsableRev(self, rev, _options):
dbeam@chromium.orge5d1e612011-12-19 19:49:19 +00001453 """Verifies the validity of the revision for this repository."""
1454 if not scm.SVN.IsValidRevision(url='%s@%s' % (self.url, rev)):
1455 raise gclient_utils.Error(
1456 ( '%s isn\'t a valid revision. Please check that your safesync_url is\n'
1457 'correct.') % rev)
1458 return rev
1459
msb@chromium.orge6f78352010-01-13 17:05:33 +00001460 def FullUrlForRelativeUrl(self, url):
1461 # Find the forth '/' and strip from there. A bit hackish.
1462 return '/'.join(self.url.split('/')[:4]) + url
tony@chromium.org99828122010-06-04 01:41:02 +00001463
maruel@chromium.org669600d2010-09-01 19:06:31 +00001464 def _Run(self, args, options, **kwargs):
1465 """Runs a commands that goes to stdout."""
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001466 kwargs.setdefault('cwd', self.checkout_path)
maruel@chromium.org669600d2010-09-01 19:06:31 +00001467 gclient_utils.CheckCallAndFilterAndHeader(['svn'] + args,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +00001468 always=options.verbose, **kwargs)
maruel@chromium.org669600d2010-09-01 19:06:31 +00001469
iannucci@chromium.org2702bcd2013-09-24 19:10:07 +00001470 def Svnversion(self):
1471 """Runs the lowest checked out revision in the current project."""
1472 info = scm.SVN.CaptureLocalInfo([], os.path.join(self.checkout_path, '.'))
1473 return info['Revision']
1474
maruel@chromium.org669600d2010-09-01 19:06:31 +00001475 def _RunAndGetFileList(self, args, options, file_list, cwd=None):
1476 """Runs a commands that goes to stdout and grabs the file listed."""
maruel@chromium.org8469bf92010-09-03 19:03:15 +00001477 cwd = cwd or self.checkout_path
maruel@chromium.orgce117f62011-01-17 20:04:25 +00001478 scm.SVN.RunAndGetFileList(
1479 options.verbose,
1480 args + ['--ignore-externals'],
1481 cwd=cwd,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +00001482 file_list=file_list)
maruel@chromium.org669600d2010-09-01 19:06:31 +00001483
maruel@chromium.org6e29d572010-06-04 17:32:20 +00001484 @staticmethod
maruel@chromium.org8e0e9262010-08-17 19:20:27 +00001485 def _AddAdditionalUpdateFlags(command, options, revision):
tony@chromium.org99828122010-06-04 01:41:02 +00001486 """Add additional flags to command depending on what options are set.
1487 command should be a list of strings that represents an svn command.
1488
1489 This method returns a new list to be used as a command."""
1490 new_command = command[:]
1491 if revision:
1492 new_command.extend(['--revision', str(revision).strip()])
maruel@chromium.org36ac2392011-10-12 16:36:11 +00001493 # We don't want interaction when jobs are used.
1494 if options.jobs > 1:
1495 new_command.append('--non-interactive')
tony@chromium.org99828122010-06-04 01:41:02 +00001496 # --force was added to 'svn update' in svn 1.5.
maruel@chromium.org36ac2392011-10-12 16:36:11 +00001497 # --accept was added to 'svn update' in svn 1.6.
1498 if not scm.SVN.AssertVersion('1.5')[0]:
1499 return new_command
1500
1501 # It's annoying to have it block in the middle of a sync, just sensible
1502 # defaults.
1503 if options.force:
tony@chromium.org99828122010-06-04 01:41:02 +00001504 new_command.append('--force')
maruel@chromium.org36ac2392011-10-12 16:36:11 +00001505 if command[0] != 'checkout' and scm.SVN.AssertVersion('1.6')[0]:
1506 new_command.extend(('--accept', 'theirs-conflict'))
1507 elif options.manually_grab_svn_rev:
1508 new_command.append('--force')
1509 if command[0] != 'checkout' and scm.SVN.AssertVersion('1.6')[0]:
1510 new_command.extend(('--accept', 'postpone'))
1511 elif command[0] != 'checkout' and scm.SVN.AssertVersion('1.6')[0]:
1512 new_command.extend(('--accept', 'postpone'))
tony@chromium.org99828122010-06-04 01:41:02 +00001513 return new_command