##// END OF EJS Templates
hgweb: pass ui to diffstatgen() explicitly
Yuya Nishihara -
r38604:66eb74f9 default
parent child Browse files
Show More
@@ -1,806 +1,806 b''
1 1 # hgweb/webutil.py - utility library for the web interface.
2 2 #
3 3 # Copyright 21 May 2005 - (c) 2005 Jake Edge <jake@edge2.net>
4 4 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
5 5 #
6 6 # This software may be used and distributed according to the terms of the
7 7 # GNU General Public License version 2 or any later version.
8 8
9 9 from __future__ import absolute_import
10 10
11 11 import copy
12 12 import difflib
13 13 import os
14 14 import re
15 15
16 16 from ..i18n import _
17 17 from ..node import hex, nullid, short
18 18
19 19 from .common import (
20 20 ErrorResponse,
21 21 HTTP_BAD_REQUEST,
22 22 HTTP_NOT_FOUND,
23 23 paritygen,
24 24 )
25 25
26 26 from .. import (
27 27 context,
28 28 error,
29 29 match,
30 30 mdiff,
31 31 obsutil,
32 32 patch,
33 33 pathutil,
34 34 pycompat,
35 35 scmutil,
36 36 templatefilters,
37 37 templatekw,
38 38 templateutil,
39 39 ui as uimod,
40 40 util,
41 41 )
42 42
43 43 from ..utils import (
44 44 diffutil,
45 45 stringutil,
46 46 )
47 47
48 48 archivespecs = util.sortdict((
49 49 ('zip', ('application/zip', 'zip', '.zip', None)),
50 50 ('gz', ('application/x-gzip', 'tgz', '.tar.gz', None)),
51 51 ('bz2', ('application/x-bzip2', 'tbz2', '.tar.bz2', None)),
52 52 ))
53 53
54 54 def archivelist(ui, nodeid, url=None):
55 55 allowed = ui.configlist('web', 'allow-archive', untrusted=True)
56 56 archives = []
57 57
58 58 for typ, spec in archivespecs.iteritems():
59 59 if typ in allowed or ui.configbool('web', 'allow' + typ,
60 60 untrusted=True):
61 61 archives.append({
62 62 'type': typ,
63 63 'extension': spec[2],
64 64 'node': nodeid,
65 65 'url': url,
66 66 })
67 67
68 68 return templateutil.mappinglist(archives)
69 69
70 70 def up(p):
71 71 if p[0:1] != "/":
72 72 p = "/" + p
73 73 if p[-1:] == "/":
74 74 p = p[:-1]
75 75 up = os.path.dirname(p)
76 76 if up == "/":
77 77 return "/"
78 78 return up + "/"
79 79
80 80 def _navseq(step, firststep=None):
81 81 if firststep:
82 82 yield firststep
83 83 if firststep >= 20 and firststep <= 40:
84 84 firststep = 50
85 85 yield firststep
86 86 assert step > 0
87 87 assert firststep > 0
88 88 while step <= firststep:
89 89 step *= 10
90 90 while True:
91 91 yield 1 * step
92 92 yield 3 * step
93 93 step *= 10
94 94
95 95 class revnav(object):
96 96
97 97 def __init__(self, repo):
98 98 """Navigation generation object
99 99
100 100 :repo: repo object we generate nav for
101 101 """
102 102 # used for hex generation
103 103 self._revlog = repo.changelog
104 104
105 105 def __nonzero__(self):
106 106 """return True if any revision to navigate over"""
107 107 return self._first() is not None
108 108
109 109 __bool__ = __nonzero__
110 110
111 111 def _first(self):
112 112 """return the minimum non-filtered changeset or None"""
113 113 try:
114 114 return next(iter(self._revlog))
115 115 except StopIteration:
116 116 return None
117 117
118 118 def hex(self, rev):
119 119 return hex(self._revlog.node(rev))
120 120
121 121 def gen(self, pos, pagelen, limit):
122 122 """computes label and revision id for navigation link
123 123
124 124 :pos: is the revision relative to which we generate navigation.
125 125 :pagelen: the size of each navigation page
126 126 :limit: how far shall we link
127 127
128 128 The return is:
129 129 - a single element mappinglist
130 130 - containing a dictionary with a `before` and `after` key
131 131 - values are dictionaries with `label` and `node` keys
132 132 """
133 133 if not self:
134 134 # empty repo
135 135 return templateutil.mappinglist([
136 136 {'before': templateutil.mappinglist([]),
137 137 'after': templateutil.mappinglist([])},
138 138 ])
139 139
140 140 targets = []
141 141 for f in _navseq(1, pagelen):
142 142 if f > limit:
143 143 break
144 144 targets.append(pos + f)
145 145 targets.append(pos - f)
146 146 targets.sort()
147 147
148 148 first = self._first()
149 149 navbefore = [{'label': '(%i)' % first, 'node': self.hex(first)}]
150 150 navafter = []
151 151 for rev in targets:
152 152 if rev not in self._revlog:
153 153 continue
154 154 if pos < rev < limit:
155 155 navafter.append({'label': '+%d' % abs(rev - pos),
156 156 'node': self.hex(rev)})
157 157 if 0 < rev < pos:
158 158 navbefore.append({'label': '-%d' % abs(rev - pos),
159 159 'node': self.hex(rev)})
160 160
161 161 navafter.append({'label': 'tip', 'node': 'tip'})
162 162
163 163 # TODO: maybe this can be a scalar object supporting tomap()
164 164 return templateutil.mappinglist([
165 165 {'before': templateutil.mappinglist(navbefore),
166 166 'after': templateutil.mappinglist(navafter)},
167 167 ])
168 168
169 169 class filerevnav(revnav):
170 170
171 171 def __init__(self, repo, path):
172 172 """Navigation generation object
173 173
174 174 :repo: repo object we generate nav for
175 175 :path: path of the file we generate nav for
176 176 """
177 177 # used for iteration
178 178 self._changelog = repo.unfiltered().changelog
179 179 # used for hex generation
180 180 self._revlog = repo.file(path)
181 181
182 182 def hex(self, rev):
183 183 return hex(self._changelog.node(self._revlog.linkrev(rev)))
184 184
185 185 # TODO: maybe this can be a wrapper class for changectx/filectx list, which
186 186 # yields {'ctx': ctx}
187 187 def _ctxsgen(context, ctxs):
188 188 for s in ctxs:
189 189 d = {
190 190 'node': s.hex(),
191 191 'rev': s.rev(),
192 192 'user': s.user(),
193 193 'date': s.date(),
194 194 'description': s.description(),
195 195 'branch': s.branch(),
196 196 }
197 197 if util.safehasattr(s, 'path'):
198 198 d['file'] = s.path()
199 199 yield d
200 200
201 201 def _siblings(siblings=None, hiderev=None):
202 202 if siblings is None:
203 203 siblings = []
204 204 siblings = [s for s in siblings if s.node() != nullid]
205 205 if len(siblings) == 1 and siblings[0].rev() == hiderev:
206 206 siblings = []
207 207 return templateutil.mappinggenerator(_ctxsgen, args=(siblings,))
208 208
209 209 def difffeatureopts(req, ui, section):
210 210 diffopts = diffutil.difffeatureopts(ui, untrusted=True,
211 211 section=section, whitespace=True)
212 212
213 213 for k in ('ignorews', 'ignorewsamount', 'ignorewseol', 'ignoreblanklines'):
214 214 v = req.qsparams.get(k)
215 215 if v is not None:
216 216 v = stringutil.parsebool(v)
217 217 setattr(diffopts, k, v if v is not None else True)
218 218
219 219 return diffopts
220 220
221 221 def annotate(req, fctx, ui):
222 222 diffopts = difffeatureopts(req, ui, 'annotate')
223 223 return fctx.annotate(follow=True, diffopts=diffopts)
224 224
225 225 def parents(ctx, hide=None):
226 226 if isinstance(ctx, context.basefilectx):
227 227 introrev = ctx.introrev()
228 228 if ctx.changectx().rev() != introrev:
229 229 return _siblings([ctx.repo()[introrev]], hide)
230 230 return _siblings(ctx.parents(), hide)
231 231
232 232 def children(ctx, hide=None):
233 233 return _siblings(ctx.children(), hide)
234 234
235 235 def renamelink(fctx):
236 236 r = fctx.renamed()
237 237 if r:
238 238 return templateutil.mappinglist([{'file': r[0], 'node': hex(r[1])}])
239 239 return templateutil.mappinglist([])
240 240
241 241 def nodetagsdict(repo, node):
242 242 return templateutil.hybridlist(repo.nodetags(node), name='name')
243 243
244 244 def nodebookmarksdict(repo, node):
245 245 return templateutil.hybridlist(repo.nodebookmarks(node), name='name')
246 246
247 247 def nodebranchdict(repo, ctx):
248 248 branches = []
249 249 branch = ctx.branch()
250 250 # If this is an empty repo, ctx.node() == nullid,
251 251 # ctx.branch() == 'default'.
252 252 try:
253 253 branchnode = repo.branchtip(branch)
254 254 except error.RepoLookupError:
255 255 branchnode = None
256 256 if branchnode == ctx.node():
257 257 branches.append(branch)
258 258 return templateutil.hybridlist(branches, name='name')
259 259
260 260 def nodeinbranch(repo, ctx):
261 261 branches = []
262 262 branch = ctx.branch()
263 263 try:
264 264 branchnode = repo.branchtip(branch)
265 265 except error.RepoLookupError:
266 266 branchnode = None
267 267 if branch != 'default' and branchnode != ctx.node():
268 268 branches.append(branch)
269 269 return templateutil.hybridlist(branches, name='name')
270 270
271 271 def nodebranchnodefault(ctx):
272 272 branches = []
273 273 branch = ctx.branch()
274 274 if branch != 'default':
275 275 branches.append(branch)
276 276 return templateutil.hybridlist(branches, name='name')
277 277
278 278 def _nodenamesgen(context, f, node, name):
279 279 for t in f(node):
280 280 yield {name: t}
281 281
282 282 def showtag(repo, t1, node=nullid):
283 283 args = (repo.nodetags, node, 'tag')
284 284 return templateutil.mappinggenerator(_nodenamesgen, args=args, name=t1)
285 285
286 286 def showbookmark(repo, t1, node=nullid):
287 287 args = (repo.nodebookmarks, node, 'bookmark')
288 288 return templateutil.mappinggenerator(_nodenamesgen, args=args, name=t1)
289 289
290 290 def branchentries(repo, stripecount, limit=0):
291 291 tips = []
292 292 heads = repo.heads()
293 293 parity = paritygen(stripecount)
294 294 sortkey = lambda item: (not item[1], item[0].rev())
295 295
296 296 def entries(context):
297 297 count = 0
298 298 if not tips:
299 299 for tag, hs, tip, closed in repo.branchmap().iterbranches():
300 300 tips.append((repo[tip], closed))
301 301 for ctx, closed in sorted(tips, key=sortkey, reverse=True):
302 302 if limit > 0 and count >= limit:
303 303 return
304 304 count += 1
305 305 if closed:
306 306 status = 'closed'
307 307 elif ctx.node() not in heads:
308 308 status = 'inactive'
309 309 else:
310 310 status = 'open'
311 311 yield {
312 312 'parity': next(parity),
313 313 'branch': ctx.branch(),
314 314 'status': status,
315 315 'node': ctx.hex(),
316 316 'date': ctx.date()
317 317 }
318 318
319 319 return templateutil.mappinggenerator(entries)
320 320
321 321 def cleanpath(repo, path):
322 322 path = path.lstrip('/')
323 323 return pathutil.canonpath(repo.root, '', path)
324 324
325 325 def changectx(repo, req):
326 326 changeid = "tip"
327 327 if 'node' in req.qsparams:
328 328 changeid = req.qsparams['node']
329 329 ipos = changeid.find(':')
330 330 if ipos != -1:
331 331 changeid = changeid[(ipos + 1):]
332 332
333 333 return scmutil.revsymbol(repo, changeid)
334 334
335 335 def basechangectx(repo, req):
336 336 if 'node' in req.qsparams:
337 337 changeid = req.qsparams['node']
338 338 ipos = changeid.find(':')
339 339 if ipos != -1:
340 340 changeid = changeid[:ipos]
341 341 return scmutil.revsymbol(repo, changeid)
342 342
343 343 return None
344 344
345 345 def filectx(repo, req):
346 346 if 'file' not in req.qsparams:
347 347 raise ErrorResponse(HTTP_NOT_FOUND, 'file not given')
348 348 path = cleanpath(repo, req.qsparams['file'])
349 349 if 'node' in req.qsparams:
350 350 changeid = req.qsparams['node']
351 351 elif 'filenode' in req.qsparams:
352 352 changeid = req.qsparams['filenode']
353 353 else:
354 354 raise ErrorResponse(HTTP_NOT_FOUND, 'node or filenode not given')
355 355 try:
356 356 fctx = scmutil.revsymbol(repo, changeid)[path]
357 357 except error.RepoError:
358 358 fctx = repo.filectx(path, fileid=changeid)
359 359
360 360 return fctx
361 361
362 362 def linerange(req):
363 363 linerange = req.qsparams.getall('linerange')
364 364 if not linerange:
365 365 return None
366 366 if len(linerange) > 1:
367 367 raise ErrorResponse(HTTP_BAD_REQUEST,
368 368 'redundant linerange parameter')
369 369 try:
370 370 fromline, toline = map(int, linerange[0].split(':', 1))
371 371 except ValueError:
372 372 raise ErrorResponse(HTTP_BAD_REQUEST,
373 373 'invalid linerange parameter')
374 374 try:
375 375 return util.processlinerange(fromline, toline)
376 376 except error.ParseError as exc:
377 377 raise ErrorResponse(HTTP_BAD_REQUEST, pycompat.bytestr(exc))
378 378
379 379 def formatlinerange(fromline, toline):
380 380 return '%d:%d' % (fromline + 1, toline)
381 381
382 382 def _succsandmarkersgen(context, mapping):
383 383 repo = context.resource(mapping, 'repo')
384 384 itemmappings = templatekw.showsuccsandmarkers(context, mapping)
385 385 for item in itemmappings.tovalue(context, mapping):
386 386 item['successors'] = _siblings(repo[successor]
387 387 for successor in item['successors'])
388 388 yield item
389 389
390 390 def succsandmarkers(context, mapping):
391 391 return templateutil.mappinggenerator(_succsandmarkersgen, args=(mapping,))
392 392
393 393 # teach templater succsandmarkers is switched to (context, mapping) API
394 394 succsandmarkers._requires = {'repo', 'ctx'}
395 395
396 396 def _whyunstablegen(context, mapping):
397 397 repo = context.resource(mapping, 'repo')
398 398 ctx = context.resource(mapping, 'ctx')
399 399
400 400 entries = obsutil.whyunstable(repo, ctx)
401 401 for entry in entries:
402 402 if entry.get('divergentnodes'):
403 403 entry['divergentnodes'] = _siblings(entry['divergentnodes'])
404 404 yield entry
405 405
406 406 def whyunstable(context, mapping):
407 407 return templateutil.mappinggenerator(_whyunstablegen, args=(mapping,))
408 408
409 409 whyunstable._requires = {'repo', 'ctx'}
410 410
411 411 def commonentry(repo, ctx):
412 412 node = ctx.node()
413 413 return {
414 414 # TODO: perhaps ctx.changectx() should be assigned if ctx is a
415 415 # filectx, but I'm not pretty sure if that would always work because
416 416 # fctx.parents() != fctx.changectx.parents() for example.
417 417 'ctx': ctx,
418 418 'rev': ctx.rev(),
419 419 'node': hex(node),
420 420 'author': ctx.user(),
421 421 'desc': ctx.description(),
422 422 'date': ctx.date(),
423 423 'extra': ctx.extra(),
424 424 'phase': ctx.phasestr(),
425 425 'obsolete': ctx.obsolete(),
426 426 'succsandmarkers': succsandmarkers,
427 427 'instabilities': templateutil.hybridlist(ctx.instabilities(),
428 428 name='instability'),
429 429 'whyunstable': whyunstable,
430 430 'branch': nodebranchnodefault(ctx),
431 431 'inbranch': nodeinbranch(repo, ctx),
432 432 'branches': nodebranchdict(repo, ctx),
433 433 'tags': nodetagsdict(repo, node),
434 434 'bookmarks': nodebookmarksdict(repo, node),
435 435 'parent': lambda **x: parents(ctx),
436 436 'child': lambda **x: children(ctx),
437 437 }
438 438
439 439 def changelistentry(web, ctx):
440 440 '''Obtain a dictionary to be used for entries in a changelist.
441 441
442 442 This function is called when producing items for the "entries" list passed
443 443 to the "shortlog" and "changelog" templates.
444 444 '''
445 445 repo = web.repo
446 446 rev = ctx.rev()
447 447 n = ctx.node()
448 448 showtags = showtag(repo, 'changelogtag', n)
449 449 files = listfilediffs(ctx.files(), n, web.maxfiles)
450 450
451 451 entry = commonentry(repo, ctx)
452 452 entry.update(
453 453 allparents=lambda **x: parents(ctx),
454 454 parent=lambda **x: parents(ctx, rev - 1),
455 455 child=lambda **x: children(ctx, rev + 1),
456 456 changelogtag=showtags,
457 457 files=files,
458 458 )
459 459 return entry
460 460
461 461 def changelistentries(web, revs, maxcount, parityfn):
462 462 """Emit up to N records for an iterable of revisions."""
463 463 repo = web.repo
464 464
465 465 count = 0
466 466 for rev in revs:
467 467 if count >= maxcount:
468 468 break
469 469
470 470 count += 1
471 471
472 472 entry = changelistentry(web, repo[rev])
473 473 entry['parity'] = next(parityfn)
474 474
475 475 yield entry
476 476
477 477 def symrevorshortnode(req, ctx):
478 478 if 'node' in req.qsparams:
479 479 return templatefilters.revescape(req.qsparams['node'])
480 480 else:
481 481 return short(ctx.node())
482 482
483 483 def _listfilesgen(context, ctx, stripecount):
484 484 parity = paritygen(stripecount)
485 485 for blockno, f in enumerate(ctx.files()):
486 486 template = 'filenodelink' if f in ctx else 'filenolink'
487 487 yield context.process(template, {
488 488 'node': ctx.hex(),
489 489 'file': f,
490 490 'blockno': blockno + 1,
491 491 'parity': next(parity),
492 492 })
493 493
494 494 def changesetentry(web, ctx):
495 495 '''Obtain a dictionary to be used to render the "changeset" template.'''
496 496
497 497 showtags = showtag(web.repo, 'changesettag', ctx.node())
498 498 showbookmarks = showbookmark(web.repo, 'changesetbookmark', ctx.node())
499 499 showbranch = nodebranchnodefault(ctx)
500 500
501 501 basectx = basechangectx(web.repo, web.req)
502 502 if basectx is None:
503 503 basectx = ctx.p1()
504 504
505 505 style = web.config('web', 'style')
506 506 if 'style' in web.req.qsparams:
507 507 style = web.req.qsparams['style']
508 508
509 509 diff = diffs(web, ctx, basectx, None, style)
510 510
511 511 parity = paritygen(web.stripecount)
512 diffstatsgen = diffstatgen(ctx, basectx)
512 diffstatsgen = diffstatgen(web.repo.ui, ctx, basectx)
513 513 diffstats = diffstat(ctx, diffstatsgen, parity)
514 514
515 515 return dict(
516 516 diff=diff,
517 517 symrev=symrevorshortnode(web.req, ctx),
518 518 basenode=basectx.hex(),
519 519 changesettag=showtags,
520 520 changesetbookmark=showbookmarks,
521 521 changesetbranch=showbranch,
522 522 files=templateutil.mappedgenerator(_listfilesgen,
523 523 args=(ctx, web.stripecount)),
524 524 diffsummary=lambda **x: diffsummary(diffstatsgen),
525 525 diffstat=diffstats,
526 526 archives=web.archivelist(ctx.hex()),
527 527 **pycompat.strkwargs(commonentry(web.repo, ctx)))
528 528
529 529 def _listfilediffsgen(context, files, node, max):
530 530 for f in files[:max]:
531 531 yield context.process('filedifflink', {'node': hex(node), 'file': f})
532 532 if len(files) > max:
533 533 yield context.process('fileellipses', {})
534 534
535 535 def listfilediffs(files, node, max):
536 536 return templateutil.mappedgenerator(_listfilediffsgen,
537 537 args=(files, node, max))
538 538
539 539 def _prettyprintdifflines(context, lines, blockno, lineidprefix):
540 540 for lineno, l in enumerate(lines, 1):
541 541 difflineno = "%d.%d" % (blockno, lineno)
542 542 if l.startswith('+'):
543 543 ltype = "difflineplus"
544 544 elif l.startswith('-'):
545 545 ltype = "difflineminus"
546 546 elif l.startswith('@'):
547 547 ltype = "difflineat"
548 548 else:
549 549 ltype = "diffline"
550 550 yield context.process(ltype, {
551 551 'line': l,
552 552 'lineno': lineno,
553 553 'lineid': lineidprefix + "l%s" % difflineno,
554 554 'linenumber': "% 8s" % difflineno,
555 555 })
556 556
557 557 def _diffsgen(context, repo, ctx, basectx, files, style, stripecount,
558 558 linerange, lineidprefix):
559 559 if files:
560 560 m = match.exact(repo.root, repo.getcwd(), files)
561 561 else:
562 562 m = match.always(repo.root, repo.getcwd())
563 563
564 564 diffopts = patch.diffopts(repo.ui, untrusted=True)
565 565 node1 = basectx.node()
566 566 node2 = ctx.node()
567 567 parity = paritygen(stripecount)
568 568
569 569 diffhunks = patch.diffhunks(repo, node1, node2, m, opts=diffopts)
570 570 for blockno, (fctx1, fctx2, header, hunks) in enumerate(diffhunks, 1):
571 571 if style != 'raw':
572 572 header = header[1:]
573 573 lines = [h + '\n' for h in header]
574 574 for hunkrange, hunklines in hunks:
575 575 if linerange is not None and hunkrange is not None:
576 576 s1, l1, s2, l2 = hunkrange
577 577 if not mdiff.hunkinrange((s2, l2), linerange):
578 578 continue
579 579 lines.extend(hunklines)
580 580 if lines:
581 581 l = templateutil.mappedgenerator(_prettyprintdifflines,
582 582 args=(lines, blockno,
583 583 lineidprefix))
584 584 yield {
585 585 'parity': next(parity),
586 586 'blockno': blockno,
587 587 'lines': l,
588 588 }
589 589
590 590 def diffs(web, ctx, basectx, files, style, linerange=None, lineidprefix=''):
591 591 args = (web.repo, ctx, basectx, files, style, web.stripecount,
592 592 linerange, lineidprefix)
593 593 return templateutil.mappinggenerator(_diffsgen, args=args, name='diffblock')
594 594
595 595 def _compline(type, leftlineno, leftline, rightlineno, rightline):
596 596 lineid = leftlineno and ("l%d" % leftlineno) or ''
597 597 lineid += rightlineno and ("r%d" % rightlineno) or ''
598 598 llno = '%d' % leftlineno if leftlineno else ''
599 599 rlno = '%d' % rightlineno if rightlineno else ''
600 600 return {
601 601 'type': type,
602 602 'lineid': lineid,
603 603 'leftlineno': leftlineno,
604 604 'leftlinenumber': "% 6s" % llno,
605 605 'leftline': leftline or '',
606 606 'rightlineno': rightlineno,
607 607 'rightlinenumber': "% 6s" % rlno,
608 608 'rightline': rightline or '',
609 609 }
610 610
611 611 def _getcompblockgen(context, leftlines, rightlines, opcodes):
612 612 for type, llo, lhi, rlo, rhi in opcodes:
613 613 len1 = lhi - llo
614 614 len2 = rhi - rlo
615 615 count = min(len1, len2)
616 616 for i in xrange(count):
617 617 yield _compline(type=type,
618 618 leftlineno=llo + i + 1,
619 619 leftline=leftlines[llo + i],
620 620 rightlineno=rlo + i + 1,
621 621 rightline=rightlines[rlo + i])
622 622 if len1 > len2:
623 623 for i in xrange(llo + count, lhi):
624 624 yield _compline(type=type,
625 625 leftlineno=i + 1,
626 626 leftline=leftlines[i],
627 627 rightlineno=None,
628 628 rightline=None)
629 629 elif len2 > len1:
630 630 for i in xrange(rlo + count, rhi):
631 631 yield _compline(type=type,
632 632 leftlineno=None,
633 633 leftline=None,
634 634 rightlineno=i + 1,
635 635 rightline=rightlines[i])
636 636
637 637 def _getcompblock(leftlines, rightlines, opcodes):
638 638 args = (leftlines, rightlines, opcodes)
639 639 return templateutil.mappinggenerator(_getcompblockgen, args=args,
640 640 name='comparisonline')
641 641
642 642 def _comparegen(context, contextnum, leftlines, rightlines):
643 643 '''Generator function that provides side-by-side comparison data.'''
644 644 s = difflib.SequenceMatcher(None, leftlines, rightlines)
645 645 if contextnum < 0:
646 646 l = _getcompblock(leftlines, rightlines, s.get_opcodes())
647 647 yield {'lines': l}
648 648 else:
649 649 for oc in s.get_grouped_opcodes(n=contextnum):
650 650 l = _getcompblock(leftlines, rightlines, oc)
651 651 yield {'lines': l}
652 652
653 653 def compare(contextnum, leftlines, rightlines):
654 654 args = (contextnum, leftlines, rightlines)
655 655 return templateutil.mappinggenerator(_comparegen, args=args,
656 656 name='comparisonblock')
657 657
658 def diffstatgen(ctx, basectx):
658 def diffstatgen(ui, ctx, basectx):
659 659 '''Generator function that provides the diffstat data.'''
660 660
661 diffopts = patch.diffopts(ctx._repo.ui, {'noprefix': False})
661 diffopts = patch.diffopts(ui, {'noprefix': False})
662 662 stats = patch.diffstatdata(
663 663 util.iterlines(ctx.diff(basectx, opts=diffopts)))
664 664 maxname, maxtotal, addtotal, removetotal, binary = patch.diffstatsum(stats)
665 665 while True:
666 666 yield stats, maxname, maxtotal, addtotal, removetotal, binary
667 667
668 668 def diffsummary(statgen):
669 669 '''Return a short summary of the diff.'''
670 670
671 671 stats, maxname, maxtotal, addtotal, removetotal, binary = next(statgen)
672 672 return _(' %d files changed, %d insertions(+), %d deletions(-)\n') % (
673 673 len(stats), addtotal, removetotal)
674 674
675 675 def _diffstattmplgen(context, ctx, statgen, parity):
676 676 stats, maxname, maxtotal, addtotal, removetotal, binary = next(statgen)
677 677 files = ctx.files()
678 678
679 679 def pct(i):
680 680 if maxtotal == 0:
681 681 return 0
682 682 return (float(i) / maxtotal) * 100
683 683
684 684 fileno = 0
685 685 for filename, adds, removes, isbinary in stats:
686 686 template = 'diffstatlink' if filename in files else 'diffstatnolink'
687 687 total = adds + removes
688 688 fileno += 1
689 689 yield context.process(template, {
690 690 'node': ctx.hex(),
691 691 'file': filename,
692 692 'fileno': fileno,
693 693 'total': total,
694 694 'addpct': pct(adds),
695 695 'removepct': pct(removes),
696 696 'parity': next(parity),
697 697 })
698 698
699 699 def diffstat(ctx, statgen, parity):
700 700 '''Return a diffstat template for each file in the diff.'''
701 701 args = (ctx, statgen, parity)
702 702 return templateutil.mappedgenerator(_diffstattmplgen, args=args)
703 703
704 704 class sessionvars(templateutil.wrapped):
705 705 def __init__(self, vars, start='?'):
706 706 self._start = start
707 707 self._vars = vars
708 708
709 709 def __getitem__(self, key):
710 710 return self._vars[key]
711 711
712 712 def __setitem__(self, key, value):
713 713 self._vars[key] = value
714 714
715 715 def __copy__(self):
716 716 return sessionvars(copy.copy(self._vars), self._start)
717 717
718 718 def contains(self, context, mapping, item):
719 719 item = templateutil.unwrapvalue(context, mapping, item)
720 720 return item in self._vars
721 721
722 722 def getmember(self, context, mapping, key):
723 723 key = templateutil.unwrapvalue(context, mapping, key)
724 724 return self._vars.get(key)
725 725
726 726 def getmin(self, context, mapping):
727 727 raise error.ParseError(_('not comparable'))
728 728
729 729 def getmax(self, context, mapping):
730 730 raise error.ParseError(_('not comparable'))
731 731
732 732 def filter(self, context, mapping, select):
733 733 # implement if necessary
734 734 raise error.ParseError(_('not filterable'))
735 735
736 736 def itermaps(self, context):
737 737 separator = self._start
738 738 for key, value in sorted(self._vars.iteritems()):
739 739 yield {'name': key,
740 740 'value': pycompat.bytestr(value),
741 741 'separator': separator,
742 742 }
743 743 separator = '&'
744 744
745 745 def join(self, context, mapping, sep):
746 746 # could be '{separator}{name}={value|urlescape}'
747 747 raise error.ParseError(_('not displayable without template'))
748 748
749 749 def show(self, context, mapping):
750 750 return self.join(context, '')
751 751
752 752 def tobool(self, context, mapping):
753 753 return bool(self._vars)
754 754
755 755 def tovalue(self, context, mapping):
756 756 return self._vars
757 757
758 758 class wsgiui(uimod.ui):
759 759 # default termwidth breaks under mod_wsgi
760 760 def termwidth(self):
761 761 return 80
762 762
763 763 def getwebsubs(repo):
764 764 websubtable = []
765 765 websubdefs = repo.ui.configitems('websub')
766 766 # we must maintain interhg backwards compatibility
767 767 websubdefs += repo.ui.configitems('interhg')
768 768 for key, pattern in websubdefs:
769 769 # grab the delimiter from the character after the "s"
770 770 unesc = pattern[1:2]
771 771 delim = stringutil.reescape(unesc)
772 772
773 773 # identify portions of the pattern, taking care to avoid escaped
774 774 # delimiters. the replace format and flags are optional, but
775 775 # delimiters are required.
776 776 match = re.match(
777 777 br'^s%s(.+)(?:(?<=\\\\)|(?<!\\))%s(.*)%s([ilmsux])*$'
778 778 % (delim, delim, delim), pattern)
779 779 if not match:
780 780 repo.ui.warn(_("websub: invalid pattern for %s: %s\n")
781 781 % (key, pattern))
782 782 continue
783 783
784 784 # we need to unescape the delimiter for regexp and format
785 785 delim_re = re.compile(br'(?<!\\)\\%s' % delim)
786 786 regexp = delim_re.sub(unesc, match.group(1))
787 787 format = delim_re.sub(unesc, match.group(2))
788 788
789 789 # the pattern allows for 6 regexp flags, so set them if necessary
790 790 flagin = match.group(3)
791 791 flags = 0
792 792 if flagin:
793 793 for flag in flagin.upper():
794 794 flags |= re.__dict__[flag]
795 795
796 796 try:
797 797 regexp = re.compile(regexp, flags)
798 798 websubtable.append((regexp, format))
799 799 except re.error:
800 800 repo.ui.warn(_("websub: invalid regexp for %s: %s\n")
801 801 % (key, regexp))
802 802 return websubtable
803 803
804 804 def getgraphnode(repo, ctx):
805 805 return (templatekw.getgraphnodecurrent(repo, ctx) +
806 806 templatekw.getgraphnodesymbol(ctx))
General Comments 0
You need to be logged in to leave comments. Login now