mirror of
				https://github.com/redmine/redmine.git
				synced 2025-11-03 20:06:24 +01:00 
			
		
		
		
	Patch by Marc Schlaich and Frédéric Fondement. git-svn-id: http://svn.redmine.org/redmine/trunk@17351 e93f8b46-1217-0410-a6f0-8f06a7374b81
		
			
				
	
	
		
			238 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			238 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# 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: 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
 | 
						|
 | 
						|
cmdtable = {}
 | 
						|
command = cmdutil.command(cmdtable)
 | 
						|
 | 
						|
_x = cgi.escape
 | 
						|
_u = lambda s: cgi.escape(urllib.quote(s))
 | 
						|
 | 
						|
def _changectx(repo, rev):
 | 
						|
    if hasattr(repo, 'changectx'):
 | 
						|
        return repo.changectx(rev)
 | 
						|
    else:
 | 
						|
        return repo[rev]
 | 
						|
 | 
						|
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 = _changectx(repo, tiprev())
 | 
						|
    ui.write('<tip revision="%d" node="%s"/>\n'
 | 
						|
             % (tipctx.rev(), _x(node.hex(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.hex(n)), _x(t)))
 | 
						|
 | 
						|
def _branches(ui, repo):
 | 
						|
    # see mercurial/commands.py:branches
 | 
						|
    def iterbranches():
 | 
						|
        if getattr(repo, 'branchtags', None) is not None:
 | 
						|
            # Mercurial < 2.9
 | 
						|
            for t, n in repo.branchtags().iteritems():
 | 
						|
                yield t, n, repo.changelog.rev(n)
 | 
						|
        else:
 | 
						|
            for tag, heads, tip, isclosed in repo.branchmap().iterbranches():
 | 
						|
                yield tag, tip, repo.changelog.rev(tip)
 | 
						|
    def branchheads(branch):
 | 
						|
        try:
 | 
						|
            return repo.branchheads(branch, closed=False)
 | 
						|
        except TypeError:  # Mercurial < 1.2
 | 
						|
            return repo.branchheads(branch)
 | 
						|
    def lookup(rev, n):
 | 
						|
        try:
 | 
						|
            return repo.lookup(rev)
 | 
						|
        except RuntimeError:
 | 
						|
            return n
 | 
						|
    for t, n, r in sorted(iterbranches(), key=lambda e: e[2], reverse=True):
 | 
						|
        if lookup(r, n) in branchheads(t):
 | 
						|
            ui.write('<branch revision="%d" node="%s" name="%s"/>\n'
 | 
						|
                     % (r, _x(node.hex(n)), _x(t)))
 | 
						|
 | 
						|
def _manifest(ui, repo, path, rev):
 | 
						|
    ctx = _changectx(repo, 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.hex(fctx.node())),
 | 
						|
                        tm, fctx.size(), ))
 | 
						|
 | 
						|
    ui.write('</manifest>\n')
 | 
						|
 | 
						|
@command('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...')
 | 
						|
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)
 | 
						|
 | 
						|
@command('rhcat',
 | 
						|
               [('r', 'rev', '', 'revision')],
 | 
						|
               'hg rhcat ([-r REV] ...) FILE...')
 | 
						|
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)
 | 
						|
 | 
						|
@command('rhdiff',
 | 
						|
               [('r', 'rev', [], 'revision'),
 | 
						|
                ('c', 'change', '', 'change made by revision')],
 | 
						|
               'hg rhdiff ([-c REV] | [-r REV] ...) [FILE]...')
 | 
						|
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 = _changectx(repo, change).parents()[0].rev()
 | 
						|
        opts['rev'] = [str(base), change]
 | 
						|
    opts['nodates'] = True
 | 
						|
    return commands.diff(ui, repo, *map(urllib.unquote_plus, pats), **opts)
 | 
						|
 | 
						|
@command('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]')
 | 
						|
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)
 | 
						|
 | 
						|
@command('rhmanifest',
 | 
						|
                   [('r', 'rev', '', 'show the specified revision')],
 | 
						|
                   'hg rhmanifest [-r REV] [PATH]')
 | 
						|
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')
 | 
						|
 | 
						|
@command('rhsummary',[], 'hg rhsummary')
 | 
						|
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')
 | 
						|
 |