##// END OF EJS Templates
pull: preindent some code...
Boris Feld -
r35780:40df727b default
parent child Browse files
Show More
@@ -1,2228 +1,2229 b''
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,
1336 streamclonerequested=streamclonerequested,
1337 **pycompat.strkwargs(opargs))
1337 **pycompat.strkwargs(opargs))
1338
1338
1339 peerlocal = pullop.remote.local()
1339 peerlocal = pullop.remote.local()
1340 if peerlocal:
1340 if peerlocal:
1341 missing = set(peerlocal.requirements) - pullop.repo.supported
1341 missing = set(peerlocal.requirements) - pullop.repo.supported
1342 if missing:
1342 if missing:
1343 msg = _("required features are not"
1343 msg = _("required features are not"
1344 " supported in the destination:"
1344 " supported in the destination:"
1345 " %s") % (', '.join(sorted(missing)))
1345 " %s") % (', '.join(sorted(missing)))
1346 raise error.Abort(msg)
1346 raise error.Abort(msg)
1347
1347
1348 pullop.trmanager = transactionmanager(repo, 'pull', remote.url())
1348 pullop.trmanager = transactionmanager(repo, 'pull', remote.url())
1349 with repo.wlock(), repo.lock(), pullop.trmanager:
1349 with repo.wlock(), repo.lock(), pullop.trmanager:
1350 # This should ideally be in _pullbundle2(). However, it needs to run
1350 # This should ideally be in _pullbundle2(). However, it needs to run
1351 # before discovery to avoid extra work.
1351 # before discovery to avoid extra work.
1352 _maybeapplyclonebundle(pullop)
1352 _maybeapplyclonebundle(pullop)
1353 streamclone.maybeperformlegacystreamclone(pullop)
1353 streamclone.maybeperformlegacystreamclone(pullop)
1354 _pulldiscovery(pullop)
1354 _pulldiscovery(pullop)
1355 if pullop.canusebundle2:
1355 if pullop.canusebundle2:
1356 _pullbundle2(pullop)
1356 _pullbundle2(pullop)
1357 _pullchangeset(pullop)
1357 _pullchangeset(pullop)
1358 _pullphase(pullop)
1358 _pullphase(pullop)
1359 _pullbookmarks(pullop)
1359 _pullbookmarks(pullop)
1360 _pullobsolete(pullop)
1360 _pullobsolete(pullop)
1361
1361
1362 # storing remotenames
1362 # storing remotenames
1363 if repo.ui.configbool('experimental', 'remotenames'):
1363 if repo.ui.configbool('experimental', 'remotenames'):
1364 logexchange.pullremotenames(repo, remote)
1364 logexchange.pullremotenames(repo, remote)
1365
1365
1366 return pullop
1366 return pullop
1367
1367
1368 # list of steps to perform discovery before pull
1368 # list of steps to perform discovery before pull
1369 pulldiscoveryorder = []
1369 pulldiscoveryorder = []
1370
1370
1371 # Mapping between step name and function
1371 # Mapping between step name and function
1372 #
1372 #
1373 # This exists to help extensions wrap steps if necessary
1373 # This exists to help extensions wrap steps if necessary
1374 pulldiscoverymapping = {}
1374 pulldiscoverymapping = {}
1375
1375
1376 def pulldiscovery(stepname):
1376 def pulldiscovery(stepname):
1377 """decorator for function performing discovery before pull
1377 """decorator for function performing discovery before pull
1378
1378
1379 The function is added to the step -> function mapping and appended to the
1379 The function is added to the step -> function mapping and appended to the
1380 list of steps. Beware that decorated function will be added in order (this
1380 list of steps. Beware that decorated function will be added in order (this
1381 may matter).
1381 may matter).
1382
1382
1383 You can only use this decorator for a new step, if you want to wrap a step
1383 You can only use this decorator for a new step, if you want to wrap a step
1384 from an extension, change the pulldiscovery dictionary directly."""
1384 from an extension, change the pulldiscovery dictionary directly."""
1385 def dec(func):
1385 def dec(func):
1386 assert stepname not in pulldiscoverymapping
1386 assert stepname not in pulldiscoverymapping
1387 pulldiscoverymapping[stepname] = func
1387 pulldiscoverymapping[stepname] = func
1388 pulldiscoveryorder.append(stepname)
1388 pulldiscoveryorder.append(stepname)
1389 return func
1389 return func
1390 return dec
1390 return dec
1391
1391
1392 def _pulldiscovery(pullop):
1392 def _pulldiscovery(pullop):
1393 """Run all discovery steps"""
1393 """Run all discovery steps"""
1394 for stepname in pulldiscoveryorder:
1394 for stepname in pulldiscoveryorder:
1395 step = pulldiscoverymapping[stepname]
1395 step = pulldiscoverymapping[stepname]
1396 step(pullop)
1396 step(pullop)
1397
1397
1398 @pulldiscovery('b1:bookmarks')
1398 @pulldiscovery('b1:bookmarks')
1399 def _pullbookmarkbundle1(pullop):
1399 def _pullbookmarkbundle1(pullop):
1400 """fetch bookmark data in bundle1 case
1400 """fetch bookmark data in bundle1 case
1401
1401
1402 If not using bundle2, we have to fetch bookmarks before changeset
1402 If not using bundle2, we have to fetch bookmarks before changeset
1403 discovery to reduce the chance and impact of race conditions."""
1403 discovery to reduce the chance and impact of race conditions."""
1404 if pullop.remotebookmarks is not None:
1404 if pullop.remotebookmarks is not None:
1405 return
1405 return
1406 if pullop.canusebundle2 and 'listkeys' in pullop.remotebundle2caps:
1406 if pullop.canusebundle2 and 'listkeys' in pullop.remotebundle2caps:
1407 # all known bundle2 servers now support listkeys, but lets be nice with
1407 # all known bundle2 servers now support listkeys, but lets be nice with
1408 # new implementation.
1408 # new implementation.
1409 return
1409 return
1410 books = pullop.remote.listkeys('bookmarks')
1410 books = pullop.remote.listkeys('bookmarks')
1411 pullop.remotebookmarks = bookmod.unhexlifybookmarks(books)
1411 pullop.remotebookmarks = bookmod.unhexlifybookmarks(books)
1412
1412
1413
1413
1414 @pulldiscovery('changegroup')
1414 @pulldiscovery('changegroup')
1415 def _pulldiscoverychangegroup(pullop):
1415 def _pulldiscoverychangegroup(pullop):
1416 """discovery phase for the pull
1416 """discovery phase for the pull
1417
1417
1418 Current handle changeset discovery only, will change handle all discovery
1418 Current handle changeset discovery only, will change handle all discovery
1419 at some point."""
1419 at some point."""
1420 tmp = discovery.findcommonincoming(pullop.repo,
1420 tmp = discovery.findcommonincoming(pullop.repo,
1421 pullop.remote,
1421 pullop.remote,
1422 heads=pullop.heads,
1422 heads=pullop.heads,
1423 force=pullop.force)
1423 force=pullop.force)
1424 common, fetch, rheads = tmp
1424 common, fetch, rheads = tmp
1425 nm = pullop.repo.unfiltered().changelog.nodemap
1425 nm = pullop.repo.unfiltered().changelog.nodemap
1426 if fetch and rheads:
1426 if fetch and rheads:
1427 # If a remote heads is filtered locally, put in back in common.
1427 # If a remote heads is filtered locally, put in back in common.
1428 #
1428 #
1429 # This is a hackish solution to catch most of "common but locally
1429 # This is a hackish solution to catch most of "common but locally
1430 # hidden situation". We do not performs discovery on unfiltered
1430 # hidden situation". We do not performs discovery on unfiltered
1431 # repository because it end up doing a pathological amount of round
1431 # repository because it end up doing a pathological amount of round
1432 # trip for w huge amount of changeset we do not care about.
1432 # trip for w huge amount of changeset we do not care about.
1433 #
1433 #
1434 # If a set of such "common but filtered" changeset exist on the server
1434 # If a set of such "common but filtered" changeset exist on the server
1435 # but are not including a remote heads, we'll not be able to detect it,
1435 # but are not including a remote heads, we'll not be able to detect it,
1436 scommon = set(common)
1436 scommon = set(common)
1437 for n in rheads:
1437 for n in rheads:
1438 if n in nm:
1438 if n in nm:
1439 if n not in scommon:
1439 if n not in scommon:
1440 common.append(n)
1440 common.append(n)
1441 if set(rheads).issubset(set(common)):
1441 if set(rheads).issubset(set(common)):
1442 fetch = []
1442 fetch = []
1443 pullop.common = common
1443 pullop.common = common
1444 pullop.fetch = fetch
1444 pullop.fetch = fetch
1445 pullop.rheads = rheads
1445 pullop.rheads = rheads
1446
1446
1447 def _pullbundle2(pullop):
1447 def _pullbundle2(pullop):
1448 """pull data using bundle2
1448 """pull data using bundle2
1449
1449
1450 For now, the only supported data are changegroup."""
1450 For now, the only supported data are changegroup."""
1451 kwargs = {'bundlecaps': caps20to10(pullop.repo)}
1451 kwargs = {'bundlecaps': caps20to10(pullop.repo)}
1452
1452
1453 # make ui easier to access
1453 # make ui easier to access
1454 ui = pullop.repo.ui
1454 ui = pullop.repo.ui
1455
1455
1456 # At the moment we don't do stream clones over bundle2. If that is
1456 # At the moment we don't do stream clones over bundle2. If that is
1457 # implemented then here's where the check for that will go.
1457 # implemented then here's where the check for that will go.
1458 streaming = False
1458 streaming = False
1459
1459
1460 # declare pull perimeters
1460 # declare pull perimeters
1461 kwargs['common'] = pullop.common
1461 kwargs['common'] = pullop.common
1462 kwargs['heads'] = pullop.heads or pullop.rheads
1462 kwargs['heads'] = pullop.heads or pullop.rheads
1463
1463
1464 # pulling changegroup
1464 if True:
1465 pullop.stepsdone.add('changegroup')
1465 # pulling changegroup
1466 pullop.stepsdone.add('changegroup')
1466
1467
1467 kwargs['cg'] = pullop.fetch
1468 kwargs['cg'] = pullop.fetch
1468
1469
1469 legacyphase = 'phases' in ui.configlist('devel', 'legacy.exchange')
1470 legacyphase = 'phases' in ui.configlist('devel', 'legacy.exchange')
1470 hasbinaryphase = 'heads' in pullop.remotebundle2caps.get('phases', ())
1471 hasbinaryphase = 'heads' in pullop.remotebundle2caps.get('phases', ())
1471 if (not legacyphase and hasbinaryphase):
1472 if (not legacyphase and hasbinaryphase):
1472 kwargs['phases'] = True
1473 kwargs['phases'] = True
1473 pullop.stepsdone.add('phases')
1474 pullop.stepsdone.add('phases')
1474
1475
1475 if 'listkeys' in pullop.remotebundle2caps:
1476 if 'listkeys' in pullop.remotebundle2caps:
1476 if 'phases' not in pullop.stepsdone:
1477 if 'phases' not in pullop.stepsdone:
1477 kwargs['listkeys'] = ['phases']
1478 kwargs['listkeys'] = ['phases']
1478
1479
1479 bookmarksrequested = False
1480 bookmarksrequested = False
1480 legacybookmark = 'bookmarks' in ui.configlist('devel', 'legacy.exchange')
1481 legacybookmark = 'bookmarks' in ui.configlist('devel', 'legacy.exchange')
1481 hasbinarybook = 'bookmarks' in pullop.remotebundle2caps
1482 hasbinarybook = 'bookmarks' in pullop.remotebundle2caps
1482
1483
1483 if pullop.remotebookmarks is not None:
1484 if pullop.remotebookmarks is not None:
1484 pullop.stepsdone.add('request-bookmarks')
1485 pullop.stepsdone.add('request-bookmarks')
1485
1486
1486 if ('request-bookmarks' not in pullop.stepsdone
1487 if ('request-bookmarks' not in pullop.stepsdone
1487 and pullop.remotebookmarks is None
1488 and pullop.remotebookmarks is None
1488 and not legacybookmark and hasbinarybook):
1489 and not legacybookmark and hasbinarybook):
1489 kwargs['bookmarks'] = True
1490 kwargs['bookmarks'] = True
1490 bookmarksrequested = True
1491 bookmarksrequested = True
1491
1492
1492 if 'listkeys' in pullop.remotebundle2caps:
1493 if 'listkeys' in pullop.remotebundle2caps:
1493 if 'request-bookmarks' not in pullop.stepsdone:
1494 if 'request-bookmarks' not in pullop.stepsdone:
1494 # make sure to always includes bookmark data when migrating
1495 # make sure to always includes bookmark data when migrating
1495 # `hg incoming --bundle` to using this function.
1496 # `hg incoming --bundle` to using this function.
1496 pullop.stepsdone.add('request-bookmarks')
1497 pullop.stepsdone.add('request-bookmarks')
1497 kwargs.setdefault('listkeys', []).append('bookmarks')
1498 kwargs.setdefault('listkeys', []).append('bookmarks')
1498
1499
1499 # If this is a full pull / clone and the server supports the clone bundles
1500 # If this is a full pull / clone and the server supports the clone bundles
1500 # feature, tell the server whether we attempted a clone bundle. The
1501 # feature, tell the server whether we attempted a clone bundle. The
1501 # presence of this flag indicates the client supports clone bundles. This
1502 # presence of this flag indicates the client supports clone bundles. This
1502 # will enable the server to treat clients that support clone bundles
1503 # will enable the server to treat clients that support clone bundles
1503 # differently from those that don't.
1504 # differently from those that don't.
1504 if (pullop.remote.capable('clonebundles')
1505 if (pullop.remote.capable('clonebundles')
1505 and pullop.heads is None and list(pullop.common) == [nullid]):
1506 and pullop.heads is None and list(pullop.common) == [nullid]):
1506 kwargs['cbattempted'] = pullop.clonebundleattempted
1507 kwargs['cbattempted'] = pullop.clonebundleattempted
1507
1508
1508 if streaming:
1509 if streaming:
1509 pullop.repo.ui.status(_('streaming all changes\n'))
1510 pullop.repo.ui.status(_('streaming all changes\n'))
1510 elif not pullop.fetch:
1511 elif not pullop.fetch:
1511 pullop.repo.ui.status(_("no changes found\n"))
1512 pullop.repo.ui.status(_("no changes found\n"))
1512 pullop.cgresult = 0
1513 pullop.cgresult = 0
1513 else:
1514 else:
1514 if pullop.heads is None and list(pullop.common) == [nullid]:
1515 if pullop.heads is None and list(pullop.common) == [nullid]:
1515 pullop.repo.ui.status(_("requesting all changes\n"))
1516 pullop.repo.ui.status(_("requesting all changes\n"))
1516 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1517 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1517 remoteversions = bundle2.obsmarkersversion(pullop.remotebundle2caps)
1518 remoteversions = bundle2.obsmarkersversion(pullop.remotebundle2caps)
1518 if obsolete.commonversion(remoteversions) is not None:
1519 if obsolete.commonversion(remoteversions) is not None:
1519 kwargs['obsmarkers'] = True
1520 kwargs['obsmarkers'] = True
1520 pullop.stepsdone.add('obsmarkers')
1521 pullop.stepsdone.add('obsmarkers')
1521 _pullbundle2extraprepare(pullop, kwargs)
1522 _pullbundle2extraprepare(pullop, kwargs)
1522 bundle = pullop.remote.getbundle('pull', **pycompat.strkwargs(kwargs))
1523 bundle = pullop.remote.getbundle('pull', **pycompat.strkwargs(kwargs))
1523 try:
1524 try:
1524 op = bundle2.bundleoperation(pullop.repo, pullop.gettransaction)
1525 op = bundle2.bundleoperation(pullop.repo, pullop.gettransaction)
1525 op.modes['bookmarks'] = 'records'
1526 op.modes['bookmarks'] = 'records'
1526 bundle2.processbundle(pullop.repo, bundle, op=op)
1527 bundle2.processbundle(pullop.repo, bundle, op=op)
1527 except bundle2.AbortFromPart as exc:
1528 except bundle2.AbortFromPart as exc:
1528 pullop.repo.ui.status(_('remote: abort: %s\n') % exc)
1529 pullop.repo.ui.status(_('remote: abort: %s\n') % exc)
1529 raise error.Abort(_('pull failed on remote'), hint=exc.hint)
1530 raise error.Abort(_('pull failed on remote'), hint=exc.hint)
1530 except error.BundleValueError as exc:
1531 except error.BundleValueError as exc:
1531 raise error.Abort(_('missing support for %s') % exc)
1532 raise error.Abort(_('missing support for %s') % exc)
1532
1533
1533 if pullop.fetch:
1534 if pullop.fetch:
1534 pullop.cgresult = bundle2.combinechangegroupresults(op)
1535 pullop.cgresult = bundle2.combinechangegroupresults(op)
1535
1536
1536 # processing phases change
1537 # processing phases change
1537 for namespace, value in op.records['listkeys']:
1538 for namespace, value in op.records['listkeys']:
1538 if namespace == 'phases':
1539 if namespace == 'phases':
1539 _pullapplyphases(pullop, value)
1540 _pullapplyphases(pullop, value)
1540
1541
1541 # processing bookmark update
1542 # processing bookmark update
1542 if bookmarksrequested:
1543 if bookmarksrequested:
1543 books = {}
1544 books = {}
1544 for record in op.records['bookmarks']:
1545 for record in op.records['bookmarks']:
1545 books[record['bookmark']] = record["node"]
1546 books[record['bookmark']] = record["node"]
1546 pullop.remotebookmarks = books
1547 pullop.remotebookmarks = books
1547 else:
1548 else:
1548 for namespace, value in op.records['listkeys']:
1549 for namespace, value in op.records['listkeys']:
1549 if namespace == 'bookmarks':
1550 if namespace == 'bookmarks':
1550 pullop.remotebookmarks = bookmod.unhexlifybookmarks(value)
1551 pullop.remotebookmarks = bookmod.unhexlifybookmarks(value)
1551
1552
1552 # bookmark data were either already there or pulled in the bundle
1553 # bookmark data were either already there or pulled in the bundle
1553 if pullop.remotebookmarks is not None:
1554 if pullop.remotebookmarks is not None:
1554 _pullbookmarks(pullop)
1555 _pullbookmarks(pullop)
1555
1556
1556 def _pullbundle2extraprepare(pullop, kwargs):
1557 def _pullbundle2extraprepare(pullop, kwargs):
1557 """hook function so that extensions can extend the getbundle call"""
1558 """hook function so that extensions can extend the getbundle call"""
1558
1559
1559 def _pullchangeset(pullop):
1560 def _pullchangeset(pullop):
1560 """pull changeset from unbundle into the local repo"""
1561 """pull changeset from unbundle into the local repo"""
1561 # We delay the open of the transaction as late as possible so we
1562 # We delay the open of the transaction as late as possible so we
1562 # don't open transaction for nothing or you break future useful
1563 # don't open transaction for nothing or you break future useful
1563 # rollback call
1564 # rollback call
1564 if 'changegroup' in pullop.stepsdone:
1565 if 'changegroup' in pullop.stepsdone:
1565 return
1566 return
1566 pullop.stepsdone.add('changegroup')
1567 pullop.stepsdone.add('changegroup')
1567 if not pullop.fetch:
1568 if not pullop.fetch:
1568 pullop.repo.ui.status(_("no changes found\n"))
1569 pullop.repo.ui.status(_("no changes found\n"))
1569 pullop.cgresult = 0
1570 pullop.cgresult = 0
1570 return
1571 return
1571 tr = pullop.gettransaction()
1572 tr = pullop.gettransaction()
1572 if pullop.heads is None and list(pullop.common) == [nullid]:
1573 if pullop.heads is None and list(pullop.common) == [nullid]:
1573 pullop.repo.ui.status(_("requesting all changes\n"))
1574 pullop.repo.ui.status(_("requesting all changes\n"))
1574 elif pullop.heads is None and pullop.remote.capable('changegroupsubset'):
1575 elif pullop.heads is None and pullop.remote.capable('changegroupsubset'):
1575 # issue1320, avoid a race if remote changed after discovery
1576 # issue1320, avoid a race if remote changed after discovery
1576 pullop.heads = pullop.rheads
1577 pullop.heads = pullop.rheads
1577
1578
1578 if pullop.remote.capable('getbundle'):
1579 if pullop.remote.capable('getbundle'):
1579 # TODO: get bundlecaps from remote
1580 # TODO: get bundlecaps from remote
1580 cg = pullop.remote.getbundle('pull', common=pullop.common,
1581 cg = pullop.remote.getbundle('pull', common=pullop.common,
1581 heads=pullop.heads or pullop.rheads)
1582 heads=pullop.heads or pullop.rheads)
1582 elif pullop.heads is None:
1583 elif pullop.heads is None:
1583 cg = pullop.remote.changegroup(pullop.fetch, 'pull')
1584 cg = pullop.remote.changegroup(pullop.fetch, 'pull')
1584 elif not pullop.remote.capable('changegroupsubset'):
1585 elif not pullop.remote.capable('changegroupsubset'):
1585 raise error.Abort(_("partial pull cannot be done because "
1586 raise error.Abort(_("partial pull cannot be done because "
1586 "other repository doesn't support "
1587 "other repository doesn't support "
1587 "changegroupsubset."))
1588 "changegroupsubset."))
1588 else:
1589 else:
1589 cg = pullop.remote.changegroupsubset(pullop.fetch, pullop.heads, 'pull')
1590 cg = pullop.remote.changegroupsubset(pullop.fetch, pullop.heads, 'pull')
1590 bundleop = bundle2.applybundle(pullop.repo, cg, tr, 'pull',
1591 bundleop = bundle2.applybundle(pullop.repo, cg, tr, 'pull',
1591 pullop.remote.url())
1592 pullop.remote.url())
1592 pullop.cgresult = bundle2.combinechangegroupresults(bundleop)
1593 pullop.cgresult = bundle2.combinechangegroupresults(bundleop)
1593
1594
1594 def _pullphase(pullop):
1595 def _pullphase(pullop):
1595 # Get remote phases data from remote
1596 # Get remote phases data from remote
1596 if 'phases' in pullop.stepsdone:
1597 if 'phases' in pullop.stepsdone:
1597 return
1598 return
1598 remotephases = pullop.remote.listkeys('phases')
1599 remotephases = pullop.remote.listkeys('phases')
1599 _pullapplyphases(pullop, remotephases)
1600 _pullapplyphases(pullop, remotephases)
1600
1601
1601 def _pullapplyphases(pullop, remotephases):
1602 def _pullapplyphases(pullop, remotephases):
1602 """apply phase movement from observed remote state"""
1603 """apply phase movement from observed remote state"""
1603 if 'phases' in pullop.stepsdone:
1604 if 'phases' in pullop.stepsdone:
1604 return
1605 return
1605 pullop.stepsdone.add('phases')
1606 pullop.stepsdone.add('phases')
1606 publishing = bool(remotephases.get('publishing', False))
1607 publishing = bool(remotephases.get('publishing', False))
1607 if remotephases and not publishing:
1608 if remotephases and not publishing:
1608 # remote is new and non-publishing
1609 # remote is new and non-publishing
1609 pheads, _dr = phases.analyzeremotephases(pullop.repo,
1610 pheads, _dr = phases.analyzeremotephases(pullop.repo,
1610 pullop.pulledsubset,
1611 pullop.pulledsubset,
1611 remotephases)
1612 remotephases)
1612 dheads = pullop.pulledsubset
1613 dheads = pullop.pulledsubset
1613 else:
1614 else:
1614 # Remote is old or publishing all common changesets
1615 # Remote is old or publishing all common changesets
1615 # should be seen as public
1616 # should be seen as public
1616 pheads = pullop.pulledsubset
1617 pheads = pullop.pulledsubset
1617 dheads = []
1618 dheads = []
1618 unfi = pullop.repo.unfiltered()
1619 unfi = pullop.repo.unfiltered()
1619 phase = unfi._phasecache.phase
1620 phase = unfi._phasecache.phase
1620 rev = unfi.changelog.nodemap.get
1621 rev = unfi.changelog.nodemap.get
1621 public = phases.public
1622 public = phases.public
1622 draft = phases.draft
1623 draft = phases.draft
1623
1624
1624 # exclude changesets already public locally and update the others
1625 # exclude changesets already public locally and update the others
1625 pheads = [pn for pn in pheads if phase(unfi, rev(pn)) > public]
1626 pheads = [pn for pn in pheads if phase(unfi, rev(pn)) > public]
1626 if pheads:
1627 if pheads:
1627 tr = pullop.gettransaction()
1628 tr = pullop.gettransaction()
1628 phases.advanceboundary(pullop.repo, tr, public, pheads)
1629 phases.advanceboundary(pullop.repo, tr, public, pheads)
1629
1630
1630 # exclude changesets already draft locally and update the others
1631 # exclude changesets already draft locally and update the others
1631 dheads = [pn for pn in dheads if phase(unfi, rev(pn)) > draft]
1632 dheads = [pn for pn in dheads if phase(unfi, rev(pn)) > draft]
1632 if dheads:
1633 if dheads:
1633 tr = pullop.gettransaction()
1634 tr = pullop.gettransaction()
1634 phases.advanceboundary(pullop.repo, tr, draft, dheads)
1635 phases.advanceboundary(pullop.repo, tr, draft, dheads)
1635
1636
1636 def _pullbookmarks(pullop):
1637 def _pullbookmarks(pullop):
1637 """process the remote bookmark information to update the local one"""
1638 """process the remote bookmark information to update the local one"""
1638 if 'bookmarks' in pullop.stepsdone:
1639 if 'bookmarks' in pullop.stepsdone:
1639 return
1640 return
1640 pullop.stepsdone.add('bookmarks')
1641 pullop.stepsdone.add('bookmarks')
1641 repo = pullop.repo
1642 repo = pullop.repo
1642 remotebookmarks = pullop.remotebookmarks
1643 remotebookmarks = pullop.remotebookmarks
1643 bookmod.updatefromremote(repo.ui, repo, remotebookmarks,
1644 bookmod.updatefromremote(repo.ui, repo, remotebookmarks,
1644 pullop.remote.url(),
1645 pullop.remote.url(),
1645 pullop.gettransaction,
1646 pullop.gettransaction,
1646 explicit=pullop.explicitbookmarks)
1647 explicit=pullop.explicitbookmarks)
1647
1648
1648 def _pullobsolete(pullop):
1649 def _pullobsolete(pullop):
1649 """utility function to pull obsolete markers from a remote
1650 """utility function to pull obsolete markers from a remote
1650
1651
1651 The `gettransaction` is function that return the pull transaction, creating
1652 The `gettransaction` is function that return the pull transaction, creating
1652 one if necessary. We return the transaction to inform the calling code that
1653 one if necessary. We return the transaction to inform the calling code that
1653 a new transaction have been created (when applicable).
1654 a new transaction have been created (when applicable).
1654
1655
1655 Exists mostly to allow overriding for experimentation purpose"""
1656 Exists mostly to allow overriding for experimentation purpose"""
1656 if 'obsmarkers' in pullop.stepsdone:
1657 if 'obsmarkers' in pullop.stepsdone:
1657 return
1658 return
1658 pullop.stepsdone.add('obsmarkers')
1659 pullop.stepsdone.add('obsmarkers')
1659 tr = None
1660 tr = None
1660 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1661 if obsolete.isenabled(pullop.repo, obsolete.exchangeopt):
1661 pullop.repo.ui.debug('fetching remote obsolete markers\n')
1662 pullop.repo.ui.debug('fetching remote obsolete markers\n')
1662 remoteobs = pullop.remote.listkeys('obsolete')
1663 remoteobs = pullop.remote.listkeys('obsolete')
1663 if 'dump0' in remoteobs:
1664 if 'dump0' in remoteobs:
1664 tr = pullop.gettransaction()
1665 tr = pullop.gettransaction()
1665 markers = []
1666 markers = []
1666 for key in sorted(remoteobs, reverse=True):
1667 for key in sorted(remoteobs, reverse=True):
1667 if key.startswith('dump'):
1668 if key.startswith('dump'):
1668 data = util.b85decode(remoteobs[key])
1669 data = util.b85decode(remoteobs[key])
1669 version, newmarks = obsolete._readmarkers(data)
1670 version, newmarks = obsolete._readmarkers(data)
1670 markers += newmarks
1671 markers += newmarks
1671 if markers:
1672 if markers:
1672 pullop.repo.obsstore.add(tr, markers)
1673 pullop.repo.obsstore.add(tr, markers)
1673 pullop.repo.invalidatevolatilesets()
1674 pullop.repo.invalidatevolatilesets()
1674 return tr
1675 return tr
1675
1676
1676 def caps20to10(repo):
1677 def caps20to10(repo):
1677 """return a set with appropriate options to use bundle20 during getbundle"""
1678 """return a set with appropriate options to use bundle20 during getbundle"""
1678 caps = {'HG20'}
1679 caps = {'HG20'}
1679 capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo))
1680 capsblob = bundle2.encodecaps(bundle2.getrepocaps(repo))
1680 caps.add('bundle2=' + urlreq.quote(capsblob))
1681 caps.add('bundle2=' + urlreq.quote(capsblob))
1681 return caps
1682 return caps
1682
1683
1683 # List of names of steps to perform for a bundle2 for getbundle, order matters.
1684 # List of names of steps to perform for a bundle2 for getbundle, order matters.
1684 getbundle2partsorder = []
1685 getbundle2partsorder = []
1685
1686
1686 # Mapping between step name and function
1687 # Mapping between step name and function
1687 #
1688 #
1688 # This exists to help extensions wrap steps if necessary
1689 # This exists to help extensions wrap steps if necessary
1689 getbundle2partsmapping = {}
1690 getbundle2partsmapping = {}
1690
1691
1691 def getbundle2partsgenerator(stepname, idx=None):
1692 def getbundle2partsgenerator(stepname, idx=None):
1692 """decorator for function generating bundle2 part for getbundle
1693 """decorator for function generating bundle2 part for getbundle
1693
1694
1694 The function is added to the step -> function mapping and appended to the
1695 The function is added to the step -> function mapping and appended to the
1695 list of steps. Beware that decorated functions will be added in order
1696 list of steps. Beware that decorated functions will be added in order
1696 (this may matter).
1697 (this may matter).
1697
1698
1698 You can only use this decorator for new steps, if you want to wrap a step
1699 You can only use this decorator for new steps, if you want to wrap a step
1699 from an extension, attack the getbundle2partsmapping dictionary directly."""
1700 from an extension, attack the getbundle2partsmapping dictionary directly."""
1700 def dec(func):
1701 def dec(func):
1701 assert stepname not in getbundle2partsmapping
1702 assert stepname not in getbundle2partsmapping
1702 getbundle2partsmapping[stepname] = func
1703 getbundle2partsmapping[stepname] = func
1703 if idx is None:
1704 if idx is None:
1704 getbundle2partsorder.append(stepname)
1705 getbundle2partsorder.append(stepname)
1705 else:
1706 else:
1706 getbundle2partsorder.insert(idx, stepname)
1707 getbundle2partsorder.insert(idx, stepname)
1707 return func
1708 return func
1708 return dec
1709 return dec
1709
1710
1710 def bundle2requested(bundlecaps):
1711 def bundle2requested(bundlecaps):
1711 if bundlecaps is not None:
1712 if bundlecaps is not None:
1712 return any(cap.startswith('HG2') for cap in bundlecaps)
1713 return any(cap.startswith('HG2') for cap in bundlecaps)
1713 return False
1714 return False
1714
1715
1715 def getbundlechunks(repo, source, heads=None, common=None, bundlecaps=None,
1716 def getbundlechunks(repo, source, heads=None, common=None, bundlecaps=None,
1716 **kwargs):
1717 **kwargs):
1717 """Return chunks constituting a bundle's raw data.
1718 """Return chunks constituting a bundle's raw data.
1718
1719
1719 Could be a bundle HG10 or a bundle HG20 depending on bundlecaps
1720 Could be a bundle HG10 or a bundle HG20 depending on bundlecaps
1720 passed.
1721 passed.
1721
1722
1722 Returns an iterator over raw chunks (of varying sizes).
1723 Returns an iterator over raw chunks (of varying sizes).
1723 """
1724 """
1724 kwargs = pycompat.byteskwargs(kwargs)
1725 kwargs = pycompat.byteskwargs(kwargs)
1725 usebundle2 = bundle2requested(bundlecaps)
1726 usebundle2 = bundle2requested(bundlecaps)
1726 # bundle10 case
1727 # bundle10 case
1727 if not usebundle2:
1728 if not usebundle2:
1728 if bundlecaps and not kwargs.get('cg', True):
1729 if bundlecaps and not kwargs.get('cg', True):
1729 raise ValueError(_('request for bundle10 must include changegroup'))
1730 raise ValueError(_('request for bundle10 must include changegroup'))
1730
1731
1731 if kwargs:
1732 if kwargs:
1732 raise ValueError(_('unsupported getbundle arguments: %s')
1733 raise ValueError(_('unsupported getbundle arguments: %s')
1733 % ', '.join(sorted(kwargs.keys())))
1734 % ', '.join(sorted(kwargs.keys())))
1734 outgoing = _computeoutgoing(repo, heads, common)
1735 outgoing = _computeoutgoing(repo, heads, common)
1735 return changegroup.makestream(repo, outgoing, '01', source,
1736 return changegroup.makestream(repo, outgoing, '01', source,
1736 bundlecaps=bundlecaps)
1737 bundlecaps=bundlecaps)
1737
1738
1738 # bundle20 case
1739 # bundle20 case
1739 b2caps = {}
1740 b2caps = {}
1740 for bcaps in bundlecaps:
1741 for bcaps in bundlecaps:
1741 if bcaps.startswith('bundle2='):
1742 if bcaps.startswith('bundle2='):
1742 blob = urlreq.unquote(bcaps[len('bundle2='):])
1743 blob = urlreq.unquote(bcaps[len('bundle2='):])
1743 b2caps.update(bundle2.decodecaps(blob))
1744 b2caps.update(bundle2.decodecaps(blob))
1744 bundler = bundle2.bundle20(repo.ui, b2caps)
1745 bundler = bundle2.bundle20(repo.ui, b2caps)
1745
1746
1746 kwargs['heads'] = heads
1747 kwargs['heads'] = heads
1747 kwargs['common'] = common
1748 kwargs['common'] = common
1748
1749
1749 for name in getbundle2partsorder:
1750 for name in getbundle2partsorder:
1750 func = getbundle2partsmapping[name]
1751 func = getbundle2partsmapping[name]
1751 func(bundler, repo, source, bundlecaps=bundlecaps, b2caps=b2caps,
1752 func(bundler, repo, source, bundlecaps=bundlecaps, b2caps=b2caps,
1752 **pycompat.strkwargs(kwargs))
1753 **pycompat.strkwargs(kwargs))
1753
1754
1754 return bundler.getchunks()
1755 return bundler.getchunks()
1755
1756
1756 @getbundle2partsgenerator('stream')
1757 @getbundle2partsgenerator('stream')
1757 def _getbundlestream(bundler, repo, source, bundlecaps=None,
1758 def _getbundlestream(bundler, repo, source, bundlecaps=None,
1758 b2caps=None, heads=None, common=None, **kwargs):
1759 b2caps=None, heads=None, common=None, **kwargs):
1759 if not kwargs.get('stream', False):
1760 if not kwargs.get('stream', False):
1760 return
1761 return
1761 filecount, bytecount, it = streamclone.generatev2(repo)
1762 filecount, bytecount, it = streamclone.generatev2(repo)
1762 requirements = ' '.join(repo.requirements)
1763 requirements = ' '.join(repo.requirements)
1763 part = bundler.newpart('stream', data=it)
1764 part = bundler.newpart('stream', data=it)
1764 part.addparam('bytecount', '%d' % bytecount, mandatory=True)
1765 part.addparam('bytecount', '%d' % bytecount, mandatory=True)
1765 part.addparam('filecount', '%d' % filecount, mandatory=True)
1766 part.addparam('filecount', '%d' % filecount, mandatory=True)
1766 part.addparam('requirements', requirements, mandatory=True)
1767 part.addparam('requirements', requirements, mandatory=True)
1767 part.addparam('version', 'v2', mandatory=True)
1768 part.addparam('version', 'v2', mandatory=True)
1768
1769
1769 @getbundle2partsgenerator('changegroup')
1770 @getbundle2partsgenerator('changegroup')
1770 def _getbundlechangegrouppart(bundler, repo, source, bundlecaps=None,
1771 def _getbundlechangegrouppart(bundler, repo, source, bundlecaps=None,
1771 b2caps=None, heads=None, common=None, **kwargs):
1772 b2caps=None, heads=None, common=None, **kwargs):
1772 """add a changegroup part to the requested bundle"""
1773 """add a changegroup part to the requested bundle"""
1773 cgstream = None
1774 cgstream = None
1774 if kwargs.get(r'cg', True):
1775 if kwargs.get(r'cg', True):
1775 # build changegroup bundle here.
1776 # build changegroup bundle here.
1776 version = '01'
1777 version = '01'
1777 cgversions = b2caps.get('changegroup')
1778 cgversions = b2caps.get('changegroup')
1778 if cgversions: # 3.1 and 3.2 ship with an empty value
1779 if cgversions: # 3.1 and 3.2 ship with an empty value
1779 cgversions = [v for v in cgversions
1780 cgversions = [v for v in cgversions
1780 if v in changegroup.supportedoutgoingversions(repo)]
1781 if v in changegroup.supportedoutgoingversions(repo)]
1781 if not cgversions:
1782 if not cgversions:
1782 raise ValueError(_('no common changegroup version'))
1783 raise ValueError(_('no common changegroup version'))
1783 version = max(cgversions)
1784 version = max(cgversions)
1784 outgoing = _computeoutgoing(repo, heads, common)
1785 outgoing = _computeoutgoing(repo, heads, common)
1785 if outgoing.missing:
1786 if outgoing.missing:
1786 cgstream = changegroup.makestream(repo, outgoing, version, source,
1787 cgstream = changegroup.makestream(repo, outgoing, version, source,
1787 bundlecaps=bundlecaps)
1788 bundlecaps=bundlecaps)
1788
1789
1789 if cgstream:
1790 if cgstream:
1790 part = bundler.newpart('changegroup', data=cgstream)
1791 part = bundler.newpart('changegroup', data=cgstream)
1791 if cgversions:
1792 if cgversions:
1792 part.addparam('version', version)
1793 part.addparam('version', version)
1793 part.addparam('nbchanges', '%d' % len(outgoing.missing),
1794 part.addparam('nbchanges', '%d' % len(outgoing.missing),
1794 mandatory=False)
1795 mandatory=False)
1795 if 'treemanifest' in repo.requirements:
1796 if 'treemanifest' in repo.requirements:
1796 part.addparam('treemanifest', '1')
1797 part.addparam('treemanifest', '1')
1797
1798
1798 @getbundle2partsgenerator('bookmarks')
1799 @getbundle2partsgenerator('bookmarks')
1799 def _getbundlebookmarkpart(bundler, repo, source, bundlecaps=None,
1800 def _getbundlebookmarkpart(bundler, repo, source, bundlecaps=None,
1800 b2caps=None, **kwargs):
1801 b2caps=None, **kwargs):
1801 """add a bookmark part to the requested bundle"""
1802 """add a bookmark part to the requested bundle"""
1802 if not kwargs.get(r'bookmarks', False):
1803 if not kwargs.get(r'bookmarks', False):
1803 return
1804 return
1804 if 'bookmarks' not in b2caps:
1805 if 'bookmarks' not in b2caps:
1805 raise ValueError(_('no common bookmarks exchange method'))
1806 raise ValueError(_('no common bookmarks exchange method'))
1806 books = bookmod.listbinbookmarks(repo)
1807 books = bookmod.listbinbookmarks(repo)
1807 data = bookmod.binaryencode(books)
1808 data = bookmod.binaryencode(books)
1808 if data:
1809 if data:
1809 bundler.newpart('bookmarks', data=data)
1810 bundler.newpart('bookmarks', data=data)
1810
1811
1811 @getbundle2partsgenerator('listkeys')
1812 @getbundle2partsgenerator('listkeys')
1812 def _getbundlelistkeysparts(bundler, repo, source, bundlecaps=None,
1813 def _getbundlelistkeysparts(bundler, repo, source, bundlecaps=None,
1813 b2caps=None, **kwargs):
1814 b2caps=None, **kwargs):
1814 """add parts containing listkeys namespaces to the requested bundle"""
1815 """add parts containing listkeys namespaces to the requested bundle"""
1815 listkeys = kwargs.get(r'listkeys', ())
1816 listkeys = kwargs.get(r'listkeys', ())
1816 for namespace in listkeys:
1817 for namespace in listkeys:
1817 part = bundler.newpart('listkeys')
1818 part = bundler.newpart('listkeys')
1818 part.addparam('namespace', namespace)
1819 part.addparam('namespace', namespace)
1819 keys = repo.listkeys(namespace).items()
1820 keys = repo.listkeys(namespace).items()
1820 part.data = pushkey.encodekeys(keys)
1821 part.data = pushkey.encodekeys(keys)
1821
1822
1822 @getbundle2partsgenerator('obsmarkers')
1823 @getbundle2partsgenerator('obsmarkers')
1823 def _getbundleobsmarkerpart(bundler, repo, source, bundlecaps=None,
1824 def _getbundleobsmarkerpart(bundler, repo, source, bundlecaps=None,
1824 b2caps=None, heads=None, **kwargs):
1825 b2caps=None, heads=None, **kwargs):
1825 """add an obsolescence markers part to the requested bundle"""
1826 """add an obsolescence markers part to the requested bundle"""
1826 if kwargs.get(r'obsmarkers', False):
1827 if kwargs.get(r'obsmarkers', False):
1827 if heads is None:
1828 if heads is None:
1828 heads = repo.heads()
1829 heads = repo.heads()
1829 subset = [c.node() for c in repo.set('::%ln', heads)]
1830 subset = [c.node() for c in repo.set('::%ln', heads)]
1830 markers = repo.obsstore.relevantmarkers(subset)
1831 markers = repo.obsstore.relevantmarkers(subset)
1831 markers = sorted(markers)
1832 markers = sorted(markers)
1832 bundle2.buildobsmarkerspart(bundler, markers)
1833 bundle2.buildobsmarkerspart(bundler, markers)
1833
1834
1834 @getbundle2partsgenerator('phases')
1835 @getbundle2partsgenerator('phases')
1835 def _getbundlephasespart(bundler, repo, source, bundlecaps=None,
1836 def _getbundlephasespart(bundler, repo, source, bundlecaps=None,
1836 b2caps=None, heads=None, **kwargs):
1837 b2caps=None, heads=None, **kwargs):
1837 """add phase heads part to the requested bundle"""
1838 """add phase heads part to the requested bundle"""
1838 if kwargs.get(r'phases', False):
1839 if kwargs.get(r'phases', False):
1839 if not 'heads' in b2caps.get('phases'):
1840 if not 'heads' in b2caps.get('phases'):
1840 raise ValueError(_('no common phases exchange method'))
1841 raise ValueError(_('no common phases exchange method'))
1841 if heads is None:
1842 if heads is None:
1842 heads = repo.heads()
1843 heads = repo.heads()
1843
1844
1844 headsbyphase = collections.defaultdict(set)
1845 headsbyphase = collections.defaultdict(set)
1845 if repo.publishing():
1846 if repo.publishing():
1846 headsbyphase[phases.public] = heads
1847 headsbyphase[phases.public] = heads
1847 else:
1848 else:
1848 # find the appropriate heads to move
1849 # find the appropriate heads to move
1849
1850
1850 phase = repo._phasecache.phase
1851 phase = repo._phasecache.phase
1851 node = repo.changelog.node
1852 node = repo.changelog.node
1852 rev = repo.changelog.rev
1853 rev = repo.changelog.rev
1853 for h in heads:
1854 for h in heads:
1854 headsbyphase[phase(repo, rev(h))].add(h)
1855 headsbyphase[phase(repo, rev(h))].add(h)
1855 seenphases = list(headsbyphase.keys())
1856 seenphases = list(headsbyphase.keys())
1856
1857
1857 # We do not handle anything but public and draft phase for now)
1858 # We do not handle anything but public and draft phase for now)
1858 if seenphases:
1859 if seenphases:
1859 assert max(seenphases) <= phases.draft
1860 assert max(seenphases) <= phases.draft
1860
1861
1861 # if client is pulling non-public changesets, we need to find
1862 # if client is pulling non-public changesets, we need to find
1862 # intermediate public heads.
1863 # intermediate public heads.
1863 draftheads = headsbyphase.get(phases.draft, set())
1864 draftheads = headsbyphase.get(phases.draft, set())
1864 if draftheads:
1865 if draftheads:
1865 publicheads = headsbyphase.get(phases.public, set())
1866 publicheads = headsbyphase.get(phases.public, set())
1866
1867
1867 revset = 'heads(only(%ln, %ln) and public())'
1868 revset = 'heads(only(%ln, %ln) and public())'
1868 extraheads = repo.revs(revset, draftheads, publicheads)
1869 extraheads = repo.revs(revset, draftheads, publicheads)
1869 for r in extraheads:
1870 for r in extraheads:
1870 headsbyphase[phases.public].add(node(r))
1871 headsbyphase[phases.public].add(node(r))
1871
1872
1872 # transform data in a format used by the encoding function
1873 # transform data in a format used by the encoding function
1873 phasemapping = []
1874 phasemapping = []
1874 for phase in phases.allphases:
1875 for phase in phases.allphases:
1875 phasemapping.append(sorted(headsbyphase[phase]))
1876 phasemapping.append(sorted(headsbyphase[phase]))
1876
1877
1877 # generate the actual part
1878 # generate the actual part
1878 phasedata = phases.binaryencode(phasemapping)
1879 phasedata = phases.binaryencode(phasemapping)
1879 bundler.newpart('phase-heads', data=phasedata)
1880 bundler.newpart('phase-heads', data=phasedata)
1880
1881
1881 @getbundle2partsgenerator('hgtagsfnodes')
1882 @getbundle2partsgenerator('hgtagsfnodes')
1882 def _getbundletagsfnodes(bundler, repo, source, bundlecaps=None,
1883 def _getbundletagsfnodes(bundler, repo, source, bundlecaps=None,
1883 b2caps=None, heads=None, common=None,
1884 b2caps=None, heads=None, common=None,
1884 **kwargs):
1885 **kwargs):
1885 """Transfer the .hgtags filenodes mapping.
1886 """Transfer the .hgtags filenodes mapping.
1886
1887
1887 Only values for heads in this bundle will be transferred.
1888 Only values for heads in this bundle will be transferred.
1888
1889
1889 The part data consists of pairs of 20 byte changeset node and .hgtags
1890 The part data consists of pairs of 20 byte changeset node and .hgtags
1890 filenodes raw values.
1891 filenodes raw values.
1891 """
1892 """
1892 # Don't send unless:
1893 # Don't send unless:
1893 # - changeset are being exchanged,
1894 # - changeset are being exchanged,
1894 # - the client supports it.
1895 # - the client supports it.
1895 if not (kwargs.get(r'cg', True) and 'hgtagsfnodes' in b2caps):
1896 if not (kwargs.get(r'cg', True) and 'hgtagsfnodes' in b2caps):
1896 return
1897 return
1897
1898
1898 outgoing = _computeoutgoing(repo, heads, common)
1899 outgoing = _computeoutgoing(repo, heads, common)
1899 bundle2.addparttagsfnodescache(repo, bundler, outgoing)
1900 bundle2.addparttagsfnodescache(repo, bundler, outgoing)
1900
1901
1901 def check_heads(repo, their_heads, context):
1902 def check_heads(repo, their_heads, context):
1902 """check if the heads of a repo have been modified
1903 """check if the heads of a repo have been modified
1903
1904
1904 Used by peer for unbundling.
1905 Used by peer for unbundling.
1905 """
1906 """
1906 heads = repo.heads()
1907 heads = repo.heads()
1907 heads_hash = hashlib.sha1(''.join(sorted(heads))).digest()
1908 heads_hash = hashlib.sha1(''.join(sorted(heads))).digest()
1908 if not (their_heads == ['force'] or their_heads == heads or
1909 if not (their_heads == ['force'] or their_heads == heads or
1909 their_heads == ['hashed', heads_hash]):
1910 their_heads == ['hashed', heads_hash]):
1910 # someone else committed/pushed/unbundled while we
1911 # someone else committed/pushed/unbundled while we
1911 # were transferring data
1912 # were transferring data
1912 raise error.PushRaced('repository changed while %s - '
1913 raise error.PushRaced('repository changed while %s - '
1913 'please try again' % context)
1914 'please try again' % context)
1914
1915
1915 def unbundle(repo, cg, heads, source, url):
1916 def unbundle(repo, cg, heads, source, url):
1916 """Apply a bundle to a repo.
1917 """Apply a bundle to a repo.
1917
1918
1918 this function makes sure the repo is locked during the application and have
1919 this function makes sure the repo is locked during the application and have
1919 mechanism to check that no push race occurred between the creation of the
1920 mechanism to check that no push race occurred between the creation of the
1920 bundle and its application.
1921 bundle and its application.
1921
1922
1922 If the push was raced as PushRaced exception is raised."""
1923 If the push was raced as PushRaced exception is raised."""
1923 r = 0
1924 r = 0
1924 # need a transaction when processing a bundle2 stream
1925 # need a transaction when processing a bundle2 stream
1925 # [wlock, lock, tr] - needs to be an array so nested functions can modify it
1926 # [wlock, lock, tr] - needs to be an array so nested functions can modify it
1926 lockandtr = [None, None, None]
1927 lockandtr = [None, None, None]
1927 recordout = None
1928 recordout = None
1928 # quick fix for output mismatch with bundle2 in 3.4
1929 # quick fix for output mismatch with bundle2 in 3.4
1929 captureoutput = repo.ui.configbool('experimental', 'bundle2-output-capture')
1930 captureoutput = repo.ui.configbool('experimental', 'bundle2-output-capture')
1930 if url.startswith('remote:http:') or url.startswith('remote:https:'):
1931 if url.startswith('remote:http:') or url.startswith('remote:https:'):
1931 captureoutput = True
1932 captureoutput = True
1932 try:
1933 try:
1933 # note: outside bundle1, 'heads' is expected to be empty and this
1934 # note: outside bundle1, 'heads' is expected to be empty and this
1934 # 'check_heads' call wil be a no-op
1935 # 'check_heads' call wil be a no-op
1935 check_heads(repo, heads, 'uploading changes')
1936 check_heads(repo, heads, 'uploading changes')
1936 # push can proceed
1937 # push can proceed
1937 if not isinstance(cg, bundle2.unbundle20):
1938 if not isinstance(cg, bundle2.unbundle20):
1938 # legacy case: bundle1 (changegroup 01)
1939 # legacy case: bundle1 (changegroup 01)
1939 txnname = "\n".join([source, util.hidepassword(url)])
1940 txnname = "\n".join([source, util.hidepassword(url)])
1940 with repo.lock(), repo.transaction(txnname) as tr:
1941 with repo.lock(), repo.transaction(txnname) as tr:
1941 op = bundle2.applybundle(repo, cg, tr, source, url)
1942 op = bundle2.applybundle(repo, cg, tr, source, url)
1942 r = bundle2.combinechangegroupresults(op)
1943 r = bundle2.combinechangegroupresults(op)
1943 else:
1944 else:
1944 r = None
1945 r = None
1945 try:
1946 try:
1946 def gettransaction():
1947 def gettransaction():
1947 if not lockandtr[2]:
1948 if not lockandtr[2]:
1948 lockandtr[0] = repo.wlock()
1949 lockandtr[0] = repo.wlock()
1949 lockandtr[1] = repo.lock()
1950 lockandtr[1] = repo.lock()
1950 lockandtr[2] = repo.transaction(source)
1951 lockandtr[2] = repo.transaction(source)
1951 lockandtr[2].hookargs['source'] = source
1952 lockandtr[2].hookargs['source'] = source
1952 lockandtr[2].hookargs['url'] = url
1953 lockandtr[2].hookargs['url'] = url
1953 lockandtr[2].hookargs['bundle2'] = '1'
1954 lockandtr[2].hookargs['bundle2'] = '1'
1954 return lockandtr[2]
1955 return lockandtr[2]
1955
1956
1956 # Do greedy locking by default until we're satisfied with lazy
1957 # Do greedy locking by default until we're satisfied with lazy
1957 # locking.
1958 # locking.
1958 if not repo.ui.configbool('experimental', 'bundle2lazylocking'):
1959 if not repo.ui.configbool('experimental', 'bundle2lazylocking'):
1959 gettransaction()
1960 gettransaction()
1960
1961
1961 op = bundle2.bundleoperation(repo, gettransaction,
1962 op = bundle2.bundleoperation(repo, gettransaction,
1962 captureoutput=captureoutput)
1963 captureoutput=captureoutput)
1963 try:
1964 try:
1964 op = bundle2.processbundle(repo, cg, op=op)
1965 op = bundle2.processbundle(repo, cg, op=op)
1965 finally:
1966 finally:
1966 r = op.reply
1967 r = op.reply
1967 if captureoutput and r is not None:
1968 if captureoutput and r is not None:
1968 repo.ui.pushbuffer(error=True, subproc=True)
1969 repo.ui.pushbuffer(error=True, subproc=True)
1969 def recordout(output):
1970 def recordout(output):
1970 r.newpart('output', data=output, mandatory=False)
1971 r.newpart('output', data=output, mandatory=False)
1971 if lockandtr[2] is not None:
1972 if lockandtr[2] is not None:
1972 lockandtr[2].close()
1973 lockandtr[2].close()
1973 except BaseException as exc:
1974 except BaseException as exc:
1974 exc.duringunbundle2 = True
1975 exc.duringunbundle2 = True
1975 if captureoutput and r is not None:
1976 if captureoutput and r is not None:
1976 parts = exc._bundle2salvagedoutput = r.salvageoutput()
1977 parts = exc._bundle2salvagedoutput = r.salvageoutput()
1977 def recordout(output):
1978 def recordout(output):
1978 part = bundle2.bundlepart('output', data=output,
1979 part = bundle2.bundlepart('output', data=output,
1979 mandatory=False)
1980 mandatory=False)
1980 parts.append(part)
1981 parts.append(part)
1981 raise
1982 raise
1982 finally:
1983 finally:
1983 lockmod.release(lockandtr[2], lockandtr[1], lockandtr[0])
1984 lockmod.release(lockandtr[2], lockandtr[1], lockandtr[0])
1984 if recordout is not None:
1985 if recordout is not None:
1985 recordout(repo.ui.popbuffer())
1986 recordout(repo.ui.popbuffer())
1986 return r
1987 return r
1987
1988
1988 def _maybeapplyclonebundle(pullop):
1989 def _maybeapplyclonebundle(pullop):
1989 """Apply a clone bundle from a remote, if possible."""
1990 """Apply a clone bundle from a remote, if possible."""
1990
1991
1991 repo = pullop.repo
1992 repo = pullop.repo
1992 remote = pullop.remote
1993 remote = pullop.remote
1993
1994
1994 if not repo.ui.configbool('ui', 'clonebundles'):
1995 if not repo.ui.configbool('ui', 'clonebundles'):
1995 return
1996 return
1996
1997
1997 # Only run if local repo is empty.
1998 # Only run if local repo is empty.
1998 if len(repo):
1999 if len(repo):
1999 return
2000 return
2000
2001
2001 if pullop.heads:
2002 if pullop.heads:
2002 return
2003 return
2003
2004
2004 if not remote.capable('clonebundles'):
2005 if not remote.capable('clonebundles'):
2005 return
2006 return
2006
2007
2007 res = remote._call('clonebundles')
2008 res = remote._call('clonebundles')
2008
2009
2009 # If we call the wire protocol command, that's good enough to record the
2010 # If we call the wire protocol command, that's good enough to record the
2010 # attempt.
2011 # attempt.
2011 pullop.clonebundleattempted = True
2012 pullop.clonebundleattempted = True
2012
2013
2013 entries = parseclonebundlesmanifest(repo, res)
2014 entries = parseclonebundlesmanifest(repo, res)
2014 if not entries:
2015 if not entries:
2015 repo.ui.note(_('no clone bundles available on remote; '
2016 repo.ui.note(_('no clone bundles available on remote; '
2016 'falling back to regular clone\n'))
2017 'falling back to regular clone\n'))
2017 return
2018 return
2018
2019
2019 entries = filterclonebundleentries(
2020 entries = filterclonebundleentries(
2020 repo, entries, streamclonerequested=pullop.streamclonerequested)
2021 repo, entries, streamclonerequested=pullop.streamclonerequested)
2021
2022
2022 if not entries:
2023 if not entries:
2023 # There is a thundering herd concern here. However, if a server
2024 # There is a thundering herd concern here. However, if a server
2024 # operator doesn't advertise bundles appropriate for its clients,
2025 # operator doesn't advertise bundles appropriate for its clients,
2025 # they deserve what's coming. Furthermore, from a client's
2026 # they deserve what's coming. Furthermore, from a client's
2026 # perspective, no automatic fallback would mean not being able to
2027 # perspective, no automatic fallback would mean not being able to
2027 # clone!
2028 # clone!
2028 repo.ui.warn(_('no compatible clone bundles available on server; '
2029 repo.ui.warn(_('no compatible clone bundles available on server; '
2029 'falling back to regular clone\n'))
2030 'falling back to regular clone\n'))
2030 repo.ui.warn(_('(you may want to report this to the server '
2031 repo.ui.warn(_('(you may want to report this to the server '
2031 'operator)\n'))
2032 'operator)\n'))
2032 return
2033 return
2033
2034
2034 entries = sortclonebundleentries(repo.ui, entries)
2035 entries = sortclonebundleentries(repo.ui, entries)
2035
2036
2036 url = entries[0]['URL']
2037 url = entries[0]['URL']
2037 repo.ui.status(_('applying clone bundle from %s\n') % url)
2038 repo.ui.status(_('applying clone bundle from %s\n') % url)
2038 if trypullbundlefromurl(repo.ui, repo, url):
2039 if trypullbundlefromurl(repo.ui, repo, url):
2039 repo.ui.status(_('finished applying clone bundle\n'))
2040 repo.ui.status(_('finished applying clone bundle\n'))
2040 # Bundle failed.
2041 # Bundle failed.
2041 #
2042 #
2042 # We abort by default to avoid the thundering herd of
2043 # We abort by default to avoid the thundering herd of
2043 # clients flooding a server that was expecting expensive
2044 # clients flooding a server that was expecting expensive
2044 # clone load to be offloaded.
2045 # clone load to be offloaded.
2045 elif repo.ui.configbool('ui', 'clonebundlefallback'):
2046 elif repo.ui.configbool('ui', 'clonebundlefallback'):
2046 repo.ui.warn(_('falling back to normal clone\n'))
2047 repo.ui.warn(_('falling back to normal clone\n'))
2047 else:
2048 else:
2048 raise error.Abort(_('error applying bundle'),
2049 raise error.Abort(_('error applying bundle'),
2049 hint=_('if this error persists, consider contacting '
2050 hint=_('if this error persists, consider contacting '
2050 'the server operator or disable clone '
2051 'the server operator or disable clone '
2051 'bundles via '
2052 'bundles via '
2052 '"--config ui.clonebundles=false"'))
2053 '"--config ui.clonebundles=false"'))
2053
2054
2054 def parseclonebundlesmanifest(repo, s):
2055 def parseclonebundlesmanifest(repo, s):
2055 """Parses the raw text of a clone bundles manifest.
2056 """Parses the raw text of a clone bundles manifest.
2056
2057
2057 Returns a list of dicts. The dicts have a ``URL`` key corresponding
2058 Returns a list of dicts. The dicts have a ``URL`` key corresponding
2058 to the URL and other keys are the attributes for the entry.
2059 to the URL and other keys are the attributes for the entry.
2059 """
2060 """
2060 m = []
2061 m = []
2061 for line in s.splitlines():
2062 for line in s.splitlines():
2062 fields = line.split()
2063 fields = line.split()
2063 if not fields:
2064 if not fields:
2064 continue
2065 continue
2065 attrs = {'URL': fields[0]}
2066 attrs = {'URL': fields[0]}
2066 for rawattr in fields[1:]:
2067 for rawattr in fields[1:]:
2067 key, value = rawattr.split('=', 1)
2068 key, value = rawattr.split('=', 1)
2068 key = urlreq.unquote(key)
2069 key = urlreq.unquote(key)
2069 value = urlreq.unquote(value)
2070 value = urlreq.unquote(value)
2070 attrs[key] = value
2071 attrs[key] = value
2071
2072
2072 # Parse BUNDLESPEC into components. This makes client-side
2073 # Parse BUNDLESPEC into components. This makes client-side
2073 # preferences easier to specify since you can prefer a single
2074 # preferences easier to specify since you can prefer a single
2074 # component of the BUNDLESPEC.
2075 # component of the BUNDLESPEC.
2075 if key == 'BUNDLESPEC':
2076 if key == 'BUNDLESPEC':
2076 try:
2077 try:
2077 comp, version, params = parsebundlespec(repo, value,
2078 comp, version, params = parsebundlespec(repo, value,
2078 externalnames=True)
2079 externalnames=True)
2079 attrs['COMPRESSION'] = comp
2080 attrs['COMPRESSION'] = comp
2080 attrs['VERSION'] = version
2081 attrs['VERSION'] = version
2081 except error.InvalidBundleSpecification:
2082 except error.InvalidBundleSpecification:
2082 pass
2083 pass
2083 except error.UnsupportedBundleSpecification:
2084 except error.UnsupportedBundleSpecification:
2084 pass
2085 pass
2085
2086
2086 m.append(attrs)
2087 m.append(attrs)
2087
2088
2088 return m
2089 return m
2089
2090
2090 def filterclonebundleentries(repo, entries, streamclonerequested=False):
2091 def filterclonebundleentries(repo, entries, streamclonerequested=False):
2091 """Remove incompatible clone bundle manifest entries.
2092 """Remove incompatible clone bundle manifest entries.
2092
2093
2093 Accepts a list of entries parsed with ``parseclonebundlesmanifest``
2094 Accepts a list of entries parsed with ``parseclonebundlesmanifest``
2094 and returns a new list consisting of only the entries that this client
2095 and returns a new list consisting of only the entries that this client
2095 should be able to apply.
2096 should be able to apply.
2096
2097
2097 There is no guarantee we'll be able to apply all returned entries because
2098 There is no guarantee we'll be able to apply all returned entries because
2098 the metadata we use to filter on may be missing or wrong.
2099 the metadata we use to filter on may be missing or wrong.
2099 """
2100 """
2100 newentries = []
2101 newentries = []
2101 for entry in entries:
2102 for entry in entries:
2102 spec = entry.get('BUNDLESPEC')
2103 spec = entry.get('BUNDLESPEC')
2103 if spec:
2104 if spec:
2104 try:
2105 try:
2105 comp, version, params = parsebundlespec(repo, spec, strict=True)
2106 comp, version, params = parsebundlespec(repo, spec, strict=True)
2106
2107
2107 # If a stream clone was requested, filter out non-streamclone
2108 # If a stream clone was requested, filter out non-streamclone
2108 # entries.
2109 # entries.
2109 if streamclonerequested and (comp != 'UN' or version != 's1'):
2110 if streamclonerequested and (comp != 'UN' or version != 's1'):
2110 repo.ui.debug('filtering %s because not a stream clone\n' %
2111 repo.ui.debug('filtering %s because not a stream clone\n' %
2111 entry['URL'])
2112 entry['URL'])
2112 continue
2113 continue
2113
2114
2114 except error.InvalidBundleSpecification as e:
2115 except error.InvalidBundleSpecification as e:
2115 repo.ui.debug(str(e) + '\n')
2116 repo.ui.debug(str(e) + '\n')
2116 continue
2117 continue
2117 except error.UnsupportedBundleSpecification as e:
2118 except error.UnsupportedBundleSpecification as e:
2118 repo.ui.debug('filtering %s because unsupported bundle '
2119 repo.ui.debug('filtering %s because unsupported bundle '
2119 'spec: %s\n' % (entry['URL'], str(e)))
2120 'spec: %s\n' % (entry['URL'], str(e)))
2120 continue
2121 continue
2121 # If we don't have a spec and requested a stream clone, we don't know
2122 # If we don't have a spec and requested a stream clone, we don't know
2122 # what the entry is so don't attempt to apply it.
2123 # what the entry is so don't attempt to apply it.
2123 elif streamclonerequested:
2124 elif streamclonerequested:
2124 repo.ui.debug('filtering %s because cannot determine if a stream '
2125 repo.ui.debug('filtering %s because cannot determine if a stream '
2125 'clone bundle\n' % entry['URL'])
2126 'clone bundle\n' % entry['URL'])
2126 continue
2127 continue
2127
2128
2128 if 'REQUIRESNI' in entry and not sslutil.hassni:
2129 if 'REQUIRESNI' in entry and not sslutil.hassni:
2129 repo.ui.debug('filtering %s because SNI not supported\n' %
2130 repo.ui.debug('filtering %s because SNI not supported\n' %
2130 entry['URL'])
2131 entry['URL'])
2131 continue
2132 continue
2132
2133
2133 newentries.append(entry)
2134 newentries.append(entry)
2134
2135
2135 return newentries
2136 return newentries
2136
2137
2137 class clonebundleentry(object):
2138 class clonebundleentry(object):
2138 """Represents an item in a clone bundles manifest.
2139 """Represents an item in a clone bundles manifest.
2139
2140
2140 This rich class is needed to support sorting since sorted() in Python 3
2141 This rich class is needed to support sorting since sorted() in Python 3
2141 doesn't support ``cmp`` and our comparison is complex enough that ``key=``
2142 doesn't support ``cmp`` and our comparison is complex enough that ``key=``
2142 won't work.
2143 won't work.
2143 """
2144 """
2144
2145
2145 def __init__(self, value, prefers):
2146 def __init__(self, value, prefers):
2146 self.value = value
2147 self.value = value
2147 self.prefers = prefers
2148 self.prefers = prefers
2148
2149
2149 def _cmp(self, other):
2150 def _cmp(self, other):
2150 for prefkey, prefvalue in self.prefers:
2151 for prefkey, prefvalue in self.prefers:
2151 avalue = self.value.get(prefkey)
2152 avalue = self.value.get(prefkey)
2152 bvalue = other.value.get(prefkey)
2153 bvalue = other.value.get(prefkey)
2153
2154
2154 # Special case for b missing attribute and a matches exactly.
2155 # Special case for b missing attribute and a matches exactly.
2155 if avalue is not None and bvalue is None and avalue == prefvalue:
2156 if avalue is not None and bvalue is None and avalue == prefvalue:
2156 return -1
2157 return -1
2157
2158
2158 # Special case for a missing attribute and b matches exactly.
2159 # Special case for a missing attribute and b matches exactly.
2159 if bvalue is not None and avalue is None and bvalue == prefvalue:
2160 if bvalue is not None and avalue is None and bvalue == prefvalue:
2160 return 1
2161 return 1
2161
2162
2162 # We can't compare unless attribute present on both.
2163 # We can't compare unless attribute present on both.
2163 if avalue is None or bvalue is None:
2164 if avalue is None or bvalue is None:
2164 continue
2165 continue
2165
2166
2166 # Same values should fall back to next attribute.
2167 # Same values should fall back to next attribute.
2167 if avalue == bvalue:
2168 if avalue == bvalue:
2168 continue
2169 continue
2169
2170
2170 # Exact matches come first.
2171 # Exact matches come first.
2171 if avalue == prefvalue:
2172 if avalue == prefvalue:
2172 return -1
2173 return -1
2173 if bvalue == prefvalue:
2174 if bvalue == prefvalue:
2174 return 1
2175 return 1
2175
2176
2176 # Fall back to next attribute.
2177 # Fall back to next attribute.
2177 continue
2178 continue
2178
2179
2179 # If we got here we couldn't sort by attributes and prefers. Fall
2180 # If we got here we couldn't sort by attributes and prefers. Fall
2180 # back to index order.
2181 # back to index order.
2181 return 0
2182 return 0
2182
2183
2183 def __lt__(self, other):
2184 def __lt__(self, other):
2184 return self._cmp(other) < 0
2185 return self._cmp(other) < 0
2185
2186
2186 def __gt__(self, other):
2187 def __gt__(self, other):
2187 return self._cmp(other) > 0
2188 return self._cmp(other) > 0
2188
2189
2189 def __eq__(self, other):
2190 def __eq__(self, other):
2190 return self._cmp(other) == 0
2191 return self._cmp(other) == 0
2191
2192
2192 def __le__(self, other):
2193 def __le__(self, other):
2193 return self._cmp(other) <= 0
2194 return self._cmp(other) <= 0
2194
2195
2195 def __ge__(self, other):
2196 def __ge__(self, other):
2196 return self._cmp(other) >= 0
2197 return self._cmp(other) >= 0
2197
2198
2198 def __ne__(self, other):
2199 def __ne__(self, other):
2199 return self._cmp(other) != 0
2200 return self._cmp(other) != 0
2200
2201
2201 def sortclonebundleentries(ui, entries):
2202 def sortclonebundleentries(ui, entries):
2202 prefers = ui.configlist('ui', 'clonebundleprefers')
2203 prefers = ui.configlist('ui', 'clonebundleprefers')
2203 if not prefers:
2204 if not prefers:
2204 return list(entries)
2205 return list(entries)
2205
2206
2206 prefers = [p.split('=', 1) for p in prefers]
2207 prefers = [p.split('=', 1) for p in prefers]
2207
2208
2208 items = sorted(clonebundleentry(v, prefers) for v in entries)
2209 items = sorted(clonebundleentry(v, prefers) for v in entries)
2209 return [i.value for i in items]
2210 return [i.value for i in items]
2210
2211
2211 def trypullbundlefromurl(ui, repo, url):
2212 def trypullbundlefromurl(ui, repo, url):
2212 """Attempt to apply a bundle from a URL."""
2213 """Attempt to apply a bundle from a URL."""
2213 with repo.lock(), repo.transaction('bundleurl') as tr:
2214 with repo.lock(), repo.transaction('bundleurl') as tr:
2214 try:
2215 try:
2215 fh = urlmod.open(ui, url)
2216 fh = urlmod.open(ui, url)
2216 cg = readbundle(ui, fh, 'stream')
2217 cg = readbundle(ui, fh, 'stream')
2217
2218
2218 if isinstance(cg, streamclone.streamcloneapplier):
2219 if isinstance(cg, streamclone.streamcloneapplier):
2219 cg.apply(repo)
2220 cg.apply(repo)
2220 else:
2221 else:
2221 bundle2.applybundle(repo, cg, tr, 'clonebundles', url)
2222 bundle2.applybundle(repo, cg, tr, 'clonebundles', url)
2222 return True
2223 return True
2223 except urlerr.httperror as e:
2224 except urlerr.httperror as e:
2224 ui.warn(_('HTTP error fetching bundle: %s\n') % str(e))
2225 ui.warn(_('HTTP error fetching bundle: %s\n') % str(e))
2225 except urlerr.urlerror as e:
2226 except urlerr.urlerror as e:
2226 ui.warn(_('error fetching bundle: %s\n') % e.reason)
2227 ui.warn(_('error fetching bundle: %s\n') % e.reason)
2227
2228
2228 return False
2229 return False
General Comments 0
You need to be logged in to leave comments. Login now