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