##// END OF EJS Templates
revset: use fastasc/fastdesc switch consistently in addset.__iter__
Yuya Nishihara -
r25130:a94ef7f5 default
parent child Browse files
Show More
@@ -1,3512 +1,3510
1 # revset.py - revision set queries for mercurial
1 # revset.py - revision set queries for mercurial
2 #
2 #
3 # Copyright 2010 Matt Mackall <mpm@selenic.com>
3 # Copyright 2010 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 import re
8 import re
9 import parser, util, error, hbisect, phases
9 import parser, util, error, hbisect, phases
10 import node
10 import node
11 import heapq
11 import heapq
12 import match as matchmod
12 import match as matchmod
13 from i18n import _
13 from i18n import _
14 import encoding
14 import encoding
15 import obsolete as obsmod
15 import obsolete as obsmod
16 import pathutil
16 import pathutil
17 import repoview
17 import repoview
18
18
19 def _revancestors(repo, revs, followfirst):
19 def _revancestors(repo, revs, followfirst):
20 """Like revlog.ancestors(), but supports followfirst."""
20 """Like revlog.ancestors(), but supports followfirst."""
21 if followfirst:
21 if followfirst:
22 cut = 1
22 cut = 1
23 else:
23 else:
24 cut = None
24 cut = None
25 cl = repo.changelog
25 cl = repo.changelog
26
26
27 def iterate():
27 def iterate():
28 revs.sort(reverse=True)
28 revs.sort(reverse=True)
29 irevs = iter(revs)
29 irevs = iter(revs)
30 h = []
30 h = []
31 try:
31 try:
32 inputrev = irevs.next()
32 inputrev = irevs.next()
33 heapq.heappush(h, -inputrev)
33 heapq.heappush(h, -inputrev)
34 except StopIteration:
34 except StopIteration:
35 return
35 return
36
36
37 seen = set()
37 seen = set()
38 while h:
38 while h:
39 current = -heapq.heappop(h)
39 current = -heapq.heappop(h)
40 if current == inputrev:
40 if current == inputrev:
41 try:
41 try:
42 inputrev = irevs.next()
42 inputrev = irevs.next()
43 heapq.heappush(h, -inputrev)
43 heapq.heappush(h, -inputrev)
44 except StopIteration:
44 except StopIteration:
45 pass
45 pass
46 if current not in seen:
46 if current not in seen:
47 seen.add(current)
47 seen.add(current)
48 yield current
48 yield current
49 for parent in cl.parentrevs(current)[:cut]:
49 for parent in cl.parentrevs(current)[:cut]:
50 if parent != node.nullrev:
50 if parent != node.nullrev:
51 heapq.heappush(h, -parent)
51 heapq.heappush(h, -parent)
52
52
53 return generatorset(iterate(), iterasc=False)
53 return generatorset(iterate(), iterasc=False)
54
54
55 def _revdescendants(repo, revs, followfirst):
55 def _revdescendants(repo, revs, followfirst):
56 """Like revlog.descendants() but supports followfirst."""
56 """Like revlog.descendants() but supports followfirst."""
57 if followfirst:
57 if followfirst:
58 cut = 1
58 cut = 1
59 else:
59 else:
60 cut = None
60 cut = None
61
61
62 def iterate():
62 def iterate():
63 cl = repo.changelog
63 cl = repo.changelog
64 first = min(revs)
64 first = min(revs)
65 nullrev = node.nullrev
65 nullrev = node.nullrev
66 if first == nullrev:
66 if first == nullrev:
67 # Are there nodes with a null first parent and a non-null
67 # Are there nodes with a null first parent and a non-null
68 # second one? Maybe. Do we care? Probably not.
68 # second one? Maybe. Do we care? Probably not.
69 for i in cl:
69 for i in cl:
70 yield i
70 yield i
71 else:
71 else:
72 seen = set(revs)
72 seen = set(revs)
73 for i in cl.revs(first + 1):
73 for i in cl.revs(first + 1):
74 for x in cl.parentrevs(i)[:cut]:
74 for x in cl.parentrevs(i)[:cut]:
75 if x != nullrev and x in seen:
75 if x != nullrev and x in seen:
76 seen.add(i)
76 seen.add(i)
77 yield i
77 yield i
78 break
78 break
79
79
80 return generatorset(iterate(), iterasc=True)
80 return generatorset(iterate(), iterasc=True)
81
81
82 def _revsbetween(repo, roots, heads):
82 def _revsbetween(repo, roots, heads):
83 """Return all paths between roots and heads, inclusive of both endpoint
83 """Return all paths between roots and heads, inclusive of both endpoint
84 sets."""
84 sets."""
85 if not roots:
85 if not roots:
86 return baseset()
86 return baseset()
87 parentrevs = repo.changelog.parentrevs
87 parentrevs = repo.changelog.parentrevs
88 visit = list(heads)
88 visit = list(heads)
89 reachable = set()
89 reachable = set()
90 seen = {}
90 seen = {}
91 minroot = min(roots)
91 minroot = min(roots)
92 roots = set(roots)
92 roots = set(roots)
93 # open-code the post-order traversal due to the tiny size of
93 # open-code the post-order traversal due to the tiny size of
94 # sys.getrecursionlimit()
94 # sys.getrecursionlimit()
95 while visit:
95 while visit:
96 rev = visit.pop()
96 rev = visit.pop()
97 if rev in roots:
97 if rev in roots:
98 reachable.add(rev)
98 reachable.add(rev)
99 parents = parentrevs(rev)
99 parents = parentrevs(rev)
100 seen[rev] = parents
100 seen[rev] = parents
101 for parent in parents:
101 for parent in parents:
102 if parent >= minroot and parent not in seen:
102 if parent >= minroot and parent not in seen:
103 visit.append(parent)
103 visit.append(parent)
104 if not reachable:
104 if not reachable:
105 return baseset()
105 return baseset()
106 for rev in sorted(seen):
106 for rev in sorted(seen):
107 for parent in seen[rev]:
107 for parent in seen[rev]:
108 if parent in reachable:
108 if parent in reachable:
109 reachable.add(rev)
109 reachable.add(rev)
110 return baseset(sorted(reachable))
110 return baseset(sorted(reachable))
111
111
112 elements = {
112 elements = {
113 "(": (21, ("group", 1, ")"), ("func", 1, ")")),
113 "(": (21, ("group", 1, ")"), ("func", 1, ")")),
114 "##": (20, None, ("_concat", 20)),
114 "##": (20, None, ("_concat", 20)),
115 "~": (18, None, ("ancestor", 18)),
115 "~": (18, None, ("ancestor", 18)),
116 "^": (18, None, ("parent", 18), ("parentpost", 18)),
116 "^": (18, None, ("parent", 18), ("parentpost", 18)),
117 "-": (5, ("negate", 19), ("minus", 5)),
117 "-": (5, ("negate", 19), ("minus", 5)),
118 "::": (17, ("dagrangepre", 17), ("dagrange", 17),
118 "::": (17, ("dagrangepre", 17), ("dagrange", 17),
119 ("dagrangepost", 17)),
119 ("dagrangepost", 17)),
120 "..": (17, ("dagrangepre", 17), ("dagrange", 17),
120 "..": (17, ("dagrangepre", 17), ("dagrange", 17),
121 ("dagrangepost", 17)),
121 ("dagrangepost", 17)),
122 ":": (15, ("rangepre", 15), ("range", 15), ("rangepost", 15)),
122 ":": (15, ("rangepre", 15), ("range", 15), ("rangepost", 15)),
123 "not": (10, ("not", 10)),
123 "not": (10, ("not", 10)),
124 "!": (10, ("not", 10)),
124 "!": (10, ("not", 10)),
125 "and": (5, None, ("and", 5)),
125 "and": (5, None, ("and", 5)),
126 "&": (5, None, ("and", 5)),
126 "&": (5, None, ("and", 5)),
127 "%": (5, None, ("only", 5), ("onlypost", 5)),
127 "%": (5, None, ("only", 5), ("onlypost", 5)),
128 "or": (4, None, ("or", 4)),
128 "or": (4, None, ("or", 4)),
129 "|": (4, None, ("or", 4)),
129 "|": (4, None, ("or", 4)),
130 "+": (4, None, ("or", 4)),
130 "+": (4, None, ("or", 4)),
131 ",": (2, None, ("list", 2)),
131 ",": (2, None, ("list", 2)),
132 ")": (0, None, None),
132 ")": (0, None, None),
133 "symbol": (0, ("symbol",), None),
133 "symbol": (0, ("symbol",), None),
134 "string": (0, ("string",), None),
134 "string": (0, ("string",), None),
135 "end": (0, None, None),
135 "end": (0, None, None),
136 }
136 }
137
137
138 keywords = set(['and', 'or', 'not'])
138 keywords = set(['and', 'or', 'not'])
139
139
140 # default set of valid characters for the initial letter of symbols
140 # default set of valid characters for the initial letter of symbols
141 _syminitletters = set(c for c in [chr(i) for i in xrange(256)]
141 _syminitletters = set(c for c in [chr(i) for i in xrange(256)]
142 if c.isalnum() or c in '._@' or ord(c) > 127)
142 if c.isalnum() or c in '._@' or ord(c) > 127)
143
143
144 # default set of valid characters for non-initial letters of symbols
144 # default set of valid characters for non-initial letters of symbols
145 _symletters = set(c for c in [chr(i) for i in xrange(256)]
145 _symletters = set(c for c in [chr(i) for i in xrange(256)]
146 if c.isalnum() or c in '-._/@' or ord(c) > 127)
146 if c.isalnum() or c in '-._/@' or ord(c) > 127)
147
147
148 def tokenize(program, lookup=None, syminitletters=None, symletters=None):
148 def tokenize(program, lookup=None, syminitletters=None, symletters=None):
149 '''
149 '''
150 Parse a revset statement into a stream of tokens
150 Parse a revset statement into a stream of tokens
151
151
152 ``syminitletters`` is the set of valid characters for the initial
152 ``syminitletters`` is the set of valid characters for the initial
153 letter of symbols.
153 letter of symbols.
154
154
155 By default, character ``c`` is recognized as valid for initial
155 By default, character ``c`` is recognized as valid for initial
156 letter of symbols, if ``c.isalnum() or c in '._@' or ord(c) > 127``.
156 letter of symbols, if ``c.isalnum() or c in '._@' or ord(c) > 127``.
157
157
158 ``symletters`` is the set of valid characters for non-initial
158 ``symletters`` is the set of valid characters for non-initial
159 letters of symbols.
159 letters of symbols.
160
160
161 By default, character ``c`` is recognized as valid for non-initial
161 By default, character ``c`` is recognized as valid for non-initial
162 letters of symbols, if ``c.isalnum() or c in '-._/@' or ord(c) > 127``.
162 letters of symbols, if ``c.isalnum() or c in '-._/@' or ord(c) > 127``.
163
163
164 Check that @ is a valid unquoted token character (issue3686):
164 Check that @ is a valid unquoted token character (issue3686):
165 >>> list(tokenize("@::"))
165 >>> list(tokenize("@::"))
166 [('symbol', '@', 0), ('::', None, 1), ('end', None, 3)]
166 [('symbol', '@', 0), ('::', None, 1), ('end', None, 3)]
167
167
168 '''
168 '''
169 if syminitletters is None:
169 if syminitletters is None:
170 syminitletters = _syminitletters
170 syminitletters = _syminitletters
171 if symletters is None:
171 if symletters is None:
172 symletters = _symletters
172 symletters = _symletters
173
173
174 pos, l = 0, len(program)
174 pos, l = 0, len(program)
175 while pos < l:
175 while pos < l:
176 c = program[pos]
176 c = program[pos]
177 if c.isspace(): # skip inter-token whitespace
177 if c.isspace(): # skip inter-token whitespace
178 pass
178 pass
179 elif c == ':' and program[pos:pos + 2] == '::': # look ahead carefully
179 elif c == ':' and program[pos:pos + 2] == '::': # look ahead carefully
180 yield ('::', None, pos)
180 yield ('::', None, pos)
181 pos += 1 # skip ahead
181 pos += 1 # skip ahead
182 elif c == '.' and program[pos:pos + 2] == '..': # look ahead carefully
182 elif c == '.' and program[pos:pos + 2] == '..': # look ahead carefully
183 yield ('..', None, pos)
183 yield ('..', None, pos)
184 pos += 1 # skip ahead
184 pos += 1 # skip ahead
185 elif c == '#' and program[pos:pos + 2] == '##': # look ahead carefully
185 elif c == '#' and program[pos:pos + 2] == '##': # look ahead carefully
186 yield ('##', None, pos)
186 yield ('##', None, pos)
187 pos += 1 # skip ahead
187 pos += 1 # skip ahead
188 elif c in "():,-|&+!~^%": # handle simple operators
188 elif c in "():,-|&+!~^%": # handle simple operators
189 yield (c, None, pos)
189 yield (c, None, pos)
190 elif (c in '"\'' or c == 'r' and
190 elif (c in '"\'' or c == 'r' and
191 program[pos:pos + 2] in ("r'", 'r"')): # handle quoted strings
191 program[pos:pos + 2] in ("r'", 'r"')): # handle quoted strings
192 if c == 'r':
192 if c == 'r':
193 pos += 1
193 pos += 1
194 c = program[pos]
194 c = program[pos]
195 decode = lambda x: x
195 decode = lambda x: x
196 else:
196 else:
197 decode = lambda x: x.decode('string-escape')
197 decode = lambda x: x.decode('string-escape')
198 pos += 1
198 pos += 1
199 s = pos
199 s = pos
200 while pos < l: # find closing quote
200 while pos < l: # find closing quote
201 d = program[pos]
201 d = program[pos]
202 if d == '\\': # skip over escaped characters
202 if d == '\\': # skip over escaped characters
203 pos += 2
203 pos += 2
204 continue
204 continue
205 if d == c:
205 if d == c:
206 yield ('string', decode(program[s:pos]), s)
206 yield ('string', decode(program[s:pos]), s)
207 break
207 break
208 pos += 1
208 pos += 1
209 else:
209 else:
210 raise error.ParseError(_("unterminated string"), s)
210 raise error.ParseError(_("unterminated string"), s)
211 # gather up a symbol/keyword
211 # gather up a symbol/keyword
212 elif c in syminitletters:
212 elif c in syminitletters:
213 s = pos
213 s = pos
214 pos += 1
214 pos += 1
215 while pos < l: # find end of symbol
215 while pos < l: # find end of symbol
216 d = program[pos]
216 d = program[pos]
217 if d not in symletters:
217 if d not in symletters:
218 break
218 break
219 if d == '.' and program[pos - 1] == '.': # special case for ..
219 if d == '.' and program[pos - 1] == '.': # special case for ..
220 pos -= 1
220 pos -= 1
221 break
221 break
222 pos += 1
222 pos += 1
223 sym = program[s:pos]
223 sym = program[s:pos]
224 if sym in keywords: # operator keywords
224 if sym in keywords: # operator keywords
225 yield (sym, None, s)
225 yield (sym, None, s)
226 elif '-' in sym:
226 elif '-' in sym:
227 # some jerk gave us foo-bar-baz, try to check if it's a symbol
227 # some jerk gave us foo-bar-baz, try to check if it's a symbol
228 if lookup and lookup(sym):
228 if lookup and lookup(sym):
229 # looks like a real symbol
229 # looks like a real symbol
230 yield ('symbol', sym, s)
230 yield ('symbol', sym, s)
231 else:
231 else:
232 # looks like an expression
232 # looks like an expression
233 parts = sym.split('-')
233 parts = sym.split('-')
234 for p in parts[:-1]:
234 for p in parts[:-1]:
235 if p: # possible consecutive -
235 if p: # possible consecutive -
236 yield ('symbol', p, s)
236 yield ('symbol', p, s)
237 s += len(p)
237 s += len(p)
238 yield ('-', None, pos)
238 yield ('-', None, pos)
239 s += 1
239 s += 1
240 if parts[-1]: # possible trailing -
240 if parts[-1]: # possible trailing -
241 yield ('symbol', parts[-1], s)
241 yield ('symbol', parts[-1], s)
242 else:
242 else:
243 yield ('symbol', sym, s)
243 yield ('symbol', sym, s)
244 pos -= 1
244 pos -= 1
245 else:
245 else:
246 raise error.ParseError(_("syntax error in revset '%s'") %
246 raise error.ParseError(_("syntax error in revset '%s'") %
247 program, pos)
247 program, pos)
248 pos += 1
248 pos += 1
249 yield ('end', None, pos)
249 yield ('end', None, pos)
250
250
251 def parseerrordetail(inst):
251 def parseerrordetail(inst):
252 """Compose error message from specified ParseError object
252 """Compose error message from specified ParseError object
253 """
253 """
254 if len(inst.args) > 1:
254 if len(inst.args) > 1:
255 return _('at %s: %s') % (inst.args[1], inst.args[0])
255 return _('at %s: %s') % (inst.args[1], inst.args[0])
256 else:
256 else:
257 return inst.args[0]
257 return inst.args[0]
258
258
259 # helpers
259 # helpers
260
260
261 def getstring(x, err):
261 def getstring(x, err):
262 if x and (x[0] == 'string' or x[0] == 'symbol'):
262 if x and (x[0] == 'string' or x[0] == 'symbol'):
263 return x[1]
263 return x[1]
264 raise error.ParseError(err)
264 raise error.ParseError(err)
265
265
266 def getlist(x):
266 def getlist(x):
267 if not x:
267 if not x:
268 return []
268 return []
269 if x[0] == 'list':
269 if x[0] == 'list':
270 return getlist(x[1]) + [x[2]]
270 return getlist(x[1]) + [x[2]]
271 return [x]
271 return [x]
272
272
273 def getargs(x, min, max, err):
273 def getargs(x, min, max, err):
274 l = getlist(x)
274 l = getlist(x)
275 if len(l) < min or (max >= 0 and len(l) > max):
275 if len(l) < min or (max >= 0 and len(l) > max):
276 raise error.ParseError(err)
276 raise error.ParseError(err)
277 return l
277 return l
278
278
279 def isvalidsymbol(tree):
279 def isvalidsymbol(tree):
280 """Examine whether specified ``tree`` is valid ``symbol`` or not
280 """Examine whether specified ``tree`` is valid ``symbol`` or not
281 """
281 """
282 return tree[0] == 'symbol' and len(tree) > 1
282 return tree[0] == 'symbol' and len(tree) > 1
283
283
284 def getsymbol(tree):
284 def getsymbol(tree):
285 """Get symbol name from valid ``symbol`` in ``tree``
285 """Get symbol name from valid ``symbol`` in ``tree``
286
286
287 This assumes that ``tree`` is already examined by ``isvalidsymbol``.
287 This assumes that ``tree`` is already examined by ``isvalidsymbol``.
288 """
288 """
289 return tree[1]
289 return tree[1]
290
290
291 def isvalidfunc(tree):
291 def isvalidfunc(tree):
292 """Examine whether specified ``tree`` is valid ``func`` or not
292 """Examine whether specified ``tree`` is valid ``func`` or not
293 """
293 """
294 return tree[0] == 'func' and len(tree) > 1 and isvalidsymbol(tree[1])
294 return tree[0] == 'func' and len(tree) > 1 and isvalidsymbol(tree[1])
295
295
296 def getfuncname(tree):
296 def getfuncname(tree):
297 """Get function name from valid ``func`` in ``tree``
297 """Get function name from valid ``func`` in ``tree``
298
298
299 This assumes that ``tree`` is already examined by ``isvalidfunc``.
299 This assumes that ``tree`` is already examined by ``isvalidfunc``.
300 """
300 """
301 return getsymbol(tree[1])
301 return getsymbol(tree[1])
302
302
303 def getfuncargs(tree):
303 def getfuncargs(tree):
304 """Get list of function arguments from valid ``func`` in ``tree``
304 """Get list of function arguments from valid ``func`` in ``tree``
305
305
306 This assumes that ``tree`` is already examined by ``isvalidfunc``.
306 This assumes that ``tree`` is already examined by ``isvalidfunc``.
307 """
307 """
308 if len(tree) > 2:
308 if len(tree) > 2:
309 return getlist(tree[2])
309 return getlist(tree[2])
310 else:
310 else:
311 return []
311 return []
312
312
313 def getset(repo, subset, x):
313 def getset(repo, subset, x):
314 if not x:
314 if not x:
315 raise error.ParseError(_("missing argument"))
315 raise error.ParseError(_("missing argument"))
316 s = methods[x[0]](repo, subset, *x[1:])
316 s = methods[x[0]](repo, subset, *x[1:])
317 if util.safehasattr(s, 'isascending'):
317 if util.safehasattr(s, 'isascending'):
318 return s
318 return s
319 return baseset(s)
319 return baseset(s)
320
320
321 def _getrevsource(repo, r):
321 def _getrevsource(repo, r):
322 extra = repo[r].extra()
322 extra = repo[r].extra()
323 for label in ('source', 'transplant_source', 'rebase_source'):
323 for label in ('source', 'transplant_source', 'rebase_source'):
324 if label in extra:
324 if label in extra:
325 try:
325 try:
326 return repo[extra[label]].rev()
326 return repo[extra[label]].rev()
327 except error.RepoLookupError:
327 except error.RepoLookupError:
328 pass
328 pass
329 return None
329 return None
330
330
331 # operator methods
331 # operator methods
332
332
333 def stringset(repo, subset, x):
333 def stringset(repo, subset, x):
334 x = repo[x].rev()
334 x = repo[x].rev()
335 if x in subset:
335 if x in subset:
336 return baseset([x])
336 return baseset([x])
337 return baseset()
337 return baseset()
338
338
339 def rangeset(repo, subset, x, y):
339 def rangeset(repo, subset, x, y):
340 m = getset(repo, fullreposet(repo), x)
340 m = getset(repo, fullreposet(repo), x)
341 n = getset(repo, fullreposet(repo), y)
341 n = getset(repo, fullreposet(repo), y)
342
342
343 if not m or not n:
343 if not m or not n:
344 return baseset()
344 return baseset()
345 m, n = m.first(), n.last()
345 m, n = m.first(), n.last()
346
346
347 if m < n:
347 if m < n:
348 r = spanset(repo, m, n + 1)
348 r = spanset(repo, m, n + 1)
349 else:
349 else:
350 r = spanset(repo, m, n - 1)
350 r = spanset(repo, m, n - 1)
351 return r & subset
351 return r & subset
352
352
353 def dagrange(repo, subset, x, y):
353 def dagrange(repo, subset, x, y):
354 r = fullreposet(repo)
354 r = fullreposet(repo)
355 xs = _revsbetween(repo, getset(repo, r, x), getset(repo, r, y))
355 xs = _revsbetween(repo, getset(repo, r, x), getset(repo, r, y))
356 return xs & subset
356 return xs & subset
357
357
358 def andset(repo, subset, x, y):
358 def andset(repo, subset, x, y):
359 return getset(repo, getset(repo, subset, x), y)
359 return getset(repo, getset(repo, subset, x), y)
360
360
361 def orset(repo, subset, x, y):
361 def orset(repo, subset, x, y):
362 xl = getset(repo, subset, x)
362 xl = getset(repo, subset, x)
363 yl = getset(repo, subset, y)
363 yl = getset(repo, subset, y)
364 return xl + yl
364 return xl + yl
365
365
366 def notset(repo, subset, x):
366 def notset(repo, subset, x):
367 return subset - getset(repo, subset, x)
367 return subset - getset(repo, subset, x)
368
368
369 def listset(repo, subset, a, b):
369 def listset(repo, subset, a, b):
370 raise error.ParseError(_("can't use a list in this context"))
370 raise error.ParseError(_("can't use a list in this context"))
371
371
372 def func(repo, subset, a, b):
372 def func(repo, subset, a, b):
373 if a[0] == 'symbol' and a[1] in symbols:
373 if a[0] == 'symbol' and a[1] in symbols:
374 return symbols[a[1]](repo, subset, b)
374 return symbols[a[1]](repo, subset, b)
375 raise error.UnknownIdentifier(a[1], symbols.keys())
375 raise error.UnknownIdentifier(a[1], symbols.keys())
376
376
377 # functions
377 # functions
378
378
379 def adds(repo, subset, x):
379 def adds(repo, subset, x):
380 """``adds(pattern)``
380 """``adds(pattern)``
381 Changesets that add a file matching pattern.
381 Changesets that add a file matching pattern.
382
382
383 The pattern without explicit kind like ``glob:`` is expected to be
383 The pattern without explicit kind like ``glob:`` is expected to be
384 relative to the current directory and match against a file or a
384 relative to the current directory and match against a file or a
385 directory.
385 directory.
386 """
386 """
387 # i18n: "adds" is a keyword
387 # i18n: "adds" is a keyword
388 pat = getstring(x, _("adds requires a pattern"))
388 pat = getstring(x, _("adds requires a pattern"))
389 return checkstatus(repo, subset, pat, 1)
389 return checkstatus(repo, subset, pat, 1)
390
390
391 def ancestor(repo, subset, x):
391 def ancestor(repo, subset, x):
392 """``ancestor(*changeset)``
392 """``ancestor(*changeset)``
393 A greatest common ancestor of the changesets.
393 A greatest common ancestor of the changesets.
394
394
395 Accepts 0 or more changesets.
395 Accepts 0 or more changesets.
396 Will return empty list when passed no args.
396 Will return empty list when passed no args.
397 Greatest common ancestor of a single changeset is that changeset.
397 Greatest common ancestor of a single changeset is that changeset.
398 """
398 """
399 # i18n: "ancestor" is a keyword
399 # i18n: "ancestor" is a keyword
400 l = getlist(x)
400 l = getlist(x)
401 rl = fullreposet(repo)
401 rl = fullreposet(repo)
402 anc = None
402 anc = None
403
403
404 # (getset(repo, rl, i) for i in l) generates a list of lists
404 # (getset(repo, rl, i) for i in l) generates a list of lists
405 for revs in (getset(repo, rl, i) for i in l):
405 for revs in (getset(repo, rl, i) for i in l):
406 for r in revs:
406 for r in revs:
407 if anc is None:
407 if anc is None:
408 anc = repo[r]
408 anc = repo[r]
409 else:
409 else:
410 anc = anc.ancestor(repo[r])
410 anc = anc.ancestor(repo[r])
411
411
412 if anc is not None and anc.rev() in subset:
412 if anc is not None and anc.rev() in subset:
413 return baseset([anc.rev()])
413 return baseset([anc.rev()])
414 return baseset()
414 return baseset()
415
415
416 def _ancestors(repo, subset, x, followfirst=False):
416 def _ancestors(repo, subset, x, followfirst=False):
417 heads = getset(repo, fullreposet(repo), x)
417 heads = getset(repo, fullreposet(repo), x)
418 if not heads:
418 if not heads:
419 return baseset()
419 return baseset()
420 s = _revancestors(repo, heads, followfirst)
420 s = _revancestors(repo, heads, followfirst)
421 return subset & s
421 return subset & s
422
422
423 def ancestors(repo, subset, x):
423 def ancestors(repo, subset, x):
424 """``ancestors(set)``
424 """``ancestors(set)``
425 Changesets that are ancestors of a changeset in set.
425 Changesets that are ancestors of a changeset in set.
426 """
426 """
427 return _ancestors(repo, subset, x)
427 return _ancestors(repo, subset, x)
428
428
429 def _firstancestors(repo, subset, x):
429 def _firstancestors(repo, subset, x):
430 # ``_firstancestors(set)``
430 # ``_firstancestors(set)``
431 # Like ``ancestors(set)`` but follows only the first parents.
431 # Like ``ancestors(set)`` but follows only the first parents.
432 return _ancestors(repo, subset, x, followfirst=True)
432 return _ancestors(repo, subset, x, followfirst=True)
433
433
434 def ancestorspec(repo, subset, x, n):
434 def ancestorspec(repo, subset, x, n):
435 """``set~n``
435 """``set~n``
436 Changesets that are the Nth ancestor (first parents only) of a changeset
436 Changesets that are the Nth ancestor (first parents only) of a changeset
437 in set.
437 in set.
438 """
438 """
439 try:
439 try:
440 n = int(n[1])
440 n = int(n[1])
441 except (TypeError, ValueError):
441 except (TypeError, ValueError):
442 raise error.ParseError(_("~ expects a number"))
442 raise error.ParseError(_("~ expects a number"))
443 ps = set()
443 ps = set()
444 cl = repo.changelog
444 cl = repo.changelog
445 for r in getset(repo, fullreposet(repo), x):
445 for r in getset(repo, fullreposet(repo), x):
446 for i in range(n):
446 for i in range(n):
447 r = cl.parentrevs(r)[0]
447 r = cl.parentrevs(r)[0]
448 ps.add(r)
448 ps.add(r)
449 return subset & ps
449 return subset & ps
450
450
451 def author(repo, subset, x):
451 def author(repo, subset, x):
452 """``author(string)``
452 """``author(string)``
453 Alias for ``user(string)``.
453 Alias for ``user(string)``.
454 """
454 """
455 # i18n: "author" is a keyword
455 # i18n: "author" is a keyword
456 n = encoding.lower(getstring(x, _("author requires a string")))
456 n = encoding.lower(getstring(x, _("author requires a string")))
457 kind, pattern, matcher = _substringmatcher(n)
457 kind, pattern, matcher = _substringmatcher(n)
458 return subset.filter(lambda x: matcher(encoding.lower(repo[x].user())))
458 return subset.filter(lambda x: matcher(encoding.lower(repo[x].user())))
459
459
460 def bisect(repo, subset, x):
460 def bisect(repo, subset, x):
461 """``bisect(string)``
461 """``bisect(string)``
462 Changesets marked in the specified bisect status:
462 Changesets marked in the specified bisect status:
463
463
464 - ``good``, ``bad``, ``skip``: csets explicitly marked as good/bad/skip
464 - ``good``, ``bad``, ``skip``: csets explicitly marked as good/bad/skip
465 - ``goods``, ``bads`` : csets topologically good/bad
465 - ``goods``, ``bads`` : csets topologically good/bad
466 - ``range`` : csets taking part in the bisection
466 - ``range`` : csets taking part in the bisection
467 - ``pruned`` : csets that are goods, bads or skipped
467 - ``pruned`` : csets that are goods, bads or skipped
468 - ``untested`` : csets whose fate is yet unknown
468 - ``untested`` : csets whose fate is yet unknown
469 - ``ignored`` : csets ignored due to DAG topology
469 - ``ignored`` : csets ignored due to DAG topology
470 - ``current`` : the cset currently being bisected
470 - ``current`` : the cset currently being bisected
471 """
471 """
472 # i18n: "bisect" is a keyword
472 # i18n: "bisect" is a keyword
473 status = getstring(x, _("bisect requires a string")).lower()
473 status = getstring(x, _("bisect requires a string")).lower()
474 state = set(hbisect.get(repo, status))
474 state = set(hbisect.get(repo, status))
475 return subset & state
475 return subset & state
476
476
477 # Backward-compatibility
477 # Backward-compatibility
478 # - no help entry so that we do not advertise it any more
478 # - no help entry so that we do not advertise it any more
479 def bisected(repo, subset, x):
479 def bisected(repo, subset, x):
480 return bisect(repo, subset, x)
480 return bisect(repo, subset, x)
481
481
482 def bookmark(repo, subset, x):
482 def bookmark(repo, subset, x):
483 """``bookmark([name])``
483 """``bookmark([name])``
484 The named bookmark or all bookmarks.
484 The named bookmark or all bookmarks.
485
485
486 If `name` starts with `re:`, the remainder of the name is treated as
486 If `name` starts with `re:`, the remainder of the name is treated as
487 a regular expression. To match a bookmark that actually starts with `re:`,
487 a regular expression. To match a bookmark that actually starts with `re:`,
488 use the prefix `literal:`.
488 use the prefix `literal:`.
489 """
489 """
490 # i18n: "bookmark" is a keyword
490 # i18n: "bookmark" is a keyword
491 args = getargs(x, 0, 1, _('bookmark takes one or no arguments'))
491 args = getargs(x, 0, 1, _('bookmark takes one or no arguments'))
492 if args:
492 if args:
493 bm = getstring(args[0],
493 bm = getstring(args[0],
494 # i18n: "bookmark" is a keyword
494 # i18n: "bookmark" is a keyword
495 _('the argument to bookmark must be a string'))
495 _('the argument to bookmark must be a string'))
496 kind, pattern, matcher = _stringmatcher(bm)
496 kind, pattern, matcher = _stringmatcher(bm)
497 bms = set()
497 bms = set()
498 if kind == 'literal':
498 if kind == 'literal':
499 bmrev = repo._bookmarks.get(pattern, None)
499 bmrev = repo._bookmarks.get(pattern, None)
500 if not bmrev:
500 if not bmrev:
501 raise error.RepoLookupError(_("bookmark '%s' does not exist")
501 raise error.RepoLookupError(_("bookmark '%s' does not exist")
502 % bm)
502 % bm)
503 bms.add(repo[bmrev].rev())
503 bms.add(repo[bmrev].rev())
504 else:
504 else:
505 matchrevs = set()
505 matchrevs = set()
506 for name, bmrev in repo._bookmarks.iteritems():
506 for name, bmrev in repo._bookmarks.iteritems():
507 if matcher(name):
507 if matcher(name):
508 matchrevs.add(bmrev)
508 matchrevs.add(bmrev)
509 if not matchrevs:
509 if not matchrevs:
510 raise error.RepoLookupError(_("no bookmarks exist"
510 raise error.RepoLookupError(_("no bookmarks exist"
511 " that match '%s'") % pattern)
511 " that match '%s'") % pattern)
512 for bmrev in matchrevs:
512 for bmrev in matchrevs:
513 bms.add(repo[bmrev].rev())
513 bms.add(repo[bmrev].rev())
514 else:
514 else:
515 bms = set([repo[r].rev()
515 bms = set([repo[r].rev()
516 for r in repo._bookmarks.values()])
516 for r in repo._bookmarks.values()])
517 bms -= set([node.nullrev])
517 bms -= set([node.nullrev])
518 return subset & bms
518 return subset & bms
519
519
520 def branch(repo, subset, x):
520 def branch(repo, subset, x):
521 """``branch(string or set)``
521 """``branch(string or set)``
522 All changesets belonging to the given branch or the branches of the given
522 All changesets belonging to the given branch or the branches of the given
523 changesets.
523 changesets.
524
524
525 If `string` starts with `re:`, the remainder of the name is treated as
525 If `string` starts with `re:`, the remainder of the name is treated as
526 a regular expression. To match a branch that actually starts with `re:`,
526 a regular expression. To match a branch that actually starts with `re:`,
527 use the prefix `literal:`.
527 use the prefix `literal:`.
528 """
528 """
529 getbi = repo.revbranchcache().branchinfo
529 getbi = repo.revbranchcache().branchinfo
530
530
531 try:
531 try:
532 b = getstring(x, '')
532 b = getstring(x, '')
533 except error.ParseError:
533 except error.ParseError:
534 # not a string, but another revspec, e.g. tip()
534 # not a string, but another revspec, e.g. tip()
535 pass
535 pass
536 else:
536 else:
537 kind, pattern, matcher = _stringmatcher(b)
537 kind, pattern, matcher = _stringmatcher(b)
538 if kind == 'literal':
538 if kind == 'literal':
539 # note: falls through to the revspec case if no branch with
539 # note: falls through to the revspec case if no branch with
540 # this name exists
540 # this name exists
541 if pattern in repo.branchmap():
541 if pattern in repo.branchmap():
542 return subset.filter(lambda r: matcher(getbi(r)[0]))
542 return subset.filter(lambda r: matcher(getbi(r)[0]))
543 else:
543 else:
544 return subset.filter(lambda r: matcher(getbi(r)[0]))
544 return subset.filter(lambda r: matcher(getbi(r)[0]))
545
545
546 s = getset(repo, fullreposet(repo), x)
546 s = getset(repo, fullreposet(repo), x)
547 b = set()
547 b = set()
548 for r in s:
548 for r in s:
549 b.add(getbi(r)[0])
549 b.add(getbi(r)[0])
550 c = s.__contains__
550 c = s.__contains__
551 return subset.filter(lambda r: c(r) or getbi(r)[0] in b)
551 return subset.filter(lambda r: c(r) or getbi(r)[0] in b)
552
552
553 def bumped(repo, subset, x):
553 def bumped(repo, subset, x):
554 """``bumped()``
554 """``bumped()``
555 Mutable changesets marked as successors of public changesets.
555 Mutable changesets marked as successors of public changesets.
556
556
557 Only non-public and non-obsolete changesets can be `bumped`.
557 Only non-public and non-obsolete changesets can be `bumped`.
558 """
558 """
559 # i18n: "bumped" is a keyword
559 # i18n: "bumped" is a keyword
560 getargs(x, 0, 0, _("bumped takes no arguments"))
560 getargs(x, 0, 0, _("bumped takes no arguments"))
561 bumped = obsmod.getrevs(repo, 'bumped')
561 bumped = obsmod.getrevs(repo, 'bumped')
562 return subset & bumped
562 return subset & bumped
563
563
564 def bundle(repo, subset, x):
564 def bundle(repo, subset, x):
565 """``bundle()``
565 """``bundle()``
566 Changesets in the bundle.
566 Changesets in the bundle.
567
567
568 Bundle must be specified by the -R option."""
568 Bundle must be specified by the -R option."""
569
569
570 try:
570 try:
571 bundlerevs = repo.changelog.bundlerevs
571 bundlerevs = repo.changelog.bundlerevs
572 except AttributeError:
572 except AttributeError:
573 raise util.Abort(_("no bundle provided - specify with -R"))
573 raise util.Abort(_("no bundle provided - specify with -R"))
574 return subset & bundlerevs
574 return subset & bundlerevs
575
575
576 def checkstatus(repo, subset, pat, field):
576 def checkstatus(repo, subset, pat, field):
577 hasset = matchmod.patkind(pat) == 'set'
577 hasset = matchmod.patkind(pat) == 'set'
578
578
579 mcache = [None]
579 mcache = [None]
580 def matches(x):
580 def matches(x):
581 c = repo[x]
581 c = repo[x]
582 if not mcache[0] or hasset:
582 if not mcache[0] or hasset:
583 mcache[0] = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=c)
583 mcache[0] = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=c)
584 m = mcache[0]
584 m = mcache[0]
585 fname = None
585 fname = None
586 if not m.anypats() and len(m.files()) == 1:
586 if not m.anypats() and len(m.files()) == 1:
587 fname = m.files()[0]
587 fname = m.files()[0]
588 if fname is not None:
588 if fname is not None:
589 if fname not in c.files():
589 if fname not in c.files():
590 return False
590 return False
591 else:
591 else:
592 for f in c.files():
592 for f in c.files():
593 if m(f):
593 if m(f):
594 break
594 break
595 else:
595 else:
596 return False
596 return False
597 files = repo.status(c.p1().node(), c.node())[field]
597 files = repo.status(c.p1().node(), c.node())[field]
598 if fname is not None:
598 if fname is not None:
599 if fname in files:
599 if fname in files:
600 return True
600 return True
601 else:
601 else:
602 for f in files:
602 for f in files:
603 if m(f):
603 if m(f):
604 return True
604 return True
605
605
606 return subset.filter(matches)
606 return subset.filter(matches)
607
607
608 def _children(repo, narrow, parentset):
608 def _children(repo, narrow, parentset):
609 cs = set()
609 cs = set()
610 if not parentset:
610 if not parentset:
611 return baseset(cs)
611 return baseset(cs)
612 pr = repo.changelog.parentrevs
612 pr = repo.changelog.parentrevs
613 minrev = min(parentset)
613 minrev = min(parentset)
614 for r in narrow:
614 for r in narrow:
615 if r <= minrev:
615 if r <= minrev:
616 continue
616 continue
617 for p in pr(r):
617 for p in pr(r):
618 if p in parentset:
618 if p in parentset:
619 cs.add(r)
619 cs.add(r)
620 return baseset(cs)
620 return baseset(cs)
621
621
622 def children(repo, subset, x):
622 def children(repo, subset, x):
623 """``children(set)``
623 """``children(set)``
624 Child changesets of changesets in set.
624 Child changesets of changesets in set.
625 """
625 """
626 s = getset(repo, fullreposet(repo), x)
626 s = getset(repo, fullreposet(repo), x)
627 cs = _children(repo, subset, s)
627 cs = _children(repo, subset, s)
628 return subset & cs
628 return subset & cs
629
629
630 def closed(repo, subset, x):
630 def closed(repo, subset, x):
631 """``closed()``
631 """``closed()``
632 Changeset is closed.
632 Changeset is closed.
633 """
633 """
634 # i18n: "closed" is a keyword
634 # i18n: "closed" is a keyword
635 getargs(x, 0, 0, _("closed takes no arguments"))
635 getargs(x, 0, 0, _("closed takes no arguments"))
636 return subset.filter(lambda r: repo[r].closesbranch())
636 return subset.filter(lambda r: repo[r].closesbranch())
637
637
638 def contains(repo, subset, x):
638 def contains(repo, subset, x):
639 """``contains(pattern)``
639 """``contains(pattern)``
640 The revision's manifest contains a file matching pattern (but might not
640 The revision's manifest contains a file matching pattern (but might not
641 modify it). See :hg:`help patterns` for information about file patterns.
641 modify it). See :hg:`help patterns` for information about file patterns.
642
642
643 The pattern without explicit kind like ``glob:`` is expected to be
643 The pattern without explicit kind like ``glob:`` is expected to be
644 relative to the current directory and match against a file exactly
644 relative to the current directory and match against a file exactly
645 for efficiency.
645 for efficiency.
646 """
646 """
647 # i18n: "contains" is a keyword
647 # i18n: "contains" is a keyword
648 pat = getstring(x, _("contains requires a pattern"))
648 pat = getstring(x, _("contains requires a pattern"))
649
649
650 def matches(x):
650 def matches(x):
651 if not matchmod.patkind(pat):
651 if not matchmod.patkind(pat):
652 pats = pathutil.canonpath(repo.root, repo.getcwd(), pat)
652 pats = pathutil.canonpath(repo.root, repo.getcwd(), pat)
653 if pats in repo[x]:
653 if pats in repo[x]:
654 return True
654 return True
655 else:
655 else:
656 c = repo[x]
656 c = repo[x]
657 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=c)
657 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=c)
658 for f in c.manifest():
658 for f in c.manifest():
659 if m(f):
659 if m(f):
660 return True
660 return True
661 return False
661 return False
662
662
663 return subset.filter(matches)
663 return subset.filter(matches)
664
664
665 def converted(repo, subset, x):
665 def converted(repo, subset, x):
666 """``converted([id])``
666 """``converted([id])``
667 Changesets converted from the given identifier in the old repository if
667 Changesets converted from the given identifier in the old repository if
668 present, or all converted changesets if no identifier is specified.
668 present, or all converted changesets if no identifier is specified.
669 """
669 """
670
670
671 # There is exactly no chance of resolving the revision, so do a simple
671 # There is exactly no chance of resolving the revision, so do a simple
672 # string compare and hope for the best
672 # string compare and hope for the best
673
673
674 rev = None
674 rev = None
675 # i18n: "converted" is a keyword
675 # i18n: "converted" is a keyword
676 l = getargs(x, 0, 1, _('converted takes one or no arguments'))
676 l = getargs(x, 0, 1, _('converted takes one or no arguments'))
677 if l:
677 if l:
678 # i18n: "converted" is a keyword
678 # i18n: "converted" is a keyword
679 rev = getstring(l[0], _('converted requires a revision'))
679 rev = getstring(l[0], _('converted requires a revision'))
680
680
681 def _matchvalue(r):
681 def _matchvalue(r):
682 source = repo[r].extra().get('convert_revision', None)
682 source = repo[r].extra().get('convert_revision', None)
683 return source is not None and (rev is None or source.startswith(rev))
683 return source is not None and (rev is None or source.startswith(rev))
684
684
685 return subset.filter(lambda r: _matchvalue(r))
685 return subset.filter(lambda r: _matchvalue(r))
686
686
687 def date(repo, subset, x):
687 def date(repo, subset, x):
688 """``date(interval)``
688 """``date(interval)``
689 Changesets within the interval, see :hg:`help dates`.
689 Changesets within the interval, see :hg:`help dates`.
690 """
690 """
691 # i18n: "date" is a keyword
691 # i18n: "date" is a keyword
692 ds = getstring(x, _("date requires a string"))
692 ds = getstring(x, _("date requires a string"))
693 dm = util.matchdate(ds)
693 dm = util.matchdate(ds)
694 return subset.filter(lambda x: dm(repo[x].date()[0]))
694 return subset.filter(lambda x: dm(repo[x].date()[0]))
695
695
696 def desc(repo, subset, x):
696 def desc(repo, subset, x):
697 """``desc(string)``
697 """``desc(string)``
698 Search commit message for string. The match is case-insensitive.
698 Search commit message for string. The match is case-insensitive.
699 """
699 """
700 # i18n: "desc" is a keyword
700 # i18n: "desc" is a keyword
701 ds = encoding.lower(getstring(x, _("desc requires a string")))
701 ds = encoding.lower(getstring(x, _("desc requires a string")))
702
702
703 def matches(x):
703 def matches(x):
704 c = repo[x]
704 c = repo[x]
705 return ds in encoding.lower(c.description())
705 return ds in encoding.lower(c.description())
706
706
707 return subset.filter(matches)
707 return subset.filter(matches)
708
708
709 def _descendants(repo, subset, x, followfirst=False):
709 def _descendants(repo, subset, x, followfirst=False):
710 roots = getset(repo, fullreposet(repo), x)
710 roots = getset(repo, fullreposet(repo), x)
711 if not roots:
711 if not roots:
712 return baseset()
712 return baseset()
713 s = _revdescendants(repo, roots, followfirst)
713 s = _revdescendants(repo, roots, followfirst)
714
714
715 # Both sets need to be ascending in order to lazily return the union
715 # Both sets need to be ascending in order to lazily return the union
716 # in the correct order.
716 # in the correct order.
717 base = subset & roots
717 base = subset & roots
718 desc = subset & s
718 desc = subset & s
719 result = base + desc
719 result = base + desc
720 if subset.isascending():
720 if subset.isascending():
721 result.sort()
721 result.sort()
722 elif subset.isdescending():
722 elif subset.isdescending():
723 result.sort(reverse=True)
723 result.sort(reverse=True)
724 else:
724 else:
725 result = subset & result
725 result = subset & result
726 return result
726 return result
727
727
728 def descendants(repo, subset, x):
728 def descendants(repo, subset, x):
729 """``descendants(set)``
729 """``descendants(set)``
730 Changesets which are descendants of changesets in set.
730 Changesets which are descendants of changesets in set.
731 """
731 """
732 return _descendants(repo, subset, x)
732 return _descendants(repo, subset, x)
733
733
734 def _firstdescendants(repo, subset, x):
734 def _firstdescendants(repo, subset, x):
735 # ``_firstdescendants(set)``
735 # ``_firstdescendants(set)``
736 # Like ``descendants(set)`` but follows only the first parents.
736 # Like ``descendants(set)`` but follows only the first parents.
737 return _descendants(repo, subset, x, followfirst=True)
737 return _descendants(repo, subset, x, followfirst=True)
738
738
739 def destination(repo, subset, x):
739 def destination(repo, subset, x):
740 """``destination([set])``
740 """``destination([set])``
741 Changesets that were created by a graft, transplant or rebase operation,
741 Changesets that were created by a graft, transplant or rebase operation,
742 with the given revisions specified as the source. Omitting the optional set
742 with the given revisions specified as the source. Omitting the optional set
743 is the same as passing all().
743 is the same as passing all().
744 """
744 """
745 if x is not None:
745 if x is not None:
746 sources = getset(repo, fullreposet(repo), x)
746 sources = getset(repo, fullreposet(repo), x)
747 else:
747 else:
748 sources = fullreposet(repo)
748 sources = fullreposet(repo)
749
749
750 dests = set()
750 dests = set()
751
751
752 # subset contains all of the possible destinations that can be returned, so
752 # subset contains all of the possible destinations that can be returned, so
753 # iterate over them and see if their source(s) were provided in the arg set.
753 # iterate over them and see if their source(s) were provided in the arg set.
754 # Even if the immediate src of r is not in the arg set, src's source (or
754 # Even if the immediate src of r is not in the arg set, src's source (or
755 # further back) may be. Scanning back further than the immediate src allows
755 # further back) may be. Scanning back further than the immediate src allows
756 # transitive transplants and rebases to yield the same results as transitive
756 # transitive transplants and rebases to yield the same results as transitive
757 # grafts.
757 # grafts.
758 for r in subset:
758 for r in subset:
759 src = _getrevsource(repo, r)
759 src = _getrevsource(repo, r)
760 lineage = None
760 lineage = None
761
761
762 while src is not None:
762 while src is not None:
763 if lineage is None:
763 if lineage is None:
764 lineage = list()
764 lineage = list()
765
765
766 lineage.append(r)
766 lineage.append(r)
767
767
768 # The visited lineage is a match if the current source is in the arg
768 # The visited lineage is a match if the current source is in the arg
769 # set. Since every candidate dest is visited by way of iterating
769 # set. Since every candidate dest is visited by way of iterating
770 # subset, any dests further back in the lineage will be tested by a
770 # subset, any dests further back in the lineage will be tested by a
771 # different iteration over subset. Likewise, if the src was already
771 # different iteration over subset. Likewise, if the src was already
772 # selected, the current lineage can be selected without going back
772 # selected, the current lineage can be selected without going back
773 # further.
773 # further.
774 if src in sources or src in dests:
774 if src in sources or src in dests:
775 dests.update(lineage)
775 dests.update(lineage)
776 break
776 break
777
777
778 r = src
778 r = src
779 src = _getrevsource(repo, r)
779 src = _getrevsource(repo, r)
780
780
781 return subset.filter(dests.__contains__)
781 return subset.filter(dests.__contains__)
782
782
783 def divergent(repo, subset, x):
783 def divergent(repo, subset, x):
784 """``divergent()``
784 """``divergent()``
785 Final successors of changesets with an alternative set of final successors.
785 Final successors of changesets with an alternative set of final successors.
786 """
786 """
787 # i18n: "divergent" is a keyword
787 # i18n: "divergent" is a keyword
788 getargs(x, 0, 0, _("divergent takes no arguments"))
788 getargs(x, 0, 0, _("divergent takes no arguments"))
789 divergent = obsmod.getrevs(repo, 'divergent')
789 divergent = obsmod.getrevs(repo, 'divergent')
790 return subset & divergent
790 return subset & divergent
791
791
792 def draft(repo, subset, x):
792 def draft(repo, subset, x):
793 """``draft()``
793 """``draft()``
794 Changeset in draft phase."""
794 Changeset in draft phase."""
795 # i18n: "draft" is a keyword
795 # i18n: "draft" is a keyword
796 getargs(x, 0, 0, _("draft takes no arguments"))
796 getargs(x, 0, 0, _("draft takes no arguments"))
797 phase = repo._phasecache.phase
797 phase = repo._phasecache.phase
798 target = phases.draft
798 target = phases.draft
799 condition = lambda r: phase(repo, r) == target
799 condition = lambda r: phase(repo, r) == target
800 return subset.filter(condition, cache=False)
800 return subset.filter(condition, cache=False)
801
801
802 def extinct(repo, subset, x):
802 def extinct(repo, subset, x):
803 """``extinct()``
803 """``extinct()``
804 Obsolete changesets with obsolete descendants only.
804 Obsolete changesets with obsolete descendants only.
805 """
805 """
806 # i18n: "extinct" is a keyword
806 # i18n: "extinct" is a keyword
807 getargs(x, 0, 0, _("extinct takes no arguments"))
807 getargs(x, 0, 0, _("extinct takes no arguments"))
808 extincts = obsmod.getrevs(repo, 'extinct')
808 extincts = obsmod.getrevs(repo, 'extinct')
809 return subset & extincts
809 return subset & extincts
810
810
811 def extra(repo, subset, x):
811 def extra(repo, subset, x):
812 """``extra(label, [value])``
812 """``extra(label, [value])``
813 Changesets with the given label in the extra metadata, with the given
813 Changesets with the given label in the extra metadata, with the given
814 optional value.
814 optional value.
815
815
816 If `value` starts with `re:`, the remainder of the value is treated as
816 If `value` starts with `re:`, the remainder of the value is treated as
817 a regular expression. To match a value that actually starts with `re:`,
817 a regular expression. To match a value that actually starts with `re:`,
818 use the prefix `literal:`.
818 use the prefix `literal:`.
819 """
819 """
820
820
821 # i18n: "extra" is a keyword
821 # i18n: "extra" is a keyword
822 l = getargs(x, 1, 2, _('extra takes at least 1 and at most 2 arguments'))
822 l = getargs(x, 1, 2, _('extra takes at least 1 and at most 2 arguments'))
823 # i18n: "extra" is a keyword
823 # i18n: "extra" is a keyword
824 label = getstring(l[0], _('first argument to extra must be a string'))
824 label = getstring(l[0], _('first argument to extra must be a string'))
825 value = None
825 value = None
826
826
827 if len(l) > 1:
827 if len(l) > 1:
828 # i18n: "extra" is a keyword
828 # i18n: "extra" is a keyword
829 value = getstring(l[1], _('second argument to extra must be a string'))
829 value = getstring(l[1], _('second argument to extra must be a string'))
830 kind, value, matcher = _stringmatcher(value)
830 kind, value, matcher = _stringmatcher(value)
831
831
832 def _matchvalue(r):
832 def _matchvalue(r):
833 extra = repo[r].extra()
833 extra = repo[r].extra()
834 return label in extra and (value is None or matcher(extra[label]))
834 return label in extra and (value is None or matcher(extra[label]))
835
835
836 return subset.filter(lambda r: _matchvalue(r))
836 return subset.filter(lambda r: _matchvalue(r))
837
837
838 def filelog(repo, subset, x):
838 def filelog(repo, subset, x):
839 """``filelog(pattern)``
839 """``filelog(pattern)``
840 Changesets connected to the specified filelog.
840 Changesets connected to the specified filelog.
841
841
842 For performance reasons, visits only revisions mentioned in the file-level
842 For performance reasons, visits only revisions mentioned in the file-level
843 filelog, rather than filtering through all changesets (much faster, but
843 filelog, rather than filtering through all changesets (much faster, but
844 doesn't include deletes or duplicate changes). For a slower, more accurate
844 doesn't include deletes or duplicate changes). For a slower, more accurate
845 result, use ``file()``.
845 result, use ``file()``.
846
846
847 The pattern without explicit kind like ``glob:`` is expected to be
847 The pattern without explicit kind like ``glob:`` is expected to be
848 relative to the current directory and match against a file exactly
848 relative to the current directory and match against a file exactly
849 for efficiency.
849 for efficiency.
850
850
851 If some linkrev points to revisions filtered by the current repoview, we'll
851 If some linkrev points to revisions filtered by the current repoview, we'll
852 work around it to return a non-filtered value.
852 work around it to return a non-filtered value.
853 """
853 """
854
854
855 # i18n: "filelog" is a keyword
855 # i18n: "filelog" is a keyword
856 pat = getstring(x, _("filelog requires a pattern"))
856 pat = getstring(x, _("filelog requires a pattern"))
857 s = set()
857 s = set()
858 cl = repo.changelog
858 cl = repo.changelog
859
859
860 if not matchmod.patkind(pat):
860 if not matchmod.patkind(pat):
861 f = pathutil.canonpath(repo.root, repo.getcwd(), pat)
861 f = pathutil.canonpath(repo.root, repo.getcwd(), pat)
862 files = [f]
862 files = [f]
863 else:
863 else:
864 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=repo[None])
864 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=repo[None])
865 files = (f for f in repo[None] if m(f))
865 files = (f for f in repo[None] if m(f))
866
866
867 for f in files:
867 for f in files:
868 backrevref = {} # final value for: filerev -> changerev
868 backrevref = {} # final value for: filerev -> changerev
869 lowestchild = {} # lowest known filerev child of a filerev
869 lowestchild = {} # lowest known filerev child of a filerev
870 delayed = [] # filerev with filtered linkrev, for post-processing
870 delayed = [] # filerev with filtered linkrev, for post-processing
871 lowesthead = None # cache for manifest content of all head revisions
871 lowesthead = None # cache for manifest content of all head revisions
872 fl = repo.file(f)
872 fl = repo.file(f)
873 for fr in list(fl):
873 for fr in list(fl):
874 rev = fl.linkrev(fr)
874 rev = fl.linkrev(fr)
875 if rev not in cl:
875 if rev not in cl:
876 # changerev pointed in linkrev is filtered
876 # changerev pointed in linkrev is filtered
877 # record it for post processing.
877 # record it for post processing.
878 delayed.append((fr, rev))
878 delayed.append((fr, rev))
879 continue
879 continue
880 for p in fl.parentrevs(fr):
880 for p in fl.parentrevs(fr):
881 if 0 <= p and p not in lowestchild:
881 if 0 <= p and p not in lowestchild:
882 lowestchild[p] = fr
882 lowestchild[p] = fr
883 backrevref[fr] = rev
883 backrevref[fr] = rev
884 s.add(rev)
884 s.add(rev)
885
885
886 # Post-processing of all filerevs we skipped because they were
886 # Post-processing of all filerevs we skipped because they were
887 # filtered. If such filerevs have known and unfiltered children, this
887 # filtered. If such filerevs have known and unfiltered children, this
888 # means they have an unfiltered appearance out there. We'll use linkrev
888 # means they have an unfiltered appearance out there. We'll use linkrev
889 # adjustment to find one of these appearances. The lowest known child
889 # adjustment to find one of these appearances. The lowest known child
890 # will be used as a starting point because it is the best upper-bound we
890 # will be used as a starting point because it is the best upper-bound we
891 # have.
891 # have.
892 #
892 #
893 # This approach will fail when an unfiltered but linkrev-shadowed
893 # This approach will fail when an unfiltered but linkrev-shadowed
894 # appearance exists in a head changeset without unfiltered filerev
894 # appearance exists in a head changeset without unfiltered filerev
895 # children anywhere.
895 # children anywhere.
896 while delayed:
896 while delayed:
897 # must be a descending iteration. To slowly fill lowest child
897 # must be a descending iteration. To slowly fill lowest child
898 # information that is of potential use by the next item.
898 # information that is of potential use by the next item.
899 fr, rev = delayed.pop()
899 fr, rev = delayed.pop()
900 lkr = rev
900 lkr = rev
901
901
902 child = lowestchild.get(fr)
902 child = lowestchild.get(fr)
903
903
904 if child is None:
904 if child is None:
905 # search for existence of this file revision in a head revision.
905 # search for existence of this file revision in a head revision.
906 # There are three possibilities:
906 # There are three possibilities:
907 # - the revision exists in a head and we can find an
907 # - the revision exists in a head and we can find an
908 # introduction from there,
908 # introduction from there,
909 # - the revision does not exist in a head because it has been
909 # - the revision does not exist in a head because it has been
910 # changed since its introduction: we would have found a child
910 # changed since its introduction: we would have found a child
911 # and be in the other 'else' clause,
911 # and be in the other 'else' clause,
912 # - all versions of the revision are hidden.
912 # - all versions of the revision are hidden.
913 if lowesthead is None:
913 if lowesthead is None:
914 lowesthead = {}
914 lowesthead = {}
915 for h in repo.heads():
915 for h in repo.heads():
916 fnode = repo[h].manifest().get(f)
916 fnode = repo[h].manifest().get(f)
917 if fnode is not None:
917 if fnode is not None:
918 lowesthead[fl.rev(fnode)] = h
918 lowesthead[fl.rev(fnode)] = h
919 headrev = lowesthead.get(fr)
919 headrev = lowesthead.get(fr)
920 if headrev is None:
920 if headrev is None:
921 # content is nowhere unfiltered
921 # content is nowhere unfiltered
922 continue
922 continue
923 rev = repo[headrev][f].introrev()
923 rev = repo[headrev][f].introrev()
924 else:
924 else:
925 # the lowest known child is a good upper bound
925 # the lowest known child is a good upper bound
926 childcrev = backrevref[child]
926 childcrev = backrevref[child]
927 # XXX this does not guarantee returning the lowest
927 # XXX this does not guarantee returning the lowest
928 # introduction of this revision, but this gives a
928 # introduction of this revision, but this gives a
929 # result which is a good start and will fit in most
929 # result which is a good start and will fit in most
930 # cases. We probably need to fix the multiple
930 # cases. We probably need to fix the multiple
931 # introductions case properly (report each
931 # introductions case properly (report each
932 # introduction, even for identical file revisions)
932 # introduction, even for identical file revisions)
933 # once and for all at some point anyway.
933 # once and for all at some point anyway.
934 for p in repo[childcrev][f].parents():
934 for p in repo[childcrev][f].parents():
935 if p.filerev() == fr:
935 if p.filerev() == fr:
936 rev = p.rev()
936 rev = p.rev()
937 break
937 break
938 if rev == lkr: # no shadowed entry found
938 if rev == lkr: # no shadowed entry found
939 # XXX This should never happen unless some manifest points
939 # XXX This should never happen unless some manifest points
940 # to biggish file revisions (like a revision that uses a
940 # to biggish file revisions (like a revision that uses a
941 # parent that never appears in the manifest ancestors)
941 # parent that never appears in the manifest ancestors)
942 continue
942 continue
943
943
944 # Fill the data for the next iteration.
944 # Fill the data for the next iteration.
945 for p in fl.parentrevs(fr):
945 for p in fl.parentrevs(fr):
946 if 0 <= p and p not in lowestchild:
946 if 0 <= p and p not in lowestchild:
947 lowestchild[p] = fr
947 lowestchild[p] = fr
948 backrevref[fr] = rev
948 backrevref[fr] = rev
949 s.add(rev)
949 s.add(rev)
950
950
951 return subset & s
951 return subset & s
952
952
953 def first(repo, subset, x):
953 def first(repo, subset, x):
954 """``first(set, [n])``
954 """``first(set, [n])``
955 An alias for limit().
955 An alias for limit().
956 """
956 """
957 return limit(repo, subset, x)
957 return limit(repo, subset, x)
958
958
959 def _follow(repo, subset, x, name, followfirst=False):
959 def _follow(repo, subset, x, name, followfirst=False):
960 l = getargs(x, 0, 1, _("%s takes no arguments or a filename") % name)
960 l = getargs(x, 0, 1, _("%s takes no arguments or a filename") % name)
961 c = repo['.']
961 c = repo['.']
962 if l:
962 if l:
963 x = getstring(l[0], _("%s expected a filename") % name)
963 x = getstring(l[0], _("%s expected a filename") % name)
964 if x in c:
964 if x in c:
965 cx = c[x]
965 cx = c[x]
966 s = set(ctx.rev() for ctx in cx.ancestors(followfirst=followfirst))
966 s = set(ctx.rev() for ctx in cx.ancestors(followfirst=followfirst))
967 # include the revision responsible for the most recent version
967 # include the revision responsible for the most recent version
968 s.add(cx.introrev())
968 s.add(cx.introrev())
969 else:
969 else:
970 return baseset()
970 return baseset()
971 else:
971 else:
972 s = _revancestors(repo, baseset([c.rev()]), followfirst)
972 s = _revancestors(repo, baseset([c.rev()]), followfirst)
973
973
974 return subset & s
974 return subset & s
975
975
976 def follow(repo, subset, x):
976 def follow(repo, subset, x):
977 """``follow([file])``
977 """``follow([file])``
978 An alias for ``::.`` (ancestors of the working directory's first parent).
978 An alias for ``::.`` (ancestors of the working directory's first parent).
979 If a filename is specified, the history of the given file is followed,
979 If a filename is specified, the history of the given file is followed,
980 including copies.
980 including copies.
981 """
981 """
982 return _follow(repo, subset, x, 'follow')
982 return _follow(repo, subset, x, 'follow')
983
983
984 def _followfirst(repo, subset, x):
984 def _followfirst(repo, subset, x):
985 # ``followfirst([file])``
985 # ``followfirst([file])``
986 # Like ``follow([file])`` but follows only the first parent of
986 # Like ``follow([file])`` but follows only the first parent of
987 # every revision or file revision.
987 # every revision or file revision.
988 return _follow(repo, subset, x, '_followfirst', followfirst=True)
988 return _follow(repo, subset, x, '_followfirst', followfirst=True)
989
989
990 def getall(repo, subset, x):
990 def getall(repo, subset, x):
991 """``all()``
991 """``all()``
992 All changesets, the same as ``0:tip``.
992 All changesets, the same as ``0:tip``.
993 """
993 """
994 # i18n: "all" is a keyword
994 # i18n: "all" is a keyword
995 getargs(x, 0, 0, _("all takes no arguments"))
995 getargs(x, 0, 0, _("all takes no arguments"))
996 return subset & spanset(repo) # drop "null" if any
996 return subset & spanset(repo) # drop "null" if any
997
997
998 def grep(repo, subset, x):
998 def grep(repo, subset, x):
999 """``grep(regex)``
999 """``grep(regex)``
1000 Like ``keyword(string)`` but accepts a regex. Use ``grep(r'...')``
1000 Like ``keyword(string)`` but accepts a regex. Use ``grep(r'...')``
1001 to ensure special escape characters are handled correctly. Unlike
1001 to ensure special escape characters are handled correctly. Unlike
1002 ``keyword(string)``, the match is case-sensitive.
1002 ``keyword(string)``, the match is case-sensitive.
1003 """
1003 """
1004 try:
1004 try:
1005 # i18n: "grep" is a keyword
1005 # i18n: "grep" is a keyword
1006 gr = re.compile(getstring(x, _("grep requires a string")))
1006 gr = re.compile(getstring(x, _("grep requires a string")))
1007 except re.error, e:
1007 except re.error, e:
1008 raise error.ParseError(_('invalid match pattern: %s') % e)
1008 raise error.ParseError(_('invalid match pattern: %s') % e)
1009
1009
1010 def matches(x):
1010 def matches(x):
1011 c = repo[x]
1011 c = repo[x]
1012 for e in c.files() + [c.user(), c.description()]:
1012 for e in c.files() + [c.user(), c.description()]:
1013 if gr.search(e):
1013 if gr.search(e):
1014 return True
1014 return True
1015 return False
1015 return False
1016
1016
1017 return subset.filter(matches)
1017 return subset.filter(matches)
1018
1018
1019 def _matchfiles(repo, subset, x):
1019 def _matchfiles(repo, subset, x):
1020 # _matchfiles takes a revset list of prefixed arguments:
1020 # _matchfiles takes a revset list of prefixed arguments:
1021 #
1021 #
1022 # [p:foo, i:bar, x:baz]
1022 # [p:foo, i:bar, x:baz]
1023 #
1023 #
1024 # builds a match object from them and filters subset. Allowed
1024 # builds a match object from them and filters subset. Allowed
1025 # prefixes are 'p:' for regular patterns, 'i:' for include
1025 # prefixes are 'p:' for regular patterns, 'i:' for include
1026 # patterns and 'x:' for exclude patterns. Use 'r:' prefix to pass
1026 # patterns and 'x:' for exclude patterns. Use 'r:' prefix to pass
1027 # a revision identifier, or the empty string to reference the
1027 # a revision identifier, or the empty string to reference the
1028 # working directory, from which the match object is
1028 # working directory, from which the match object is
1029 # initialized. Use 'd:' to set the default matching mode, default
1029 # initialized. Use 'd:' to set the default matching mode, default
1030 # to 'glob'. At most one 'r:' and 'd:' argument can be passed.
1030 # to 'glob'. At most one 'r:' and 'd:' argument can be passed.
1031
1031
1032 # i18n: "_matchfiles" is a keyword
1032 # i18n: "_matchfiles" is a keyword
1033 l = getargs(x, 1, -1, _("_matchfiles requires at least one argument"))
1033 l = getargs(x, 1, -1, _("_matchfiles requires at least one argument"))
1034 pats, inc, exc = [], [], []
1034 pats, inc, exc = [], [], []
1035 rev, default = None, None
1035 rev, default = None, None
1036 for arg in l:
1036 for arg in l:
1037 # i18n: "_matchfiles" is a keyword
1037 # i18n: "_matchfiles" is a keyword
1038 s = getstring(arg, _("_matchfiles requires string arguments"))
1038 s = getstring(arg, _("_matchfiles requires string arguments"))
1039 prefix, value = s[:2], s[2:]
1039 prefix, value = s[:2], s[2:]
1040 if prefix == 'p:':
1040 if prefix == 'p:':
1041 pats.append(value)
1041 pats.append(value)
1042 elif prefix == 'i:':
1042 elif prefix == 'i:':
1043 inc.append(value)
1043 inc.append(value)
1044 elif prefix == 'x:':
1044 elif prefix == 'x:':
1045 exc.append(value)
1045 exc.append(value)
1046 elif prefix == 'r:':
1046 elif prefix == 'r:':
1047 if rev is not None:
1047 if rev is not None:
1048 # i18n: "_matchfiles" is a keyword
1048 # i18n: "_matchfiles" is a keyword
1049 raise error.ParseError(_('_matchfiles expected at most one '
1049 raise error.ParseError(_('_matchfiles expected at most one '
1050 'revision'))
1050 'revision'))
1051 if value != '': # empty means working directory; leave rev as None
1051 if value != '': # empty means working directory; leave rev as None
1052 rev = value
1052 rev = value
1053 elif prefix == 'd:':
1053 elif prefix == 'd:':
1054 if default is not None:
1054 if default is not None:
1055 # i18n: "_matchfiles" is a keyword
1055 # i18n: "_matchfiles" is a keyword
1056 raise error.ParseError(_('_matchfiles expected at most one '
1056 raise error.ParseError(_('_matchfiles expected at most one '
1057 'default mode'))
1057 'default mode'))
1058 default = value
1058 default = value
1059 else:
1059 else:
1060 # i18n: "_matchfiles" is a keyword
1060 # i18n: "_matchfiles" is a keyword
1061 raise error.ParseError(_('invalid _matchfiles prefix: %s') % prefix)
1061 raise error.ParseError(_('invalid _matchfiles prefix: %s') % prefix)
1062 if not default:
1062 if not default:
1063 default = 'glob'
1063 default = 'glob'
1064
1064
1065 m = matchmod.match(repo.root, repo.getcwd(), pats, include=inc,
1065 m = matchmod.match(repo.root, repo.getcwd(), pats, include=inc,
1066 exclude=exc, ctx=repo[rev], default=default)
1066 exclude=exc, ctx=repo[rev], default=default)
1067
1067
1068 def matches(x):
1068 def matches(x):
1069 for f in repo[x].files():
1069 for f in repo[x].files():
1070 if m(f):
1070 if m(f):
1071 return True
1071 return True
1072 return False
1072 return False
1073
1073
1074 return subset.filter(matches)
1074 return subset.filter(matches)
1075
1075
1076 def hasfile(repo, subset, x):
1076 def hasfile(repo, subset, x):
1077 """``file(pattern)``
1077 """``file(pattern)``
1078 Changesets affecting files matched by pattern.
1078 Changesets affecting files matched by pattern.
1079
1079
1080 For a faster but less accurate result, consider using ``filelog()``
1080 For a faster but less accurate result, consider using ``filelog()``
1081 instead.
1081 instead.
1082
1082
1083 This predicate uses ``glob:`` as the default kind of pattern.
1083 This predicate uses ``glob:`` as the default kind of pattern.
1084 """
1084 """
1085 # i18n: "file" is a keyword
1085 # i18n: "file" is a keyword
1086 pat = getstring(x, _("file requires a pattern"))
1086 pat = getstring(x, _("file requires a pattern"))
1087 return _matchfiles(repo, subset, ('string', 'p:' + pat))
1087 return _matchfiles(repo, subset, ('string', 'p:' + pat))
1088
1088
1089 def head(repo, subset, x):
1089 def head(repo, subset, x):
1090 """``head()``
1090 """``head()``
1091 Changeset is a named branch head.
1091 Changeset is a named branch head.
1092 """
1092 """
1093 # i18n: "head" is a keyword
1093 # i18n: "head" is a keyword
1094 getargs(x, 0, 0, _("head takes no arguments"))
1094 getargs(x, 0, 0, _("head takes no arguments"))
1095 hs = set()
1095 hs = set()
1096 for b, ls in repo.branchmap().iteritems():
1096 for b, ls in repo.branchmap().iteritems():
1097 hs.update(repo[h].rev() for h in ls)
1097 hs.update(repo[h].rev() for h in ls)
1098 return baseset(hs).filter(subset.__contains__)
1098 return baseset(hs).filter(subset.__contains__)
1099
1099
1100 def heads(repo, subset, x):
1100 def heads(repo, subset, x):
1101 """``heads(set)``
1101 """``heads(set)``
1102 Members of set with no children in set.
1102 Members of set with no children in set.
1103 """
1103 """
1104 s = getset(repo, subset, x)
1104 s = getset(repo, subset, x)
1105 ps = parents(repo, subset, x)
1105 ps = parents(repo, subset, x)
1106 return s - ps
1106 return s - ps
1107
1107
1108 def hidden(repo, subset, x):
1108 def hidden(repo, subset, x):
1109 """``hidden()``
1109 """``hidden()``
1110 Hidden changesets.
1110 Hidden changesets.
1111 """
1111 """
1112 # i18n: "hidden" is a keyword
1112 # i18n: "hidden" is a keyword
1113 getargs(x, 0, 0, _("hidden takes no arguments"))
1113 getargs(x, 0, 0, _("hidden takes no arguments"))
1114 hiddenrevs = repoview.filterrevs(repo, 'visible')
1114 hiddenrevs = repoview.filterrevs(repo, 'visible')
1115 return subset & hiddenrevs
1115 return subset & hiddenrevs
1116
1116
1117 def keyword(repo, subset, x):
1117 def keyword(repo, subset, x):
1118 """``keyword(string)``
1118 """``keyword(string)``
1119 Search commit message, user name, and names of changed files for
1119 Search commit message, user name, and names of changed files for
1120 string. The match is case-insensitive.
1120 string. The match is case-insensitive.
1121 """
1121 """
1122 # i18n: "keyword" is a keyword
1122 # i18n: "keyword" is a keyword
1123 kw = encoding.lower(getstring(x, _("keyword requires a string")))
1123 kw = encoding.lower(getstring(x, _("keyword requires a string")))
1124
1124
1125 def matches(r):
1125 def matches(r):
1126 c = repo[r]
1126 c = repo[r]
1127 return util.any(kw in encoding.lower(t) for t in c.files() + [c.user(),
1127 return util.any(kw in encoding.lower(t) for t in c.files() + [c.user(),
1128 c.description()])
1128 c.description()])
1129
1129
1130 return subset.filter(matches)
1130 return subset.filter(matches)
1131
1131
1132 def limit(repo, subset, x):
1132 def limit(repo, subset, x):
1133 """``limit(set, [n])``
1133 """``limit(set, [n])``
1134 First n members of set, defaulting to 1.
1134 First n members of set, defaulting to 1.
1135 """
1135 """
1136 # i18n: "limit" is a keyword
1136 # i18n: "limit" is a keyword
1137 l = getargs(x, 1, 2, _("limit requires one or two arguments"))
1137 l = getargs(x, 1, 2, _("limit requires one or two arguments"))
1138 try:
1138 try:
1139 lim = 1
1139 lim = 1
1140 if len(l) == 2:
1140 if len(l) == 2:
1141 # i18n: "limit" is a keyword
1141 # i18n: "limit" is a keyword
1142 lim = int(getstring(l[1], _("limit requires a number")))
1142 lim = int(getstring(l[1], _("limit requires a number")))
1143 except (TypeError, ValueError):
1143 except (TypeError, ValueError):
1144 # i18n: "limit" is a keyword
1144 # i18n: "limit" is a keyword
1145 raise error.ParseError(_("limit expects a number"))
1145 raise error.ParseError(_("limit expects a number"))
1146 ss = subset
1146 ss = subset
1147 os = getset(repo, fullreposet(repo), l[0])
1147 os = getset(repo, fullreposet(repo), l[0])
1148 result = []
1148 result = []
1149 it = iter(os)
1149 it = iter(os)
1150 for x in xrange(lim):
1150 for x in xrange(lim):
1151 try:
1151 try:
1152 y = it.next()
1152 y = it.next()
1153 if y in ss:
1153 if y in ss:
1154 result.append(y)
1154 result.append(y)
1155 except (StopIteration):
1155 except (StopIteration):
1156 break
1156 break
1157 return baseset(result)
1157 return baseset(result)
1158
1158
1159 def last(repo, subset, x):
1159 def last(repo, subset, x):
1160 """``last(set, [n])``
1160 """``last(set, [n])``
1161 Last n members of set, defaulting to 1.
1161 Last n members of set, defaulting to 1.
1162 """
1162 """
1163 # i18n: "last" is a keyword
1163 # i18n: "last" is a keyword
1164 l = getargs(x, 1, 2, _("last requires one or two arguments"))
1164 l = getargs(x, 1, 2, _("last requires one or two arguments"))
1165 try:
1165 try:
1166 lim = 1
1166 lim = 1
1167 if len(l) == 2:
1167 if len(l) == 2:
1168 # i18n: "last" is a keyword
1168 # i18n: "last" is a keyword
1169 lim = int(getstring(l[1], _("last requires a number")))
1169 lim = int(getstring(l[1], _("last requires a number")))
1170 except (TypeError, ValueError):
1170 except (TypeError, ValueError):
1171 # i18n: "last" is a keyword
1171 # i18n: "last" is a keyword
1172 raise error.ParseError(_("last expects a number"))
1172 raise error.ParseError(_("last expects a number"))
1173 ss = subset
1173 ss = subset
1174 os = getset(repo, fullreposet(repo), l[0])
1174 os = getset(repo, fullreposet(repo), l[0])
1175 os.reverse()
1175 os.reverse()
1176 result = []
1176 result = []
1177 it = iter(os)
1177 it = iter(os)
1178 for x in xrange(lim):
1178 for x in xrange(lim):
1179 try:
1179 try:
1180 y = it.next()
1180 y = it.next()
1181 if y in ss:
1181 if y in ss:
1182 result.append(y)
1182 result.append(y)
1183 except (StopIteration):
1183 except (StopIteration):
1184 break
1184 break
1185 return baseset(result)
1185 return baseset(result)
1186
1186
1187 def maxrev(repo, subset, x):
1187 def maxrev(repo, subset, x):
1188 """``max(set)``
1188 """``max(set)``
1189 Changeset with highest revision number in set.
1189 Changeset with highest revision number in set.
1190 """
1190 """
1191 os = getset(repo, fullreposet(repo), x)
1191 os = getset(repo, fullreposet(repo), x)
1192 if os:
1192 if os:
1193 m = os.max()
1193 m = os.max()
1194 if m in subset:
1194 if m in subset:
1195 return baseset([m])
1195 return baseset([m])
1196 return baseset()
1196 return baseset()
1197
1197
1198 def merge(repo, subset, x):
1198 def merge(repo, subset, x):
1199 """``merge()``
1199 """``merge()``
1200 Changeset is a merge changeset.
1200 Changeset is a merge changeset.
1201 """
1201 """
1202 # i18n: "merge" is a keyword
1202 # i18n: "merge" is a keyword
1203 getargs(x, 0, 0, _("merge takes no arguments"))
1203 getargs(x, 0, 0, _("merge takes no arguments"))
1204 cl = repo.changelog
1204 cl = repo.changelog
1205 return subset.filter(lambda r: cl.parentrevs(r)[1] != -1)
1205 return subset.filter(lambda r: cl.parentrevs(r)[1] != -1)
1206
1206
1207 def branchpoint(repo, subset, x):
1207 def branchpoint(repo, subset, x):
1208 """``branchpoint()``
1208 """``branchpoint()``
1209 Changesets with more than one child.
1209 Changesets with more than one child.
1210 """
1210 """
1211 # i18n: "branchpoint" is a keyword
1211 # i18n: "branchpoint" is a keyword
1212 getargs(x, 0, 0, _("branchpoint takes no arguments"))
1212 getargs(x, 0, 0, _("branchpoint takes no arguments"))
1213 cl = repo.changelog
1213 cl = repo.changelog
1214 if not subset:
1214 if not subset:
1215 return baseset()
1215 return baseset()
1216 baserev = min(subset)
1216 baserev = min(subset)
1217 parentscount = [0]*(len(repo) - baserev)
1217 parentscount = [0]*(len(repo) - baserev)
1218 for r in cl.revs(start=baserev + 1):
1218 for r in cl.revs(start=baserev + 1):
1219 for p in cl.parentrevs(r):
1219 for p in cl.parentrevs(r):
1220 if p >= baserev:
1220 if p >= baserev:
1221 parentscount[p - baserev] += 1
1221 parentscount[p - baserev] += 1
1222 return subset.filter(lambda r: parentscount[r - baserev] > 1)
1222 return subset.filter(lambda r: parentscount[r - baserev] > 1)
1223
1223
1224 def minrev(repo, subset, x):
1224 def minrev(repo, subset, x):
1225 """``min(set)``
1225 """``min(set)``
1226 Changeset with lowest revision number in set.
1226 Changeset with lowest revision number in set.
1227 """
1227 """
1228 os = getset(repo, fullreposet(repo), x)
1228 os = getset(repo, fullreposet(repo), x)
1229 if os:
1229 if os:
1230 m = os.min()
1230 m = os.min()
1231 if m in subset:
1231 if m in subset:
1232 return baseset([m])
1232 return baseset([m])
1233 return baseset()
1233 return baseset()
1234
1234
1235 def modifies(repo, subset, x):
1235 def modifies(repo, subset, x):
1236 """``modifies(pattern)``
1236 """``modifies(pattern)``
1237 Changesets modifying files matched by pattern.
1237 Changesets modifying files matched by pattern.
1238
1238
1239 The pattern without explicit kind like ``glob:`` is expected to be
1239 The pattern without explicit kind like ``glob:`` is expected to be
1240 relative to the current directory and match against a file or a
1240 relative to the current directory and match against a file or a
1241 directory.
1241 directory.
1242 """
1242 """
1243 # i18n: "modifies" is a keyword
1243 # i18n: "modifies" is a keyword
1244 pat = getstring(x, _("modifies requires a pattern"))
1244 pat = getstring(x, _("modifies requires a pattern"))
1245 return checkstatus(repo, subset, pat, 0)
1245 return checkstatus(repo, subset, pat, 0)
1246
1246
1247 def named(repo, subset, x):
1247 def named(repo, subset, x):
1248 """``named(namespace)``
1248 """``named(namespace)``
1249 The changesets in a given namespace.
1249 The changesets in a given namespace.
1250
1250
1251 If `namespace` starts with `re:`, the remainder of the string is treated as
1251 If `namespace` starts with `re:`, the remainder of the string is treated as
1252 a regular expression. To match a namespace that actually starts with `re:`,
1252 a regular expression. To match a namespace that actually starts with `re:`,
1253 use the prefix `literal:`.
1253 use the prefix `literal:`.
1254 """
1254 """
1255 # i18n: "named" is a keyword
1255 # i18n: "named" is a keyword
1256 args = getargs(x, 1, 1, _('named requires a namespace argument'))
1256 args = getargs(x, 1, 1, _('named requires a namespace argument'))
1257
1257
1258 ns = getstring(args[0],
1258 ns = getstring(args[0],
1259 # i18n: "named" is a keyword
1259 # i18n: "named" is a keyword
1260 _('the argument to named must be a string'))
1260 _('the argument to named must be a string'))
1261 kind, pattern, matcher = _stringmatcher(ns)
1261 kind, pattern, matcher = _stringmatcher(ns)
1262 namespaces = set()
1262 namespaces = set()
1263 if kind == 'literal':
1263 if kind == 'literal':
1264 if pattern not in repo.names:
1264 if pattern not in repo.names:
1265 raise error.RepoLookupError(_("namespace '%s' does not exist")
1265 raise error.RepoLookupError(_("namespace '%s' does not exist")
1266 % ns)
1266 % ns)
1267 namespaces.add(repo.names[pattern])
1267 namespaces.add(repo.names[pattern])
1268 else:
1268 else:
1269 for name, ns in repo.names.iteritems():
1269 for name, ns in repo.names.iteritems():
1270 if matcher(name):
1270 if matcher(name):
1271 namespaces.add(ns)
1271 namespaces.add(ns)
1272 if not namespaces:
1272 if not namespaces:
1273 raise error.RepoLookupError(_("no namespace exists"
1273 raise error.RepoLookupError(_("no namespace exists"
1274 " that match '%s'") % pattern)
1274 " that match '%s'") % pattern)
1275
1275
1276 names = set()
1276 names = set()
1277 for ns in namespaces:
1277 for ns in namespaces:
1278 for name in ns.listnames(repo):
1278 for name in ns.listnames(repo):
1279 if name not in ns.deprecated:
1279 if name not in ns.deprecated:
1280 names.update(repo[n].rev() for n in ns.nodes(repo, name))
1280 names.update(repo[n].rev() for n in ns.nodes(repo, name))
1281
1281
1282 names -= set([node.nullrev])
1282 names -= set([node.nullrev])
1283 return subset & names
1283 return subset & names
1284
1284
1285 def node_(repo, subset, x):
1285 def node_(repo, subset, x):
1286 """``id(string)``
1286 """``id(string)``
1287 Revision non-ambiguously specified by the given hex string prefix.
1287 Revision non-ambiguously specified by the given hex string prefix.
1288 """
1288 """
1289 # i18n: "id" is a keyword
1289 # i18n: "id" is a keyword
1290 l = getargs(x, 1, 1, _("id requires one argument"))
1290 l = getargs(x, 1, 1, _("id requires one argument"))
1291 # i18n: "id" is a keyword
1291 # i18n: "id" is a keyword
1292 n = getstring(l[0], _("id requires a string"))
1292 n = getstring(l[0], _("id requires a string"))
1293 if len(n) == 40:
1293 if len(n) == 40:
1294 try:
1294 try:
1295 rn = repo.changelog.rev(node.bin(n))
1295 rn = repo.changelog.rev(node.bin(n))
1296 except (LookupError, TypeError):
1296 except (LookupError, TypeError):
1297 rn = None
1297 rn = None
1298 else:
1298 else:
1299 rn = None
1299 rn = None
1300 pm = repo.changelog._partialmatch(n)
1300 pm = repo.changelog._partialmatch(n)
1301 if pm is not None:
1301 if pm is not None:
1302 rn = repo.changelog.rev(pm)
1302 rn = repo.changelog.rev(pm)
1303
1303
1304 if rn is None:
1304 if rn is None:
1305 return baseset()
1305 return baseset()
1306 result = baseset([rn])
1306 result = baseset([rn])
1307 return result & subset
1307 return result & subset
1308
1308
1309 def obsolete(repo, subset, x):
1309 def obsolete(repo, subset, x):
1310 """``obsolete()``
1310 """``obsolete()``
1311 Mutable changeset with a newer version."""
1311 Mutable changeset with a newer version."""
1312 # i18n: "obsolete" is a keyword
1312 # i18n: "obsolete" is a keyword
1313 getargs(x, 0, 0, _("obsolete takes no arguments"))
1313 getargs(x, 0, 0, _("obsolete takes no arguments"))
1314 obsoletes = obsmod.getrevs(repo, 'obsolete')
1314 obsoletes = obsmod.getrevs(repo, 'obsolete')
1315 return subset & obsoletes
1315 return subset & obsoletes
1316
1316
1317 def only(repo, subset, x):
1317 def only(repo, subset, x):
1318 """``only(set, [set])``
1318 """``only(set, [set])``
1319 Changesets that are ancestors of the first set that are not ancestors
1319 Changesets that are ancestors of the first set that are not ancestors
1320 of any other head in the repo. If a second set is specified, the result
1320 of any other head in the repo. If a second set is specified, the result
1321 is ancestors of the first set that are not ancestors of the second set
1321 is ancestors of the first set that are not ancestors of the second set
1322 (i.e. ::<set1> - ::<set2>).
1322 (i.e. ::<set1> - ::<set2>).
1323 """
1323 """
1324 cl = repo.changelog
1324 cl = repo.changelog
1325 # i18n: "only" is a keyword
1325 # i18n: "only" is a keyword
1326 args = getargs(x, 1, 2, _('only takes one or two arguments'))
1326 args = getargs(x, 1, 2, _('only takes one or two arguments'))
1327 include = getset(repo, fullreposet(repo), args[0])
1327 include = getset(repo, fullreposet(repo), args[0])
1328 if len(args) == 1:
1328 if len(args) == 1:
1329 if not include:
1329 if not include:
1330 return baseset()
1330 return baseset()
1331
1331
1332 descendants = set(_revdescendants(repo, include, False))
1332 descendants = set(_revdescendants(repo, include, False))
1333 exclude = [rev for rev in cl.headrevs()
1333 exclude = [rev for rev in cl.headrevs()
1334 if not rev in descendants and not rev in include]
1334 if not rev in descendants and not rev in include]
1335 else:
1335 else:
1336 exclude = getset(repo, fullreposet(repo), args[1])
1336 exclude = getset(repo, fullreposet(repo), args[1])
1337
1337
1338 results = set(cl.findmissingrevs(common=exclude, heads=include))
1338 results = set(cl.findmissingrevs(common=exclude, heads=include))
1339 return subset & results
1339 return subset & results
1340
1340
1341 def origin(repo, subset, x):
1341 def origin(repo, subset, x):
1342 """``origin([set])``
1342 """``origin([set])``
1343 Changesets that were specified as a source for the grafts, transplants or
1343 Changesets that were specified as a source for the grafts, transplants or
1344 rebases that created the given revisions. Omitting the optional set is the
1344 rebases that created the given revisions. Omitting the optional set is the
1345 same as passing all(). If a changeset created by these operations is itself
1345 same as passing all(). If a changeset created by these operations is itself
1346 specified as a source for one of these operations, only the source changeset
1346 specified as a source for one of these operations, only the source changeset
1347 for the first operation is selected.
1347 for the first operation is selected.
1348 """
1348 """
1349 if x is not None:
1349 if x is not None:
1350 dests = getset(repo, fullreposet(repo), x)
1350 dests = getset(repo, fullreposet(repo), x)
1351 else:
1351 else:
1352 dests = fullreposet(repo)
1352 dests = fullreposet(repo)
1353
1353
1354 def _firstsrc(rev):
1354 def _firstsrc(rev):
1355 src = _getrevsource(repo, rev)
1355 src = _getrevsource(repo, rev)
1356 if src is None:
1356 if src is None:
1357 return None
1357 return None
1358
1358
1359 while True:
1359 while True:
1360 prev = _getrevsource(repo, src)
1360 prev = _getrevsource(repo, src)
1361
1361
1362 if prev is None:
1362 if prev is None:
1363 return src
1363 return src
1364 src = prev
1364 src = prev
1365
1365
1366 o = set([_firstsrc(r) for r in dests])
1366 o = set([_firstsrc(r) for r in dests])
1367 o -= set([None])
1367 o -= set([None])
1368 return subset & o
1368 return subset & o
1369
1369
1370 def outgoing(repo, subset, x):
1370 def outgoing(repo, subset, x):
1371 """``outgoing([path])``
1371 """``outgoing([path])``
1372 Changesets not found in the specified destination repository, or the
1372 Changesets not found in the specified destination repository, or the
1373 default push location.
1373 default push location.
1374 """
1374 """
1375 # Avoid cycles.
1375 # Avoid cycles.
1376 import discovery
1376 import discovery
1377 import hg
1377 import hg
1378 # i18n: "outgoing" is a keyword
1378 # i18n: "outgoing" is a keyword
1379 l = getargs(x, 0, 1, _("outgoing takes one or no arguments"))
1379 l = getargs(x, 0, 1, _("outgoing takes one or no arguments"))
1380 # i18n: "outgoing" is a keyword
1380 # i18n: "outgoing" is a keyword
1381 dest = l and getstring(l[0], _("outgoing requires a repository path")) or ''
1381 dest = l and getstring(l[0], _("outgoing requires a repository path")) or ''
1382 dest = repo.ui.expandpath(dest or 'default-push', dest or 'default')
1382 dest = repo.ui.expandpath(dest or 'default-push', dest or 'default')
1383 dest, branches = hg.parseurl(dest)
1383 dest, branches = hg.parseurl(dest)
1384 revs, checkout = hg.addbranchrevs(repo, repo, branches, [])
1384 revs, checkout = hg.addbranchrevs(repo, repo, branches, [])
1385 if revs:
1385 if revs:
1386 revs = [repo.lookup(rev) for rev in revs]
1386 revs = [repo.lookup(rev) for rev in revs]
1387 other = hg.peer(repo, {}, dest)
1387 other = hg.peer(repo, {}, dest)
1388 repo.ui.pushbuffer()
1388 repo.ui.pushbuffer()
1389 outgoing = discovery.findcommonoutgoing(repo, other, onlyheads=revs)
1389 outgoing = discovery.findcommonoutgoing(repo, other, onlyheads=revs)
1390 repo.ui.popbuffer()
1390 repo.ui.popbuffer()
1391 cl = repo.changelog
1391 cl = repo.changelog
1392 o = set([cl.rev(r) for r in outgoing.missing])
1392 o = set([cl.rev(r) for r in outgoing.missing])
1393 return subset & o
1393 return subset & o
1394
1394
1395 def p1(repo, subset, x):
1395 def p1(repo, subset, x):
1396 """``p1([set])``
1396 """``p1([set])``
1397 First parent of changesets in set, or the working directory.
1397 First parent of changesets in set, or the working directory.
1398 """
1398 """
1399 if x is None:
1399 if x is None:
1400 p = repo[x].p1().rev()
1400 p = repo[x].p1().rev()
1401 if p >= 0:
1401 if p >= 0:
1402 return subset & baseset([p])
1402 return subset & baseset([p])
1403 return baseset()
1403 return baseset()
1404
1404
1405 ps = set()
1405 ps = set()
1406 cl = repo.changelog
1406 cl = repo.changelog
1407 for r in getset(repo, fullreposet(repo), x):
1407 for r in getset(repo, fullreposet(repo), x):
1408 ps.add(cl.parentrevs(r)[0])
1408 ps.add(cl.parentrevs(r)[0])
1409 ps -= set([node.nullrev])
1409 ps -= set([node.nullrev])
1410 return subset & ps
1410 return subset & ps
1411
1411
1412 def p2(repo, subset, x):
1412 def p2(repo, subset, x):
1413 """``p2([set])``
1413 """``p2([set])``
1414 Second parent of changesets in set, or the working directory.
1414 Second parent of changesets in set, or the working directory.
1415 """
1415 """
1416 if x is None:
1416 if x is None:
1417 ps = repo[x].parents()
1417 ps = repo[x].parents()
1418 try:
1418 try:
1419 p = ps[1].rev()
1419 p = ps[1].rev()
1420 if p >= 0:
1420 if p >= 0:
1421 return subset & baseset([p])
1421 return subset & baseset([p])
1422 return baseset()
1422 return baseset()
1423 except IndexError:
1423 except IndexError:
1424 return baseset()
1424 return baseset()
1425
1425
1426 ps = set()
1426 ps = set()
1427 cl = repo.changelog
1427 cl = repo.changelog
1428 for r in getset(repo, fullreposet(repo), x):
1428 for r in getset(repo, fullreposet(repo), x):
1429 ps.add(cl.parentrevs(r)[1])
1429 ps.add(cl.parentrevs(r)[1])
1430 ps -= set([node.nullrev])
1430 ps -= set([node.nullrev])
1431 return subset & ps
1431 return subset & ps
1432
1432
1433 def parents(repo, subset, x):
1433 def parents(repo, subset, x):
1434 """``parents([set])``
1434 """``parents([set])``
1435 The set of all parents for all changesets in set, or the working directory.
1435 The set of all parents for all changesets in set, or the working directory.
1436 """
1436 """
1437 if x is None:
1437 if x is None:
1438 ps = set(p.rev() for p in repo[x].parents())
1438 ps = set(p.rev() for p in repo[x].parents())
1439 else:
1439 else:
1440 ps = set()
1440 ps = set()
1441 cl = repo.changelog
1441 cl = repo.changelog
1442 for r in getset(repo, fullreposet(repo), x):
1442 for r in getset(repo, fullreposet(repo), x):
1443 ps.update(cl.parentrevs(r))
1443 ps.update(cl.parentrevs(r))
1444 ps -= set([node.nullrev])
1444 ps -= set([node.nullrev])
1445 return subset & ps
1445 return subset & ps
1446
1446
1447 def parentspec(repo, subset, x, n):
1447 def parentspec(repo, subset, x, n):
1448 """``set^0``
1448 """``set^0``
1449 The set.
1449 The set.
1450 ``set^1`` (or ``set^``), ``set^2``
1450 ``set^1`` (or ``set^``), ``set^2``
1451 First or second parent, respectively, of all changesets in set.
1451 First or second parent, respectively, of all changesets in set.
1452 """
1452 """
1453 try:
1453 try:
1454 n = int(n[1])
1454 n = int(n[1])
1455 if n not in (0, 1, 2):
1455 if n not in (0, 1, 2):
1456 raise ValueError
1456 raise ValueError
1457 except (TypeError, ValueError):
1457 except (TypeError, ValueError):
1458 raise error.ParseError(_("^ expects a number 0, 1, or 2"))
1458 raise error.ParseError(_("^ expects a number 0, 1, or 2"))
1459 ps = set()
1459 ps = set()
1460 cl = repo.changelog
1460 cl = repo.changelog
1461 for r in getset(repo, fullreposet(repo), x):
1461 for r in getset(repo, fullreposet(repo), x):
1462 if n == 0:
1462 if n == 0:
1463 ps.add(r)
1463 ps.add(r)
1464 elif n == 1:
1464 elif n == 1:
1465 ps.add(cl.parentrevs(r)[0])
1465 ps.add(cl.parentrevs(r)[0])
1466 elif n == 2:
1466 elif n == 2:
1467 parents = cl.parentrevs(r)
1467 parents = cl.parentrevs(r)
1468 if len(parents) > 1:
1468 if len(parents) > 1:
1469 ps.add(parents[1])
1469 ps.add(parents[1])
1470 return subset & ps
1470 return subset & ps
1471
1471
1472 def present(repo, subset, x):
1472 def present(repo, subset, x):
1473 """``present(set)``
1473 """``present(set)``
1474 An empty set, if any revision in set isn't found; otherwise,
1474 An empty set, if any revision in set isn't found; otherwise,
1475 all revisions in set.
1475 all revisions in set.
1476
1476
1477 If any of specified revisions is not present in the local repository,
1477 If any of specified revisions is not present in the local repository,
1478 the query is normally aborted. But this predicate allows the query
1478 the query is normally aborted. But this predicate allows the query
1479 to continue even in such cases.
1479 to continue even in such cases.
1480 """
1480 """
1481 try:
1481 try:
1482 return getset(repo, subset, x)
1482 return getset(repo, subset, x)
1483 except error.RepoLookupError:
1483 except error.RepoLookupError:
1484 return baseset()
1484 return baseset()
1485
1485
1486 def public(repo, subset, x):
1486 def public(repo, subset, x):
1487 """``public()``
1487 """``public()``
1488 Changeset in public phase."""
1488 Changeset in public phase."""
1489 # i18n: "public" is a keyword
1489 # i18n: "public" is a keyword
1490 getargs(x, 0, 0, _("public takes no arguments"))
1490 getargs(x, 0, 0, _("public takes no arguments"))
1491 phase = repo._phasecache.phase
1491 phase = repo._phasecache.phase
1492 target = phases.public
1492 target = phases.public
1493 condition = lambda r: phase(repo, r) == target
1493 condition = lambda r: phase(repo, r) == target
1494 return subset.filter(condition, cache=False)
1494 return subset.filter(condition, cache=False)
1495
1495
1496 def remote(repo, subset, x):
1496 def remote(repo, subset, x):
1497 """``remote([id [,path]])``
1497 """``remote([id [,path]])``
1498 Local revision that corresponds to the given identifier in a
1498 Local revision that corresponds to the given identifier in a
1499 remote repository, if present. Here, the '.' identifier is a
1499 remote repository, if present. Here, the '.' identifier is a
1500 synonym for the current local branch.
1500 synonym for the current local branch.
1501 """
1501 """
1502
1502
1503 import hg # avoid start-up nasties
1503 import hg # avoid start-up nasties
1504 # i18n: "remote" is a keyword
1504 # i18n: "remote" is a keyword
1505 l = getargs(x, 0, 2, _("remote takes one, two or no arguments"))
1505 l = getargs(x, 0, 2, _("remote takes one, two or no arguments"))
1506
1506
1507 q = '.'
1507 q = '.'
1508 if len(l) > 0:
1508 if len(l) > 0:
1509 # i18n: "remote" is a keyword
1509 # i18n: "remote" is a keyword
1510 q = getstring(l[0], _("remote requires a string id"))
1510 q = getstring(l[0], _("remote requires a string id"))
1511 if q == '.':
1511 if q == '.':
1512 q = repo['.'].branch()
1512 q = repo['.'].branch()
1513
1513
1514 dest = ''
1514 dest = ''
1515 if len(l) > 1:
1515 if len(l) > 1:
1516 # i18n: "remote" is a keyword
1516 # i18n: "remote" is a keyword
1517 dest = getstring(l[1], _("remote requires a repository path"))
1517 dest = getstring(l[1], _("remote requires a repository path"))
1518 dest = repo.ui.expandpath(dest or 'default')
1518 dest = repo.ui.expandpath(dest or 'default')
1519 dest, branches = hg.parseurl(dest)
1519 dest, branches = hg.parseurl(dest)
1520 revs, checkout = hg.addbranchrevs(repo, repo, branches, [])
1520 revs, checkout = hg.addbranchrevs(repo, repo, branches, [])
1521 if revs:
1521 if revs:
1522 revs = [repo.lookup(rev) for rev in revs]
1522 revs = [repo.lookup(rev) for rev in revs]
1523 other = hg.peer(repo, {}, dest)
1523 other = hg.peer(repo, {}, dest)
1524 n = other.lookup(q)
1524 n = other.lookup(q)
1525 if n in repo:
1525 if n in repo:
1526 r = repo[n].rev()
1526 r = repo[n].rev()
1527 if r in subset:
1527 if r in subset:
1528 return baseset([r])
1528 return baseset([r])
1529 return baseset()
1529 return baseset()
1530
1530
1531 def removes(repo, subset, x):
1531 def removes(repo, subset, x):
1532 """``removes(pattern)``
1532 """``removes(pattern)``
1533 Changesets which remove files matching pattern.
1533 Changesets which remove files matching pattern.
1534
1534
1535 The pattern without explicit kind like ``glob:`` is expected to be
1535 The pattern without explicit kind like ``glob:`` is expected to be
1536 relative to the current directory and match against a file or a
1536 relative to the current directory and match against a file or a
1537 directory.
1537 directory.
1538 """
1538 """
1539 # i18n: "removes" is a keyword
1539 # i18n: "removes" is a keyword
1540 pat = getstring(x, _("removes requires a pattern"))
1540 pat = getstring(x, _("removes requires a pattern"))
1541 return checkstatus(repo, subset, pat, 2)
1541 return checkstatus(repo, subset, pat, 2)
1542
1542
1543 def rev(repo, subset, x):
1543 def rev(repo, subset, x):
1544 """``rev(number)``
1544 """``rev(number)``
1545 Revision with the given numeric identifier.
1545 Revision with the given numeric identifier.
1546 """
1546 """
1547 # i18n: "rev" is a keyword
1547 # i18n: "rev" is a keyword
1548 l = getargs(x, 1, 1, _("rev requires one argument"))
1548 l = getargs(x, 1, 1, _("rev requires one argument"))
1549 try:
1549 try:
1550 # i18n: "rev" is a keyword
1550 # i18n: "rev" is a keyword
1551 l = int(getstring(l[0], _("rev requires a number")))
1551 l = int(getstring(l[0], _("rev requires a number")))
1552 except (TypeError, ValueError):
1552 except (TypeError, ValueError):
1553 # i18n: "rev" is a keyword
1553 # i18n: "rev" is a keyword
1554 raise error.ParseError(_("rev expects a number"))
1554 raise error.ParseError(_("rev expects a number"))
1555 if l not in repo.changelog and l != node.nullrev:
1555 if l not in repo.changelog and l != node.nullrev:
1556 return baseset()
1556 return baseset()
1557 return subset & baseset([l])
1557 return subset & baseset([l])
1558
1558
1559 def matching(repo, subset, x):
1559 def matching(repo, subset, x):
1560 """``matching(revision [, field])``
1560 """``matching(revision [, field])``
1561 Changesets in which a given set of fields match the set of fields in the
1561 Changesets in which a given set of fields match the set of fields in the
1562 selected revision or set.
1562 selected revision or set.
1563
1563
1564 To match more than one field pass the list of fields to match separated
1564 To match more than one field pass the list of fields to match separated
1565 by spaces (e.g. ``author description``).
1565 by spaces (e.g. ``author description``).
1566
1566
1567 Valid fields are most regular revision fields and some special fields.
1567 Valid fields are most regular revision fields and some special fields.
1568
1568
1569 Regular revision fields are ``description``, ``author``, ``branch``,
1569 Regular revision fields are ``description``, ``author``, ``branch``,
1570 ``date``, ``files``, ``phase``, ``parents``, ``substate``, ``user``
1570 ``date``, ``files``, ``phase``, ``parents``, ``substate``, ``user``
1571 and ``diff``.
1571 and ``diff``.
1572 Note that ``author`` and ``user`` are synonyms. ``diff`` refers to the
1572 Note that ``author`` and ``user`` are synonyms. ``diff`` refers to the
1573 contents of the revision. Two revisions matching their ``diff`` will
1573 contents of the revision. Two revisions matching their ``diff`` will
1574 also match their ``files``.
1574 also match their ``files``.
1575
1575
1576 Special fields are ``summary`` and ``metadata``:
1576 Special fields are ``summary`` and ``metadata``:
1577 ``summary`` matches the first line of the description.
1577 ``summary`` matches the first line of the description.
1578 ``metadata`` is equivalent to matching ``description user date``
1578 ``metadata`` is equivalent to matching ``description user date``
1579 (i.e. it matches the main metadata fields).
1579 (i.e. it matches the main metadata fields).
1580
1580
1581 ``metadata`` is the default field which is used when no fields are
1581 ``metadata`` is the default field which is used when no fields are
1582 specified. You can match more than one field at a time.
1582 specified. You can match more than one field at a time.
1583 """
1583 """
1584 # i18n: "matching" is a keyword
1584 # i18n: "matching" is a keyword
1585 l = getargs(x, 1, 2, _("matching takes 1 or 2 arguments"))
1585 l = getargs(x, 1, 2, _("matching takes 1 or 2 arguments"))
1586
1586
1587 revs = getset(repo, fullreposet(repo), l[0])
1587 revs = getset(repo, fullreposet(repo), l[0])
1588
1588
1589 fieldlist = ['metadata']
1589 fieldlist = ['metadata']
1590 if len(l) > 1:
1590 if len(l) > 1:
1591 fieldlist = getstring(l[1],
1591 fieldlist = getstring(l[1],
1592 # i18n: "matching" is a keyword
1592 # i18n: "matching" is a keyword
1593 _("matching requires a string "
1593 _("matching requires a string "
1594 "as its second argument")).split()
1594 "as its second argument")).split()
1595
1595
1596 # Make sure that there are no repeated fields,
1596 # Make sure that there are no repeated fields,
1597 # expand the 'special' 'metadata' field type
1597 # expand the 'special' 'metadata' field type
1598 # and check the 'files' whenever we check the 'diff'
1598 # and check the 'files' whenever we check the 'diff'
1599 fields = []
1599 fields = []
1600 for field in fieldlist:
1600 for field in fieldlist:
1601 if field == 'metadata':
1601 if field == 'metadata':
1602 fields += ['user', 'description', 'date']
1602 fields += ['user', 'description', 'date']
1603 elif field == 'diff':
1603 elif field == 'diff':
1604 # a revision matching the diff must also match the files
1604 # a revision matching the diff must also match the files
1605 # since matching the diff is very costly, make sure to
1605 # since matching the diff is very costly, make sure to
1606 # also match the files first
1606 # also match the files first
1607 fields += ['files', 'diff']
1607 fields += ['files', 'diff']
1608 else:
1608 else:
1609 if field == 'author':
1609 if field == 'author':
1610 field = 'user'
1610 field = 'user'
1611 fields.append(field)
1611 fields.append(field)
1612 fields = set(fields)
1612 fields = set(fields)
1613 if 'summary' in fields and 'description' in fields:
1613 if 'summary' in fields and 'description' in fields:
1614 # If a revision matches its description it also matches its summary
1614 # If a revision matches its description it also matches its summary
1615 fields.discard('summary')
1615 fields.discard('summary')
1616
1616
1617 # We may want to match more than one field
1617 # We may want to match more than one field
1618 # Not all fields take the same amount of time to be matched
1618 # Not all fields take the same amount of time to be matched
1619 # Sort the selected fields in order of increasing matching cost
1619 # Sort the selected fields in order of increasing matching cost
1620 fieldorder = ['phase', 'parents', 'user', 'date', 'branch', 'summary',
1620 fieldorder = ['phase', 'parents', 'user', 'date', 'branch', 'summary',
1621 'files', 'description', 'substate', 'diff']
1621 'files', 'description', 'substate', 'diff']
1622 def fieldkeyfunc(f):
1622 def fieldkeyfunc(f):
1623 try:
1623 try:
1624 return fieldorder.index(f)
1624 return fieldorder.index(f)
1625 except ValueError:
1625 except ValueError:
1626 # assume an unknown field is very costly
1626 # assume an unknown field is very costly
1627 return len(fieldorder)
1627 return len(fieldorder)
1628 fields = list(fields)
1628 fields = list(fields)
1629 fields.sort(key=fieldkeyfunc)
1629 fields.sort(key=fieldkeyfunc)
1630
1630
1631 # Each field will be matched with its own "getfield" function
1631 # Each field will be matched with its own "getfield" function
1632 # which will be added to the getfieldfuncs array of functions
1632 # which will be added to the getfieldfuncs array of functions
1633 getfieldfuncs = []
1633 getfieldfuncs = []
1634 _funcs = {
1634 _funcs = {
1635 'user': lambda r: repo[r].user(),
1635 'user': lambda r: repo[r].user(),
1636 'branch': lambda r: repo[r].branch(),
1636 'branch': lambda r: repo[r].branch(),
1637 'date': lambda r: repo[r].date(),
1637 'date': lambda r: repo[r].date(),
1638 'description': lambda r: repo[r].description(),
1638 'description': lambda r: repo[r].description(),
1639 'files': lambda r: repo[r].files(),
1639 'files': lambda r: repo[r].files(),
1640 'parents': lambda r: repo[r].parents(),
1640 'parents': lambda r: repo[r].parents(),
1641 'phase': lambda r: repo[r].phase(),
1641 'phase': lambda r: repo[r].phase(),
1642 'substate': lambda r: repo[r].substate,
1642 'substate': lambda r: repo[r].substate,
1643 'summary': lambda r: repo[r].description().splitlines()[0],
1643 'summary': lambda r: repo[r].description().splitlines()[0],
1644 'diff': lambda r: list(repo[r].diff(git=True),)
1644 'diff': lambda r: list(repo[r].diff(git=True),)
1645 }
1645 }
1646 for info in fields:
1646 for info in fields:
1647 getfield = _funcs.get(info, None)
1647 getfield = _funcs.get(info, None)
1648 if getfield is None:
1648 if getfield is None:
1649 raise error.ParseError(
1649 raise error.ParseError(
1650 # i18n: "matching" is a keyword
1650 # i18n: "matching" is a keyword
1651 _("unexpected field name passed to matching: %s") % info)
1651 _("unexpected field name passed to matching: %s") % info)
1652 getfieldfuncs.append(getfield)
1652 getfieldfuncs.append(getfield)
1653 # convert the getfield array of functions into a "getinfo" function
1653 # convert the getfield array of functions into a "getinfo" function
1654 # which returns an array of field values (or a single value if there
1654 # which returns an array of field values (or a single value if there
1655 # is only one field to match)
1655 # is only one field to match)
1656 getinfo = lambda r: [f(r) for f in getfieldfuncs]
1656 getinfo = lambda r: [f(r) for f in getfieldfuncs]
1657
1657
1658 def matches(x):
1658 def matches(x):
1659 for rev in revs:
1659 for rev in revs:
1660 target = getinfo(rev)
1660 target = getinfo(rev)
1661 match = True
1661 match = True
1662 for n, f in enumerate(getfieldfuncs):
1662 for n, f in enumerate(getfieldfuncs):
1663 if target[n] != f(x):
1663 if target[n] != f(x):
1664 match = False
1664 match = False
1665 if match:
1665 if match:
1666 return True
1666 return True
1667 return False
1667 return False
1668
1668
1669 return subset.filter(matches)
1669 return subset.filter(matches)
1670
1670
1671 def reverse(repo, subset, x):
1671 def reverse(repo, subset, x):
1672 """``reverse(set)``
1672 """``reverse(set)``
1673 Reverse order of set.
1673 Reverse order of set.
1674 """
1674 """
1675 l = getset(repo, subset, x)
1675 l = getset(repo, subset, x)
1676 l.reverse()
1676 l.reverse()
1677 return l
1677 return l
1678
1678
1679 def roots(repo, subset, x):
1679 def roots(repo, subset, x):
1680 """``roots(set)``
1680 """``roots(set)``
1681 Changesets in set with no parent changeset in set.
1681 Changesets in set with no parent changeset in set.
1682 """
1682 """
1683 s = getset(repo, fullreposet(repo), x)
1683 s = getset(repo, fullreposet(repo), x)
1684 subset = subset & s# baseset([r for r in s if r in subset])
1684 subset = subset & s# baseset([r for r in s if r in subset])
1685 cs = _children(repo, subset, s)
1685 cs = _children(repo, subset, s)
1686 return subset - cs
1686 return subset - cs
1687
1687
1688 def secret(repo, subset, x):
1688 def secret(repo, subset, x):
1689 """``secret()``
1689 """``secret()``
1690 Changeset in secret phase."""
1690 Changeset in secret phase."""
1691 # i18n: "secret" is a keyword
1691 # i18n: "secret" is a keyword
1692 getargs(x, 0, 0, _("secret takes no arguments"))
1692 getargs(x, 0, 0, _("secret takes no arguments"))
1693 phase = repo._phasecache.phase
1693 phase = repo._phasecache.phase
1694 target = phases.secret
1694 target = phases.secret
1695 condition = lambda r: phase(repo, r) == target
1695 condition = lambda r: phase(repo, r) == target
1696 return subset.filter(condition, cache=False)
1696 return subset.filter(condition, cache=False)
1697
1697
1698 def sort(repo, subset, x):
1698 def sort(repo, subset, x):
1699 """``sort(set[, [-]key...])``
1699 """``sort(set[, [-]key...])``
1700 Sort set by keys. The default sort order is ascending, specify a key
1700 Sort set by keys. The default sort order is ascending, specify a key
1701 as ``-key`` to sort in descending order.
1701 as ``-key`` to sort in descending order.
1702
1702
1703 The keys can be:
1703 The keys can be:
1704
1704
1705 - ``rev`` for the revision number,
1705 - ``rev`` for the revision number,
1706 - ``branch`` for the branch name,
1706 - ``branch`` for the branch name,
1707 - ``desc`` for the commit message (description),
1707 - ``desc`` for the commit message (description),
1708 - ``user`` for user name (``author`` can be used as an alias),
1708 - ``user`` for user name (``author`` can be used as an alias),
1709 - ``date`` for the commit date
1709 - ``date`` for the commit date
1710 """
1710 """
1711 # i18n: "sort" is a keyword
1711 # i18n: "sort" is a keyword
1712 l = getargs(x, 1, 2, _("sort requires one or two arguments"))
1712 l = getargs(x, 1, 2, _("sort requires one or two arguments"))
1713 keys = "rev"
1713 keys = "rev"
1714 if len(l) == 2:
1714 if len(l) == 2:
1715 # i18n: "sort" is a keyword
1715 # i18n: "sort" is a keyword
1716 keys = getstring(l[1], _("sort spec must be a string"))
1716 keys = getstring(l[1], _("sort spec must be a string"))
1717
1717
1718 s = l[0]
1718 s = l[0]
1719 keys = keys.split()
1719 keys = keys.split()
1720 l = []
1720 l = []
1721 def invert(s):
1721 def invert(s):
1722 return "".join(chr(255 - ord(c)) for c in s)
1722 return "".join(chr(255 - ord(c)) for c in s)
1723 revs = getset(repo, subset, s)
1723 revs = getset(repo, subset, s)
1724 if keys == ["rev"]:
1724 if keys == ["rev"]:
1725 revs.sort()
1725 revs.sort()
1726 return revs
1726 return revs
1727 elif keys == ["-rev"]:
1727 elif keys == ["-rev"]:
1728 revs.sort(reverse=True)
1728 revs.sort(reverse=True)
1729 return revs
1729 return revs
1730 for r in revs:
1730 for r in revs:
1731 c = repo[r]
1731 c = repo[r]
1732 e = []
1732 e = []
1733 for k in keys:
1733 for k in keys:
1734 if k == 'rev':
1734 if k == 'rev':
1735 e.append(r)
1735 e.append(r)
1736 elif k == '-rev':
1736 elif k == '-rev':
1737 e.append(-r)
1737 e.append(-r)
1738 elif k == 'branch':
1738 elif k == 'branch':
1739 e.append(c.branch())
1739 e.append(c.branch())
1740 elif k == '-branch':
1740 elif k == '-branch':
1741 e.append(invert(c.branch()))
1741 e.append(invert(c.branch()))
1742 elif k == 'desc':
1742 elif k == 'desc':
1743 e.append(c.description())
1743 e.append(c.description())
1744 elif k == '-desc':
1744 elif k == '-desc':
1745 e.append(invert(c.description()))
1745 e.append(invert(c.description()))
1746 elif k in 'user author':
1746 elif k in 'user author':
1747 e.append(c.user())
1747 e.append(c.user())
1748 elif k in '-user -author':
1748 elif k in '-user -author':
1749 e.append(invert(c.user()))
1749 e.append(invert(c.user()))
1750 elif k == 'date':
1750 elif k == 'date':
1751 e.append(c.date()[0])
1751 e.append(c.date()[0])
1752 elif k == '-date':
1752 elif k == '-date':
1753 e.append(-c.date()[0])
1753 e.append(-c.date()[0])
1754 else:
1754 else:
1755 raise error.ParseError(_("unknown sort key %r") % k)
1755 raise error.ParseError(_("unknown sort key %r") % k)
1756 e.append(r)
1756 e.append(r)
1757 l.append(e)
1757 l.append(e)
1758 l.sort()
1758 l.sort()
1759 return baseset([e[-1] for e in l])
1759 return baseset([e[-1] for e in l])
1760
1760
1761 def subrepo(repo, subset, x):
1761 def subrepo(repo, subset, x):
1762 """``subrepo([pattern])``
1762 """``subrepo([pattern])``
1763 Changesets that add, modify or remove the given subrepo. If no subrepo
1763 Changesets that add, modify or remove the given subrepo. If no subrepo
1764 pattern is named, any subrepo changes are returned.
1764 pattern is named, any subrepo changes are returned.
1765 """
1765 """
1766 # i18n: "subrepo" is a keyword
1766 # i18n: "subrepo" is a keyword
1767 args = getargs(x, 0, 1, _('subrepo takes at most one argument'))
1767 args = getargs(x, 0, 1, _('subrepo takes at most one argument'))
1768 if len(args) != 0:
1768 if len(args) != 0:
1769 pat = getstring(args[0], _("subrepo requires a pattern"))
1769 pat = getstring(args[0], _("subrepo requires a pattern"))
1770
1770
1771 m = matchmod.exact(repo.root, repo.root, ['.hgsubstate'])
1771 m = matchmod.exact(repo.root, repo.root, ['.hgsubstate'])
1772
1772
1773 def submatches(names):
1773 def submatches(names):
1774 k, p, m = _stringmatcher(pat)
1774 k, p, m = _stringmatcher(pat)
1775 for name in names:
1775 for name in names:
1776 if m(name):
1776 if m(name):
1777 yield name
1777 yield name
1778
1778
1779 def matches(x):
1779 def matches(x):
1780 c = repo[x]
1780 c = repo[x]
1781 s = repo.status(c.p1().node(), c.node(), match=m)
1781 s = repo.status(c.p1().node(), c.node(), match=m)
1782
1782
1783 if len(args) == 0:
1783 if len(args) == 0:
1784 return s.added or s.modified or s.removed
1784 return s.added or s.modified or s.removed
1785
1785
1786 if s.added:
1786 if s.added:
1787 return util.any(submatches(c.substate.keys()))
1787 return util.any(submatches(c.substate.keys()))
1788
1788
1789 if s.modified:
1789 if s.modified:
1790 subs = set(c.p1().substate.keys())
1790 subs = set(c.p1().substate.keys())
1791 subs.update(c.substate.keys())
1791 subs.update(c.substate.keys())
1792
1792
1793 for path in submatches(subs):
1793 for path in submatches(subs):
1794 if c.p1().substate.get(path) != c.substate.get(path):
1794 if c.p1().substate.get(path) != c.substate.get(path):
1795 return True
1795 return True
1796
1796
1797 if s.removed:
1797 if s.removed:
1798 return util.any(submatches(c.p1().substate.keys()))
1798 return util.any(submatches(c.p1().substate.keys()))
1799
1799
1800 return False
1800 return False
1801
1801
1802 return subset.filter(matches)
1802 return subset.filter(matches)
1803
1803
1804 def _stringmatcher(pattern):
1804 def _stringmatcher(pattern):
1805 """
1805 """
1806 accepts a string, possibly starting with 're:' or 'literal:' prefix.
1806 accepts a string, possibly starting with 're:' or 'literal:' prefix.
1807 returns the matcher name, pattern, and matcher function.
1807 returns the matcher name, pattern, and matcher function.
1808 missing or unknown prefixes are treated as literal matches.
1808 missing or unknown prefixes are treated as literal matches.
1809
1809
1810 helper for tests:
1810 helper for tests:
1811 >>> def test(pattern, *tests):
1811 >>> def test(pattern, *tests):
1812 ... kind, pattern, matcher = _stringmatcher(pattern)
1812 ... kind, pattern, matcher = _stringmatcher(pattern)
1813 ... return (kind, pattern, [bool(matcher(t)) for t in tests])
1813 ... return (kind, pattern, [bool(matcher(t)) for t in tests])
1814
1814
1815 exact matching (no prefix):
1815 exact matching (no prefix):
1816 >>> test('abcdefg', 'abc', 'def', 'abcdefg')
1816 >>> test('abcdefg', 'abc', 'def', 'abcdefg')
1817 ('literal', 'abcdefg', [False, False, True])
1817 ('literal', 'abcdefg', [False, False, True])
1818
1818
1819 regex matching ('re:' prefix)
1819 regex matching ('re:' prefix)
1820 >>> test('re:a.+b', 'nomatch', 'fooadef', 'fooadefbar')
1820 >>> test('re:a.+b', 'nomatch', 'fooadef', 'fooadefbar')
1821 ('re', 'a.+b', [False, False, True])
1821 ('re', 'a.+b', [False, False, True])
1822
1822
1823 force exact matches ('literal:' prefix)
1823 force exact matches ('literal:' prefix)
1824 >>> test('literal:re:foobar', 'foobar', 're:foobar')
1824 >>> test('literal:re:foobar', 'foobar', 're:foobar')
1825 ('literal', 're:foobar', [False, True])
1825 ('literal', 're:foobar', [False, True])
1826
1826
1827 unknown prefixes are ignored and treated as literals
1827 unknown prefixes are ignored and treated as literals
1828 >>> test('foo:bar', 'foo', 'bar', 'foo:bar')
1828 >>> test('foo:bar', 'foo', 'bar', 'foo:bar')
1829 ('literal', 'foo:bar', [False, False, True])
1829 ('literal', 'foo:bar', [False, False, True])
1830 """
1830 """
1831 if pattern.startswith('re:'):
1831 if pattern.startswith('re:'):
1832 pattern = pattern[3:]
1832 pattern = pattern[3:]
1833 try:
1833 try:
1834 regex = re.compile(pattern)
1834 regex = re.compile(pattern)
1835 except re.error, e:
1835 except re.error, e:
1836 raise error.ParseError(_('invalid regular expression: %s')
1836 raise error.ParseError(_('invalid regular expression: %s')
1837 % e)
1837 % e)
1838 return 're', pattern, regex.search
1838 return 're', pattern, regex.search
1839 elif pattern.startswith('literal:'):
1839 elif pattern.startswith('literal:'):
1840 pattern = pattern[8:]
1840 pattern = pattern[8:]
1841 return 'literal', pattern, pattern.__eq__
1841 return 'literal', pattern, pattern.__eq__
1842
1842
1843 def _substringmatcher(pattern):
1843 def _substringmatcher(pattern):
1844 kind, pattern, matcher = _stringmatcher(pattern)
1844 kind, pattern, matcher = _stringmatcher(pattern)
1845 if kind == 'literal':
1845 if kind == 'literal':
1846 matcher = lambda s: pattern in s
1846 matcher = lambda s: pattern in s
1847 return kind, pattern, matcher
1847 return kind, pattern, matcher
1848
1848
1849 def tag(repo, subset, x):
1849 def tag(repo, subset, x):
1850 """``tag([name])``
1850 """``tag([name])``
1851 The specified tag by name, or all tagged revisions if no name is given.
1851 The specified tag by name, or all tagged revisions if no name is given.
1852
1852
1853 If `name` starts with `re:`, the remainder of the name is treated as
1853 If `name` starts with `re:`, the remainder of the name is treated as
1854 a regular expression. To match a tag that actually starts with `re:`,
1854 a regular expression. To match a tag that actually starts with `re:`,
1855 use the prefix `literal:`.
1855 use the prefix `literal:`.
1856 """
1856 """
1857 # i18n: "tag" is a keyword
1857 # i18n: "tag" is a keyword
1858 args = getargs(x, 0, 1, _("tag takes one or no arguments"))
1858 args = getargs(x, 0, 1, _("tag takes one or no arguments"))
1859 cl = repo.changelog
1859 cl = repo.changelog
1860 if args:
1860 if args:
1861 pattern = getstring(args[0],
1861 pattern = getstring(args[0],
1862 # i18n: "tag" is a keyword
1862 # i18n: "tag" is a keyword
1863 _('the argument to tag must be a string'))
1863 _('the argument to tag must be a string'))
1864 kind, pattern, matcher = _stringmatcher(pattern)
1864 kind, pattern, matcher = _stringmatcher(pattern)
1865 if kind == 'literal':
1865 if kind == 'literal':
1866 # avoid resolving all tags
1866 # avoid resolving all tags
1867 tn = repo._tagscache.tags.get(pattern, None)
1867 tn = repo._tagscache.tags.get(pattern, None)
1868 if tn is None:
1868 if tn is None:
1869 raise error.RepoLookupError(_("tag '%s' does not exist")
1869 raise error.RepoLookupError(_("tag '%s' does not exist")
1870 % pattern)
1870 % pattern)
1871 s = set([repo[tn].rev()])
1871 s = set([repo[tn].rev()])
1872 else:
1872 else:
1873 s = set([cl.rev(n) for t, n in repo.tagslist() if matcher(t)])
1873 s = set([cl.rev(n) for t, n in repo.tagslist() if matcher(t)])
1874 else:
1874 else:
1875 s = set([cl.rev(n) for t, n in repo.tagslist() if t != 'tip'])
1875 s = set([cl.rev(n) for t, n in repo.tagslist() if t != 'tip'])
1876 return subset & s
1876 return subset & s
1877
1877
1878 def tagged(repo, subset, x):
1878 def tagged(repo, subset, x):
1879 return tag(repo, subset, x)
1879 return tag(repo, subset, x)
1880
1880
1881 def unstable(repo, subset, x):
1881 def unstable(repo, subset, x):
1882 """``unstable()``
1882 """``unstable()``
1883 Non-obsolete changesets with obsolete ancestors.
1883 Non-obsolete changesets with obsolete ancestors.
1884 """
1884 """
1885 # i18n: "unstable" is a keyword
1885 # i18n: "unstable" is a keyword
1886 getargs(x, 0, 0, _("unstable takes no arguments"))
1886 getargs(x, 0, 0, _("unstable takes no arguments"))
1887 unstables = obsmod.getrevs(repo, 'unstable')
1887 unstables = obsmod.getrevs(repo, 'unstable')
1888 return subset & unstables
1888 return subset & unstables
1889
1889
1890
1890
1891 def user(repo, subset, x):
1891 def user(repo, subset, x):
1892 """``user(string)``
1892 """``user(string)``
1893 User name contains string. The match is case-insensitive.
1893 User name contains string. The match is case-insensitive.
1894
1894
1895 If `string` starts with `re:`, the remainder of the string is treated as
1895 If `string` starts with `re:`, the remainder of the string is treated as
1896 a regular expression. To match a user that actually contains `re:`, use
1896 a regular expression. To match a user that actually contains `re:`, use
1897 the prefix `literal:`.
1897 the prefix `literal:`.
1898 """
1898 """
1899 return author(repo, subset, x)
1899 return author(repo, subset, x)
1900
1900
1901 # experimental
1901 # experimental
1902 def wdir(repo, subset, x):
1902 def wdir(repo, subset, x):
1903 # i18n: "wdir" is a keyword
1903 # i18n: "wdir" is a keyword
1904 getargs(x, 0, 0, _("wdir takes no arguments"))
1904 getargs(x, 0, 0, _("wdir takes no arguments"))
1905 if None in subset:
1905 if None in subset:
1906 return baseset([None])
1906 return baseset([None])
1907 return baseset()
1907 return baseset()
1908
1908
1909 # for internal use
1909 # for internal use
1910 def _list(repo, subset, x):
1910 def _list(repo, subset, x):
1911 s = getstring(x, "internal error")
1911 s = getstring(x, "internal error")
1912 if not s:
1912 if not s:
1913 return baseset()
1913 return baseset()
1914 ls = [repo[r].rev() for r in s.split('\0')]
1914 ls = [repo[r].rev() for r in s.split('\0')]
1915 s = subset
1915 s = subset
1916 return baseset([r for r in ls if r in s])
1916 return baseset([r for r in ls if r in s])
1917
1917
1918 # for internal use
1918 # for internal use
1919 def _intlist(repo, subset, x):
1919 def _intlist(repo, subset, x):
1920 s = getstring(x, "internal error")
1920 s = getstring(x, "internal error")
1921 if not s:
1921 if not s:
1922 return baseset()
1922 return baseset()
1923 ls = [int(r) for r in s.split('\0')]
1923 ls = [int(r) for r in s.split('\0')]
1924 s = subset
1924 s = subset
1925 return baseset([r for r in ls if r in s])
1925 return baseset([r for r in ls if r in s])
1926
1926
1927 # for internal use
1927 # for internal use
1928 def _hexlist(repo, subset, x):
1928 def _hexlist(repo, subset, x):
1929 s = getstring(x, "internal error")
1929 s = getstring(x, "internal error")
1930 if not s:
1930 if not s:
1931 return baseset()
1931 return baseset()
1932 cl = repo.changelog
1932 cl = repo.changelog
1933 ls = [cl.rev(node.bin(r)) for r in s.split('\0')]
1933 ls = [cl.rev(node.bin(r)) for r in s.split('\0')]
1934 s = subset
1934 s = subset
1935 return baseset([r for r in ls if r in s])
1935 return baseset([r for r in ls if r in s])
1936
1936
1937 symbols = {
1937 symbols = {
1938 "adds": adds,
1938 "adds": adds,
1939 "all": getall,
1939 "all": getall,
1940 "ancestor": ancestor,
1940 "ancestor": ancestor,
1941 "ancestors": ancestors,
1941 "ancestors": ancestors,
1942 "_firstancestors": _firstancestors,
1942 "_firstancestors": _firstancestors,
1943 "author": author,
1943 "author": author,
1944 "bisect": bisect,
1944 "bisect": bisect,
1945 "bisected": bisected,
1945 "bisected": bisected,
1946 "bookmark": bookmark,
1946 "bookmark": bookmark,
1947 "branch": branch,
1947 "branch": branch,
1948 "branchpoint": branchpoint,
1948 "branchpoint": branchpoint,
1949 "bumped": bumped,
1949 "bumped": bumped,
1950 "bundle": bundle,
1950 "bundle": bundle,
1951 "children": children,
1951 "children": children,
1952 "closed": closed,
1952 "closed": closed,
1953 "contains": contains,
1953 "contains": contains,
1954 "converted": converted,
1954 "converted": converted,
1955 "date": date,
1955 "date": date,
1956 "desc": desc,
1956 "desc": desc,
1957 "descendants": descendants,
1957 "descendants": descendants,
1958 "_firstdescendants": _firstdescendants,
1958 "_firstdescendants": _firstdescendants,
1959 "destination": destination,
1959 "destination": destination,
1960 "divergent": divergent,
1960 "divergent": divergent,
1961 "draft": draft,
1961 "draft": draft,
1962 "extinct": extinct,
1962 "extinct": extinct,
1963 "extra": extra,
1963 "extra": extra,
1964 "file": hasfile,
1964 "file": hasfile,
1965 "filelog": filelog,
1965 "filelog": filelog,
1966 "first": first,
1966 "first": first,
1967 "follow": follow,
1967 "follow": follow,
1968 "_followfirst": _followfirst,
1968 "_followfirst": _followfirst,
1969 "grep": grep,
1969 "grep": grep,
1970 "head": head,
1970 "head": head,
1971 "heads": heads,
1971 "heads": heads,
1972 "hidden": hidden,
1972 "hidden": hidden,
1973 "id": node_,
1973 "id": node_,
1974 "keyword": keyword,
1974 "keyword": keyword,
1975 "last": last,
1975 "last": last,
1976 "limit": limit,
1976 "limit": limit,
1977 "_matchfiles": _matchfiles,
1977 "_matchfiles": _matchfiles,
1978 "max": maxrev,
1978 "max": maxrev,
1979 "merge": merge,
1979 "merge": merge,
1980 "min": minrev,
1980 "min": minrev,
1981 "modifies": modifies,
1981 "modifies": modifies,
1982 "named": named,
1982 "named": named,
1983 "obsolete": obsolete,
1983 "obsolete": obsolete,
1984 "only": only,
1984 "only": only,
1985 "origin": origin,
1985 "origin": origin,
1986 "outgoing": outgoing,
1986 "outgoing": outgoing,
1987 "p1": p1,
1987 "p1": p1,
1988 "p2": p2,
1988 "p2": p2,
1989 "parents": parents,
1989 "parents": parents,
1990 "present": present,
1990 "present": present,
1991 "public": public,
1991 "public": public,
1992 "remote": remote,
1992 "remote": remote,
1993 "removes": removes,
1993 "removes": removes,
1994 "rev": rev,
1994 "rev": rev,
1995 "reverse": reverse,
1995 "reverse": reverse,
1996 "roots": roots,
1996 "roots": roots,
1997 "sort": sort,
1997 "sort": sort,
1998 "secret": secret,
1998 "secret": secret,
1999 "subrepo": subrepo,
1999 "subrepo": subrepo,
2000 "matching": matching,
2000 "matching": matching,
2001 "tag": tag,
2001 "tag": tag,
2002 "tagged": tagged,
2002 "tagged": tagged,
2003 "user": user,
2003 "user": user,
2004 "unstable": unstable,
2004 "unstable": unstable,
2005 "wdir": wdir,
2005 "wdir": wdir,
2006 "_list": _list,
2006 "_list": _list,
2007 "_intlist": _intlist,
2007 "_intlist": _intlist,
2008 "_hexlist": _hexlist,
2008 "_hexlist": _hexlist,
2009 }
2009 }
2010
2010
2011 # symbols which can't be used for a DoS attack for any given input
2011 # symbols which can't be used for a DoS attack for any given input
2012 # (e.g. those which accept regexes as plain strings shouldn't be included)
2012 # (e.g. those which accept regexes as plain strings shouldn't be included)
2013 # functions that just return a lot of changesets (like all) don't count here
2013 # functions that just return a lot of changesets (like all) don't count here
2014 safesymbols = set([
2014 safesymbols = set([
2015 "adds",
2015 "adds",
2016 "all",
2016 "all",
2017 "ancestor",
2017 "ancestor",
2018 "ancestors",
2018 "ancestors",
2019 "_firstancestors",
2019 "_firstancestors",
2020 "author",
2020 "author",
2021 "bisect",
2021 "bisect",
2022 "bisected",
2022 "bisected",
2023 "bookmark",
2023 "bookmark",
2024 "branch",
2024 "branch",
2025 "branchpoint",
2025 "branchpoint",
2026 "bumped",
2026 "bumped",
2027 "bundle",
2027 "bundle",
2028 "children",
2028 "children",
2029 "closed",
2029 "closed",
2030 "converted",
2030 "converted",
2031 "date",
2031 "date",
2032 "desc",
2032 "desc",
2033 "descendants",
2033 "descendants",
2034 "_firstdescendants",
2034 "_firstdescendants",
2035 "destination",
2035 "destination",
2036 "divergent",
2036 "divergent",
2037 "draft",
2037 "draft",
2038 "extinct",
2038 "extinct",
2039 "extra",
2039 "extra",
2040 "file",
2040 "file",
2041 "filelog",
2041 "filelog",
2042 "first",
2042 "first",
2043 "follow",
2043 "follow",
2044 "_followfirst",
2044 "_followfirst",
2045 "head",
2045 "head",
2046 "heads",
2046 "heads",
2047 "hidden",
2047 "hidden",
2048 "id",
2048 "id",
2049 "keyword",
2049 "keyword",
2050 "last",
2050 "last",
2051 "limit",
2051 "limit",
2052 "_matchfiles",
2052 "_matchfiles",
2053 "max",
2053 "max",
2054 "merge",
2054 "merge",
2055 "min",
2055 "min",
2056 "modifies",
2056 "modifies",
2057 "obsolete",
2057 "obsolete",
2058 "only",
2058 "only",
2059 "origin",
2059 "origin",
2060 "outgoing",
2060 "outgoing",
2061 "p1",
2061 "p1",
2062 "p2",
2062 "p2",
2063 "parents",
2063 "parents",
2064 "present",
2064 "present",
2065 "public",
2065 "public",
2066 "remote",
2066 "remote",
2067 "removes",
2067 "removes",
2068 "rev",
2068 "rev",
2069 "reverse",
2069 "reverse",
2070 "roots",
2070 "roots",
2071 "sort",
2071 "sort",
2072 "secret",
2072 "secret",
2073 "matching",
2073 "matching",
2074 "tag",
2074 "tag",
2075 "tagged",
2075 "tagged",
2076 "user",
2076 "user",
2077 "unstable",
2077 "unstable",
2078 "wdir",
2078 "wdir",
2079 "_list",
2079 "_list",
2080 "_intlist",
2080 "_intlist",
2081 "_hexlist",
2081 "_hexlist",
2082 ])
2082 ])
2083
2083
2084 methods = {
2084 methods = {
2085 "range": rangeset,
2085 "range": rangeset,
2086 "dagrange": dagrange,
2086 "dagrange": dagrange,
2087 "string": stringset,
2087 "string": stringset,
2088 "symbol": stringset,
2088 "symbol": stringset,
2089 "and": andset,
2089 "and": andset,
2090 "or": orset,
2090 "or": orset,
2091 "not": notset,
2091 "not": notset,
2092 "list": listset,
2092 "list": listset,
2093 "func": func,
2093 "func": func,
2094 "ancestor": ancestorspec,
2094 "ancestor": ancestorspec,
2095 "parent": parentspec,
2095 "parent": parentspec,
2096 "parentpost": p1,
2096 "parentpost": p1,
2097 }
2097 }
2098
2098
2099 def optimize(x, small):
2099 def optimize(x, small):
2100 if x is None:
2100 if x is None:
2101 return 0, x
2101 return 0, x
2102
2102
2103 smallbonus = 1
2103 smallbonus = 1
2104 if small:
2104 if small:
2105 smallbonus = .5
2105 smallbonus = .5
2106
2106
2107 op = x[0]
2107 op = x[0]
2108 if op == 'minus':
2108 if op == 'minus':
2109 return optimize(('and', x[1], ('not', x[2])), small)
2109 return optimize(('and', x[1], ('not', x[2])), small)
2110 elif op == 'only':
2110 elif op == 'only':
2111 return optimize(('func', ('symbol', 'only'),
2111 return optimize(('func', ('symbol', 'only'),
2112 ('list', x[1], x[2])), small)
2112 ('list', x[1], x[2])), small)
2113 elif op == 'onlypost':
2113 elif op == 'onlypost':
2114 return optimize(('func', ('symbol', 'only'), x[1]), small)
2114 return optimize(('func', ('symbol', 'only'), x[1]), small)
2115 elif op == 'dagrangepre':
2115 elif op == 'dagrangepre':
2116 return optimize(('func', ('symbol', 'ancestors'), x[1]), small)
2116 return optimize(('func', ('symbol', 'ancestors'), x[1]), small)
2117 elif op == 'dagrangepost':
2117 elif op == 'dagrangepost':
2118 return optimize(('func', ('symbol', 'descendants'), x[1]), small)
2118 return optimize(('func', ('symbol', 'descendants'), x[1]), small)
2119 elif op == 'rangepre':
2119 elif op == 'rangepre':
2120 return optimize(('range', ('string', '0'), x[1]), small)
2120 return optimize(('range', ('string', '0'), x[1]), small)
2121 elif op == 'rangepost':
2121 elif op == 'rangepost':
2122 return optimize(('range', x[1], ('string', 'tip')), small)
2122 return optimize(('range', x[1], ('string', 'tip')), small)
2123 elif op == 'negate':
2123 elif op == 'negate':
2124 return optimize(('string',
2124 return optimize(('string',
2125 '-' + getstring(x[1], _("can't negate that"))), small)
2125 '-' + getstring(x[1], _("can't negate that"))), small)
2126 elif op in 'string symbol negate':
2126 elif op in 'string symbol negate':
2127 return smallbonus, x # single revisions are small
2127 return smallbonus, x # single revisions are small
2128 elif op == 'and':
2128 elif op == 'and':
2129 wa, ta = optimize(x[1], True)
2129 wa, ta = optimize(x[1], True)
2130 wb, tb = optimize(x[2], True)
2130 wb, tb = optimize(x[2], True)
2131
2131
2132 # (::x and not ::y)/(not ::y and ::x) have a fast path
2132 # (::x and not ::y)/(not ::y and ::x) have a fast path
2133 def isonly(revs, bases):
2133 def isonly(revs, bases):
2134 return (
2134 return (
2135 revs[0] == 'func'
2135 revs[0] == 'func'
2136 and getstring(revs[1], _('not a symbol')) == 'ancestors'
2136 and getstring(revs[1], _('not a symbol')) == 'ancestors'
2137 and bases[0] == 'not'
2137 and bases[0] == 'not'
2138 and bases[1][0] == 'func'
2138 and bases[1][0] == 'func'
2139 and getstring(bases[1][1], _('not a symbol')) == 'ancestors')
2139 and getstring(bases[1][1], _('not a symbol')) == 'ancestors')
2140
2140
2141 w = min(wa, wb)
2141 w = min(wa, wb)
2142 if isonly(ta, tb):
2142 if isonly(ta, tb):
2143 return w, ('func', ('symbol', 'only'), ('list', ta[2], tb[1][2]))
2143 return w, ('func', ('symbol', 'only'), ('list', ta[2], tb[1][2]))
2144 if isonly(tb, ta):
2144 if isonly(tb, ta):
2145 return w, ('func', ('symbol', 'only'), ('list', tb[2], ta[1][2]))
2145 return w, ('func', ('symbol', 'only'), ('list', tb[2], ta[1][2]))
2146
2146
2147 if wa > wb:
2147 if wa > wb:
2148 return w, (op, tb, ta)
2148 return w, (op, tb, ta)
2149 return w, (op, ta, tb)
2149 return w, (op, ta, tb)
2150 elif op == 'or':
2150 elif op == 'or':
2151 wa, ta = optimize(x[1], False)
2151 wa, ta = optimize(x[1], False)
2152 wb, tb = optimize(x[2], False)
2152 wb, tb = optimize(x[2], False)
2153 if wb < wa:
2153 if wb < wa:
2154 wb, wa = wa, wb
2154 wb, wa = wa, wb
2155 return max(wa, wb), (op, ta, tb)
2155 return max(wa, wb), (op, ta, tb)
2156 elif op == 'not':
2156 elif op == 'not':
2157 o = optimize(x[1], not small)
2157 o = optimize(x[1], not small)
2158 return o[0], (op, o[1])
2158 return o[0], (op, o[1])
2159 elif op == 'parentpost':
2159 elif op == 'parentpost':
2160 o = optimize(x[1], small)
2160 o = optimize(x[1], small)
2161 return o[0], (op, o[1])
2161 return o[0], (op, o[1])
2162 elif op == 'group':
2162 elif op == 'group':
2163 return optimize(x[1], small)
2163 return optimize(x[1], small)
2164 elif op in 'dagrange range list parent ancestorspec':
2164 elif op in 'dagrange range list parent ancestorspec':
2165 if op == 'parent':
2165 if op == 'parent':
2166 # x^:y means (x^) : y, not x ^ (:y)
2166 # x^:y means (x^) : y, not x ^ (:y)
2167 post = ('parentpost', x[1])
2167 post = ('parentpost', x[1])
2168 if x[2][0] == 'dagrangepre':
2168 if x[2][0] == 'dagrangepre':
2169 return optimize(('dagrange', post, x[2][1]), small)
2169 return optimize(('dagrange', post, x[2][1]), small)
2170 elif x[2][0] == 'rangepre':
2170 elif x[2][0] == 'rangepre':
2171 return optimize(('range', post, x[2][1]), small)
2171 return optimize(('range', post, x[2][1]), small)
2172
2172
2173 wa, ta = optimize(x[1], small)
2173 wa, ta = optimize(x[1], small)
2174 wb, tb = optimize(x[2], small)
2174 wb, tb = optimize(x[2], small)
2175 return wa + wb, (op, ta, tb)
2175 return wa + wb, (op, ta, tb)
2176 elif op == 'func':
2176 elif op == 'func':
2177 f = getstring(x[1], _("not a symbol"))
2177 f = getstring(x[1], _("not a symbol"))
2178 wa, ta = optimize(x[2], small)
2178 wa, ta = optimize(x[2], small)
2179 if f in ("author branch closed date desc file grep keyword "
2179 if f in ("author branch closed date desc file grep keyword "
2180 "outgoing user"):
2180 "outgoing user"):
2181 w = 10 # slow
2181 w = 10 # slow
2182 elif f in "modifies adds removes":
2182 elif f in "modifies adds removes":
2183 w = 30 # slower
2183 w = 30 # slower
2184 elif f == "contains":
2184 elif f == "contains":
2185 w = 100 # very slow
2185 w = 100 # very slow
2186 elif f == "ancestor":
2186 elif f == "ancestor":
2187 w = 1 * smallbonus
2187 w = 1 * smallbonus
2188 elif f in "reverse limit first _intlist":
2188 elif f in "reverse limit first _intlist":
2189 w = 0
2189 w = 0
2190 elif f in "sort":
2190 elif f in "sort":
2191 w = 10 # assume most sorts look at changelog
2191 w = 10 # assume most sorts look at changelog
2192 else:
2192 else:
2193 w = 1
2193 w = 1
2194 return w + wa, (op, x[1], ta)
2194 return w + wa, (op, x[1], ta)
2195 return 1, x
2195 return 1, x
2196
2196
2197 _aliasarg = ('func', ('symbol', '_aliasarg'))
2197 _aliasarg = ('func', ('symbol', '_aliasarg'))
2198 def _getaliasarg(tree):
2198 def _getaliasarg(tree):
2199 """If tree matches ('func', ('symbol', '_aliasarg'), ('string', X))
2199 """If tree matches ('func', ('symbol', '_aliasarg'), ('string', X))
2200 return X, None otherwise.
2200 return X, None otherwise.
2201 """
2201 """
2202 if (len(tree) == 3 and tree[:2] == _aliasarg
2202 if (len(tree) == 3 and tree[:2] == _aliasarg
2203 and tree[2][0] == 'string'):
2203 and tree[2][0] == 'string'):
2204 return tree[2][1]
2204 return tree[2][1]
2205 return None
2205 return None
2206
2206
2207 def _checkaliasarg(tree, known=None):
2207 def _checkaliasarg(tree, known=None):
2208 """Check tree contains no _aliasarg construct or only ones which
2208 """Check tree contains no _aliasarg construct or only ones which
2209 value is in known. Used to avoid alias placeholders injection.
2209 value is in known. Used to avoid alias placeholders injection.
2210 """
2210 """
2211 if isinstance(tree, tuple):
2211 if isinstance(tree, tuple):
2212 arg = _getaliasarg(tree)
2212 arg = _getaliasarg(tree)
2213 if arg is not None and (not known or arg not in known):
2213 if arg is not None and (not known or arg not in known):
2214 raise error.UnknownIdentifier('_aliasarg', [])
2214 raise error.UnknownIdentifier('_aliasarg', [])
2215 for t in tree:
2215 for t in tree:
2216 _checkaliasarg(t, known)
2216 _checkaliasarg(t, known)
2217
2217
2218 # the set of valid characters for the initial letter of symbols in
2218 # the set of valid characters for the initial letter of symbols in
2219 # alias declarations and definitions
2219 # alias declarations and definitions
2220 _aliassyminitletters = set(c for c in [chr(i) for i in xrange(256)]
2220 _aliassyminitletters = set(c for c in [chr(i) for i in xrange(256)]
2221 if c.isalnum() or c in '._@$' or ord(c) > 127)
2221 if c.isalnum() or c in '._@$' or ord(c) > 127)
2222
2222
2223 def _tokenizealias(program, lookup=None):
2223 def _tokenizealias(program, lookup=None):
2224 """Parse alias declaration/definition into a stream of tokens
2224 """Parse alias declaration/definition into a stream of tokens
2225
2225
2226 This allows symbol names to use also ``$`` as an initial letter
2226 This allows symbol names to use also ``$`` as an initial letter
2227 (for backward compatibility), and callers of this function should
2227 (for backward compatibility), and callers of this function should
2228 examine whether ``$`` is used also for unexpected symbols or not.
2228 examine whether ``$`` is used also for unexpected symbols or not.
2229 """
2229 """
2230 return tokenize(program, lookup=lookup,
2230 return tokenize(program, lookup=lookup,
2231 syminitletters=_aliassyminitletters)
2231 syminitletters=_aliassyminitletters)
2232
2232
2233 def _parsealiasdecl(decl):
2233 def _parsealiasdecl(decl):
2234 """Parse alias declaration ``decl``
2234 """Parse alias declaration ``decl``
2235
2235
2236 This returns ``(name, tree, args, errorstr)`` tuple:
2236 This returns ``(name, tree, args, errorstr)`` tuple:
2237
2237
2238 - ``name``: of declared alias (may be ``decl`` itself at error)
2238 - ``name``: of declared alias (may be ``decl`` itself at error)
2239 - ``tree``: parse result (or ``None`` at error)
2239 - ``tree``: parse result (or ``None`` at error)
2240 - ``args``: list of alias argument names (or None for symbol declaration)
2240 - ``args``: list of alias argument names (or None for symbol declaration)
2241 - ``errorstr``: detail about detected error (or None)
2241 - ``errorstr``: detail about detected error (or None)
2242
2242
2243 >>> _parsealiasdecl('foo')
2243 >>> _parsealiasdecl('foo')
2244 ('foo', ('symbol', 'foo'), None, None)
2244 ('foo', ('symbol', 'foo'), None, None)
2245 >>> _parsealiasdecl('$foo')
2245 >>> _parsealiasdecl('$foo')
2246 ('$foo', None, None, "'$' not for alias arguments")
2246 ('$foo', None, None, "'$' not for alias arguments")
2247 >>> _parsealiasdecl('foo::bar')
2247 >>> _parsealiasdecl('foo::bar')
2248 ('foo::bar', None, None, 'invalid format')
2248 ('foo::bar', None, None, 'invalid format')
2249 >>> _parsealiasdecl('foo bar')
2249 >>> _parsealiasdecl('foo bar')
2250 ('foo bar', None, None, 'at 4: invalid token')
2250 ('foo bar', None, None, 'at 4: invalid token')
2251 >>> _parsealiasdecl('foo()')
2251 >>> _parsealiasdecl('foo()')
2252 ('foo', ('func', ('symbol', 'foo')), [], None)
2252 ('foo', ('func', ('symbol', 'foo')), [], None)
2253 >>> _parsealiasdecl('$foo()')
2253 >>> _parsealiasdecl('$foo()')
2254 ('$foo()', None, None, "'$' not for alias arguments")
2254 ('$foo()', None, None, "'$' not for alias arguments")
2255 >>> _parsealiasdecl('foo($1, $2)')
2255 >>> _parsealiasdecl('foo($1, $2)')
2256 ('foo', ('func', ('symbol', 'foo')), ['$1', '$2'], None)
2256 ('foo', ('func', ('symbol', 'foo')), ['$1', '$2'], None)
2257 >>> _parsealiasdecl('foo(bar_bar, baz.baz)')
2257 >>> _parsealiasdecl('foo(bar_bar, baz.baz)')
2258 ('foo', ('func', ('symbol', 'foo')), ['bar_bar', 'baz.baz'], None)
2258 ('foo', ('func', ('symbol', 'foo')), ['bar_bar', 'baz.baz'], None)
2259 >>> _parsealiasdecl('foo($1, $2, nested($1, $2))')
2259 >>> _parsealiasdecl('foo($1, $2, nested($1, $2))')
2260 ('foo($1, $2, nested($1, $2))', None, None, 'invalid argument list')
2260 ('foo($1, $2, nested($1, $2))', None, None, 'invalid argument list')
2261 >>> _parsealiasdecl('foo(bar($1, $2))')
2261 >>> _parsealiasdecl('foo(bar($1, $2))')
2262 ('foo(bar($1, $2))', None, None, 'invalid argument list')
2262 ('foo(bar($1, $2))', None, None, 'invalid argument list')
2263 >>> _parsealiasdecl('foo("string")')
2263 >>> _parsealiasdecl('foo("string")')
2264 ('foo("string")', None, None, 'invalid argument list')
2264 ('foo("string")', None, None, 'invalid argument list')
2265 >>> _parsealiasdecl('foo($1, $2')
2265 >>> _parsealiasdecl('foo($1, $2')
2266 ('foo($1, $2', None, None, 'at 10: unexpected token: end')
2266 ('foo($1, $2', None, None, 'at 10: unexpected token: end')
2267 >>> _parsealiasdecl('foo("string')
2267 >>> _parsealiasdecl('foo("string')
2268 ('foo("string', None, None, 'at 5: unterminated string')
2268 ('foo("string', None, None, 'at 5: unterminated string')
2269 >>> _parsealiasdecl('foo($1, $2, $1)')
2269 >>> _parsealiasdecl('foo($1, $2, $1)')
2270 ('foo', None, None, 'argument names collide with each other')
2270 ('foo', None, None, 'argument names collide with each other')
2271 """
2271 """
2272 p = parser.parser(_tokenizealias, elements)
2272 p = parser.parser(_tokenizealias, elements)
2273 try:
2273 try:
2274 tree, pos = p.parse(decl)
2274 tree, pos = p.parse(decl)
2275 if (pos != len(decl)):
2275 if (pos != len(decl)):
2276 raise error.ParseError(_('invalid token'), pos)
2276 raise error.ParseError(_('invalid token'), pos)
2277
2277
2278 if isvalidsymbol(tree):
2278 if isvalidsymbol(tree):
2279 # "name = ...." style
2279 # "name = ...." style
2280 name = getsymbol(tree)
2280 name = getsymbol(tree)
2281 if name.startswith('$'):
2281 if name.startswith('$'):
2282 return (decl, None, None, _("'$' not for alias arguments"))
2282 return (decl, None, None, _("'$' not for alias arguments"))
2283 return (name, ('symbol', name), None, None)
2283 return (name, ('symbol', name), None, None)
2284
2284
2285 if isvalidfunc(tree):
2285 if isvalidfunc(tree):
2286 # "name(arg, ....) = ...." style
2286 # "name(arg, ....) = ...." style
2287 name = getfuncname(tree)
2287 name = getfuncname(tree)
2288 if name.startswith('$'):
2288 if name.startswith('$'):
2289 return (decl, None, None, _("'$' not for alias arguments"))
2289 return (decl, None, None, _("'$' not for alias arguments"))
2290 args = []
2290 args = []
2291 for arg in getfuncargs(tree):
2291 for arg in getfuncargs(tree):
2292 if not isvalidsymbol(arg):
2292 if not isvalidsymbol(arg):
2293 return (decl, None, None, _("invalid argument list"))
2293 return (decl, None, None, _("invalid argument list"))
2294 args.append(getsymbol(arg))
2294 args.append(getsymbol(arg))
2295 if len(args) != len(set(args)):
2295 if len(args) != len(set(args)):
2296 return (name, None, None,
2296 return (name, None, None,
2297 _("argument names collide with each other"))
2297 _("argument names collide with each other"))
2298 return (name, ('func', ('symbol', name)), args, None)
2298 return (name, ('func', ('symbol', name)), args, None)
2299
2299
2300 return (decl, None, None, _("invalid format"))
2300 return (decl, None, None, _("invalid format"))
2301 except error.ParseError, inst:
2301 except error.ParseError, inst:
2302 return (decl, None, None, parseerrordetail(inst))
2302 return (decl, None, None, parseerrordetail(inst))
2303
2303
2304 def _parsealiasdefn(defn, args):
2304 def _parsealiasdefn(defn, args):
2305 """Parse alias definition ``defn``
2305 """Parse alias definition ``defn``
2306
2306
2307 This function also replaces alias argument references in the
2307 This function also replaces alias argument references in the
2308 specified definition by ``_aliasarg(ARGNAME)``.
2308 specified definition by ``_aliasarg(ARGNAME)``.
2309
2309
2310 ``args`` is a list of alias argument names, or None if the alias
2310 ``args`` is a list of alias argument names, or None if the alias
2311 is declared as a symbol.
2311 is declared as a symbol.
2312
2312
2313 This returns "tree" as parsing result.
2313 This returns "tree" as parsing result.
2314
2314
2315 >>> args = ['$1', '$2', 'foo']
2315 >>> args = ['$1', '$2', 'foo']
2316 >>> print prettyformat(_parsealiasdefn('$1 or foo', args))
2316 >>> print prettyformat(_parsealiasdefn('$1 or foo', args))
2317 (or
2317 (or
2318 (func
2318 (func
2319 ('symbol', '_aliasarg')
2319 ('symbol', '_aliasarg')
2320 ('string', '$1'))
2320 ('string', '$1'))
2321 (func
2321 (func
2322 ('symbol', '_aliasarg')
2322 ('symbol', '_aliasarg')
2323 ('string', 'foo')))
2323 ('string', 'foo')))
2324 >>> try:
2324 >>> try:
2325 ... _parsealiasdefn('$1 or $bar', args)
2325 ... _parsealiasdefn('$1 or $bar', args)
2326 ... except error.ParseError, inst:
2326 ... except error.ParseError, inst:
2327 ... print parseerrordetail(inst)
2327 ... print parseerrordetail(inst)
2328 at 6: '$' not for alias arguments
2328 at 6: '$' not for alias arguments
2329 >>> args = ['$1', '$10', 'foo']
2329 >>> args = ['$1', '$10', 'foo']
2330 >>> print prettyformat(_parsealiasdefn('$10 or foobar', args))
2330 >>> print prettyformat(_parsealiasdefn('$10 or foobar', args))
2331 (or
2331 (or
2332 (func
2332 (func
2333 ('symbol', '_aliasarg')
2333 ('symbol', '_aliasarg')
2334 ('string', '$10'))
2334 ('string', '$10'))
2335 ('symbol', 'foobar'))
2335 ('symbol', 'foobar'))
2336 >>> print prettyformat(_parsealiasdefn('"$1" or "foo"', args))
2336 >>> print prettyformat(_parsealiasdefn('"$1" or "foo"', args))
2337 (or
2337 (or
2338 ('string', '$1')
2338 ('string', '$1')
2339 ('string', 'foo'))
2339 ('string', 'foo'))
2340 """
2340 """
2341 def tokenizedefn(program, lookup=None):
2341 def tokenizedefn(program, lookup=None):
2342 if args:
2342 if args:
2343 argset = set(args)
2343 argset = set(args)
2344 else:
2344 else:
2345 argset = set()
2345 argset = set()
2346
2346
2347 for t, value, pos in _tokenizealias(program, lookup=lookup):
2347 for t, value, pos in _tokenizealias(program, lookup=lookup):
2348 if t == 'symbol':
2348 if t == 'symbol':
2349 if value in argset:
2349 if value in argset:
2350 # emulate tokenization of "_aliasarg('ARGNAME')":
2350 # emulate tokenization of "_aliasarg('ARGNAME')":
2351 # "_aliasarg()" is an unknown symbol only used separate
2351 # "_aliasarg()" is an unknown symbol only used separate
2352 # alias argument placeholders from regular strings.
2352 # alias argument placeholders from regular strings.
2353 yield ('symbol', '_aliasarg', pos)
2353 yield ('symbol', '_aliasarg', pos)
2354 yield ('(', None, pos)
2354 yield ('(', None, pos)
2355 yield ('string', value, pos)
2355 yield ('string', value, pos)
2356 yield (')', None, pos)
2356 yield (')', None, pos)
2357 continue
2357 continue
2358 elif value.startswith('$'):
2358 elif value.startswith('$'):
2359 raise error.ParseError(_("'$' not for alias arguments"),
2359 raise error.ParseError(_("'$' not for alias arguments"),
2360 pos)
2360 pos)
2361 yield (t, value, pos)
2361 yield (t, value, pos)
2362
2362
2363 p = parser.parser(tokenizedefn, elements)
2363 p = parser.parser(tokenizedefn, elements)
2364 tree, pos = p.parse(defn)
2364 tree, pos = p.parse(defn)
2365 if pos != len(defn):
2365 if pos != len(defn):
2366 raise error.ParseError(_('invalid token'), pos)
2366 raise error.ParseError(_('invalid token'), pos)
2367 return tree
2367 return tree
2368
2368
2369 class revsetalias(object):
2369 class revsetalias(object):
2370 # whether own `error` information is already shown or not.
2370 # whether own `error` information is already shown or not.
2371 # this avoids showing same warning multiple times at each `findaliases`.
2371 # this avoids showing same warning multiple times at each `findaliases`.
2372 warned = False
2372 warned = False
2373
2373
2374 def __init__(self, name, value):
2374 def __init__(self, name, value):
2375 '''Aliases like:
2375 '''Aliases like:
2376
2376
2377 h = heads(default)
2377 h = heads(default)
2378 b($1) = ancestors($1) - ancestors(default)
2378 b($1) = ancestors($1) - ancestors(default)
2379 '''
2379 '''
2380 self.name, self.tree, self.args, self.error = _parsealiasdecl(name)
2380 self.name, self.tree, self.args, self.error = _parsealiasdecl(name)
2381 if self.error:
2381 if self.error:
2382 self.error = _('failed to parse the declaration of revset alias'
2382 self.error = _('failed to parse the declaration of revset alias'
2383 ' "%s": %s') % (self.name, self.error)
2383 ' "%s": %s') % (self.name, self.error)
2384 return
2384 return
2385
2385
2386 try:
2386 try:
2387 self.replacement = _parsealiasdefn(value, self.args)
2387 self.replacement = _parsealiasdefn(value, self.args)
2388 # Check for placeholder injection
2388 # Check for placeholder injection
2389 _checkaliasarg(self.replacement, self.args)
2389 _checkaliasarg(self.replacement, self.args)
2390 except error.ParseError, inst:
2390 except error.ParseError, inst:
2391 self.error = _('failed to parse the definition of revset alias'
2391 self.error = _('failed to parse the definition of revset alias'
2392 ' "%s": %s') % (self.name, parseerrordetail(inst))
2392 ' "%s": %s') % (self.name, parseerrordetail(inst))
2393
2393
2394 def _getalias(aliases, tree):
2394 def _getalias(aliases, tree):
2395 """If tree looks like an unexpanded alias, return it. Return None
2395 """If tree looks like an unexpanded alias, return it. Return None
2396 otherwise.
2396 otherwise.
2397 """
2397 """
2398 if isinstance(tree, tuple) and tree:
2398 if isinstance(tree, tuple) and tree:
2399 if tree[0] == 'symbol' and len(tree) == 2:
2399 if tree[0] == 'symbol' and len(tree) == 2:
2400 name = tree[1]
2400 name = tree[1]
2401 alias = aliases.get(name)
2401 alias = aliases.get(name)
2402 if alias and alias.args is None and alias.tree == tree:
2402 if alias and alias.args is None and alias.tree == tree:
2403 return alias
2403 return alias
2404 if tree[0] == 'func' and len(tree) > 1:
2404 if tree[0] == 'func' and len(tree) > 1:
2405 if tree[1][0] == 'symbol' and len(tree[1]) == 2:
2405 if tree[1][0] == 'symbol' and len(tree[1]) == 2:
2406 name = tree[1][1]
2406 name = tree[1][1]
2407 alias = aliases.get(name)
2407 alias = aliases.get(name)
2408 if alias and alias.args is not None and alias.tree == tree[:2]:
2408 if alias and alias.args is not None and alias.tree == tree[:2]:
2409 return alias
2409 return alias
2410 return None
2410 return None
2411
2411
2412 def _expandargs(tree, args):
2412 def _expandargs(tree, args):
2413 """Replace _aliasarg instances with the substitution value of the
2413 """Replace _aliasarg instances with the substitution value of the
2414 same name in args, recursively.
2414 same name in args, recursively.
2415 """
2415 """
2416 if not tree or not isinstance(tree, tuple):
2416 if not tree or not isinstance(tree, tuple):
2417 return tree
2417 return tree
2418 arg = _getaliasarg(tree)
2418 arg = _getaliasarg(tree)
2419 if arg is not None:
2419 if arg is not None:
2420 return args[arg]
2420 return args[arg]
2421 return tuple(_expandargs(t, args) for t in tree)
2421 return tuple(_expandargs(t, args) for t in tree)
2422
2422
2423 def _expandaliases(aliases, tree, expanding, cache):
2423 def _expandaliases(aliases, tree, expanding, cache):
2424 """Expand aliases in tree, recursively.
2424 """Expand aliases in tree, recursively.
2425
2425
2426 'aliases' is a dictionary mapping user defined aliases to
2426 'aliases' is a dictionary mapping user defined aliases to
2427 revsetalias objects.
2427 revsetalias objects.
2428 """
2428 """
2429 if not isinstance(tree, tuple):
2429 if not isinstance(tree, tuple):
2430 # Do not expand raw strings
2430 # Do not expand raw strings
2431 return tree
2431 return tree
2432 alias = _getalias(aliases, tree)
2432 alias = _getalias(aliases, tree)
2433 if alias is not None:
2433 if alias is not None:
2434 if alias.error:
2434 if alias.error:
2435 raise util.Abort(alias.error)
2435 raise util.Abort(alias.error)
2436 if alias in expanding:
2436 if alias in expanding:
2437 raise error.ParseError(_('infinite expansion of revset alias "%s" '
2437 raise error.ParseError(_('infinite expansion of revset alias "%s" '
2438 'detected') % alias.name)
2438 'detected') % alias.name)
2439 expanding.append(alias)
2439 expanding.append(alias)
2440 if alias.name not in cache:
2440 if alias.name not in cache:
2441 cache[alias.name] = _expandaliases(aliases, alias.replacement,
2441 cache[alias.name] = _expandaliases(aliases, alias.replacement,
2442 expanding, cache)
2442 expanding, cache)
2443 result = cache[alias.name]
2443 result = cache[alias.name]
2444 expanding.pop()
2444 expanding.pop()
2445 if alias.args is not None:
2445 if alias.args is not None:
2446 l = getlist(tree[2])
2446 l = getlist(tree[2])
2447 if len(l) != len(alias.args):
2447 if len(l) != len(alias.args):
2448 raise error.ParseError(
2448 raise error.ParseError(
2449 _('invalid number of arguments: %s') % len(l))
2449 _('invalid number of arguments: %s') % len(l))
2450 l = [_expandaliases(aliases, a, [], cache) for a in l]
2450 l = [_expandaliases(aliases, a, [], cache) for a in l]
2451 result = _expandargs(result, dict(zip(alias.args, l)))
2451 result = _expandargs(result, dict(zip(alias.args, l)))
2452 else:
2452 else:
2453 result = tuple(_expandaliases(aliases, t, expanding, cache)
2453 result = tuple(_expandaliases(aliases, t, expanding, cache)
2454 for t in tree)
2454 for t in tree)
2455 return result
2455 return result
2456
2456
2457 def findaliases(ui, tree, showwarning=None):
2457 def findaliases(ui, tree, showwarning=None):
2458 _checkaliasarg(tree)
2458 _checkaliasarg(tree)
2459 aliases = {}
2459 aliases = {}
2460 for k, v in ui.configitems('revsetalias'):
2460 for k, v in ui.configitems('revsetalias'):
2461 alias = revsetalias(k, v)
2461 alias = revsetalias(k, v)
2462 aliases[alias.name] = alias
2462 aliases[alias.name] = alias
2463 tree = _expandaliases(aliases, tree, [], {})
2463 tree = _expandaliases(aliases, tree, [], {})
2464 if showwarning:
2464 if showwarning:
2465 # warn about problematic (but not referred) aliases
2465 # warn about problematic (but not referred) aliases
2466 for name, alias in sorted(aliases.iteritems()):
2466 for name, alias in sorted(aliases.iteritems()):
2467 if alias.error and not alias.warned:
2467 if alias.error and not alias.warned:
2468 showwarning(_('warning: %s\n') % (alias.error))
2468 showwarning(_('warning: %s\n') % (alias.error))
2469 alias.warned = True
2469 alias.warned = True
2470 return tree
2470 return tree
2471
2471
2472 def foldconcat(tree):
2472 def foldconcat(tree):
2473 """Fold elements to be concatenated by `##`
2473 """Fold elements to be concatenated by `##`
2474 """
2474 """
2475 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2475 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2476 return tree
2476 return tree
2477 if tree[0] == '_concat':
2477 if tree[0] == '_concat':
2478 pending = [tree]
2478 pending = [tree]
2479 l = []
2479 l = []
2480 while pending:
2480 while pending:
2481 e = pending.pop()
2481 e = pending.pop()
2482 if e[0] == '_concat':
2482 if e[0] == '_concat':
2483 pending.extend(reversed(e[1:]))
2483 pending.extend(reversed(e[1:]))
2484 elif e[0] in ('string', 'symbol'):
2484 elif e[0] in ('string', 'symbol'):
2485 l.append(e[1])
2485 l.append(e[1])
2486 else:
2486 else:
2487 msg = _("\"##\" can't concatenate \"%s\" element") % (e[0])
2487 msg = _("\"##\" can't concatenate \"%s\" element") % (e[0])
2488 raise error.ParseError(msg)
2488 raise error.ParseError(msg)
2489 return ('string', ''.join(l))
2489 return ('string', ''.join(l))
2490 else:
2490 else:
2491 return tuple(foldconcat(t) for t in tree)
2491 return tuple(foldconcat(t) for t in tree)
2492
2492
2493 def parse(spec, lookup=None):
2493 def parse(spec, lookup=None):
2494 p = parser.parser(tokenize, elements)
2494 p = parser.parser(tokenize, elements)
2495 return p.parse(spec, lookup=lookup)
2495 return p.parse(spec, lookup=lookup)
2496
2496
2497 def posttreebuilthook(tree, repo):
2497 def posttreebuilthook(tree, repo):
2498 # hook for extensions to execute code on the optimized tree
2498 # hook for extensions to execute code on the optimized tree
2499 pass
2499 pass
2500
2500
2501 def match(ui, spec, repo=None):
2501 def match(ui, spec, repo=None):
2502 if not spec:
2502 if not spec:
2503 raise error.ParseError(_("empty query"))
2503 raise error.ParseError(_("empty query"))
2504 lookup = None
2504 lookup = None
2505 if repo:
2505 if repo:
2506 lookup = repo.__contains__
2506 lookup = repo.__contains__
2507 tree, pos = parse(spec, lookup)
2507 tree, pos = parse(spec, lookup)
2508 if (pos != len(spec)):
2508 if (pos != len(spec)):
2509 raise error.ParseError(_("invalid token"), pos)
2509 raise error.ParseError(_("invalid token"), pos)
2510 if ui:
2510 if ui:
2511 tree = findaliases(ui, tree, showwarning=ui.warn)
2511 tree = findaliases(ui, tree, showwarning=ui.warn)
2512 tree = foldconcat(tree)
2512 tree = foldconcat(tree)
2513 weight, tree = optimize(tree, True)
2513 weight, tree = optimize(tree, True)
2514 posttreebuilthook(tree, repo)
2514 posttreebuilthook(tree, repo)
2515 def mfunc(repo, subset=None):
2515 def mfunc(repo, subset=None):
2516 if subset is None:
2516 if subset is None:
2517 subset = fullreposet(repo)
2517 subset = fullreposet(repo)
2518 if util.safehasattr(subset, 'isascending'):
2518 if util.safehasattr(subset, 'isascending'):
2519 result = getset(repo, subset, tree)
2519 result = getset(repo, subset, tree)
2520 else:
2520 else:
2521 result = getset(repo, baseset(subset), tree)
2521 result = getset(repo, baseset(subset), tree)
2522 return result
2522 return result
2523 return mfunc
2523 return mfunc
2524
2524
2525 def formatspec(expr, *args):
2525 def formatspec(expr, *args):
2526 '''
2526 '''
2527 This is a convenience function for using revsets internally, and
2527 This is a convenience function for using revsets internally, and
2528 escapes arguments appropriately. Aliases are intentionally ignored
2528 escapes arguments appropriately. Aliases are intentionally ignored
2529 so that intended expression behavior isn't accidentally subverted.
2529 so that intended expression behavior isn't accidentally subverted.
2530
2530
2531 Supported arguments:
2531 Supported arguments:
2532
2532
2533 %r = revset expression, parenthesized
2533 %r = revset expression, parenthesized
2534 %d = int(arg), no quoting
2534 %d = int(arg), no quoting
2535 %s = string(arg), escaped and single-quoted
2535 %s = string(arg), escaped and single-quoted
2536 %b = arg.branch(), escaped and single-quoted
2536 %b = arg.branch(), escaped and single-quoted
2537 %n = hex(arg), single-quoted
2537 %n = hex(arg), single-quoted
2538 %% = a literal '%'
2538 %% = a literal '%'
2539
2539
2540 Prefixing the type with 'l' specifies a parenthesized list of that type.
2540 Prefixing the type with 'l' specifies a parenthesized list of that type.
2541
2541
2542 >>> formatspec('%r:: and %lr', '10 or 11', ("this()", "that()"))
2542 >>> formatspec('%r:: and %lr', '10 or 11', ("this()", "that()"))
2543 '(10 or 11):: and ((this()) or (that()))'
2543 '(10 or 11):: and ((this()) or (that()))'
2544 >>> formatspec('%d:: and not %d::', 10, 20)
2544 >>> formatspec('%d:: and not %d::', 10, 20)
2545 '10:: and not 20::'
2545 '10:: and not 20::'
2546 >>> formatspec('%ld or %ld', [], [1])
2546 >>> formatspec('%ld or %ld', [], [1])
2547 "_list('') or 1"
2547 "_list('') or 1"
2548 >>> formatspec('keyword(%s)', 'foo\\xe9')
2548 >>> formatspec('keyword(%s)', 'foo\\xe9')
2549 "keyword('foo\\\\xe9')"
2549 "keyword('foo\\\\xe9')"
2550 >>> b = lambda: 'default'
2550 >>> b = lambda: 'default'
2551 >>> b.branch = b
2551 >>> b.branch = b
2552 >>> formatspec('branch(%b)', b)
2552 >>> formatspec('branch(%b)', b)
2553 "branch('default')"
2553 "branch('default')"
2554 >>> formatspec('root(%ls)', ['a', 'b', 'c', 'd'])
2554 >>> formatspec('root(%ls)', ['a', 'b', 'c', 'd'])
2555 "root(_list('a\\x00b\\x00c\\x00d'))"
2555 "root(_list('a\\x00b\\x00c\\x00d'))"
2556 '''
2556 '''
2557
2557
2558 def quote(s):
2558 def quote(s):
2559 return repr(str(s))
2559 return repr(str(s))
2560
2560
2561 def argtype(c, arg):
2561 def argtype(c, arg):
2562 if c == 'd':
2562 if c == 'd':
2563 return str(int(arg))
2563 return str(int(arg))
2564 elif c == 's':
2564 elif c == 's':
2565 return quote(arg)
2565 return quote(arg)
2566 elif c == 'r':
2566 elif c == 'r':
2567 parse(arg) # make sure syntax errors are confined
2567 parse(arg) # make sure syntax errors are confined
2568 return '(%s)' % arg
2568 return '(%s)' % arg
2569 elif c == 'n':
2569 elif c == 'n':
2570 return quote(node.hex(arg))
2570 return quote(node.hex(arg))
2571 elif c == 'b':
2571 elif c == 'b':
2572 return quote(arg.branch())
2572 return quote(arg.branch())
2573
2573
2574 def listexp(s, t):
2574 def listexp(s, t):
2575 l = len(s)
2575 l = len(s)
2576 if l == 0:
2576 if l == 0:
2577 return "_list('')"
2577 return "_list('')"
2578 elif l == 1:
2578 elif l == 1:
2579 return argtype(t, s[0])
2579 return argtype(t, s[0])
2580 elif t == 'd':
2580 elif t == 'd':
2581 return "_intlist('%s')" % "\0".join(str(int(a)) for a in s)
2581 return "_intlist('%s')" % "\0".join(str(int(a)) for a in s)
2582 elif t == 's':
2582 elif t == 's':
2583 return "_list('%s')" % "\0".join(s)
2583 return "_list('%s')" % "\0".join(s)
2584 elif t == 'n':
2584 elif t == 'n':
2585 return "_hexlist('%s')" % "\0".join(node.hex(a) for a in s)
2585 return "_hexlist('%s')" % "\0".join(node.hex(a) for a in s)
2586 elif t == 'b':
2586 elif t == 'b':
2587 return "_list('%s')" % "\0".join(a.branch() for a in s)
2587 return "_list('%s')" % "\0".join(a.branch() for a in s)
2588
2588
2589 m = l // 2
2589 m = l // 2
2590 return '(%s or %s)' % (listexp(s[:m], t), listexp(s[m:], t))
2590 return '(%s or %s)' % (listexp(s[:m], t), listexp(s[m:], t))
2591
2591
2592 ret = ''
2592 ret = ''
2593 pos = 0
2593 pos = 0
2594 arg = 0
2594 arg = 0
2595 while pos < len(expr):
2595 while pos < len(expr):
2596 c = expr[pos]
2596 c = expr[pos]
2597 if c == '%':
2597 if c == '%':
2598 pos += 1
2598 pos += 1
2599 d = expr[pos]
2599 d = expr[pos]
2600 if d == '%':
2600 if d == '%':
2601 ret += d
2601 ret += d
2602 elif d in 'dsnbr':
2602 elif d in 'dsnbr':
2603 ret += argtype(d, args[arg])
2603 ret += argtype(d, args[arg])
2604 arg += 1
2604 arg += 1
2605 elif d == 'l':
2605 elif d == 'l':
2606 # a list of some type
2606 # a list of some type
2607 pos += 1
2607 pos += 1
2608 d = expr[pos]
2608 d = expr[pos]
2609 ret += listexp(list(args[arg]), d)
2609 ret += listexp(list(args[arg]), d)
2610 arg += 1
2610 arg += 1
2611 else:
2611 else:
2612 raise util.Abort('unexpected revspec format character %s' % d)
2612 raise util.Abort('unexpected revspec format character %s' % d)
2613 else:
2613 else:
2614 ret += c
2614 ret += c
2615 pos += 1
2615 pos += 1
2616
2616
2617 return ret
2617 return ret
2618
2618
2619 def prettyformat(tree):
2619 def prettyformat(tree):
2620 def _prettyformat(tree, level, lines):
2620 def _prettyformat(tree, level, lines):
2621 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2621 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2622 lines.append((level, str(tree)))
2622 lines.append((level, str(tree)))
2623 else:
2623 else:
2624 lines.append((level, '(%s' % tree[0]))
2624 lines.append((level, '(%s' % tree[0]))
2625 for s in tree[1:]:
2625 for s in tree[1:]:
2626 _prettyformat(s, level + 1, lines)
2626 _prettyformat(s, level + 1, lines)
2627 lines[-1:] = [(lines[-1][0], lines[-1][1] + ')')]
2627 lines[-1:] = [(lines[-1][0], lines[-1][1] + ')')]
2628
2628
2629 lines = []
2629 lines = []
2630 _prettyformat(tree, 0, lines)
2630 _prettyformat(tree, 0, lines)
2631 output = '\n'.join((' '*l + s) for l, s in lines)
2631 output = '\n'.join((' '*l + s) for l, s in lines)
2632 return output
2632 return output
2633
2633
2634 def depth(tree):
2634 def depth(tree):
2635 if isinstance(tree, tuple):
2635 if isinstance(tree, tuple):
2636 return max(map(depth, tree)) + 1
2636 return max(map(depth, tree)) + 1
2637 else:
2637 else:
2638 return 0
2638 return 0
2639
2639
2640 def funcsused(tree):
2640 def funcsused(tree):
2641 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2641 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2642 return set()
2642 return set()
2643 else:
2643 else:
2644 funcs = set()
2644 funcs = set()
2645 for s in tree[1:]:
2645 for s in tree[1:]:
2646 funcs |= funcsused(s)
2646 funcs |= funcsused(s)
2647 if tree[0] == 'func':
2647 if tree[0] == 'func':
2648 funcs.add(tree[1][1])
2648 funcs.add(tree[1][1])
2649 return funcs
2649 return funcs
2650
2650
2651 class abstractsmartset(object):
2651 class abstractsmartset(object):
2652
2652
2653 def __nonzero__(self):
2653 def __nonzero__(self):
2654 """True if the smartset is not empty"""
2654 """True if the smartset is not empty"""
2655 raise NotImplementedError()
2655 raise NotImplementedError()
2656
2656
2657 def __contains__(self, rev):
2657 def __contains__(self, rev):
2658 """provide fast membership testing"""
2658 """provide fast membership testing"""
2659 raise NotImplementedError()
2659 raise NotImplementedError()
2660
2660
2661 def __iter__(self):
2661 def __iter__(self):
2662 """iterate the set in the order it is supposed to be iterated"""
2662 """iterate the set in the order it is supposed to be iterated"""
2663 raise NotImplementedError()
2663 raise NotImplementedError()
2664
2664
2665 # Attributes containing a function to perform a fast iteration in a given
2665 # Attributes containing a function to perform a fast iteration in a given
2666 # direction. A smartset can have none, one, or both defined.
2666 # direction. A smartset can have none, one, or both defined.
2667 #
2667 #
2668 # Default value is None instead of a function returning None to avoid
2668 # Default value is None instead of a function returning None to avoid
2669 # initializing an iterator just for testing if a fast method exists.
2669 # initializing an iterator just for testing if a fast method exists.
2670 fastasc = None
2670 fastasc = None
2671 fastdesc = None
2671 fastdesc = None
2672
2672
2673 def isascending(self):
2673 def isascending(self):
2674 """True if the set will iterate in ascending order"""
2674 """True if the set will iterate in ascending order"""
2675 raise NotImplementedError()
2675 raise NotImplementedError()
2676
2676
2677 def isdescending(self):
2677 def isdescending(self):
2678 """True if the set will iterate in descending order"""
2678 """True if the set will iterate in descending order"""
2679 raise NotImplementedError()
2679 raise NotImplementedError()
2680
2680
2681 def min(self):
2681 def min(self):
2682 """return the minimum element in the set"""
2682 """return the minimum element in the set"""
2683 if self.fastasc is not None:
2683 if self.fastasc is not None:
2684 for r in self.fastasc():
2684 for r in self.fastasc():
2685 return r
2685 return r
2686 raise ValueError('arg is an empty sequence')
2686 raise ValueError('arg is an empty sequence')
2687 return min(self)
2687 return min(self)
2688
2688
2689 def max(self):
2689 def max(self):
2690 """return the maximum element in the set"""
2690 """return the maximum element in the set"""
2691 if self.fastdesc is not None:
2691 if self.fastdesc is not None:
2692 for r in self.fastdesc():
2692 for r in self.fastdesc():
2693 return r
2693 return r
2694 raise ValueError('arg is an empty sequence')
2694 raise ValueError('arg is an empty sequence')
2695 return max(self)
2695 return max(self)
2696
2696
2697 def first(self):
2697 def first(self):
2698 """return the first element in the set (user iteration perspective)
2698 """return the first element in the set (user iteration perspective)
2699
2699
2700 Return None if the set is empty"""
2700 Return None if the set is empty"""
2701 raise NotImplementedError()
2701 raise NotImplementedError()
2702
2702
2703 def last(self):
2703 def last(self):
2704 """return the last element in the set (user iteration perspective)
2704 """return the last element in the set (user iteration perspective)
2705
2705
2706 Return None if the set is empty"""
2706 Return None if the set is empty"""
2707 raise NotImplementedError()
2707 raise NotImplementedError()
2708
2708
2709 def __len__(self):
2709 def __len__(self):
2710 """return the length of the smartsets
2710 """return the length of the smartsets
2711
2711
2712 This can be expensive on smartset that could be lazy otherwise."""
2712 This can be expensive on smartset that could be lazy otherwise."""
2713 raise NotImplementedError()
2713 raise NotImplementedError()
2714
2714
2715 def reverse(self):
2715 def reverse(self):
2716 """reverse the expected iteration order"""
2716 """reverse the expected iteration order"""
2717 raise NotImplementedError()
2717 raise NotImplementedError()
2718
2718
2719 def sort(self, reverse=True):
2719 def sort(self, reverse=True):
2720 """get the set to iterate in an ascending or descending order"""
2720 """get the set to iterate in an ascending or descending order"""
2721 raise NotImplementedError()
2721 raise NotImplementedError()
2722
2722
2723 def __and__(self, other):
2723 def __and__(self, other):
2724 """Returns a new object with the intersection of the two collections.
2724 """Returns a new object with the intersection of the two collections.
2725
2725
2726 This is part of the mandatory API for smartset."""
2726 This is part of the mandatory API for smartset."""
2727 if isinstance(other, fullreposet):
2727 if isinstance(other, fullreposet):
2728 return self
2728 return self
2729 return self.filter(other.__contains__, cache=False)
2729 return self.filter(other.__contains__, cache=False)
2730
2730
2731 def __add__(self, other):
2731 def __add__(self, other):
2732 """Returns a new object with the union of the two collections.
2732 """Returns a new object with the union of the two collections.
2733
2733
2734 This is part of the mandatory API for smartset."""
2734 This is part of the mandatory API for smartset."""
2735 return addset(self, other)
2735 return addset(self, other)
2736
2736
2737 def __sub__(self, other):
2737 def __sub__(self, other):
2738 """Returns a new object with the substraction of the two collections.
2738 """Returns a new object with the substraction of the two collections.
2739
2739
2740 This is part of the mandatory API for smartset."""
2740 This is part of the mandatory API for smartset."""
2741 c = other.__contains__
2741 c = other.__contains__
2742 return self.filter(lambda r: not c(r), cache=False)
2742 return self.filter(lambda r: not c(r), cache=False)
2743
2743
2744 def filter(self, condition, cache=True):
2744 def filter(self, condition, cache=True):
2745 """Returns this smartset filtered by condition as a new smartset.
2745 """Returns this smartset filtered by condition as a new smartset.
2746
2746
2747 `condition` is a callable which takes a revision number and returns a
2747 `condition` is a callable which takes a revision number and returns a
2748 boolean.
2748 boolean.
2749
2749
2750 This is part of the mandatory API for smartset."""
2750 This is part of the mandatory API for smartset."""
2751 # builtin cannot be cached. but do not needs to
2751 # builtin cannot be cached. but do not needs to
2752 if cache and util.safehasattr(condition, 'func_code'):
2752 if cache and util.safehasattr(condition, 'func_code'):
2753 condition = util.cachefunc(condition)
2753 condition = util.cachefunc(condition)
2754 return filteredset(self, condition)
2754 return filteredset(self, condition)
2755
2755
2756 class baseset(abstractsmartset):
2756 class baseset(abstractsmartset):
2757 """Basic data structure that represents a revset and contains the basic
2757 """Basic data structure that represents a revset and contains the basic
2758 operation that it should be able to perform.
2758 operation that it should be able to perform.
2759
2759
2760 Every method in this class should be implemented by any smartset class.
2760 Every method in this class should be implemented by any smartset class.
2761 """
2761 """
2762 def __init__(self, data=()):
2762 def __init__(self, data=()):
2763 if not isinstance(data, list):
2763 if not isinstance(data, list):
2764 data = list(data)
2764 data = list(data)
2765 self._list = data
2765 self._list = data
2766 self._ascending = None
2766 self._ascending = None
2767
2767
2768 @util.propertycache
2768 @util.propertycache
2769 def _set(self):
2769 def _set(self):
2770 return set(self._list)
2770 return set(self._list)
2771
2771
2772 @util.propertycache
2772 @util.propertycache
2773 def _asclist(self):
2773 def _asclist(self):
2774 asclist = self._list[:]
2774 asclist = self._list[:]
2775 asclist.sort()
2775 asclist.sort()
2776 return asclist
2776 return asclist
2777
2777
2778 def __iter__(self):
2778 def __iter__(self):
2779 if self._ascending is None:
2779 if self._ascending is None:
2780 return iter(self._list)
2780 return iter(self._list)
2781 elif self._ascending:
2781 elif self._ascending:
2782 return iter(self._asclist)
2782 return iter(self._asclist)
2783 else:
2783 else:
2784 return reversed(self._asclist)
2784 return reversed(self._asclist)
2785
2785
2786 def fastasc(self):
2786 def fastasc(self):
2787 return iter(self._asclist)
2787 return iter(self._asclist)
2788
2788
2789 def fastdesc(self):
2789 def fastdesc(self):
2790 return reversed(self._asclist)
2790 return reversed(self._asclist)
2791
2791
2792 @util.propertycache
2792 @util.propertycache
2793 def __contains__(self):
2793 def __contains__(self):
2794 return self._set.__contains__
2794 return self._set.__contains__
2795
2795
2796 def __nonzero__(self):
2796 def __nonzero__(self):
2797 return bool(self._list)
2797 return bool(self._list)
2798
2798
2799 def sort(self, reverse=False):
2799 def sort(self, reverse=False):
2800 self._ascending = not bool(reverse)
2800 self._ascending = not bool(reverse)
2801
2801
2802 def reverse(self):
2802 def reverse(self):
2803 if self._ascending is None:
2803 if self._ascending is None:
2804 self._list.reverse()
2804 self._list.reverse()
2805 else:
2805 else:
2806 self._ascending = not self._ascending
2806 self._ascending = not self._ascending
2807
2807
2808 def __len__(self):
2808 def __len__(self):
2809 return len(self._list)
2809 return len(self._list)
2810
2810
2811 def isascending(self):
2811 def isascending(self):
2812 """Returns True if the collection is ascending order, False if not.
2812 """Returns True if the collection is ascending order, False if not.
2813
2813
2814 This is part of the mandatory API for smartset."""
2814 This is part of the mandatory API for smartset."""
2815 if len(self) <= 1:
2815 if len(self) <= 1:
2816 return True
2816 return True
2817 return self._ascending is not None and self._ascending
2817 return self._ascending is not None and self._ascending
2818
2818
2819 def isdescending(self):
2819 def isdescending(self):
2820 """Returns True if the collection is descending order, False if not.
2820 """Returns True if the collection is descending order, False if not.
2821
2821
2822 This is part of the mandatory API for smartset."""
2822 This is part of the mandatory API for smartset."""
2823 if len(self) <= 1:
2823 if len(self) <= 1:
2824 return True
2824 return True
2825 return self._ascending is not None and not self._ascending
2825 return self._ascending is not None and not self._ascending
2826
2826
2827 def first(self):
2827 def first(self):
2828 if self:
2828 if self:
2829 if self._ascending is None:
2829 if self._ascending is None:
2830 return self._list[0]
2830 return self._list[0]
2831 elif self._ascending:
2831 elif self._ascending:
2832 return self._asclist[0]
2832 return self._asclist[0]
2833 else:
2833 else:
2834 return self._asclist[-1]
2834 return self._asclist[-1]
2835 return None
2835 return None
2836
2836
2837 def last(self):
2837 def last(self):
2838 if self:
2838 if self:
2839 if self._ascending is None:
2839 if self._ascending is None:
2840 return self._list[-1]
2840 return self._list[-1]
2841 elif self._ascending:
2841 elif self._ascending:
2842 return self._asclist[-1]
2842 return self._asclist[-1]
2843 else:
2843 else:
2844 return self._asclist[0]
2844 return self._asclist[0]
2845 return None
2845 return None
2846
2846
2847 def __repr__(self):
2847 def __repr__(self):
2848 d = {None: '', False: '-', True: '+'}[self._ascending]
2848 d = {None: '', False: '-', True: '+'}[self._ascending]
2849 return '<%s%s %r>' % (type(self).__name__, d, self._list)
2849 return '<%s%s %r>' % (type(self).__name__, d, self._list)
2850
2850
2851 class filteredset(abstractsmartset):
2851 class filteredset(abstractsmartset):
2852 """Duck type for baseset class which iterates lazily over the revisions in
2852 """Duck type for baseset class which iterates lazily over the revisions in
2853 the subset and contains a function which tests for membership in the
2853 the subset and contains a function which tests for membership in the
2854 revset
2854 revset
2855 """
2855 """
2856 def __init__(self, subset, condition=lambda x: True):
2856 def __init__(self, subset, condition=lambda x: True):
2857 """
2857 """
2858 condition: a function that decide whether a revision in the subset
2858 condition: a function that decide whether a revision in the subset
2859 belongs to the revset or not.
2859 belongs to the revset or not.
2860 """
2860 """
2861 self._subset = subset
2861 self._subset = subset
2862 self._condition = condition
2862 self._condition = condition
2863 self._cache = {}
2863 self._cache = {}
2864
2864
2865 def __contains__(self, x):
2865 def __contains__(self, x):
2866 c = self._cache
2866 c = self._cache
2867 if x not in c:
2867 if x not in c:
2868 v = c[x] = x in self._subset and self._condition(x)
2868 v = c[x] = x in self._subset and self._condition(x)
2869 return v
2869 return v
2870 return c[x]
2870 return c[x]
2871
2871
2872 def __iter__(self):
2872 def __iter__(self):
2873 return self._iterfilter(self._subset)
2873 return self._iterfilter(self._subset)
2874
2874
2875 def _iterfilter(self, it):
2875 def _iterfilter(self, it):
2876 cond = self._condition
2876 cond = self._condition
2877 for x in it:
2877 for x in it:
2878 if cond(x):
2878 if cond(x):
2879 yield x
2879 yield x
2880
2880
2881 @property
2881 @property
2882 def fastasc(self):
2882 def fastasc(self):
2883 it = self._subset.fastasc
2883 it = self._subset.fastasc
2884 if it is None:
2884 if it is None:
2885 return None
2885 return None
2886 return lambda: self._iterfilter(it())
2886 return lambda: self._iterfilter(it())
2887
2887
2888 @property
2888 @property
2889 def fastdesc(self):
2889 def fastdesc(self):
2890 it = self._subset.fastdesc
2890 it = self._subset.fastdesc
2891 if it is None:
2891 if it is None:
2892 return None
2892 return None
2893 return lambda: self._iterfilter(it())
2893 return lambda: self._iterfilter(it())
2894
2894
2895 def __nonzero__(self):
2895 def __nonzero__(self):
2896 for r in self:
2896 for r in self:
2897 return True
2897 return True
2898 return False
2898 return False
2899
2899
2900 def __len__(self):
2900 def __len__(self):
2901 # Basic implementation to be changed in future patches.
2901 # Basic implementation to be changed in future patches.
2902 l = baseset([r for r in self])
2902 l = baseset([r for r in self])
2903 return len(l)
2903 return len(l)
2904
2904
2905 def sort(self, reverse=False):
2905 def sort(self, reverse=False):
2906 self._subset.sort(reverse=reverse)
2906 self._subset.sort(reverse=reverse)
2907
2907
2908 def reverse(self):
2908 def reverse(self):
2909 self._subset.reverse()
2909 self._subset.reverse()
2910
2910
2911 def isascending(self):
2911 def isascending(self):
2912 return self._subset.isascending()
2912 return self._subset.isascending()
2913
2913
2914 def isdescending(self):
2914 def isdescending(self):
2915 return self._subset.isdescending()
2915 return self._subset.isdescending()
2916
2916
2917 def first(self):
2917 def first(self):
2918 for x in self:
2918 for x in self:
2919 return x
2919 return x
2920 return None
2920 return None
2921
2921
2922 def last(self):
2922 def last(self):
2923 it = None
2923 it = None
2924 if self._subset.isascending:
2924 if self._subset.isascending:
2925 it = self.fastdesc
2925 it = self.fastdesc
2926 elif self._subset.isdescending:
2926 elif self._subset.isdescending:
2927 it = self.fastdesc
2927 it = self.fastdesc
2928 if it is None:
2928 if it is None:
2929 # slowly consume everything. This needs improvement
2929 # slowly consume everything. This needs improvement
2930 it = lambda: reversed(list(self))
2930 it = lambda: reversed(list(self))
2931 for x in it():
2931 for x in it():
2932 return x
2932 return x
2933 return None
2933 return None
2934
2934
2935 def __repr__(self):
2935 def __repr__(self):
2936 return '<%s %r>' % (type(self).__name__, self._subset)
2936 return '<%s %r>' % (type(self).__name__, self._subset)
2937
2937
2938 class addset(abstractsmartset):
2938 class addset(abstractsmartset):
2939 """Represent the addition of two sets
2939 """Represent the addition of two sets
2940
2940
2941 Wrapper structure for lazily adding two structures without losing much
2941 Wrapper structure for lazily adding two structures without losing much
2942 performance on the __contains__ method
2942 performance on the __contains__ method
2943
2943
2944 If the ascending attribute is set, that means the two structures are
2944 If the ascending attribute is set, that means the two structures are
2945 ordered in either an ascending or descending way. Therefore, we can add
2945 ordered in either an ascending or descending way. Therefore, we can add
2946 them maintaining the order by iterating over both at the same time
2946 them maintaining the order by iterating over both at the same time
2947
2947
2948 >>> xs = baseset([0, 3, 2])
2948 >>> xs = baseset([0, 3, 2])
2949 >>> ys = baseset([5, 2, 4])
2949 >>> ys = baseset([5, 2, 4])
2950
2950
2951 >>> rs = addset(xs, ys)
2951 >>> rs = addset(xs, ys)
2952 >>> bool(rs), 0 in rs, 1 in rs, 5 in rs, rs.first(), rs.last()
2952 >>> bool(rs), 0 in rs, 1 in rs, 5 in rs, rs.first(), rs.last()
2953 (True, True, False, True, 0, 4)
2953 (True, True, False, True, 0, 4)
2954 >>> rs = addset(xs, baseset([]))
2954 >>> rs = addset(xs, baseset([]))
2955 >>> bool(rs), 0 in rs, 1 in rs, rs.first(), rs.last()
2955 >>> bool(rs), 0 in rs, 1 in rs, rs.first(), rs.last()
2956 (True, True, False, 0, 2)
2956 (True, True, False, 0, 2)
2957 >>> rs = addset(baseset([]), baseset([]))
2957 >>> rs = addset(baseset([]), baseset([]))
2958 >>> bool(rs), 0 in rs, rs.first(), rs.last()
2958 >>> bool(rs), 0 in rs, rs.first(), rs.last()
2959 (False, False, None, None)
2959 (False, False, None, None)
2960
2960
2961 iterate unsorted:
2961 iterate unsorted:
2962 >>> rs = addset(xs, ys)
2962 >>> rs = addset(xs, ys)
2963 >>> [x for x in rs] # without _genlist
2963 >>> [x for x in rs] # without _genlist
2964 [0, 3, 2, 5, 4]
2964 [0, 3, 2, 5, 4]
2965 >>> assert not rs._genlist
2965 >>> assert not rs._genlist
2966 >>> len(rs)
2966 >>> len(rs)
2967 5
2967 5
2968 >>> [x for x in rs] # with _genlist
2968 >>> [x for x in rs] # with _genlist
2969 [0, 3, 2, 5, 4]
2969 [0, 3, 2, 5, 4]
2970 >>> assert rs._genlist
2970 >>> assert rs._genlist
2971
2971
2972 iterate ascending:
2972 iterate ascending:
2973 >>> rs = addset(xs, ys, ascending=True)
2973 >>> rs = addset(xs, ys, ascending=True)
2974 >>> [x for x in rs], [x for x in rs.fastasc()] # without _asclist
2974 >>> [x for x in rs], [x for x in rs.fastasc()] # without _asclist
2975 ([0, 2, 3, 4, 5], [0, 2, 3, 4, 5])
2975 ([0, 2, 3, 4, 5], [0, 2, 3, 4, 5])
2976 >>> assert not rs._asclist
2976 >>> assert not rs._asclist
2977 >>> len(rs)
2977 >>> len(rs)
2978 5
2978 5
2979 >>> [x for x in rs], [x for x in rs.fastasc()]
2979 >>> [x for x in rs], [x for x in rs.fastasc()]
2980 ([0, 2, 3, 4, 5], [0, 2, 3, 4, 5])
2980 ([0, 2, 3, 4, 5], [0, 2, 3, 4, 5])
2981 >>> assert rs._asclist
2981 >>> assert rs._asclist
2982
2982
2983 iterate descending:
2983 iterate descending:
2984 >>> rs = addset(xs, ys, ascending=False)
2984 >>> rs = addset(xs, ys, ascending=False)
2985 >>> [x for x in rs], [x for x in rs.fastdesc()] # without _asclist
2985 >>> [x for x in rs], [x for x in rs.fastdesc()] # without _asclist
2986 ([5, 4, 3, 2, 0], [5, 4, 3, 2, 0])
2986 ([5, 4, 3, 2, 0], [5, 4, 3, 2, 0])
2987 >>> assert not rs._asclist
2987 >>> assert not rs._asclist
2988 >>> len(rs)
2988 >>> len(rs)
2989 5
2989 5
2990 >>> [x for x in rs], [x for x in rs.fastdesc()]
2990 >>> [x for x in rs], [x for x in rs.fastdesc()]
2991 ([5, 4, 3, 2, 0], [5, 4, 3, 2, 0])
2991 ([5, 4, 3, 2, 0], [5, 4, 3, 2, 0])
2992 >>> assert rs._asclist
2992 >>> assert rs._asclist
2993
2993
2994 iterate ascending without fastasc:
2994 iterate ascending without fastasc:
2995 >>> rs = addset(xs, generatorset(ys), ascending=True)
2995 >>> rs = addset(xs, generatorset(ys), ascending=True)
2996 >>> assert rs.fastasc is None
2996 >>> assert rs.fastasc is None
2997 >>> [x for x in rs]
2997 >>> [x for x in rs]
2998 [0, 2, 3, 4, 5]
2998 [0, 2, 3, 4, 5]
2999
2999
3000 iterate descending without fastdesc:
3000 iterate descending without fastdesc:
3001 >>> rs = addset(generatorset(xs), ys, ascending=False)
3001 >>> rs = addset(generatorset(xs), ys, ascending=False)
3002 >>> assert rs.fastdesc is None
3002 >>> assert rs.fastdesc is None
3003 >>> [x for x in rs]
3003 >>> [x for x in rs]
3004 [5, 4, 3, 2, 0]
3004 [5, 4, 3, 2, 0]
3005 """
3005 """
3006 def __init__(self, revs1, revs2, ascending=None):
3006 def __init__(self, revs1, revs2, ascending=None):
3007 self._r1 = revs1
3007 self._r1 = revs1
3008 self._r2 = revs2
3008 self._r2 = revs2
3009 self._iter = None
3009 self._iter = None
3010 self._ascending = ascending
3010 self._ascending = ascending
3011 self._genlist = None
3011 self._genlist = None
3012 self._asclist = None
3012 self._asclist = None
3013
3013
3014 def __len__(self):
3014 def __len__(self):
3015 return len(self._list)
3015 return len(self._list)
3016
3016
3017 def __nonzero__(self):
3017 def __nonzero__(self):
3018 return bool(self._r1) or bool(self._r2)
3018 return bool(self._r1) or bool(self._r2)
3019
3019
3020 @util.propertycache
3020 @util.propertycache
3021 def _list(self):
3021 def _list(self):
3022 if not self._genlist:
3022 if not self._genlist:
3023 self._genlist = baseset(iter(self))
3023 self._genlist = baseset(iter(self))
3024 return self._genlist
3024 return self._genlist
3025
3025
3026 def __iter__(self):
3026 def __iter__(self):
3027 """Iterate over both collections without repeating elements
3027 """Iterate over both collections without repeating elements
3028
3028
3029 If the ascending attribute is not set, iterate over the first one and
3029 If the ascending attribute is not set, iterate over the first one and
3030 then over the second one checking for membership on the first one so we
3030 then over the second one checking for membership on the first one so we
3031 dont yield any duplicates.
3031 dont yield any duplicates.
3032
3032
3033 If the ascending attribute is set, iterate over both collections at the
3033 If the ascending attribute is set, iterate over both collections at the
3034 same time, yielding only one value at a time in the given order.
3034 same time, yielding only one value at a time in the given order.
3035 """
3035 """
3036 if self._ascending is None:
3036 if self._ascending is None:
3037 if self._genlist:
3037 if self._genlist:
3038 return iter(self._genlist)
3038 return iter(self._genlist)
3039 def arbitraryordergen():
3039 def arbitraryordergen():
3040 for r in self._r1:
3040 for r in self._r1:
3041 yield r
3041 yield r
3042 inr1 = self._r1.__contains__
3042 inr1 = self._r1.__contains__
3043 for r in self._r2:
3043 for r in self._r2:
3044 if not inr1(r):
3044 if not inr1(r):
3045 yield r
3045 yield r
3046 return arbitraryordergen()
3046 return arbitraryordergen()
3047 # try to use our own fast iterator if it exists
3047 # try to use our own fast iterator if it exists
3048 self._trysetasclist()
3048 self._trysetasclist()
3049 if self._ascending:
3049 if self._ascending:
3050 it = self.fastasc
3050 attr = 'fastasc'
3051 else:
3051 else:
3052 it = self.fastdesc
3052 attr = 'fastdesc'
3053 it = getattr(self, attr)
3053 if it is not None:
3054 if it is not None:
3054 return it()
3055 return it()
3055 # maybe half of the component supports fast
3056 # maybe half of the component supports fast
3056 attr = 'fastdesc'
3057 if self._ascending:
3058 attr = 'fastasc'
3059 # get iterator for _r1
3057 # get iterator for _r1
3060 iter1 = getattr(self._r1, attr)
3058 iter1 = getattr(self._r1, attr)
3061 if iter1 is None:
3059 if iter1 is None:
3062 # let's avoid side effect (not sure it matters)
3060 # let's avoid side effect (not sure it matters)
3063 iter1 = iter(sorted(self._r1, reverse=not self._ascending))
3061 iter1 = iter(sorted(self._r1, reverse=not self._ascending))
3064 else:
3062 else:
3065 iter1 = iter1()
3063 iter1 = iter1()
3066 # get iterator for _r2
3064 # get iterator for _r2
3067 iter2 = getattr(self._r2, attr)
3065 iter2 = getattr(self._r2, attr)
3068 if iter2 is None:
3066 if iter2 is None:
3069 # let's avoid side effect (not sure it matters)
3067 # let's avoid side effect (not sure it matters)
3070 iter2 = iter(sorted(self._r2, reverse=not self._ascending))
3068 iter2 = iter(sorted(self._r2, reverse=not self._ascending))
3071 else:
3069 else:
3072 iter2 = iter2()
3070 iter2 = iter2()
3073 return self._iterordered(self._ascending, iter1, iter2)
3071 return self._iterordered(self._ascending, iter1, iter2)
3074
3072
3075 def _trysetasclist(self):
3073 def _trysetasclist(self):
3076 """populate the _asclist attribute if possible and necessary"""
3074 """populate the _asclist attribute if possible and necessary"""
3077 if self._genlist is not None and self._asclist is None:
3075 if self._genlist is not None and self._asclist is None:
3078 self._asclist = sorted(self._genlist)
3076 self._asclist = sorted(self._genlist)
3079
3077
3080 @property
3078 @property
3081 def fastasc(self):
3079 def fastasc(self):
3082 self._trysetasclist()
3080 self._trysetasclist()
3083 if self._asclist is not None:
3081 if self._asclist is not None:
3084 return self._asclist.__iter__
3082 return self._asclist.__iter__
3085 iter1 = self._r1.fastasc
3083 iter1 = self._r1.fastasc
3086 iter2 = self._r2.fastasc
3084 iter2 = self._r2.fastasc
3087 if None in (iter1, iter2):
3085 if None in (iter1, iter2):
3088 return None
3086 return None
3089 return lambda: self._iterordered(True, iter1(), iter2())
3087 return lambda: self._iterordered(True, iter1(), iter2())
3090
3088
3091 @property
3089 @property
3092 def fastdesc(self):
3090 def fastdesc(self):
3093 self._trysetasclist()
3091 self._trysetasclist()
3094 if self._asclist is not None:
3092 if self._asclist is not None:
3095 return self._asclist.__reversed__
3093 return self._asclist.__reversed__
3096 iter1 = self._r1.fastdesc
3094 iter1 = self._r1.fastdesc
3097 iter2 = self._r2.fastdesc
3095 iter2 = self._r2.fastdesc
3098 if None in (iter1, iter2):
3096 if None in (iter1, iter2):
3099 return None
3097 return None
3100 return lambda: self._iterordered(False, iter1(), iter2())
3098 return lambda: self._iterordered(False, iter1(), iter2())
3101
3099
3102 def _iterordered(self, ascending, iter1, iter2):
3100 def _iterordered(self, ascending, iter1, iter2):
3103 """produce an ordered iteration from two iterators with the same order
3101 """produce an ordered iteration from two iterators with the same order
3104
3102
3105 The ascending is used to indicated the iteration direction.
3103 The ascending is used to indicated the iteration direction.
3106 """
3104 """
3107 choice = max
3105 choice = max
3108 if ascending:
3106 if ascending:
3109 choice = min
3107 choice = min
3110
3108
3111 val1 = None
3109 val1 = None
3112 val2 = None
3110 val2 = None
3113 try:
3111 try:
3114 # Consume both iterators in an ordered way until one is
3112 # Consume both iterators in an ordered way until one is
3115 # empty
3113 # empty
3116 while True:
3114 while True:
3117 if val1 is None:
3115 if val1 is None:
3118 val1 = iter1.next()
3116 val1 = iter1.next()
3119 if val2 is None:
3117 if val2 is None:
3120 val2 = iter2.next()
3118 val2 = iter2.next()
3121 next = choice(val1, val2)
3119 next = choice(val1, val2)
3122 yield next
3120 yield next
3123 if val1 == next:
3121 if val1 == next:
3124 val1 = None
3122 val1 = None
3125 if val2 == next:
3123 if val2 == next:
3126 val2 = None
3124 val2 = None
3127 except StopIteration:
3125 except StopIteration:
3128 # Flush any remaining values and consume the other one
3126 # Flush any remaining values and consume the other one
3129 it = iter2
3127 it = iter2
3130 if val1 is not None:
3128 if val1 is not None:
3131 yield val1
3129 yield val1
3132 it = iter1
3130 it = iter1
3133 elif val2 is not None:
3131 elif val2 is not None:
3134 # might have been equality and both are empty
3132 # might have been equality and both are empty
3135 yield val2
3133 yield val2
3136 for val in it:
3134 for val in it:
3137 yield val
3135 yield val
3138
3136
3139 def __contains__(self, x):
3137 def __contains__(self, x):
3140 return x in self._r1 or x in self._r2
3138 return x in self._r1 or x in self._r2
3141
3139
3142 def sort(self, reverse=False):
3140 def sort(self, reverse=False):
3143 """Sort the added set
3141 """Sort the added set
3144
3142
3145 For this we use the cached list with all the generated values and if we
3143 For this we use the cached list with all the generated values and if we
3146 know they are ascending or descending we can sort them in a smart way.
3144 know they are ascending or descending we can sort them in a smart way.
3147 """
3145 """
3148 self._ascending = not reverse
3146 self._ascending = not reverse
3149
3147
3150 def isascending(self):
3148 def isascending(self):
3151 return self._ascending is not None and self._ascending
3149 return self._ascending is not None and self._ascending
3152
3150
3153 def isdescending(self):
3151 def isdescending(self):
3154 return self._ascending is not None and not self._ascending
3152 return self._ascending is not None and not self._ascending
3155
3153
3156 def reverse(self):
3154 def reverse(self):
3157 if self._ascending is None:
3155 if self._ascending is None:
3158 self._list.reverse()
3156 self._list.reverse()
3159 else:
3157 else:
3160 self._ascending = not self._ascending
3158 self._ascending = not self._ascending
3161
3159
3162 def first(self):
3160 def first(self):
3163 for x in self:
3161 for x in self:
3164 return x
3162 return x
3165 return None
3163 return None
3166
3164
3167 def last(self):
3165 def last(self):
3168 self.reverse()
3166 self.reverse()
3169 val = self.first()
3167 val = self.first()
3170 self.reverse()
3168 self.reverse()
3171 return val
3169 return val
3172
3170
3173 def __repr__(self):
3171 def __repr__(self):
3174 d = {None: '', False: '-', True: '+'}[self._ascending]
3172 d = {None: '', False: '-', True: '+'}[self._ascending]
3175 return '<%s%s %r, %r>' % (type(self).__name__, d, self._r1, self._r2)
3173 return '<%s%s %r, %r>' % (type(self).__name__, d, self._r1, self._r2)
3176
3174
3177 class generatorset(abstractsmartset):
3175 class generatorset(abstractsmartset):
3178 """Wrap a generator for lazy iteration
3176 """Wrap a generator for lazy iteration
3179
3177
3180 Wrapper structure for generators that provides lazy membership and can
3178 Wrapper structure for generators that provides lazy membership and can
3181 be iterated more than once.
3179 be iterated more than once.
3182 When asked for membership it generates values until either it finds the
3180 When asked for membership it generates values until either it finds the
3183 requested one or has gone through all the elements in the generator
3181 requested one or has gone through all the elements in the generator
3184 """
3182 """
3185 def __init__(self, gen, iterasc=None):
3183 def __init__(self, gen, iterasc=None):
3186 """
3184 """
3187 gen: a generator producing the values for the generatorset.
3185 gen: a generator producing the values for the generatorset.
3188 """
3186 """
3189 self._gen = gen
3187 self._gen = gen
3190 self._asclist = None
3188 self._asclist = None
3191 self._cache = {}
3189 self._cache = {}
3192 self._genlist = []
3190 self._genlist = []
3193 self._finished = False
3191 self._finished = False
3194 self._ascending = True
3192 self._ascending = True
3195 if iterasc is not None:
3193 if iterasc is not None:
3196 if iterasc:
3194 if iterasc:
3197 self.fastasc = self._iterator
3195 self.fastasc = self._iterator
3198 self.__contains__ = self._asccontains
3196 self.__contains__ = self._asccontains
3199 else:
3197 else:
3200 self.fastdesc = self._iterator
3198 self.fastdesc = self._iterator
3201 self.__contains__ = self._desccontains
3199 self.__contains__ = self._desccontains
3202
3200
3203 def __nonzero__(self):
3201 def __nonzero__(self):
3204 # Do not use 'for r in self' because it will enforce the iteration
3202 # Do not use 'for r in self' because it will enforce the iteration
3205 # order (default ascending), possibly unrolling a whole descending
3203 # order (default ascending), possibly unrolling a whole descending
3206 # iterator.
3204 # iterator.
3207 if self._genlist:
3205 if self._genlist:
3208 return True
3206 return True
3209 for r in self._consumegen():
3207 for r in self._consumegen():
3210 return True
3208 return True
3211 return False
3209 return False
3212
3210
3213 def __contains__(self, x):
3211 def __contains__(self, x):
3214 if x in self._cache:
3212 if x in self._cache:
3215 return self._cache[x]
3213 return self._cache[x]
3216
3214
3217 # Use new values only, as existing values would be cached.
3215 # Use new values only, as existing values would be cached.
3218 for l in self._consumegen():
3216 for l in self._consumegen():
3219 if l == x:
3217 if l == x:
3220 return True
3218 return True
3221
3219
3222 self._cache[x] = False
3220 self._cache[x] = False
3223 return False
3221 return False
3224
3222
3225 def _asccontains(self, x):
3223 def _asccontains(self, x):
3226 """version of contains optimised for ascending generator"""
3224 """version of contains optimised for ascending generator"""
3227 if x in self._cache:
3225 if x in self._cache:
3228 return self._cache[x]
3226 return self._cache[x]
3229
3227
3230 # Use new values only, as existing values would be cached.
3228 # Use new values only, as existing values would be cached.
3231 for l in self._consumegen():
3229 for l in self._consumegen():
3232 if l == x:
3230 if l == x:
3233 return True
3231 return True
3234 if l > x:
3232 if l > x:
3235 break
3233 break
3236
3234
3237 self._cache[x] = False
3235 self._cache[x] = False
3238 return False
3236 return False
3239
3237
3240 def _desccontains(self, x):
3238 def _desccontains(self, x):
3241 """version of contains optimised for descending generator"""
3239 """version of contains optimised for descending generator"""
3242 if x in self._cache:
3240 if x in self._cache:
3243 return self._cache[x]
3241 return self._cache[x]
3244
3242
3245 # Use new values only, as existing values would be cached.
3243 # Use new values only, as existing values would be cached.
3246 for l in self._consumegen():
3244 for l in self._consumegen():
3247 if l == x:
3245 if l == x:
3248 return True
3246 return True
3249 if l < x:
3247 if l < x:
3250 break
3248 break
3251
3249
3252 self._cache[x] = False
3250 self._cache[x] = False
3253 return False
3251 return False
3254
3252
3255 def __iter__(self):
3253 def __iter__(self):
3256 if self._ascending:
3254 if self._ascending:
3257 it = self.fastasc
3255 it = self.fastasc
3258 else:
3256 else:
3259 it = self.fastdesc
3257 it = self.fastdesc
3260 if it is not None:
3258 if it is not None:
3261 return it()
3259 return it()
3262 # we need to consume the iterator
3260 # we need to consume the iterator
3263 for x in self._consumegen():
3261 for x in self._consumegen():
3264 pass
3262 pass
3265 # recall the same code
3263 # recall the same code
3266 return iter(self)
3264 return iter(self)
3267
3265
3268 def _iterator(self):
3266 def _iterator(self):
3269 if self._finished:
3267 if self._finished:
3270 return iter(self._genlist)
3268 return iter(self._genlist)
3271
3269
3272 # We have to use this complex iteration strategy to allow multiple
3270 # We have to use this complex iteration strategy to allow multiple
3273 # iterations at the same time. We need to be able to catch revision
3271 # iterations at the same time. We need to be able to catch revision
3274 # removed from _consumegen and added to genlist in another instance.
3272 # removed from _consumegen and added to genlist in another instance.
3275 #
3273 #
3276 # Getting rid of it would provide an about 15% speed up on this
3274 # Getting rid of it would provide an about 15% speed up on this
3277 # iteration.
3275 # iteration.
3278 genlist = self._genlist
3276 genlist = self._genlist
3279 nextrev = self._consumegen().next
3277 nextrev = self._consumegen().next
3280 _len = len # cache global lookup
3278 _len = len # cache global lookup
3281 def gen():
3279 def gen():
3282 i = 0
3280 i = 0
3283 while True:
3281 while True:
3284 if i < _len(genlist):
3282 if i < _len(genlist):
3285 yield genlist[i]
3283 yield genlist[i]
3286 else:
3284 else:
3287 yield nextrev()
3285 yield nextrev()
3288 i += 1
3286 i += 1
3289 return gen()
3287 return gen()
3290
3288
3291 def _consumegen(self):
3289 def _consumegen(self):
3292 cache = self._cache
3290 cache = self._cache
3293 genlist = self._genlist.append
3291 genlist = self._genlist.append
3294 for item in self._gen:
3292 for item in self._gen:
3295 cache[item] = True
3293 cache[item] = True
3296 genlist(item)
3294 genlist(item)
3297 yield item
3295 yield item
3298 if not self._finished:
3296 if not self._finished:
3299 self._finished = True
3297 self._finished = True
3300 asc = self._genlist[:]
3298 asc = self._genlist[:]
3301 asc.sort()
3299 asc.sort()
3302 self._asclist = asc
3300 self._asclist = asc
3303 self.fastasc = asc.__iter__
3301 self.fastasc = asc.__iter__
3304 self.fastdesc = asc.__reversed__
3302 self.fastdesc = asc.__reversed__
3305
3303
3306 def __len__(self):
3304 def __len__(self):
3307 for x in self._consumegen():
3305 for x in self._consumegen():
3308 pass
3306 pass
3309 return len(self._genlist)
3307 return len(self._genlist)
3310
3308
3311 def sort(self, reverse=False):
3309 def sort(self, reverse=False):
3312 self._ascending = not reverse
3310 self._ascending = not reverse
3313
3311
3314 def reverse(self):
3312 def reverse(self):
3315 self._ascending = not self._ascending
3313 self._ascending = not self._ascending
3316
3314
3317 def isascending(self):
3315 def isascending(self):
3318 return self._ascending
3316 return self._ascending
3319
3317
3320 def isdescending(self):
3318 def isdescending(self):
3321 return not self._ascending
3319 return not self._ascending
3322
3320
3323 def first(self):
3321 def first(self):
3324 if self._ascending:
3322 if self._ascending:
3325 it = self.fastasc
3323 it = self.fastasc
3326 else:
3324 else:
3327 it = self.fastdesc
3325 it = self.fastdesc
3328 if it is None:
3326 if it is None:
3329 # we need to consume all and try again
3327 # we need to consume all and try again
3330 for x in self._consumegen():
3328 for x in self._consumegen():
3331 pass
3329 pass
3332 return self.first()
3330 return self.first()
3333 if self:
3331 if self:
3334 return it().next()
3332 return it().next()
3335 return None
3333 return None
3336
3334
3337 def last(self):
3335 def last(self):
3338 if self._ascending:
3336 if self._ascending:
3339 it = self.fastdesc
3337 it = self.fastdesc
3340 else:
3338 else:
3341 it = self.fastasc
3339 it = self.fastasc
3342 if it is None:
3340 if it is None:
3343 # we need to consume all and try again
3341 # we need to consume all and try again
3344 for x in self._consumegen():
3342 for x in self._consumegen():
3345 pass
3343 pass
3346 return self.first()
3344 return self.first()
3347 if self:
3345 if self:
3348 return it().next()
3346 return it().next()
3349 return None
3347 return None
3350
3348
3351 def __repr__(self):
3349 def __repr__(self):
3352 d = {False: '-', True: '+'}[self._ascending]
3350 d = {False: '-', True: '+'}[self._ascending]
3353 return '<%s%s>' % (type(self).__name__, d)
3351 return '<%s%s>' % (type(self).__name__, d)
3354
3352
3355 class spanset(abstractsmartset):
3353 class spanset(abstractsmartset):
3356 """Duck type for baseset class which represents a range of revisions and
3354 """Duck type for baseset class which represents a range of revisions and
3357 can work lazily and without having all the range in memory
3355 can work lazily and without having all the range in memory
3358
3356
3359 Note that spanset(x, y) behave almost like xrange(x, y) except for two
3357 Note that spanset(x, y) behave almost like xrange(x, y) except for two
3360 notable points:
3358 notable points:
3361 - when x < y it will be automatically descending,
3359 - when x < y it will be automatically descending,
3362 - revision filtered with this repoview will be skipped.
3360 - revision filtered with this repoview will be skipped.
3363
3361
3364 """
3362 """
3365 def __init__(self, repo, start=0, end=None):
3363 def __init__(self, repo, start=0, end=None):
3366 """
3364 """
3367 start: first revision included the set
3365 start: first revision included the set
3368 (default to 0)
3366 (default to 0)
3369 end: first revision excluded (last+1)
3367 end: first revision excluded (last+1)
3370 (default to len(repo)
3368 (default to len(repo)
3371
3369
3372 Spanset will be descending if `end` < `start`.
3370 Spanset will be descending if `end` < `start`.
3373 """
3371 """
3374 if end is None:
3372 if end is None:
3375 end = len(repo)
3373 end = len(repo)
3376 self._ascending = start <= end
3374 self._ascending = start <= end
3377 if not self._ascending:
3375 if not self._ascending:
3378 start, end = end + 1, start +1
3376 start, end = end + 1, start +1
3379 self._start = start
3377 self._start = start
3380 self._end = end
3378 self._end = end
3381 self._hiddenrevs = repo.changelog.filteredrevs
3379 self._hiddenrevs = repo.changelog.filteredrevs
3382
3380
3383 def sort(self, reverse=False):
3381 def sort(self, reverse=False):
3384 self._ascending = not reverse
3382 self._ascending = not reverse
3385
3383
3386 def reverse(self):
3384 def reverse(self):
3387 self._ascending = not self._ascending
3385 self._ascending = not self._ascending
3388
3386
3389 def _iterfilter(self, iterrange):
3387 def _iterfilter(self, iterrange):
3390 s = self._hiddenrevs
3388 s = self._hiddenrevs
3391 for r in iterrange:
3389 for r in iterrange:
3392 if r not in s:
3390 if r not in s:
3393 yield r
3391 yield r
3394
3392
3395 def __iter__(self):
3393 def __iter__(self):
3396 if self._ascending:
3394 if self._ascending:
3397 return self.fastasc()
3395 return self.fastasc()
3398 else:
3396 else:
3399 return self.fastdesc()
3397 return self.fastdesc()
3400
3398
3401 def fastasc(self):
3399 def fastasc(self):
3402 iterrange = xrange(self._start, self._end)
3400 iterrange = xrange(self._start, self._end)
3403 if self._hiddenrevs:
3401 if self._hiddenrevs:
3404 return self._iterfilter(iterrange)
3402 return self._iterfilter(iterrange)
3405 return iter(iterrange)
3403 return iter(iterrange)
3406
3404
3407 def fastdesc(self):
3405 def fastdesc(self):
3408 iterrange = xrange(self._end - 1, self._start - 1, -1)
3406 iterrange = xrange(self._end - 1, self._start - 1, -1)
3409 if self._hiddenrevs:
3407 if self._hiddenrevs:
3410 return self._iterfilter(iterrange)
3408 return self._iterfilter(iterrange)
3411 return iter(iterrange)
3409 return iter(iterrange)
3412
3410
3413 def __contains__(self, rev):
3411 def __contains__(self, rev):
3414 hidden = self._hiddenrevs
3412 hidden = self._hiddenrevs
3415 return ((self._start <= rev < self._end)
3413 return ((self._start <= rev < self._end)
3416 and not (hidden and rev in hidden))
3414 and not (hidden and rev in hidden))
3417
3415
3418 def __nonzero__(self):
3416 def __nonzero__(self):
3419 for r in self:
3417 for r in self:
3420 return True
3418 return True
3421 return False
3419 return False
3422
3420
3423 def __len__(self):
3421 def __len__(self):
3424 if not self._hiddenrevs:
3422 if not self._hiddenrevs:
3425 return abs(self._end - self._start)
3423 return abs(self._end - self._start)
3426 else:
3424 else:
3427 count = 0
3425 count = 0
3428 start = self._start
3426 start = self._start
3429 end = self._end
3427 end = self._end
3430 for rev in self._hiddenrevs:
3428 for rev in self._hiddenrevs:
3431 if (end < rev <= start) or (start <= rev < end):
3429 if (end < rev <= start) or (start <= rev < end):
3432 count += 1
3430 count += 1
3433 return abs(self._end - self._start) - count
3431 return abs(self._end - self._start) - count
3434
3432
3435 def isascending(self):
3433 def isascending(self):
3436 return self._ascending
3434 return self._ascending
3437
3435
3438 def isdescending(self):
3436 def isdescending(self):
3439 return not self._ascending
3437 return not self._ascending
3440
3438
3441 def first(self):
3439 def first(self):
3442 if self._ascending:
3440 if self._ascending:
3443 it = self.fastasc
3441 it = self.fastasc
3444 else:
3442 else:
3445 it = self.fastdesc
3443 it = self.fastdesc
3446 for x in it():
3444 for x in it():
3447 return x
3445 return x
3448 return None
3446 return None
3449
3447
3450 def last(self):
3448 def last(self):
3451 if self._ascending:
3449 if self._ascending:
3452 it = self.fastdesc
3450 it = self.fastdesc
3453 else:
3451 else:
3454 it = self.fastasc
3452 it = self.fastasc
3455 for x in it():
3453 for x in it():
3456 return x
3454 return x
3457 return None
3455 return None
3458
3456
3459 def __repr__(self):
3457 def __repr__(self):
3460 d = {False: '-', True: '+'}[self._ascending]
3458 d = {False: '-', True: '+'}[self._ascending]
3461 return '<%s%s %d:%d>' % (type(self).__name__, d,
3459 return '<%s%s %d:%d>' % (type(self).__name__, d,
3462 self._start, self._end - 1)
3460 self._start, self._end - 1)
3463
3461
3464 class fullreposet(spanset):
3462 class fullreposet(spanset):
3465 """a set containing all revisions in the repo
3463 """a set containing all revisions in the repo
3466
3464
3467 This class exists to host special optimization and magic to handle virtual
3465 This class exists to host special optimization and magic to handle virtual
3468 revisions such as "null".
3466 revisions such as "null".
3469 """
3467 """
3470
3468
3471 def __init__(self, repo):
3469 def __init__(self, repo):
3472 super(fullreposet, self).__init__(repo)
3470 super(fullreposet, self).__init__(repo)
3473
3471
3474 def __contains__(self, rev):
3472 def __contains__(self, rev):
3475 # assumes the given rev is valid
3473 # assumes the given rev is valid
3476 hidden = self._hiddenrevs
3474 hidden = self._hiddenrevs
3477 return not (hidden and rev in hidden)
3475 return not (hidden and rev in hidden)
3478
3476
3479 def __and__(self, other):
3477 def __and__(self, other):
3480 """As self contains the whole repo, all of the other set should also be
3478 """As self contains the whole repo, all of the other set should also be
3481 in self. Therefore `self & other = other`.
3479 in self. Therefore `self & other = other`.
3482
3480
3483 This boldly assumes the other contains valid revs only.
3481 This boldly assumes the other contains valid revs only.
3484 """
3482 """
3485 # other not a smartset, make is so
3483 # other not a smartset, make is so
3486 if not util.safehasattr(other, 'isascending'):
3484 if not util.safehasattr(other, 'isascending'):
3487 # filter out hidden revision
3485 # filter out hidden revision
3488 # (this boldly assumes all smartset are pure)
3486 # (this boldly assumes all smartset are pure)
3489 #
3487 #
3490 # `other` was used with "&", let's assume this is a set like
3488 # `other` was used with "&", let's assume this is a set like
3491 # object.
3489 # object.
3492 other = baseset(other - self._hiddenrevs)
3490 other = baseset(other - self._hiddenrevs)
3493
3491
3494 other.sort(reverse=self.isdescending())
3492 other.sort(reverse=self.isdescending())
3495 return other
3493 return other
3496
3494
3497 def prettyformatset(revs):
3495 def prettyformatset(revs):
3498 lines = []
3496 lines = []
3499 rs = repr(revs)
3497 rs = repr(revs)
3500 p = 0
3498 p = 0
3501 while p < len(rs):
3499 while p < len(rs):
3502 q = rs.find('<', p + 1)
3500 q = rs.find('<', p + 1)
3503 if q < 0:
3501 if q < 0:
3504 q = len(rs)
3502 q = len(rs)
3505 l = rs.count('<', 0, p) - rs.count('>', 0, p)
3503 l = rs.count('<', 0, p) - rs.count('>', 0, p)
3506 assert l >= 0
3504 assert l >= 0
3507 lines.append((l, rs[p:q].rstrip()))
3505 lines.append((l, rs[p:q].rstrip()))
3508 p = q
3506 p = q
3509 return '\n'.join(' ' * l + s for l, s in lines)
3507 return '\n'.join(' ' * l + s for l, s in lines)
3510
3508
3511 # tell hggettext to extract docstrings from these functions:
3509 # tell hggettext to extract docstrings from these functions:
3512 i18nfunctions = symbols.values()
3510 i18nfunctions = symbols.values()
General Comments 0
You need to be logged in to leave comments. Login now