/lib/redmine/scm/adapters/mercurial/redminehelper.py
Python | 222 lines | 200 code | 0 blank | 22 comment | 0 complexity | b1f9ccea6b8fa7f2963e1389f3982872 MD5 | raw file
- # redminehelper: Redmine helper extension for Mercurial
- #
- # Copyright 2010 Alessio Franceschelli (alefranz.net)
- # Copyright 2010-2011 Yuya Nishihara <yuya@tcha.org>
- #
- # This software may be used and distributed according to the terms of the
- # GNU General Public License version 2 or any later version.
- """helper commands for Redmine to reduce the number of hg calls
- To test this extension, please try::
- $ hg --config extensions.redminehelper=redminehelper.py rhsummary
- I/O encoding:
- :file path: urlencoded, raw string
- :tag name: utf-8
- :branch name: utf-8
- :node: 12-digits (short) hex string
- Output example of rhsummary::
- <?xml version="1.0"?>
- <rhsummary>
- <repository root="/foo/bar">
- <tip revision="1234" node="abcdef0123..."/>
- <tag revision="123" node="34567abc..." name="1.1.1"/>
- <branch .../>
- ...
- </repository>
- </rhsummary>
- Output example of rhmanifest::
- <?xml version="1.0"?>
- <rhmanifest>
- <repository root="/foo/bar">
- <manifest revision="1234" path="lib">
- <file name="diff.rb" revision="123" node="34567abc..." time="12345"
- size="100"/>
- ...
- <dir name="redmine"/>
- ...
- </manifest>
- </repository>
- </rhmanifest>
- """
- import re, time, cgi, urllib
- from mercurial import cmdutil, commands, node, error, hg
- _x = cgi.escape
- _u = lambda s: cgi.escape(urllib.quote(s))
- def _tip(ui, repo):
- # see mercurial/commands.py:tip
- def tiprev():
- try:
- return len(repo) - 1
- except TypeError: # Mercurial < 1.1
- return repo.changelog.count() - 1
- tipctx = repo.changectx(tiprev())
- ui.write('<tip revision="%d" node="%s"/>\n'
- % (tipctx.rev(), _x(node.short(tipctx.node()))))
- _SPECIAL_TAGS = ('tip',)
- def _tags(ui, repo):
- # see mercurial/commands.py:tags
- for t, n in reversed(repo.tagslist()):
- if t in _SPECIAL_TAGS:
- continue
- try:
- r = repo.changelog.rev(n)
- except error.LookupError:
- continue
- ui.write('<tag revision="%d" node="%s" name="%s"/>\n'
- % (r, _x(node.short(n)), _x(t)))
- def _branches(ui, repo):
- # see mercurial/commands.py:branches
- def iterbranches():
- for t, n in repo.branchtags().iteritems():
- yield t, n, repo.changelog.rev(n)
- def branchheads(branch):
- try:
- return repo.branchheads(branch, closed=False)
- except TypeError: # Mercurial < 1.2
- return repo.branchheads(branch)
- for t, n, r in sorted(iterbranches(), key=lambda e: e[2], reverse=True):
- if repo.lookup(r) in branchheads(t):
- ui.write('<branch revision="%d" node="%s" name="%s"/>\n'
- % (r, _x(node.short(n)), _x(t)))
- def _manifest(ui, repo, path, rev):
- ctx = repo.changectx(rev)
- ui.write('<manifest revision="%d" path="%s">\n'
- % (ctx.rev(), _u(path)))
- known = set()
- pathprefix = (path.rstrip('/') + '/').lstrip('/')
- for f, n in sorted(ctx.manifest().iteritems(), key=lambda e: e[0]):
- if not f.startswith(pathprefix):
- continue
- name = re.sub(r'/.*', '/', f[len(pathprefix):])
- if name in known:
- continue
- known.add(name)
- if name.endswith('/'):
- ui.write('<dir name="%s"/>\n'
- % _x(urllib.quote(name[:-1])))
- else:
- fctx = repo.filectx(f, fileid=n)
- tm, tzoffset = fctx.date()
- ui.write('<file name="%s" revision="%d" node="%s" '
- 'time="%d" size="%d"/>\n'
- % (_u(name), fctx.rev(), _x(node.short(fctx.node())),
- tm, fctx.size(), ))
- ui.write('</manifest>\n')
- def rhannotate(ui, repo, *pats, **opts):
- rev = urllib.unquote_plus(opts.pop('rev', None))
- opts['rev'] = rev
- return commands.annotate(ui, repo, *map(urllib.unquote_plus, pats), **opts)
- def rhcat(ui, repo, file1, *pats, **opts):
- rev = urllib.unquote_plus(opts.pop('rev', None))
- opts['rev'] = rev
- return commands.cat(ui, repo, urllib.unquote_plus(file1), *map(urllib.unquote_plus, pats), **opts)
- def rhdiff(ui, repo, *pats, **opts):
- """diff repository (or selected files)"""
- change = opts.pop('change', None)
- if change: # add -c option for Mercurial<1.1
- base = repo.changectx(change).parents()[0].rev()
- opts['rev'] = [str(base), change]
- opts['nodates'] = True
- return commands.diff(ui, repo, *map(urllib.unquote_plus, pats), **opts)
- def rhlog(ui, repo, *pats, **opts):
- rev = opts.pop('rev')
- bra0 = opts.pop('branch')
- from_rev = urllib.unquote_plus(opts.pop('from', None))
- to_rev = urllib.unquote_plus(opts.pop('to' , None))
- bra = urllib.unquote_plus(opts.pop('rhbranch', None))
- from_rev = from_rev.replace('"', '\\"')
- to_rev = to_rev.replace('"', '\\"')
- if hg.util.version() >= '1.6':
- opts['rev'] = ['"%s":"%s"' % (from_rev, to_rev)]
- else:
- opts['rev'] = ['%s:%s' % (from_rev, to_rev)]
- opts['branch'] = [bra]
- return commands.log(ui, repo, *map(urllib.unquote_plus, pats), **opts)
- def rhmanifest(ui, repo, path='', **opts):
- """output the sub-manifest of the specified directory"""
- ui.write('<?xml version="1.0"?>\n')
- ui.write('<rhmanifest>\n')
- ui.write('<repository root="%s">\n' % _u(repo.root))
- try:
- _manifest(ui, repo, urllib.unquote_plus(path), urllib.unquote_plus(opts.get('rev')))
- finally:
- ui.write('</repository>\n')
- ui.write('</rhmanifest>\n')
- def rhsummary(ui, repo, **opts):
- """output the summary of the repository"""
- ui.write('<?xml version="1.0"?>\n')
- ui.write('<rhsummary>\n')
- ui.write('<repository root="%s">\n' % _u(repo.root))
- try:
- _tip(ui, repo)
- _tags(ui, repo)
- _branches(ui, repo)
- # TODO: bookmarks in core (Mercurial>=1.8)
- finally:
- ui.write('</repository>\n')
- ui.write('</rhsummary>\n')
- # This extension should be compatible with Mercurial 0.9.5.
- # Note that Mercurial 0.9.5 doesn't have extensions.wrapfunction().
- cmdtable = {
- 'rhannotate': (rhannotate,
- [('r', 'rev', '', 'revision'),
- ('u', 'user', None, 'list the author (long with -v)'),
- ('n', 'number', None, 'list the revision number (default)'),
- ('c', 'changeset', None, 'list the changeset'),
- ],
- 'hg rhannotate [-r REV] [-u] [-n] [-c] FILE...'),
- 'rhcat': (rhcat,
- [('r', 'rev', '', 'revision')],
- 'hg rhcat ([-r REV] ...) FILE...'),
- 'rhdiff': (rhdiff,
- [('r', 'rev', [], 'revision'),
- ('c', 'change', '', 'change made by revision')],
- 'hg rhdiff ([-c REV] | [-r REV] ...) [FILE]...'),
- 'rhlog': (rhlog,
- [
- ('r', 'rev', [], 'show the specified revision'),
- ('b', 'branch', [],
- 'show changesets within the given named branch'),
- ('l', 'limit', '',
- 'limit number of changes displayed'),
- ('d', 'date', '',
- 'show revisions matching date spec'),
- ('u', 'user', [],
- 'revisions committed by user'),
- ('', 'from', '',
- ''),
- ('', 'to', '',
- ''),
- ('', 'rhbranch', '',
- ''),
- ('', 'template', '',
- 'display with template')],
- 'hg rhlog [OPTION]... [FILE]'),
- 'rhmanifest': (rhmanifest,
- [('r', 'rev', '', 'show the specified revision')],
- 'hg rhmanifest [-r REV] [PATH]'),
- 'rhsummary': (rhsummary, [], 'hg rhsummary'),
- }