comparison contrib/phabricator.py @ 33200:04cf9927f350

phabricator: add phabread command to read patches

This patch adds a `phabread` command generating plain-text patches from
Phabricator, suitable for `hg import`. It respects `hg:meta` so user and
date information might be preserved. And it removes `Summary:` field name
which makes the commit message a bit tidier.

To support stacked diffs, a `--stack` flag was added to read dependent
patches recursively.

33199:228ad1e58a85

# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
"""simple Phabricator integration

This extension provides a ``phabsend`` command which sends a stack of
changesets to Phabricator without amending commit messages.

By default, Phabricator requires ``Test Plan`` which might prevent some
changeset from being sent. The requirement could be disabled by changing
``differential.require-test-plan-field`` config server side.

    token = cli-xxxxxxxxxxxxxxxxxxxxxxxxxxxx

    # Repo callsign. If a repo has a URL https://$HOST/diffusion/FOO, then its
    # callsign is "FOO".
    callsign = FOO
"""

from __future__ import absolute_import

import json

            action = _('skipped')

        ui.write(_('D%s: %s - %s: %s\n') % (newrevid, action, ctx,
                                            ctx.description().split('\n')[0]))
        lastrevid = newrevid

33200:04cf9927f350

# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.
"""simple Phabricator integration

This extension provides a ``phabsend`` command which sends a stack of
changesets to Phabricator without amending commit messages, and a ``phabread``
command which prints a stack of revisions in a format suitable
for :hg:`import`.

By default, Phabricator requires ``Test Plan`` which might prevent some
changeset from being sent. The requirement could be disabled by changing
``differential.require-test-plan-field`` config server side.

    token = cli-xxxxxxxxxxxxxxxxxxxxxxxxxxxx

    # Repo callsign. If a repo has a URL https://$HOST/diffusion/FOO, then its
    # callsign is "FOO".
    callsign = FOO

"""

from __future__ import absolute_import

import json

            action = _('skipped')

        ui.write(_('D%s: %s - %s: %s\n') % (newrevid, action, ctx,
                                            ctx.description().split('\n')[0]))
        lastrevid = newrevid

_summaryre = re.compile('^Summary:\s*', re.M)

def readpatch(repo, params, recursive=False):
    """generate plain-text patch readable by 'hg import'

    params is passed to "differential.query". If recursive is True, also return
    dependent patches.
    """
    # Differential Revisions
    drevs = callconduit(repo, 'differential.query', params)
    if len(drevs) == 1:
        drev = drevs[0]
    else:
        raise error.Abort(_('cannot get Differential Revision %r') % params)

    repo.ui.note(_('reading D%s\n') % drev[r'id'])

    diffid = max(int(v) for v in drev[r'diffs'])
    body = callconduit(repo, 'differential.getrawdiff', {'diffID': diffid})
    desc = callconduit(repo, 'differential.getcommitmessage',
                       {'revision_id': drev[r'id']})
    header = '# HG changeset patch\n'

    # Remove potential empty "Summary:"
    desc = _summaryre.sub('', desc)

    # Try to preserve metadata (user, date) from hg:meta property
    diffs = callconduit(repo, 'differential.querydiffs', {'ids': [diffid]})
    props = diffs[str(diffid)][r'properties'] # could be empty list or dict
    if props and r'hg:meta' in props:
        meta = props[r'hg:meta']
        for k, v in meta.items():
            header += '# %s %s\n' % (k.capitalize(), v)

    patch = ('%s%s\n%s') % (header, desc, body)

    # Check dependencies
    if recursive:
        auxiliary = drev.get(r'auxiliary', {})
        depends = auxiliary.get(r'phabricator:depends-on', [])
        for phid in depends:
            patch = readpatch(repo, {'phids': [phid]}, recursive=True) + patch
    return patch

@command('phabread',
         [('', 'stack', False, _('read dependencies'))],
         _('REVID [OPTIONS]'))
def phabread(ui, repo, revid, **opts):
    """print patches from Phabricator suitable for importing

    REVID could be a Differential Revision identity, like ``D123``, or just the
    number ``123``, or a full URL like ``https://phab.example.com/D123``.

    If --stack is given, follow dependencies information and read all patches.
    """
    try:
        revid = int(revid.split('/')[-1].replace('D', ''))
    except ValueError:
        raise error.Abort(_('invalid Revision ID: %s') % revid)
    patch = readpatch(repo, {'ids': [revid]}, recursive=opts.get('stack'))
    ui.write(patch)