##// END OF EJS Templates
revset: added basic operators to orderedlazyset...
Lucas Moscovicz -
r20612:60c308b9 default
parent child Browse files
Show More
@@ -1,2376 +1,2384
1 1 # revset.py - revision set queries for mercurial
2 2 #
3 3 # Copyright 2010 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7
8 8 import re
9 9 import parser, util, error, discovery, hbisect, phases
10 10 import node
11 11 import match as matchmod
12 12 from i18n import _
13 13 import encoding
14 14 import obsolete as obsmod
15 15 import pathutil
16 16 import repoview
17 17
18 18 def _revancestors(repo, revs, followfirst):
19 19 """Like revlog.ancestors(), but supports followfirst."""
20 20 cut = followfirst and 1 or None
21 21 cl = repo.changelog
22 22 visit = util.deque(revs)
23 23 seen = set([node.nullrev])
24 24 while visit:
25 25 for parent in cl.parentrevs(visit.popleft())[:cut]:
26 26 if parent not in seen:
27 27 visit.append(parent)
28 28 seen.add(parent)
29 29 yield parent
30 30
31 31 def _revdescendants(repo, revs, followfirst):
32 32 """Like revlog.descendants() but supports followfirst."""
33 33 cut = followfirst and 1 or None
34 34 cl = repo.changelog
35 35 first = min(revs)
36 36 nullrev = node.nullrev
37 37 if first == nullrev:
38 38 # Are there nodes with a null first parent and a non-null
39 39 # second one? Maybe. Do we care? Probably not.
40 40 for i in cl:
41 41 yield i
42 42 return
43 43
44 44 seen = set(revs)
45 45 for i in cl.revs(first + 1):
46 46 for x in cl.parentrevs(i)[:cut]:
47 47 if x != nullrev and x in seen:
48 48 seen.add(i)
49 49 yield i
50 50 break
51 51
52 52 def _revsbetween(repo, roots, heads):
53 53 """Return all paths between roots and heads, inclusive of both endpoint
54 54 sets."""
55 55 if not roots:
56 56 return baseset([])
57 57 parentrevs = repo.changelog.parentrevs
58 58 visit = baseset(heads)
59 59 reachable = set()
60 60 seen = {}
61 61 minroot = min(roots)
62 62 roots = set(roots)
63 63 # open-code the post-order traversal due to the tiny size of
64 64 # sys.getrecursionlimit()
65 65 while visit:
66 66 rev = visit.pop()
67 67 if rev in roots:
68 68 reachable.add(rev)
69 69 parents = parentrevs(rev)
70 70 seen[rev] = parents
71 71 for parent in parents:
72 72 if parent >= minroot and parent not in seen:
73 73 visit.append(parent)
74 74 if not reachable:
75 75 return baseset([])
76 76 for rev in sorted(seen):
77 77 for parent in seen[rev]:
78 78 if parent in reachable:
79 79 reachable.add(rev)
80 80 return baseset(sorted(reachable))
81 81
82 82 elements = {
83 83 "(": (20, ("group", 1, ")"), ("func", 1, ")")),
84 84 "~": (18, None, ("ancestor", 18)),
85 85 "^": (18, None, ("parent", 18), ("parentpost", 18)),
86 86 "-": (5, ("negate", 19), ("minus", 5)),
87 87 "::": (17, ("dagrangepre", 17), ("dagrange", 17),
88 88 ("dagrangepost", 17)),
89 89 "..": (17, ("dagrangepre", 17), ("dagrange", 17),
90 90 ("dagrangepost", 17)),
91 91 ":": (15, ("rangepre", 15), ("range", 15), ("rangepost", 15)),
92 92 "not": (10, ("not", 10)),
93 93 "!": (10, ("not", 10)),
94 94 "and": (5, None, ("and", 5)),
95 95 "&": (5, None, ("and", 5)),
96 96 "or": (4, None, ("or", 4)),
97 97 "|": (4, None, ("or", 4)),
98 98 "+": (4, None, ("or", 4)),
99 99 ",": (2, None, ("list", 2)),
100 100 ")": (0, None, None),
101 101 "symbol": (0, ("symbol",), None),
102 102 "string": (0, ("string",), None),
103 103 "end": (0, None, None),
104 104 }
105 105
106 106 keywords = set(['and', 'or', 'not'])
107 107
108 108 def tokenize(program):
109 109 '''
110 110 Parse a revset statement into a stream of tokens
111 111
112 112 Check that @ is a valid unquoted token character (issue3686):
113 113 >>> list(tokenize("@::"))
114 114 [('symbol', '@', 0), ('::', None, 1), ('end', None, 3)]
115 115
116 116 '''
117 117
118 118 pos, l = 0, len(program)
119 119 while pos < l:
120 120 c = program[pos]
121 121 if c.isspace(): # skip inter-token whitespace
122 122 pass
123 123 elif c == ':' and program[pos:pos + 2] == '::': # look ahead carefully
124 124 yield ('::', None, pos)
125 125 pos += 1 # skip ahead
126 126 elif c == '.' and program[pos:pos + 2] == '..': # look ahead carefully
127 127 yield ('..', None, pos)
128 128 pos += 1 # skip ahead
129 129 elif c in "():,-|&+!~^": # handle simple operators
130 130 yield (c, None, pos)
131 131 elif (c in '"\'' or c == 'r' and
132 132 program[pos:pos + 2] in ("r'", 'r"')): # handle quoted strings
133 133 if c == 'r':
134 134 pos += 1
135 135 c = program[pos]
136 136 decode = lambda x: x
137 137 else:
138 138 decode = lambda x: x.decode('string-escape')
139 139 pos += 1
140 140 s = pos
141 141 while pos < l: # find closing quote
142 142 d = program[pos]
143 143 if d == '\\': # skip over escaped characters
144 144 pos += 2
145 145 continue
146 146 if d == c:
147 147 yield ('string', decode(program[s:pos]), s)
148 148 break
149 149 pos += 1
150 150 else:
151 151 raise error.ParseError(_("unterminated string"), s)
152 152 # gather up a symbol/keyword
153 153 elif c.isalnum() or c in '._@' or ord(c) > 127:
154 154 s = pos
155 155 pos += 1
156 156 while pos < l: # find end of symbol
157 157 d = program[pos]
158 158 if not (d.isalnum() or d in "._/@" or ord(d) > 127):
159 159 break
160 160 if d == '.' and program[pos - 1] == '.': # special case for ..
161 161 pos -= 1
162 162 break
163 163 pos += 1
164 164 sym = program[s:pos]
165 165 if sym in keywords: # operator keywords
166 166 yield (sym, None, s)
167 167 else:
168 168 yield ('symbol', sym, s)
169 169 pos -= 1
170 170 else:
171 171 raise error.ParseError(_("syntax error"), pos)
172 172 pos += 1
173 173 yield ('end', None, pos)
174 174
175 175 # helpers
176 176
177 177 def getstring(x, err):
178 178 if x and (x[0] == 'string' or x[0] == 'symbol'):
179 179 return x[1]
180 180 raise error.ParseError(err)
181 181
182 182 def getlist(x):
183 183 if not x:
184 184 return []
185 185 if x[0] == 'list':
186 186 return getlist(x[1]) + [x[2]]
187 187 return [x]
188 188
189 189 def getargs(x, min, max, err):
190 190 l = getlist(x)
191 191 if len(l) < min or (max >= 0 and len(l) > max):
192 192 raise error.ParseError(err)
193 193 return l
194 194
195 195 def getset(repo, subset, x):
196 196 if not x:
197 197 raise error.ParseError(_("missing argument"))
198 198 s = methods[x[0]](repo, subset, *x[1:])
199 199 if util.safehasattr(s, 'set'):
200 200 return s
201 201 return baseset(s)
202 202
203 203 def _getrevsource(repo, r):
204 204 extra = repo[r].extra()
205 205 for label in ('source', 'transplant_source', 'rebase_source'):
206 206 if label in extra:
207 207 try:
208 208 return repo[extra[label]].rev()
209 209 except error.RepoLookupError:
210 210 pass
211 211 return None
212 212
213 213 # operator methods
214 214
215 215 def stringset(repo, subset, x):
216 216 x = repo[x].rev()
217 217 if x == -1 and len(subset) == len(repo):
218 218 return baseset([-1])
219 219 if len(subset) == len(repo) or x in subset:
220 220 return baseset([x])
221 221 return baseset([])
222 222
223 223 def symbolset(repo, subset, x):
224 224 if x in symbols:
225 225 raise error.ParseError(_("can't use %s here") % x)
226 226 return stringset(repo, subset, x)
227 227
228 228 def rangeset(repo, subset, x, y):
229 229 cl = baseset(repo.changelog)
230 230 m = getset(repo, cl, x)
231 231 n = getset(repo, cl, y)
232 232
233 233 if not m or not n:
234 234 return baseset([])
235 235 m, n = m[0], n[-1]
236 236
237 237 if m < n:
238 238 r = spanset(repo, m, n + 1)
239 239 else:
240 240 r = spanset(repo, m, n - 1)
241 241 return r & subset
242 242
243 243 def dagrange(repo, subset, x, y):
244 244 r = spanset(repo)
245 245 xs = _revsbetween(repo, getset(repo, r, x), getset(repo, r, y))
246 246 s = subset.set()
247 247 return xs.filter(lambda r: r in s)
248 248
249 249 def andset(repo, subset, x, y):
250 250 return getset(repo, getset(repo, subset, x), y)
251 251
252 252 def orset(repo, subset, x, y):
253 253 xl = getset(repo, subset, x)
254 254 yl = getset(repo, subset - xl, y)
255 255 return xl + yl
256 256
257 257 def notset(repo, subset, x):
258 258 return subset - getset(repo, subset, x)
259 259
260 260 def listset(repo, subset, a, b):
261 261 raise error.ParseError(_("can't use a list in this context"))
262 262
263 263 def func(repo, subset, a, b):
264 264 if a[0] == 'symbol' and a[1] in symbols:
265 265 return symbols[a[1]](repo, subset, b)
266 266 raise error.ParseError(_("not a function: %s") % a[1])
267 267
268 268 # functions
269 269
270 270 def adds(repo, subset, x):
271 271 """``adds(pattern)``
272 272 Changesets that add a file matching pattern.
273 273
274 274 The pattern without explicit kind like ``glob:`` is expected to be
275 275 relative to the current directory and match against a file or a
276 276 directory.
277 277 """
278 278 # i18n: "adds" is a keyword
279 279 pat = getstring(x, _("adds requires a pattern"))
280 280 return checkstatus(repo, subset, pat, 1)
281 281
282 282 def ancestor(repo, subset, x):
283 283 """``ancestor(*changeset)``
284 284 Greatest common ancestor of the changesets.
285 285
286 286 Accepts 0 or more changesets.
287 287 Will return empty list when passed no args.
288 288 Greatest common ancestor of a single changeset is that changeset.
289 289 """
290 290 # i18n: "ancestor" is a keyword
291 291 l = getlist(x)
292 292 rl = spanset(repo)
293 293 anc = None
294 294
295 295 # (getset(repo, rl, i) for i in l) generates a list of lists
296 296 rev = repo.changelog.rev
297 297 ancestor = repo.changelog.ancestor
298 298 node = repo.changelog.node
299 299 for revs in (getset(repo, rl, i) for i in l):
300 300 for r in revs:
301 301 if anc is None:
302 302 anc = r
303 303 else:
304 304 anc = rev(ancestor(node(anc), node(r)))
305 305
306 306 if anc is not None and anc in subset:
307 307 return baseset([anc])
308 308 return baseset([])
309 309
310 310 def _ancestors(repo, subset, x, followfirst=False):
311 311 args = getset(repo, spanset(repo), x)
312 312 if not args:
313 313 return baseset([])
314 314 s = set(_revancestors(repo, args, followfirst)) | set(args)
315 315 return subset.filter(lambda r: r in s)
316 316
317 317 def ancestors(repo, subset, x):
318 318 """``ancestors(set)``
319 319 Changesets that are ancestors of a changeset in set.
320 320 """
321 321 return _ancestors(repo, subset, x)
322 322
323 323 def _firstancestors(repo, subset, x):
324 324 # ``_firstancestors(set)``
325 325 # Like ``ancestors(set)`` but follows only the first parents.
326 326 return _ancestors(repo, subset, x, followfirst=True)
327 327
328 328 def ancestorspec(repo, subset, x, n):
329 329 """``set~n``
330 330 Changesets that are the Nth ancestor (first parents only) of a changeset
331 331 in set.
332 332 """
333 333 try:
334 334 n = int(n[1])
335 335 except (TypeError, ValueError):
336 336 raise error.ParseError(_("~ expects a number"))
337 337 ps = set()
338 338 cl = repo.changelog
339 339 for r in getset(repo, baseset(cl), x):
340 340 for i in range(n):
341 341 r = cl.parentrevs(r)[0]
342 342 ps.add(r)
343 343 return subset.filter(lambda r: r in ps)
344 344
345 345 def author(repo, subset, x):
346 346 """``author(string)``
347 347 Alias for ``user(string)``.
348 348 """
349 349 # i18n: "author" is a keyword
350 350 n = encoding.lower(getstring(x, _("author requires a string")))
351 351 kind, pattern, matcher = _substringmatcher(n)
352 352 return subset.filter(lambda x: matcher(encoding.lower(repo[x].user())))
353 353
354 354 def bisect(repo, subset, x):
355 355 """``bisect(string)``
356 356 Changesets marked in the specified bisect status:
357 357
358 358 - ``good``, ``bad``, ``skip``: csets explicitly marked as good/bad/skip
359 359 - ``goods``, ``bads`` : csets topologically good/bad
360 360 - ``range`` : csets taking part in the bisection
361 361 - ``pruned`` : csets that are goods, bads or skipped
362 362 - ``untested`` : csets whose fate is yet unknown
363 363 - ``ignored`` : csets ignored due to DAG topology
364 364 - ``current`` : the cset currently being bisected
365 365 """
366 366 # i18n: "bisect" is a keyword
367 367 status = getstring(x, _("bisect requires a string")).lower()
368 368 state = set(hbisect.get(repo, status))
369 369 return subset.filter(lambda r: r in state)
370 370
371 371 # Backward-compatibility
372 372 # - no help entry so that we do not advertise it any more
373 373 def bisected(repo, subset, x):
374 374 return bisect(repo, subset, x)
375 375
376 376 def bookmark(repo, subset, x):
377 377 """``bookmark([name])``
378 378 The named bookmark or all bookmarks.
379 379
380 380 If `name` starts with `re:`, the remainder of the name is treated as
381 381 a regular expression. To match a bookmark that actually starts with `re:`,
382 382 use the prefix `literal:`.
383 383 """
384 384 # i18n: "bookmark" is a keyword
385 385 args = getargs(x, 0, 1, _('bookmark takes one or no arguments'))
386 386 if args:
387 387 bm = getstring(args[0],
388 388 # i18n: "bookmark" is a keyword
389 389 _('the argument to bookmark must be a string'))
390 390 kind, pattern, matcher = _stringmatcher(bm)
391 391 if kind == 'literal':
392 392 bmrev = repo._bookmarks.get(bm, None)
393 393 if not bmrev:
394 394 raise util.Abort(_("bookmark '%s' does not exist") % bm)
395 395 bmrev = repo[bmrev].rev()
396 396 return subset.filter(lambda r: r == bmrev)
397 397 else:
398 398 matchrevs = set()
399 399 for name, bmrev in repo._bookmarks.iteritems():
400 400 if matcher(name):
401 401 matchrevs.add(bmrev)
402 402 if not matchrevs:
403 403 raise util.Abort(_("no bookmarks exist that match '%s'")
404 404 % pattern)
405 405 bmrevs = set()
406 406 for bmrev in matchrevs:
407 407 bmrevs.add(repo[bmrev].rev())
408 408 return subset & bmrevs
409 409
410 410 bms = set([repo[r].rev()
411 411 for r in repo._bookmarks.values()])
412 412 return subset.filter(lambda r: r in bms)
413 413
414 414 def branch(repo, subset, x):
415 415 """``branch(string or set)``
416 416 All changesets belonging to the given branch or the branches of the given
417 417 changesets.
418 418
419 419 If `string` starts with `re:`, the remainder of the name is treated as
420 420 a regular expression. To match a branch that actually starts with `re:`,
421 421 use the prefix `literal:`.
422 422 """
423 423 try:
424 424 b = getstring(x, '')
425 425 except error.ParseError:
426 426 # not a string, but another revspec, e.g. tip()
427 427 pass
428 428 else:
429 429 kind, pattern, matcher = _stringmatcher(b)
430 430 if kind == 'literal':
431 431 # note: falls through to the revspec case if no branch with
432 432 # this name exists
433 433 if pattern in repo.branchmap():
434 434 return subset.filter(lambda r: matcher(repo[r].branch()))
435 435 else:
436 436 return subset.filter(lambda r: matcher(repo[r].branch()))
437 437
438 438 s = getset(repo, spanset(repo), x)
439 439 b = set()
440 440 for r in s:
441 441 b.add(repo[r].branch())
442 442 s = s.set()
443 443 return subset.filter(lambda r: r in s or repo[r].branch() in b)
444 444
445 445 def bumped(repo, subset, x):
446 446 """``bumped()``
447 447 Mutable changesets marked as successors of public changesets.
448 448
449 449 Only non-public and non-obsolete changesets can be `bumped`.
450 450 """
451 451 # i18n: "bumped" is a keyword
452 452 getargs(x, 0, 0, _("bumped takes no arguments"))
453 453 bumped = obsmod.getrevs(repo, 'bumped')
454 454 return subset & bumped
455 455
456 456 def bundle(repo, subset, x):
457 457 """``bundle()``
458 458 Changesets in the bundle.
459 459
460 460 Bundle must be specified by the -R option."""
461 461
462 462 try:
463 463 bundlerevs = repo.changelog.bundlerevs
464 464 except AttributeError:
465 465 raise util.Abort(_("no bundle provided - specify with -R"))
466 466 return subset & bundlerevs
467 467
468 468 def checkstatus(repo, subset, pat, field):
469 469 hasset = matchmod.patkind(pat) == 'set'
470 470
471 471 def matches(x):
472 472 m = None
473 473 fname = None
474 474 c = repo[x]
475 475 if not m or hasset:
476 476 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=c)
477 477 if not m.anypats() and len(m.files()) == 1:
478 478 fname = m.files()[0]
479 479 if fname is not None:
480 480 if fname not in c.files():
481 481 return False
482 482 else:
483 483 for f in c.files():
484 484 if m(f):
485 485 break
486 486 else:
487 487 return False
488 488 files = repo.status(c.p1().node(), c.node())[field]
489 489 if fname is not None:
490 490 if fname in files:
491 491 return True
492 492 else:
493 493 for f in files:
494 494 if m(f):
495 495 return True
496 496
497 497 return subset.filter(matches)
498 498
499 499 def _children(repo, narrow, parentset):
500 500 cs = set()
501 501 if not parentset:
502 502 return baseset(cs)
503 503 pr = repo.changelog.parentrevs
504 504 minrev = min(parentset)
505 505 for r in narrow:
506 506 if r <= minrev:
507 507 continue
508 508 for p in pr(r):
509 509 if p in parentset:
510 510 cs.add(r)
511 511 return baseset(cs)
512 512
513 513 def children(repo, subset, x):
514 514 """``children(set)``
515 515 Child changesets of changesets in set.
516 516 """
517 517 s = getset(repo, baseset(repo), x).set()
518 518 cs = _children(repo, subset, s)
519 519 return subset & cs
520 520
521 521 def closed(repo, subset, x):
522 522 """``closed()``
523 523 Changeset is closed.
524 524 """
525 525 # i18n: "closed" is a keyword
526 526 getargs(x, 0, 0, _("closed takes no arguments"))
527 527 return subset.filter(lambda r: repo[r].closesbranch())
528 528
529 529 def contains(repo, subset, x):
530 530 """``contains(pattern)``
531 531 Revision contains a file matching pattern. See :hg:`help patterns`
532 532 for information about file patterns.
533 533
534 534 The pattern without explicit kind like ``glob:`` is expected to be
535 535 relative to the current directory and match against a file exactly
536 536 for efficiency.
537 537 """
538 538 # i18n: "contains" is a keyword
539 539 pat = getstring(x, _("contains requires a pattern"))
540 540
541 541 def matches(x):
542 542 if not matchmod.patkind(pat):
543 543 pats = pathutil.canonpath(repo.root, repo.getcwd(), pat)
544 544 if pats in repo[x]:
545 545 return True
546 546 else:
547 547 c = repo[x]
548 548 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=c)
549 549 for f in c.manifest():
550 550 if m(f):
551 551 return True
552 552 return False
553 553
554 554 return subset.filter(matches)
555 555
556 556 def converted(repo, subset, x):
557 557 """``converted([id])``
558 558 Changesets converted from the given identifier in the old repository if
559 559 present, or all converted changesets if no identifier is specified.
560 560 """
561 561
562 562 # There is exactly no chance of resolving the revision, so do a simple
563 563 # string compare and hope for the best
564 564
565 565 rev = None
566 566 # i18n: "converted" is a keyword
567 567 l = getargs(x, 0, 1, _('converted takes one or no arguments'))
568 568 if l:
569 569 # i18n: "converted" is a keyword
570 570 rev = getstring(l[0], _('converted requires a revision'))
571 571
572 572 def _matchvalue(r):
573 573 source = repo[r].extra().get('convert_revision', None)
574 574 return source is not None and (rev is None or source.startswith(rev))
575 575
576 576 return subset.filter(lambda r: _matchvalue(r))
577 577
578 578 def date(repo, subset, x):
579 579 """``date(interval)``
580 580 Changesets within the interval, see :hg:`help dates`.
581 581 """
582 582 # i18n: "date" is a keyword
583 583 ds = getstring(x, _("date requires a string"))
584 584 dm = util.matchdate(ds)
585 585 return subset.filter(lambda x: dm(repo[x].date()[0]))
586 586
587 587 def desc(repo, subset, x):
588 588 """``desc(string)``
589 589 Search commit message for string. The match is case-insensitive.
590 590 """
591 591 # i18n: "desc" is a keyword
592 592 ds = encoding.lower(getstring(x, _("desc requires a string")))
593 593
594 594 def matches(x):
595 595 c = repo[x]
596 596 return ds in encoding.lower(c.description())
597 597
598 598 return subset.filter(matches)
599 599
600 600 def _descendants(repo, subset, x, followfirst=False):
601 601 args = getset(repo, spanset(repo), x)
602 602 if not args:
603 603 return baseset([])
604 604 s = set(_revdescendants(repo, args, followfirst)) | set(args)
605 605 return subset & s
606 606
607 607 def descendants(repo, subset, x):
608 608 """``descendants(set)``
609 609 Changesets which are descendants of changesets in set.
610 610 """
611 611 return _descendants(repo, subset, x)
612 612
613 613 def _firstdescendants(repo, subset, x):
614 614 # ``_firstdescendants(set)``
615 615 # Like ``descendants(set)`` but follows only the first parents.
616 616 return _descendants(repo, subset, x, followfirst=True)
617 617
618 618 def destination(repo, subset, x):
619 619 """``destination([set])``
620 620 Changesets that were created by a graft, transplant or rebase operation,
621 621 with the given revisions specified as the source. Omitting the optional set
622 622 is the same as passing all().
623 623 """
624 624 if x is not None:
625 625 args = getset(repo, spanset(repo), x).set()
626 626 else:
627 627 args = getall(repo, spanset(repo), x).set()
628 628
629 629 dests = set()
630 630
631 631 # subset contains all of the possible destinations that can be returned, so
632 632 # iterate over them and see if their source(s) were provided in the args.
633 633 # Even if the immediate src of r is not in the args, src's source (or
634 634 # further back) may be. Scanning back further than the immediate src allows
635 635 # transitive transplants and rebases to yield the same results as transitive
636 636 # grafts.
637 637 for r in subset:
638 638 src = _getrevsource(repo, r)
639 639 lineage = None
640 640
641 641 while src is not None:
642 642 if lineage is None:
643 643 lineage = list()
644 644
645 645 lineage.append(r)
646 646
647 647 # The visited lineage is a match if the current source is in the arg
648 648 # set. Since every candidate dest is visited by way of iterating
649 649 # subset, any dests further back in the lineage will be tested by a
650 650 # different iteration over subset. Likewise, if the src was already
651 651 # selected, the current lineage can be selected without going back
652 652 # further.
653 653 if src in args or src in dests:
654 654 dests.update(lineage)
655 655 break
656 656
657 657 r = src
658 658 src = _getrevsource(repo, r)
659 659
660 660 return subset.filter(lambda r: r in dests)
661 661
662 662 def divergent(repo, subset, x):
663 663 """``divergent()``
664 664 Final successors of changesets with an alternative set of final successors.
665 665 """
666 666 # i18n: "divergent" is a keyword
667 667 getargs(x, 0, 0, _("divergent takes no arguments"))
668 668 divergent = obsmod.getrevs(repo, 'divergent')
669 669 return subset.filter(lambda r: r in divergent)
670 670
671 671 def draft(repo, subset, x):
672 672 """``draft()``
673 673 Changeset in draft phase."""
674 674 # i18n: "draft" is a keyword
675 675 getargs(x, 0, 0, _("draft takes no arguments"))
676 676 pc = repo._phasecache
677 677 return subset.filter(lambda r: pc.phase(repo, r) == phases.draft)
678 678
679 679 def extinct(repo, subset, x):
680 680 """``extinct()``
681 681 Obsolete changesets with obsolete descendants only.
682 682 """
683 683 # i18n: "extinct" is a keyword
684 684 getargs(x, 0, 0, _("extinct takes no arguments"))
685 685 extincts = obsmod.getrevs(repo, 'extinct')
686 686 return subset & extincts
687 687
688 688 def extra(repo, subset, x):
689 689 """``extra(label, [value])``
690 690 Changesets with the given label in the extra metadata, with the given
691 691 optional value.
692 692
693 693 If `value` starts with `re:`, the remainder of the value is treated as
694 694 a regular expression. To match a value that actually starts with `re:`,
695 695 use the prefix `literal:`.
696 696 """
697 697
698 698 # i18n: "extra" is a keyword
699 699 l = getargs(x, 1, 2, _('extra takes at least 1 and at most 2 arguments'))
700 700 # i18n: "extra" is a keyword
701 701 label = getstring(l[0], _('first argument to extra must be a string'))
702 702 value = None
703 703
704 704 if len(l) > 1:
705 705 # i18n: "extra" is a keyword
706 706 value = getstring(l[1], _('second argument to extra must be a string'))
707 707 kind, value, matcher = _stringmatcher(value)
708 708
709 709 def _matchvalue(r):
710 710 extra = repo[r].extra()
711 711 return label in extra and (value is None or matcher(extra[label]))
712 712
713 713 return subset.filter(lambda r: _matchvalue(r))
714 714
715 715 def filelog(repo, subset, x):
716 716 """``filelog(pattern)``
717 717 Changesets connected to the specified filelog.
718 718
719 719 For performance reasons, ``filelog()`` does not show every changeset
720 720 that affects the requested file(s). See :hg:`help log` for details. For
721 721 a slower, more accurate result, use ``file()``.
722 722
723 723 The pattern without explicit kind like ``glob:`` is expected to be
724 724 relative to the current directory and match against a file exactly
725 725 for efficiency.
726 726 """
727 727
728 728 # i18n: "filelog" is a keyword
729 729 pat = getstring(x, _("filelog requires a pattern"))
730 730 s = set()
731 731
732 732 if not matchmod.patkind(pat):
733 733 f = pathutil.canonpath(repo.root, repo.getcwd(), pat)
734 734 fl = repo.file(f)
735 735 for fr in fl:
736 736 s.add(fl.linkrev(fr))
737 737 else:
738 738 m = matchmod.match(repo.root, repo.getcwd(), [pat], ctx=repo[None])
739 739 for f in repo[None]:
740 740 if m(f):
741 741 fl = repo.file(f)
742 742 for fr in fl:
743 743 s.add(fl.linkrev(fr))
744 744
745 745 return subset.filter(lambda r: r in s)
746 746
747 747 def first(repo, subset, x):
748 748 """``first(set, [n])``
749 749 An alias for limit().
750 750 """
751 751 return limit(repo, subset, x)
752 752
753 753 def _follow(repo, subset, x, name, followfirst=False):
754 754 l = getargs(x, 0, 1, _("%s takes no arguments or a filename") % name)
755 755 c = repo['.']
756 756 if l:
757 757 x = getstring(l[0], _("%s expected a filename") % name)
758 758 if x in c:
759 759 cx = c[x]
760 760 s = set(ctx.rev() for ctx in cx.ancestors(followfirst=followfirst))
761 761 # include the revision responsible for the most recent version
762 762 s.add(cx.linkrev())
763 763 else:
764 764 return baseset([])
765 765 else:
766 766 s = set(_revancestors(repo, [c.rev()], followfirst)) | set([c.rev()])
767 767
768 768 return subset.filter(lambda r: r in s)
769 769
770 770 def follow(repo, subset, x):
771 771 """``follow([file])``
772 772 An alias for ``::.`` (ancestors of the working copy's first parent).
773 773 If a filename is specified, the history of the given file is followed,
774 774 including copies.
775 775 """
776 776 return _follow(repo, subset, x, 'follow')
777 777
778 778 def _followfirst(repo, subset, x):
779 779 # ``followfirst([file])``
780 780 # Like ``follow([file])`` but follows only the first parent of
781 781 # every revision or file revision.
782 782 return _follow(repo, subset, x, '_followfirst', followfirst=True)
783 783
784 784 def getall(repo, subset, x):
785 785 """``all()``
786 786 All changesets, the same as ``0:tip``.
787 787 """
788 788 # i18n: "all" is a keyword
789 789 getargs(x, 0, 0, _("all takes no arguments"))
790 790 return subset
791 791
792 792 def grep(repo, subset, x):
793 793 """``grep(regex)``
794 794 Like ``keyword(string)`` but accepts a regex. Use ``grep(r'...')``
795 795 to ensure special escape characters are handled correctly. Unlike
796 796 ``keyword(string)``, the match is case-sensitive.
797 797 """
798 798 try:
799 799 # i18n: "grep" is a keyword
800 800 gr = re.compile(getstring(x, _("grep requires a string")))
801 801 except re.error, e:
802 802 raise error.ParseError(_('invalid match pattern: %s') % e)
803 803
804 804 def matches(x):
805 805 c = repo[x]
806 806 for e in c.files() + [c.user(), c.description()]:
807 807 if gr.search(e):
808 808 return True
809 809 return False
810 810
811 811 return subset.filter(matches)
812 812
813 813 def _matchfiles(repo, subset, x):
814 814 # _matchfiles takes a revset list of prefixed arguments:
815 815 #
816 816 # [p:foo, i:bar, x:baz]
817 817 #
818 818 # builds a match object from them and filters subset. Allowed
819 819 # prefixes are 'p:' for regular patterns, 'i:' for include
820 820 # patterns and 'x:' for exclude patterns. Use 'r:' prefix to pass
821 821 # a revision identifier, or the empty string to reference the
822 822 # working directory, from which the match object is
823 823 # initialized. Use 'd:' to set the default matching mode, default
824 824 # to 'glob'. At most one 'r:' and 'd:' argument can be passed.
825 825
826 826 # i18n: "_matchfiles" is a keyword
827 827 l = getargs(x, 1, -1, _("_matchfiles requires at least one argument"))
828 828 pats, inc, exc = [], [], []
829 829 hasset = False
830 830 rev, default = None, None
831 831 for arg in l:
832 832 # i18n: "_matchfiles" is a keyword
833 833 s = getstring(arg, _("_matchfiles requires string arguments"))
834 834 prefix, value = s[:2], s[2:]
835 835 if prefix == 'p:':
836 836 pats.append(value)
837 837 elif prefix == 'i:':
838 838 inc.append(value)
839 839 elif prefix == 'x:':
840 840 exc.append(value)
841 841 elif prefix == 'r:':
842 842 if rev is not None:
843 843 # i18n: "_matchfiles" is a keyword
844 844 raise error.ParseError(_('_matchfiles expected at most one '
845 845 'revision'))
846 846 rev = value
847 847 elif prefix == 'd:':
848 848 if default is not None:
849 849 # i18n: "_matchfiles" is a keyword
850 850 raise error.ParseError(_('_matchfiles expected at most one '
851 851 'default mode'))
852 852 default = value
853 853 else:
854 854 # i18n: "_matchfiles" is a keyword
855 855 raise error.ParseError(_('invalid _matchfiles prefix: %s') % prefix)
856 856 if not hasset and matchmod.patkind(value) == 'set':
857 857 hasset = True
858 858 if not default:
859 859 default = 'glob'
860 860
861 861 def matches(x):
862 862 m = None
863 863 c = repo[x]
864 864 if not m or (hasset and rev is None):
865 865 ctx = c
866 866 if rev is not None:
867 867 ctx = repo[rev or None]
868 868 m = matchmod.match(repo.root, repo.getcwd(), pats, include=inc,
869 869 exclude=exc, ctx=ctx, default=default)
870 870 for f in c.files():
871 871 if m(f):
872 872 return True
873 873 return False
874 874
875 875 return subset.filter(matches)
876 876
877 877 def hasfile(repo, subset, x):
878 878 """``file(pattern)``
879 879 Changesets affecting files matched by pattern.
880 880
881 881 For a faster but less accurate result, consider using ``filelog()``
882 882 instead.
883 883
884 884 This predicate uses ``glob:`` as the default kind of pattern.
885 885 """
886 886 # i18n: "file" is a keyword
887 887 pat = getstring(x, _("file requires a pattern"))
888 888 return _matchfiles(repo, subset, ('string', 'p:' + pat))
889 889
890 890 def head(repo, subset, x):
891 891 """``head()``
892 892 Changeset is a named branch head.
893 893 """
894 894 # i18n: "head" is a keyword
895 895 getargs(x, 0, 0, _("head takes no arguments"))
896 896 hs = set()
897 897 for b, ls in repo.branchmap().iteritems():
898 898 hs.update(repo[h].rev() for h in ls)
899 899 return subset.filter(lambda r: r in hs)
900 900
901 901 def heads(repo, subset, x):
902 902 """``heads(set)``
903 903 Members of set with no children in set.
904 904 """
905 905 s = getset(repo, subset, x)
906 906 ps = parents(repo, subset, x)
907 907 return s - ps
908 908
909 909 def hidden(repo, subset, x):
910 910 """``hidden()``
911 911 Hidden changesets.
912 912 """
913 913 # i18n: "hidden" is a keyword
914 914 getargs(x, 0, 0, _("hidden takes no arguments"))
915 915 hiddenrevs = repoview.filterrevs(repo, 'visible')
916 916 return subset & hiddenrevs
917 917
918 918 def keyword(repo, subset, x):
919 919 """``keyword(string)``
920 920 Search commit message, user name, and names of changed files for
921 921 string. The match is case-insensitive.
922 922 """
923 923 # i18n: "keyword" is a keyword
924 924 kw = encoding.lower(getstring(x, _("keyword requires a string")))
925 925
926 926 def matches(r):
927 927 c = repo[r]
928 928 return util.any(kw in encoding.lower(t) for t in c.files() + [c.user(),
929 929 c.description()])
930 930
931 931 return subset.filter(matches)
932 932
933 933 def limit(repo, subset, x):
934 934 """``limit(set, [n])``
935 935 First n members of set, defaulting to 1.
936 936 """
937 937 # i18n: "limit" is a keyword
938 938 l = getargs(x, 1, 2, _("limit requires one or two arguments"))
939 939 try:
940 940 lim = 1
941 941 if len(l) == 2:
942 942 # i18n: "limit" is a keyword
943 943 lim = int(getstring(l[1], _("limit requires a number")))
944 944 except (TypeError, ValueError):
945 945 # i18n: "limit" is a keyword
946 946 raise error.ParseError(_("limit expects a number"))
947 947 ss = subset.set()
948 948 os = getset(repo, spanset(repo), l[0])
949 949 bs = baseset([])
950 950 it = iter(os)
951 951 for x in xrange(lim):
952 952 try:
953 953 y = it.next()
954 954 if y in ss:
955 955 bs.append(y)
956 956 except (StopIteration):
957 957 break
958 958 return bs
959 959
960 960 def last(repo, subset, x):
961 961 """``last(set, [n])``
962 962 Last n members of set, defaulting to 1.
963 963 """
964 964 # i18n: "last" is a keyword
965 965 l = getargs(x, 1, 2, _("last requires one or two arguments"))
966 966 try:
967 967 lim = 1
968 968 if len(l) == 2:
969 969 # i18n: "last" is a keyword
970 970 lim = int(getstring(l[1], _("last requires a number")))
971 971 except (TypeError, ValueError):
972 972 # i18n: "last" is a keyword
973 973 raise error.ParseError(_("last expects a number"))
974 974 ss = subset.set()
975 975 os = getset(repo, spanset(repo), l[0])
976 976 os.reverse()
977 977 bs = baseset([])
978 978 it = iter(os)
979 979 for x in xrange(lim):
980 980 try:
981 981 y = it.next()
982 982 if y in ss:
983 983 bs.append(y)
984 984 except (StopIteration):
985 985 break
986 986 return bs
987 987
988 988 def maxrev(repo, subset, x):
989 989 """``max(set)``
990 990 Changeset with highest revision number in set.
991 991 """
992 992 os = getset(repo, spanset(repo), x)
993 993 if os:
994 994 m = max(os)
995 995 if m in subset:
996 996 return baseset([m])
997 997 return baseset([])
998 998
999 999 def merge(repo, subset, x):
1000 1000 """``merge()``
1001 1001 Changeset is a merge changeset.
1002 1002 """
1003 1003 # i18n: "merge" is a keyword
1004 1004 getargs(x, 0, 0, _("merge takes no arguments"))
1005 1005 cl = repo.changelog
1006 1006 return subset.filter(lambda r: cl.parentrevs(r)[1] != -1)
1007 1007
1008 1008 def branchpoint(repo, subset, x):
1009 1009 """``branchpoint()``
1010 1010 Changesets with more than one child.
1011 1011 """
1012 1012 # i18n: "branchpoint" is a keyword
1013 1013 getargs(x, 0, 0, _("branchpoint takes no arguments"))
1014 1014 cl = repo.changelog
1015 1015 if not subset:
1016 1016 return baseset([])
1017 1017 baserev = min(subset)
1018 1018 parentscount = [0]*(len(repo) - baserev)
1019 1019 for r in cl.revs(start=baserev + 1):
1020 1020 for p in cl.parentrevs(r):
1021 1021 if p >= baserev:
1022 1022 parentscount[p - baserev] += 1
1023 1023 return subset.filter(lambda r: parentscount[r - baserev] > 1)
1024 1024
1025 1025 def minrev(repo, subset, x):
1026 1026 """``min(set)``
1027 1027 Changeset with lowest revision number in set.
1028 1028 """
1029 1029 os = getset(repo, spanset(repo), x)
1030 1030 if os:
1031 1031 m = min(os)
1032 1032 if m in subset:
1033 1033 return baseset([m])
1034 1034 return baseset([])
1035 1035
1036 1036 def _missingancestors(repo, subset, x):
1037 1037 # i18n: "_missingancestors" is a keyword
1038 1038 revs, bases = getargs(x, 2, 2,
1039 1039 _("_missingancestors requires two arguments"))
1040 1040 rs = baseset(repo)
1041 1041 revs = getset(repo, rs, revs)
1042 1042 bases = getset(repo, rs, bases)
1043 1043 missing = set(repo.changelog.findmissingrevs(bases, revs))
1044 1044 return baseset([r for r in subset if r in missing])
1045 1045
1046 1046 def modifies(repo, subset, x):
1047 1047 """``modifies(pattern)``
1048 1048 Changesets modifying files matched by pattern.
1049 1049
1050 1050 The pattern without explicit kind like ``glob:`` is expected to be
1051 1051 relative to the current directory and match against a file or a
1052 1052 directory.
1053 1053 """
1054 1054 # i18n: "modifies" is a keyword
1055 1055 pat = getstring(x, _("modifies requires a pattern"))
1056 1056 return checkstatus(repo, subset, pat, 0)
1057 1057
1058 1058 def node_(repo, subset, x):
1059 1059 """``id(string)``
1060 1060 Revision non-ambiguously specified by the given hex string prefix.
1061 1061 """
1062 1062 # i18n: "id" is a keyword
1063 1063 l = getargs(x, 1, 1, _("id requires one argument"))
1064 1064 # i18n: "id" is a keyword
1065 1065 n = getstring(l[0], _("id requires a string"))
1066 1066 if len(n) == 40:
1067 1067 rn = repo[n].rev()
1068 1068 else:
1069 1069 rn = None
1070 1070 pm = repo.changelog._partialmatch(n)
1071 1071 if pm is not None:
1072 1072 rn = repo.changelog.rev(pm)
1073 1073
1074 1074 return subset.filter(lambda r: r == rn)
1075 1075
1076 1076 def obsolete(repo, subset, x):
1077 1077 """``obsolete()``
1078 1078 Mutable changeset with a newer version."""
1079 1079 # i18n: "obsolete" is a keyword
1080 1080 getargs(x, 0, 0, _("obsolete takes no arguments"))
1081 1081 obsoletes = obsmod.getrevs(repo, 'obsolete')
1082 1082 return subset & obsoletes
1083 1083
1084 1084 def origin(repo, subset, x):
1085 1085 """``origin([set])``
1086 1086 Changesets that were specified as a source for the grafts, transplants or
1087 1087 rebases that created the given revisions. Omitting the optional set is the
1088 1088 same as passing all(). If a changeset created by these operations is itself
1089 1089 specified as a source for one of these operations, only the source changeset
1090 1090 for the first operation is selected.
1091 1091 """
1092 1092 if x is not None:
1093 1093 args = getset(repo, spanset(repo), x).set()
1094 1094 else:
1095 1095 args = getall(repo, spanset(repo), x).set()
1096 1096
1097 1097 def _firstsrc(rev):
1098 1098 src = _getrevsource(repo, rev)
1099 1099 if src is None:
1100 1100 return None
1101 1101
1102 1102 while True:
1103 1103 prev = _getrevsource(repo, src)
1104 1104
1105 1105 if prev is None:
1106 1106 return src
1107 1107 src = prev
1108 1108
1109 1109 o = set([_firstsrc(r) for r in args])
1110 1110 return subset.filter(lambda r: r in o)
1111 1111
1112 1112 def outgoing(repo, subset, x):
1113 1113 """``outgoing([path])``
1114 1114 Changesets not found in the specified destination repository, or the
1115 1115 default push location.
1116 1116 """
1117 1117 import hg # avoid start-up nasties
1118 1118 # i18n: "outgoing" is a keyword
1119 1119 l = getargs(x, 0, 1, _("outgoing takes one or no arguments"))
1120 1120 # i18n: "outgoing" is a keyword
1121 1121 dest = l and getstring(l[0], _("outgoing requires a repository path")) or ''
1122 1122 dest = repo.ui.expandpath(dest or 'default-push', dest or 'default')
1123 1123 dest, branches = hg.parseurl(dest)
1124 1124 revs, checkout = hg.addbranchrevs(repo, repo, branches, [])
1125 1125 if revs:
1126 1126 revs = [repo.lookup(rev) for rev in revs]
1127 1127 other = hg.peer(repo, {}, dest)
1128 1128 repo.ui.pushbuffer()
1129 1129 outgoing = discovery.findcommonoutgoing(repo, other, onlyheads=revs)
1130 1130 repo.ui.popbuffer()
1131 1131 cl = repo.changelog
1132 1132 o = set([cl.rev(r) for r in outgoing.missing])
1133 1133 return subset.filter(lambda r: r in o)
1134 1134
1135 1135 def p1(repo, subset, x):
1136 1136 """``p1([set])``
1137 1137 First parent of changesets in set, or the working directory.
1138 1138 """
1139 1139 if x is None:
1140 1140 p = repo[x].p1().rev()
1141 1141 return subset.filter(lambda r: r == p)
1142 1142
1143 1143 ps = set()
1144 1144 cl = repo.changelog
1145 1145 for r in getset(repo, spanset(repo), x):
1146 1146 ps.add(cl.parentrevs(r)[0])
1147 1147 return subset & ps
1148 1148
1149 1149 def p2(repo, subset, x):
1150 1150 """``p2([set])``
1151 1151 Second parent of changesets in set, or the working directory.
1152 1152 """
1153 1153 if x is None:
1154 1154 ps = repo[x].parents()
1155 1155 try:
1156 1156 p = ps[1].rev()
1157 1157 return subset.filter(lambda r: r == p)
1158 1158 except IndexError:
1159 1159 return baseset([])
1160 1160
1161 1161 ps = set()
1162 1162 cl = repo.changelog
1163 1163 for r in getset(repo, spanset(repo), x):
1164 1164 ps.add(cl.parentrevs(r)[1])
1165 1165 return subset & ps
1166 1166
1167 1167 def parents(repo, subset, x):
1168 1168 """``parents([set])``
1169 1169 The set of all parents for all changesets in set, or the working directory.
1170 1170 """
1171 1171 if x is None:
1172 1172 ps = tuple(p.rev() for p in repo[x].parents())
1173 1173 return subset & ps
1174 1174
1175 1175 ps = set()
1176 1176 cl = repo.changelog
1177 1177 for r in getset(repo, spanset(repo), x):
1178 1178 ps.update(cl.parentrevs(r))
1179 1179 return subset & ps
1180 1180
1181 1181 def parentspec(repo, subset, x, n):
1182 1182 """``set^0``
1183 1183 The set.
1184 1184 ``set^1`` (or ``set^``), ``set^2``
1185 1185 First or second parent, respectively, of all changesets in set.
1186 1186 """
1187 1187 try:
1188 1188 n = int(n[1])
1189 1189 if n not in (0, 1, 2):
1190 1190 raise ValueError
1191 1191 except (TypeError, ValueError):
1192 1192 raise error.ParseError(_("^ expects a number 0, 1, or 2"))
1193 1193 ps = set()
1194 1194 cl = repo.changelog
1195 1195 for r in getset(repo, baseset(cl), x):
1196 1196 if n == 0:
1197 1197 ps.add(r)
1198 1198 elif n == 1:
1199 1199 ps.add(cl.parentrevs(r)[0])
1200 1200 elif n == 2:
1201 1201 parents = cl.parentrevs(r)
1202 1202 if len(parents) > 1:
1203 1203 ps.add(parents[1])
1204 1204 return subset & ps
1205 1205
1206 1206 def present(repo, subset, x):
1207 1207 """``present(set)``
1208 1208 An empty set, if any revision in set isn't found; otherwise,
1209 1209 all revisions in set.
1210 1210
1211 1211 If any of specified revisions is not present in the local repository,
1212 1212 the query is normally aborted. But this predicate allows the query
1213 1213 to continue even in such cases.
1214 1214 """
1215 1215 try:
1216 1216 return getset(repo, subset, x)
1217 1217 except error.RepoLookupError:
1218 1218 return baseset([])
1219 1219
1220 1220 def public(repo, subset, x):
1221 1221 """``public()``
1222 1222 Changeset in public phase."""
1223 1223 # i18n: "public" is a keyword
1224 1224 getargs(x, 0, 0, _("public takes no arguments"))
1225 1225 pc = repo._phasecache
1226 1226 return subset.filter(lambda r: pc.phase(repo, r) == phases.public)
1227 1227
1228 1228 def remote(repo, subset, x):
1229 1229 """``remote([id [,path]])``
1230 1230 Local revision that corresponds to the given identifier in a
1231 1231 remote repository, if present. Here, the '.' identifier is a
1232 1232 synonym for the current local branch.
1233 1233 """
1234 1234
1235 1235 import hg # avoid start-up nasties
1236 1236 # i18n: "remote" is a keyword
1237 1237 l = getargs(x, 0, 2, _("remote takes one, two or no arguments"))
1238 1238
1239 1239 q = '.'
1240 1240 if len(l) > 0:
1241 1241 # i18n: "remote" is a keyword
1242 1242 q = getstring(l[0], _("remote requires a string id"))
1243 1243 if q == '.':
1244 1244 q = repo['.'].branch()
1245 1245
1246 1246 dest = ''
1247 1247 if len(l) > 1:
1248 1248 # i18n: "remote" is a keyword
1249 1249 dest = getstring(l[1], _("remote requires a repository path"))
1250 1250 dest = repo.ui.expandpath(dest or 'default')
1251 1251 dest, branches = hg.parseurl(dest)
1252 1252 revs, checkout = hg.addbranchrevs(repo, repo, branches, [])
1253 1253 if revs:
1254 1254 revs = [repo.lookup(rev) for rev in revs]
1255 1255 other = hg.peer(repo, {}, dest)
1256 1256 n = other.lookup(q)
1257 1257 if n in repo:
1258 1258 r = repo[n].rev()
1259 1259 if r in subset:
1260 1260 return baseset([r])
1261 1261 return baseset([])
1262 1262
1263 1263 def removes(repo, subset, x):
1264 1264 """``removes(pattern)``
1265 1265 Changesets which remove files matching pattern.
1266 1266
1267 1267 The pattern without explicit kind like ``glob:`` is expected to be
1268 1268 relative to the current directory and match against a file or a
1269 1269 directory.
1270 1270 """
1271 1271 # i18n: "removes" is a keyword
1272 1272 pat = getstring(x, _("removes requires a pattern"))
1273 1273 return checkstatus(repo, subset, pat, 2)
1274 1274
1275 1275 def rev(repo, subset, x):
1276 1276 """``rev(number)``
1277 1277 Revision with the given numeric identifier.
1278 1278 """
1279 1279 # i18n: "rev" is a keyword
1280 1280 l = getargs(x, 1, 1, _("rev requires one argument"))
1281 1281 try:
1282 1282 # i18n: "rev" is a keyword
1283 1283 l = int(getstring(l[0], _("rev requires a number")))
1284 1284 except (TypeError, ValueError):
1285 1285 # i18n: "rev" is a keyword
1286 1286 raise error.ParseError(_("rev expects a number"))
1287 1287 return subset.filter(lambda r: r == l)
1288 1288
1289 1289 def matching(repo, subset, x):
1290 1290 """``matching(revision [, field])``
1291 1291 Changesets in which a given set of fields match the set of fields in the
1292 1292 selected revision or set.
1293 1293
1294 1294 To match more than one field pass the list of fields to match separated
1295 1295 by spaces (e.g. ``author description``).
1296 1296
1297 1297 Valid fields are most regular revision fields and some special fields.
1298 1298
1299 1299 Regular revision fields are ``description``, ``author``, ``branch``,
1300 1300 ``date``, ``files``, ``phase``, ``parents``, ``substate``, ``user``
1301 1301 and ``diff``.
1302 1302 Note that ``author`` and ``user`` are synonyms. ``diff`` refers to the
1303 1303 contents of the revision. Two revisions matching their ``diff`` will
1304 1304 also match their ``files``.
1305 1305
1306 1306 Special fields are ``summary`` and ``metadata``:
1307 1307 ``summary`` matches the first line of the description.
1308 1308 ``metadata`` is equivalent to matching ``description user date``
1309 1309 (i.e. it matches the main metadata fields).
1310 1310
1311 1311 ``metadata`` is the default field which is used when no fields are
1312 1312 specified. You can match more than one field at a time.
1313 1313 """
1314 1314 # i18n: "matching" is a keyword
1315 1315 l = getargs(x, 1, 2, _("matching takes 1 or 2 arguments"))
1316 1316
1317 1317 revs = getset(repo, baseset(repo.changelog), l[0])
1318 1318
1319 1319 fieldlist = ['metadata']
1320 1320 if len(l) > 1:
1321 1321 fieldlist = getstring(l[1],
1322 1322 # i18n: "matching" is a keyword
1323 1323 _("matching requires a string "
1324 1324 "as its second argument")).split()
1325 1325
1326 1326 # Make sure that there are no repeated fields,
1327 1327 # expand the 'special' 'metadata' field type
1328 1328 # and check the 'files' whenever we check the 'diff'
1329 1329 fields = []
1330 1330 for field in fieldlist:
1331 1331 if field == 'metadata':
1332 1332 fields += ['user', 'description', 'date']
1333 1333 elif field == 'diff':
1334 1334 # a revision matching the diff must also match the files
1335 1335 # since matching the diff is very costly, make sure to
1336 1336 # also match the files first
1337 1337 fields += ['files', 'diff']
1338 1338 else:
1339 1339 if field == 'author':
1340 1340 field = 'user'
1341 1341 fields.append(field)
1342 1342 fields = set(fields)
1343 1343 if 'summary' in fields and 'description' in fields:
1344 1344 # If a revision matches its description it also matches its summary
1345 1345 fields.discard('summary')
1346 1346
1347 1347 # We may want to match more than one field
1348 1348 # Not all fields take the same amount of time to be matched
1349 1349 # Sort the selected fields in order of increasing matching cost
1350 1350 fieldorder = ['phase', 'parents', 'user', 'date', 'branch', 'summary',
1351 1351 'files', 'description', 'substate', 'diff']
1352 1352 def fieldkeyfunc(f):
1353 1353 try:
1354 1354 return fieldorder.index(f)
1355 1355 except ValueError:
1356 1356 # assume an unknown field is very costly
1357 1357 return len(fieldorder)
1358 1358 fields = list(fields)
1359 1359 fields.sort(key=fieldkeyfunc)
1360 1360
1361 1361 # Each field will be matched with its own "getfield" function
1362 1362 # which will be added to the getfieldfuncs array of functions
1363 1363 getfieldfuncs = []
1364 1364 _funcs = {
1365 1365 'user': lambda r: repo[r].user(),
1366 1366 'branch': lambda r: repo[r].branch(),
1367 1367 'date': lambda r: repo[r].date(),
1368 1368 'description': lambda r: repo[r].description(),
1369 1369 'files': lambda r: repo[r].files(),
1370 1370 'parents': lambda r: repo[r].parents(),
1371 1371 'phase': lambda r: repo[r].phase(),
1372 1372 'substate': lambda r: repo[r].substate,
1373 1373 'summary': lambda r: repo[r].description().splitlines()[0],
1374 1374 'diff': lambda r: list(repo[r].diff(git=True),)
1375 1375 }
1376 1376 for info in fields:
1377 1377 getfield = _funcs.get(info, None)
1378 1378 if getfield is None:
1379 1379 raise error.ParseError(
1380 1380 # i18n: "matching" is a keyword
1381 1381 _("unexpected field name passed to matching: %s") % info)
1382 1382 getfieldfuncs.append(getfield)
1383 1383 # convert the getfield array of functions into a "getinfo" function
1384 1384 # which returns an array of field values (or a single value if there
1385 1385 # is only one field to match)
1386 1386 getinfo = lambda r: [f(r) for f in getfieldfuncs]
1387 1387
1388 1388 def matches(x):
1389 1389 for rev in revs:
1390 1390 target = getinfo(rev)
1391 1391 match = True
1392 1392 for n, f in enumerate(getfieldfuncs):
1393 1393 if target[n] != f(x):
1394 1394 match = False
1395 1395 if match:
1396 1396 return True
1397 1397 return False
1398 1398
1399 1399 return subset.filter(matches)
1400 1400
1401 1401 def reverse(repo, subset, x):
1402 1402 """``reverse(set)``
1403 1403 Reverse order of set.
1404 1404 """
1405 1405 l = getset(repo, subset, x)
1406 1406 l.reverse()
1407 1407 return l
1408 1408
1409 1409 def roots(repo, subset, x):
1410 1410 """``roots(set)``
1411 1411 Changesets in set with no parent changeset in set.
1412 1412 """
1413 1413 s = getset(repo, baseset(repo.changelog), x).set()
1414 1414 subset = baseset([r for r in subset if r in s])
1415 1415 cs = _children(repo, subset, s)
1416 1416 return subset - cs
1417 1417
1418 1418 def secret(repo, subset, x):
1419 1419 """``secret()``
1420 1420 Changeset in secret phase."""
1421 1421 # i18n: "secret" is a keyword
1422 1422 getargs(x, 0, 0, _("secret takes no arguments"))
1423 1423 pc = repo._phasecache
1424 1424 return subset.filter(lambda x: pc.phase(repo, x) == phases.secret)
1425 1425
1426 1426 def sort(repo, subset, x):
1427 1427 """``sort(set[, [-]key...])``
1428 1428 Sort set by keys. The default sort order is ascending, specify a key
1429 1429 as ``-key`` to sort in descending order.
1430 1430
1431 1431 The keys can be:
1432 1432
1433 1433 - ``rev`` for the revision number,
1434 1434 - ``branch`` for the branch name,
1435 1435 - ``desc`` for the commit message (description),
1436 1436 - ``user`` for user name (``author`` can be used as an alias),
1437 1437 - ``date`` for the commit date
1438 1438 """
1439 1439 # i18n: "sort" is a keyword
1440 1440 l = getargs(x, 1, 2, _("sort requires one or two arguments"))
1441 1441 keys = "rev"
1442 1442 if len(l) == 2:
1443 1443 # i18n: "sort" is a keyword
1444 1444 keys = getstring(l[1], _("sort spec must be a string"))
1445 1445
1446 1446 s = l[0]
1447 1447 keys = keys.split()
1448 1448 l = []
1449 1449 def invert(s):
1450 1450 return "".join(chr(255 - ord(c)) for c in s)
1451 1451 for r in getset(repo, subset, s):
1452 1452 c = repo[r]
1453 1453 e = []
1454 1454 for k in keys:
1455 1455 if k == 'rev':
1456 1456 e.append(r)
1457 1457 elif k == '-rev':
1458 1458 e.append(-r)
1459 1459 elif k == 'branch':
1460 1460 e.append(c.branch())
1461 1461 elif k == '-branch':
1462 1462 e.append(invert(c.branch()))
1463 1463 elif k == 'desc':
1464 1464 e.append(c.description())
1465 1465 elif k == '-desc':
1466 1466 e.append(invert(c.description()))
1467 1467 elif k in 'user author':
1468 1468 e.append(c.user())
1469 1469 elif k in '-user -author':
1470 1470 e.append(invert(c.user()))
1471 1471 elif k == 'date':
1472 1472 e.append(c.date()[0])
1473 1473 elif k == '-date':
1474 1474 e.append(-c.date()[0])
1475 1475 else:
1476 1476 raise error.ParseError(_("unknown sort key %r") % k)
1477 1477 e.append(r)
1478 1478 l.append(e)
1479 1479 l.sort()
1480 1480 return baseset([e[-1] for e in l])
1481 1481
1482 1482 def _stringmatcher(pattern):
1483 1483 """
1484 1484 accepts a string, possibly starting with 're:' or 'literal:' prefix.
1485 1485 returns the matcher name, pattern, and matcher function.
1486 1486 missing or unknown prefixes are treated as literal matches.
1487 1487
1488 1488 helper for tests:
1489 1489 >>> def test(pattern, *tests):
1490 1490 ... kind, pattern, matcher = _stringmatcher(pattern)
1491 1491 ... return (kind, pattern, [bool(matcher(t)) for t in tests])
1492 1492
1493 1493 exact matching (no prefix):
1494 1494 >>> test('abcdefg', 'abc', 'def', 'abcdefg')
1495 1495 ('literal', 'abcdefg', [False, False, True])
1496 1496
1497 1497 regex matching ('re:' prefix)
1498 1498 >>> test('re:a.+b', 'nomatch', 'fooadef', 'fooadefbar')
1499 1499 ('re', 'a.+b', [False, False, True])
1500 1500
1501 1501 force exact matches ('literal:' prefix)
1502 1502 >>> test('literal:re:foobar', 'foobar', 're:foobar')
1503 1503 ('literal', 're:foobar', [False, True])
1504 1504
1505 1505 unknown prefixes are ignored and treated as literals
1506 1506 >>> test('foo:bar', 'foo', 'bar', 'foo:bar')
1507 1507 ('literal', 'foo:bar', [False, False, True])
1508 1508 """
1509 1509 if pattern.startswith('re:'):
1510 1510 pattern = pattern[3:]
1511 1511 try:
1512 1512 regex = re.compile(pattern)
1513 1513 except re.error, e:
1514 1514 raise error.ParseError(_('invalid regular expression: %s')
1515 1515 % e)
1516 1516 return 're', pattern, regex.search
1517 1517 elif pattern.startswith('literal:'):
1518 1518 pattern = pattern[8:]
1519 1519 return 'literal', pattern, pattern.__eq__
1520 1520
1521 1521 def _substringmatcher(pattern):
1522 1522 kind, pattern, matcher = _stringmatcher(pattern)
1523 1523 if kind == 'literal':
1524 1524 matcher = lambda s: pattern in s
1525 1525 return kind, pattern, matcher
1526 1526
1527 1527 def tag(repo, subset, x):
1528 1528 """``tag([name])``
1529 1529 The specified tag by name, or all tagged revisions if no name is given.
1530 1530 """
1531 1531 # i18n: "tag" is a keyword
1532 1532 args = getargs(x, 0, 1, _("tag takes one or no arguments"))
1533 1533 cl = repo.changelog
1534 1534 if args:
1535 1535 pattern = getstring(args[0],
1536 1536 # i18n: "tag" is a keyword
1537 1537 _('the argument to tag must be a string'))
1538 1538 kind, pattern, matcher = _stringmatcher(pattern)
1539 1539 if kind == 'literal':
1540 1540 # avoid resolving all tags
1541 1541 tn = repo._tagscache.tags.get(pattern, None)
1542 1542 if tn is None:
1543 1543 raise util.Abort(_("tag '%s' does not exist") % pattern)
1544 1544 s = set([repo[tn].rev()])
1545 1545 else:
1546 1546 s = set([cl.rev(n) for t, n in repo.tagslist() if matcher(t)])
1547 1547 else:
1548 1548 s = set([cl.rev(n) for t, n in repo.tagslist() if t != 'tip'])
1549 1549 return subset & s
1550 1550
1551 1551 def tagged(repo, subset, x):
1552 1552 return tag(repo, subset, x)
1553 1553
1554 1554 def unstable(repo, subset, x):
1555 1555 """``unstable()``
1556 1556 Non-obsolete changesets with obsolete ancestors.
1557 1557 """
1558 1558 # i18n: "unstable" is a keyword
1559 1559 getargs(x, 0, 0, _("unstable takes no arguments"))
1560 1560 unstables = obsmod.getrevs(repo, 'unstable')
1561 1561 return subset & unstables
1562 1562
1563 1563
1564 1564 def user(repo, subset, x):
1565 1565 """``user(string)``
1566 1566 User name contains string. The match is case-insensitive.
1567 1567
1568 1568 If `string` starts with `re:`, the remainder of the string is treated as
1569 1569 a regular expression. To match a user that actually contains `re:`, use
1570 1570 the prefix `literal:`.
1571 1571 """
1572 1572 return author(repo, subset, x)
1573 1573
1574 1574 # for internal use
1575 1575 def _list(repo, subset, x):
1576 1576 s = getstring(x, "internal error")
1577 1577 if not s:
1578 1578 return baseset([])
1579 1579 ls = [repo[r].rev() for r in s.split('\0')]
1580 1580 s = subset.set()
1581 1581 return baseset([r for r in ls if r in s])
1582 1582
1583 1583 # for internal use
1584 1584 def _intlist(repo, subset, x):
1585 1585 s = getstring(x, "internal error")
1586 1586 if not s:
1587 1587 return baseset([])
1588 1588 ls = [int(r) for r in s.split('\0')]
1589 1589 s = subset.set()
1590 1590 return baseset([r for r in ls if r in s])
1591 1591
1592 1592 # for internal use
1593 1593 def _hexlist(repo, subset, x):
1594 1594 s = getstring(x, "internal error")
1595 1595 if not s:
1596 1596 return baseset([])
1597 1597 cl = repo.changelog
1598 1598 ls = [cl.rev(node.bin(r)) for r in s.split('\0')]
1599 1599 s = subset.set()
1600 1600 return baseset([r for r in ls if r in s])
1601 1601
1602 1602 symbols = {
1603 1603 "adds": adds,
1604 1604 "all": getall,
1605 1605 "ancestor": ancestor,
1606 1606 "ancestors": ancestors,
1607 1607 "_firstancestors": _firstancestors,
1608 1608 "author": author,
1609 1609 "bisect": bisect,
1610 1610 "bisected": bisected,
1611 1611 "bookmark": bookmark,
1612 1612 "branch": branch,
1613 1613 "branchpoint": branchpoint,
1614 1614 "bumped": bumped,
1615 1615 "bundle": bundle,
1616 1616 "children": children,
1617 1617 "closed": closed,
1618 1618 "contains": contains,
1619 1619 "converted": converted,
1620 1620 "date": date,
1621 1621 "desc": desc,
1622 1622 "descendants": descendants,
1623 1623 "_firstdescendants": _firstdescendants,
1624 1624 "destination": destination,
1625 1625 "divergent": divergent,
1626 1626 "draft": draft,
1627 1627 "extinct": extinct,
1628 1628 "extra": extra,
1629 1629 "file": hasfile,
1630 1630 "filelog": filelog,
1631 1631 "first": first,
1632 1632 "follow": follow,
1633 1633 "_followfirst": _followfirst,
1634 1634 "grep": grep,
1635 1635 "head": head,
1636 1636 "heads": heads,
1637 1637 "hidden": hidden,
1638 1638 "id": node_,
1639 1639 "keyword": keyword,
1640 1640 "last": last,
1641 1641 "limit": limit,
1642 1642 "_matchfiles": _matchfiles,
1643 1643 "max": maxrev,
1644 1644 "merge": merge,
1645 1645 "min": minrev,
1646 1646 "_missingancestors": _missingancestors,
1647 1647 "modifies": modifies,
1648 1648 "obsolete": obsolete,
1649 1649 "origin": origin,
1650 1650 "outgoing": outgoing,
1651 1651 "p1": p1,
1652 1652 "p2": p2,
1653 1653 "parents": parents,
1654 1654 "present": present,
1655 1655 "public": public,
1656 1656 "remote": remote,
1657 1657 "removes": removes,
1658 1658 "rev": rev,
1659 1659 "reverse": reverse,
1660 1660 "roots": roots,
1661 1661 "sort": sort,
1662 1662 "secret": secret,
1663 1663 "matching": matching,
1664 1664 "tag": tag,
1665 1665 "tagged": tagged,
1666 1666 "user": user,
1667 1667 "unstable": unstable,
1668 1668 "_list": _list,
1669 1669 "_intlist": _intlist,
1670 1670 "_hexlist": _hexlist,
1671 1671 }
1672 1672
1673 1673 # symbols which can't be used for a DoS attack for any given input
1674 1674 # (e.g. those which accept regexes as plain strings shouldn't be included)
1675 1675 # functions that just return a lot of changesets (like all) don't count here
1676 1676 safesymbols = set([
1677 1677 "adds",
1678 1678 "all",
1679 1679 "ancestor",
1680 1680 "ancestors",
1681 1681 "_firstancestors",
1682 1682 "author",
1683 1683 "bisect",
1684 1684 "bisected",
1685 1685 "bookmark",
1686 1686 "branch",
1687 1687 "branchpoint",
1688 1688 "bumped",
1689 1689 "bundle",
1690 1690 "children",
1691 1691 "closed",
1692 1692 "converted",
1693 1693 "date",
1694 1694 "desc",
1695 1695 "descendants",
1696 1696 "_firstdescendants",
1697 1697 "destination",
1698 1698 "divergent",
1699 1699 "draft",
1700 1700 "extinct",
1701 1701 "extra",
1702 1702 "file",
1703 1703 "filelog",
1704 1704 "first",
1705 1705 "follow",
1706 1706 "_followfirst",
1707 1707 "head",
1708 1708 "heads",
1709 1709 "hidden",
1710 1710 "id",
1711 1711 "keyword",
1712 1712 "last",
1713 1713 "limit",
1714 1714 "_matchfiles",
1715 1715 "max",
1716 1716 "merge",
1717 1717 "min",
1718 1718 "_missingancestors",
1719 1719 "modifies",
1720 1720 "obsolete",
1721 1721 "origin",
1722 1722 "outgoing",
1723 1723 "p1",
1724 1724 "p2",
1725 1725 "parents",
1726 1726 "present",
1727 1727 "public",
1728 1728 "remote",
1729 1729 "removes",
1730 1730 "rev",
1731 1731 "reverse",
1732 1732 "roots",
1733 1733 "sort",
1734 1734 "secret",
1735 1735 "matching",
1736 1736 "tag",
1737 1737 "tagged",
1738 1738 "user",
1739 1739 "unstable",
1740 1740 "_list",
1741 1741 "_intlist",
1742 1742 "_hexlist",
1743 1743 ])
1744 1744
1745 1745 methods = {
1746 1746 "range": rangeset,
1747 1747 "dagrange": dagrange,
1748 1748 "string": stringset,
1749 1749 "symbol": symbolset,
1750 1750 "and": andset,
1751 1751 "or": orset,
1752 1752 "not": notset,
1753 1753 "list": listset,
1754 1754 "func": func,
1755 1755 "ancestor": ancestorspec,
1756 1756 "parent": parentspec,
1757 1757 "parentpost": p1,
1758 1758 }
1759 1759
1760 1760 def optimize(x, small):
1761 1761 if x is None:
1762 1762 return 0, x
1763 1763
1764 1764 smallbonus = 1
1765 1765 if small:
1766 1766 smallbonus = .5
1767 1767
1768 1768 op = x[0]
1769 1769 if op == 'minus':
1770 1770 return optimize(('and', x[1], ('not', x[2])), small)
1771 1771 elif op == 'dagrangepre':
1772 1772 return optimize(('func', ('symbol', 'ancestors'), x[1]), small)
1773 1773 elif op == 'dagrangepost':
1774 1774 return optimize(('func', ('symbol', 'descendants'), x[1]), small)
1775 1775 elif op == 'rangepre':
1776 1776 return optimize(('range', ('string', '0'), x[1]), small)
1777 1777 elif op == 'rangepost':
1778 1778 return optimize(('range', x[1], ('string', 'tip')), small)
1779 1779 elif op == 'negate':
1780 1780 return optimize(('string',
1781 1781 '-' + getstring(x[1], _("can't negate that"))), small)
1782 1782 elif op in 'string symbol negate':
1783 1783 return smallbonus, x # single revisions are small
1784 1784 elif op == 'and':
1785 1785 wa, ta = optimize(x[1], True)
1786 1786 wb, tb = optimize(x[2], True)
1787 1787
1788 1788 # (::x and not ::y)/(not ::y and ::x) have a fast path
1789 1789 def ismissingancestors(revs, bases):
1790 1790 return (
1791 1791 revs[0] == 'func'
1792 1792 and getstring(revs[1], _('not a symbol')) == 'ancestors'
1793 1793 and bases[0] == 'not'
1794 1794 and bases[1][0] == 'func'
1795 1795 and getstring(bases[1][1], _('not a symbol')) == 'ancestors')
1796 1796
1797 1797 w = min(wa, wb)
1798 1798 if ismissingancestors(ta, tb):
1799 1799 return w, ('func', ('symbol', '_missingancestors'),
1800 1800 ('list', ta[2], tb[1][2]))
1801 1801 if ismissingancestors(tb, ta):
1802 1802 return w, ('func', ('symbol', '_missingancestors'),
1803 1803 ('list', tb[2], ta[1][2]))
1804 1804
1805 1805 if wa > wb:
1806 1806 return w, (op, tb, ta)
1807 1807 return w, (op, ta, tb)
1808 1808 elif op == 'or':
1809 1809 wa, ta = optimize(x[1], False)
1810 1810 wb, tb = optimize(x[2], False)
1811 1811 if wb < wa:
1812 1812 wb, wa = wa, wb
1813 1813 return max(wa, wb), (op, ta, tb)
1814 1814 elif op == 'not':
1815 1815 o = optimize(x[1], not small)
1816 1816 return o[0], (op, o[1])
1817 1817 elif op == 'parentpost':
1818 1818 o = optimize(x[1], small)
1819 1819 return o[0], (op, o[1])
1820 1820 elif op == 'group':
1821 1821 return optimize(x[1], small)
1822 1822 elif op in 'dagrange range list parent ancestorspec':
1823 1823 if op == 'parent':
1824 1824 # x^:y means (x^) : y, not x ^ (:y)
1825 1825 post = ('parentpost', x[1])
1826 1826 if x[2][0] == 'dagrangepre':
1827 1827 return optimize(('dagrange', post, x[2][1]), small)
1828 1828 elif x[2][0] == 'rangepre':
1829 1829 return optimize(('range', post, x[2][1]), small)
1830 1830
1831 1831 wa, ta = optimize(x[1], small)
1832 1832 wb, tb = optimize(x[2], small)
1833 1833 return wa + wb, (op, ta, tb)
1834 1834 elif op == 'func':
1835 1835 f = getstring(x[1], _("not a symbol"))
1836 1836 wa, ta = optimize(x[2], small)
1837 1837 if f in ("author branch closed date desc file grep keyword "
1838 1838 "outgoing user"):
1839 1839 w = 10 # slow
1840 1840 elif f in "modifies adds removes":
1841 1841 w = 30 # slower
1842 1842 elif f == "contains":
1843 1843 w = 100 # very slow
1844 1844 elif f == "ancestor":
1845 1845 w = 1 * smallbonus
1846 1846 elif f in "reverse limit first":
1847 1847 w = 0
1848 1848 elif f in "sort":
1849 1849 w = 10 # assume most sorts look at changelog
1850 1850 else:
1851 1851 w = 1
1852 1852 return w + wa, (op, x[1], ta)
1853 1853 return 1, x
1854 1854
1855 1855 _aliasarg = ('func', ('symbol', '_aliasarg'))
1856 1856 def _getaliasarg(tree):
1857 1857 """If tree matches ('func', ('symbol', '_aliasarg'), ('string', X))
1858 1858 return X, None otherwise.
1859 1859 """
1860 1860 if (len(tree) == 3 and tree[:2] == _aliasarg
1861 1861 and tree[2][0] == 'string'):
1862 1862 return tree[2][1]
1863 1863 return None
1864 1864
1865 1865 def _checkaliasarg(tree, known=None):
1866 1866 """Check tree contains no _aliasarg construct or only ones which
1867 1867 value is in known. Used to avoid alias placeholders injection.
1868 1868 """
1869 1869 if isinstance(tree, tuple):
1870 1870 arg = _getaliasarg(tree)
1871 1871 if arg is not None and (not known or arg not in known):
1872 1872 raise error.ParseError(_("not a function: %s") % '_aliasarg')
1873 1873 for t in tree:
1874 1874 _checkaliasarg(t, known)
1875 1875
1876 1876 class revsetalias(object):
1877 1877 funcre = re.compile('^([^(]+)\(([^)]+)\)$')
1878 1878 args = None
1879 1879
1880 1880 def __init__(self, name, value):
1881 1881 '''Aliases like:
1882 1882
1883 1883 h = heads(default)
1884 1884 b($1) = ancestors($1) - ancestors(default)
1885 1885 '''
1886 1886 m = self.funcre.search(name)
1887 1887 if m:
1888 1888 self.name = m.group(1)
1889 1889 self.tree = ('func', ('symbol', m.group(1)))
1890 1890 self.args = [x.strip() for x in m.group(2).split(',')]
1891 1891 for arg in self.args:
1892 1892 # _aliasarg() is an unknown symbol only used separate
1893 1893 # alias argument placeholders from regular strings.
1894 1894 value = value.replace(arg, '_aliasarg(%r)' % (arg,))
1895 1895 else:
1896 1896 self.name = name
1897 1897 self.tree = ('symbol', name)
1898 1898
1899 1899 self.replacement, pos = parse(value)
1900 1900 if pos != len(value):
1901 1901 raise error.ParseError(_('invalid token'), pos)
1902 1902 # Check for placeholder injection
1903 1903 _checkaliasarg(self.replacement, self.args)
1904 1904
1905 1905 def _getalias(aliases, tree):
1906 1906 """If tree looks like an unexpanded alias, return it. Return None
1907 1907 otherwise.
1908 1908 """
1909 1909 if isinstance(tree, tuple) and tree:
1910 1910 if tree[0] == 'symbol' and len(tree) == 2:
1911 1911 name = tree[1]
1912 1912 alias = aliases.get(name)
1913 1913 if alias and alias.args is None and alias.tree == tree:
1914 1914 return alias
1915 1915 if tree[0] == 'func' and len(tree) > 1:
1916 1916 if tree[1][0] == 'symbol' and len(tree[1]) == 2:
1917 1917 name = tree[1][1]
1918 1918 alias = aliases.get(name)
1919 1919 if alias and alias.args is not None and alias.tree == tree[:2]:
1920 1920 return alias
1921 1921 return None
1922 1922
1923 1923 def _expandargs(tree, args):
1924 1924 """Replace _aliasarg instances with the substitution value of the
1925 1925 same name in args, recursively.
1926 1926 """
1927 1927 if not tree or not isinstance(tree, tuple):
1928 1928 return tree
1929 1929 arg = _getaliasarg(tree)
1930 1930 if arg is not None:
1931 1931 return args[arg]
1932 1932 return tuple(_expandargs(t, args) for t in tree)
1933 1933
1934 1934 def _expandaliases(aliases, tree, expanding, cache):
1935 1935 """Expand aliases in tree, recursively.
1936 1936
1937 1937 'aliases' is a dictionary mapping user defined aliases to
1938 1938 revsetalias objects.
1939 1939 """
1940 1940 if not isinstance(tree, tuple):
1941 1941 # Do not expand raw strings
1942 1942 return tree
1943 1943 alias = _getalias(aliases, tree)
1944 1944 if alias is not None:
1945 1945 if alias in expanding:
1946 1946 raise error.ParseError(_('infinite expansion of revset alias "%s" '
1947 1947 'detected') % alias.name)
1948 1948 expanding.append(alias)
1949 1949 if alias.name not in cache:
1950 1950 cache[alias.name] = _expandaliases(aliases, alias.replacement,
1951 1951 expanding, cache)
1952 1952 result = cache[alias.name]
1953 1953 expanding.pop()
1954 1954 if alias.args is not None:
1955 1955 l = getlist(tree[2])
1956 1956 if len(l) != len(alias.args):
1957 1957 raise error.ParseError(
1958 1958 _('invalid number of arguments: %s') % len(l))
1959 1959 l = [_expandaliases(aliases, a, [], cache) for a in l]
1960 1960 result = _expandargs(result, dict(zip(alias.args, l)))
1961 1961 else:
1962 1962 result = tuple(_expandaliases(aliases, t, expanding, cache)
1963 1963 for t in tree)
1964 1964 return result
1965 1965
1966 1966 def findaliases(ui, tree):
1967 1967 _checkaliasarg(tree)
1968 1968 aliases = {}
1969 1969 for k, v in ui.configitems('revsetalias'):
1970 1970 alias = revsetalias(k, v)
1971 1971 aliases[alias.name] = alias
1972 1972 return _expandaliases(aliases, tree, [], {})
1973 1973
1974 1974 def parse(spec):
1975 1975 p = parser.parser(tokenize, elements)
1976 1976 return p.parse(spec)
1977 1977
1978 1978 def match(ui, spec):
1979 1979 if not spec:
1980 1980 raise error.ParseError(_("empty query"))
1981 1981 tree, pos = parse(spec)
1982 1982 if (pos != len(spec)):
1983 1983 raise error.ParseError(_("invalid token"), pos)
1984 1984 if ui:
1985 1985 tree = findaliases(ui, tree)
1986 1986 weight, tree = optimize(tree, True)
1987 1987 def mfunc(repo, subset):
1988 1988 if util.safehasattr(subset, 'set'):
1989 1989 return getset(repo, subset, tree)
1990 1990 return getset(repo, baseset(subset), tree)
1991 1991 return mfunc
1992 1992
1993 1993 def formatspec(expr, *args):
1994 1994 '''
1995 1995 This is a convenience function for using revsets internally, and
1996 1996 escapes arguments appropriately. Aliases are intentionally ignored
1997 1997 so that intended expression behavior isn't accidentally subverted.
1998 1998
1999 1999 Supported arguments:
2000 2000
2001 2001 %r = revset expression, parenthesized
2002 2002 %d = int(arg), no quoting
2003 2003 %s = string(arg), escaped and single-quoted
2004 2004 %b = arg.branch(), escaped and single-quoted
2005 2005 %n = hex(arg), single-quoted
2006 2006 %% = a literal '%'
2007 2007
2008 2008 Prefixing the type with 'l' specifies a parenthesized list of that type.
2009 2009
2010 2010 >>> formatspec('%r:: and %lr', '10 or 11', ("this()", "that()"))
2011 2011 '(10 or 11):: and ((this()) or (that()))'
2012 2012 >>> formatspec('%d:: and not %d::', 10, 20)
2013 2013 '10:: and not 20::'
2014 2014 >>> formatspec('%ld or %ld', [], [1])
2015 2015 "_list('') or 1"
2016 2016 >>> formatspec('keyword(%s)', 'foo\\xe9')
2017 2017 "keyword('foo\\\\xe9')"
2018 2018 >>> b = lambda: 'default'
2019 2019 >>> b.branch = b
2020 2020 >>> formatspec('branch(%b)', b)
2021 2021 "branch('default')"
2022 2022 >>> formatspec('root(%ls)', ['a', 'b', 'c', 'd'])
2023 2023 "root(_list('a\\x00b\\x00c\\x00d'))"
2024 2024 '''
2025 2025
2026 2026 def quote(s):
2027 2027 return repr(str(s))
2028 2028
2029 2029 def argtype(c, arg):
2030 2030 if c == 'd':
2031 2031 return str(int(arg))
2032 2032 elif c == 's':
2033 2033 return quote(arg)
2034 2034 elif c == 'r':
2035 2035 parse(arg) # make sure syntax errors are confined
2036 2036 return '(%s)' % arg
2037 2037 elif c == 'n':
2038 2038 return quote(node.hex(arg))
2039 2039 elif c == 'b':
2040 2040 return quote(arg.branch())
2041 2041
2042 2042 def listexp(s, t):
2043 2043 l = len(s)
2044 2044 if l == 0:
2045 2045 return "_list('')"
2046 2046 elif l == 1:
2047 2047 return argtype(t, s[0])
2048 2048 elif t == 'd':
2049 2049 return "_intlist('%s')" % "\0".join(str(int(a)) for a in s)
2050 2050 elif t == 's':
2051 2051 return "_list('%s')" % "\0".join(s)
2052 2052 elif t == 'n':
2053 2053 return "_hexlist('%s')" % "\0".join(node.hex(a) for a in s)
2054 2054 elif t == 'b':
2055 2055 return "_list('%s')" % "\0".join(a.branch() for a in s)
2056 2056
2057 2057 m = l // 2
2058 2058 return '(%s or %s)' % (listexp(s[:m], t), listexp(s[m:], t))
2059 2059
2060 2060 ret = ''
2061 2061 pos = 0
2062 2062 arg = 0
2063 2063 while pos < len(expr):
2064 2064 c = expr[pos]
2065 2065 if c == '%':
2066 2066 pos += 1
2067 2067 d = expr[pos]
2068 2068 if d == '%':
2069 2069 ret += d
2070 2070 elif d in 'dsnbr':
2071 2071 ret += argtype(d, args[arg])
2072 2072 arg += 1
2073 2073 elif d == 'l':
2074 2074 # a list of some type
2075 2075 pos += 1
2076 2076 d = expr[pos]
2077 2077 ret += listexp(list(args[arg]), d)
2078 2078 arg += 1
2079 2079 else:
2080 2080 raise util.Abort('unexpected revspec format character %s' % d)
2081 2081 else:
2082 2082 ret += c
2083 2083 pos += 1
2084 2084
2085 2085 return ret
2086 2086
2087 2087 def prettyformat(tree):
2088 2088 def _prettyformat(tree, level, lines):
2089 2089 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2090 2090 lines.append((level, str(tree)))
2091 2091 else:
2092 2092 lines.append((level, '(%s' % tree[0]))
2093 2093 for s in tree[1:]:
2094 2094 _prettyformat(s, level + 1, lines)
2095 2095 lines[-1:] = [(lines[-1][0], lines[-1][1] + ')')]
2096 2096
2097 2097 lines = []
2098 2098 _prettyformat(tree, 0, lines)
2099 2099 output = '\n'.join((' '*l + s) for l, s in lines)
2100 2100 return output
2101 2101
2102 2102 def depth(tree):
2103 2103 if isinstance(tree, tuple):
2104 2104 return max(map(depth, tree)) + 1
2105 2105 else:
2106 2106 return 0
2107 2107
2108 2108 def funcsused(tree):
2109 2109 if not isinstance(tree, tuple) or tree[0] in ('string', 'symbol'):
2110 2110 return set()
2111 2111 else:
2112 2112 funcs = set()
2113 2113 for s in tree[1:]:
2114 2114 funcs |= funcsused(s)
2115 2115 if tree[0] == 'func':
2116 2116 funcs.add(tree[1][1])
2117 2117 return funcs
2118 2118
2119 2119 class baseset(list):
2120 2120 """Basic data structure that represents a revset and contains the basic
2121 2121 operation that it should be able to perform.
2122 2122 """
2123 2123 def __init__(self, data):
2124 2124 super(baseset, self).__init__(data)
2125 2125 self._set = None
2126 2126
2127 2127 def set(self):
2128 2128 if not self._set:
2129 2129 self._set = set(self)
2130 2130 return self._set
2131 2131
2132 2132 def __sub__(self, x):
2133 2133 if isinstance(x, baseset):
2134 2134 s = x.set()
2135 2135 else:
2136 2136 s = set(x)
2137 2137 return baseset(self.set() - s)
2138 2138
2139 2139 def __and__(self, x):
2140 2140 if isinstance(x, baseset):
2141 2141 x = x.set()
2142 2142 return baseset([y for y in self if y in x])
2143 2143
2144 2144 def __add__(self, x):
2145 2145 s = self.set()
2146 2146 l = [r for r in x if r not in s]
2147 2147 return baseset(list(self) + l)
2148 2148
2149 2149 def filter(self, l):
2150 2150 return lazyset(self, l)
2151 2151
2152 2152 class lazyset(object):
2153 2153 """Duck type for baseset class which iterates lazily over the revisions in
2154 2154 the subset and contains a function which tests for membership in the
2155 2155 revset
2156 2156 """
2157 2157 def __init__(self, subset, condition=lambda x: True):
2158 2158 self._subset = subset
2159 2159 self._condition = condition
2160 2160 self._cache = {}
2161 2161
2162 2162 def __contains__(self, x):
2163 2163 c = self._cache
2164 2164 if x not in c:
2165 2165 c[x] = x in self._subset and self._condition(x)
2166 2166 return c[x]
2167 2167
2168 2168 def __iter__(self):
2169 2169 cond = self._condition
2170 2170 for x in self._subset:
2171 2171 if cond(x):
2172 2172 yield x
2173 2173
2174 2174 def __and__(self, x):
2175 2175 return lazyset(self, lambda r: r in x)
2176 2176
2177 2177 def __sub__(self, x):
2178 2178 return lazyset(self, lambda r: r not in x)
2179 2179
2180 2180 def __add__(self, x):
2181 2181 def iterates():
2182 2182 for r in self:
2183 2183 yield r
2184 2184 for r in x:
2185 2185 if r not in self:
2186 2186 yield r
2187 2187
2188 2188 return lazyset(generatorset(iterates()))
2189 2189
2190 2190 def __nonzero__(self):
2191 2191 for r in self:
2192 2192 return True
2193 2193 return False
2194 2194
2195 2195 def __len__(self):
2196 2196 # Basic implementation to be changed in future patches.
2197 2197 l = baseset([r for r in self])
2198 2198 return len(l)
2199 2199
2200 2200 def __getitem__(self, x):
2201 2201 # Basic implementation to be changed in future patches.
2202 2202 l = baseset([r for r in self])
2203 2203 return l[x]
2204 2204
2205 2205 def sort(self, reverse=False):
2206 2206 # Basic implementation to be changed in future patches.
2207 2207 self._subset = baseset(self._subset)
2208 2208 self._subset.sort(reverse=reverse)
2209 2209
2210 2210 def reverse(self):
2211 2211 self._subset.reverse()
2212 2212
2213 2213 def set(self):
2214 2214 return set([r for r in self])
2215 2215
2216 2216 def filter(self, l):
2217 2217 return lazyset(self, l)
2218 2218
2219 2219 class orderedlazyset(lazyset):
2220 2220 """Subclass of lazyset which subset can be ordered either ascending or
2221 2221 descendingly
2222 2222 """
2223 2223 def __init__(self, subset, condition, ascending=True):
2224 2224 super(orderedlazyset, self).__init__(subset, condition)
2225 2225 self._ascending = ascending
2226 2226
2227 2227 def filter(self, l):
2228 2228 return orderedlazyset(self, l, ascending=self._ascending)
2229 2229
2230 def __and__(self, x):
2231 return orderedlazyset(self, lambda r: r in x,
2232 ascending=self._ascending)
2233
2234 def __sub__(self, x):
2235 return orderedlazyset(self, lambda r: r not in x,
2236 ascending=self._ascending)
2237
2230 2238 class generatorset(object):
2231 2239 """Wrapper structure for generators that provides lazy membership and can
2232 2240 be iterated more than once.
2233 2241 When asked for membership it generates values until either it finds the
2234 2242 requested one or has gone through all the elements in the generator
2235 2243 """
2236 2244 def __init__(self, gen):
2237 2245 self._gen = gen
2238 2246 self._iter = iter(gen)
2239 2247 self._cache = {}
2240 2248 self._genlist = baseset([])
2241 2249 self._iterated = False
2242 2250
2243 2251 def _nextitem(self):
2244 2252 l = self._iter.next()
2245 2253 self._cache[l] = True
2246 2254 self._genlist.append(l)
2247 2255 return l
2248 2256
2249 2257 def __contains__(self, x):
2250 2258 if x in self._cache:
2251 2259 return self._cache[x]
2252 2260
2253 2261 self._iterated = True
2254 2262 while True:
2255 2263 try:
2256 2264 l = self._nextitem()
2257 2265 if l == x:
2258 2266 return True
2259 2267 except (StopIteration):
2260 2268 break
2261 2269
2262 2270 self._cache[x] = False
2263 2271 return False
2264 2272
2265 2273 def __iter__(self):
2266 2274 if self._iterated:
2267 2275 for l in self._genlist:
2268 2276 yield l
2269 2277 while True:
2270 2278 try:
2271 2279 item = self._nextitem()
2272 2280 yield item
2273 2281 except (StopIteration):
2274 2282 break
2275 2283 else:
2276 2284 self._iterated = True
2277 2285 for item in self._gen:
2278 2286 self._cache[item] = True
2279 2287 self._genlist.append(item)
2280 2288 yield item
2281 2289
2282 2290 def set(self):
2283 2291 return self
2284 2292
2285 2293 class spanset(object):
2286 2294 """Duck type for baseset class which represents a range of revisions and
2287 2295 can work lazily and without having all the range in memory
2288 2296 """
2289 2297 def __init__(self, repo, start=0, end=None):
2290 2298 self._start = start
2291 2299 if end is not None:
2292 2300 self._end = end
2293 2301 else:
2294 2302 self._end = len(repo)
2295 2303 self._hiddenrevs = repo.changelog.filteredrevs
2296 2304
2297 2305 def _contained(self, rev):
2298 2306 return (rev <= self._start and rev > self._end) or (rev >= self._start
2299 2307 and rev < self._end)
2300 2308
2301 2309 def __iter__(self):
2302 2310 if self._start <= self._end:
2303 2311 iterrange = xrange(self._start, self._end)
2304 2312 else:
2305 2313 iterrange = xrange(self._start, self._end, -1)
2306 2314
2307 2315 if self._hiddenrevs:
2308 2316 s = self._hiddenrevs
2309 2317 for r in iterrange:
2310 2318 if r not in s:
2311 2319 yield r
2312 2320 else:
2313 2321 for r in iterrange:
2314 2322 yield r
2315 2323
2316 2324 def __contains__(self, x):
2317 2325 return self._contained(x) and not (self._hiddenrevs and rev in
2318 2326 self._hiddenrevs)
2319 2327
2320 2328 def __and__(self, x):
2321 2329 if isinstance(x, baseset):
2322 2330 x = x.set()
2323 2331 return lazyset(self, lambda r: r in x)
2324 2332
2325 2333 def __sub__(self, x):
2326 2334 if isinstance(x, baseset):
2327 2335 x = x.set()
2328 2336 return lazyset(self, lambda r: r not in x)
2329 2337
2330 2338 def __add__(self, x):
2331 2339 def iterates():
2332 2340 for r in self:
2333 2341 yield r
2334 2342 for r in x:
2335 2343 if r not in self:
2336 2344 yield r
2337 2345
2338 2346 return lazyset(generatorset(iterates()))
2339 2347
2340 2348 def __len__(self):
2341 2349 if not self._hiddenrevs:
2342 2350 return abs(self._end - self._start)
2343 2351 else:
2344 2352 count = 0
2345 2353 for rev in self._hiddenrevs:
2346 2354 if self._contained(rev):
2347 2355 count += 1
2348 2356 return abs(self._end - self._start) - count
2349 2357
2350 2358 def __getitem__(self, x):
2351 2359 # Basic implementation to be changed in future patches.
2352 2360 l = baseset([r for r in self])
2353 2361 return l[x]
2354 2362
2355 2363 def sort(self, reverse=False):
2356 2364 # Basic implementation to be changed in future patches.
2357 2365 if reverse:
2358 2366 self.reverse()
2359 2367
2360 2368 def reverse(self):
2361 2369 if self._start <= self._end:
2362 2370 self._start, self._end = self._end - 1, self._start - 1
2363 2371 else:
2364 2372 self._start, self._end = self._end + 1, self._start + 1
2365 2373
2366 2374 def set(self):
2367 2375 return self
2368 2376
2369 2377 def filter(self, l):
2370 2378 if self._start <= self._end:
2371 2379 return orderedlazyset(self, l)
2372 2380 else:
2373 2381 return orderedlazyset(self, l, ascending=False)
2374 2382
2375 2383 # tell hggettext to extract docstrings from these functions:
2376 2384 i18nfunctions = symbols.values()
General Comments 0
You need to be logged in to leave comments. Login now