##// END OF EJS Templates
py3: handle keyword arguments correctly in exchange.py...
Pulkit Goyal -
r35356:a119e97b default
parent child Browse files
Show More
@@ -1,2214 +1,2215
1 # exchange.py - utility to exchange data between repos.
1 # exchange.py - utility to exchange data between repos.
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import collections
10 import collections
11 import errno
11 import errno
12 import hashlib
12 import hashlib
13
13
14 from .i18n import _
14 from .i18n import _
15 from .node import (
15 from .node import (
16 bin,
16 bin,
17 hex,
17 hex,
18 nullid,
18 nullid,
19 )
19 )
20 from . import (
20 from . import (
21 bookmarks as bookmod,
21 bookmarks as bookmod,
22 bundle2,
22 bundle2,
23 changegroup,
23 changegroup,
24 discovery,
24 discovery,
25 error,
25 error,
26 lock as lockmod,
26 lock as lockmod,
27 logexchange,
27 logexchange,
28 obsolete,
28 obsolete,
29 phases,
29 phases,
30 pushkey,
30 pushkey,
31 pycompat,
31 pycompat,
32 scmutil,
32 scmutil,
33 sslutil,
33 sslutil,
34 streamclone,
34 streamclone,
35 url as urlmod,
35 url as urlmod,
36 util,
36 util,
37 )
37 )
38
38
39 urlerr = util.urlerr
39 urlerr = util.urlerr
40 urlreq = util.urlreq
40 urlreq = util.urlreq
41
41
42 # Maps bundle version human names to changegroup versions.
42 # Maps bundle version human names to changegroup versions.
43 _bundlespeccgversions = {'v1': '01',
43 _bundlespeccgversions = {'v1': '01',
44 'v2': '02',
44 'v2': '02',
45 'packed1': 's1',
45 'packed1': 's1',
46 'bundle2': '02', #legacy
46 'bundle2': '02', #legacy
47 }
47 }
48
48
49 # Compression engines allowed in version 1. THIS SHOULD NEVER CHANGE.
49 # Compression engines allowed in version 1. THIS SHOULD NEVER CHANGE.
50 _bundlespecv1compengines = {'gzip', 'bzip2', 'none'}
50 _bundlespecv1compengines = {'gzip', 'bzip2', 'none'}
51
51
52 def parsebundlespec(repo, spec, strict=True, externalnames=False):
52 def parsebundlespec(repo, spec, strict=True, externalnames=False):
53 """Parse a bundle string specification into parts.
53 """Parse a bundle string specification into parts.
54
54
55 Bundle specifications denote a well-defined bundle/exchange format.
55 Bundle specifications denote a well-defined bundle/exchange format.
56 The content of a given specification should not change over time in
56 The content of a given specification should not change over time in
57 order to ensure that bundles produced by a newer version of Mercurial are
57 order to ensure that bundles produced by a newer version of Mercurial are
58 readable from an older version.
58 readable from an older version.
59
59
60 The string currently has the form:
60 The string currently has the form:
61
61
62 <compression>-<type>[;<parameter0>[;<parameter1>]]
62 <compression>-<type>[;<parameter0>[;<parameter1>]]
63
63
64 Where <compression> is one of the supported compression formats
64 Where <compression> is one of the supported compression formats
65 and <type> is (currently) a version string. A ";" can follow the type and
65 and <type> is (currently) a version string. A ";" can follow the type and
66 all text afterwards is interpreted as URI encoded, ";" delimited key=value
66 all text afterwards is interpreted as URI encoded, ";" delimited key=value
67 pairs.
67 pairs.
68
68
69 If ``strict`` is True (the default) <compression> is required. Otherwise,
69 If ``strict`` is True (the default) <compression> is required. Otherwise,
70 it is optional.
70 it is optional.
71
71
72 If ``externalnames`` is False (the default), the human-centric names will
72 If ``externalnames`` is False (the default), the human-centric names will
73 be converted to their internal representation.
73 be converted to their internal representation.
74
74
75 Returns a 3-tuple of (compression, version, parameters). Compression will
75 Returns a 3-tuple of (compression, version, parameters). Compression will
76 be ``None`` if not in strict mode and a compression isn't defined.
76 be ``None`` if not in strict mode and a compression isn't defined.
77
77
78 An ``InvalidBundleSpecification`` is raised when the specification is
78 An ``InvalidBundleSpecification`` is raised when the specification is
79 not syntactically well formed.
79 not syntactically well formed.
80
80
81 An ``UnsupportedBundleSpecification`` is raised when the compression or
81 An ``UnsupportedBundleSpecification`` is raised when the compression or
82 bundle type/version is not recognized.
82 bundle type/version is not recognized.
83
83
84 Note: this function will likely eventually return a more complex data
84 Note: this function will likely eventually return a more complex data
85 structure, including bundle2 part information.
85 structure, including bundle2 part information.
86 """
86 """
87 def parseparams(s):
87 def parseparams(s):
88 if ';' not in s:
88 if ';' not in s:
89 return s, {}
89 return s, {}
90
90
91 params = {}
91 params = {}
92 version, paramstr = s.split(';', 1)
92 version, paramstr = s.split(';', 1)
93
93
94 for p in paramstr.split(';'):
94 for p in paramstr.split(';'):
95 if '=' not in p:
95 if '=' not in p:
96 raise error.InvalidBundleSpecification(
96 raise error.InvalidBundleSpecification(
97 _('invalid bundle specification: '
97 _('invalid bundle specification: '
98 'missing "=" in parameter: %s') % p)
98 'missing "=" in parameter: %s') % p)
99
99
100 key, value = p.split('=', 1)
100 key, value = p.split('=', 1)
101 key = urlreq.unquote(key)
101 key = urlreq.unquote(key)
102 value = urlreq.unquote(value)
102 value = urlreq.unquote(value)
103 params[key] = value
103 params[key] = value
104
104
105 return version, params
105 return version, params
106
106
107
107
108 if strict and '-' not in spec:
108 if strict and '-' not in spec:
109 raise error.InvalidBundleSpecification(
109 raise error.InvalidBundleSpecification(
110 _('invalid bundle specification; '
110 _('invalid bundle specification; '
111 'must be prefixed with compression: %s') % spec)
111 'must be prefixed with compression: %s') % spec)
112
112
113 if '-' in spec:
113 if '-' in spec:
114 compression, version = spec.split('-', 1)
114 compression, version = spec.split('-', 1)
115
115
116 if compression not in util.compengines.supportedbundlenames:
116 if compression not in util.compengines.supportedbundlenames:
117 raise error.UnsupportedBundleSpecification(
117 raise error.UnsupportedBundleSpecification(
118 _('%s compression is not supported') % compression)
118 _('%s compression is not supported') % compression)
119
119
120 version, params = parseparams(version)
120 version, params = parseparams(version)
121
121
122 if version not in _bundlespeccgversions:
122 if version not in _bundlespeccgversions:
123 raise error.UnsupportedBundleSpecification(
123 raise error.UnsupportedBundleSpecification(
124 _('%s is not a recognized bundle version') % version)
124 _('%s is not a recognized bundle version') % version)
125 else:
125 else:
126 # Value could be just the compression or just the version, in which
126 # Value could be just the compression or just the version, in which
127 # case some defaults are assumed (but only when not in strict mode).
127 # case some defaults are assumed (but only when not in strict mode).
128 assert not strict
128 assert not strict
129
129
130 spec, params = parseparams(spec)
130 spec, params = parseparams(spec)
131
131
132 if spec in util.compengines.supportedbundlenames:
132 if spec in util.compengines.supportedbundlenames:
133 compression = spec
133 compression = spec
134 version = 'v1'
134 version = 'v1'
135 # Generaldelta repos require v2.
135 # Generaldelta repos require v2.
136 if 'generaldelta' in repo.requirements:
136 if 'generaldelta' in repo.requirements:
137 version = 'v2'
137 version = 'v2'
138 # Modern compression engines require v2.
138 # Modern compression engines require v2.
139 if compression not in _bundlespecv1compengines:
139 if compression not in _bundlespecv1compengines:
140 version = 'v2'
140 version = 'v2'
141 elif spec in _bundlespeccgversions:
141 elif spec in _bundlespeccgversions:
142 if spec == 'packed1':
142 if spec == 'packed1':
143 compression = 'none'
143 compression = 'none'
144 else:
144 else:
145 compression = 'bzip2'
145 compression = 'bzip2'
146 version = spec
146 version = spec
147 else:
147 else:
148 raise error.UnsupportedBundleSpecification(
148 raise error.UnsupportedBundleSpecification(
149 _('%s is not a recognized bundle specification') % spec)
149 _('%s is not a recognized bundle specification') % spec)
150
150
151 # Bundle version 1 only supports a known set of compression engines.
151 # Bundle version 1 only supports a known set of compression engines.
152 if version == 'v1' and compression not in _bundlespecv1compengines:
152 if version == 'v1' and compression not in _bundlespecv1compengines:
153 raise error.UnsupportedBundleSpecification(
153 raise error.UnsupportedBundleSpecification(
154 _('compression engine %s is not supported on v1 bundles') %
154 _('compression engine %s is not supported on v1 bundles') %
155 compression)
155 compression)
156
156
157 # The specification for packed1 can optionally declare the data formats
157 # The specification for packed1 can optionally declare the data formats
158 # required to apply it. If we see this metadata, compare against what the
158 # required to apply it. If we see this metadata, compare against what the
159 # repo supports and error if the bundle isn't compatible.
159 # repo supports and error if the bundle isn't compatible.
160 if version == 'packed1' and 'requirements' in params:
160 if version == 'packed1' and 'requirements' in params:
161 requirements = set(params['requirements'].split(','))
161 requirements = set(params['requirements'].split(','))
162 missingreqs = requirements - repo.supportedformats
162 missingreqs = requirements - repo.supportedformats
163 if missingreqs:
163 if missingreqs:
164 raise error.UnsupportedBundleSpecification(
164 raise error.UnsupportedBundleSpecification(
165 _('missing support for repository features: %s') %
165 _('missing support for repository features: %s') %
166 ', '.join(sorted(missingreqs)))
166 ', '.join(sorted(missingreqs)))
167
167
168 if not externalnames:
168 if not externalnames:
169 engine = util.compengines.forbundlename(compression)
169 engine = util.compengines.forbundlename(compression)
170 compression = engine.bundletype()[1]
170 compression = engine.bundletype()[1]
171 version = _bundlespeccgversions[version]
171 version = _bundlespeccgversions[version]
172 return compression, version, params
172 return compression, version, params
173
173
174 def readbundle(ui, fh, fname, vfs=None):
174 def readbundle(ui, fh, fname, vfs=None):
175 header = changegroup.readexactly(fh, 4)
175 header = changegroup.readexactly(fh, 4)
176
176
177 alg = None
177 alg = None
178 if not fname:
178 if not fname:
179 fname = "stream"
179 fname = "stream"
180 if not header.startswith('HG') and header.startswith('\0'):
180 if not header.startswith('HG') and header.startswith('\0'):
181 fh = changegroup.headerlessfixup(fh, header)
181 fh = changegroup.headerlessfixup(fh, header)
182 header = "HG10"
182 header = "HG10"
183 alg = 'UN'
183 alg = 'UN'
184 elif vfs:
184 elif vfs:
185 fname = vfs.join(fname)
185 fname = vfs.join(fname)
186
186
187 magic, version = header[0:2], header[2:4]
187 magic, version = header[0:2], header[2:4]
188
188
189 if magic != 'HG':
189 if magic != 'HG':
190 raise error.Abort(_('%s: not a Mercurial bundle') % fname)
190 raise error.Abort(_('%s: not a Mercurial bundle') % fname)
191 if version == '10':
191 if version == '10':
192 if alg is None:
192 if alg is None:
193 alg = changegroup.readexactly(fh, 2)
193 alg = changegroup.readexactly(fh, 2)
194 return changegroup.cg1unpacker(fh, alg)
194 return changegroup.cg1unpacker(fh, alg)
195 elif version.startswith('2'):
195 elif version.startswith('2'):
196 return bundle2.getunbundler(ui, fh, magicstring=magic + version)
196 return bundle2.getunbundler(ui, fh, magicstring=magic + version)
197 elif version == 'S1':
197 elif version == 'S1':
198 return streamclone.streamcloneapplier(fh)
198 return streamclone.streamcloneapplier(fh)
199 else:
199 else:
200 raise error.Abort(_('%s: unknown bundle version %s') % (fname, version))
200 raise error.Abort(_('%s: unknown bundle version %s') % (fname, version))
201
201
202 def getbundlespec(ui, fh):
202 def getbundlespec(ui, fh):
203 """Infer the bundlespec from a bundle file handle.
203 """Infer the bundlespec from a bundle file handle.
204
204
205 The input file handle is seeked and the original seek position is not
205 The input file handle is seeked and the original seek position is not
206 restored.
206 restored.
207 """
207 """
208 def speccompression(alg):
208 def speccompression(alg):
209 try:
209 try:
210 return util.compengines.forbundletype(alg).bundletype()[0]
210 return util.compengines.forbundletype(alg).bundletype()[0]
211 except KeyError:
211 except KeyError:
212 return None
212 return None
213
213
214 b = readbundle(ui, fh, None)
214 b = readbundle(ui, fh, None)
215 if isinstance(b, changegroup.cg1unpacker):
215 if isinstance(b, changegroup.cg1unpacker):
216 alg = b._type
216 alg = b._type
217 if alg == '_truncatedBZ':
217 if alg == '_truncatedBZ':
218 alg = 'BZ'
218 alg = 'BZ'
219 comp = speccompression(alg)
219 comp = speccompression(alg)
220 if not comp:
220 if not comp:
221 raise error.Abort(_('unknown compression algorithm: %s') % alg)
221 raise error.Abort(_('unknown compression algorithm: %s') % alg)
222 return '%s-v1' % comp
222 return '%s-v1' % comp
223 elif isinstance(b, bundle2.unbundle20):
223 elif isinstance(b, bundle2.unbundle20):
224 if 'Compression' in b.params:
224 if 'Compression' in b.params:
225 comp = speccompression(b.params['Compression'])
225 comp = speccompression(b.params['Compression'])
226 if not comp:
226 if not comp:
227 raise error.Abort(_('unknown compression algorithm: %s') % comp)
227 raise error.Abort(_('unknown compression algorithm: %s') % comp)
228 else:
228 else:
229 comp = 'none'
229 comp = 'none'
230
230
231 version = None
231 version = None
232 for part in b.iterparts():
232 for part in b.iterparts():
233 if part.type == 'changegroup':
233 if part.type == 'changegroup':
234 version = part.params['version']
234 version = part.params['version']
235 if version in ('01', '02'):
235 if version in ('01', '02'):
236 version = 'v2'
236 version = 'v2'
237 else:
237 else:
238 raise error.Abort(_('changegroup version %s does not have '
238 raise error.Abort(_('changegroup version %s does not have '
239 'a known bundlespec') % version,
239 'a known bundlespec') % version,
240 hint=_('try upgrading your Mercurial '
240 hint=_('try upgrading your Mercurial '
241 'client'))
241 'client'))
242
242
243 if not version:
243 if not version:
244 raise error.Abort(_('could not identify changegroup version in '
244 raise error.Abort(_('could not identify changegroup version in '
245 'bundle'))
245 'bundle'))
246
246
247 return '%s-%s' % (comp, version)
247 return '%s-%s' % (comp, version)
248 elif isinstance(b, streamclone.streamcloneapplier):
248 elif isinstance(b, streamclone.streamcloneapplier):
249 requirements = streamclone.readbundle1header(fh)[2]
249 requirements = streamclone.readbundle1header(fh)[2]
250 params = 'requirements=%s' % ','.join(sorted(requirements))
250 params = 'requirements=%s' % ','.join(sorted(requirements))
251 return 'none-packed1;%s' % urlreq.quote(params)
251 return 'none-packed1;%s' % urlreq.quote(params)
252 else:
252 else:
253 raise error.Abort(_('unknown bundle type: %s') % b)
253 raise error.Abort(_('unknown bundle type: %s') % b)
254
254
255 def _computeoutgoing(repo, heads, common):
255 def _computeoutgoing(repo, heads, common):
256 """Computes which revs are outgoing given a set of common
256 """Computes which revs are outgoing given a set of common
257 and a set of heads.
257 and a set of heads.
258
258
259 This is a separate function so extensions can have access to
259 This is a separate function so extensions can have access to
260 the logic.
260 the logic.
261
261
262 Returns a discovery.outgoing object.
262 Returns a discovery.outgoing object.
263 """
263 """
264 cl = repo.changelog
264 cl = repo.changelog
265 if common:
265 if common:
266 hasnode = cl.hasnode
266 hasnode = cl.hasnode
267 common = [n for n in common if hasnode(n)]
267 common = [n for n in common if hasnode(n)]
268 else:
268 else:
269 common = [nullid]
269 common = [nullid]
270 if not heads:
270 if not heads:
271 heads = cl.heads()
271 heads = cl.heads()
272 return discovery.outgoing(repo, common, heads)
272 return discovery.outgoing(repo, common, heads)
273
273
274 def _forcebundle1(op):
274 def _forcebundle1(op):
275 """return true if a pull/push must use bundle1
275 """return true if a pull/push must use bundle1
276
276
277 This function is used to allow testing of the older bundle version"""
277 This function is used to allow testing of the older bundle version"""
278 ui = op.repo.ui
278 ui = op.repo.ui
279 forcebundle1 = False
279 forcebundle1 = False
280 # The goal is this config is to allow developer to choose the bundle
280 # The goal is this config is to allow developer to choose the bundle
281 # version used during exchanged. This is especially handy during test.
281 # version used during exchanged. This is especially handy during test.
282 # Value is a list of bundle version to be picked from, highest version
282 # Value is a list of bundle version to be picked from, highest version
283 # should be used.
283 # should be used.
284 #
284 #
285 # developer config: devel.legacy.exchange
285 # developer config: devel.legacy.exchange
286 exchange = ui.configlist('devel', 'legacy.exchange')
286 exchange = ui.configlist('devel', 'legacy.exchange')
287 forcebundle1 = 'bundle2' not in exchange and 'bundle1' in exchange
287 forcebundle1 = 'bundle2' not in exchange and 'bundle1' in exchange
288 return forcebundle1 or not op.remote.capable('bundle2')
288 return forcebundle1 or not op.remote.capable('bundle2')
289
289
290 class pushoperation(object):
290 class pushoperation(object):
291 """A object that represent a single push operation
291 """A object that represent a single push operation
292
292
293 Its purpose is to carry push related state and very common operations.
293 Its purpose is to carry push related state and very common operations.
294
294
295 A new pushoperation should be created at the beginning of each push and
295 A new pushoperation should be created at the beginning of each push and
296 discarded afterward.
296 discarded afterward.
297 """
297 """
298
298
299 def __init__(self, repo, remote, force=False, revs=None, newbranch=False,
299 def __init__(self, repo, remote, force=False, revs=None, newbranch=False,
300 bookmarks=(), pushvars=None):
300 bookmarks=(), pushvars=None):
301 # repo we push from
301 # repo we push from
302 self.repo = repo
302 self.repo = repo
303 self.ui = repo.ui
303 self.ui = repo.ui
304 # repo we push to
304 # repo we push to
305 self.remote = remote
305 self.remote = remote
306 # force option provided
306 # force option provided
307 self.force = force
307 self.force = force
308 # revs to be pushed (None is "all")
308 # revs to be pushed (None is "all")
309 self.revs = revs
309 self.revs = revs
310 # bookmark explicitly pushed
310 # bookmark explicitly pushed
311 self.bookmarks = bookmarks
311 self.bookmarks = bookmarks
312 # allow push of new branch
312 # allow push of new branch
313 self.newbranch = newbranch
313 self.newbranch = newbranch
314 # step already performed
314 # step already performed
315 # (used to check what steps have been already performed through bundle2)
315 # (used to check what steps have been already performed through bundle2)
316 self.stepsdone = set()
316 self.stepsdone = set()
317 # Integer version of the changegroup push result
317 # Integer version of the changegroup push result
318 # - None means nothing to push
318 # - None means nothing to push
319 # - 0 means HTTP error
319 # - 0 means HTTP error
320 # - 1 means we pushed and remote head count is unchanged *or*
320 # - 1 means we pushed and remote head count is unchanged *or*
321 # we have outgoing changesets but refused to push
321 # we have outgoing changesets but refused to push
322 # - other values as described by addchangegroup()
322 # - other values as described by addchangegroup()
323 self.cgresult = None
323 self.cgresult = None
324 # Boolean value for the bookmark push
324 # Boolean value for the bookmark push
325 self.bkresult = None
325 self.bkresult = None
326 # discover.outgoing object (contains common and outgoing data)
326 # discover.outgoing object (contains common and outgoing data)
327 self.outgoing = None
327 self.outgoing = None
328 # all remote topological heads before the push
328 # all remote topological heads before the push
329 self.remoteheads = None
329 self.remoteheads = None
330 # Details of the remote branch pre and post push
330 # Details of the remote branch pre and post push
331 #
331 #
332 # mapping: {'branch': ([remoteheads],
332 # mapping: {'branch': ([remoteheads],
333 # [newheads],
333 # [newheads],
334 # [unsyncedheads],
334 # [unsyncedheads],
335 # [discardedheads])}
335 # [discardedheads])}
336 # - branch: the branch name
336 # - branch: the branch name
337 # - remoteheads: the list of remote heads known locally
337 # - remoteheads: the list of remote heads known locally
338 # None if the branch is new
338 # None if the branch is new
339 # - newheads: the new remote heads (known locally) with outgoing pushed
339 # - newheads: the new remote heads (known locally) with outgoing pushed
340 # - unsyncedheads: the list of remote heads unknown locally.
340 # - unsyncedheads: the list of remote heads unknown locally.
341 # - discardedheads: the list of remote heads made obsolete by the push
341 # - discardedheads: the list of remote heads made obsolete by the push
342 self.pushbranchmap = None
342 self.pushbranchmap = None
343 # testable as a boolean indicating if any nodes are missing locally.
343 # testable as a boolean indicating if any nodes are missing locally.
344 self.incoming = None
344 self.incoming = None
345 # summary of the remote phase situation
345 # summary of the remote phase situation
346 self.remotephases = None
346 self.remotephases = None
347 # phases changes that must be pushed along side the changesets
347 # phases changes that must be pushed along side the changesets
348 self.outdatedphases = None
348 self.outdatedphases = None
349 # phases changes that must be pushed if changeset push fails
349 # phases changes that must be pushed if changeset push fails
350 self.fallbackoutdatedphases = None
350 self.fallbackoutdatedphases = None
351 # outgoing obsmarkers
351 # outgoing obsmarkers
352 self.outobsmarkers = set()
352 self.outobsmarkers = set()
353 # outgoing bookmarks
353 # outgoing bookmarks
354 self.outbookmarks = []
354 self.outbookmarks = []
355 # transaction manager
355 # transaction manager
356 self.trmanager = None
356 self.trmanager = None
357 # map { pushkey partid -> callback handling failure}
357 # map { pushkey partid -> callback handling failure}
358 # used to handle exception from mandatory pushkey part failure
358 # used to handle exception from mandatory pushkey part failure
359 self.pkfailcb = {}
359 self.pkfailcb = {}
360 # an iterable of pushvars or None
360 # an iterable of pushvars or None
361 self.pushvars = pushvars
361 self.pushvars = pushvars
362
362
363 @util.propertycache
363 @util.propertycache
364 def futureheads(self):
364 def futureheads(self):
365 """future remote heads if the changeset push succeeds"""
365 """future remote heads if the changeset push succeeds"""
366 return self.outgoing.missingheads
366 return self.outgoing.missingheads
367
367
368 @util.propertycache
368 @util.propertycache
369 def fallbackheads(self):
369 def fallbackheads(self):
370 """future remote heads if the changeset push fails"""
370 """future remote heads if the changeset push fails"""
371 if self.revs is None:
371 if self.revs is None:
372 # not target to push, all common are relevant
372 # not target to push, all common are relevant
373 return self.outgoing.commonheads
373 return self.outgoing.commonheads
374 unfi = self.repo.unfiltered()
374 unfi = self.repo.unfiltered()
375 # I want cheads = heads(::missingheads and ::commonheads)
375 # I want cheads = heads(::missingheads and ::commonheads)
376 # (missingheads is revs with secret changeset filtered out)
376 # (missingheads is revs with secret changeset filtered out)
377 #
377 #
378 # This can be expressed as:
378 # This can be expressed as:
379 # cheads = ( (missingheads and ::commonheads)
379 # cheads = ( (missingheads and ::commonheads)
380 # + (commonheads and ::missingheads))"
380 # + (commonheads and ::missingheads))"
381 # )
381 # )
382 #
382 #
383 # while trying to push we already computed the following:
383 # while trying to push we already computed the following:
384 # common = (::commonheads)
384 # common = (::commonheads)
385 # missing = ((commonheads::missingheads) - commonheads)
385 # missing = ((commonheads::missingheads) - commonheads)
386 #
386 #
387 # We can pick:
387 # We can pick:
388 # * missingheads part of common (::commonheads)
388 # * missingheads part of common (::commonheads)
389 common = self.outgoing.common
389 common = self.outgoing.common
390 nm = self.repo.changelog.nodemap
390 nm = self.repo.changelog.nodemap
391 cheads = [node for node in self.revs if nm[node] in common]
391 cheads = [node for node in self.revs if nm[node] in common]
392 # and
392 # and
393 # * commonheads parents on missing
393 # * commonheads parents on missing
394 revset = unfi.set('%ln and parents(roots(%ln))',
394 revset = unfi.set('%ln and parents(roots(%ln))',
395 self.outgoing.commonheads,
395 self.outgoing.commonheads,
396 self.outgoing.missing)
396 self.outgoing.missing)
397 cheads.extend(c.node() for c in revset)
397 cheads.extend(c.node() for c in revset)
398 return cheads
398 return cheads
399
399
400 @property
400 @property
401 def commonheads(self):
401 def commonheads(self):
402 """set of all common heads after changeset bundle push"""
402 """set of all common heads after changeset bundle push"""
403 if self.cgresult:
403 if self.cgresult:
404 return self.futureheads
404 return self.futureheads
405 else:
405 else:
406 return self.fallbackheads
406 return self.fallbackheads
407
407
408 # mapping of message used when pushing bookmark
408 # mapping of message used when pushing bookmark
409 bookmsgmap = {'update': (_("updating bookmark %s\n"),
409 bookmsgmap = {'update': (_("updating bookmark %s\n"),
410 _('updating bookmark %s failed!\n')),
410 _('updating bookmark %s failed!\n')),
411 'export': (_("exporting bookmark %s\n"),
411 'export': (_("exporting bookmark %s\n"),
412 _('exporting bookmark %s failed!\n')),
412 _('exporting bookmark %s failed!\n')),
413 'delete': (_("deleting remote bookmark %s\n"),
413 'delete': (_("deleting remote bookmark %s\n"),
414 _('deleting remote bookmark %s failed!\n')),
414 _('deleting remote bookmark %s failed!\n')),
415 }
415 }
416
416
417
417
418 def push(repo, remote, force=False, revs=None, newbranch=False, bookmarks=(),
418 def push(repo, remote, force=False, revs=None, newbranch=False, bookmarks=(),
419 opargs=None):
419 opargs=None):
420 '''Push outgoing changesets (limited by revs) from a local
420 '''Push outgoing changesets (limited by revs) from a local
421 repository to remote. Return an integer:
421 repository to remote. Return an integer:
422 - None means nothing to push
422 - None means nothing to push
423 - 0 means HTTP error
423 - 0 means HTTP error
424 - 1 means we pushed and remote head count is unchanged *or*
424 - 1 means we pushed and remote head count is unchanged *or*
425 we have outgoing changesets but refused to push
425 we have outgoing changesets but refused to push
426 - other values as described by addchangegroup()
426 - other values as described by addchangegroup()
427 '''
427 '''
428 if opargs is None:
428 if opargs is None:
429 opargs = {}
429 opargs = {}
430 pushop = pushoperation(repo, remote, force, revs, newbranch, bookmarks,
430 pushop = pushoperation(repo, remote, force, revs, newbranch, bookmarks,
431 **pycompat.strkwargs(opargs))
431 **pycompat.strkwargs(opargs))
432 if pushop.remote.local():
432 if pushop.remote.local():
433 missing = (set(pushop.repo.requirements)
433 missing = (set(pushop.repo.requirements)
434 - pushop.remote.local().supported)
434 - pushop.remote.local().supported)
435 if missing:
435 if missing:
436 msg = _("required features are not"
436 msg = _("required features are not"
437 " supported in the destination:"
437 " supported in the destination:"
438 " %s") % (', '.join(sorted(missing)))
438 " %s") % (', '.join(sorted(missing)))
439 raise error.Abort(msg)
439 raise error.Abort(msg)
440
440
441 if not pushop.remote.canpush():
441 if not pushop.remote.canpush():
442 raise error.Abort(_("destination does not support push"))
442 raise error.Abort(_("destination does not support push"))
443
443
444 if not pushop.remote.capable('unbundle'):
444 if not pushop.remote.capable('unbundle'):
445 raise error.Abort(_('cannot push: destination does not support the '
445 raise error.Abort(_('cannot push: destination does not support the '
446 'unbundle wire protocol command'))
446 'unbundle wire protocol command'))
447
447
448 # get lock as we might write phase data
448 # get lock as we might write phase data
449 wlock = lock = None
449 wlock = lock = None
450 try:
450 try:
451 # bundle2 push may receive a reply bundle touching bookmarks or other
451 # bundle2 push may receive a reply bundle touching bookmarks or other
452 # things requiring the wlock. Take it now to ensure proper ordering.
452 # things requiring the wlock. Take it now to ensure proper ordering.
453 maypushback = pushop.ui.configbool('experimental', 'bundle2.pushback')
453 maypushback = pushop.ui.configbool('experimental', 'bundle2.pushback')
454 if (not _forcebundle1(pushop)) and maypushback:
454 if (not _forcebundle1(pushop)) and maypushback:
455 wlock = pushop.repo.wlock()
455 wlock = pushop.repo.wlock()
456 lock = pushop.repo.lock()
456 lock = pushop.repo.lock()
457 pushop.trmanager = transactionmanager(pushop.repo,
457 pushop.trmanager = transactionmanager(pushop.repo,
458 'push-response',
458 'push-response',
459 pushop.remote.url())
459 pushop.remote.url())
460 except IOError as err:
460 except IOError as err:
461 if err.errno != errno.EACCES:
461 if err.errno != errno.EACCES:
462 raise
462 raise
463 # source repo cannot be locked.
463 # source repo cannot be locked.
464 # We do not abort the push, but just disable the local phase
464 # We do not abort the push, but just disable the local phase
465 # synchronisation.
465 # synchronisation.
466 msg = 'cannot lock source repository: %s\n' % err
466 msg = 'cannot lock source repository: %s\n' % err
467 pushop.ui.debug(msg)
467 pushop.ui.debug(msg)
468
468
469 with wlock or util.nullcontextmanager(), \
469 with wlock or util.nullcontextmanager(), \
470 lock or util.nullcontextmanager(), \
470 lock or util.nullcontextmanager(), \
471 pushop.trmanager or util.nullcontextmanager():
471 pushop.trmanager or util.nullcontextmanager():
472 pushop.repo.checkpush(pushop)
472 pushop.repo.checkpush(pushop)
473 _pushdiscovery(pushop)
473 _pushdiscovery(pushop)
474 if not _forcebundle1(pushop):
474 if not _forcebundle1(pushop):
475 _pushbundle2(pushop)
475 _pushbundle2(pushop)
476 _pushchangeset(pushop)
476 _pushchangeset(pushop)
477 _pushsyncphase(pushop)
477 _pushsyncphase(pushop)
478 _pushobsolete(pushop)
478 _pushobsolete(pushop)
479 _pushbookmark(pushop)
479 _pushbookmark(pushop)
480
480
481 return pushop
481 return pushop
482
482
483 # list of steps to perform discovery before push
483 # list of steps to perform discovery before push
484 pushdiscoveryorder = []
484 pushdiscoveryorder = []
485
485
486 # Mapping between step name and function
486 # Mapping between step name and function
487 #
487 #
488 # This exists to help extensions wrap steps if necessary
488 # This exists to help extensions wrap steps if necessary
489 pushdiscoverymapping = {}
489 pushdiscoverymapping = {}
490
490
491 def pushdiscovery(stepname):
491 def pushdiscovery(stepname):
492 """decorator for function performing discovery before push
492 """decorator for function performing discovery before push
493
493
494 The function is added to the step -> function mapping and appended to the
494 The function is added to the step -> function mapping and appended to the
495 list of steps. Beware that decorated function will be added in order (this
495 list of steps. Beware that decorated function will be added in order (this
496 may matter).
496 may matter).
497
497
498 You can only use this decorator for a new step, if you want to wrap a step
498 You can only use this decorator for a new step, if you want to wrap a step
499 from an extension, change the pushdiscovery dictionary directly."""
499 from an extension, change the pushdiscovery dictionary directly."""
500 def dec(func):
500 def dec(func):
501 assert stepname not in pushdiscoverymapping
501 assert stepname not in pushdiscoverymapping
502 pushdiscoverymapping[stepname] = func
502 pushdiscoverymapping[stepname] = func
503 pushdiscoveryorder.append(stepname)
503 pushdiscoveryorder.append(stepname)
504 return func
504 return func
505 return dec
505 return dec
506
506
507 def _pushdiscovery(pushop):
507 def _pushdiscovery(pushop):
508 """Run all discovery steps"""
508 """Run all discovery steps"""
509 for stepname in pushdiscoveryorder:
509 for stepname in pushdiscoveryorder:
510 step = pushdiscoverymapping[stepname]
510 step = pushdiscoverymapping[stepname]
511 step(pushop)
511 step(pushop)
512
512
513 @pushdiscovery('changeset')
513 @pushdiscovery('changeset')
514 def _pushdiscoverychangeset(pushop):
514 def _pushdiscoverychangeset(pushop):
515 """discover the changeset that need to be pushed"""
515 """discover the changeset that need to be pushed"""
516 fci = discovery.findcommonincoming
516 fci = discovery.findcommonincoming
517 if pushop.revs:
517 if pushop.revs:
518 commoninc = fci(pushop.repo, pushop.remote, force=pushop.force,
518 commoninc = fci(pushop.repo, pushop.remote, force=pushop.force,
519 ancestorsof=pushop.revs)
519 ancestorsof=pushop.revs)
520 else:
520 else:
521 commoninc = fci(pushop.repo, pushop.remote, force=pushop.force)
521 commoninc = fci(pushop.repo, pushop.remote, force=pushop.force)
522 common, inc, remoteheads = commoninc
522 common, inc, remoteheads = commoninc
523 fco = discovery.findcommonoutgoing
523 fco = discovery.findcommonoutgoing
524 outgoing = fco(pushop.repo, pushop.remote, onlyheads=pushop.revs,
524 outgoing = fco(pushop.repo, pushop.remote, onlyheads=pushop.revs,
525 commoninc=commoninc, force=pushop.force)
525 commoninc=commoninc, force=pushop.force)
526 pushop.outgoing = outgoing
526 pushop.outgoing = outgoing
527 pushop.remoteheads = remoteheads
527 pushop.remoteheads = remoteheads
528 pushop.incoming = inc
528 pushop.incoming = inc
529
529
530 @pushdiscovery('phase')
530 @pushdiscovery('phase')
531 def _pushdiscoveryphase(pushop):
531 def _pushdiscoveryphase(pushop):
532 """discover the phase that needs to be pushed
532 """discover the phase that needs to be pushed
533
533
534 (computed for both success and failure case for changesets push)"""
534 (computed for both success and failure case for changesets push)"""
535 outgoing = pushop.outgoing
535 outgoing = pushop.outgoing
536 unfi = pushop.repo.unfiltered()
536 unfi = pushop.repo.unfiltered()
537 remotephases = pushop.remote.listkeys('phases')
537 remotephases = pushop.remote.listkeys('phases')
538 if (pushop.ui.configbool('ui', '_usedassubrepo')
538 if (pushop.ui.configbool('ui', '_usedassubrepo')
539 and remotephases # server supports phases
539 and remotephases # server supports phases
540 and not pushop.outgoing.missing # no changesets to be pushed
540 and not pushop.outgoing.missing # no changesets to be pushed
541 and remotephases.get('publishing', False)):
541 and remotephases.get('publishing', False)):
542 # When:
542 # When:
543 # - this is a subrepo push
543 # - this is a subrepo push
544 # - and remote support phase
544 # - and remote support phase
545 # - and no changeset are to be pushed
545 # - and no changeset are to be pushed
546 # - and remote is publishing
546 # - and remote is publishing
547 # We may be in issue 3781 case!
547 # We may be in issue 3781 case!
548 # We drop the possible phase synchronisation done by
548 # We drop the possible phase synchronisation done by
549 # courtesy to publish changesets possibly locally draft
549 # courtesy to publish changesets possibly locally draft
550 # on the remote.
550 # on the remote.
551 pushop.outdatedphases = []
551 pushop.outdatedphases = []
552 pushop.fallbackoutdatedphases = []
552 pushop.fallbackoutdatedphases = []
553 return
553 return
554
554
555 pushop.remotephases = phases.remotephasessummary(pushop.repo,
555 pushop.remotephases = phases.remotephasessummary(pushop.repo,
556 pushop.fallbackheads,
556 pushop.fallbackheads,
557 remotephases)
557 remotephases)
558 droots = pushop.remotephases.draftroots
558 droots = pushop.remotephases.draftroots
559
559
560 extracond = ''
560 extracond = ''
561 if not pushop.remotephases.publishing:
561 if not pushop.remotephases.publishing:
562 extracond = ' and public()'
562 extracond = ' and public()'
563 revset = 'heads((%%ln::%%ln) %s)' % extracond
563 revset = 'heads((%%ln::%%ln) %s)' % extracond
564 # Get the list of all revs draft on remote by public here.
564 # Get the list of all revs draft on remote by public here.
565 # XXX Beware that revset break if droots is not strictly
565 # XXX Beware that revset break if droots is not strictly
566 # XXX root we may want to ensure it is but it is costly
566 # XXX root we may want to ensure it is but it is costly
567 fallback = list(unfi.set(revset, droots, pushop.fallbackheads))
567 fallback = list(unfi.set(revset, droots, pushop.fallbackheads))
568 if not outgoing.missing:
568 if not outgoing.missing:
569 future = fallback
569 future = fallback
570 else:
570 else:
571 # adds changeset we are going to push as draft
571 # adds changeset we are going to push as draft
572 #
572 #
573 # should not be necessary for publishing server, but because of an
573 # should not be necessary for publishing server, but because of an
574 # issue fixed in xxxxx we have to do it anyway.
574 # issue fixed in xxxxx we have to do it anyway.
575 fdroots = list(unfi.set('roots(%ln + %ln::)',
575 fdroots = list(unfi.set('roots(%ln + %ln::)',
576 outgoing.missing, droots))
576 outgoing.missing, droots))
577 fdroots = [f.node() for f in fdroots]
577 fdroots = [f.node() for f in fdroots]
578 future = list(unfi.set(revset, fdroots, pushop.futureheads))
578 future = list(unfi.set(revset, fdroots, pushop.futureheads))
579 pushop.outdatedphases = future
579 pushop.outdatedphases = future
580 pushop.fallbackoutdatedphases = fallback
580 pushop.fallbackoutdatedphases = fallback
581
581
582 @pushdiscovery('obsmarker')
582 @pushdiscovery('obsmarker')
583 def _pushdiscoveryobsmarkers(pushop):
583 def _pushdiscoveryobsmarkers(pushop):
584 if (obsolete.isenabled(pushop.repo, obsolete.exchangeopt)
584 if (obsolete.isenabled(pushop.repo, obsolete.exchangeopt)
585 and pushop.repo.obsstore
585 and pushop.repo.obsstore
586 and 'obsolete' in pushop.remote.listkeys('namespaces')):
586 and 'obsolete' in pushop.remote.listkeys('namespaces')):
587 repo = pushop.repo
587 repo = pushop.repo
588 # very naive computation, that can be quite expensive on big repo.
588 # very naive computation, that can be quite expensive on big repo.
589 # However: evolution is currently slow on them anyway.
589 # However: evolution is currently slow on them anyway.
590 nodes = (c.node() for c in repo.set('::%ln', pushop.futureheads))
590 nodes = (c.node() for c in repo.set('::%ln', pushop.futureheads))
591 pushop.outobsmarkers = pushop.repo.obsstore.relevantmarkers(nodes)
591 pushop.outobsmarkers = pushop.repo.obsstore.relevantmarkers(nodes)
592
592
593 @pushdiscovery('bookmarks')
593 @pushdiscovery('bookmarks')
594 def _pushdiscoverybookmarks(pushop):
594 def _pushdiscoverybookmarks(pushop):
595 ui = pushop.ui
595 ui = pushop.ui
596 repo = pushop.repo.unfiltered()
596 repo = pushop.repo.unfiltered()
597 remote = pushop.remote
597 remote = pushop.remote
598 ui.debug("checking for updated bookmarks\n")
598 ui.debug("checking for updated bookmarks\n")
599 ancestors = ()
599 ancestors = ()
600 if pushop.revs:
600 if pushop.revs:
601 revnums = map(repo.changelog.rev, pushop.revs)
601 revnums = map(repo.changelog.rev, pushop.revs)
602 ancestors = repo.changelog.ancestors(revnums, inclusive=True)
602 ancestors = repo.changelog.ancestors(revnums, inclusive=True)
603 remotebookmark = remote.listkeys('bookmarks')
603 remotebookmark = remote.listkeys('bookmarks')
604
604
605 explicit = set([repo._bookmarks.expandname(bookmark)
605 explicit = set([repo._bookmarks.expandname(bookmark)
606 for bookmark in pushop.bookmarks])
606 for bookmark in pushop.bookmarks])
607
607
608 remotebookmark = bookmod.unhexlifybookmarks(remotebookmark)
608 remotebookmark = bookmod.unhexlifybookmarks(remotebookmark)
609 comp = bookmod.comparebookmarks(repo, repo._bookmarks, remotebookmark)
609 comp = bookmod.comparebookmarks(repo, repo._bookmarks, remotebookmark)
610
610
611 def safehex(x):
611 def safehex(x):
612 if x is None:
612 if x is None:
613 return x
613 return x
614 return hex(x)
614 return hex(x)
615
615
616 def hexifycompbookmarks(bookmarks):
616 def hexifycompbookmarks(bookmarks):
617 for b, scid, dcid in bookmarks:
617 for b, scid, dcid in bookmarks:
618 yield b, safehex(scid), safehex(dcid)
618 yield b, safehex(scid), safehex(dcid)
619
619
620 comp = [hexifycompbookmarks(marks) for marks in comp]
620 comp = [hexifycompbookmarks(marks) for marks in comp]
621 addsrc, adddst, advsrc, advdst, diverge, differ, invalid, same = comp
621 addsrc, adddst, advsrc, advdst, diverge, differ, invalid, same = comp
622
622
623 for b, scid, dcid in advsrc:
623 for b, scid, dcid in advsrc:
624 if b in explicit:
624 if b in explicit:
625 explicit.remove(b)
625 explicit.remove(b)
626 if not ancestors or repo[scid].rev() in ancestors:
626 if not ancestors or repo[scid].rev() in ancestors:
627 pushop.outbookmarks.append((b, dcid, scid))
627 pushop.outbookmarks.append((b, dcid, scid))
628 # search added bookmark
628 # search added bookmark
629 for b, scid, dcid in addsrc:
629 for b, scid, dcid in addsrc:
630 if b in explicit:
630 if b in explicit:
631 explicit.remove(b)
631 explicit.remove(b)
632 pushop.outbookmarks.append((b, '', scid))
632 pushop.outbookmarks.append((b, '', scid))
633 # search for overwritten bookmark
633 # search for overwritten bookmark
634 for b, scid, dcid in list(advdst) + list(diverge) + list(differ):
634 for b, scid, dcid in list(advdst) + list(diverge) + list(differ):
635 if b in explicit:
635 if b in explicit:
636 explicit.remove(b)
636 explicit.remove(b)
637 pushop.outbookmarks.append((b, dcid, scid))
637 pushop.outbookmarks.append((b, dcid, scid))
638 # search for bookmark to delete
638 # search for bookmark to delete
639 for b, scid, dcid in adddst:
639 for b, scid, dcid in adddst:
640 if b in explicit:
640 if b in explicit:
641 explicit.remove(b)
641 explicit.remove(b)
642 # treat as "deleted locally"
642 # treat as "deleted locally"
643 pushop.outbookmarks.append((b, dcid, ''))
643 pushop.outbookmarks.append((b, dcid, ''))
644 # identical bookmarks shouldn't get reported
644 # identical bookmarks shouldn't get reported
645 for b, scid, dcid in same:
645 for b, scid, dcid in same:
646 if b in explicit:
646 if b in explicit:
647 explicit.remove(b)
647 explicit.remove(b)
648
648
649 if explicit:
649 if explicit:
650 explicit = sorted(explicit)
650 explicit = sorted(explicit)
651 # we should probably list all of them
651 # we should probably list all of them
652 ui.warn(_('bookmark %s does not exist on the local '
652 ui.warn(_('bookmark %s does not exist on the local '
653 'or remote repository!\n') % explicit[0])
653 'or remote repository!\n') % explicit[0])
654 pushop.bkresult = 2
654 pushop.bkresult = 2
655
655
656 pushop.outbookmarks.sort()
656 pushop.outbookmarks.sort()
657
657
658 def _pushcheckoutgoing(pushop):
658 def _pushcheckoutgoing(pushop):
659 outgoing = pushop.outgoing
659 outgoing = pushop.outgoing
660 unfi = pushop.repo.unfiltered()
660 unfi = pushop.repo.unfiltered()
661 if not outgoing.missing:
661 if not outgoing.missing:
662 # nothing to push
662 # nothing to push
663 scmutil.nochangesfound(unfi.ui, unfi, outgoing.excluded)
663 scmutil.nochangesfound(unfi.ui, unfi, outgoing.excluded)
664 return False
664 return False
665 # something to push
665 # something to push
666 if not pushop.force:
666 if not pushop.force:
667 # if repo.obsstore == False --> no obsolete
667 # if repo.obsstore == False --> no obsolete
668 # then, save the iteration
668 # then, save the iteration
669 if unfi.obsstore:
669 if unfi.obsstore:
670 # this message are here for 80 char limit reason
670 # this message are here for 80 char limit reason
671 mso = _("push includes obsolete changeset: %s!")
671 mso = _("push includes obsolete changeset: %s!")
672 mspd = _("push includes phase-divergent changeset: %s!")
672 mspd = _("push includes phase-divergent changeset: %s!")
673 mscd = _("push includes content-divergent changeset: %s!")
673 mscd = _("push includes content-divergent changeset: %s!")
674 mst = {"orphan": _("push includes orphan changeset: %s!"),
674 mst = {"orphan": _("push includes orphan changeset: %s!"),
675 "phase-divergent": mspd,
675 "phase-divergent": mspd,
676 "content-divergent": mscd}
676 "content-divergent": mscd}
677 # If we are to push if there is at least one
677 # If we are to push if there is at least one
678 # obsolete or unstable changeset in missing, at
678 # obsolete or unstable changeset in missing, at
679 # least one of the missinghead will be obsolete or
679 # least one of the missinghead will be obsolete or
680 # unstable. So checking heads only is ok
680 # unstable. So checking heads only is ok
681 for node in outgoing.missingheads:
681 for node in outgoing.missingheads:
682 ctx = unfi[node]
682 ctx = unfi[node]
683 if ctx.obsolete():
683 if ctx.obsolete():
684 raise error.Abort(mso % ctx)
684 raise error.Abort(mso % ctx)
685 elif ctx.isunstable():
685 elif ctx.isunstable():
686 # TODO print more than one instability in the abort
686 # TODO print more than one instability in the abort
687 # message
687 # message
688 raise error.Abort(mst[ctx.instabilities()[0]] % ctx)
688 raise error.Abort(mst[ctx.instabilities()[0]] % ctx)
689
689
690 discovery.checkheads(pushop)
690 discovery.checkheads(pushop)
691 return True
691 return True
692
692
693 # List of names of steps to perform for an outgoing bundle2, order matters.
693 # List of names of steps to perform for an outgoing bundle2, order matters.
694 b2partsgenorder = []
694 b2partsgenorder = []
695
695
696 # Mapping between step name and function
696 # Mapping between step name and function
697 #
697 #
698 # This exists to help extensions wrap steps if necessary
698 # This exists to help extensions wrap steps if necessary
699 b2partsgenmapping = {}
699 b2partsgenmapping = {}
700
700
701 def b2partsgenerator(stepname, idx=None):
701 def b2partsgenerator(stepname, idx=None):
702 """decorator for function generating bundle2 part
702 """decorator for function generating bundle2 part
703
703
704 The function is added to the step -> function mapping and appended to the
704 The function is added to the step -> function mapping and appended to the
705 list of steps. Beware that decorated functions will be added in order
705 list of steps. Beware that decorated functions will be added in order
706 (this may matter).
706 (this may matter).
707
707
708 You can only use this decorator for new steps, if you want to wrap a step
708 You can only use this decorator for new steps, if you want to wrap a step
709 from an extension, attack the b2partsgenmapping dictionary directly."""
709 from an extension, attack the b2partsgenmapping dictionary directly."""
710 def dec(func):
710 def dec(func):
711 assert stepname not in b2partsgenmapping
711 assert stepname not in b2partsgenmapping
712 b2partsgenmapping[stepname] = func
712 b2partsgenmapping[stepname] = func
713 if idx is None:
713 if idx is None:
714 b2partsgenorder.append(stepname)
714 b2partsgenorder.append(stepname)
715 else:
715 else:
716 b2partsgenorder.insert(idx, stepname)
716 b2partsgenorder.insert(idx, stepname)
717 return func
717 return func
718 return dec
718 return dec
719
719
720 def _pushb2ctxcheckheads(pushop, bundler):
720 def _pushb2ctxcheckheads(pushop, bundler):
721 """Generate race condition checking parts
721 """Generate race condition checking parts
722
722
723 Exists as an independent function to aid extensions
723 Exists as an independent function to aid extensions
724 """
724 """
725 # * 'force' do not check for push race,
725 # * 'force' do not check for push race,
726 # * if we don't push anything, there are nothing to check.
726 # * if we don't push anything, there are nothing to check.
727 if not pushop.force and pushop.outgoing.missingheads:
727 if not pushop.force and pushop.outgoing.missingheads:
728 allowunrelated = 'related' in bundler.capabilities.get('checkheads', ())
728 allowunrelated = 'related' in bundler.capabilities.get('checkheads', ())
729 emptyremote = pushop.pushbranchmap is None
729 emptyremote = pushop.pushbranchmap is None
730 if not allowunrelated or emptyremote:
730 if not allowunrelated or emptyremote:
731 bundler.newpart('check:heads', data=iter(pushop.remoteheads))
731 bundler.newpart('check:heads', data=iter(pushop.remoteheads))
732 else:
732 else:
733 affected = set()
733 affected = set()
734 for branch, heads in pushop.pushbranchmap.iteritems():
734 for branch, heads in pushop.pushbranchmap.iteritems():
735 remoteheads, newheads, unsyncedheads, discardedheads = heads
735 remoteheads, newheads, unsyncedheads, discardedheads = heads
736 if remoteheads is not None:
736 if remoteheads is not None:
737 remote = set(remoteheads)
737 remote = set(remoteheads)
738 affected |= set(discardedheads) & remote
738 affected |= set(discardedheads) & remote
739 affected |= remote - set(newheads)
739 affected |= remote - set(newheads)
740 if affected:
740 if affected:
741 data = iter(sorted(affected))
741 data = iter(sorted(affected))
742 bundler.newpart('check:updated-heads', data=data)
742 bundler.newpart('check:updated-heads', data=data)
743
743
744 def _pushing(pushop):
744 def _pushing(pushop):
745 """return True if we are pushing anything"""
745 """return True if we are pushing anything"""
746 return bool(pushop.outgoing.missing
746 return bool(pushop.outgoing.missing
747 or pushop.outdatedphases
747 or pushop.outdatedphases
748 or pushop.outobsmarkers
748 or pushop.outobsmarkers
749 or pushop.outbookmarks)
749 or pushop.outbookmarks)
750
750
751 @b2partsgenerator('check-bookmarks')
751 @b2partsgenerator('check-bookmarks')
752 def _pushb2checkbookmarks(pushop, bundler):
752 def _pushb2checkbookmarks(pushop, bundler):
753 """insert bookmark move checking"""
753 """insert bookmark move checking"""
754 if not _pushing(pushop) or pushop.force:
754 if not _pushing(pushop) or pushop.force:
755 return
755 return
756 b2caps = bundle2.bundle2caps(pushop.remote)
756 b2caps = bundle2.bundle2caps(pushop.remote)
757 hasbookmarkcheck = 'bookmarks' in b2caps
757 hasbookmarkcheck = 'bookmarks' in b2caps
758 if not (pushop.outbookmarks and hasbookmarkcheck):
758 if not (pushop.outbookmarks and hasbookmarkcheck):
759 return
759 return
760 data = []
760 data = []
761 for book, old, new in pushop.outbookmarks:
761 for book, old, new in pushop.outbookmarks:
762 old = bin(old)
762 old = bin(old)
763 data.append((book, old))
763 data.append((book, old))
764 checkdata = bookmod.binaryencode(data)
764 checkdata = bookmod.binaryencode(data)
765 bundler.newpart('check:bookmarks', data=checkdata)
765 bundler.newpart('check:bookmarks', data=checkdata)
766
766
767 @b2partsgenerator('check-phases')
767 @b2partsgenerator('check-phases')
768 def _pushb2checkphases(pushop, bundler):
768 def _pushb2checkphases(pushop, bundler):
769 """insert phase move checking"""
769 """insert phase move checking"""
770 if not _pushing(pushop) or pushop.force:
770 if not _pushing(pushop) or pushop.force:
771 return
771 return
772 b2caps = bundle2.bundle2caps(pushop.remote)
772 b2caps = bundle2.bundle2caps(pushop.remote)
773 hasphaseheads = 'heads' in b2caps.get('phases', ())
773 hasphaseheads = 'heads' in b2caps.get('phases', ())
774 if pushop.remotephases is not None and hasphaseheads:
774 if pushop.remotephases is not None and hasphaseheads:
775 # check that the remote phase has not changed
775 # check that the remote phase has not changed
776 checks = [[] for p in phases.allphases]
776 checks = [[] for p in phases.allphases]
777 checks[phases.public].extend(pushop.remotephases.publicheads)
777 checks[phases.public].extend(pushop.remotephases.publicheads)
778 checks[phases.draft].extend(pushop.remotephases.draftroots)
778 checks[phases.draft].extend(pushop.remotephases.draftroots)
779 if any(checks):
779 if any(checks):
780 for nodes in checks:
780 for nodes in checks:
781 nodes.sort()
781 nodes.sort()
782 checkdata = phases.binaryencode(checks)
782 checkdata = phases.binaryencode(checks)
783 bundler.newpart('check:phases', data=checkdata)
783 bundler.newpart('check:phases', data=checkdata)
784
784
785 @b2partsgenerator('changeset')
785 @b2partsgenerator('changeset')
786 def _pushb2ctx(pushop, bundler):
786 def _pushb2ctx(pushop, bundler):
787 """handle changegroup push through bundle2
787 """handle changegroup push through bundle2
788
788
789 addchangegroup result is stored in the ``pushop.cgresult`` attribute.
789 addchangegroup result is stored in the ``pushop.cgresult`` attribute.
790 """
790 """
791 if 'changesets' in pushop.stepsdone:
791 if 'changesets' in pushop.stepsdone:
792 return
792 return
793 pushop.stepsdone.add('changesets')
793 pushop.stepsdone.add('changesets')
794 # Send known heads to the server for race detection.
794 # Send known heads to the server for race detection.
795 if not _pushcheckoutgoing(pushop):
795 if not _pushcheckoutgoing(pushop):
796 return
796 return
797 pushop.repo.prepushoutgoinghooks(pushop)
797 pushop.repo.prepushoutgoinghooks(pushop)
798
798
799 _pushb2ctxcheckheads(pushop, bundler)
799 _pushb2ctxcheckheads(pushop, bundler)
800
800
801 b2caps = bundle2.bundle2caps(pushop.remote)
801 b2caps = bundle2.bundle2caps(pushop.remote)
802 version = '01'
802 version = '01'
803 cgversions = b2caps.get('changegroup')
803 cgversions = b2caps.get('changegroup')
804 if cgversions: # 3.1 and 3.2 ship with an empty value
804 if cgversions: # 3.1 and 3.2 ship with an empty value
805 cgversions = [v for v in cgversions
805 cgversions = [v for v in cgversions
806 if v in changegroup.supportedoutgoingversions(
806 if v in changegroup.supportedoutgoingversions(
807 pushop.repo)]
807 pushop.repo)]
808 if not cgversions:
808 if not cgversions:
809 raise ValueError(_('no common changegroup version'))
809 raise ValueError(_('no common changegroup version'))
810 version = max(cgversions)
810 version = max(cgversions)
811 cgstream = changegroup.makestream(pushop.repo, pushop.outgoing, version,
811 cgstream = changegroup.makestream(pushop.repo, pushop.outgoing, version,
812 'push')
812 'push')
813 cgpart = bundler.newpart('changegroup', data=cgstream)
813 cgpart = bundler.newpart('changegroup', data=cgstream)
814 if cgversions:
814 if cgversions:
815 cgpart.addparam('version', version)
815 cgpart.addparam('version', version)
816 if 'treemanifest' in pushop.repo.requirements:
816 if 'treemanifest' in pushop.repo.requirements:
817 cgpart.addparam('treemanifest', '1')
817 cgpart.addparam('treemanifest', '1')
818 def handlereply(op):
818 def handlereply(op):
819 """extract addchangegroup returns from server reply"""
819 """extract addchangegroup returns from server reply"""
820 cgreplies = op.records.getreplies(cgpart.id)
820 cgreplies = op.records.getreplies(cgpart.id)
821 assert len(cgreplies['changegroup']) == 1
821 assert len(cgreplies['changegroup']) == 1
822 pushop.cgresult = cgreplies['changegroup'][0]['return']
822 pushop.cgresult = cgreplies['changegroup'][0]['return']
823 return handlereply
823 return handlereply
824
824
825 @b2partsgenerator('phase')
825 @b2partsgenerator('phase')
826 def _pushb2phases(pushop, bundler):
826 def _pushb2phases(pushop, bundler):
827 """handle phase push through bundle2"""
827 """handle phase push through bundle2"""
828 if 'phases' in pushop.stepsdone:
828 if 'phases' in pushop.stepsdone:
829 return
829 return
830 b2caps = bundle2.bundle2caps(pushop.remote)
830 b2caps = bundle2.bundle2caps(pushop.remote)
831 ui = pushop.repo.ui
831 ui = pushop.repo.ui
832
832
833 legacyphase = 'phases' in ui.configlist('devel', 'legacy.exchange')
833 legacyphase = 'phases' in ui.configlist('devel', 'legacy.exchange')
834 haspushkey = 'pushkey' in b2caps
834 haspushkey = 'pushkey' in b2caps
835 hasphaseheads = 'heads' in b2caps.get('phases', ())
835 hasphaseheads = 'heads' in b2caps.get('phases', ())
836
836
837 if hasphaseheads and not legacyphase:
837 if hasphaseheads and not legacyphase:
838 return _pushb2phaseheads(pushop, bundler)
838 return _pushb2phaseheads(pushop, bundler)
839 elif haspushkey:
839 elif haspushkey:
840 return _pushb2phasespushkey(pushop, bundler)
840 return _pushb2phasespushkey(pushop, bundler)
841
841
842 def _pushb2phaseheads(pushop, bundler):
842 def _pushb2phaseheads(pushop, bundler):
843 """push phase information through a bundle2 - binary part"""
843 """push phase information through a bundle2 - binary part"""
844 pushop.stepsdone.add('phases')
844 pushop.stepsdone.add('phases')
845 if pushop.outdatedphases:
845 if pushop.outdatedphases:
846 updates = [[] for p in phases.allphases]
846 updates = [[] for p in phases.allphases]
847 updates[0].extend(h.node() for h in pushop.outdatedphases)
847 updates[0].extend(h.node() for h in pushop.outdatedphases)
848 phasedata = phases.binaryencode(updates)
848 phasedata = phases.binaryencode(updates)
849 bundler.newpart('phase-heads', data=phasedata)
849 bundler.newpart('phase-heads', data=phasedata)
850
850
851 def _pushb2phasespushkey(pushop, bundler):
851 def _pushb2phasespushkey(pushop, bundler):
852 """push phase information through a bundle2 - pushkey part"""
852 """push phase information through a bundle2 - pushkey part"""
853 pushop.stepsdone.add('phases')
853 pushop.stepsdone.add('phases')
854 part2node = []
854 part2node = []
855
855
856 def handlefailure(pushop, exc):
856 def handlefailure(pushop, exc):
857 targetid = int(exc.partid)
857 targetid = int(exc.partid)
858 for partid, node in part2node:
858 for partid, node in part2node:
859 if partid == targetid:
859 if partid == targetid:
860 raise error.Abort(_('updating %s to public failed') % node)
860 raise error.Abort(_('updating %s to public failed') % node)
861
861
862 enc = pushkey.encode
862 enc = pushkey.encode
863 for newremotehead in pushop.outdatedphases:
863 for newremotehead in pushop.outdatedphases:
864 part = bundler.newpart('pushkey')
864 part = bundler.newpart('pushkey')
865 part.addparam('namespace', enc('phases'))
865 part.addparam('namespace', enc('phases'))
866 part.addparam('key', enc(newremotehead.hex()))
866 part.addparam('key', enc(newremotehead.hex()))
867 part.addparam('old', enc('%d' % phases.draft))
867 part.addparam('old', enc('%d' % phases.draft))
868 part.addparam('new', enc('%d' % phases.public))
868 part.addparam('new', enc('%d' % phases.public))
869 part2node.append((part.id, newremotehead))
869 part2node.append((part.id, newremotehead))
870 pushop.pkfailcb[part.id] = handlefailure
870 pushop.pkfailcb[part.id] = handlefailure
871
871
872 def handlereply(op):
872 def handlereply(op):
873 for partid, node in part2node:
873 for partid, node in part2node:
874 partrep = op.records.getreplies(partid)
874 partrep = op.records.getreplies(partid)
875 results = partrep['pushkey']
875 results = partrep['pushkey']
876 assert len(results) <= 1
876 assert len(results) <= 1
877 msg = None
877 msg = None
878 if not results:
878 if not results:
879 msg = _('server ignored update of %s to public!\n') % node
879 msg = _('server ignored update of %s to public!\n') % node
880 elif not int(results[0]['return']):
880 elif not int(results[0]['return']):
881 msg = _('updating %s to public failed!\n') % node
881 msg = _('updating %s to public failed!\n') % node
882 if msg is not None:
882 if msg is not None:
883 pushop.ui.warn(msg)
883 pushop.ui.warn(msg)
884 return handlereply
884 return handlereply
885
885
886 @b2partsgenerator('obsmarkers')
886 @b2partsgenerator('obsmarkers')
887 def _pushb2obsmarkers(pushop, bundler):
887 def _pushb2obsmarkers(pushop, bundler):
888 if 'obsmarkers' in pushop.stepsdone:
888 if 'obsmarkers' in pushop.stepsdone:
889 return
889 return
890 remoteversions = bundle2.obsmarkersversion(bundler.capabilities)
890 remoteversions = bundle2.obsmarkersversion(bundler.capabilities)
891 if obsolete.commonversion(remoteversions) is None:
891 if obsolete.commonversion(remoteversions) is None:
892 return
892 return
893 pushop.stepsdone.add('obsmarkers')
893 pushop.stepsdone.add('obsmarkers')
894 if pushop.outobsmarkers:
894 if pushop.outobsmarkers:
895 markers = sorted(pushop.outobsmarkers)
895 markers = sorted(pushop.outobsmarkers)
896 bundle2.buildobsmarkerspart(bundler, markers)
896 bundle2.buildobsmarkerspart(bundler, markers)
897
897
898 @b2partsgenerator('bookmarks')
898 @b2partsgenerator('bookmarks')
899 def _pushb2bookmarks(pushop, bundler):
899 def _pushb2bookmarks(pushop, bundler):
900 """handle bookmark push through bundle2"""
900 """handle bookmark push through bundle2"""
901 if 'bookmarks' in pushop.stepsdone:
901 if 'bookmarks' in pushop.stepsdone:
902 return
902 return
903 b2caps = bundle2.bundle2caps(pushop.remote)
903 b2caps = bundle2.bundle2caps(pushop.remote)
904
904
905 legacy = pushop.repo.ui.configlist('devel', 'legacy.exchange')
905 legacy = pushop.repo.ui.configlist('devel', 'legacy.exchange')
906 legacybooks = 'bookmarks' in legacy
906 legacybooks = 'bookmarks' in legacy
907
907
908 if not legacybooks and 'bookmarks' in b2caps:
908 if not legacybooks and 'bookmarks' in b2caps:
909 return _pushb2bookmarkspart(pushop, bundler)
909 return _pushb2bookmarkspart(pushop, bundler)
910 elif 'pushkey' in b2caps:
910 elif 'pushkey' in b2caps:
911 return _pushb2bookmarkspushkey(pushop, bundler)
911 return _pushb2bookmarkspushkey(pushop, bundler)
912
912
913 def _bmaction(old, new):
913 def _bmaction(old, new):
914 """small utility for bookmark pushing"""
914 """small utility for bookmark pushing"""
915 if not old:
915 if not old:
916 return 'export'
916 return 'export'
917 elif not new:
917 elif not new:
918 return 'delete'
918 return 'delete'
919 return 'update'
919 return 'update'
920
920
921 def _pushb2bookmarkspart(pushop, bundler):
921 def _pushb2bookmarkspart(pushop, bundler):
922 pushop.stepsdone.add('bookmarks')
922 pushop.stepsdone.add('bookmarks')
923 if not pushop.outbookmarks:
923 if not pushop.outbookmarks:
924 return
924 return
925
925
926 allactions = []
926 allactions = []
927 data = []
927 data = []
928 for book, old, new in pushop.outbookmarks:
928 for book, old, new in pushop.outbookmarks:
929 new = bin(new)
929 new = bin(new)
930 data.append((book, new))
930 data.append((book, new))
931 allactions.append((book, _bmaction(old, new)))
931 allactions.append((book, _bmaction(old, new)))
932 checkdata = bookmod.binaryencode(data)
932 checkdata = bookmod.binaryencode(data)
933 bundler.newpart('bookmarks', data=checkdata)
933 bundler.newpart('bookmarks', data=checkdata)
934
934
935 def handlereply(op):
935 def handlereply(op):
936 ui = pushop.ui
936 ui = pushop.ui
937 # if success
937 # if success
938 for book, action in allactions:
938 for book, action in allactions:
939 ui.status(bookmsgmap[action][0] % book)
939 ui.status(bookmsgmap[action][0] % book)
940
940
941 return handlereply
941 return handlereply
942
942
943 def _pushb2bookmarkspushkey(pushop, bundler):
943 def _pushb2bookmarkspushkey(pushop, bundler):
944 pushop.stepsdone.add('bookmarks')
944 pushop.stepsdone.add('bookmarks')
945 part2book = []
945 part2book = []
946 enc = pushkey.encode
946 enc = pushkey.encode
947
947
948 def handlefailure(pushop, exc):
948 def handlefailure(pushop, exc):
949 targetid = int(exc.partid)
949 targetid = int(exc.partid)
950 for partid, book, action in part2book:
950 for partid, book, action in part2book:
951 if partid == targetid:
951 if partid == targetid:
952 raise error.Abort(bookmsgmap[action][1].rstrip() % book)
952 raise error.Abort(bookmsgmap[action][1].rstrip() % book)
953 # we should not be called for part we did not generated
953 # we should not be called for part we did not generated
954 assert False
954 assert False
955
955
956 for book, old, new in pushop.outbookmarks:
956 for book, old, new in pushop.outbookmarks:
957 part = bundler.newpart('pushkey')
957 part = bundler.newpart('pushkey')
958 part.addparam('namespace', enc('bookmarks'))
958 part.addparam('namespace', enc('bookmarks'))
959 part.addparam('key', enc(book))
959 part.addparam('key', enc(book))
960 part.addparam('old', enc(old))
960 part.addparam('old', enc(old))
961 part.addparam('new', enc(new))
961 part.addparam('new', enc(new))
962 action = 'update'
962 action = 'update'
963 if not old:
963 if not old:
964 action = 'export'
964 action = 'export'
965 elif not new:
965 elif not new:
966 action = 'delete'
966 action = 'delete'
967 part2book.append((part.id, book, action))
967 part2book.append((part.id, book, action))
968 pushop.pkfailcb[part.id] = handlefailure
968 pushop.pkfailcb[part.id] = handlefailure
969
969
970 def handlereply(op):
970 def handlereply(op):
971 ui = pushop.ui
971 ui = pushop.ui
972 for partid, book, action in part2book:
972 for partid, book, action in part2book:
973 partrep = op.records.getreplies(partid)
973 partrep = op.records.getreplies(partid)
974 results = partrep['pushkey']
974 results = partrep['pushkey']
975 assert len(results) <= 1
975 assert len(results) <= 1
976 if not results:
976 if not results:
977 pushop.ui.warn(_('server ignored bookmark %s update\n') % book)
977 pushop.ui.warn(_('server ignored bookmark %s update\n') % book)
978 else:
978 else:
979 ret = int(results[0]['return'])
979 ret = int(results[0]['return'])
980 if ret:
980 if ret:
981 ui.status(bookmsgmap[action][0] % book)
981 ui.status(bookmsgmap[action][0] % book)
982 else:
982 else:
983 ui.warn(bookmsgmap[action][1] % book)
983 ui.warn(bookmsgmap[action][1] % book)
984 if pushop.bkresult is not None:
984 if pushop.bkresult is not None:
985 pushop.bkresult = 1
985 pushop.bkresult = 1
986 return handlereply
986 return handlereply
987
987
988 @b2partsgenerator('pushvars', idx=0)
988 @b2partsgenerator('pushvars', idx=0)
989 def _getbundlesendvars(pushop, bundler):
989 def _getbundlesendvars(pushop, bundler):
990 '''send shellvars via bundle2'''
990 '''send shellvars via bundle2'''
991 pushvars = pushop.pushvars
991 pushvars = pushop.pushvars
992 if pushvars:
992 if pushvars:
993 shellvars = {}
993 shellvars = {}
994 for raw in pushvars:
994 for raw in pushvars:
995 if '=' not in raw:
995 if '=' not in raw:
996 msg = ("unable to parse variable '%s', should follow "
996 msg = ("unable to parse variable '%s', should follow "
997 "'KEY=VALUE' or 'KEY=' format")
997 "'KEY=VALUE' or 'KEY=' format")
998 raise error.Abort(msg % raw)
998 raise error.Abort(msg % raw)
999 k, v = raw.split('=', 1)
999 k, v = raw.split('=', 1)
1000 shellvars[k] = v
1000 shellvars[k] = v
1001
1001
1002 part = bundler.newpart('pushvars')
1002 part = bundler.newpart('pushvars')
1003
1003
1004 for key, value in shellvars.iteritems():
1004 for key, value in shellvars.iteritems():
1005 part.addparam(key, value, mandatory=False)
1005 part.addparam(key, value, mandatory=False)
1006
1006
1007 def _pushbundle2(pushop):
1007 def _pushbundle2(pushop):
1008 """push data to the remote using bundle2
1008 """push data to the remote using bundle2
1009
1009
1010 The only currently supported type of data is changegroup but this will
1010 The only currently supported type of data is changegroup but this will
1011 evolve in the future."""
1011 evolve in the future."""
1012 bundler = bundle2.bundle20(pushop.ui, bundle2.bundle2caps(pushop.remote))
1012 bundler = bundle2.bundle20(pushop.ui, bundle2.bundle2caps(pushop.remote))
1013 pushback = (pushop.trmanager
1013 pushback = (pushop.trmanager
1014 and pushop.ui.configbool('experimental', 'bundle2.pushback'))
1014 and pushop.ui.configbool('experimental', 'bundle2.pushback'))
1015
1015
1016 # create reply capability
1016 # create reply capability
1017 capsblob = bundle2.encodecaps(bundle2.getrepocaps(pushop.repo,
1017 capsblob = bundle2.encodecaps(bundle2.getrepocaps(pushop.repo,
1018 allowpushback=pushback))
1018 allowpushback=pushback))
1019 bundler.newpart('replycaps', data=capsblob)
1019 bundler.newpart('replycaps', data=capsblob)
1020 replyhandlers = []
1020 replyhandlers = []
1021 for partgenname in b2partsgenorder:
1021 for partgenname in b2partsgenorder:
1022 partgen = b2partsgenmapping[partgenname]
1022 partgen = b2partsgenmapping[partgenname]
1023 ret = partgen(pushop, bundler)
1023 ret = partgen(pushop, bundler)
1024 if callable(ret):
1024 if callable(ret):
1025 replyhandlers.append(ret)
1025 replyhandlers.append(ret)
1026 # do not push if nothing to push
1026 # do not push if nothing to push
1027 if bundler.nbparts <= 1:
1027 if bundler.nbparts <= 1:
1028 return
1028 return
1029 stream = util.chunkbuffer(bundler.getchunks())
1029 stream = util.chunkbuffer(bundler.getchunks())
1030 try:
1030 try:
1031 try:
1031 try:
1032 reply = pushop.remote.unbundle(
1032 reply = pushop.remote.unbundle(
1033 stream, ['force'], pushop.remote.url())
1033 stream, ['force'], pushop.remote.url())
1034 except error.BundleValueError as exc:
1034 except error.BundleValueError as exc:
1035 raise error.Abort(_('missing support for %s') % exc)
1035 raise error.Abort(_('missing support for %s') % exc)
1036 try:
1036 try:
1037 trgetter = None
1037 trgetter = None
1038 if pushback:
1038 if pushback:
1039 trgetter = pushop.trmanager.transaction
1039 trgetter = pushop.trmanager.transaction
1040 op = bundle2.processbundle(pushop.repo, reply, trgetter)
1040 op = bundle2.processbundle(pushop.repo, reply, trgetter)
1041 except error.BundleValueError as exc:
1041 except error.BundleValueError as exc:
1042 raise error.Abort(_('missing support for %s') % exc)
1042 raise error.Abort(_('missing support for %s') % exc)
1043 except bundle2.AbortFromPart as exc:
1043 except bundle2.AbortFromPart as exc:
1044 pushop.ui.status(_('remote: %s\n') % exc)
1044 pushop.ui.status(_('remote: %s\n') % exc)
1045 if exc.hint is not None:
1045 if exc.hint is not None:
1046 pushop.ui.status(_('remote: %s\n') % ('(%s)' % exc.hint))
1046 pushop.ui.status(_('remote: %s\n') % ('(%s)' % exc.hint))
1047 raise error.Abort(_('push failed on remote'))
1047 raise error.Abort(_('push failed on remote'))
1048 except error.PushkeyFailed as exc:
1048 except error.PushkeyFailed as exc:
1049 partid = int(exc.partid)
1049 partid = int(exc.partid)
1050 if partid not in pushop.pkfailcb:
1050 if partid not in pushop.pkfailcb:
1051 raise
1051 raise
1052 pushop.pkfailcb[partid](pushop, exc)
1052 pushop.pkfailcb[partid](pushop, exc)
1053 for rephand in replyhandlers:
1053 for rephand in replyhandlers:
1054 rephand(op)
1054 rephand(op)
1055
1055
1056 def _pushchangeset(pushop):
1056 def _pushchangeset(pushop):
1057 """Make the actual push of changeset bundle to remote repo"""
1057 """Make the actual push of changeset bundle to remote repo"""
1058 if 'changesets' in pushop.stepsdone:
1058 if 'changesets' in pushop.stepsdone:
1059 return
1059 return
1060 pushop.stepsdone.add('changesets')
1060 pushop.stepsdone.add('changesets')
1061 if not _pushcheckoutgoing(pushop):
1061 if not _pushcheckoutgoing(pushop):
1062 return
1062 return
1063
1063
1064 # Should have verified this in push().
1064 # Should have verified this in push().
1065 assert pushop.remote.capable('unbundle')
1065 assert pushop.remote.capable('unbundle')
1066
1066
1067 pushop.repo.prepushoutgoinghooks(pushop)
1067 pushop.repo.prepushoutgoinghooks(pushop)
1068 outgoing = pushop.outgoing
1068 outgoing = pushop.outgoing
1069 # TODO: get bundlecaps from remote
1069 # TODO: get bundlecaps from remote
1070 bundlecaps = None
1070 bundlecaps = None
1071 # create a changegroup from local
1071 # create a changegroup from local
1072 if pushop.revs is None and not (outgoing.excluded
1072 if pushop.revs is None and not (outgoing.excluded
1073 or pushop.repo.changelog.filteredrevs):
1073 or pushop.repo.changelog.filteredrevs):
1074 # push everything,
1074 # push everything,
1075 # use the fast path, no race possible on push
1075 # use the fast path, no race possible on push
1076 cg = changegroup.makechangegroup(pushop.repo, outgoing, '01', 'push',
1076 cg = changegroup.makechangegroup(pushop.repo, outgoing, '01', 'push',
1077 fastpath=True, bundlecaps=bundlecaps)
1077 fastpath=True, bundlecaps=bundlecaps)
1078 else:
1078 else:
1079 cg = changegroup.makechangegroup(pushop.repo, outgoing, '01',
1079 cg = changegroup.makechangegroup(pushop.repo, outgoing, '01',
1080 'push', bundlecaps=bundlecaps)
1080 'push', bundlecaps=bundlecaps)
1081
1081
1082 # apply changegroup to remote
1082 # apply changegroup to remote
1083 # local repo finds heads on server, finds out what
1083 # local repo finds heads on server, finds out what
1084 # revs it must push. once revs transferred, if server
1084 # revs it must push. once revs transferred, if server
1085 # finds it has different heads (someone else won
1085 # finds it has different heads (someone else won
1086 # commit/push race), server aborts.
1086 # commit/push race), server aborts.
1087 if pushop.force:
1087 if pushop.force:
1088 remoteheads = ['force']
1088 remoteheads = ['force']
1089 else:
1089 else:
1090 remoteheads = pushop.remoteheads
1090 remoteheads = pushop.remoteheads
1091 # ssh: return remote's addchangegroup()
1091 # ssh: return remote's addchangegroup()
1092 # http: return remote's addchangegroup() or 0 for error
1092 # http: return remote's addchangegroup() or 0 for error
1093 pushop.cgresult = pushop.remote.unbundle(cg, remoteheads,
1093 pushop.cgresult = pushop.remote.unbundle(cg, remoteheads,
1094 pushop.repo.url())
1094 pushop.repo.url())
1095
1095
1096 def _pushsyncphase(pushop):
1096 def _pushsyncphase(pushop):
1097 """synchronise phase information locally and remotely"""
1097 """synchronise phase information locally and remotely"""
1098 cheads = pushop.commonheads
1098 cheads = pushop.commonheads
1099 # even when we don't push, exchanging phase data is useful
1099 # even when we don't push, exchanging phase data is useful
1100 remotephases = pushop.remote.listkeys('phases')
1100 remotephases = pushop.remote.listkeys('phases')
1101 if (pushop.ui.configbool('ui', '_usedassubrepo')
1101 if (pushop.ui.configbool('ui', '_usedassubrepo')
1102 and remotephases # server supports phases
1102 and remotephases # server supports phases
1103 and pushop.cgresult is None # nothing was pushed
1103 and pushop.cgresult is None # nothing was pushed
1104 and remotephases.get('publishing', False)):
1104 and remotephases.get('publishing', False)):
1105 # When:
1105 # When:
1106 # - this is a subrepo push
1106 # - this is a subrepo push
1107 # - and remote support phase
1107 # - and remote support phase
1108 # - and no changeset was pushed
1108 # - and no changeset was pushed
1109 # - and remote is publishing
1109 # - and remote is publishing
1110 # We may be in issue 3871 case!
1110 # We may be in issue 3871 case!
1111 # We drop the possible phase synchronisation done by
1111 # We drop the possible phase synchronisation done by
1112 # courtesy to publish changesets possibly locally draft
1112 # courtesy to publish changesets possibly locally draft
1113 # on the remote.
1113 # on the remote.
1114 remotephases = {'publishing': 'True'}
1114 remotephases = {'publishing': 'True'}
1115 if not remotephases: # old server or public only reply from non-publishing
1115 if not remotephases: # old server or public only reply from non-publishing
1116 _localphasemove(pushop, cheads)
1116 _localphasemove(pushop, cheads)
1117 # don't push any phase data as there is nothing to push
1117 # don't push any phase data as there is nothing to push
1118 else:
1118 else:
1119 ana = phases.analyzeremotephases(pushop.repo, cheads,
1119 ana = phases.analyzeremotephases(pushop.repo, cheads,
1120 remotephases)
1120 remotephases)
1121 pheads, droots = ana
1121 pheads, droots = ana
1122 ### Apply remote phase on local
1122 ### Apply remote phase on local
1123 if remotephases.get('publishing', False):
1123 if remotephases.get('publishing', False):
1124 _localphasemove(pushop, cheads)
1124 _localphasemove(pushop, cheads)
1125 else: # publish = False
1125 else: # publish = False
1126 _localphasemove(pushop, pheads)
1126 _localphasemove(pushop, pheads)
1127 _localphasemove(pushop, cheads, phases.draft)
1127 _localphasemove(pushop, cheads, phases.draft)
1128 ### Apply local phase on remote
1128 ### Apply local phase on remote
1129
1129
1130 if pushop.cgresult:
1130 if pushop.cgresult:
1131 if 'phases' in pushop.stepsdone:
1131 if 'phases' in pushop.stepsdone:
1132 # phases already pushed though bundle2
1132 # phases already pushed though bundle2
1133 return
1133 return
1134 outdated = pushop.outdatedphases
1134 outdated = pushop.outdatedphases
1135 else:
1135 else:
1136 outdated = pushop.fallbackoutdatedphases
1136 outdated = pushop.fallbackoutdatedphases
1137
1137
1138 pushop.stepsdone.add('phases')
1138 pushop.stepsdone.add('phases')
1139
1139
1140 # filter heads already turned public by the push
1140 # filter heads already turned public by the push
1141 outdated = [c for c in outdated if c.node() not in pheads]
1141 outdated = [c for c in outdated if c.node() not in pheads]
1142 # fallback to independent pushkey command
1142 # fallback to independent pushkey command
1143 for newremotehead in outdated:
1143 for newremotehead in outdated:
1144 r = pushop.remote.pushkey('phases',
1144 r = pushop.remote.pushkey('phases',
1145 newremotehead.hex(),
1145 newremotehead.hex(),
1146 str(phases.draft),
1146 str(phases.draft),
1147 str(phases.public))
1147 str(phases.public))
1148 if not r:
1148 if not r:
1149 pushop.ui.warn(_('updating %s to public failed!\n')
1149 pushop.ui.warn(_('updating %s to public failed!\n')
1150 % newremotehead)
1150 % newremotehead)
1151
1151
1152 def _localphasemove(pushop, nodes, phase=phases.public):
1152 def _localphasemove(pushop, nodes, phase=phases.public):
1153 """move <nodes> to <phase> in the local source repo"""
1153 """move <nodes> to <phase> in the local source repo"""
1154 if pushop.trmanager:
1154 if pushop.trmanager:
1155 phases.advanceboundary(pushop.repo,
1155 phases.advanceboundary(pushop.repo,
1156 pushop.trmanager.transaction(),
1156 pushop.trmanager.transaction(),
1157 phase,
1157 phase,
1158 nodes)
1158 nodes)
1159 else:
1159 else:
1160 # repo is not locked, do not change any phases!
1160 # repo is not locked, do not change any phases!
1161 # Informs the user that phases should have been moved when
1161 # Informs the user that phases should have been moved when
1162 # applicable.
1162 # applicable.
1163 actualmoves = [n for n in nodes if phase < pushop.repo[n].phase()]
1163 actualmoves = [n for n in nodes if phase < pushop.repo[n].phase()]
1164 phasestr = phases.phasenames[phase]
1164 phasestr = phases.phasenames[phase]
1165 if actualmoves:
1165 if actualmoves:
1166 pushop.ui.status(_('cannot lock source repo, skipping '
1166 pushop.ui.status(_('cannot lock source repo, skipping '
1167 'local %s phase update\n') % phasestr)
1167 'local %s phase update\n') % phasestr)
1168
1168
1169 def _pushobsolete(pushop):
1169 def _pushobsolete(pushop):
1170 """utility function to push obsolete markers to a remote"""
1170 """utility function to push obsolete markers to a remote"""
1171 if 'obsmarkers' in pushop.stepsdone:
1171 if 'obsmarkers' in pushop.stepsdone:
1172 return
1172 return
1173 repo = pushop.repo
1173 repo = pushop.repo
1174 remote = pushop.remote
1174 remote = pushop.remote
1175 pushop.stepsdone.add('obsmarkers')
1175 pushop.stepsdone.add('obsmarkers')
1176 if pushop.outobsmarkers:
1176 if pushop.outobsmarkers:
1177 pushop.ui.debug('try to push obsolete markers to remote\n')
1177 pushop.ui.debug('try to push obsolete markers to remote\n')
1178 rslts = []
1178 rslts = []
1179 remotedata = obsolete._pushkeyescape(sorted(pushop.outobsmarkers))
1179 remotedata = obsolete._pushkeyescape(sorted(pushop.outobsmarkers))
1180 for key in sorted(remotedata, reverse=True):
1180 for key in sorted(remotedata, reverse=True):
1181 # reverse sort to ensure we end with dump0
1181 # reverse sort to ensure we end with dump0
1182 data = remotedata[key]
1182 data = remotedata[key]
1183 rslts.append(remote.pushkey('obsolete', key, '', data))
1183 rslts.append(remote.pushkey('obsolete', key, '', data))
1184 if [r for r in rslts if not r]:
1184 if [r for r in rslts if not r]:
1185 msg = _('failed to push some obsolete markers!\n')
1185 msg = _('failed to push some obsolete markers!\n')
1186 repo.ui.warn(msg)
1186 repo.ui.warn(msg)
1187
1187
1188 def _pushbookmark(pushop):
1188 def _pushbookmark(pushop):
1189 """Update bookmark position on remote"""
1189 """Update bookmark position on remote"""
1190 if pushop.cgresult == 0 or 'bookmarks' in pushop.stepsdone:
1190 if pushop.cgresult == 0 or 'bookmarks' in pushop.stepsdone:
1191 return
1191 return
1192 pushop.stepsdone.add('bookmarks')
1192 pushop.stepsdone.add('bookmarks')
1193 ui = pushop.ui
1193 ui = pushop.ui
1194 remote = pushop.remote
1194 remote = pushop.remote
1195
1195
1196 for b, old, new in pushop.outbookmarks:
1196 for b, old, new in pushop.outbookmarks:
1197 action = 'update'
1197 action = 'update'
1198 if not old:
1198 if not old:
1199 action = 'export'
1199 action = 'export'
1200 elif not new:
1200 elif not new:
1201 action = 'delete'
1201 action = 'delete'
1202 if remote.pushkey('bookmarks', b, old, new):
1202 if remote.pushkey('bookmarks', b, old, new):
1203 ui.status(bookmsgmap[action][0] % b)
1203 ui.status(bookmsgmap[action][0] % b)
1204 else:
1204 else:
1205 ui.warn(bookmsgmap[action][1] % b)
1205 ui.warn(bookmsgmap[action][1] % b)
1206 # discovery can have set the value form invalid entry
1206 # discovery can have set the value form invalid entry
1207 if pushop.bkresult is not None:
1207 if pushop.bkresult is not None:
1208 pushop.bkresult = 1
1208 pushop.bkresult = 1
1209
1209
1210 class pulloperation(object):
1210 class pulloperation(object):
1211 """A object that represent a single pull operation
1211 """A object that represent a single pull operation
1212
1212
1213 It purpose is to carry pull related state and very common operation.
1213 It purpose is to carry pull related state and very common operation.
1214
1214
1215 A new should be created at the beginning of each pull and discarded
1215 A new should be created at the beginning of each pull and discarded
1216 afterward.
1216 afterward.
1217 """
1217 """
1218
1218
1219 def __init__(self, repo, remote, heads=None, force=False, bookmarks=(),
1219 def __init__(self, repo, remote, heads=None, force=False, bookmarks=(),
1220 remotebookmarks=None, streamclonerequested=None):
1220 remotebookmarks=None, streamclonerequested=None):
1221 # repo we pull into
1221 # repo we pull into
1222 self.repo = repo
1222 self.repo = repo
1223 # repo we pull from
1223 # repo we pull from
1224 self.remote = remote
1224 self.remote = remote
1225 # revision we try to pull (None is "all")
1225 # revision we try to pull (None is "all")
1226 self.heads = heads
1226 self.heads = heads
1227 # bookmark pulled explicitly
1227 # bookmark pulled explicitly
1228 self.explicitbookmarks = [repo._bookmarks.expandname(bookmark)
1228 self.explicitbookmarks = [repo._bookmarks.expandname(bookmark)
1229 for bookmark in bookmarks]
1229 for bookmark in bookmarks]
1230 # do we force pull?
1230 # do we force pull?
1231 self.force = force
1231 self.force = force
1232 # whether a streaming clone was requested
1232 # whether a streaming clone was requested
1233 self.streamclonerequested = streamclonerequested
1233 self.streamclonerequested = streamclonerequested
1234 # transaction manager
1234 # transaction manager
1235 self.trmanager = None
1235 self.trmanager = None
1236 # set of common changeset between local and remote before pull
1236 # set of common changeset between local and remote before pull
1237 self.common = None
1237 self.common = None
1238 # set of pulled head
1238 # set of pulled head
1239 self.rheads = None
1239 self.rheads = None
1240 # list of missing changeset to fetch remotely
1240 # list of missing changeset to fetch remotely
1241 self.fetch = None
1241 self.fetch = None
1242 # remote bookmarks data
1242 # remote bookmarks data
1243 self.remotebookmarks = remotebookmarks
1243 self.remotebookmarks = remotebookmarks
1244 # result of changegroup pulling (used as return code by pull)
1244 # result of changegroup pulling (used as return code by pull)
1245 self.cgresult = None
1245 self.cgresult = None
1246 # list of step already done
1246 # list of step already done
1247 self.stepsdone = set()
1247 self.stepsdone = set()
1248 # Whether we attempted a clone from pre-generated bundles.
1248 # Whether we attempted a clone from pre-generated bundles.
1249 self.clonebundleattempted = False
1249 self.clonebundleattempted = False
1250
1250
1251 @util.propertycache
1251 @util.propertycache
1252 def pulledsubset(self):
1252 def pulledsubset(self):
1253 """heads of the set of changeset target by the pull"""
1253 """heads of the set of changeset target by the pull"""
1254 # compute target subset
1254 # compute target subset
1255 if self.heads is None:
1255 if self.heads is None:
1256 # We pulled every thing possible
1256 # We pulled every thing possible
1257 # sync on everything common
1257 # sync on everything common
1258 c = set(self.common)
1258 c = set(self.common)
1259 ret = list(self.common)
1259 ret = list(self.common)
1260 for n in self.rheads:
1260 for n in self.rheads:
1261 if n not in c:
1261 if n not in c:
1262 ret.append(n)
1262 ret.append(n)
1263 return ret
1263 return ret
1264 else:
1264 else:
1265 # We pulled a specific subset
1265 # We pulled a specific subset
1266 # sync on this subset
1266 # sync on this subset
1267 return self.heads
1267 return self.heads
1268
1268
1269 @util.propertycache
1269 @util.propertycache
1270 def canusebundle2(self):
1270 def canusebundle2(self):
1271 return not _forcebundle1(self)
1271 return not _forcebundle1(self)
1272
1272
1273 @util.propertycache
1273 @util.propertycache
1274 def remotebundle2caps(self):
1274 def remotebundle2caps(self):
1275 return bundle2.bundle2caps(self.remote)
1275 return bundle2.bundle2caps(self.remote)
1276
1276
1277 def gettransaction(self):
1277 def gettransaction(self):
1278 # deprecated; talk to trmanager directly
1278 # deprecated; talk to trmanager directly
1279 return self.trmanager.transaction()
1279 return self.trmanager.transaction()
1280
1280
1281 class transactionmanager(util.transactional):
1281 class transactionmanager(util.transactional):
1282 """An object to manage the life cycle of a transaction
1282 """An object to manage the life cycle of a transaction
1283
1283
1284 It creates the transaction on demand and calls the appropriate hooks when
1284 It creates the transaction on demand and calls the appropriate hooks when
1285 closing the transaction."""
1285 closing the transaction."""
1286 def __init__(self, repo, source, url):
1286 def __init__(self, repo, source, url):
1287 self.repo = repo
1287 self.repo = repo
1288 self.source = source
1288 self.source = source
1289 self.url = url
1289 self.url = url
1290 self._tr = None
1290 self._tr = None
1291
1291
1292 def transaction(self):
1292 def transaction(self):
1293 """Return an open transaction object, constructing if necessary"""
1293 """Return an open transaction object, constructing if necessary"""
1294 if not self._tr:
1294 if not self._tr:
1295 trname = '%s\n%s' % (self.source, util.hidepassword(self.url))
1295 trname = '%s\n%s' % (self.source, util.hidepassword(self.url))
1296 self._tr = self.repo.transaction(trname)
1296 self._tr = self.repo.transaction(trname)
1297 self._tr.hookargs['source'] = self.source
1297 self._tr.hookargs['source'] = self.source
1298 self._tr.hookargs['url'] = self.url
1298 self._tr.hookargs['url'] = self.url
1299 return self._tr
1299 return self._tr
1300
1300
1301 def close(self):
1301 def close(self):
1302 """close transaction if created"""
1302 """close transaction if created"""
1303 if self._tr is not None:
1303 if self._tr is not None:
1304 self._tr.close()
1304 self._tr.close()
1305
1305
1306 def release(self):
1306 def release(self):
1307 """release transaction if created"""
1307 """release transaction if created"""
1308 if self._tr is not None:
1308 if self._tr is not None:
1309 self._tr.release()
1309 self._tr.release()
1310
1310
1311 def pull(repo, remote, heads=None, force=False, bookmarks=(), opargs=None,
1311 def pull(repo, remote, heads=None, force=False, bookmarks=(), opargs=None,
1312 streamclonerequested=None):
1312 streamclonerequested=None):
1313 """Fetch repository data from a remote.
1313 """Fetch repository data from a remote.
1314
1314
1315 This is the main function used to retrieve data from a remote repository.
1315 This is the main function used to retrieve data from a remote repository.
1316
1316
1317 ``repo`` is the local repository to clone into.
1317 ``repo`` is the local repository to clone into.
1318 ``remote`` is a peer instance.
1318 ``remote`` is a peer instance.
1319 ``heads`` is an iterable of revisions we want to pull. ``None`` (the
1319 ``heads`` is an iterable of revisions we want to pull. ``None`` (the
1320 default) means to pull everything from the remote.
1320 default) means to pull everything from the remote.
1321 ``bookmarks`` is an iterable of bookmarks requesting to be pulled. By
1321 ``bookmarks`` is an iterable of bookmarks requesting to be pulled. By
1322 default, all remote bookmarks are pulled.
1322 default, all remote bookmarks are pulled.
1323 ``opargs`` are additional keyword arguments to pass to ``pulloperation``
1323 ``opargs`` are additional keyword arguments to pass to ``pulloperation``
1324 initialization.
1324 initialization.
1325 ``streamclonerequested`` is a boolean indicating whether a "streaming
1325 ``streamclonerequested`` is a boolean indicating whether a "streaming
1326 clone" is requested. A "streaming clone" is essentially a raw file copy
1326 clone" is requested. A "streaming clone" is essentially a raw file copy
1327 of revlogs from the server. This only works when the local repository is
1327 of revlogs from the server. This only works when the local repository is
1328 empty. The default value of ``None`` means to respect the server
1328 empty. The default value of ``None`` means to respect the server
1329 configuration for preferring stream clones.
1329 configuration for preferring stream clones.
1330
1330
1331 Returns the ``pulloperation`` created for this pull.
1331 Returns the ``pulloperation`` created for this pull.
1332 """
1332 """
1333 if opargs is None:
1333 if opargs is None:
1334 opargs = {}
1334 opargs = {}
1335 pullop = pulloperation(repo, remote, heads, force, bookmarks=bookmarks,
1335 pullop = pulloperation(repo, remote, heads, force, bookmarks=bookmarks,
1336 streamclonerequested=streamclonerequested, **opargs)
1336 streamclonerequested=streamclonerequested,
1337 **pycompat.strkwargs(opargs))
1337
1338
1338 peerlocal = pullop.remote.local()
1339 peerlocal = pullop.remote.local()
1339 if peerlocal:
1340 if peerlocal:
1340 missing = set(peerlocal.requirements) - pullop.repo.supported
1341 missing = set(peerlocal.requirements) - pullop.repo.supported
1341 if missing:
1342 if missing:
1342 msg = _("required features are not"
1343 msg = _("required features are not"
1343 " supported in the destination:"
1344 " supported in the destination:"
1344 " %s") % (', '.join(sorted(missing)))
1345 " %s") % (', '.join(sorted(missing)))
1345 raise error.Abort(msg)
1346 raise error.Abort(msg)
1346
1347
1347 wlock = lock = None
1348 wlock = lock = None
1348 try:
1349 try:
1349 wlock = pullop.repo.wlock()
1350 wlock = pullop.repo.wlock()
1350 lock = pullop.repo.lock()
1351 lock = pullop.repo.lock()
1351 pullop.trmanager = transactionmanager(repo, 'pull', remote.url())
1352 pullop.trmanager = transactionmanager(repo, 'pull', remote.url())
1352 # This should ideally be in _pullbundle2(). However, it needs to run
1353 # This should ideally be in _pullbundle2(). However, it needs to run
1353 # before discovery to avoid extra work.
1354 # before discovery to avoid extra work.
1354 _maybeapplyclonebundle(pullop)
1355 _maybeapplyclonebundle(pullop)
1355 streamclone.maybeperformlegacystreamclone(pullop)
1356 streamclone.maybeperformlegacystreamclone(pullop)
1356 _pulldiscovery(pullop)
1357 _pulldiscovery(pullop)
1357 if pullop.canusebundle2:
1358 if pullop.canusebundle2:
1358 _pullbundle2(pullop)
1359 _pullbundle2(pullop)
1359 _pullchangeset(pullop)
1360 _pullchangeset(pullop)
1360 _pullphase(pullop)
1361 _pullphase(pullop)
1361 _pullbookmarks(pullop)
1362 _pullbookmarks(pullop)
1362 _pullobsolete(pullop)
1363 _pullobsolete(pullop)
1363 pullop.trmanager.close()
1364 pullop.trmanager.close()
1364 finally:
1365 finally:
1365 lockmod.release(pullop.trmanager, lock, wlock)
1366 lockmod.release(pullop.trmanager, lock, wlock)
1366
1367
1367 # storing remotenames
1368 # storing remotenames
1368 if repo.ui.configbool('experimental', 'remotenames'):
1369 if repo.ui.configbool('experimental', 'remotenames'):
1369 logexchange.pullremotenames(repo, remote)
1370 logexchange.pullremotenames(repo, remote)
1370
1371
1371 return pullop
1372 return pullop
1372
1373
1373 # list of steps to perform discovery before pull
1374 # list of steps to perform discovery before pull
1374 pulldiscoveryorder = []
1375 pulldiscoveryorder = []
1375
1376
1376 # Mapping between step name and function
1377 # Mapping between step name and function
1377 #
1378 #
1378 # This exists to help extensions wrap steps if necessary
1379 # This exists to help extensions wrap steps if necessary
1379 pulldiscoverymapping = {}
1380 pulldiscoverymapping = {}
1380
1381
1381 def pulldiscovery(stepname):
1382 def pulldiscovery(stepname):
1382 """decorator for function performing discovery before pull
1383 """decorator for function performing discovery before pull
1383
1384
1384 The function is added to the step -> function mapping and appended to the
1385 The function is added to the step -> function mapping and appended to the
1385 list of steps. Beware that decorated function will be added in order (this
1386 list of steps. Beware that decorated function will be added in order (this
1386 may matter).
1387 may matter).
1387
1388
1388 You can only use this decorator for a new step, if you want to wrap a step
1389 You can only use this decorator for a new step, if you want to wrap a step
1389 from an extension, change the pulldiscovery dictionary directly."""
1390 from an extension, change the pulldiscovery dictionary directly."""
1390 def dec(func):
1391 def dec(func):
1391 assert stepname not in pulldiscoverymapping
1392 assert stepname not in pulldiscoverymapping
1392 pulldiscoverymapping[stepname] = func
1393 pulldiscoverymapping[stepname] = func
1393 pulldiscoveryorder.append(stepname)
1394 pulldiscoveryorder.append(stepname)
1394 return func
1395 return func
1395 return dec
1396 return dec
1396
1397
1397 def _pulldiscovery(pullop):
1398 def _pulldiscovery(pullop):
1398 """Run all discovery steps"""
1399 """Run all discovery steps"""
1399 for stepname in pulldiscoveryorder:
1400 for stepname in pulldiscoveryorder:
1400 step = pulldiscoverymapping[stepname]
1401 step = pulldiscoverymapping[stepname]
1401 step(pullop)
1402 step(pullop)
1402
1403
1403 @pulldiscovery('b1:bookmarks')
1404 @pulldiscovery('b1:bookmarks')
1404 def _pullbookmarkbundle1(pullop):
1405 def _pullbookmarkbundle1(pullop):
1405 """fetch bookmark data in bundle1 case
1406 """fetch bookmark data in bundle1 case
1406
1407
1407 If not using bundle2, we have to fetch bookmarks before changeset
1408 If not using bundle2, we have to fetch bookmarks before changeset
1408 discovery to reduce the chance and impact of race conditions."""
1409 discovery to reduce the chance and impact of race conditions."""
1409 if pullop.remotebookmarks is not None:
1410 if pullop.remotebookmarks is not None:
1410 return
1411 return
1411 if pullop.canusebundle2 and 'listkeys' in pullop.remotebundle2caps:
1412 if pullop.canusebundle2 and 'listkeys' in pullop.remotebundle2caps:
1412 # all known bundle2 servers now support listkeys, but lets be nice with
1413 # all known bundle2 servers now support listkeys, but lets be nice with
1413 # new implementation.
1414 # new implementation.
1414 return
1415 return
1415 books = pullop.remote.listkeys('bookmarks')
1416 books = pullop.remote.listkeys('bookmarks')
1416 pullop.remotebookmarks = bookmod.unhexlifybookmarks(books)
1417 pullop.remotebookmarks = bookmod.unhexlifybookmarks(books)
1417
1418
1418
1419
1419 @pulldiscovery('changegroup')
1420 @pulldiscovery('changegroup')
1420 def _pulldiscoverychangegroup(pullop):
1421 def _pulldiscoverychangegroup(pullop):
1421 """discovery phase for the pull
1422 """discovery phase for the pull
1422
1423
1423 Current handle changeset discovery only, will change handle all discovery
1424 Current handle changeset discovery only, will change handle all discovery
1424 at some point."""
1425 at some point."""
1425 tmp = discovery.findcommonincoming(pullop.repo,
1426 tmp = discovery.findcommonincoming(pullop.repo,
1426 pullop.remote,
1427 pullop.remote,
1427 heads=pullop.heads,
1428 heads=pullop.heads,
1428 force=pullop.force)
1429 force=pullop.force)
1429 common, fetch, rheads = tmp
1430 common, fetch, rheads = tmp
1430 nm = pullop.repo.unfiltered().changelog.nodemap
1431 nm = pullop.repo.unfiltered().changelog.nodemap
1431 if fetch and rheads:
1432 if fetch and rheads:
1432 # If a remote heads is filtered locally, put in back in common.
1433 # If a remote heads is filtered locally, put in back in common.
1433 #
1434 #
1434 # This is a hackish solution to catch most of "common but locally
1435 # This is a hackish solution to catch most of "common but locally
1435 # hidden situation". We do not performs discovery on unfiltered
1436 # hidden situation". We do not performs discovery on unfiltered
1436 # repository because it end up doing a pathological amount of round
1437 # repository because it end up doing a pathological amount of round
1437 # trip for w huge amount of changeset we do not care about.
1438 # trip for w huge amount of changeset we do not care about.
1438 #
1439 #
1439 # If a set of such "common but filtered" changeset exist on the server
1440 # If a set of such "common but filtered" changeset exist on the server
1440 # but are not including a remote heads, we'll not be able to detect it,
1441 # but are not including a remote heads, we'll not be able to detect it,
1441 scommon = set(common)
1442 scommon = set(common)
1442 for n in rheads:
1443 for n in rheads:
1443 if n in nm:
1444 if n in nm:
1444 if n not in scommon:
1445 if n not in scommon:
1445 common.append(n)
1446 common.append(n)
1446 if set(rheads).issubset(set(common)):
1447 if set(rheads).issubset(set(common)):
1447 fetch = []
1448 fetch = []
1448 pullop.common = common
1449 pullop.common = common
1449 pullop.fetch = fetch
1450 pullop.fetch = fetch
1450 pullop.rheads = rheads
1451 pullop.rheads = rheads
1451
1452
1452 def _pullbundle2(pullop):
1453 def _pullbundle2(pullop):
1453 """pull data using bundle2
1454 """pull data using bundle2
1454
1455
1455 For now, the only supported data are changegroup."""
1456 For now, the only supported data are changegroup."""
1456 kwargs = {'bundlecaps': caps20to10(pullop.repo)}
1457 kwargs = {'bundlecaps': caps20to10(pullop.repo)}
1457
1458
1458 # At the moment we don't do stream clones over bundle2. If that is
1459 # At the moment we don't do stream clones over bundle2. If that is
1459 # implemented then here's where the check for that will go.
1460 # implemented then here's where the check for that will go.
1460 streaming = False
1461 streaming = False
1461
1462
1462 # pulling changegroup
1463 # pulling changegroup
1463 pullop.stepsdone.add('changegroup')
1464 pullop.stepsdone.add('changegroup')
1464
1465
1465 kwargs['common'] = pullop.common
1466 kwargs['common'] = pullop.common
1466 kwargs['heads'] = pullop.heads or pullop.rheads
1467 kwargs['heads'] = pullop.heads or pullop.rheads
1467 kwargs['cg'] = pullop.fetch
1468 kwargs['cg'] = pullop.fetch
1468
1469
1469 ui = pullop.repo.ui
1470 ui = pullop.repo.ui
1470 legacyphase = 'phases' in ui.configlist('devel', 'legacy.exchange')
1471 legacyphase = 'phases' in ui.configlist('devel', 'legacy.exchange')
1471 hasbinaryphase = 'heads' in pullop.remotebundle2caps.get('phases', ())
1472 hasbinaryphase = 'heads' in pullop.remotebundle2caps.get('phases', ())
1472 if (not legacyphase and hasbinaryphase):
1473 if (not legacyphase and hasbinaryphase):
1473 kwargs['phases'] = True
1474 kwargs['phases'] = True
1474 pullop.stepsdone.add('phases')
1475 pullop.stepsdone.add('phases')
1475
1476
1476 bookmarksrequested = False
1477 bookmarksrequested = False
1477 legacybookmark = 'bookmarks' in ui.configlist('devel', 'legacy.exchange')
1478 legacybookmark = 'bookmarks' in ui.configlist('devel', 'legacy.exchange')
1478 hasbinarybook = 'bookmarks' in pullop.remotebundle2caps
1479 hasbinarybook = 'bookmarks' in pullop.remotebundle2caps
1479
1480
1480 if pullop.remotebookmarks is not None:
1481 if pullop.remotebookmarks is not None:
1481 pullop.stepsdone.add('request-bookmarks')
1482 pullop.stepsdone.add('request-bookmarks')
1482
1483
1483 if ('request-bookmarks' not in pullop.stepsdone
1484 if ('request-bookmarks' not in pullop.stepsdone
1484 and pullop.remotebookmarks is None
1485 and pullop.remotebookmarks is None
1485 and not legacybookmark and hasbinarybook):
1486 and not legacybookmark and hasbinarybook):
1486 kwargs['bookmarks'] = True
1487 kwargs['bookmarks'] = True
1487 bookmarksrequested = True
1488 bookmarksrequested = True
1488
1489
1489 if 'listkeys' in pullop.remotebundle2caps:
1490 if 'listkeys' in pullop.remotebundle2caps:
1490 if 'phases' not in pullop.stepsdone:
1491 if 'phases' not in pullop.stepsdone:
1491 kwargs['listkeys'] = ['phases']
1492 kwargs['listkeys'] = ['phases']
1492 if 'request-bookmarks' not in pullop.stepsdone:
1493 if 'request-bookmarks' not in pullop.stepsdone:
1493 # make sure to always includes bookmark data when migrating
1494 # make sure to always includes bookmark data when migrating
1494 # `hg incoming --bundle` to using this function.
1495 # `hg incoming --bundle` to using this function.
1495 pullop.stepsdone.add('request-bookmarks')
1496 pullop.stepsdone.add('request-bookmarks')
1496 kwargs.setdefault('listkeys', []).append('bookmarks')
1497 kwargs.setdefault('listkeys', []).append('bookmarks')
1497
1498
1498 # If this is a full pull / clone and the server supports the clone bundles
1499 # If this is a full pull / clone and the server supports the clone bundles
1499 # feature, tell the server whether we attempted a clone bundle. The
1500 # feature, tell the server whether we attempted a clone bundle. The
1500 # presence of this flag indicates the client supports clone bundles. This
1501 # presence of this flag indicates the client supports clone bundles. This
1501 # will enable the server to treat clients that support clone bundles
1502 # will enable the server to treat clients that support clone bundles
1502 # differently from those that don't.
1503 # differently from those that don't.
1503 if (pullop.remote.capable('clonebundles')
1504 if (pullop.remote.capable('clonebundles')
1504 and pullop.heads is None and list(pullop.common) == [nullid]):
1505 and pullop.heads is None and list(pullop.common) == [nullid]):
1505 kwargs['cbattempted'] = pullop.clonebundleattempted
1506 kwargs['cbattempted'] = pullop.clonebundleattempted
1506
1507
1507 if streaming:
1508 if streaming:
1508 pullop.repo.ui.status(_('streaming all changes\n'))
1509 pullop.repo.ui.status(_('streaming all changes\n'))
1509 elif not pullop.fetch:
1510 elif not pullop.fetch:
1510 pullop.repo.ui.status(_("no changes found\n"))
1511 pullop.repo.ui.status(_("no changes found\n"))
1511 pullop.cgresult = 0
1512 pullop.cgresult = 0
1512 else:
1513 else:
1513 if pullop.heads is None and list(pullop.common) == [nullid]:
1514 if pullop.heads is None and list(pullop.common) == [nullid]:
1514 pullop.repo.ui.status(_("requesting all changes\n"))
1515 pullop.repo.ui.status(_("requesting all changes\n"))
1515 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1516 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1516 remoteversions = bundle2.obsmarkersversion(pullop.remotebundle2caps)
1517 remoteversions = bundle2.obsmarkersversion(pullop.remotebundle2caps)
1517 if obsolete.commonversion(remoteversions) is not None:
1518 if obsolete.commonversion(remoteversions) is not None:
1518 kwargs['obsmarkers'] = True
1519 kwargs['obsmarkers'] = True
1519 pullop.stepsdone.add('obsmarkers')
1520 pullop.stepsdone.add('obsmarkers')
1520 _pullbundle2extraprepare(pullop, kwargs)
1521 _pullbundle2extraprepare(pullop, kwargs)
1521 bundle = pullop.remote.getbundle('pull', **pycompat.strkwargs(kwargs))
1522 bundle = pullop.remote.getbundle('pull', **pycompat.strkwargs(kwargs))
1522 try:
1523 try:
1523 op = bundle2.bundleoperation(pullop.repo, pullop.gettransaction)
1524 op = bundle2.bundleoperation(pullop.repo, pullop.gettransaction)
1524 op.modes['bookmarks'] = 'records'
1525 op.modes['bookmarks'] = 'records'
1525 bundle2.processbundle(pullop.repo, bundle, op=op)
1526 bundle2.processbundle(pullop.repo, bundle, op=op)
1526 except bundle2.AbortFromPart as exc:
1527 except bundle2.AbortFromPart as exc:
1527 pullop.repo.ui.status(_('remote: abort: %s\n') % exc)
1528 pullop.repo.ui.status(_('remote: abort: %s\n') % exc)
1528 raise error.Abort(_('pull failed on remote'), hint=exc.hint)
1529 raise error.Abort(_('pull failed on remote'), hint=exc.hint)
1529 except error.BundleValueError as exc:
1530 except error.BundleValueError as exc:
1530 raise error.Abort(_('missing support for %s') % exc)
1531 raise error.Abort(_('missing support for %s') % exc)
1531
1532
1532 if pullop.fetch:
1533 if pullop.fetch:
1533 pullop.cgresult = bundle2.combinechangegroupresults(op)
1534 pullop.cgresult = bundle2.combinechangegroupresults(op)
1534
1535
1535 # processing phases change
1536 # processing phases change
1536 for namespace, value in op.records['listkeys']:
1537 for namespace, value in op.records['listkeys']:
1537 if namespace == 'phases':
1538 if namespace == 'phases':
1538 _pullapplyphases(pullop, value)
1539 _pullapplyphases(pullop, value)
1539
1540
1540 # processing bookmark update
1541 # processing bookmark update
1541 if bookmarksrequested:
1542 if bookmarksrequested:
1542 books = {}
1543 books = {}
1543 for record in op.records['bookmarks']:
1544 for record in op.records['bookmarks']:
1544 books[record['bookmark']] = record["node"]
1545 books[record['bookmark']] = record["node"]
1545 pullop.remotebookmarks = books
1546 pullop.remotebookmarks = books
1546 else:
1547 else:
1547 for namespace, value in op.records['listkeys']:
1548 for namespace, value in op.records['listkeys']:
1548 if namespace == 'bookmarks':
1549 if namespace == 'bookmarks':
1549 pullop.remotebookmarks = bookmod.unhexlifybookmarks(value)
1550 pullop.remotebookmarks = bookmod.unhexlifybookmarks(value)
1550
1551
1551 # bookmark data were either already there or pulled in the bundle
1552 # bookmark data were either already there or pulled in the bundle
1552 if pullop.remotebookmarks is not None:
1553 if pullop.remotebookmarks is not None:
1553 _pullbookmarks(pullop)
1554 _pullbookmarks(pullop)
1554
1555
1555 def _pullbundle2extraprepare(pullop, kwargs):
1556 def _pullbundle2extraprepare(pullop, kwargs):
1556 """hook function so that extensions can extend the getbundle call"""
1557 """hook function so that extensions can extend the getbundle call"""
1557
1558
1558 def _pullchangeset(pullop):
1559 def _pullchangeset(pullop):
1559 """pull changeset from unbundle into the local repo"""
1560 """pull changeset from unbundle into the local repo"""
1560 # We delay the open of the transaction as late as possible so we
1561 # We delay the open of the transaction as late as possible so we
1561 # don't open transaction for nothing or you break future useful
1562 # don't open transaction for nothing or you break future useful
1562 # rollback call
1563 # rollback call
1563 if 'changegroup' in pullop.stepsdone:
1564 if 'changegroup' in pullop.stepsdone:
1564 return
1565 return
1565 pullop.stepsdone.add('changegroup')
1566 pullop.stepsdone.add('changegroup')
1566 if not pullop.fetch:
1567 if not pullop.fetch:
1567 pullop.repo.ui.status(_("no changes found\n"))
1568 pullop.repo.ui.status(_("no changes found\n"))
1568 pullop.cgresult = 0
1569 pullop.cgresult = 0
1569 return
1570 return
1570 tr = pullop.gettransaction()
1571 tr = pullop.gettransaction()
1571 if pullop.heads is None and list(pullop.common) == [nullid]:
1572 if pullop.heads is None and list(pullop.common) == [nullid]:
1572 pullop.repo.ui.status(_("requesting all changes\n"))
1573 pullop.repo.ui.status(_("requesting all changes\n"))
1573 elif pullop.heads is None and pullop.remote.capable('changegroupsubset'):
1574 elif pullop.heads is None and pullop.remote.capable('changegroupsubset'):
1574 # issue1320, avoid a race if remote changed after discovery
1575 # issue1320, avoid a race if remote changed after discovery
1575 pullop.heads = pullop.rheads
1576 pullop.heads = pullop.rheads
1576
1577
1577 if pullop.remote.capable('getbundle'):
1578 if pullop.remote.capable('getbundle'):
1578 # TODO: get bundlecaps from remote
1579 # TODO: get bundlecaps from remote
1579 cg = pullop.remote.getbundle('pull', common=pullop.common,
1580 cg = pullop.remote.getbundle('pull', common=pullop.common,
1580 heads=pullop.heads or pullop.rheads)
1581 heads=pullop.heads or pullop.rheads)
1581 elif pullop.heads is None:
1582 elif pullop.heads is None:
1582 cg = pullop.remote.changegroup(pullop.fetch, 'pull')
1583 cg = pullop.remote.changegroup(pullop.fetch, 'pull')
1583 elif not pullop.remote.capable('changegroupsubset'):
1584 elif not pullop.remote.capable('changegroupsubset'):
1584 raise error.Abort(_("partial pull cannot be done because "
1585 raise error.Abort(_("partial pull cannot be done because "
1585 "other repository doesn't support "
1586 "other repository doesn't support "
1586 "changegroupsubset."))
1587 "changegroupsubset."))
1587 else:
1588 else:
1588 cg = pullop.remote.changegroupsubset(pullop.fetch, pullop.heads, 'pull')
1589 cg = pullop.remote.changegroupsubset(pullop.fetch, pullop.heads, 'pull')
1589 bundleop = bundle2.applybundle(pullop.repo, cg, tr, 'pull',
1590 bundleop = bundle2.applybundle(pullop.repo, cg, tr, 'pull',
1590 pullop.remote.url())
1591 pullop.remote.url())
1591 pullop.cgresult = bundle2.combinechangegroupresults(bundleop)
1592 pullop.cgresult = bundle2.combinechangegroupresults(bundleop)
1592
1593
1593 def _pullphase(pullop):
1594 def _pullphase(pullop):
1594 # Get remote phases data from remote
1595 # Get remote phases data from remote
1595 if 'phases' in pullop.stepsdone:
1596 if 'phases' in pullop.stepsdone:
1596 return
1597 return
1597 remotephases = pullop.remote.listkeys('phases')
1598 remotephases = pullop.remote.listkeys('phases')
1598 _pullapplyphases(pullop, remotephases)
1599 _pullapplyphases(pullop, remotephases)
1599
1600
1600 def _pullapplyphases(pullop, remotephases):
1601 def _pullapplyphases(pullop, remotephases):
1601 """apply phase movement from observed remote state"""
1602 """apply phase movement from observed remote state"""
1602 if 'phases' in pullop.stepsdone:
1603 if 'phases' in pullop.stepsdone:
1603 return
1604 return
1604 pullop.stepsdone.add('phases')
1605 pullop.stepsdone.add('phases')
1605 publishing = bool(remotephases.get('publishing', False))
1606 publishing = bool(remotephases.get('publishing', False))
1606 if remotephases and not publishing:
1607 if remotephases and not publishing:
1607 # remote is new and non-publishing
1608 # remote is new and non-publishing
1608 pheads, _dr = phases.analyzeremotephases(pullop.repo,
1609 pheads, _dr = phases.analyzeremotephases(pullop.repo,
1609 pullop.pulledsubset,
1610 pullop.pulledsubset,
1610 remotephases)
1611 remotephases)
1611 dheads = pullop.pulledsubset
1612 dheads = pullop.pulledsubset
1612 else:
1613 else:
1613 # Remote is old or publishing all common changesets
1614 # Remote is old or publishing all common changesets
1614 # should be seen as public
1615 # should be seen as public
1615 pheads = pullop.pulledsubset
1616 pheads = pullop.pulledsubset
1616 dheads = []
1617 dheads = []
1617 unfi = pullop.repo.unfiltered()
1618 unfi = pullop.repo.unfiltered()
1618 phase = unfi._phasecache.phase
1619 phase = unfi._phasecache.phase
1619 rev = unfi.changelog.nodemap.get
1620 rev = unfi.changelog.nodemap.get
1620 public = phases.public
1621 public = phases.public
1621 draft = phases.draft
1622 draft = phases.draft
1622
1623
1623 # exclude changesets already public locally and update the others
1624 # exclude changesets already public locally and update the others
1624 pheads = [pn for pn in pheads if phase(unfi, rev(pn)) > public]
1625 pheads = [pn for pn in pheads if phase(unfi, rev(pn)) > public]
1625 if pheads:
1626 if pheads:
1626 tr = pullop.gettransaction()
1627 tr = pullop.gettransaction()
1627 phases.advanceboundary(pullop.repo, tr, public, pheads)
1628 phases.advanceboundary(pullop.repo, tr, public, pheads)
1628
1629
1629 # exclude changesets already draft locally and update the others
1630 # exclude changesets already draft locally and update the others
1630 dheads = [pn for pn in dheads if phase(unfi, rev(pn)) > draft]
1631 dheads = [pn for pn in dheads if phase(unfi, rev(pn)) > draft]
1631 if dheads:
1632 if dheads:
1632 tr = pullop.gettransaction()
1633 tr = pullop.gettransaction()
1633 phases.advanceboundary(pullop.repo, tr, draft, dheads)
1634 phases.advanceboundary(pullop.repo, tr, draft, dheads)
1634
1635
1635 def _pullbookmarks(pullop):
1636 def _pullbookmarks(pullop):
1636 """process the remote bookmark information to update the local one"""
1637 """process the remote bookmark information to update the local one"""
1637 if 'bookmarks' in pullop.stepsdone:
1638 if 'bookmarks' in pullop.stepsdone:
1638 return
1639 return
1639 pullop.stepsdone.add('bookmarks')
1640 pullop.stepsdone.add('bookmarks')
1640 repo = pullop.repo
1641 repo = pullop.repo
1641 remotebookmarks = pullop.remotebookmarks
1642 remotebookmarks = pullop.remotebookmarks
1642 bookmod.updatefromremote(repo.ui, repo, remotebookmarks,
1643 bookmod.updatefromremote(repo.ui, repo, remotebookmarks,
1643 pullop.remote.url(),
1644 pullop.remote.url(),
1644 pullop.gettransaction,
1645 pullop.gettransaction,
1645 explicit=pullop.explicitbookmarks)
1646 explicit=pullop.explicitbookmarks)
1646
1647
1647 def _pullobsolete(pullop):
1648 def _pullobsolete(pullop):
1648 """utility function to pull obsolete markers from a remote
1649 """utility function to pull obsolete markers from a remote
1649
1650
1650 The `gettransaction` is function that return the pull transaction, creating
1651 The `gettransaction` is function that return the pull transaction, creating
1651 one if necessary. We return the transaction to inform the calling code that
1652 one if necessary. We return the transaction to inform the calling code that
1652 a new transaction have been created (when applicable).
1653 a new transaction have been created (when applicable).
1653
1654
1654 Exists mostly to allow overriding for experimentation purpose"""
1655 Exists mostly to allow overriding for experimentation purpose"""
1655 if 'obsmarkers' in pullop.stepsdone:
1656 if 'obsmarkers' in pullop.stepsdone:
1656 return
1657 return
1657 pullop.stepsdone.add('obsmarkers')
1658 pullop.stepsdone.add('obsmarkers')
1658 tr = None
1659 tr = None
1659 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1660 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1660 pullop.repo.ui.debug('fetching remote obsolete markers\n')
1661 pullop.repo.ui.debug('fetching remote obsolete markers\n')
1661 remoteobs = pullop.remote.listkeys('obsolete')
1662 remoteobs = pullop.remote.listkeys('obsolete')
1662 if 'dump0' in remoteobs:
1663 if 'dump0' in remoteobs:
1663 tr = pullop.gettransaction()
1664 tr = pullop.gettransaction()
1664 markers = []
1665 markers = []
1665 for key in sorted(remoteobs, reverse=True):
1666 for key in sorted(remoteobs, reverse=True):
1666 if key.startswith('dump'):
1667 if key.startswith('dump'):
1667 data = util.b85decode(remoteobs[key])
1668 data = util.b85decode(remoteobs[key])
1668 version, newmarks = obsolete._readmarkers(data)
1669 version, newmarks = obsolete._readmarkers(data)
1669 markers += newmarks
1670 markers += newmarks
1670 if markers:
1671 if markers:
1671 pullop.repo.obsstore.add(tr, markers)
1672 pullop.repo.obsstore.add(tr, markers)
1672 pullop.repo.invalidatevolatilesets()
1673 pullop.repo.invalidatevolatilesets()
1673 return tr
1674 return tr
1674
1675
1675 def caps20to10(repo):
1676 def caps20to10(repo):
1676 """return a set with appropriate options to use bundle20 during getbundle"""
1677 """return a set with appropriate options to use bundle20 during getbundle"""
1677 caps = {'HG20'}
1678 caps = {'HG20'}
1678 capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo))
1679 capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo))
1679 caps.add('bundle2=' + urlreq.quote(capsblob))
1680 caps.add('bundle2=' + urlreq.quote(capsblob))
1680 return caps
1681 return caps
1681
1682
1682 # List of names of steps to perform for a bundle2 for getbundle, order matters.
1683 # List of names of steps to perform for a bundle2 for getbundle, order matters.
1683 getbundle2partsorder = []
1684 getbundle2partsorder = []
1684
1685
1685 # Mapping between step name and function
1686 # Mapping between step name and function
1686 #
1687 #
1687 # This exists to help extensions wrap steps if necessary
1688 # This exists to help extensions wrap steps if necessary
1688 getbundle2partsmapping = {}
1689 getbundle2partsmapping = {}
1689
1690
1690 def getbundle2partsgenerator(stepname, idx=None):
1691 def getbundle2partsgenerator(stepname, idx=None):
1691 """decorator for function generating bundle2 part for getbundle
1692 """decorator for function generating bundle2 part for getbundle
1692
1693
1693 The function is added to the step -> function mapping and appended to the
1694 The function is added to the step -> function mapping and appended to the
1694 list of steps. Beware that decorated functions will be added in order
1695 list of steps. Beware that decorated functions will be added in order
1695 (this may matter).
1696 (this may matter).
1696
1697
1697 You can only use this decorator for new steps, if you want to wrap a step
1698 You can only use this decorator for new steps, if you want to wrap a step
1698 from an extension, attack the getbundle2partsmapping dictionary directly."""
1699 from an extension, attack the getbundle2partsmapping dictionary directly."""
1699 def dec(func):
1700 def dec(func):
1700 assert stepname not in getbundle2partsmapping
1701 assert stepname not in getbundle2partsmapping
1701 getbundle2partsmapping[stepname] = func
1702 getbundle2partsmapping[stepname] = func
1702 if idx is None:
1703 if idx is None:
1703 getbundle2partsorder.append(stepname)
1704 getbundle2partsorder.append(stepname)
1704 else:
1705 else:
1705 getbundle2partsorder.insert(idx, stepname)
1706 getbundle2partsorder.insert(idx, stepname)
1706 return func
1707 return func
1707 return dec
1708 return dec
1708
1709
1709 def bundle2requested(bundlecaps):
1710 def bundle2requested(bundlecaps):
1710 if bundlecaps is not None:
1711 if bundlecaps is not None:
1711 return any(cap.startswith('HG2') for cap in bundlecaps)
1712 return any(cap.startswith('HG2') for cap in bundlecaps)
1712 return False
1713 return False
1713
1714
1714 def getbundlechunks(repo, source, heads=None, common=None, bundlecaps=None,
1715 def getbundlechunks(repo, source, heads=None, common=None, bundlecaps=None,
1715 **kwargs):
1716 **kwargs):
1716 """Return chunks constituting a bundle's raw data.
1717 """Return chunks constituting a bundle's raw data.
1717
1718
1718 Could be a bundle HG10 or a bundle HG20 depending on bundlecaps
1719 Could be a bundle HG10 or a bundle HG20 depending on bundlecaps
1719 passed.
1720 passed.
1720
1721
1721 Returns an iterator over raw chunks (of varying sizes).
1722 Returns an iterator over raw chunks (of varying sizes).
1722 """
1723 """
1723 kwargs = pycompat.byteskwargs(kwargs)
1724 kwargs = pycompat.byteskwargs(kwargs)
1724 usebundle2 = bundle2requested(bundlecaps)
1725 usebundle2 = bundle2requested(bundlecaps)
1725 # bundle10 case
1726 # bundle10 case
1726 if not usebundle2:
1727 if not usebundle2:
1727 if bundlecaps and not kwargs.get('cg', True):
1728 if bundlecaps and not kwargs.get('cg', True):
1728 raise ValueError(_('request for bundle10 must include changegroup'))
1729 raise ValueError(_('request for bundle10 must include changegroup'))
1729
1730
1730 if kwargs:
1731 if kwargs:
1731 raise ValueError(_('unsupported getbundle arguments: %s')
1732 raise ValueError(_('unsupported getbundle arguments: %s')
1732 % ', '.join(sorted(kwargs.keys())))
1733 % ', '.join(sorted(kwargs.keys())))
1733 outgoing = _computeoutgoing(repo, heads, common)
1734 outgoing = _computeoutgoing(repo, heads, common)
1734 return changegroup.makestream(repo, outgoing, '01', source,
1735 return changegroup.makestream(repo, outgoing, '01', source,
1735 bundlecaps=bundlecaps)
1736 bundlecaps=bundlecaps)
1736
1737
1737 # bundle20 case
1738 # bundle20 case
1738 b2caps = {}
1739 b2caps = {}
1739 for bcaps in bundlecaps:
1740 for bcaps in bundlecaps:
1740 if bcaps.startswith('bundle2='):
1741 if bcaps.startswith('bundle2='):
1741 blob = urlreq.unquote(bcaps[len('bundle2='):])
1742 blob = urlreq.unquote(bcaps[len('bundle2='):])
1742 b2caps.update(bundle2.decodecaps(blob))
1743 b2caps.update(bundle2.decodecaps(blob))
1743 bundler = bundle2.bundle20(repo.ui, b2caps)
1744 bundler = bundle2.bundle20(repo.ui, b2caps)
1744
1745
1745 kwargs['heads'] = heads
1746 kwargs['heads'] = heads
1746 kwargs['common'] = common
1747 kwargs['common'] = common
1747
1748
1748 for name in getbundle2partsorder:
1749 for name in getbundle2partsorder:
1749 func = getbundle2partsmapping[name]
1750 func = getbundle2partsmapping[name]
1750 func(bundler, repo, source, bundlecaps=bundlecaps, b2caps=b2caps,
1751 func(bundler, repo, source, bundlecaps=bundlecaps, b2caps=b2caps,
1751 **pycompat.strkwargs(kwargs))
1752 **pycompat.strkwargs(kwargs))
1752
1753
1753 return bundler.getchunks()
1754 return bundler.getchunks()
1754
1755
1755 @getbundle2partsgenerator('changegroup')
1756 @getbundle2partsgenerator('changegroup')
1756 def _getbundlechangegrouppart(bundler, repo, source, bundlecaps=None,
1757 def _getbundlechangegrouppart(bundler, repo, source, bundlecaps=None,
1757 b2caps=None, heads=None, common=None, **kwargs):
1758 b2caps=None, heads=None, common=None, **kwargs):
1758 """add a changegroup part to the requested bundle"""
1759 """add a changegroup part to the requested bundle"""
1759 cgstream = None
1760 cgstream = None
1760 if kwargs.get('cg', True):
1761 if kwargs.get(r'cg', True):
1761 # build changegroup bundle here.
1762 # build changegroup bundle here.
1762 version = '01'
1763 version = '01'
1763 cgversions = b2caps.get('changegroup')
1764 cgversions = b2caps.get('changegroup')
1764 if cgversions: # 3.1 and 3.2 ship with an empty value
1765 if cgversions: # 3.1 and 3.2 ship with an empty value
1765 cgversions = [v for v in cgversions
1766 cgversions = [v for v in cgversions
1766 if v in changegroup.supportedoutgoingversions(repo)]
1767 if v in changegroup.supportedoutgoingversions(repo)]
1767 if not cgversions:
1768 if not cgversions:
1768 raise ValueError(_('no common changegroup version'))
1769 raise ValueError(_('no common changegroup version'))
1769 version = max(cgversions)
1770 version = max(cgversions)
1770 outgoing = _computeoutgoing(repo, heads, common)
1771 outgoing = _computeoutgoing(repo, heads, common)
1771 if outgoing.missing:
1772 if outgoing.missing:
1772 cgstream = changegroup.makestream(repo, outgoing, version, source,
1773 cgstream = changegroup.makestream(repo, outgoing, version, source,
1773 bundlecaps=bundlecaps)
1774 bundlecaps=bundlecaps)
1774
1775
1775 if cgstream:
1776 if cgstream:
1776 part = bundler.newpart('changegroup', data=cgstream)
1777 part = bundler.newpart('changegroup', data=cgstream)
1777 if cgversions:
1778 if cgversions:
1778 part.addparam('version', version)
1779 part.addparam('version', version)
1779 part.addparam('nbchanges', '%d' % len(outgoing.missing),
1780 part.addparam('nbchanges', '%d' % len(outgoing.missing),
1780 mandatory=False)
1781 mandatory=False)
1781 if 'treemanifest' in repo.requirements:
1782 if 'treemanifest' in repo.requirements:
1782 part.addparam('treemanifest', '1')
1783 part.addparam('treemanifest', '1')
1783
1784
1784 @getbundle2partsgenerator('bookmarks')
1785 @getbundle2partsgenerator('bookmarks')
1785 def _getbundlebookmarkpart(bundler, repo, source, bundlecaps=None,
1786 def _getbundlebookmarkpart(bundler, repo, source, bundlecaps=None,
1786 b2caps=None, **kwargs):
1787 b2caps=None, **kwargs):
1787 """add a bookmark part to the requested bundle"""
1788 """add a bookmark part to the requested bundle"""
1788 if not kwargs.get('bookmarks', False):
1789 if not kwargs.get(r'bookmarks', False):
1789 return
1790 return
1790 if 'bookmarks' not in b2caps:
1791 if 'bookmarks' not in b2caps:
1791 raise ValueError(_('no common bookmarks exchange method'))
1792 raise ValueError(_('no common bookmarks exchange method'))
1792 books = bookmod.listbinbookmarks(repo)
1793 books = bookmod.listbinbookmarks(repo)
1793 data = bookmod.binaryencode(books)
1794 data = bookmod.binaryencode(books)
1794 if data:
1795 if data:
1795 bundler.newpart('bookmarks', data=data)
1796 bundler.newpart('bookmarks', data=data)
1796
1797
1797 @getbundle2partsgenerator('listkeys')
1798 @getbundle2partsgenerator('listkeys')
1798 def _getbundlelistkeysparts(bundler, repo, source, bundlecaps=None,
1799 def _getbundlelistkeysparts(bundler, repo, source, bundlecaps=None,
1799 b2caps=None, **kwargs):
1800 b2caps=None, **kwargs):
1800 """add parts containing listkeys namespaces to the requested bundle"""
1801 """add parts containing listkeys namespaces to the requested bundle"""
1801 listkeys = kwargs.get('listkeys', ())
1802 listkeys = kwargs.get(r'listkeys', ())
1802 for namespace in listkeys:
1803 for namespace in listkeys:
1803 part = bundler.newpart('listkeys')
1804 part = bundler.newpart('listkeys')
1804 part.addparam('namespace', namespace)
1805 part.addparam('namespace', namespace)
1805 keys = repo.listkeys(namespace).items()
1806 keys = repo.listkeys(namespace).items()
1806 part.data = pushkey.encodekeys(keys)
1807 part.data = pushkey.encodekeys(keys)
1807
1808
1808 @getbundle2partsgenerator('obsmarkers')
1809 @getbundle2partsgenerator('obsmarkers')
1809 def _getbundleobsmarkerpart(bundler, repo, source, bundlecaps=None,
1810 def _getbundleobsmarkerpart(bundler, repo, source, bundlecaps=None,
1810 b2caps=None, heads=None, **kwargs):
1811 b2caps=None, heads=None, **kwargs):
1811 """add an obsolescence markers part to the requested bundle"""
1812 """add an obsolescence markers part to the requested bundle"""
1812 if kwargs.get('obsmarkers', False):
1813 if kwargs.get(r'obsmarkers', False):
1813 if heads is None:
1814 if heads is None:
1814 heads = repo.heads()
1815 heads = repo.heads()
1815 subset = [c.node() for c in repo.set('::%ln', heads)]
1816 subset = [c.node() for c in repo.set('::%ln', heads)]
1816 markers = repo.obsstore.relevantmarkers(subset)
1817 markers = repo.obsstore.relevantmarkers(subset)
1817 markers = sorted(markers)
1818 markers = sorted(markers)
1818 bundle2.buildobsmarkerspart(bundler, markers)
1819 bundle2.buildobsmarkerspart(bundler, markers)
1819
1820
1820 @getbundle2partsgenerator('phases')
1821 @getbundle2partsgenerator('phases')
1821 def _getbundlephasespart(bundler, repo, source, bundlecaps=None,
1822 def _getbundlephasespart(bundler, repo, source, bundlecaps=None,
1822 b2caps=None, heads=None, **kwargs):
1823 b2caps=None, heads=None, **kwargs):
1823 """add phase heads part to the requested bundle"""
1824 """add phase heads part to the requested bundle"""
1824 if kwargs.get('phases', False):
1825 if kwargs.get(r'phases', False):
1825 if not 'heads' in b2caps.get('phases'):
1826 if not 'heads' in b2caps.get('phases'):
1826 raise ValueError(_('no common phases exchange method'))
1827 raise ValueError(_('no common phases exchange method'))
1827 if heads is None:
1828 if heads is None:
1828 heads = repo.heads()
1829 heads = repo.heads()
1829
1830
1830 headsbyphase = collections.defaultdict(set)
1831 headsbyphase = collections.defaultdict(set)
1831 if repo.publishing():
1832 if repo.publishing():
1832 headsbyphase[phases.public] = heads
1833 headsbyphase[phases.public] = heads
1833 else:
1834 else:
1834 # find the appropriate heads to move
1835 # find the appropriate heads to move
1835
1836
1836 phase = repo._phasecache.phase
1837 phase = repo._phasecache.phase
1837 node = repo.changelog.node
1838 node = repo.changelog.node
1838 rev = repo.changelog.rev
1839 rev = repo.changelog.rev
1839 for h in heads:
1840 for h in heads:
1840 headsbyphase[phase(repo, rev(h))].add(h)
1841 headsbyphase[phase(repo, rev(h))].add(h)
1841 seenphases = list(headsbyphase.keys())
1842 seenphases = list(headsbyphase.keys())
1842
1843
1843 # We do not handle anything but public and draft phase for now)
1844 # We do not handle anything but public and draft phase for now)
1844 if seenphases:
1845 if seenphases:
1845 assert max(seenphases) <= phases.draft
1846 assert max(seenphases) <= phases.draft
1846
1847
1847 # if client is pulling non-public changesets, we need to find
1848 # if client is pulling non-public changesets, we need to find
1848 # intermediate public heads.
1849 # intermediate public heads.
1849 draftheads = headsbyphase.get(phases.draft, set())
1850 draftheads = headsbyphase.get(phases.draft, set())
1850 if draftheads:
1851 if draftheads:
1851 publicheads = headsbyphase.get(phases.public, set())
1852 publicheads = headsbyphase.get(phases.public, set())
1852
1853
1853 revset = 'heads(only(%ln, %ln) and public())'
1854 revset = 'heads(only(%ln, %ln) and public())'
1854 extraheads = repo.revs(revset, draftheads, publicheads)
1855 extraheads = repo.revs(revset, draftheads, publicheads)
1855 for r in extraheads:
1856 for r in extraheads:
1856 headsbyphase[phases.public].add(node(r))
1857 headsbyphase[phases.public].add(node(r))
1857
1858
1858 # transform data in a format used by the encoding function
1859 # transform data in a format used by the encoding function
1859 phasemapping = []
1860 phasemapping = []
1860 for phase in phases.allphases:
1861 for phase in phases.allphases:
1861 phasemapping.append(sorted(headsbyphase[phase]))
1862 phasemapping.append(sorted(headsbyphase[phase]))
1862
1863
1863 # generate the actual part
1864 # generate the actual part
1864 phasedata = phases.binaryencode(phasemapping)
1865 phasedata = phases.binaryencode(phasemapping)
1865 bundler.newpart('phase-heads', data=phasedata)
1866 bundler.newpart('phase-heads', data=phasedata)
1866
1867
1867 @getbundle2partsgenerator('hgtagsfnodes')
1868 @getbundle2partsgenerator('hgtagsfnodes')
1868 def _getbundletagsfnodes(bundler, repo, source, bundlecaps=None,
1869 def _getbundletagsfnodes(bundler, repo, source, bundlecaps=None,
1869 b2caps=None, heads=None, common=None,
1870 b2caps=None, heads=None, common=None,
1870 **kwargs):
1871 **kwargs):
1871 """Transfer the .hgtags filenodes mapping.
1872 """Transfer the .hgtags filenodes mapping.
1872
1873
1873 Only values for heads in this bundle will be transferred.
1874 Only values for heads in this bundle will be transferred.
1874
1875
1875 The part data consists of pairs of 20 byte changeset node and .hgtags
1876 The part data consists of pairs of 20 byte changeset node and .hgtags
1876 filenodes raw values.
1877 filenodes raw values.
1877 """
1878 """
1878 # Don't send unless:
1879 # Don't send unless:
1879 # - changeset are being exchanged,
1880 # - changeset are being exchanged,
1880 # - the client supports it.
1881 # - the client supports it.
1881 if not (kwargs.get('cg', True) and 'hgtagsfnodes' in b2caps):
1882 if not (kwargs.get(r'cg', True) and 'hgtagsfnodes' in b2caps):
1882 return
1883 return
1883
1884
1884 outgoing = _computeoutgoing(repo, heads, common)
1885 outgoing = _computeoutgoing(repo, heads, common)
1885 bundle2.addparttagsfnodescache(repo, bundler, outgoing)
1886 bundle2.addparttagsfnodescache(repo, bundler, outgoing)
1886
1887
1887 def check_heads(repo, their_heads, context):
1888 def check_heads(repo, their_heads, context):
1888 """check if the heads of a repo have been modified
1889 """check if the heads of a repo have been modified
1889
1890
1890 Used by peer for unbundling.
1891 Used by peer for unbundling.
1891 """
1892 """
1892 heads = repo.heads()
1893 heads = repo.heads()
1893 heads_hash = hashlib.sha1(''.join(sorted(heads))).digest()
1894 heads_hash = hashlib.sha1(''.join(sorted(heads))).digest()
1894 if not (their_heads == ['force'] or their_heads == heads or
1895 if not (their_heads == ['force'] or their_heads == heads or
1895 their_heads == ['hashed', heads_hash]):
1896 their_heads == ['hashed', heads_hash]):
1896 # someone else committed/pushed/unbundled while we
1897 # someone else committed/pushed/unbundled while we
1897 # were transferring data
1898 # were transferring data
1898 raise error.PushRaced('repository changed while %s - '
1899 raise error.PushRaced('repository changed while %s - '
1899 'please try again' % context)
1900 'please try again' % context)
1900
1901
1901 def unbundle(repo, cg, heads, source, url):
1902 def unbundle(repo, cg, heads, source, url):
1902 """Apply a bundle to a repo.
1903 """Apply a bundle to a repo.
1903
1904
1904 this function makes sure the repo is locked during the application and have
1905 this function makes sure the repo is locked during the application and have
1905 mechanism to check that no push race occurred between the creation of the
1906 mechanism to check that no push race occurred between the creation of the
1906 bundle and its application.
1907 bundle and its application.
1907
1908
1908 If the push was raced as PushRaced exception is raised."""
1909 If the push was raced as PushRaced exception is raised."""
1909 r = 0
1910 r = 0
1910 # need a transaction when processing a bundle2 stream
1911 # need a transaction when processing a bundle2 stream
1911 # [wlock, lock, tr] - needs to be an array so nested functions can modify it
1912 # [wlock, lock, tr] - needs to be an array so nested functions can modify it
1912 lockandtr = [None, None, None]
1913 lockandtr = [None, None, None]
1913 recordout = None
1914 recordout = None
1914 # quick fix for output mismatch with bundle2 in 3.4
1915 # quick fix for output mismatch with bundle2 in 3.4
1915 captureoutput = repo.ui.configbool('experimental', 'bundle2-output-capture')
1916 captureoutput = repo.ui.configbool('experimental', 'bundle2-output-capture')
1916 if url.startswith('remote:http:') or url.startswith('remote:https:'):
1917 if url.startswith('remote:http:') or url.startswith('remote:https:'):
1917 captureoutput = True
1918 captureoutput = True
1918 try:
1919 try:
1919 # note: outside bundle1, 'heads' is expected to be empty and this
1920 # note: outside bundle1, 'heads' is expected to be empty and this
1920 # 'check_heads' call wil be a no-op
1921 # 'check_heads' call wil be a no-op
1921 check_heads(repo, heads, 'uploading changes')
1922 check_heads(repo, heads, 'uploading changes')
1922 # push can proceed
1923 # push can proceed
1923 if not isinstance(cg, bundle2.unbundle20):
1924 if not isinstance(cg, bundle2.unbundle20):
1924 # legacy case: bundle1 (changegroup 01)
1925 # legacy case: bundle1 (changegroup 01)
1925 txnname = "\n".join([source, util.hidepassword(url)])
1926 txnname = "\n".join([source, util.hidepassword(url)])
1926 with repo.lock(), repo.transaction(txnname) as tr:
1927 with repo.lock(), repo.transaction(txnname) as tr:
1927 op = bundle2.applybundle(repo, cg, tr, source, url)
1928 op = bundle2.applybundle(repo, cg, tr, source, url)
1928 r = bundle2.combinechangegroupresults(op)
1929 r = bundle2.combinechangegroupresults(op)
1929 else:
1930 else:
1930 r = None
1931 r = None
1931 try:
1932 try:
1932 def gettransaction():
1933 def gettransaction():
1933 if not lockandtr[2]:
1934 if not lockandtr[2]:
1934 lockandtr[0] = repo.wlock()
1935 lockandtr[0] = repo.wlock()
1935 lockandtr[1] = repo.lock()
1936 lockandtr[1] = repo.lock()
1936 lockandtr[2] = repo.transaction(source)
1937 lockandtr[2] = repo.transaction(source)
1937 lockandtr[2].hookargs['source'] = source
1938 lockandtr[2].hookargs['source'] = source
1938 lockandtr[2].hookargs['url'] = url
1939 lockandtr[2].hookargs['url'] = url
1939 lockandtr[2].hookargs['bundle2'] = '1'
1940 lockandtr[2].hookargs['bundle2'] = '1'
1940 return lockandtr[2]
1941 return lockandtr[2]
1941
1942
1942 # Do greedy locking by default until we're satisfied with lazy
1943 # Do greedy locking by default until we're satisfied with lazy
1943 # locking.
1944 # locking.
1944 if not repo.ui.configbool('experimental', 'bundle2lazylocking'):
1945 if not repo.ui.configbool('experimental', 'bundle2lazylocking'):
1945 gettransaction()
1946 gettransaction()
1946
1947
1947 op = bundle2.bundleoperation(repo, gettransaction,
1948 op = bundle2.bundleoperation(repo, gettransaction,
1948 captureoutput=captureoutput)
1949 captureoutput=captureoutput)
1949 try:
1950 try:
1950 op = bundle2.processbundle(repo, cg, op=op)
1951 op = bundle2.processbundle(repo, cg, op=op)
1951 finally:
1952 finally:
1952 r = op.reply
1953 r = op.reply
1953 if captureoutput and r is not None:
1954 if captureoutput and r is not None:
1954 repo.ui.pushbuffer(error=True, subproc=True)
1955 repo.ui.pushbuffer(error=True, subproc=True)
1955 def recordout(output):
1956 def recordout(output):
1956 r.newpart('output', data=output, mandatory=False)
1957 r.newpart('output', data=output, mandatory=False)
1957 if lockandtr[2] is not None:
1958 if lockandtr[2] is not None:
1958 lockandtr[2].close()
1959 lockandtr[2].close()
1959 except BaseException as exc:
1960 except BaseException as exc:
1960 exc.duringunbundle2 = True
1961 exc.duringunbundle2 = True
1961 if captureoutput and r is not None:
1962 if captureoutput and r is not None:
1962 parts = exc._bundle2salvagedoutput = r.salvageoutput()
1963 parts = exc._bundle2salvagedoutput = r.salvageoutput()
1963 def recordout(output):
1964 def recordout(output):
1964 part = bundle2.bundlepart('output', data=output,
1965 part = bundle2.bundlepart('output', data=output,
1965 mandatory=False)
1966 mandatory=False)
1966 parts.append(part)
1967 parts.append(part)
1967 raise
1968 raise
1968 finally:
1969 finally:
1969 lockmod.release(lockandtr[2], lockandtr[1], lockandtr[0])
1970 lockmod.release(lockandtr[2], lockandtr[1], lockandtr[0])
1970 if recordout is not None:
1971 if recordout is not None:
1971 recordout(repo.ui.popbuffer())
1972 recordout(repo.ui.popbuffer())
1972 return r
1973 return r
1973
1974
1974 def _maybeapplyclonebundle(pullop):
1975 def _maybeapplyclonebundle(pullop):
1975 """Apply a clone bundle from a remote, if possible."""
1976 """Apply a clone bundle from a remote, if possible."""
1976
1977
1977 repo = pullop.repo
1978 repo = pullop.repo
1978 remote = pullop.remote
1979 remote = pullop.remote
1979
1980
1980 if not repo.ui.configbool('ui', 'clonebundles'):
1981 if not repo.ui.configbool('ui', 'clonebundles'):
1981 return
1982 return
1982
1983
1983 # Only run if local repo is empty.
1984 # Only run if local repo is empty.
1984 if len(repo):
1985 if len(repo):
1985 return
1986 return
1986
1987
1987 if pullop.heads:
1988 if pullop.heads:
1988 return
1989 return
1989
1990
1990 if not remote.capable('clonebundles'):
1991 if not remote.capable('clonebundles'):
1991 return
1992 return
1992
1993
1993 res = remote._call('clonebundles')
1994 res = remote._call('clonebundles')
1994
1995
1995 # If we call the wire protocol command, that's good enough to record the
1996 # If we call the wire protocol command, that's good enough to record the
1996 # attempt.
1997 # attempt.
1997 pullop.clonebundleattempted = True
1998 pullop.clonebundleattempted = True
1998
1999
1999 entries = parseclonebundlesmanifest(repo, res)
2000 entries = parseclonebundlesmanifest(repo, res)
2000 if not entries:
2001 if not entries:
2001 repo.ui.note(_('no clone bundles available on remote; '
2002 repo.ui.note(_('no clone bundles available on remote; '
2002 'falling back to regular clone\n'))
2003 'falling back to regular clone\n'))
2003 return
2004 return
2004
2005
2005 entries = filterclonebundleentries(
2006 entries = filterclonebundleentries(
2006 repo, entries, streamclonerequested=pullop.streamclonerequested)
2007 repo, entries, streamclonerequested=pullop.streamclonerequested)
2007
2008
2008 if not entries:
2009 if not entries:
2009 # There is a thundering herd concern here. However, if a server
2010 # There is a thundering herd concern here. However, if a server
2010 # operator doesn't advertise bundles appropriate for its clients,
2011 # operator doesn't advertise bundles appropriate for its clients,
2011 # they deserve what's coming. Furthermore, from a client's
2012 # they deserve what's coming. Furthermore, from a client's
2012 # perspective, no automatic fallback would mean not being able to
2013 # perspective, no automatic fallback would mean not being able to
2013 # clone!
2014 # clone!
2014 repo.ui.warn(_('no compatible clone bundles available on server; '
2015 repo.ui.warn(_('no compatible clone bundles available on server; '
2015 'falling back to regular clone\n'))
2016 'falling back to regular clone\n'))
2016 repo.ui.warn(_('(you may want to report this to the server '
2017 repo.ui.warn(_('(you may want to report this to the server '
2017 'operator)\n'))
2018 'operator)\n'))
2018 return
2019 return
2019
2020
2020 entries = sortclonebundleentries(repo.ui, entries)
2021 entries = sortclonebundleentries(repo.ui, entries)
2021
2022
2022 url = entries[0]['URL']
2023 url = entries[0]['URL']
2023 repo.ui.status(_('applying clone bundle from %s\n') % url)
2024 repo.ui.status(_('applying clone bundle from %s\n') % url)
2024 if trypullbundlefromurl(repo.ui, repo, url):
2025 if trypullbundlefromurl(repo.ui, repo, url):
2025 repo.ui.status(_('finished applying clone bundle\n'))
2026 repo.ui.status(_('finished applying clone bundle\n'))
2026 # Bundle failed.
2027 # Bundle failed.
2027 #
2028 #
2028 # We abort by default to avoid the thundering herd of
2029 # We abort by default to avoid the thundering herd of
2029 # clients flooding a server that was expecting expensive
2030 # clients flooding a server that was expecting expensive
2030 # clone load to be offloaded.
2031 # clone load to be offloaded.
2031 elif repo.ui.configbool('ui', 'clonebundlefallback'):
2032 elif repo.ui.configbool('ui', 'clonebundlefallback'):
2032 repo.ui.warn(_('falling back to normal clone\n'))
2033 repo.ui.warn(_('falling back to normal clone\n'))
2033 else:
2034 else:
2034 raise error.Abort(_('error applying bundle'),
2035 raise error.Abort(_('error applying bundle'),
2035 hint=_('if this error persists, consider contacting '
2036 hint=_('if this error persists, consider contacting '
2036 'the server operator or disable clone '
2037 'the server operator or disable clone '
2037 'bundles via '
2038 'bundles via '
2038 '"--config ui.clonebundles=false"'))
2039 '"--config ui.clonebundles=false"'))
2039
2040
2040 def parseclonebundlesmanifest(repo, s):
2041 def parseclonebundlesmanifest(repo, s):
2041 """Parses the raw text of a clone bundles manifest.
2042 """Parses the raw text of a clone bundles manifest.
2042
2043
2043 Returns a list of dicts. The dicts have a ``URL`` key corresponding
2044 Returns a list of dicts. The dicts have a ``URL`` key corresponding
2044 to the URL and other keys are the attributes for the entry.
2045 to the URL and other keys are the attributes for the entry.
2045 """
2046 """
2046 m = []
2047 m = []
2047 for line in s.splitlines():
2048 for line in s.splitlines():
2048 fields = line.split()
2049 fields = line.split()
2049 if not fields:
2050 if not fields:
2050 continue
2051 continue
2051 attrs = {'URL': fields[0]}
2052 attrs = {'URL': fields[0]}
2052 for rawattr in fields[1:]:
2053 for rawattr in fields[1:]:
2053 key, value = rawattr.split('=', 1)
2054 key, value = rawattr.split('=', 1)
2054 key = urlreq.unquote(key)
2055 key = urlreq.unquote(key)
2055 value = urlreq.unquote(value)
2056 value = urlreq.unquote(value)
2056 attrs[key] = value
2057 attrs[key] = value
2057
2058
2058 # Parse BUNDLESPEC into components. This makes client-side
2059 # Parse BUNDLESPEC into components. This makes client-side
2059 # preferences easier to specify since you can prefer a single
2060 # preferences easier to specify since you can prefer a single
2060 # component of the BUNDLESPEC.
2061 # component of the BUNDLESPEC.
2061 if key == 'BUNDLESPEC':
2062 if key == 'BUNDLESPEC':
2062 try:
2063 try:
2063 comp, version, params = parsebundlespec(repo, value,
2064 comp, version, params = parsebundlespec(repo, value,
2064 externalnames=True)
2065 externalnames=True)
2065 attrs['COMPRESSION'] = comp
2066 attrs['COMPRESSION'] = comp
2066 attrs['VERSION'] = version
2067 attrs['VERSION'] = version
2067 except error.InvalidBundleSpecification:
2068 except error.InvalidBundleSpecification:
2068 pass
2069 pass
2069 except error.UnsupportedBundleSpecification:
2070 except error.UnsupportedBundleSpecification:
2070 pass
2071 pass
2071
2072
2072 m.append(attrs)
2073 m.append(attrs)
2073
2074
2074 return m
2075 return m
2075
2076
2076 def filterclonebundleentries(repo, entries, streamclonerequested=False):
2077 def filterclonebundleentries(repo, entries, streamclonerequested=False):
2077 """Remove incompatible clone bundle manifest entries.
2078 """Remove incompatible clone bundle manifest entries.
2078
2079
2079 Accepts a list of entries parsed with ``parseclonebundlesmanifest``
2080 Accepts a list of entries parsed with ``parseclonebundlesmanifest``
2080 and returns a new list consisting of only the entries that this client
2081 and returns a new list consisting of only the entries that this client
2081 should be able to apply.
2082 should be able to apply.
2082
2083
2083 There is no guarantee we'll be able to apply all returned entries because
2084 There is no guarantee we'll be able to apply all returned entries because
2084 the metadata we use to filter on may be missing or wrong.
2085 the metadata we use to filter on may be missing or wrong.
2085 """
2086 """
2086 newentries = []
2087 newentries = []
2087 for entry in entries:
2088 for entry in entries:
2088 spec = entry.get('BUNDLESPEC')
2089 spec = entry.get('BUNDLESPEC')
2089 if spec:
2090 if spec:
2090 try:
2091 try:
2091 comp, version, params = parsebundlespec(repo, spec, strict=True)
2092 comp, version, params = parsebundlespec(repo, spec, strict=True)
2092
2093
2093 # If a stream clone was requested, filter out non-streamclone
2094 # If a stream clone was requested, filter out non-streamclone
2094 # entries.
2095 # entries.
2095 if streamclonerequested and (comp != 'UN' or version != 's1'):
2096 if streamclonerequested and (comp != 'UN' or version != 's1'):
2096 repo.ui.debug('filtering %s because not a stream clone\n' %
2097 repo.ui.debug('filtering %s because not a stream clone\n' %
2097 entry['URL'])
2098 entry['URL'])
2098 continue
2099 continue
2099
2100
2100 except error.InvalidBundleSpecification as e:
2101 except error.InvalidBundleSpecification as e:
2101 repo.ui.debug(str(e) + '\n')
2102 repo.ui.debug(str(e) + '\n')
2102 continue
2103 continue
2103 except error.UnsupportedBundleSpecification as e:
2104 except error.UnsupportedBundleSpecification as e:
2104 repo.ui.debug('filtering %s because unsupported bundle '
2105 repo.ui.debug('filtering %s because unsupported bundle '
2105 'spec: %s\n' % (entry['URL'], str(e)))
2106 'spec: %s\n' % (entry['URL'], str(e)))
2106 continue
2107 continue
2107 # If we don't have a spec and requested a stream clone, we don't know
2108 # If we don't have a spec and requested a stream clone, we don't know
2108 # what the entry is so don't attempt to apply it.
2109 # what the entry is so don't attempt to apply it.
2109 elif streamclonerequested:
2110 elif streamclonerequested:
2110 repo.ui.debug('filtering %s because cannot determine if a stream '
2111 repo.ui.debug('filtering %s because cannot determine if a stream '
2111 'clone bundle\n' % entry['URL'])
2112 'clone bundle\n' % entry['URL'])
2112 continue
2113 continue
2113
2114
2114 if 'REQUIRESNI' in entry and not sslutil.hassni:
2115 if 'REQUIRESNI' in entry and not sslutil.hassni:
2115 repo.ui.debug('filtering %s because SNI not supported\n' %
2116 repo.ui.debug('filtering %s because SNI not supported\n' %
2116 entry['URL'])
2117 entry['URL'])
2117 continue
2118 continue
2118
2119
2119 newentries.append(entry)
2120 newentries.append(entry)
2120
2121
2121 return newentries
2122 return newentries
2122
2123
2123 class clonebundleentry(object):
2124 class clonebundleentry(object):
2124 """Represents an item in a clone bundles manifest.
2125 """Represents an item in a clone bundles manifest.
2125
2126
2126 This rich class is needed to support sorting since sorted() in Python 3
2127 This rich class is needed to support sorting since sorted() in Python 3
2127 doesn't support ``cmp`` and our comparison is complex enough that ``key=``
2128 doesn't support ``cmp`` and our comparison is complex enough that ``key=``
2128 won't work.
2129 won't work.
2129 """
2130 """
2130
2131
2131 def __init__(self, value, prefers):
2132 def __init__(self, value, prefers):
2132 self.value = value
2133 self.value = value
2133 self.prefers = prefers
2134 self.prefers = prefers
2134
2135
2135 def _cmp(self, other):
2136 def _cmp(self, other):
2136 for prefkey, prefvalue in self.prefers:
2137 for prefkey, prefvalue in self.prefers:
2137 avalue = self.value.get(prefkey)
2138 avalue = self.value.get(prefkey)
2138 bvalue = other.value.get(prefkey)
2139 bvalue = other.value.get(prefkey)
2139
2140
2140 # Special case for b missing attribute and a matches exactly.
2141 # Special case for b missing attribute and a matches exactly.
2141 if avalue is not None and bvalue is None and avalue == prefvalue:
2142 if avalue is not None and bvalue is None and avalue == prefvalue:
2142 return -1
2143 return -1
2143
2144
2144 # Special case for a missing attribute and b matches exactly.
2145 # Special case for a missing attribute and b matches exactly.
2145 if bvalue is not None and avalue is None and bvalue == prefvalue:
2146 if bvalue is not None and avalue is None and bvalue == prefvalue:
2146 return 1
2147 return 1
2147
2148
2148 # We can't compare unless attribute present on both.
2149 # We can't compare unless attribute present on both.
2149 if avalue is None or bvalue is None:
2150 if avalue is None or bvalue is None:
2150 continue
2151 continue
2151
2152
2152 # Same values should fall back to next attribute.
2153 # Same values should fall back to next attribute.
2153 if avalue == bvalue:
2154 if avalue == bvalue:
2154 continue
2155 continue
2155
2156
2156 # Exact matches come first.
2157 # Exact matches come first.
2157 if avalue == prefvalue:
2158 if avalue == prefvalue:
2158 return -1
2159 return -1
2159 if bvalue == prefvalue:
2160 if bvalue == prefvalue:
2160 return 1
2161 return 1
2161
2162
2162 # Fall back to next attribute.
2163 # Fall back to next attribute.
2163 continue
2164 continue
2164
2165
2165 # If we got here we couldn't sort by attributes and prefers. Fall
2166 # If we got here we couldn't sort by attributes and prefers. Fall
2166 # back to index order.
2167 # back to index order.
2167 return 0
2168 return 0
2168
2169
2169 def __lt__(self, other):
2170 def __lt__(self, other):
2170 return self._cmp(other) < 0
2171 return self._cmp(other) < 0
2171
2172
2172 def __gt__(self, other):
2173 def __gt__(self, other):
2173 return self._cmp(other) > 0
2174 return self._cmp(other) > 0
2174
2175
2175 def __eq__(self, other):
2176 def __eq__(self, other):
2176 return self._cmp(other) == 0
2177 return self._cmp(other) == 0
2177
2178
2178 def __le__(self, other):
2179 def __le__(self, other):
2179 return self._cmp(other) <= 0
2180 return self._cmp(other) <= 0
2180
2181
2181 def __ge__(self, other):
2182 def __ge__(self, other):
2182 return self._cmp(other) >= 0
2183 return self._cmp(other) >= 0
2183
2184
2184 def __ne__(self, other):
2185 def __ne__(self, other):
2185 return self._cmp(other) != 0
2186 return self._cmp(other) != 0
2186
2187
2187 def sortclonebundleentries(ui, entries):
2188 def sortclonebundleentries(ui, entries):
2188 prefers = ui.configlist('ui', 'clonebundleprefers')
2189 prefers = ui.configlist('ui', 'clonebundleprefers')
2189 if not prefers:
2190 if not prefers:
2190 return list(entries)
2191 return list(entries)
2191
2192
2192 prefers = [p.split('=', 1) for p in prefers]
2193 prefers = [p.split('=', 1) for p in prefers]
2193
2194
2194 items = sorted(clonebundleentry(v, prefers) for v in entries)
2195 items = sorted(clonebundleentry(v, prefers) for v in entries)
2195 return [i.value for i in items]
2196 return [i.value for i in items]
2196
2197
2197 def trypullbundlefromurl(ui, repo, url):
2198 def trypullbundlefromurl(ui, repo, url):
2198 """Attempt to apply a bundle from a URL."""
2199 """Attempt to apply a bundle from a URL."""
2199 with repo.lock(), repo.transaction('bundleurl') as tr:
2200 with repo.lock(), repo.transaction('bundleurl') as tr:
2200 try:
2201 try:
2201 fh = urlmod.open(ui, url)
2202 fh = urlmod.open(ui, url)
2202 cg = readbundle(ui, fh, 'stream')
2203 cg = readbundle(ui, fh, 'stream')
2203
2204
2204 if isinstance(cg, streamclone.streamcloneapplier):
2205 if isinstance(cg, streamclone.streamcloneapplier):
2205 cg.apply(repo)
2206 cg.apply(repo)
2206 else:
2207 else:
2207 bundle2.applybundle(repo, cg, tr, 'clonebundles', url)
2208 bundle2.applybundle(repo, cg, tr, 'clonebundles', url)
2208 return True
2209 return True
2209 except urlerr.httperror as e:
2210 except urlerr.httperror as e:
2210 ui.warn(_('HTTP error fetching bundle: %s\n') % str(e))
2211 ui.warn(_('HTTP error fetching bundle: %s\n') % str(e))
2211 except urlerr.urlerror as e:
2212 except urlerr.urlerror as e:
2212 ui.warn(_('error fetching bundle: %s\n') % e.reason)
2213 ui.warn(_('error fetching bundle: %s\n') % e.reason)
2213
2214
2214 return False
2215 return False
General Comments 0
You need to be logged in to leave comments. Login now