mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 17:58:27 +03:00
af1b898311
Summary: Add a debug option for HTTP data fetching. The intended usage is for it to gate verbose debug messages; the option can be set by Chef for users in the hg_dev tier. Differential Revision: D15040988 fbshipit-source-id: b7eaa3bab4200e083cffc5822fb9873611725e6b
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
# Copyright 2018 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.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from edenscm.mercurial import error, httpconnection
|
|
from edenscm.mercurial.i18n import _
|
|
from edenscm.mercurial.rust.bindings import edenapi
|
|
|
|
from . import shallowutil
|
|
|
|
|
|
def enabled(ui):
|
|
"""Check whether HTTP data fetching is enabled."""
|
|
return ui.configbool("edenapi", "enabled")
|
|
|
|
|
|
def debug(ui):
|
|
"""Check whether HTTP data fetching is in debug mode."""
|
|
return ui.configbool("edenapi", "debug")
|
|
|
|
|
|
def bailifdisabled(ui):
|
|
"""Abort if HTTP data fetching is disabled."""
|
|
if not enabled(ui):
|
|
raise error.Abort(_("HTTP data fetching is disabled"))
|
|
|
|
|
|
def getbaseurl(ui):
|
|
"""Get the base URL of the API server."""
|
|
url = ui.config("edenapi", "url")
|
|
if url is None:
|
|
raise error.Abort(_("No Eden API base URL configured"))
|
|
return url
|
|
|
|
|
|
def getcreds(ui, url):
|
|
"""Get the TLS mutual authentication credentials for the given URL."""
|
|
res = httpconnection.readauthforuri(ui, url, None)
|
|
if res is None:
|
|
return None
|
|
group, auth = res
|
|
if "cert" not in auth or "key" not in auth:
|
|
return None
|
|
return (auth["cert"], auth["key"])
|
|
|
|
|
|
def initclient(ui, repo):
|
|
"""Initialize a new Eden API client using the user's config."""
|
|
url = getbaseurl(ui)
|
|
kwargs = {
|
|
"url": url,
|
|
"cachepath": shallowutil.getcachepath(ui),
|
|
"repo": repo.name,
|
|
"backend": ui.config("edenapi", "backend"),
|
|
"creds": getcreds(ui, url),
|
|
"databatchsize": ui.config("edenapi", "databatchsize"),
|
|
"historybatchsize": ui.config("edenapi", "historybatchsize"),
|
|
"validatefiles": ui.configbool("edenapi", "validatefiles"),
|
|
}
|
|
return edenapi.client(**kwargs)
|