mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 17:27:53 +03:00
cbdaa23f88
Summary: conduit HTTP POST request did not have correct content-type header set. Now they do. Ideally we should use python-request package to do HTTP but I do not know enough how to import these dependencies so I just went with the minimal changes. Test Plan: ran hg log from dev environment and eventually got it all working. {P57411895} Reviewers: simpkins Reviewed By: simpkins Subscribers: net-systems-diffs@fb.com, mjpieters Differential Revision: https://phabricator.intern.facebook.com/D5080725 Tasks: 18294945 Signature: t1:5080725:1495052201:1bbd6edad4c1c4b3482a7479259460a815947630
137 lines
4.8 KiB
Python
137 lines
4.8 KiB
Python
# conduit.py
|
|
#
|
|
# A library function to call a phabricator conduit RPC.
|
|
# It's different from fbconduit in that this is an authenticated
|
|
# conduit client.
|
|
|
|
import hashlib
|
|
from mercurial.util import httplib
|
|
|
|
import json
|
|
import os
|
|
import time
|
|
from mercurial import util
|
|
import arcconfig
|
|
|
|
urlreq = util.urlreq
|
|
|
|
DEFAULT_HOST = 'https://phabricator.intern.facebook.com/api/'
|
|
DEFAULT_TIMEOUT = 60
|
|
mocked_responses = None
|
|
|
|
class ClientError(Exception):
|
|
def __init__(self, code, msg):
|
|
Exception.__init__(self, msg)
|
|
self.code = code
|
|
|
|
class Client(object):
|
|
def __init__(self, host=None, user=None, cert=None, act_as=None):
|
|
self._host = host or DEFAULT_HOST
|
|
self._user = user
|
|
self._cert = cert
|
|
self._actas = act_as or self._user
|
|
self._connection = None
|
|
|
|
def apply_arcconfig(self, config):
|
|
self._host = config.get('conduit_uri', DEFAULT_HOST)
|
|
try:
|
|
hostconfig = config['hosts'][self._host]
|
|
self._user = hostconfig['user']
|
|
self._cert = hostconfig['cert']
|
|
except KeyError:
|
|
try:
|
|
hostconfig = config['hosts'][config['hosts'].keys()[0]]
|
|
self._user = hostconfig['user']
|
|
self._cert = hostconfig['cert']
|
|
except KeyError:
|
|
raise arcconfig.ArcConfigError(
|
|
'arcrc is missing user '
|
|
'credentials for host %s. use '
|
|
'"arc install-certificate" to fix.' % self._host)
|
|
self._actas = self._user
|
|
self._connection = None
|
|
|
|
def call(self, method, args, timeout=DEFAULT_TIMEOUT):
|
|
token = '%d' % time.time()
|
|
sig = token + self._cert
|
|
args['__conduit__'] = {
|
|
'authUser': self._user,
|
|
'actAsUser': self._actas,
|
|
'authToken': token,
|
|
'authSignature': hashlib.sha1(sig.encode('utf-8')).hexdigest(),
|
|
}
|
|
req_data = util.urlreq.urlencode(
|
|
{
|
|
'params': json.dumps(args),
|
|
'output': 'json',
|
|
}
|
|
)
|
|
urlparts = urlreq.urlparse(self._host)
|
|
# TODO: move to python-requests
|
|
if self._connection is None:
|
|
if urlparts.scheme == 'http':
|
|
self._connection = httplib.HTTPConnection(
|
|
urlparts.netloc, timeout=timeout)
|
|
elif urlparts.scheme == 'https':
|
|
self._connection = httplib.HTTPSConnection(
|
|
urlparts.netloc, timeout=timeout)
|
|
else:
|
|
raise ClientError(
|
|
None, 'Unknown host scheme: %s', urlparts.scheme)
|
|
|
|
headers = {
|
|
'Connection': 'Keep-Alive',
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|
}
|
|
# self._connection.set_debuglevel(1)
|
|
self._connection.request('POST', (urlparts.path + method), req_data,
|
|
headers)
|
|
|
|
response = json.load(self._connection.getresponse())
|
|
if response['error_code'] is not None:
|
|
raise ClientError(response['error_code'], response['error_info'])
|
|
return response['result']
|
|
|
|
class MockClient(object):
|
|
def __init__(self, **kwargs):
|
|
pass
|
|
|
|
def apply_arcconfig(self, config):
|
|
pass
|
|
|
|
def call(self, method, args, timeout=DEFAULT_TIMEOUT):
|
|
global mocked_responses
|
|
|
|
cmd = json.dumps([method, args], sort_keys=True)
|
|
try:
|
|
response = mocked_responses.pop(0)
|
|
# Check expectations via a deep compare of the json representation.
|
|
# We need this because child objects and values are compared by
|
|
# address rather than value.
|
|
expect = json.dumps(response.get('cmd', None), sort_keys=True)
|
|
if cmd != expect:
|
|
raise ClientError(None,
|
|
'mock mismatch got %s expected %s' % (
|
|
cmd, expect))
|
|
if 'error_info' in response:
|
|
raise ClientError(response.get('error_code', None),
|
|
response['error_info'])
|
|
return response['result']
|
|
except IndexError:
|
|
raise ClientError(None,
|
|
'No more mocked responses available for call to %s' % cmd)
|
|
|
|
|
|
if 'HG_ARC_CONDUIT_MOCK' in os.environ:
|
|
# To facilitate testing, we replace the client object with this
|
|
# fake implementation that returns responses from a file that
|
|
# contains a series of json serialized object values.
|
|
with open(os.environ['HG_ARC_CONDUIT_MOCK'], 'r') as f:
|
|
mocked_responses = json.load(f)
|
|
Client = MockClient
|
|
|
|
def call_conduit(method, args, timeout=DEFAULT_TIMEOUT):
|
|
client = Client()
|
|
client.apply_arcconfig(arcconfig.load_for_path(os.getcwd()))
|
|
return client.call(method, args, timeout=timeout)
|