# phabricator.py - simple Phabricator integration

#

# Copyright 2017 Facebook, Inc.

#

# 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 (EXPERIMENTAL)

This extension provides a ``phabsend`` command which sends a stack of

changesets to Phabricator, and a ``phabread`` command which prints a stack of

revisions in a format suitable for :hg:`import`, and a ``phabupdate`` command

to update statuses in batch.

A "phabstatus" view for :hg:`show` is also provided; it displays status

information of Phabricator differentials associated with unfinished

changesets.

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.

Config::

    [phabricator]

    # Phabricator URL

    url = https://phab.example.com/

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

    # callsign is "FOO".

    callsign = FOO

    # curl command to use. If not set (default), use builtin HTTP library to

    # communicate. If set, use the specified curl command. This could be useful

    # if you need to specify advanced options that is not easily supported by

    # the internal library.

    curlcmd = curl --connect-timeout 2 --retry 3 --silent

    [auth]

    example.schemes = https

    example.prefix = phab.example.com

    # API token. Get it from https://$HOST/conduit/login/

    example.phabtoken = cli-xxxxxxxxxxxxxxxxxxxxxxxxxxxx

"""

from __future__ import absolute_import

import base64

import contextlib

import hashlib

import itertools

import json

import mimetypes

import operator

import re

from mercurial.node import bin, nullid

from mercurial.i18n import _

from mercurial.pycompat import getattr

from mercurial.thirdparty import attr

from mercurial import (

    cmdutil,

    context,

    encoding,

    error,

    exthelper,

    graphmod,

    httpconnection as httpconnectionmod,

    localrepo,

    logcmdutil,

    match,

    mdiff,

    obsutil,

    parser,

    patch,

    phases,

    pycompat,

    scmutil,

    smartset,

    tags,

    templatefilters,

    templateutil,

    url as urlmod,

    util,

)

from mercurial.utils import (

    procutil,

    stringutil,

)

from . import show

# Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for

# extensions which SHIP WITH MERCURIAL. Non-mainline extensions should

# be specifying the version(s) of Mercurial they are tested with, or

# leave the attribute unspecified.

testedwith = b'ships-with-hg-core'

eh = exthelper.exthelper()

cmdtable = eh.cmdtable

command = eh.command

configtable = eh.configtable

templatekeyword = eh.templatekeyword

uisetup = eh.finaluisetup

# developer config: phabricator.batchsize

eh.configitem(

    b'phabricator', b'batchsize', default=12,

)

eh.configitem(

    b'phabricator', b'callsign', default=None,

)

eh.configitem(

    b'phabricator', b'curlcmd', default=None,

)

# developer config: phabricator.repophid

eh.configitem(

    b'phabricator', b'repophid', default=None,

)

eh.configitem(

    b'phabricator', b'url', default=None,

)

eh.configitem(

    b'phabsend', b'confirm', default=False,

)

colortable = {

    b'phabricator.action.created': b'green',

    b'phabricator.action.skipped': b'magenta',

    b'phabricator.action.updated': b'magenta',

    b'phabricator.desc': b'',

    b'phabricator.drev': b'bold',

    b'phabricator.node': b'',

    b'phabricator.status.abandoned': b'magenta dim',

    b'phabricator.status.accepted': b'green bold',

    b'phabricator.status.closed': b'green',

    b'phabricator.status.needsreview': b'yellow',

    b'phabricator.status.needsrevision': b'red',

    b'phabricator.status.changesplanned': b'red',

}

_VCR_FLAGS = [

    (

        b'',

        b'test-vcr',

        b'',

        _(

            b'Path to a vcr file. If nonexistent, will record a new vcr transcript'

            b', otherwise will mock all http requests using the specified vcr file.'

            b' (ADVANCED)'

        ),

    ),

]

@eh.wrapfunction(localrepo, "loadhgrc")

def _loadhgrc(orig, ui, wdirvfs, hgvfs, requirements):

    """Load ``.arcconfig`` content into a ui instance on repository open.

    """

    result = False

    arcconfig = {}

    try:

        # json.loads only accepts bytes from 3.6+

        rawparams = encoding.unifromlocal(wdirvfs.read(b".arcconfig"))

        # json.loads only returns unicode strings

        arcconfig = pycompat.rapply(

            lambda x: encoding.unitolocal(x)

            if isinstance(x, pycompat.unicode)

            else x,

            pycompat.json_loads(rawparams),

        )

        result = True

    except ValueError:

        ui.warn(_(b"invalid JSON in %s\n") % wdirvfs.join(b".arcconfig"))

    except IOError:

        pass

    cfg = util.sortdict()

    if b"repository.callsign" in arcconfig:

        cfg[(b"phabricator", b"callsign")] = arcconfig[b"repository.callsign"]

    if b"phabricator.uri" in arcconfig:

        cfg[(b"phabricator", b"url")] = arcconfig[b"phabricator.uri"]

    if cfg:

        ui.applyconfig(cfg, source=wdirvfs.join(b".arcconfig"))

    return orig(ui, wdirvfs, hgvfs, requirements) or result  # Load .hg/hgrc

def vcrcommand(name, flags, spec, helpcategory=None, optionalrepo=False):

    fullflags = flags + _VCR_FLAGS

    def hgmatcher(r1, r2):

        if r1.uri != r2.uri or r1.method != r2.method:

            return False

        r1params = util.urlreq.parseqs(r1.body)

        r2params = util.urlreq.parseqs(r2.body)

        for key in r1params:

            if key not in r2params:

                return False

            value = r1params[key][0]

            # we want to compare json payloads without worrying about ordering

            if value.startswith(b'{') and value.endswith(b'}'):

                r1json = pycompat.json_loads(value)

                r2json = pycompat.json_loads(r2params[key][0])

                if r1json != r2json:

                    return False

            elif r2params[key][0] != value:

                return False

        return True

    def sanitiserequest(request):

        request.body = re.sub(

            br'cli-[a-z0-9]+', br'cli-hahayouwish', request.body

        )

        return request

    def sanitiseresponse(response):

        if 'set-cookie' in response['headers']:

            del response['headers']['set-cookie']

        return response

    def decorate(fn):

        def inner(*args, **kwargs):

            cassette = pycompat.fsdecode(kwargs.pop('test_vcr', None))

            if cassette:

                import hgdemandimport

                with hgdemandimport.deactivated():

                    import vcr as vcrmod

                    import vcr.stubs as stubs

                    vcr = vcrmod.VCR(

                        serializer='json',

                        before_record_request=sanitiserequest,

                        before_record_response=sanitiseresponse,

                        custom_patches=[

                            (

                                urlmod,

                                'httpconnection',

                                stubs.VCRHTTPConnection,

                            ),

                            (

                                urlmod,

                                'httpsconnection',

                                stubs.VCRHTTPSConnection,

                            ),

                        ],

                    )

                    vcr.register_matcher('hgmatcher', hgmatcher)

                    with vcr.use_cassette(cassette, match_on=['hgmatcher']):

                        return fn(*args, **kwargs)

            return fn(*args, **kwargs)

        inner.__name__ = fn.__name__

        inner.__doc__ = fn.__doc__

        return command(

            name,

            fullflags,

            spec,

            helpcategory=helpcategory,

            optionalrepo=optionalrepo,

        )(inner)

    return decorate

def urlencodenested(params):

    """like urlencode, but works with nested parameters.

    For example, if params is {'a': ['b', 'c'], 'd': {'e': 'f'}}, it will be

    flattened to {'a[0]': 'b', 'a[1]': 'c', 'd[e]': 'f'} and then passed to

    urlencode. Note: the encoding is consistent with PHP's http_build_query.

    """

    flatparams = util.sortdict()

    def process(prefix, obj):

        if isinstance(obj, bool):

            obj = {True: b'true', False: b'false'}[obj]  # Python -> PHP form

        lister = lambda l: [(b'%d' % k, v) for k, v in enumerate(l)]

        items = {list: lister, dict: lambda x: x.items()}.get(type(obj))

        if items is None:

            flatparams[prefix] = obj

        else:

            for k, v in items(obj):

                if prefix:

                    process(b'%s[%s]' % (prefix, k), v)

                else:

                    process(k, v)

    process(b'', params)

    return util.urlreq.urlencode(flatparams)

def readurltoken(ui):

    """return conduit url, token and make sure they exist

    Currently read from [auth] config section. In the future, it might

    make sense to read from .arcconfig and .arcrc as well.

    """

    url = ui.config(b'phabricator', b'url')

    if not url:

        raise error.Abort(

            _(b'config %s.%s is required') % (b'phabricator', b'url')

        )

    res = httpconnectionmod.readauthforuri(ui, url, util.url(url).user)

    token = None

    if res:

        group, auth = res

        ui.debug(b"using auth.%s.* for authentication\n" % group)

        token = auth.get(b'phabtoken')

    if not token:

        raise error.Abort(

            _(b'Can\'t find conduit token associated to %s') % (url,)

        )

    return url, token

def callconduit(ui, name, params):

    """call Conduit API, params is a dict. return json.loads result, or None"""

    host, token = readurltoken(ui)

    url, authinfo = util.url(b'/'.join([host, b'api', name])).authinfo()

    ui.debug(b'Conduit Call: %s %s\n' % (url, pycompat.byterepr(params)))

    params = params.copy()

    params[b'__conduit__'] = {

        b'token': token,

    }

    rawdata = {

        b'params': templatefilters.json(params),

        b'output': b'json',

        b'__conduit__': 1,

    }

    data = urlencodenested(rawdata)

    curlcmd = ui.config(b'phabricator', b'curlcmd')

    if curlcmd:

        sin, sout = procutil.popen2(

            b'%s -d @- %s' % (curlcmd, procutil.shellquote(url))

        )

        sin.write(data)

        sin.close()

        body = sout.read()

    else:

        urlopener = urlmod.opener(ui, authinfo)

        request = util.urlreq.request(pycompat.strurl(url), data=data)

        with contextlib.closing(urlopener.open(request)) as rsp:

            body = rsp.read()

    ui.debug(b'Conduit Response: %s\n' % body)

    parsed = pycompat.rapply(

        lambda x: encoding.unitolocal(x)

        if isinstance(x, pycompat.unicode)

        else x,

        # json.loads only accepts bytes from py3.6+

        pycompat.json_loads(encoding.unifromlocal(body)),

    )

    if parsed.get(b'error_code'):

        msg = _(b'Conduit Error (%s): %s') % (

            parsed[b'error_code'],

            parsed[b'error_info'],

        )

        raise error.Abort(msg)

    return parsed[b'result']

@vcrcommand(b'debugcallconduit', [], _(b'METHOD'), optionalrepo=True)

def debugcallconduit(ui, repo, name):

    """call Conduit API

    Call parameters are read from stdin as a JSON blob. Result will be written

    to stdout as a JSON blob.

    """

    # json.loads only accepts bytes from 3.6+

    rawparams = encoding.unifromlocal(ui.fin.read())

    # json.loads only returns unicode strings

    params = pycompat.rapply(

        lambda x: encoding.unitolocal(x)

        if isinstance(x, pycompat.unicode)

        else x,