|
@@
-1,2164
+1,2173
b''
|
|
1
|
1
|
# localrepo.py - read/write repository class for mercurial
|
|
2
|
2
|
#
|
|
3
|
3
|
# Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
|
|
4
|
4
|
#
|
|
5
|
5
|
# This software may be used and distributed according to the terms of the
|
|
6
|
6
|
# GNU General Public License version 2, incorporated herein by reference.
|
|
7
|
7
|
|
|
8
|
8
|
from node import bin, hex, nullid, nullrev, short
|
|
9
|
9
|
from i18n import _
|
|
10
|
10
|
import repo, changegroup, subrepo
|
|
11
|
11
|
import changelog, dirstate, filelog, manifest, context
|
|
12
|
12
|
import lock, transaction, store, encoding
|
|
13
|
13
|
import util, extensions, hook, error
|
|
14
|
14
|
import match as match_
|
|
15
|
15
|
import merge as merge_
|
|
16
|
16
|
import tags as tags_
|
|
17
|
17
|
from lock import release
|
|
18
|
18
|
import weakref, stat, errno, os, time, inspect
|
|
19
|
19
|
propertycache = util.propertycache
|
|
20
|
20
|
|
|
21
|
21
|
class localrepository(repo.repository):
|
|
22
|
22
|
capabilities = set(('lookup', 'changegroupsubset', 'branchmap'))
|
|
23
|
23
|
supported = set('revlogv1 store fncache shared'.split())
|
|
24
|
24
|
|
|
25
|
25
|
def __init__(self, baseui, path=None, create=0):
|
|
26
|
26
|
repo.repository.__init__(self)
|
|
27
|
27
|
self.root = os.path.realpath(path)
|
|
28
|
28
|
self.path = os.path.join(self.root, ".hg")
|
|
29
|
29
|
self.origroot = path
|
|
30
|
30
|
self.opener = util.opener(self.path)
|
|
31
|
31
|
self.wopener = util.opener(self.root)
|
|
32
|
32
|
self.baseui = baseui
|
|
33
|
33
|
self.ui = baseui.copy()
|
|
34
|
34
|
|
|
35
|
35
|
try:
|
|
36
|
36
|
self.ui.readconfig(self.join("hgrc"), self.root)
|
|
37
|
37
|
extensions.loadall(self.ui)
|
|
38
|
38
|
except IOError:
|
|
39
|
39
|
pass
|
|
40
|
40
|
|
|
41
|
41
|
if not os.path.isdir(self.path):
|
|
42
|
42
|
if create:
|
|
43
|
43
|
if not os.path.exists(path):
|
|
44
|
44
|
os.mkdir(path)
|
|
45
|
45
|
os.mkdir(self.path)
|
|
46
|
46
|
requirements = ["revlogv1"]
|
|
47
|
47
|
if self.ui.configbool('format', 'usestore', True):
|
|
48
|
48
|
os.mkdir(os.path.join(self.path, "store"))
|
|
49
|
49
|
requirements.append("store")
|
|
50
|
50
|
if self.ui.configbool('format', 'usefncache', True):
|
|
51
|
51
|
requirements.append("fncache")
|
|
52
|
52
|
# create an invalid changelog
|
|
53
|
53
|
self.opener("00changelog.i", "a").write(
|
|
54
|
54
|
'\0\0\0\2' # represents revlogv2
|
|
55
|
55
|
' dummy changelog to prevent using the old repo layout'
|
|
56
|
56
|
)
|
|
57
|
57
|
reqfile = self.opener("requires", "w")
|
|
58
|
58
|
for r in requirements:
|
|
59
|
59
|
reqfile.write("%s\n" % r)
|
|
60
|
60
|
reqfile.close()
|
|
61
|
61
|
else:
|
|
62
|
62
|
raise error.RepoError(_("repository %s not found") % path)
|
|
63
|
63
|
elif create:
|
|
64
|
64
|
raise error.RepoError(_("repository %s already exists") % path)
|
|
65
|
65
|
else:
|
|
66
|
66
|
# find requirements
|
|
67
|
67
|
requirements = set()
|
|
68
|
68
|
try:
|
|
69
|
69
|
requirements = set(self.opener("requires").read().splitlines())
|
|
70
|
70
|
except IOError, inst:
|
|
71
|
71
|
if inst.errno != errno.ENOENT:
|
|
72
|
72
|
raise
|
|
73
|
73
|
for r in requirements - self.supported:
|
|
74
|
74
|
raise error.RepoError(_("requirement '%s' not supported") % r)
|
|
75
|
75
|
|
|
76
|
76
|
self.sharedpath = self.path
|
|
77
|
77
|
try:
|
|
78
|
78
|
s = os.path.realpath(self.opener("sharedpath").read())
|
|
79
|
79
|
if not os.path.exists(s):
|
|
80
|
80
|
raise error.RepoError(
|
|
81
|
81
|
_('.hg/sharedpath points to nonexistent directory %s') % s)
|
|
82
|
82
|
self.sharedpath = s
|
|
83
|
83
|
except IOError, inst:
|
|
84
|
84
|
if inst.errno != errno.ENOENT:
|
|
85
|
85
|
raise
|
|
86
|
86
|
|
|
87
|
87
|
self.store = store.store(requirements, self.sharedpath, util.opener)
|
|
88
|
88
|
self.spath = self.store.path
|
|
89
|
89
|
self.sopener = self.store.opener
|
|
90
|
90
|
self.sjoin = self.store.join
|
|
91
|
91
|
self.opener.createmode = self.store.createmode
|
|
92
|
92
|
|
|
93
|
93
|
# These two define the set of tags for this repository. _tags
|
|
94
|
94
|
# maps tag name to node; _tagtypes maps tag name to 'global' or
|
|
95
|
95
|
# 'local'. (Global tags are defined by .hgtags across all
|
|
96
|
96
|
# heads, and local tags are defined in .hg/localtags.) They
|
|
97
|
97
|
# constitute the in-memory cache of tags.
|
|
98
|
98
|
self._tags = None
|
|
99
|
99
|
self._tagtypes = None
|
|
100
|
100
|
|
|
101
|
101
|
self._branchcache = None # in UTF-8
|
|
102
|
102
|
self._branchcachetip = None
|
|
103
|
103
|
self.nodetagscache = None
|
|
104
|
104
|
self.filterpats = {}
|
|
105
|
105
|
self._datafilters = {}
|
|
106
|
106
|
self._transref = self._lockref = self._wlockref = None
|
|
107
|
107
|
|
|
108
|
108
|
@propertycache
|
|
109
|
109
|
def changelog(self):
|
|
110
|
110
|
c = changelog.changelog(self.sopener)
|
|
111
|
111
|
if 'HG_PENDING' in os.environ:
|
|
112
|
112
|
p = os.environ['HG_PENDING']
|
|
113
|
113
|
if p.startswith(self.root):
|
|
114
|
114
|
c.readpending('00changelog.i.a')
|
|
115
|
115
|
self.sopener.defversion = c.version
|
|
116
|
116
|
return c
|
|
117
|
117
|
|
|
118
|
118
|
@propertycache
|
|
119
|
119
|
def manifest(self):
|
|
120
|
120
|
return manifest.manifest(self.sopener)
|
|
121
|
121
|
|
|
122
|
122
|
@propertycache
|
|
123
|
123
|
def dirstate(self):
|
|
124
|
124
|
return dirstate.dirstate(self.opener, self.ui, self.root)
|
|
125
|
125
|
|
|
126
|
126
|
def __getitem__(self, changeid):
|
|
127
|
127
|
if changeid is None:
|
|
128
|
128
|
return context.workingctx(self)
|
|
129
|
129
|
return context.changectx(self, changeid)
|
|
130
|
130
|
|
|
131
|
131
|
def __contains__(self, changeid):
|
|
132
|
132
|
try:
|
|
133
|
133
|
return bool(self.lookup(changeid))
|
|
134
|
134
|
except error.RepoLookupError:
|
|
135
|
135
|
return False
|
|
136
|
136
|
|
|
137
|
137
|
def __nonzero__(self):
|
|
138
|
138
|
return True
|
|
139
|
139
|
|
|
140
|
140
|
def __len__(self):
|
|
141
|
141
|
return len(self.changelog)
|
|
142
|
142
|
|
|
143
|
143
|
def __iter__(self):
|
|
144
|
144
|
for i in xrange(len(self)):
|
|
145
|
145
|
yield i
|
|
146
|
146
|
|
|
147
|
147
|
def url(self):
|
|
148
|
148
|
return 'file:' + self.root
|
|
149
|
149
|
|
|
150
|
150
|
def hook(self, name, throw=False, **args):
|
|
151
|
151
|
return hook.hook(self.ui, self, name, throw, **args)
|
|
152
|
152
|
|
|
153
|
153
|
tag_disallowed = ':\r\n'
|
|
154
|
154
|
|
|
155
|
155
|
def _tag(self, names, node, message, local, user, date, extra={}):
|
|
156
|
156
|
if isinstance(names, str):
|
|
157
|
157
|
allchars = names
|
|
158
|
158
|
names = (names,)
|
|
159
|
159
|
else:
|
|
160
|
160
|
allchars = ''.join(names)
|
|
161
|
161
|
for c in self.tag_disallowed:
|
|
162
|
162
|
if c in allchars:
|
|
163
|
163
|
raise util.Abort(_('%r cannot be used in a tag name') % c)
|
|
164
|
164
|
|
|
165
|
165
|
for name in names:
|
|
166
|
166
|
self.hook('pretag', throw=True, node=hex(node), tag=name,
|
|
167
|
167
|
local=local)
|
|
168
|
168
|
|
|
169
|
169
|
def writetags(fp, names, munge, prevtags):
|
|
170
|
170
|
fp.seek(0, 2)
|
|
171
|
171
|
if prevtags and prevtags[-1] != '\n':
|
|
172
|
172
|
fp.write('\n')
|
|
173
|
173
|
for name in names:
|
|
174
|
174
|
m = munge and munge(name) or name
|
|
175
|
175
|
if self._tagtypes and name in self._tagtypes:
|
|
176
|
176
|
old = self._tags.get(name, nullid)
|
|
177
|
177
|
fp.write('%s %s\n' % (hex(old), m))
|
|
178
|
178
|
fp.write('%s %s\n' % (hex(node), m))
|
|
179
|
179
|
fp.close()
|
|
180
|
180
|
|
|
181
|
181
|
prevtags = ''
|
|
182
|
182
|
if local:
|
|
183
|
183
|
try:
|
|
184
|
184
|
fp = self.opener('localtags', 'r+')
|
|
185
|
185
|
except IOError:
|
|
186
|
186
|
fp = self.opener('localtags', 'a')
|
|
187
|
187
|
else:
|
|
188
|
188
|
prevtags = fp.read()
|
|
189
|
189
|
|
|
190
|
190
|
# local tags are stored in the current charset
|
|
191
|
191
|
writetags(fp, names, None, prevtags)
|
|
192
|
192
|
for name in names:
|
|
193
|
193
|
self.hook('tag', node=hex(node), tag=name, local=local)
|
|
194
|
194
|
return
|
|
195
|
195
|
|
|
196
|
196
|
try:
|
|
197
|
197
|
fp = self.wfile('.hgtags', 'rb+')
|
|
198
|
198
|
except IOError:
|
|
199
|
199
|
fp = self.wfile('.hgtags', 'ab')
|
|
200
|
200
|
else:
|
|
201
|
201
|
prevtags = fp.read()
|
|
202
|
202
|
|
|
203
|
203
|
# committed tags are stored in UTF-8
|
|
204
|
204
|
writetags(fp, names, encoding.fromlocal, prevtags)
|
|
205
|
205
|
|
|
206
|
206
|
if '.hgtags' not in self.dirstate:
|
|
207
|
207
|
self.add(['.hgtags'])
|
|
208
|
208
|
|
|
209
|
209
|
m = match_.exact(self.root, '', ['.hgtags'])
|
|
210
|
210
|
tagnode = self.commit(message, user, date, extra=extra, match=m)
|
|
211
|
211
|
|
|
212
|
212
|
for name in names:
|
|
213
|
213
|
self.hook('tag', node=hex(node), tag=name, local=local)
|
|
214
|
214
|
|
|
215
|
215
|
return tagnode
|
|
216
|
216
|
|
|
217
|
217
|
def tag(self, names, node, message, local, user, date):
|
|
218
|
218
|
'''tag a revision with one or more symbolic names.
|
|
219
|
219
|
|
|
220
|
220
|
names is a list of strings or, when adding a single tag, names may be a
|
|
221
|
221
|
string.
|
|
222
|
222
|
|
|
223
|
223
|
if local is True, the tags are stored in a per-repository file.
|
|
224
|
224
|
otherwise, they are stored in the .hgtags file, and a new
|
|
225
|
225
|
changeset is committed with the change.
|
|
226
|
226
|
|
|
227
|
227
|
keyword arguments:
|
|
228
|
228
|
|
|
229
|
229
|
local: whether to store tags in non-version-controlled file
|
|
230
|
230
|
(default False)
|
|
231
|
231
|
|
|
232
|
232
|
message: commit message to use if committing
|
|
233
|
233
|
|
|
234
|
234
|
user: name of user to use if committing
|
|
235
|
235
|
|
|
236
|
236
|
date: date tuple to use if committing'''
|
|
237
|
237
|
|
|
238
|
238
|
for x in self.status()[:5]:
|
|
239
|
239
|
if '.hgtags' in x:
|
|
240
|
240
|
raise util.Abort(_('working copy of .hgtags is changed '
|
|
241
|
241
|
'(please commit .hgtags manually)'))
|
|
242
|
242
|
|
|
243
|
243
|
self.tags() # instantiate the cache
|
|
244
|
244
|
self._tag(names, node, message, local, user, date)
|
|
245
|
245
|
|
|
246
|
246
|
def tags(self):
|
|
247
|
247
|
'''return a mapping of tag to node'''
|
|
248
|
248
|
if self._tags is None:
|
|
249
|
249
|
(self._tags, self._tagtypes) = self._findtags()
|
|
250
|
250
|
|
|
251
|
251
|
return self._tags
|
|
252
|
252
|
|
|
253
|
253
|
def _findtags(self):
|
|
254
|
254
|
'''Do the hard work of finding tags. Return a pair of dicts
|
|
255
|
255
|
(tags, tagtypes) where tags maps tag name to node, and tagtypes
|
|
256
|
256
|
maps tag name to a string like \'global\' or \'local\'.
|
|
257
|
257
|
Subclasses or extensions are free to add their own tags, but
|
|
258
|
258
|
should be aware that the returned dicts will be retained for the
|
|
259
|
259
|
duration of the localrepo object.'''
|
|
260
|
260
|
|
|
261
|
261
|
# XXX what tagtype should subclasses/extensions use? Currently
|
|
262
|
262
|
# mq and bookmarks add tags, but do not set the tagtype at all.
|
|
263
|
263
|
# Should each extension invent its own tag type? Should there
|
|
264
|
264
|
# be one tagtype for all such "virtual" tags? Or is the status
|
|
265
|
265
|
# quo fine?
|
|
266
|
266
|
|
|
267
|
267
|
alltags = {} # map tag name to (node, hist)
|
|
268
|
268
|
tagtypes = {}
|
|
269
|
269
|
|
|
270
|
270
|
tags_.findglobaltags(self.ui, self, alltags, tagtypes)
|
|
271
|
271
|
tags_.readlocaltags(self.ui, self, alltags, tagtypes)
|
|
272
|
272
|
|
|
273
|
273
|
# Build the return dicts. Have to re-encode tag names because
|
|
274
|
274
|
# the tags module always uses UTF-8 (in order not to lose info
|
|
275
|
275
|
# writing to the cache), but the rest of Mercurial wants them in
|
|
276
|
276
|
# local encoding.
|
|
277
|
277
|
tags = {}
|
|
278
|
278
|
for (name, (node, hist)) in alltags.iteritems():
|
|
279
|
279
|
if node != nullid:
|
|
280
|
280
|
tags[encoding.tolocal(name)] = node
|
|
281
|
281
|
tags['tip'] = self.changelog.tip()
|
|
282
|
282
|
tagtypes = dict([(encoding.tolocal(name), value)
|
|
283
|
283
|
for (name, value) in tagtypes.iteritems()])
|
|
284
|
284
|
return (tags, tagtypes)
|
|
285
|
285
|
|
|
286
|
286
|
def tagtype(self, tagname):
|
|
287
|
287
|
'''
|
|
288
|
288
|
return the type of the given tag. result can be:
|
|
289
|
289
|
|
|
290
|
290
|
'local' : a local tag
|
|
291
|
291
|
'global' : a global tag
|
|
292
|
292
|
None : tag does not exist
|
|
293
|
293
|
'''
|
|
294
|
294
|
|
|
295
|
295
|
self.tags()
|
|
296
|
296
|
|
|
297
|
297
|
return self._tagtypes.get(tagname)
|
|
298
|
298
|
|
|
299
|
299
|
def tagslist(self):
|
|
300
|
300
|
'''return a list of tags ordered by revision'''
|
|
301
|
301
|
l = []
|
|
302
|
302
|
for t, n in self.tags().iteritems():
|
|
303
|
303
|
try:
|
|
304
|
304
|
r = self.changelog.rev(n)
|
|
305
|
305
|
except:
|
|
306
|
306
|
r = -2 # sort to the beginning of the list if unknown
|
|
307
|
307
|
l.append((r, t, n))
|
|
308
|
308
|
return [(t, n) for r, t, n in sorted(l)]
|
|
309
|
309
|
|
|
310
|
310
|
def nodetags(self, node):
|
|
311
|
311
|
'''return the tags associated with a node'''
|
|
312
|
312
|
if not self.nodetagscache:
|
|
313
|
313
|
self.nodetagscache = {}
|
|
314
|
314
|
for t, n in self.tags().iteritems():
|
|
315
|
315
|
self.nodetagscache.setdefault(n, []).append(t)
|
|
316
|
316
|
return self.nodetagscache.get(node, [])
|
|
317
|
317
|
|
|
318
|
318
|
def _branchtags(self, partial, lrev):
|
|
319
|
319
|
# TODO: rename this function?
|
|
320
|
320
|
tiprev = len(self) - 1
|
|
321
|
321
|
if lrev != tiprev:
|
|
322
|
322
|
self._updatebranchcache(partial, lrev+1, tiprev+1)
|
|
323
|
323
|
self._writebranchcache(partial, self.changelog.tip(), tiprev)
|
|
324
|
324
|
|
|
325
|
325
|
return partial
|
|
326
|
326
|
|
|
327
|
327
|
def branchmap(self):
|
|
328
|
328
|
tip = self.changelog.tip()
|
|
329
|
329
|
if self._branchcache is not None and self._branchcachetip == tip:
|
|
330
|
330
|
return self._branchcache
|
|
331
|
331
|
|
|
332
|
332
|
oldtip = self._branchcachetip
|
|
333
|
333
|
self._branchcachetip = tip
|
|
334
|
334
|
if oldtip is None or oldtip not in self.changelog.nodemap:
|
|
335
|
335
|
partial, last, lrev = self._readbranchcache()
|
|
336
|
336
|
else:
|
|
337
|
337
|
lrev = self.changelog.rev(oldtip)
|
|
338
|
338
|
partial = self._branchcache
|
|
339
|
339
|
|
|
340
|
340
|
self._branchtags(partial, lrev)
|
|
341
|
341
|
# this private cache holds all heads (not just tips)
|
|
342
|
342
|
self._branchcache = partial
|
|
343
|
343
|
|
|
344
|
344
|
return self._branchcache
|
|
345
|
345
|
|
|
346
|
346
|
def branchtags(self):
|
|
347
|
347
|
'''return a dict where branch names map to the tipmost head of
|
|
348
|
348
|
the branch, open heads come before closed'''
|
|
349
|
349
|
bt = {}
|
|
350
|
350
|
for bn, heads in self.branchmap().iteritems():
|
|
351
|
351
|
head = None
|
|
352
|
352
|
for i in range(len(heads)-1, -1, -1):
|
|
353
|
353
|
h = heads[i]
|
|
354
|
354
|
if 'close' not in self.changelog.read(h)[5]:
|
|
355
|
355
|
head = h
|
|
356
|
356
|
break
|
|
357
|
357
|
# no open heads were found
|
|
358
|
358
|
if head is None:
|
|
359
|
359
|
head = heads[-1]
|
|
360
|
360
|
bt[bn] = head
|
|
361
|
361
|
return bt
|
|
362
|
362
|
|
|
363
|
363
|
|
|
364
|
364
|
def _readbranchcache(self):
|
|
365
|
365
|
partial = {}
|
|
366
|
366
|
try:
|
|
367
|
367
|
f = self.opener("branchheads.cache")
|
|
368
|
368
|
lines = f.read().split('\n')
|
|
369
|
369
|
f.close()
|
|
370
|
370
|
except (IOError, OSError):
|
|
371
|
371
|
return {}, nullid, nullrev
|
|
372
|
372
|
|
|
373
|
373
|
try:
|
|
374
|
374
|
last, lrev = lines.pop(0).split(" ", 1)
|
|
375
|
375
|
last, lrev = bin(last), int(lrev)
|
|
376
|
376
|
if lrev >= len(self) or self[lrev].node() != last:
|
|
377
|
377
|
# invalidate the cache
|
|
378
|
378
|
raise ValueError('invalidating branch cache (tip differs)')
|
|
379
|
379
|
for l in lines:
|
|
380
|
380
|
if not l: continue
|
|
381
|
381
|
node, label = l.split(" ", 1)
|
|
382
|
382
|
partial.setdefault(label.strip(), []).append(bin(node))
|
|
383
|
383
|
except KeyboardInterrupt:
|
|
384
|
384
|
raise
|
|
385
|
385
|
except Exception, inst:
|
|
386
|
386
|
if self.ui.debugflag:
|
|
387
|
387
|
self.ui.warn(str(inst), '\n')
|
|
388
|
388
|
partial, last, lrev = {}, nullid, nullrev
|
|
389
|
389
|
return partial, last, lrev
|
|
390
|
390
|
|
|
391
|
391
|
def _writebranchcache(self, branches, tip, tiprev):
|
|
392
|
392
|
try:
|
|
393
|
393
|
f = self.opener("branchheads.cache", "w", atomictemp=True)
|
|
394
|
394
|
f.write("%s %s\n" % (hex(tip), tiprev))
|
|
395
|
395
|
for label, nodes in branches.iteritems():
|
|
396
|
396
|
for node in nodes:
|
|
397
|
397
|
f.write("%s %s\n" % (hex(node), label))
|
|
398
|
398
|
f.rename()
|
|
399
|
399
|
except (IOError, OSError):
|
|
400
|
400
|
pass
|
|
401
|
401
|
|
|
402
|
402
|
def _updatebranchcache(self, partial, start, end):
|
|
403
|
403
|
# collect new branch entries
|
|
404
|
404
|
newbranches = {}
|
|
405
|
405
|
for r in xrange(start, end):
|
|
406
|
406
|
c = self[r]
|
|
407
|
407
|
newbranches.setdefault(c.branch(), []).append(c.node())
|
|
408
|
408
|
# if older branchheads are reachable from new ones, they aren't
|
|
409
|
409
|
# really branchheads. Note checking parents is insufficient:
|
|
410
|
410
|
# 1 (branch a) -> 2 (branch b) -> 3 (branch a)
|
|
411
|
411
|
for branch, newnodes in newbranches.iteritems():
|
|
412
|
412
|
bheads = partial.setdefault(branch, [])
|
|
413
|
413
|
bheads.extend(newnodes)
|
|
414
|
414
|
if len(bheads) < 2:
|
|
415
|
415
|
continue
|
|
416
|
416
|
newbheads = []
|
|
417
|
417
|
# starting from tip means fewer passes over reachable
|
|
418
|
418
|
while newnodes:
|
|
419
|
419
|
latest = newnodes.pop()
|
|
420
|
420
|
if latest not in bheads:
|
|
421
|
421
|
continue
|
|
422
|
422
|
minbhrev = self[min([self[bh].rev() for bh in bheads])].node()
|
|
423
|
423
|
reachable = self.changelog.reachable(latest, minbhrev)
|
|
424
|
424
|
bheads = [b for b in bheads if b not in reachable]
|
|
425
|
425
|
newbheads.insert(0, latest)
|
|
426
|
426
|
bheads.extend(newbheads)
|
|
427
|
427
|
partial[branch] = bheads
|
|
428
|
428
|
|
|
429
|
429
|
def lookup(self, key):
|
|
430
|
430
|
if isinstance(key, int):
|
|
431
|
431
|
return self.changelog.node(key)
|
|
432
|
432
|
elif key == '.':
|
|
433
|
433
|
return self.dirstate.parents()[0]
|
|
434
|
434
|
elif key == 'null':
|
|
435
|
435
|
return nullid
|
|
436
|
436
|
elif key == 'tip':
|
|
437
|
437
|
return self.changelog.tip()
|
|
438
|
438
|
n = self.changelog._match(key)
|
|
439
|
439
|
if n:
|
|
440
|
440
|
return n
|
|
441
|
441
|
if key in self.tags():
|
|
442
|
442
|
return self.tags()[key]
|
|
443
|
443
|
if key in self.branchtags():
|
|
444
|
444
|
return self.branchtags()[key]
|
|
445
|
445
|
n = self.changelog._partialmatch(key)
|
|
446
|
446
|
if n:
|
|
447
|
447
|
return n
|
|
448
|
448
|
|
|
449
|
449
|
# can't find key, check if it might have come from damaged dirstate
|
|
450
|
450
|
if key in self.dirstate.parents():
|
|
451
|
451
|
raise error.Abort(_("working directory has unknown parent '%s'!")
|
|
452
|
452
|
% short(key))
|
|
453
|
453
|
try:
|
|
454
|
454
|
if len(key) == 20:
|
|
455
|
455
|
key = hex(key)
|
|
456
|
456
|
except:
|
|
457
|
457
|
pass
|
|
458
|
458
|
raise error.RepoLookupError(_("unknown revision '%s'") % key)
|
|
459
|
459
|
|
|
460
|
460
|
def local(self):
|
|
461
|
461
|
return True
|
|
462
|
462
|
|
|
463
|
463
|
def join(self, f):
|
|
464
|
464
|
return os.path.join(self.path, f)
|
|
465
|
465
|
|
|
466
|
466
|
def wjoin(self, f):
|
|
467
|
467
|
return os.path.join(self.root, f)
|
|
468
|
468
|
|
|
469
|
469
|
def rjoin(self, f):
|
|
470
|
470
|
return os.path.join(self.root, util.pconvert(f))
|
|
471
|
471
|
|
|
472
|
472
|
def file(self, f):
|
|
473
|
473
|
if f[0] == '/':
|
|
474
|
474
|
f = f[1:]
|
|
475
|
475
|
return filelog.filelog(self.sopener, f)
|
|
476
|
476
|
|
|
477
|
477
|
def changectx(self, changeid):
|
|
478
|
478
|
return self[changeid]
|
|
479
|
479
|
|
|
480
|
480
|
def parents(self, changeid=None):
|
|
481
|
481
|
'''get list of changectxs for parents of changeid'''
|
|
482
|
482
|
return self[changeid].parents()
|
|
483
|
483
|
|
|
484
|
484
|
def filectx(self, path, changeid=None, fileid=None):
|
|
485
|
485
|
"""changeid can be a changeset revision, node, or tag.
|
|
486
|
486
|
fileid can be a file revision or node."""
|
|
487
|
487
|
return context.filectx(self, path, changeid, fileid)
|
|
488
|
488
|
|
|
489
|
489
|
def getcwd(self):
|
|
490
|
490
|
return self.dirstate.getcwd()
|
|
491
|
491
|
|
|
492
|
492
|
def pathto(self, f, cwd=None):
|
|
493
|
493
|
return self.dirstate.pathto(f, cwd)
|
|
494
|
494
|
|
|
495
|
495
|
def wfile(self, f, mode='r'):
|
|
496
|
496
|
return self.wopener(f, mode)
|
|
497
|
497
|
|
|
498
|
498
|
def _link(self, f):
|
|
499
|
499
|
return os.path.islink(self.wjoin(f))
|
|
500
|
500
|
|
|
501
|
501
|
def _filter(self, filter, filename, data):
|
|
502
|
502
|
if filter not in self.filterpats:
|
|
503
|
503
|
l = []
|
|
504
|
504
|
for pat, cmd in self.ui.configitems(filter):
|
|
505
|
505
|
if cmd == '!':
|
|
506
|
506
|
continue
|
|
507
|
507
|
mf = match_.match(self.root, '', [pat])
|
|
508
|
508
|
fn = None
|
|
509
|
509
|
params = cmd
|
|
510
|
510
|
for name, filterfn in self._datafilters.iteritems():
|
|
511
|
511
|
if cmd.startswith(name):
|
|
512
|
512
|
fn = filterfn
|
|
513
|
513
|
params = cmd[len(name):].lstrip()
|
|
514
|
514
|
break
|
|
515
|
515
|
if not fn:
|
|
516
|
516
|
fn = lambda s, c, **kwargs: util.filter(s, c)
|
|
517
|
517
|
# Wrap old filters not supporting keyword arguments
|
|
518
|
518
|
if not inspect.getargspec(fn)[2]:
|
|
519
|
519
|
oldfn = fn
|
|
520
|
520
|
fn = lambda s, c, **kwargs: oldfn(s, c)
|
|
521
|
521
|
l.append((mf, fn, params))
|
|
522
|
522
|
self.filterpats[filter] = l
|
|
523
|
523
|
|
|
524
|
524
|
for mf, fn, cmd in self.filterpats[filter]:
|
|
525
|
525
|
if mf(filename):
|
|
526
|
526
|
self.ui.debug("filtering %s through %s\n" % (filename, cmd))
|
|
527
|
527
|
data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
|
|
528
|
528
|
break
|
|
529
|
529
|
|
|
530
|
530
|
return data
|
|
531
|
531
|
|
|
532
|
532
|
def adddatafilter(self, name, filter):
|
|
533
|
533
|
self._datafilters[name] = filter
|
|
534
|
534
|
|
|
535
|
535
|
def wread(self, filename):
|
|
536
|
536
|
if self._link(filename):
|
|
537
|
537
|
data = os.readlink(self.wjoin(filename))
|
|
538
|
538
|
else:
|
|
539
|
539
|
data = self.wopener(filename, 'r').read()
|
|
540
|
540
|
return self._filter("encode", filename, data)
|
|
541
|
541
|
|
|
542
|
542
|
def wwrite(self, filename, data, flags):
|
|
543
|
543
|
data = self._filter("decode", filename, data)
|
|
544
|
544
|
try:
|
|
545
|
545
|
os.unlink(self.wjoin(filename))
|
|
546
|
546
|
except OSError:
|
|
547
|
547
|
pass
|
|
548
|
548
|
if 'l' in flags:
|
|
549
|
549
|
self.wopener.symlink(data, filename)
|
|
550
|
550
|
else:
|
|
551
|
551
|
self.wopener(filename, 'w').write(data)
|
|
552
|
552
|
if 'x' in flags:
|
|
553
|
553
|
util.set_flags(self.wjoin(filename), False, True)
|
|
554
|
554
|
|
|
555
|
555
|
def wwritedata(self, filename, data):
|
|
556
|
556
|
return self._filter("decode", filename, data)
|
|
557
|
557
|
|
|
558
|
558
|
def transaction(self):
|
|
559
|
559
|
tr = self._transref and self._transref() or None
|
|
560
|
560
|
if tr and tr.running():
|
|
561
|
561
|
return tr.nest()
|
|
562
|
562
|
|
|
563
|
563
|
# abort here if the journal already exists
|
|
564
|
564
|
if os.path.exists(self.sjoin("journal")):
|
|
565
|
565
|
raise error.RepoError(_("abandoned transaction found - run hg recover"))
|
|
566
|
566
|
|
|
567
|
567
|
# save dirstate for rollback
|
|
568
|
568
|
try:
|
|
569
|
569
|
ds = self.opener("dirstate").read()
|
|
570
|
570
|
except IOError:
|
|
571
|
571
|
ds = ""
|
|
572
|
572
|
self.opener("journal.dirstate", "w").write(ds)
|
|
573
|
573
|
self.opener("journal.branch", "w").write(self.dirstate.branch())
|
|
574
|
574
|
|
|
575
|
575
|
renames = [(self.sjoin("journal"), self.sjoin("undo")),
|
|
576
|
576
|
(self.join("journal.dirstate"), self.join("undo.dirstate")),
|
|
577
|
577
|
(self.join("journal.branch"), self.join("undo.branch"))]
|
|
578
|
578
|
tr = transaction.transaction(self.ui.warn, self.sopener,
|
|
579
|
579
|
self.sjoin("journal"),
|
|
580
|
580
|
aftertrans(renames),
|
|
581
|
581
|
self.store.createmode)
|
|
582
|
582
|
self._transref = weakref.ref(tr)
|
|
583
|
583
|
return tr
|
|
584
|
584
|
|
|
585
|
585
|
def recover(self):
|
|
586
|
586
|
lock = self.lock()
|
|
587
|
587
|
try:
|
|
588
|
588
|
if os.path.exists(self.sjoin("journal")):
|
|
589
|
589
|
self.ui.status(_("rolling back interrupted transaction\n"))
|
|
590
|
590
|
transaction.rollback(self.sopener, self.sjoin("journal"), self.ui.warn)
|
|
591
|
591
|
self.invalidate()
|
|
592
|
592
|
return True
|
|
593
|
593
|
else:
|
|
594
|
594
|
self.ui.warn(_("no interrupted transaction available\n"))
|
|
595
|
595
|
return False
|
|
596
|
596
|
finally:
|
|
597
|
597
|
lock.release()
|
|
598
|
598
|
|
|
599
|
599
|
def rollback(self):
|
|
600
|
600
|
wlock = lock = None
|
|
601
|
601
|
try:
|
|
602
|
602
|
wlock = self.wlock()
|
|
603
|
603
|
lock = self.lock()
|
|
604
|
604
|
if os.path.exists(self.sjoin("undo")):
|
|
605
|
605
|
self.ui.status(_("rolling back last transaction\n"))
|
|
606
|
606
|
transaction.rollback(self.sopener, self.sjoin("undo"), self.ui.warn)
|
|
607
|
607
|
util.rename(self.join("undo.dirstate"), self.join("dirstate"))
|
|
608
|
608
|
try:
|
|
609
|
609
|
branch = self.opener("undo.branch").read()
|
|
610
|
610
|
self.dirstate.setbranch(branch)
|
|
611
|
611
|
except IOError:
|
|
612
|
612
|
self.ui.warn(_("Named branch could not be reset, "
|
|
613
|
613
|
"current branch still is: %s\n")
|
|
614
|
614
|
% encoding.tolocal(self.dirstate.branch()))
|
|
615
|
615
|
self.invalidate()
|
|
616
|
616
|
self.dirstate.invalidate()
|
|
617
|
617
|
self.destroyed()
|
|
618
|
618
|
else:
|
|
619
|
619
|
self.ui.warn(_("no rollback information available\n"))
|
|
620
|
620
|
finally:
|
|
621
|
621
|
release(lock, wlock)
|
|
622
|
622
|
|
|
623
|
623
|
def invalidate(self):
|
|
624
|
624
|
for a in "changelog manifest".split():
|
|
625
|
625
|
if a in self.__dict__:
|
|
626
|
626
|
delattr(self, a)
|
|
627
|
627
|
self._tags = None
|
|
628
|
628
|
self._tagtypes = None
|
|
629
|
629
|
self.nodetagscache = None
|
|
630
|
630
|
self._branchcache = None # in UTF-8
|
|
631
|
631
|
self._branchcachetip = None
|
|
632
|
632
|
|
|
633
|
633
|
def _lock(self, lockname, wait, releasefn, acquirefn, desc):
|
|
634
|
634
|
try:
|
|
635
|
635
|
l = lock.lock(lockname, 0, releasefn, desc=desc)
|
|
636
|
636
|
except error.LockHeld, inst:
|
|
637
|
637
|
if not wait:
|
|
638
|
638
|
raise
|
|
639
|
639
|
self.ui.warn(_("waiting for lock on %s held by %r\n") %
|
|
640
|
640
|
(desc, inst.locker))
|
|
641
|
641
|
# default to 600 seconds timeout
|
|
642
|
642
|
l = lock.lock(lockname, int(self.ui.config("ui", "timeout", "600")),
|
|
643
|
643
|
releasefn, desc=desc)
|
|
644
|
644
|
if acquirefn:
|
|
645
|
645
|
acquirefn()
|
|
646
|
646
|
return l
|
|
647
|
647
|
|
|
648
|
648
|
def lock(self, wait=True):
|
|
649
|
649
|
'''Lock the repository store (.hg/store) and return a weak reference
|
|
650
|
650
|
to the lock. Use this before modifying the store (e.g. committing or
|
|
651
|
651
|
stripping). If you are opening a transaction, get a lock as well.)'''
|
|
652
|
652
|
l = self._lockref and self._lockref()
|
|
653
|
653
|
if l is not None and l.held:
|
|
654
|
654
|
l.lock()
|
|
655
|
655
|
return l
|
|
656
|
656
|
|
|
657
|
657
|
l = self._lock(self.sjoin("lock"), wait, None, self.invalidate,
|
|
658
|
658
|
_('repository %s') % self.origroot)
|
|
659
|
659
|
self._lockref = weakref.ref(l)
|
|
660
|
660
|
return l
|
|
661
|
661
|
|
|
662
|
662
|
def wlock(self, wait=True):
|
|
663
|
663
|
'''Lock the non-store parts of the repository (everything under
|
|
664
|
664
|
.hg except .hg/store) and return a weak reference to the lock.
|
|
665
|
665
|
Use this before modifying files in .hg.'''
|
|
666
|
666
|
l = self._wlockref and self._wlockref()
|
|
667
|
667
|
if l is not None and l.held:
|
|
668
|
668
|
l.lock()
|
|
669
|
669
|
return l
|
|
670
|
670
|
|
|
671
|
671
|
l = self._lock(self.join("wlock"), wait, self.dirstate.write,
|
|
672
|
672
|
self.dirstate.invalidate, _('working directory of %s') %
|
|
673
|
673
|
self.origroot)
|
|
674
|
674
|
self._wlockref = weakref.ref(l)
|
|
675
|
675
|
return l
|
|
676
|
676
|
|
|
677
|
677
|
def _filecommit(self, fctx, manifest1, manifest2, linkrev, tr, changelist):
|
|
678
|
678
|
"""
|
|
679
|
679
|
commit an individual file as part of a larger transaction
|
|
680
|
680
|
"""
|
|
681
|
681
|
|
|
682
|
682
|
fname = fctx.path()
|
|
683
|
683
|
text = fctx.data()
|
|
684
|
684
|
flog = self.file(fname)
|
|
685
|
685
|
fparent1 = manifest1.get(fname, nullid)
|
|
686
|
686
|
fparent2 = fparent2o = manifest2.get(fname, nullid)
|
|
687
|
687
|
|
|
688
|
688
|
meta = {}
|
|
689
|
689
|
copy = fctx.renamed()
|
|
690
|
690
|
if copy and copy[0] != fname:
|
|
691
|
691
|
# Mark the new revision of this file as a copy of another
|
|
692
|
692
|
# file. This copy data will effectively act as a parent
|
|
693
|
693
|
# of this new revision. If this is a merge, the first
|
|
694
|
694
|
# parent will be the nullid (meaning "look up the copy data")
|
|
695
|
695
|
# and the second one will be the other parent. For example:
|
|
696
|
696
|
#
|
|
697
|
697
|
# 0 --- 1 --- 3 rev1 changes file foo
|
|
698
|
698
|
# \ / rev2 renames foo to bar and changes it
|
|
699
|
699
|
# \- 2 -/ rev3 should have bar with all changes and
|
|
700
|
700
|
# should record that bar descends from
|
|
701
|
701
|
# bar in rev2 and foo in rev1
|
|
702
|
702
|
#
|
|
703
|
703
|
# this allows this merge to succeed:
|
|
704
|
704
|
#
|
|
705
|
705
|
# 0 --- 1 --- 3 rev4 reverts the content change from rev2
|
|
706
|
706
|
# \ / merging rev3 and rev4 should use bar@rev2
|
|
707
|
707
|
# \- 2 --- 4 as the merge base
|
|
708
|
708
|
#
|
|
709
|
709
|
|
|
710
|
710
|
cfname = copy[0]
|
|
711
|
711
|
crev = manifest1.get(cfname)
|
|
712
|
712
|
newfparent = fparent2
|
|
713
|
713
|
|
|
714
|
714
|
if manifest2: # branch merge
|
|
715
|
715
|
if fparent2 == nullid or crev is None: # copied on remote side
|
|
716
|
716
|
if cfname in manifest2:
|
|
717
|
717
|
crev = manifest2[cfname]
|
|
718
|
718
|
newfparent = fparent1
|
|
719
|
719
|
|
|
720
|
720
|
# find source in nearest ancestor if we've lost track
|
|
721
|
721
|
if not crev:
|
|
722
|
722
|
self.ui.debug(" %s: searching for copy revision for %s\n" %
|
|
723
|
723
|
(fname, cfname))
|
|
724
|
724
|
for ancestor in self['.'].ancestors():
|
|
725
|
725
|
if cfname in ancestor:
|
|
726
|
726
|
crev = ancestor[cfname].filenode()
|
|
727
|
727
|
break
|
|
728
|
728
|
|
|
729
|
729
|
self.ui.debug(" %s: copy %s:%s\n" % (fname, cfname, hex(crev)))
|
|
730
|
730
|
meta["copy"] = cfname
|
|
731
|
731
|
meta["copyrev"] = hex(crev)
|
|
732
|
732
|
fparent1, fparent2 = nullid, newfparent
|
|
733
|
733
|
elif fparent2 != nullid:
|
|
734
|
734
|
# is one parent an ancestor of the other?
|
|
735
|
735
|
fparentancestor = flog.ancestor(fparent1, fparent2)
|
|
736
|
736
|
if fparentancestor == fparent1:
|
|
737
|
737
|
fparent1, fparent2 = fparent2, nullid
|
|
738
|
738
|
elif fparentancestor == fparent2:
|
|
739
|
739
|
fparent2 = nullid
|
|
740
|
740
|
|
|
741
|
741
|
# is the file changed?
|
|
742
|
742
|
if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
|
|
743
|
743
|
changelist.append(fname)
|
|
744
|
744
|
return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
|
|
745
|
745
|
|
|
746
|
746
|
# are just the flags changed during merge?
|
|
747
|
747
|
if fparent1 != fparent2o and manifest1.flags(fname) != fctx.flags():
|
|
748
|
748
|
changelist.append(fname)
|
|
749
|
749
|
|
|
750
|
750
|
return fparent1
|
|
751
|
751
|
|
|
752
|
752
|
def commit(self, text="", user=None, date=None, match=None, force=False,
|
|
753
|
753
|
editor=False, extra={}):
|
|
754
|
754
|
"""Add a new revision to current repository.
|
|
755
|
755
|
|
|
756
|
756
|
Revision information is gathered from the working directory,
|
|
757
|
757
|
match can be used to filter the committed files. If editor is
|
|
758
|
758
|
supplied, it is called to get a commit message.
|
|
759
|
759
|
"""
|
|
760
|
760
|
|
|
761
|
761
|
def fail(f, msg):
|
|
762
|
762
|
raise util.Abort('%s: %s' % (f, msg))
|
|
763
|
763
|
|
|
764
|
764
|
if not match:
|
|
765
|
765
|
match = match_.always(self.root, '')
|
|
766
|
766
|
|
|
767
|
767
|
if not force:
|
|
768
|
768
|
vdirs = []
|
|
769
|
769
|
match.dir = vdirs.append
|
|
770
|
770
|
match.bad = fail
|
|
771
|
771
|
|
|
772
|
772
|
wlock = self.wlock()
|
|
773
|
773
|
try:
|
|
774
|
774
|
p1, p2 = self.dirstate.parents()
|
|
775
|
775
|
wctx = self[None]
|
|
776
|
776
|
|
|
777
|
777
|
if (not force and p2 != nullid and match and
|
|
778
|
778
|
(match.files() or match.anypats())):
|
|
779
|
779
|
raise util.Abort(_('cannot partially commit a merge '
|
|
780
|
780
|
'(do not specify files or patterns)'))
|
|
781
|
781
|
|
|
782
|
782
|
changes = self.status(match=match, clean=force)
|
|
783
|
783
|
if force:
|
|
784
|
784
|
changes[0].extend(changes[6]) # mq may commit unchanged files
|
|
785
|
785
|
|
|
786
|
786
|
# check subrepos
|
|
787
|
787
|
subs = []
|
|
788
|
788
|
for s in wctx.substate:
|
|
789
|
789
|
if match(s) and wctx.sub(s).dirty():
|
|
790
|
790
|
subs.append(s)
|
|
791
|
791
|
if subs and '.hgsubstate' not in changes[0]:
|
|
792
|
792
|
changes[0].insert(0, '.hgsubstate')
|
|
793
|
793
|
|
|
794
|
794
|
# make sure all explicit patterns are matched
|
|
795
|
795
|
if not force and match.files():
|
|
796
|
796
|
matched = set(changes[0] + changes[1] + changes[2])
|
|
797
|
797
|
|
|
798
|
798
|
for f in match.files():
|
|
799
|
799
|
if f == '.' or f in matched or f in wctx.substate:
|
|
800
|
800
|
continue
|
|
801
|
801
|
if f in changes[3]: # missing
|
|
802
|
802
|
fail(f, _('file not found!'))
|
|
803
|
803
|
if f in vdirs: # visited directory
|
|
804
|
804
|
d = f + '/'
|
|
805
|
805
|
for mf in matched:
|
|
806
|
806
|
if mf.startswith(d):
|
|
807
|
807
|
break
|
|
808
|
808
|
else:
|
|
809
|
809
|
fail(f, _("no match under directory!"))
|
|
810
|
810
|
elif f not in self.dirstate:
|
|
811
|
811
|
fail(f, _("file not tracked!"))
|
|
812
|
812
|
|
|
813
|
813
|
if (not force and not extra.get("close") and p2 == nullid
|
|
814
|
814
|
and not (changes[0] or changes[1] or changes[2])
|
|
815
|
815
|
and self[None].branch() == self['.'].branch()):
|
|
816
|
816
|
return None
|
|
817
|
817
|
|
|
818
|
818
|
ms = merge_.mergestate(self)
|
|
819
|
819
|
for f in changes[0]:
|
|
820
|
820
|
if f in ms and ms[f] == 'u':
|
|
821
|
821
|
raise util.Abort(_("unresolved merge conflicts "
|
|
822
|
822
|
"(see hg resolve)"))
|
|
823
|
823
|
|
|
824
|
824
|
cctx = context.workingctx(self, (p1, p2), text, user, date,
|
|
825
|
825
|
extra, changes)
|
|
826
|
826
|
if editor:
|
|
827
|
827
|
cctx._text = editor(self, cctx, subs)
|
|
828
|
828
|
|
|
829
|
829
|
# commit subs
|
|
830
|
830
|
if subs:
|
|
831
|
831
|
state = wctx.substate.copy()
|
|
832
|
832
|
for s in subs:
|
|
833
|
833
|
self.ui.status(_('committing subrepository %s\n') % s)
|
|
834
|
834
|
sr = wctx.sub(s).commit(cctx._text, user, date)
|
|
835
|
835
|
state[s] = (state[s][0], sr)
|
|
836
|
836
|
subrepo.writestate(self, state)
|
|
837
|
837
|
|
|
|
838
|
# Save commit message in case this transaction gets rolled back
|
|
|
839
|
# (e.g. by a pretxncommit hook). (Save in text mode in case a
|
|
|
840
|
# Windows user wants to edit it with Notepad. Normalize
|
|
|
841
|
# trailing whitespace so the file always looks the same --
|
|
|
842
|
# makes testing easier.)
|
|
|
843
|
msgfile = self.opener('last-message.txt', 'w')
|
|
|
844
|
msgfile.write(cctx._text.rstrip() + '\n')
|
|
|
845
|
msgfile.close()
|
|
|
846
|
|
|
838
|
847
|
ret = self.commitctx(cctx, True)
|
|
839
|
848
|
|
|
840
|
849
|
# update dirstate and mergestate
|
|
841
|
850
|
for f in changes[0] + changes[1]:
|
|
842
|
851
|
self.dirstate.normal(f)
|
|
843
|
852
|
for f in changes[2]:
|
|
844
|
853
|
self.dirstate.forget(f)
|
|
845
|
854
|
self.dirstate.setparents(ret)
|
|
846
|
855
|
ms.reset()
|
|
847
|
856
|
|
|
848
|
857
|
return ret
|
|
849
|
858
|
|
|
850
|
859
|
finally:
|
|
851
|
860
|
wlock.release()
|
|
852
|
861
|
|
|
853
|
862
|
def commitctx(self, ctx, error=False):
|
|
854
|
863
|
"""Add a new revision to current repository.
|
|
855
|
864
|
|
|
856
|
865
|
Revision information is passed via the context argument.
|
|
857
|
866
|
"""
|
|
858
|
867
|
|
|
859
|
868
|
tr = lock = None
|
|
860
|
869
|
removed = ctx.removed()
|
|
861
|
870
|
p1, p2 = ctx.p1(), ctx.p2()
|
|
862
|
871
|
m1 = p1.manifest().copy()
|
|
863
|
872
|
m2 = p2.manifest()
|
|
864
|
873
|
user = ctx.user()
|
|
865
|
874
|
|
|
866
|
875
|
xp1, xp2 = p1.hex(), p2 and p2.hex() or ''
|
|
867
|
876
|
self.hook("precommit", throw=True, parent1=xp1, parent2=xp2)
|
|
868
|
877
|
|
|
869
|
878
|
lock = self.lock()
|
|
870
|
879
|
try:
|
|
871
|
880
|
tr = self.transaction()
|
|
872
|
881
|
trp = weakref.proxy(tr)
|
|
873
|
882
|
|
|
874
|
883
|
# check in files
|
|
875
|
884
|
new = {}
|
|
876
|
885
|
changed = []
|
|
877
|
886
|
linkrev = len(self)
|
|
878
|
887
|
for f in sorted(ctx.modified() + ctx.added()):
|
|
879
|
888
|
self.ui.note(f + "\n")
|
|
880
|
889
|
try:
|
|
881
|
890
|
fctx = ctx[f]
|
|
882
|
891
|
new[f] = self._filecommit(fctx, m1, m2, linkrev, trp,
|
|
883
|
892
|
changed)
|
|
884
|
893
|
m1.set(f, fctx.flags())
|
|
885
|
894
|
except (OSError, IOError):
|
|
886
|
895
|
if error:
|
|
887
|
896
|
self.ui.warn(_("trouble committing %s!\n") % f)
|
|
888
|
897
|
raise
|
|
889
|
898
|
else:
|
|
890
|
899
|
removed.append(f)
|
|
891
|
900
|
|
|
892
|
901
|
# update manifest
|
|
893
|
902
|
m1.update(new)
|
|
894
|
903
|
removed = [f for f in sorted(removed) if f in m1 or f in m2]
|
|
895
|
904
|
drop = [f for f in removed if f in m1]
|
|
896
|
905
|
for f in drop:
|
|
897
|
906
|
del m1[f]
|
|
898
|
907
|
mn = self.manifest.add(m1, trp, linkrev, p1.manifestnode(),
|
|
899
|
908
|
p2.manifestnode(), (new, drop))
|
|
900
|
909
|
|
|
901
|
910
|
# update changelog
|
|
902
|
911
|
self.changelog.delayupdate()
|
|
903
|
912
|
n = self.changelog.add(mn, changed + removed, ctx.description(),
|
|
904
|
913
|
trp, p1.node(), p2.node(),
|
|
905
|
914
|
user, ctx.date(), ctx.extra().copy())
|
|
906
|
915
|
p = lambda: self.changelog.writepending() and self.root or ""
|
|
907
|
916
|
self.hook('pretxncommit', throw=True, node=hex(n), parent1=xp1,
|
|
908
|
917
|
parent2=xp2, pending=p)
|
|
909
|
918
|
self.changelog.finalize(trp)
|
|
910
|
919
|
tr.close()
|
|
911
|
920
|
|
|
912
|
921
|
if self._branchcache:
|
|
913
|
922
|
self.branchtags()
|
|
914
|
923
|
|
|
915
|
924
|
self.hook("commit", node=hex(n), parent1=xp1, parent2=xp2)
|
|
916
|
925
|
return n
|
|
917
|
926
|
finally:
|
|
918
|
927
|
del tr
|
|
919
|
928
|
lock.release()
|
|
920
|
929
|
|
|
921
|
930
|
def destroyed(self):
|
|
922
|
931
|
'''Inform the repository that nodes have been destroyed.
|
|
923
|
932
|
Intended for use by strip and rollback, so there's a common
|
|
924
|
933
|
place for anything that has to be done after destroying history.'''
|
|
925
|
934
|
# XXX it might be nice if we could take the list of destroyed
|
|
926
|
935
|
# nodes, but I don't see an easy way for rollback() to do that
|
|
927
|
936
|
|
|
928
|
937
|
# Ensure the persistent tag cache is updated. Doing it now
|
|
929
|
938
|
# means that the tag cache only has to worry about destroyed
|
|
930
|
939
|
# heads immediately after a strip/rollback. That in turn
|
|
931
|
940
|
# guarantees that "cachetip == currenttip" (comparing both rev
|
|
932
|
941
|
# and node) always means no nodes have been added or destroyed.
|
|
933
|
942
|
|
|
934
|
943
|
# XXX this is suboptimal when qrefresh'ing: we strip the current
|
|
935
|
944
|
# head, refresh the tag cache, then immediately add a new head.
|
|
936
|
945
|
# But I think doing it this way is necessary for the "instant
|
|
937
|
946
|
# tag cache retrieval" case to work.
|
|
938
|
947
|
tags_.findglobaltags(self.ui, self, {}, {})
|
|
939
|
948
|
|
|
940
|
949
|
def walk(self, match, node=None):
|
|
941
|
950
|
'''
|
|
942
|
951
|
walk recursively through the directory tree or a given
|
|
943
|
952
|
changeset, finding all files matched by the match
|
|
944
|
953
|
function
|
|
945
|
954
|
'''
|
|
946
|
955
|
return self[node].walk(match)
|
|
947
|
956
|
|
|
948
|
957
|
def status(self, node1='.', node2=None, match=None,
|
|
949
|
958
|
ignored=False, clean=False, unknown=False):
|
|
950
|
959
|
"""return status of files between two nodes or node and working directory
|
|
951
|
960
|
|
|
952
|
961
|
If node1 is None, use the first dirstate parent instead.
|
|
953
|
962
|
If node2 is None, compare node1 with working directory.
|
|
954
|
963
|
"""
|
|
955
|
964
|
|
|
956
|
965
|
def mfmatches(ctx):
|
|
957
|
966
|
mf = ctx.manifest().copy()
|
|
958
|
967
|
for fn in mf.keys():
|
|
959
|
968
|
if not match(fn):
|
|
960
|
969
|
del mf[fn]
|
|
961
|
970
|
return mf
|
|
962
|
971
|
|
|
963
|
972
|
if isinstance(node1, context.changectx):
|
|
964
|
973
|
ctx1 = node1
|
|
965
|
974
|
else:
|
|
966
|
975
|
ctx1 = self[node1]
|
|
967
|
976
|
if isinstance(node2, context.changectx):
|
|
968
|
977
|
ctx2 = node2
|
|
969
|
978
|
else:
|
|
970
|
979
|
ctx2 = self[node2]
|
|
971
|
980
|
|
|
972
|
981
|
working = ctx2.rev() is None
|
|
973
|
982
|
parentworking = working and ctx1 == self['.']
|
|
974
|
983
|
match = match or match_.always(self.root, self.getcwd())
|
|
975
|
984
|
listignored, listclean, listunknown = ignored, clean, unknown
|
|
976
|
985
|
|
|
977
|
986
|
# load earliest manifest first for caching reasons
|
|
978
|
987
|
if not working and ctx2.rev() < ctx1.rev():
|
|
979
|
988
|
ctx2.manifest()
|
|
980
|
989
|
|
|
981
|
990
|
if not parentworking:
|
|
982
|
991
|
def bad(f, msg):
|
|
983
|
992
|
if f not in ctx1:
|
|
984
|
993
|
self.ui.warn('%s: %s\n' % (self.dirstate.pathto(f), msg))
|
|
985
|
994
|
match.bad = bad
|
|
986
|
995
|
|
|
987
|
996
|
if working: # we need to scan the working dir
|
|
988
|
997
|
s = self.dirstate.status(match, listignored, listclean, listunknown)
|
|
989
|
998
|
cmp, modified, added, removed, deleted, unknown, ignored, clean = s
|
|
990
|
999
|
|
|
991
|
1000
|
# check for any possibly clean files
|
|
992
|
1001
|
if parentworking and cmp:
|
|
993
|
1002
|
fixup = []
|
|
994
|
1003
|
# do a full compare of any files that might have changed
|
|
995
|
1004
|
for f in sorted(cmp):
|
|
996
|
1005
|
if (f not in ctx1 or ctx2.flags(f) != ctx1.flags(f)
|
|
997
|
1006
|
or ctx1[f].cmp(ctx2[f].data())):
|
|
998
|
1007
|
modified.append(f)
|
|
999
|
1008
|
else:
|
|
1000
|
1009
|
fixup.append(f)
|
|
1001
|
1010
|
|
|
1002
|
1011
|
if listclean:
|
|
1003
|
1012
|
clean += fixup
|
|
1004
|
1013
|
|
|
1005
|
1014
|
# update dirstate for files that are actually clean
|
|
1006
|
1015
|
if fixup:
|
|
1007
|
1016
|
try:
|
|
1008
|
1017
|
# updating the dirstate is optional
|
|
1009
|
1018
|
# so we don't wait on the lock
|
|
1010
|
1019
|
wlock = self.wlock(False)
|
|
1011
|
1020
|
try:
|
|
1012
|
1021
|
for f in fixup:
|
|
1013
|
1022
|
self.dirstate.normal(f)
|
|
1014
|
1023
|
finally:
|
|
1015
|
1024
|
wlock.release()
|
|
1016
|
1025
|
except error.LockError:
|
|
1017
|
1026
|
pass
|
|
1018
|
1027
|
|
|
1019
|
1028
|
if not parentworking:
|
|
1020
|
1029
|
mf1 = mfmatches(ctx1)
|
|
1021
|
1030
|
if working:
|
|
1022
|
1031
|
# we are comparing working dir against non-parent
|
|
1023
|
1032
|
# generate a pseudo-manifest for the working dir
|
|
1024
|
1033
|
mf2 = mfmatches(self['.'])
|
|
1025
|
1034
|
for f in cmp + modified + added:
|
|
1026
|
1035
|
mf2[f] = None
|
|
1027
|
1036
|
mf2.set(f, ctx2.flags(f))
|
|
1028
|
1037
|
for f in removed:
|
|
1029
|
1038
|
if f in mf2:
|
|
1030
|
1039
|
del mf2[f]
|
|
1031
|
1040
|
else:
|
|
1032
|
1041
|
# we are comparing two revisions
|
|
1033
|
1042
|
deleted, unknown, ignored = [], [], []
|
|
1034
|
1043
|
mf2 = mfmatches(ctx2)
|
|
1035
|
1044
|
|
|
1036
|
1045
|
modified, added, clean = [], [], []
|
|
1037
|
1046
|
for fn in mf2:
|
|
1038
|
1047
|
if fn in mf1:
|
|
1039
|
1048
|
if (mf1.flags(fn) != mf2.flags(fn) or
|
|
1040
|
1049
|
(mf1[fn] != mf2[fn] and
|
|
1041
|
1050
|
(mf2[fn] or ctx1[fn].cmp(ctx2[fn].data())))):
|
|
1042
|
1051
|
modified.append(fn)
|
|
1043
|
1052
|
elif listclean:
|
|
1044
|
1053
|
clean.append(fn)
|
|
1045
|
1054
|
del mf1[fn]
|
|
1046
|
1055
|
else:
|
|
1047
|
1056
|
added.append(fn)
|
|
1048
|
1057
|
removed = mf1.keys()
|
|
1049
|
1058
|
|
|
1050
|
1059
|
r = modified, added, removed, deleted, unknown, ignored, clean
|
|
1051
|
1060
|
[l.sort() for l in r]
|
|
1052
|
1061
|
return r
|
|
1053
|
1062
|
|
|
1054
|
1063
|
def add(self, list):
|
|
1055
|
1064
|
wlock = self.wlock()
|
|
1056
|
1065
|
try:
|
|
1057
|
1066
|
rejected = []
|
|
1058
|
1067
|
for f in list:
|
|
1059
|
1068
|
p = self.wjoin(f)
|
|
1060
|
1069
|
try:
|
|
1061
|
1070
|
st = os.lstat(p)
|
|
1062
|
1071
|
except:
|
|
1063
|
1072
|
self.ui.warn(_("%s does not exist!\n") % f)
|
|
1064
|
1073
|
rejected.append(f)
|
|
1065
|
1074
|
continue
|
|
1066
|
1075
|
if st.st_size > 10000000:
|
|
1067
|
1076
|
self.ui.warn(_("%s: files over 10MB may cause memory and"
|
|
1068
|
1077
|
" performance problems\n"
|
|
1069
|
1078
|
"(use 'hg revert %s' to unadd the file)\n")
|
|
1070
|
1079
|
% (f, f))
|
|
1071
|
1080
|
if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
|
|
1072
|
1081
|
self.ui.warn(_("%s not added: only files and symlinks "
|
|
1073
|
1082
|
"supported currently\n") % f)
|
|
1074
|
1083
|
rejected.append(p)
|
|
1075
|
1084
|
elif self.dirstate[f] in 'amn':
|
|
1076
|
1085
|
self.ui.warn(_("%s already tracked!\n") % f)
|
|
1077
|
1086
|
elif self.dirstate[f] == 'r':
|
|
1078
|
1087
|
self.dirstate.normallookup(f)
|
|
1079
|
1088
|
else:
|
|
1080
|
1089
|
self.dirstate.add(f)
|
|
1081
|
1090
|
return rejected
|
|
1082
|
1091
|
finally:
|
|
1083
|
1092
|
wlock.release()
|
|
1084
|
1093
|
|
|
1085
|
1094
|
def forget(self, list):
|
|
1086
|
1095
|
wlock = self.wlock()
|
|
1087
|
1096
|
try:
|
|
1088
|
1097
|
for f in list:
|
|
1089
|
1098
|
if self.dirstate[f] != 'a':
|
|
1090
|
1099
|
self.ui.warn(_("%s not added!\n") % f)
|
|
1091
|
1100
|
else:
|
|
1092
|
1101
|
self.dirstate.forget(f)
|
|
1093
|
1102
|
finally:
|
|
1094
|
1103
|
wlock.release()
|
|
1095
|
1104
|
|
|
1096
|
1105
|
def remove(self, list, unlink=False):
|
|
1097
|
1106
|
if unlink:
|
|
1098
|
1107
|
for f in list:
|
|
1099
|
1108
|
try:
|
|
1100
|
1109
|
util.unlink(self.wjoin(f))
|
|
1101
|
1110
|
except OSError, inst:
|
|
1102
|
1111
|
if inst.errno != errno.ENOENT:
|
|
1103
|
1112
|
raise
|
|
1104
|
1113
|
wlock = self.wlock()
|
|
1105
|
1114
|
try:
|
|
1106
|
1115
|
for f in list:
|
|
1107
|
1116
|
if unlink and os.path.exists(self.wjoin(f)):
|
|
1108
|
1117
|
self.ui.warn(_("%s still exists!\n") % f)
|
|
1109
|
1118
|
elif self.dirstate[f] == 'a':
|
|
1110
|
1119
|
self.dirstate.forget(f)
|
|
1111
|
1120
|
elif f not in self.dirstate:
|
|
1112
|
1121
|
self.ui.warn(_("%s not tracked!\n") % f)
|
|
1113
|
1122
|
else:
|
|
1114
|
1123
|
self.dirstate.remove(f)
|
|
1115
|
1124
|
finally:
|
|
1116
|
1125
|
wlock.release()
|
|
1117
|
1126
|
|
|
1118
|
1127
|
def undelete(self, list):
|
|
1119
|
1128
|
manifests = [self.manifest.read(self.changelog.read(p)[0])
|
|
1120
|
1129
|
for p in self.dirstate.parents() if p != nullid]
|
|
1121
|
1130
|
wlock = self.wlock()
|
|
1122
|
1131
|
try:
|
|
1123
|
1132
|
for f in list:
|
|
1124
|
1133
|
if self.dirstate[f] != 'r':
|
|
1125
|
1134
|
self.ui.warn(_("%s not removed!\n") % f)
|
|
1126
|
1135
|
else:
|
|
1127
|
1136
|
m = f in manifests[0] and manifests[0] or manifests[1]
|
|
1128
|
1137
|
t = self.file(f).read(m[f])
|
|
1129
|
1138
|
self.wwrite(f, t, m.flags(f))
|
|
1130
|
1139
|
self.dirstate.normal(f)
|
|
1131
|
1140
|
finally:
|
|
1132
|
1141
|
wlock.release()
|
|
1133
|
1142
|
|
|
1134
|
1143
|
def copy(self, source, dest):
|
|
1135
|
1144
|
p = self.wjoin(dest)
|
|
1136
|
1145
|
if not (os.path.exists(p) or os.path.islink(p)):
|
|
1137
|
1146
|
self.ui.warn(_("%s does not exist!\n") % dest)
|
|
1138
|
1147
|
elif not (os.path.isfile(p) or os.path.islink(p)):
|
|
1139
|
1148
|
self.ui.warn(_("copy failed: %s is not a file or a "
|
|
1140
|
1149
|
"symbolic link\n") % dest)
|
|
1141
|
1150
|
else:
|
|
1142
|
1151
|
wlock = self.wlock()
|
|
1143
|
1152
|
try:
|
|
1144
|
1153
|
if self.dirstate[dest] in '?r':
|
|
1145
|
1154
|
self.dirstate.add(dest)
|
|
1146
|
1155
|
self.dirstate.copy(source, dest)
|
|
1147
|
1156
|
finally:
|
|
1148
|
1157
|
wlock.release()
|
|
1149
|
1158
|
|
|
1150
|
1159
|
def heads(self, start=None):
|
|
1151
|
1160
|
heads = self.changelog.heads(start)
|
|
1152
|
1161
|
# sort the output in rev descending order
|
|
1153
|
1162
|
heads = [(-self.changelog.rev(h), h) for h in heads]
|
|
1154
|
1163
|
return [n for (r, n) in sorted(heads)]
|
|
1155
|
1164
|
|
|
1156
|
1165
|
def branchheads(self, branch=None, start=None, closed=False):
|
|
1157
|
1166
|
'''return a (possibly filtered) list of heads for the given branch
|
|
1158
|
1167
|
|
|
1159
|
1168
|
Heads are returned in topological order, from newest to oldest.
|
|
1160
|
1169
|
If branch is None, use the dirstate branch.
|
|
1161
|
1170
|
If start is not None, return only heads reachable from start.
|
|
1162
|
1171
|
If closed is True, return heads that are marked as closed as well.
|
|
1163
|
1172
|
'''
|
|
1164
|
1173
|
if branch is None:
|
|
1165
|
1174
|
branch = self[None].branch()
|
|
1166
|
1175
|
branches = self.branchmap()
|
|
1167
|
1176
|
if branch not in branches:
|
|
1168
|
1177
|
return []
|
|
1169
|
1178
|
# the cache returns heads ordered lowest to highest
|
|
1170
|
1179
|
bheads = list(reversed(branches[branch]))
|
|
1171
|
1180
|
if start is not None:
|
|
1172
|
1181
|
# filter out the heads that cannot be reached from startrev
|
|
1173
|
1182
|
fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
|
|
1174
|
1183
|
bheads = [h for h in bheads if h in fbheads]
|
|
1175
|
1184
|
if not closed:
|
|
1176
|
1185
|
bheads = [h for h in bheads if
|
|
1177
|
1186
|
('close' not in self.changelog.read(h)[5])]
|
|
1178
|
1187
|
return bheads
|
|
1179
|
1188
|
|
|
1180
|
1189
|
def branches(self, nodes):
|
|
1181
|
1190
|
if not nodes:
|
|
1182
|
1191
|
nodes = [self.changelog.tip()]
|
|
1183
|
1192
|
b = []
|
|
1184
|
1193
|
for n in nodes:
|
|
1185
|
1194
|
t = n
|
|
1186
|
1195
|
while 1:
|
|
1187
|
1196
|
p = self.changelog.parents(n)
|
|
1188
|
1197
|
if p[1] != nullid or p[0] == nullid:
|
|
1189
|
1198
|
b.append((t, n, p[0], p[1]))
|
|
1190
|
1199
|
break
|
|
1191
|
1200
|
n = p[0]
|
|
1192
|
1201
|
return b
|
|
1193
|
1202
|
|
|
1194
|
1203
|
def between(self, pairs):
|
|
1195
|
1204
|
r = []
|
|
1196
|
1205
|
|
|
1197
|
1206
|
for top, bottom in pairs:
|
|
1198
|
1207
|
n, l, i = top, [], 0
|
|
1199
|
1208
|
f = 1
|
|
1200
|
1209
|
|
|
1201
|
1210
|
while n != bottom and n != nullid:
|
|
1202
|
1211
|
p = self.changelog.parents(n)[0]
|
|
1203
|
1212
|
if i == f:
|
|
1204
|
1213
|
l.append(n)
|
|
1205
|
1214
|
f = f * 2
|
|
1206
|
1215
|
n = p
|
|
1207
|
1216
|
i += 1
|
|
1208
|
1217
|
|
|
1209
|
1218
|
r.append(l)
|
|
1210
|
1219
|
|
|
1211
|
1220
|
return r
|
|
1212
|
1221
|
|
|
1213
|
1222
|
def findincoming(self, remote, base=None, heads=None, force=False):
|
|
1214
|
1223
|
"""Return list of roots of the subsets of missing nodes from remote
|
|
1215
|
1224
|
|
|
1216
|
1225
|
If base dict is specified, assume that these nodes and their parents
|
|
1217
|
1226
|
exist on the remote side and that no child of a node of base exists
|
|
1218
|
1227
|
in both remote and self.
|
|
1219
|
1228
|
Furthermore base will be updated to include the nodes that exists
|
|
1220
|
1229
|
in self and remote but no children exists in self and remote.
|
|
1221
|
1230
|
If a list of heads is specified, return only nodes which are heads
|
|
1222
|
1231
|
or ancestors of these heads.
|
|
1223
|
1232
|
|
|
1224
|
1233
|
All the ancestors of base are in self and in remote.
|
|
1225
|
1234
|
All the descendants of the list returned are missing in self.
|
|
1226
|
1235
|
(and so we know that the rest of the nodes are missing in remote, see
|
|
1227
|
1236
|
outgoing)
|
|
1228
|
1237
|
"""
|
|
1229
|
1238
|
return self.findcommonincoming(remote, base, heads, force)[1]
|
|
1230
|
1239
|
|
|
1231
|
1240
|
def findcommonincoming(self, remote, base=None, heads=None, force=False):
|
|
1232
|
1241
|
"""Return a tuple (common, missing roots, heads) used to identify
|
|
1233
|
1242
|
missing nodes from remote.
|
|
1234
|
1243
|
|
|
1235
|
1244
|
If base dict is specified, assume that these nodes and their parents
|
|
1236
|
1245
|
exist on the remote side and that no child of a node of base exists
|
|
1237
|
1246
|
in both remote and self.
|
|
1238
|
1247
|
Furthermore base will be updated to include the nodes that exists
|
|
1239
|
1248
|
in self and remote but no children exists in self and remote.
|
|
1240
|
1249
|
If a list of heads is specified, return only nodes which are heads
|
|
1241
|
1250
|
or ancestors of these heads.
|
|
1242
|
1251
|
|
|
1243
|
1252
|
All the ancestors of base are in self and in remote.
|
|
1244
|
1253
|
"""
|
|
1245
|
1254
|
m = self.changelog.nodemap
|
|
1246
|
1255
|
search = []
|
|
1247
|
1256
|
fetch = set()
|
|
1248
|
1257
|
seen = set()
|
|
1249
|
1258
|
seenbranch = set()
|
|
1250
|
1259
|
if base is None:
|
|
1251
|
1260
|
base = {}
|
|
1252
|
1261
|
|
|
1253
|
1262
|
if not heads:
|
|
1254
|
1263
|
heads = remote.heads()
|
|
1255
|
1264
|
|
|
1256
|
1265
|
if self.changelog.tip() == nullid:
|
|
1257
|
1266
|
base[nullid] = 1
|
|
1258
|
1267
|
if heads != [nullid]:
|
|
1259
|
1268
|
return [nullid], [nullid], list(heads)
|
|
1260
|
1269
|
return [nullid], [], []
|
|
1261
|
1270
|
|
|
1262
|
1271
|
# assume we're closer to the tip than the root
|
|
1263
|
1272
|
# and start by examining the heads
|
|
1264
|
1273
|
self.ui.status(_("searching for changes\n"))
|
|
1265
|
1274
|
|
|
1266
|
1275
|
unknown = []
|
|
1267
|
1276
|
for h in heads:
|
|
1268
|
1277
|
if h not in m:
|
|
1269
|
1278
|
unknown.append(h)
|
|
1270
|
1279
|
else:
|
|
1271
|
1280
|
base[h] = 1
|
|
1272
|
1281
|
|
|
1273
|
1282
|
heads = unknown
|
|
1274
|
1283
|
if not unknown:
|
|
1275
|
1284
|
return base.keys(), [], []
|
|
1276
|
1285
|
|
|
1277
|
1286
|
req = set(unknown)
|
|
1278
|
1287
|
reqcnt = 0
|
|
1279
|
1288
|
|
|
1280
|
1289
|
# search through remote branches
|
|
1281
|
1290
|
# a 'branch' here is a linear segment of history, with four parts:
|
|
1282
|
1291
|
# head, root, first parent, second parent
|
|
1283
|
1292
|
# (a branch always has two parents (or none) by definition)
|
|
1284
|
1293
|
unknown = remote.branches(unknown)
|
|
1285
|
1294
|
while unknown:
|
|
1286
|
1295
|
r = []
|
|
1287
|
1296
|
while unknown:
|
|
1288
|
1297
|
n = unknown.pop(0)
|
|
1289
|
1298
|
if n[0] in seen:
|
|
1290
|
1299
|
continue
|
|
1291
|
1300
|
|
|
1292
|
1301
|
self.ui.debug("examining %s:%s\n"
|
|
1293
|
1302
|
% (short(n[0]), short(n[1])))
|
|
1294
|
1303
|
if n[0] == nullid: # found the end of the branch
|
|
1295
|
1304
|
pass
|
|
1296
|
1305
|
elif n in seenbranch:
|
|
1297
|
1306
|
self.ui.debug("branch already found\n")
|
|
1298
|
1307
|
continue
|
|
1299
|
1308
|
elif n[1] and n[1] in m: # do we know the base?
|
|
1300
|
1309
|
self.ui.debug("found incomplete branch %s:%s\n"
|
|
1301
|
1310
|
% (short(n[0]), short(n[1])))
|
|
1302
|
1311
|
search.append(n[0:2]) # schedule branch range for scanning
|
|
1303
|
1312
|
seenbranch.add(n)
|
|
1304
|
1313
|
else:
|
|
1305
|
1314
|
if n[1] not in seen and n[1] not in fetch:
|
|
1306
|
1315
|
if n[2] in m and n[3] in m:
|
|
1307
|
1316
|
self.ui.debug("found new changeset %s\n" %
|
|
1308
|
1317
|
short(n[1]))
|
|
1309
|
1318
|
fetch.add(n[1]) # earliest unknown
|
|
1310
|
1319
|
for p in n[2:4]:
|
|
1311
|
1320
|
if p in m:
|
|
1312
|
1321
|
base[p] = 1 # latest known
|
|
1313
|
1322
|
|
|
1314
|
1323
|
for p in n[2:4]:
|
|
1315
|
1324
|
if p not in req and p not in m:
|
|
1316
|
1325
|
r.append(p)
|
|
1317
|
1326
|
req.add(p)
|
|
1318
|
1327
|
seen.add(n[0])
|
|
1319
|
1328
|
|
|
1320
|
1329
|
if r:
|
|
1321
|
1330
|
reqcnt += 1
|
|
1322
|
1331
|
self.ui.debug("request %d: %s\n" %
|
|
1323
|
1332
|
(reqcnt, " ".join(map(short, r))))
|
|
1324
|
1333
|
for p in xrange(0, len(r), 10):
|
|
1325
|
1334
|
for b in remote.branches(r[p:p+10]):
|
|
1326
|
1335
|
self.ui.debug("received %s:%s\n" %
|
|
1327
|
1336
|
(short(b[0]), short(b[1])))
|
|
1328
|
1337
|
unknown.append(b)
|
|
1329
|
1338
|
|
|
1330
|
1339
|
# do binary search on the branches we found
|
|
1331
|
1340
|
while search:
|
|
1332
|
1341
|
newsearch = []
|
|
1333
|
1342
|
reqcnt += 1
|
|
1334
|
1343
|
for n, l in zip(search, remote.between(search)):
|
|
1335
|
1344
|
l.append(n[1])
|
|
1336
|
1345
|
p = n[0]
|
|
1337
|
1346
|
f = 1
|
|
1338
|
1347
|
for i in l:
|
|
1339
|
1348
|
self.ui.debug("narrowing %d:%d %s\n" % (f, len(l), short(i)))
|
|
1340
|
1349
|
if i in m:
|
|
1341
|
1350
|
if f <= 2:
|
|
1342
|
1351
|
self.ui.debug("found new branch changeset %s\n" %
|
|
1343
|
1352
|
short(p))
|
|
1344
|
1353
|
fetch.add(p)
|
|
1345
|
1354
|
base[i] = 1
|
|
1346
|
1355
|
else:
|
|
1347
|
1356
|
self.ui.debug("narrowed branch search to %s:%s\n"
|
|
1348
|
1357
|
% (short(p), short(i)))
|
|
1349
|
1358
|
newsearch.append((p, i))
|
|
1350
|
1359
|
break
|
|
1351
|
1360
|
p, f = i, f * 2
|
|
1352
|
1361
|
search = newsearch
|
|
1353
|
1362
|
|
|
1354
|
1363
|
# sanity check our fetch list
|
|
1355
|
1364
|
for f in fetch:
|
|
1356
|
1365
|
if f in m:
|
|
1357
|
1366
|
raise error.RepoError(_("already have changeset ")
|
|
1358
|
1367
|
+ short(f[:4]))
|
|
1359
|
1368
|
|
|
1360
|
1369
|
if base.keys() == [nullid]:
|
|
1361
|
1370
|
if force:
|
|
1362
|
1371
|
self.ui.warn(_("warning: repository is unrelated\n"))
|
|
1363
|
1372
|
else:
|
|
1364
|
1373
|
raise util.Abort(_("repository is unrelated"))
|
|
1365
|
1374
|
|
|
1366
|
1375
|
self.ui.debug("found new changesets starting at " +
|
|
1367
|
1376
|
" ".join([short(f) for f in fetch]) + "\n")
|
|
1368
|
1377
|
|
|
1369
|
1378
|
self.ui.debug("%d total queries\n" % reqcnt)
|
|
1370
|
1379
|
|
|
1371
|
1380
|
return base.keys(), list(fetch), heads
|
|
1372
|
1381
|
|
|
1373
|
1382
|
def findoutgoing(self, remote, base=None, heads=None, force=False):
|
|
1374
|
1383
|
"""Return list of nodes that are roots of subsets not in remote
|
|
1375
|
1384
|
|
|
1376
|
1385
|
If base dict is specified, assume that these nodes and their parents
|
|
1377
|
1386
|
exist on the remote side.
|
|
1378
|
1387
|
If a list of heads is specified, return only nodes which are heads
|
|
1379
|
1388
|
or ancestors of these heads, and return a second element which
|
|
1380
|
1389
|
contains all remote heads which get new children.
|
|
1381
|
1390
|
"""
|
|
1382
|
1391
|
if base is None:
|
|
1383
|
1392
|
base = {}
|
|
1384
|
1393
|
self.findincoming(remote, base, heads, force=force)
|
|
1385
|
1394
|
|
|
1386
|
1395
|
self.ui.debug("common changesets up to "
|
|
1387
|
1396
|
+ " ".join(map(short, base.keys())) + "\n")
|
|
1388
|
1397
|
|
|
1389
|
1398
|
remain = set(self.changelog.nodemap)
|
|
1390
|
1399
|
|
|
1391
|
1400
|
# prune everything remote has from the tree
|
|
1392
|
1401
|
remain.remove(nullid)
|
|
1393
|
1402
|
remove = base.keys()
|
|
1394
|
1403
|
while remove:
|
|
1395
|
1404
|
n = remove.pop(0)
|
|
1396
|
1405
|
if n in remain:
|
|
1397
|
1406
|
remain.remove(n)
|
|
1398
|
1407
|
for p in self.changelog.parents(n):
|
|
1399
|
1408
|
remove.append(p)
|
|
1400
|
1409
|
|
|
1401
|
1410
|
# find every node whose parents have been pruned
|
|
1402
|
1411
|
subset = []
|
|
1403
|
1412
|
# find every remote head that will get new children
|
|
1404
|
1413
|
updated_heads = set()
|
|
1405
|
1414
|
for n in remain:
|
|
1406
|
1415
|
p1, p2 = self.changelog.parents(n)
|
|
1407
|
1416
|
if p1 not in remain and p2 not in remain:
|
|
1408
|
1417
|
subset.append(n)
|
|
1409
|
1418
|
if heads:
|
|
1410
|
1419
|
if p1 in heads:
|
|
1411
|
1420
|
updated_heads.add(p1)
|
|
1412
|
1421
|
if p2 in heads:
|
|
1413
|
1422
|
updated_heads.add(p2)
|
|
1414
|
1423
|
|
|
1415
|
1424
|
# this is the set of all roots we have to push
|
|
1416
|
1425
|
if heads:
|
|
1417
|
1426
|
return subset, list(updated_heads)
|
|
1418
|
1427
|
else:
|
|
1419
|
1428
|
return subset
|
|
1420
|
1429
|
|
|
1421
|
1430
|
def pull(self, remote, heads=None, force=False):
|
|
1422
|
1431
|
lock = self.lock()
|
|
1423
|
1432
|
try:
|
|
1424
|
1433
|
common, fetch, rheads = self.findcommonincoming(remote, heads=heads,
|
|
1425
|
1434
|
force=force)
|
|
1426
|
1435
|
if fetch == [nullid]:
|
|
1427
|
1436
|
self.ui.status(_("requesting all changes\n"))
|
|
1428
|
1437
|
|
|
1429
|
1438
|
if not fetch:
|
|
1430
|
1439
|
self.ui.status(_("no changes found\n"))
|
|
1431
|
1440
|
return 0
|
|
1432
|
1441
|
|
|
1433
|
1442
|
if heads is None and remote.capable('changegroupsubset'):
|
|
1434
|
1443
|
heads = rheads
|
|
1435
|
1444
|
|
|
1436
|
1445
|
if heads is None:
|
|
1437
|
1446
|
cg = remote.changegroup(fetch, 'pull')
|
|
1438
|
1447
|
else:
|
|
1439
|
1448
|
if not remote.capable('changegroupsubset'):
|
|
1440
|
1449
|
raise util.Abort(_("Partial pull cannot be done because "
|
|
1441
|
1450
|
"other repository doesn't support "
|
|
1442
|
1451
|
"changegroupsubset."))
|
|
1443
|
1452
|
cg = remote.changegroupsubset(fetch, heads, 'pull')
|
|
1444
|
1453
|
return self.addchangegroup(cg, 'pull', remote.url())
|
|
1445
|
1454
|
finally:
|
|
1446
|
1455
|
lock.release()
|
|
1447
|
1456
|
|
|
1448
|
1457
|
def push(self, remote, force=False, revs=None):
|
|
1449
|
1458
|
# there are two ways to push to remote repo:
|
|
1450
|
1459
|
#
|
|
1451
|
1460
|
# addchangegroup assumes local user can lock remote
|
|
1452
|
1461
|
# repo (local filesystem, old ssh servers).
|
|
1453
|
1462
|
#
|
|
1454
|
1463
|
# unbundle assumes local user cannot lock remote repo (new ssh
|
|
1455
|
1464
|
# servers, http servers).
|
|
1456
|
1465
|
|
|
1457
|
1466
|
if remote.capable('unbundle'):
|
|
1458
|
1467
|
return self.push_unbundle(remote, force, revs)
|
|
1459
|
1468
|
return self.push_addchangegroup(remote, force, revs)
|
|
1460
|
1469
|
|
|
1461
|
1470
|
def prepush(self, remote, force, revs):
|
|
1462
|
1471
|
'''Analyze the local and remote repositories and determine which
|
|
1463
|
1472
|
changesets need to be pushed to the remote. Return a tuple
|
|
1464
|
1473
|
(changegroup, remoteheads). changegroup is a readable file-like
|
|
1465
|
1474
|
object whose read() returns successive changegroup chunks ready to
|
|
1466
|
1475
|
be sent over the wire. remoteheads is the list of remote heads.
|
|
1467
|
1476
|
'''
|
|
1468
|
1477
|
common = {}
|
|
1469
|
1478
|
remote_heads = remote.heads()
|
|
1470
|
1479
|
inc = self.findincoming(remote, common, remote_heads, force=force)
|
|
1471
|
1480
|
|
|
1472
|
1481
|
update, updated_heads = self.findoutgoing(remote, common, remote_heads)
|
|
1473
|
1482
|
msng_cl, bases, heads = self.changelog.nodesbetween(update, revs)
|
|
1474
|
1483
|
|
|
1475
|
1484
|
def checkbranch(lheads, rheads, updatelb):
|
|
1476
|
1485
|
'''
|
|
1477
|
1486
|
check whether there are more local heads than remote heads on
|
|
1478
|
1487
|
a specific branch.
|
|
1479
|
1488
|
|
|
1480
|
1489
|
lheads: local branch heads
|
|
1481
|
1490
|
rheads: remote branch heads
|
|
1482
|
1491
|
updatelb: outgoing local branch bases
|
|
1483
|
1492
|
'''
|
|
1484
|
1493
|
|
|
1485
|
1494
|
warn = 0
|
|
1486
|
1495
|
|
|
1487
|
1496
|
if not revs and len(lheads) > len(rheads):
|
|
1488
|
1497
|
warn = 1
|
|
1489
|
1498
|
else:
|
|
1490
|
1499
|
# add local heads involved in the push
|
|
1491
|
1500
|
updatelheads = [self.changelog.heads(x, lheads)
|
|
1492
|
1501
|
for x in updatelb]
|
|
1493
|
1502
|
newheads = set(sum(updatelheads, [])) & set(lheads)
|
|
1494
|
1503
|
|
|
1495
|
1504
|
if not newheads:
|
|
1496
|
1505
|
return True
|
|
1497
|
1506
|
|
|
1498
|
1507
|
# add heads we don't have or that are not involved in the push
|
|
1499
|
1508
|
for r in rheads:
|
|
1500
|
1509
|
if r in self.changelog.nodemap:
|
|
1501
|
1510
|
desc = self.changelog.heads(r, heads)
|
|
1502
|
1511
|
l = [h for h in heads if h in desc]
|
|
1503
|
1512
|
if not l:
|
|
1504
|
1513
|
newheads.add(r)
|
|
1505
|
1514
|
else:
|
|
1506
|
1515
|
newheads.add(r)
|
|
1507
|
1516
|
if len(newheads) > len(rheads):
|
|
1508
|
1517
|
warn = 1
|
|
1509
|
1518
|
|
|
1510
|
1519
|
if warn:
|
|
1511
|
1520
|
if not rheads: # new branch requires --force
|
|
1512
|
1521
|
self.ui.warn(_("abort: push creates new"
|
|
1513
|
1522
|
" remote branch '%s'!\n") %
|
|
1514
|
1523
|
self[updatelb[0]].branch())
|
|
1515
|
1524
|
else:
|
|
1516
|
1525
|
self.ui.warn(_("abort: push creates new remote heads!\n"))
|
|
1517
|
1526
|
|
|
1518
|
1527
|
self.ui.status(_("(did you forget to merge?"
|
|
1519
|
1528
|
" use push -f to force)\n"))
|
|
1520
|
1529
|
return False
|
|
1521
|
1530
|
return True
|
|
1522
|
1531
|
|
|
1523
|
1532
|
if not bases:
|
|
1524
|
1533
|
self.ui.status(_("no changes found\n"))
|
|
1525
|
1534
|
return None, 1
|
|
1526
|
1535
|
elif not force:
|
|
1527
|
1536
|
# Check for each named branch if we're creating new remote heads.
|
|
1528
|
1537
|
# To be a remote head after push, node must be either:
|
|
1529
|
1538
|
# - unknown locally
|
|
1530
|
1539
|
# - a local outgoing head descended from update
|
|
1531
|
1540
|
# - a remote head that's known locally and not
|
|
1532
|
1541
|
# ancestral to an outgoing head
|
|
1533
|
1542
|
#
|
|
1534
|
1543
|
# New named branches cannot be created without --force.
|
|
1535
|
1544
|
|
|
1536
|
1545
|
if remote_heads != [nullid]:
|
|
1537
|
1546
|
if remote.capable('branchmap'):
|
|
1538
|
1547
|
localhds = {}
|
|
1539
|
1548
|
if not revs:
|
|
1540
|
1549
|
localhds = self.branchmap()
|
|
1541
|
1550
|
else:
|
|
1542
|
1551
|
for n in heads:
|
|
1543
|
1552
|
branch = self[n].branch()
|
|
1544
|
1553
|
if branch in localhds:
|
|
1545
|
1554
|
localhds[branch].append(n)
|
|
1546
|
1555
|
else:
|
|
1547
|
1556
|
localhds[branch] = [n]
|
|
1548
|
1557
|
|
|
1549
|
1558
|
remotehds = remote.branchmap()
|
|
1550
|
1559
|
|
|
1551
|
1560
|
for lh in localhds:
|
|
1552
|
1561
|
if lh in remotehds:
|
|
1553
|
1562
|
rheads = remotehds[lh]
|
|
1554
|
1563
|
else:
|
|
1555
|
1564
|
rheads = []
|
|
1556
|
1565
|
lheads = localhds[lh]
|
|
1557
|
1566
|
updatelb = [upd for upd in update
|
|
1558
|
1567
|
if self[upd].branch() == lh]
|
|
1559
|
1568
|
if not updatelb:
|
|
1560
|
1569
|
continue
|
|
1561
|
1570
|
if not checkbranch(lheads, rheads, updatelb):
|
|
1562
|
1571
|
return None, 0
|
|
1563
|
1572
|
else:
|
|
1564
|
1573
|
if not checkbranch(heads, remote_heads, update):
|
|
1565
|
1574
|
return None, 0
|
|
1566
|
1575
|
|
|
1567
|
1576
|
if inc:
|
|
1568
|
1577
|
self.ui.warn(_("note: unsynced remote changes!\n"))
|
|
1569
|
1578
|
|
|
1570
|
1579
|
|
|
1571
|
1580
|
if revs is None:
|
|
1572
|
1581
|
# use the fast path, no race possible on push
|
|
1573
|
1582
|
nodes = self.changelog.findmissing(common.keys())
|
|
1574
|
1583
|
cg = self._changegroup(nodes, 'push')
|
|
1575
|
1584
|
else:
|
|
1576
|
1585
|
cg = self.changegroupsubset(update, revs, 'push')
|
|
1577
|
1586
|
return cg, remote_heads
|
|
1578
|
1587
|
|
|
1579
|
1588
|
def push_addchangegroup(self, remote, force, revs):
|
|
1580
|
1589
|
lock = remote.lock()
|
|
1581
|
1590
|
try:
|
|
1582
|
1591
|
ret = self.prepush(remote, force, revs)
|
|
1583
|
1592
|
if ret[0] is not None:
|
|
1584
|
1593
|
cg, remote_heads = ret
|
|
1585
|
1594
|
return remote.addchangegroup(cg, 'push', self.url())
|
|
1586
|
1595
|
return ret[1]
|
|
1587
|
1596
|
finally:
|
|
1588
|
1597
|
lock.release()
|
|
1589
|
1598
|
|
|
1590
|
1599
|
def push_unbundle(self, remote, force, revs):
|
|
1591
|
1600
|
# local repo finds heads on server, finds out what revs it
|
|
1592
|
1601
|
# must push. once revs transferred, if server finds it has
|
|
1593
|
1602
|
# different heads (someone else won commit/push race), server
|
|
1594
|
1603
|
# aborts.
|
|
1595
|
1604
|
|
|
1596
|
1605
|
ret = self.prepush(remote, force, revs)
|
|
1597
|
1606
|
if ret[0] is not None:
|
|
1598
|
1607
|
cg, remote_heads = ret
|
|
1599
|
1608
|
if force: remote_heads = ['force']
|
|
1600
|
1609
|
return remote.unbundle(cg, remote_heads, 'push')
|
|
1601
|
1610
|
return ret[1]
|
|
1602
|
1611
|
|
|
1603
|
1612
|
def changegroupinfo(self, nodes, source):
|
|
1604
|
1613
|
if self.ui.verbose or source == 'bundle':
|
|
1605
|
1614
|
self.ui.status(_("%d changesets found\n") % len(nodes))
|
|
1606
|
1615
|
if self.ui.debugflag:
|
|
1607
|
1616
|
self.ui.debug("list of changesets:\n")
|
|
1608
|
1617
|
for node in nodes:
|
|
1609
|
1618
|
self.ui.debug("%s\n" % hex(node))
|
|
1610
|
1619
|
|
|
1611
|
1620
|
def changegroupsubset(self, bases, heads, source, extranodes=None):
|
|
1612
|
1621
|
"""Compute a changegroup consisting of all the nodes that are
|
|
1613
|
1622
|
descendents of any of the bases and ancestors of any of the heads.
|
|
1614
|
1623
|
Return a chunkbuffer object whose read() method will return
|
|
1615
|
1624
|
successive changegroup chunks.
|
|
1616
|
1625
|
|
|
1617
|
1626
|
It is fairly complex as determining which filenodes and which
|
|
1618
|
1627
|
manifest nodes need to be included for the changeset to be complete
|
|
1619
|
1628
|
is non-trivial.
|
|
1620
|
1629
|
|
|
1621
|
1630
|
Another wrinkle is doing the reverse, figuring out which changeset in
|
|
1622
|
1631
|
the changegroup a particular filenode or manifestnode belongs to.
|
|
1623
|
1632
|
|
|
1624
|
1633
|
The caller can specify some nodes that must be included in the
|
|
1625
|
1634
|
changegroup using the extranodes argument. It should be a dict
|
|
1626
|
1635
|
where the keys are the filenames (or 1 for the manifest), and the
|
|
1627
|
1636
|
values are lists of (node, linknode) tuples, where node is a wanted
|
|
1628
|
1637
|
node and linknode is the changelog node that should be transmitted as
|
|
1629
|
1638
|
the linkrev.
|
|
1630
|
1639
|
"""
|
|
1631
|
1640
|
|
|
1632
|
1641
|
# Set up some initial variables
|
|
1633
|
1642
|
# Make it easy to refer to self.changelog
|
|
1634
|
1643
|
cl = self.changelog
|
|
1635
|
1644
|
# msng is short for missing - compute the list of changesets in this
|
|
1636
|
1645
|
# changegroup.
|
|
1637
|
1646
|
if not bases:
|
|
1638
|
1647
|
bases = [nullid]
|
|
1639
|
1648
|
msng_cl_lst, bases, heads = cl.nodesbetween(bases, heads)
|
|
1640
|
1649
|
|
|
1641
|
1650
|
if extranodes is None:
|
|
1642
|
1651
|
# can we go through the fast path ?
|
|
1643
|
1652
|
heads.sort()
|
|
1644
|
1653
|
allheads = self.heads()
|
|
1645
|
1654
|
allheads.sort()
|
|
1646
|
1655
|
if heads == allheads:
|
|
1647
|
1656
|
return self._changegroup(msng_cl_lst, source)
|
|
1648
|
1657
|
|
|
1649
|
1658
|
# slow path
|
|
1650
|
1659
|
self.hook('preoutgoing', throw=True, source=source)
|
|
1651
|
1660
|
|
|
1652
|
1661
|
self.changegroupinfo(msng_cl_lst, source)
|
|
1653
|
1662
|
# Some bases may turn out to be superfluous, and some heads may be
|
|
1654
|
1663
|
# too. nodesbetween will return the minimal set of bases and heads
|
|
1655
|
1664
|
# necessary to re-create the changegroup.
|
|
1656
|
1665
|
|
|
1657
|
1666
|
# Known heads are the list of heads that it is assumed the recipient
|
|
1658
|
1667
|
# of this changegroup will know about.
|
|
1659
|
1668
|
knownheads = set()
|
|
1660
|
1669
|
# We assume that all parents of bases are known heads.
|
|
1661
|
1670
|
for n in bases:
|
|
1662
|
1671
|
knownheads.update(cl.parents(n))
|
|
1663
|
1672
|
knownheads.discard(nullid)
|
|
1664
|
1673
|
knownheads = list(knownheads)
|
|
1665
|
1674
|
if knownheads:
|
|
1666
|
1675
|
# Now that we know what heads are known, we can compute which
|
|
1667
|
1676
|
# changesets are known. The recipient must know about all
|
|
1668
|
1677
|
# changesets required to reach the known heads from the null
|
|
1669
|
1678
|
# changeset.
|
|
1670
|
1679
|
has_cl_set, junk, junk = cl.nodesbetween(None, knownheads)
|
|
1671
|
1680
|
junk = None
|
|
1672
|
1681
|
# Transform the list into a set.
|
|
1673
|
1682
|
has_cl_set = set(has_cl_set)
|
|
1674
|
1683
|
else:
|
|
1675
|
1684
|
# If there were no known heads, the recipient cannot be assumed to
|
|
1676
|
1685
|
# know about any changesets.
|
|
1677
|
1686
|
has_cl_set = set()
|
|
1678
|
1687
|
|
|
1679
|
1688
|
# Make it easy to refer to self.manifest
|
|
1680
|
1689
|
mnfst = self.manifest
|
|
1681
|
1690
|
# We don't know which manifests are missing yet
|
|
1682
|
1691
|
msng_mnfst_set = {}
|
|
1683
|
1692
|
# Nor do we know which filenodes are missing.
|
|
1684
|
1693
|
msng_filenode_set = {}
|
|
1685
|
1694
|
|
|
1686
|
1695
|
junk = mnfst.index[len(mnfst) - 1] # Get around a bug in lazyindex
|
|
1687
|
1696
|
junk = None
|
|
1688
|
1697
|
|
|
1689
|
1698
|
# A changeset always belongs to itself, so the changenode lookup
|
|
1690
|
1699
|
# function for a changenode is identity.
|
|
1691
|
1700
|
def identity(x):
|
|
1692
|
1701
|
return x
|
|
1693
|
1702
|
|
|
1694
|
1703
|
# If we determine that a particular file or manifest node must be a
|
|
1695
|
1704
|
# node that the recipient of the changegroup will already have, we can
|
|
1696
|
1705
|
# also assume the recipient will have all the parents. This function
|
|
1697
|
1706
|
# prunes them from the set of missing nodes.
|
|
1698
|
1707
|
def prune_parents(revlog, hasset, msngset):
|
|
1699
|
1708
|
haslst = list(hasset)
|
|
1700
|
1709
|
haslst.sort(key=revlog.rev)
|
|
1701
|
1710
|
for node in haslst:
|
|
1702
|
1711
|
parentlst = [p for p in revlog.parents(node) if p != nullid]
|
|
1703
|
1712
|
while parentlst:
|
|
1704
|
1713
|
n = parentlst.pop()
|
|
1705
|
1714
|
if n not in hasset:
|
|
1706
|
1715
|
hasset.add(n)
|
|
1707
|
1716
|
p = [p for p in revlog.parents(n) if p != nullid]
|
|
1708
|
1717
|
parentlst.extend(p)
|
|
1709
|
1718
|
for n in hasset:
|
|
1710
|
1719
|
msngset.pop(n, None)
|
|
1711
|
1720
|
|
|
1712
|
1721
|
# This is a function generating function used to set up an environment
|
|
1713
|
1722
|
# for the inner function to execute in.
|
|
1714
|
1723
|
def manifest_and_file_collector(changedfileset):
|
|
1715
|
1724
|
# This is an information gathering function that gathers
|
|
1716
|
1725
|
# information from each changeset node that goes out as part of
|
|
1717
|
1726
|
# the changegroup. The information gathered is a list of which
|
|
1718
|
1727
|
# manifest nodes are potentially required (the recipient may
|
|
1719
|
1728
|
# already have them) and total list of all files which were
|
|
1720
|
1729
|
# changed in any changeset in the changegroup.
|
|
1721
|
1730
|
#
|
|
1722
|
1731
|
# We also remember the first changenode we saw any manifest
|
|
1723
|
1732
|
# referenced by so we can later determine which changenode 'owns'
|
|
1724
|
1733
|
# the manifest.
|
|
1725
|
1734
|
def collect_manifests_and_files(clnode):
|
|
1726
|
1735
|
c = cl.read(clnode)
|
|
1727
|
1736
|
for f in c[3]:
|
|
1728
|
1737
|
# This is to make sure we only have one instance of each
|
|
1729
|
1738
|
# filename string for each filename.
|
|
1730
|
1739
|
changedfileset.setdefault(f, f)
|
|
1731
|
1740
|
msng_mnfst_set.setdefault(c[0], clnode)
|
|
1732
|
1741
|
return collect_manifests_and_files
|
|
1733
|
1742
|
|
|
1734
|
1743
|
# Figure out which manifest nodes (of the ones we think might be part
|
|
1735
|
1744
|
# of the changegroup) the recipient must know about and remove them
|
|
1736
|
1745
|
# from the changegroup.
|
|
1737
|
1746
|
def prune_manifests():
|
|
1738
|
1747
|
has_mnfst_set = set()
|
|
1739
|
1748
|
for n in msng_mnfst_set:
|
|
1740
|
1749
|
# If a 'missing' manifest thinks it belongs to a changenode
|
|
1741
|
1750
|
# the recipient is assumed to have, obviously the recipient
|
|
1742
|
1751
|
# must have that manifest.
|
|
1743
|
1752
|
linknode = cl.node(mnfst.linkrev(mnfst.rev(n)))
|
|
1744
|
1753
|
if linknode in has_cl_set:
|
|
1745
|
1754
|
has_mnfst_set.add(n)
|
|
1746
|
1755
|
prune_parents(mnfst, has_mnfst_set, msng_mnfst_set)
|
|
1747
|
1756
|
|
|
1748
|
1757
|
# Use the information collected in collect_manifests_and_files to say
|
|
1749
|
1758
|
# which changenode any manifestnode belongs to.
|
|
1750
|
1759
|
def lookup_manifest_link(mnfstnode):
|
|
1751
|
1760
|
return msng_mnfst_set[mnfstnode]
|
|
1752
|
1761
|
|
|
1753
|
1762
|
# A function generating function that sets up the initial environment
|
|
1754
|
1763
|
# the inner function.
|
|
1755
|
1764
|
def filenode_collector(changedfiles):
|
|
1756
|
1765
|
next_rev = [0]
|
|
1757
|
1766
|
# This gathers information from each manifestnode included in the
|
|
1758
|
1767
|
# changegroup about which filenodes the manifest node references
|
|
1759
|
1768
|
# so we can include those in the changegroup too.
|
|
1760
|
1769
|
#
|
|
1761
|
1770
|
# It also remembers which changenode each filenode belongs to. It
|
|
1762
|
1771
|
# does this by assuming the a filenode belongs to the changenode
|
|
1763
|
1772
|
# the first manifest that references it belongs to.
|
|
1764
|
1773
|
def collect_msng_filenodes(mnfstnode):
|
|
1765
|
1774
|
r = mnfst.rev(mnfstnode)
|
|
1766
|
1775
|
if r == next_rev[0]:
|
|
1767
|
1776
|
# If the last rev we looked at was the one just previous,
|
|
1768
|
1777
|
# we only need to see a diff.
|
|
1769
|
1778
|
deltamf = mnfst.readdelta(mnfstnode)
|
|
1770
|
1779
|
# For each line in the delta
|
|
1771
|
1780
|
for f, fnode in deltamf.iteritems():
|
|
1772
|
1781
|
f = changedfiles.get(f, None)
|
|
1773
|
1782
|
# And if the file is in the list of files we care
|
|
1774
|
1783
|
# about.
|
|
1775
|
1784
|
if f is not None:
|
|
1776
|
1785
|
# Get the changenode this manifest belongs to
|
|
1777
|
1786
|
clnode = msng_mnfst_set[mnfstnode]
|
|
1778
|
1787
|
# Create the set of filenodes for the file if
|
|
1779
|
1788
|
# there isn't one already.
|
|
1780
|
1789
|
ndset = msng_filenode_set.setdefault(f, {})
|
|
1781
|
1790
|
# And set the filenode's changelog node to the
|
|
1782
|
1791
|
# manifest's if it hasn't been set already.
|
|
1783
|
1792
|
ndset.setdefault(fnode, clnode)
|
|
1784
|
1793
|
else:
|
|
1785
|
1794
|
# Otherwise we need a full manifest.
|
|
1786
|
1795
|
m = mnfst.read(mnfstnode)
|
|
1787
|
1796
|
# For every file in we care about.
|
|
1788
|
1797
|
for f in changedfiles:
|
|
1789
|
1798
|
fnode = m.get(f, None)
|
|
1790
|
1799
|
# If it's in the manifest
|
|
1791
|
1800
|
if fnode is not None:
|
|
1792
|
1801
|
# See comments above.
|
|
1793
|
1802
|
clnode = msng_mnfst_set[mnfstnode]
|
|
1794
|
1803
|
ndset = msng_filenode_set.setdefault(f, {})
|
|
1795
|
1804
|
ndset.setdefault(fnode, clnode)
|
|
1796
|
1805
|
# Remember the revision we hope to see next.
|
|
1797
|
1806
|
next_rev[0] = r + 1
|
|
1798
|
1807
|
return collect_msng_filenodes
|
|
1799
|
1808
|
|
|
1800
|
1809
|
# We have a list of filenodes we think we need for a file, lets remove
|
|
1801
|
1810
|
# all those we know the recipient must have.
|
|
1802
|
1811
|
def prune_filenodes(f, filerevlog):
|
|
1803
|
1812
|
msngset = msng_filenode_set[f]
|
|
1804
|
1813
|
hasset = set()
|
|
1805
|
1814
|
# If a 'missing' filenode thinks it belongs to a changenode we
|
|
1806
|
1815
|
# assume the recipient must have, then the recipient must have
|
|
1807
|
1816
|
# that filenode.
|
|
1808
|
1817
|
for n in msngset:
|
|
1809
|
1818
|
clnode = cl.node(filerevlog.linkrev(filerevlog.rev(n)))
|
|
1810
|
1819
|
if clnode in has_cl_set:
|
|
1811
|
1820
|
hasset.add(n)
|
|
1812
|
1821
|
prune_parents(filerevlog, hasset, msngset)
|
|
1813
|
1822
|
|
|
1814
|
1823
|
# A function generator function that sets up the a context for the
|
|
1815
|
1824
|
# inner function.
|
|
1816
|
1825
|
def lookup_filenode_link_func(fname):
|
|
1817
|
1826
|
msngset = msng_filenode_set[fname]
|
|
1818
|
1827
|
# Lookup the changenode the filenode belongs to.
|
|
1819
|
1828
|
def lookup_filenode_link(fnode):
|
|
1820
|
1829
|
return msngset[fnode]
|
|
1821
|
1830
|
return lookup_filenode_link
|
|
1822
|
1831
|
|
|
1823
|
1832
|
# Add the nodes that were explicitly requested.
|
|
1824
|
1833
|
def add_extra_nodes(name, nodes):
|
|
1825
|
1834
|
if not extranodes or name not in extranodes:
|
|
1826
|
1835
|
return
|
|
1827
|
1836
|
|
|
1828
|
1837
|
for node, linknode in extranodes[name]:
|
|
1829
|
1838
|
if node not in nodes:
|
|
1830
|
1839
|
nodes[node] = linknode
|
|
1831
|
1840
|
|
|
1832
|
1841
|
# Now that we have all theses utility functions to help out and
|
|
1833
|
1842
|
# logically divide up the task, generate the group.
|
|
1834
|
1843
|
def gengroup():
|
|
1835
|
1844
|
# The set of changed files starts empty.
|
|
1836
|
1845
|
changedfiles = {}
|
|
1837
|
1846
|
# Create a changenode group generator that will call our functions
|
|
1838
|
1847
|
# back to lookup the owning changenode and collect information.
|
|
1839
|
1848
|
group = cl.group(msng_cl_lst, identity,
|
|
1840
|
1849
|
manifest_and_file_collector(changedfiles))
|
|
1841
|
1850
|
for chnk in group:
|
|
1842
|
1851
|
yield chnk
|
|
1843
|
1852
|
|
|
1844
|
1853
|
# The list of manifests has been collected by the generator
|
|
1845
|
1854
|
# calling our functions back.
|
|
1846
|
1855
|
prune_manifests()
|
|
1847
|
1856
|
add_extra_nodes(1, msng_mnfst_set)
|
|
1848
|
1857
|
msng_mnfst_lst = msng_mnfst_set.keys()
|
|
1849
|
1858
|
# Sort the manifestnodes by revision number.
|
|
1850
|
1859
|
msng_mnfst_lst.sort(key=mnfst.rev)
|
|
1851
|
1860
|
# Create a generator for the manifestnodes that calls our lookup
|
|
1852
|
1861
|
# and data collection functions back.
|
|
1853
|
1862
|
group = mnfst.group(msng_mnfst_lst, lookup_manifest_link,
|
|
1854
|
1863
|
filenode_collector(changedfiles))
|
|
1855
|
1864
|
for chnk in group:
|
|
1856
|
1865
|
yield chnk
|
|
1857
|
1866
|
|
|
1858
|
1867
|
# These are no longer needed, dereference and toss the memory for
|
|
1859
|
1868
|
# them.
|
|
1860
|
1869
|
msng_mnfst_lst = None
|
|
1861
|
1870
|
msng_mnfst_set.clear()
|
|
1862
|
1871
|
|
|
1863
|
1872
|
if extranodes:
|
|
1864
|
1873
|
for fname in extranodes:
|
|
1865
|
1874
|
if isinstance(fname, int):
|
|
1866
|
1875
|
continue
|
|
1867
|
1876
|
msng_filenode_set.setdefault(fname, {})
|
|
1868
|
1877
|
changedfiles[fname] = 1
|
|
1869
|
1878
|
# Go through all our files in order sorted by name.
|
|
1870
|
1879
|
for fname in sorted(changedfiles):
|
|
1871
|
1880
|
filerevlog = self.file(fname)
|
|
1872
|
1881
|
if not len(filerevlog):
|
|
1873
|
1882
|
raise util.Abort(_("empty or missing revlog for %s") % fname)
|
|
1874
|
1883
|
# Toss out the filenodes that the recipient isn't really
|
|
1875
|
1884
|
# missing.
|
|
1876
|
1885
|
if fname in msng_filenode_set:
|
|
1877
|
1886
|
prune_filenodes(fname, filerevlog)
|
|
1878
|
1887
|
add_extra_nodes(fname, msng_filenode_set[fname])
|
|
1879
|
1888
|
msng_filenode_lst = msng_filenode_set[fname].keys()
|
|
1880
|
1889
|
else:
|
|
1881
|
1890
|
msng_filenode_lst = []
|
|
1882
|
1891
|
# If any filenodes are left, generate the group for them,
|
|
1883
|
1892
|
# otherwise don't bother.
|
|
1884
|
1893
|
if len(msng_filenode_lst) > 0:
|
|
1885
|
1894
|
yield changegroup.chunkheader(len(fname))
|
|
1886
|
1895
|
yield fname
|
|
1887
|
1896
|
# Sort the filenodes by their revision #
|
|
1888
|
1897
|
msng_filenode_lst.sort(key=filerevlog.rev)
|
|
1889
|
1898
|
# Create a group generator and only pass in a changenode
|
|
1890
|
1899
|
# lookup function as we need to collect no information
|
|
1891
|
1900
|
# from filenodes.
|
|
1892
|
1901
|
group = filerevlog.group(msng_filenode_lst,
|
|
1893
|
1902
|
lookup_filenode_link_func(fname))
|
|
1894
|
1903
|
for chnk in group:
|
|
1895
|
1904
|
yield chnk
|
|
1896
|
1905
|
if fname in msng_filenode_set:
|
|
1897
|
1906
|
# Don't need this anymore, toss it to free memory.
|
|
1898
|
1907
|
del msng_filenode_set[fname]
|
|
1899
|
1908
|
# Signal that no more groups are left.
|
|
1900
|
1909
|
yield changegroup.closechunk()
|
|
1901
|
1910
|
|
|
1902
|
1911
|
if msng_cl_lst:
|
|
1903
|
1912
|
self.hook('outgoing', node=hex(msng_cl_lst[0]), source=source)
|
|
1904
|
1913
|
|
|
1905
|
1914
|
return util.chunkbuffer(gengroup())
|
|
1906
|
1915
|
|
|
1907
|
1916
|
def changegroup(self, basenodes, source):
|
|
1908
|
1917
|
# to avoid a race we use changegroupsubset() (issue1320)
|
|
1909
|
1918
|
return self.changegroupsubset(basenodes, self.heads(), source)
|
|
1910
|
1919
|
|
|
1911
|
1920
|
def _changegroup(self, nodes, source):
|
|
1912
|
1921
|
"""Compute the changegroup of all nodes that we have that a recipient
|
|
1913
|
1922
|
doesn't. Return a chunkbuffer object whose read() method will return
|
|
1914
|
1923
|
successive changegroup chunks.
|
|
1915
|
1924
|
|
|
1916
|
1925
|
This is much easier than the previous function as we can assume that
|
|
1917
|
1926
|
the recipient has any changenode we aren't sending them.
|
|
1918
|
1927
|
|
|
1919
|
1928
|
nodes is the set of nodes to send"""
|
|
1920
|
1929
|
|
|
1921
|
1930
|
self.hook('preoutgoing', throw=True, source=source)
|
|
1922
|
1931
|
|
|
1923
|
1932
|
cl = self.changelog
|
|
1924
|
1933
|
revset = set([cl.rev(n) for n in nodes])
|
|
1925
|
1934
|
self.changegroupinfo(nodes, source)
|
|
1926
|
1935
|
|
|
1927
|
1936
|
def identity(x):
|
|
1928
|
1937
|
return x
|
|
1929
|
1938
|
|
|
1930
|
1939
|
def gennodelst(log):
|
|
1931
|
1940
|
for r in log:
|
|
1932
|
1941
|
if log.linkrev(r) in revset:
|
|
1933
|
1942
|
yield log.node(r)
|
|
1934
|
1943
|
|
|
1935
|
1944
|
def changed_file_collector(changedfileset):
|
|
1936
|
1945
|
def collect_changed_files(clnode):
|
|
1937
|
1946
|
c = cl.read(clnode)
|
|
1938
|
1947
|
changedfileset.update(c[3])
|
|
1939
|
1948
|
return collect_changed_files
|
|
1940
|
1949
|
|
|
1941
|
1950
|
def lookuprevlink_func(revlog):
|
|
1942
|
1951
|
def lookuprevlink(n):
|
|
1943
|
1952
|
return cl.node(revlog.linkrev(revlog.rev(n)))
|
|
1944
|
1953
|
return lookuprevlink
|
|
1945
|
1954
|
|
|
1946
|
1955
|
def gengroup():
|
|
1947
|
1956
|
'''yield a sequence of changegroup chunks (strings)'''
|
|
1948
|
1957
|
# construct a list of all changed files
|
|
1949
|
1958
|
changedfiles = set()
|
|
1950
|
1959
|
|
|
1951
|
1960
|
for chnk in cl.group(nodes, identity,
|
|
1952
|
1961
|
changed_file_collector(changedfiles)):
|
|
1953
|
1962
|
yield chnk
|
|
1954
|
1963
|
|
|
1955
|
1964
|
mnfst = self.manifest
|
|
1956
|
1965
|
nodeiter = gennodelst(mnfst)
|
|
1957
|
1966
|
for chnk in mnfst.group(nodeiter, lookuprevlink_func(mnfst)):
|
|
1958
|
1967
|
yield chnk
|
|
1959
|
1968
|
|
|
1960
|
1969
|
for fname in sorted(changedfiles):
|
|
1961
|
1970
|
filerevlog = self.file(fname)
|
|
1962
|
1971
|
if not len(filerevlog):
|
|
1963
|
1972
|
raise util.Abort(_("empty or missing revlog for %s") % fname)
|
|
1964
|
1973
|
nodeiter = gennodelst(filerevlog)
|
|
1965
|
1974
|
nodeiter = list(nodeiter)
|
|
1966
|
1975
|
if nodeiter:
|
|
1967
|
1976
|
yield changegroup.chunkheader(len(fname))
|
|
1968
|
1977
|
yield fname
|
|
1969
|
1978
|
lookup = lookuprevlink_func(filerevlog)
|
|
1970
|
1979
|
for chnk in filerevlog.group(nodeiter, lookup):
|
|
1971
|
1980
|
yield chnk
|
|
1972
|
1981
|
|
|
1973
|
1982
|
yield changegroup.closechunk()
|
|
1974
|
1983
|
|
|
1975
|
1984
|
if nodes:
|
|
1976
|
1985
|
self.hook('outgoing', node=hex(nodes[0]), source=source)
|
|
1977
|
1986
|
|
|
1978
|
1987
|
return util.chunkbuffer(gengroup())
|
|
1979
|
1988
|
|
|
1980
|
1989
|
def addchangegroup(self, source, srctype, url, emptyok=False):
|
|
1981
|
1990
|
"""add changegroup to repo.
|
|
1982
|
1991
|
|
|
1983
|
1992
|
return values:
|
|
1984
|
1993
|
- nothing changed or no source: 0
|
|
1985
|
1994
|
- more heads than before: 1+added heads (2..n)
|
|
1986
|
1995
|
- less heads than before: -1-removed heads (-2..-n)
|
|
1987
|
1996
|
- number of heads stays the same: 1
|
|
1988
|
1997
|
"""
|
|
1989
|
1998
|
def csmap(x):
|
|
1990
|
1999
|
self.ui.debug("add changeset %s\n" % short(x))
|
|
1991
|
2000
|
return len(cl)
|
|
1992
|
2001
|
|
|
1993
|
2002
|
def revmap(x):
|
|
1994
|
2003
|
return cl.rev(x)
|
|
1995
|
2004
|
|
|
1996
|
2005
|
if not source:
|
|
1997
|
2006
|
return 0
|
|
1998
|
2007
|
|
|
1999
|
2008
|
self.hook('prechangegroup', throw=True, source=srctype, url=url)
|
|
2000
|
2009
|
|
|
2001
|
2010
|
changesets = files = revisions = 0
|
|
2002
|
2011
|
|
|
2003
|
2012
|
# write changelog data to temp files so concurrent readers will not see
|
|
2004
|
2013
|
# inconsistent view
|
|
2005
|
2014
|
cl = self.changelog
|
|
2006
|
2015
|
cl.delayupdate()
|
|
2007
|
2016
|
oldheads = len(cl.heads())
|
|
2008
|
2017
|
|
|
2009
|
2018
|
tr = self.transaction()
|
|
2010
|
2019
|
try:
|
|
2011
|
2020
|
trp = weakref.proxy(tr)
|
|
2012
|
2021
|
# pull off the changeset group
|
|
2013
|
2022
|
self.ui.status(_("adding changesets\n"))
|
|
2014
|
2023
|
clstart = len(cl)
|
|
2015
|
2024
|
chunkiter = changegroup.chunkiter(source)
|
|
2016
|
2025
|
if cl.addgroup(chunkiter, csmap, trp) is None and not emptyok:
|
|
2017
|
2026
|
raise util.Abort(_("received changelog group is empty"))
|
|
2018
|
2027
|
clend = len(cl)
|
|
2019
|
2028
|
changesets = clend - clstart
|
|
2020
|
2029
|
|
|
2021
|
2030
|
# pull off the manifest group
|
|
2022
|
2031
|
self.ui.status(_("adding manifests\n"))
|
|
2023
|
2032
|
chunkiter = changegroup.chunkiter(source)
|
|
2024
|
2033
|
# no need to check for empty manifest group here:
|
|
2025
|
2034
|
# if the result of the merge of 1 and 2 is the same in 3 and 4,
|
|
2026
|
2035
|
# no new manifest will be created and the manifest group will
|
|
2027
|
2036
|
# be empty during the pull
|
|
2028
|
2037
|
self.manifest.addgroup(chunkiter, revmap, trp)
|
|
2029
|
2038
|
|
|
2030
|
2039
|
# process the files
|
|
2031
|
2040
|
self.ui.status(_("adding file changes\n"))
|
|
2032
|
2041
|
while 1:
|
|
2033
|
2042
|
f = changegroup.getchunk(source)
|
|
2034
|
2043
|
if not f:
|
|
2035
|
2044
|
break
|
|
2036
|
2045
|
self.ui.debug("adding %s revisions\n" % f)
|
|
2037
|
2046
|
fl = self.file(f)
|
|
2038
|
2047
|
o = len(fl)
|
|
2039
|
2048
|
chunkiter = changegroup.chunkiter(source)
|
|
2040
|
2049
|
if fl.addgroup(chunkiter, revmap, trp) is None:
|
|
2041
|
2050
|
raise util.Abort(_("received file revlog group is empty"))
|
|
2042
|
2051
|
revisions += len(fl) - o
|
|
2043
|
2052
|
files += 1
|
|
2044
|
2053
|
|
|
2045
|
2054
|
newheads = len(cl.heads())
|
|
2046
|
2055
|
heads = ""
|
|
2047
|
2056
|
if oldheads and newheads != oldheads:
|
|
2048
|
2057
|
heads = _(" (%+d heads)") % (newheads - oldheads)
|
|
2049
|
2058
|
|
|
2050
|
2059
|
self.ui.status(_("added %d changesets"
|
|
2051
|
2060
|
" with %d changes to %d files%s\n")
|
|
2052
|
2061
|
% (changesets, revisions, files, heads))
|
|
2053
|
2062
|
|
|
2054
|
2063
|
if changesets > 0:
|
|
2055
|
2064
|
p = lambda: cl.writepending() and self.root or ""
|
|
2056
|
2065
|
self.hook('pretxnchangegroup', throw=True,
|
|
2057
|
2066
|
node=hex(cl.node(clstart)), source=srctype,
|
|
2058
|
2067
|
url=url, pending=p)
|
|
2059
|
2068
|
|
|
2060
|
2069
|
# make changelog see real files again
|
|
2061
|
2070
|
cl.finalize(trp)
|
|
2062
|
2071
|
|
|
2063
|
2072
|
tr.close()
|
|
2064
|
2073
|
finally:
|
|
2065
|
2074
|
del tr
|
|
2066
|
2075
|
|
|
2067
|
2076
|
if changesets > 0:
|
|
2068
|
2077
|
# forcefully update the on-disk branch cache
|
|
2069
|
2078
|
self.ui.debug("updating the branch cache\n")
|
|
2070
|
2079
|
self.branchtags()
|
|
2071
|
2080
|
self.hook("changegroup", node=hex(cl.node(clstart)),
|
|
2072
|
2081
|
source=srctype, url=url)
|
|
2073
|
2082
|
|
|
2074
|
2083
|
for i in xrange(clstart, clend):
|
|
2075
|
2084
|
self.hook("incoming", node=hex(cl.node(i)),
|
|
2076
|
2085
|
source=srctype, url=url)
|
|
2077
|
2086
|
|
|
2078
|
2087
|
# never return 0 here:
|
|
2079
|
2088
|
if newheads < oldheads:
|
|
2080
|
2089
|
return newheads - oldheads - 1
|
|
2081
|
2090
|
else:
|
|
2082
|
2091
|
return newheads - oldheads + 1
|
|
2083
|
2092
|
|
|
2084
|
2093
|
|
|
2085
|
2094
|
def stream_in(self, remote):
|
|
2086
|
2095
|
fp = remote.stream_out()
|
|
2087
|
2096
|
l = fp.readline()
|
|
2088
|
2097
|
try:
|
|
2089
|
2098
|
resp = int(l)
|
|
2090
|
2099
|
except ValueError:
|
|
2091
|
2100
|
raise error.ResponseError(
|
|
2092
|
2101
|
_('Unexpected response from remote server:'), l)
|
|
2093
|
2102
|
if resp == 1:
|
|
2094
|
2103
|
raise util.Abort(_('operation forbidden by server'))
|
|
2095
|
2104
|
elif resp == 2:
|
|
2096
|
2105
|
raise util.Abort(_('locking the remote repository failed'))
|
|
2097
|
2106
|
elif resp != 0:
|
|
2098
|
2107
|
raise util.Abort(_('the server sent an unknown error code'))
|
|
2099
|
2108
|
self.ui.status(_('streaming all changes\n'))
|
|
2100
|
2109
|
l = fp.readline()
|
|
2101
|
2110
|
try:
|
|
2102
|
2111
|
total_files, total_bytes = map(int, l.split(' ', 1))
|
|
2103
|
2112
|
except (ValueError, TypeError):
|
|
2104
|
2113
|
raise error.ResponseError(
|
|
2105
|
2114
|
_('Unexpected response from remote server:'), l)
|
|
2106
|
2115
|
self.ui.status(_('%d files to transfer, %s of data\n') %
|
|
2107
|
2116
|
(total_files, util.bytecount(total_bytes)))
|
|
2108
|
2117
|
start = time.time()
|
|
2109
|
2118
|
for i in xrange(total_files):
|
|
2110
|
2119
|
# XXX doesn't support '\n' or '\r' in filenames
|
|
2111
|
2120
|
l = fp.readline()
|
|
2112
|
2121
|
try:
|
|
2113
|
2122
|
name, size = l.split('\0', 1)
|
|
2114
|
2123
|
size = int(size)
|
|
2115
|
2124
|
except (ValueError, TypeError):
|
|
2116
|
2125
|
raise error.ResponseError(
|
|
2117
|
2126
|
_('Unexpected response from remote server:'), l)
|
|
2118
|
2127
|
self.ui.debug('adding %s (%s)\n' % (name, util.bytecount(size)))
|
|
2119
|
2128
|
# for backwards compat, name was partially encoded
|
|
2120
|
2129
|
ofp = self.sopener(store.decodedir(name), 'w')
|
|
2121
|
2130
|
for chunk in util.filechunkiter(fp, limit=size):
|
|
2122
|
2131
|
ofp.write(chunk)
|
|
2123
|
2132
|
ofp.close()
|
|
2124
|
2133
|
elapsed = time.time() - start
|
|
2125
|
2134
|
if elapsed <= 0:
|
|
2126
|
2135
|
elapsed = 0.001
|
|
2127
|
2136
|
self.ui.status(_('transferred %s in %.1f seconds (%s/sec)\n') %
|
|
2128
|
2137
|
(util.bytecount(total_bytes), elapsed,
|
|
2129
|
2138
|
util.bytecount(total_bytes / elapsed)))
|
|
2130
|
2139
|
self.invalidate()
|
|
2131
|
2140
|
return len(self.heads()) + 1
|
|
2132
|
2141
|
|
|
2133
|
2142
|
def clone(self, remote, heads=[], stream=False):
|
|
2134
|
2143
|
'''clone remote repository.
|
|
2135
|
2144
|
|
|
2136
|
2145
|
keyword arguments:
|
|
2137
|
2146
|
heads: list of revs to clone (forces use of pull)
|
|
2138
|
2147
|
stream: use streaming clone if possible'''
|
|
2139
|
2148
|
|
|
2140
|
2149
|
# now, all clients that can request uncompressed clones can
|
|
2141
|
2150
|
# read repo formats supported by all servers that can serve
|
|
2142
|
2151
|
# them.
|
|
2143
|
2152
|
|
|
2144
|
2153
|
# if revlog format changes, client will have to check version
|
|
2145
|
2154
|
# and format flags on "stream" capability, and use
|
|
2146
|
2155
|
# uncompressed only if compatible.
|
|
2147
|
2156
|
|
|
2148
|
2157
|
if stream and not heads and remote.capable('stream'):
|
|
2149
|
2158
|
return self.stream_in(remote)
|
|
2150
|
2159
|
return self.pull(remote, heads)
|
|
2151
|
2160
|
|
|
2152
|
2161
|
# used to avoid circular references so destructors work
|
|
2153
|
2162
|
def aftertrans(files):
|
|
2154
|
2163
|
renamefiles = [tuple(t) for t in files]
|
|
2155
|
2164
|
def a():
|
|
2156
|
2165
|
for src, dest in renamefiles:
|
|
2157
|
2166
|
util.rename(src, dest)
|
|
2158
|
2167
|
return a
|
|
2159
|
2168
|
|
|
2160
|
2169
|
def instance(ui, path, create):
|
|
2161
|
2170
|
return localrepository(ui, util.drop_scheme('file', path), create)
|
|
2162
|
2171
|
|
|
2163
|
2172
|
def islocal(path):
|
|
2164
|
2173
|
return True
|