mirror of
https://github.com/facebook/sapling.git
synced 2024-10-05 14:28:17 +03:00
44343769f8
Summary: We want to rename away from "mercurial". Rather than rename the "mercurial" Python package, we opted to just collapse it into the parent "edenscm" package. This is also a step towards further organizing we want to do around the new project name. To ease the transition wrt hotfixes, we now replace "edenscm.mercurial" with "mercurial" to fix imports within base64-python extensions. Reviewed By: sggutier Differential Revision: D38943169 fbshipit-source-id: 03fa18079c51e2f7fac05d65b127095da3ab7c99
103 lines
2.7 KiB
Python
Executable File
103 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Portions Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# Copyright 2006, 2007 Olivia Mackall <olivia@selenic.com>
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
"""This does HTTP GET requests given a host:port and path and returns
|
|
a subset of the headers plus the body of the result."""
|
|
|
|
from __future__ import absolute_import, print_function
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
from edenscm import util
|
|
|
|
|
|
httplib = util.httplib
|
|
|
|
try:
|
|
import msvcrt
|
|
|
|
msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
|
|
msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
|
|
except ImportError:
|
|
pass
|
|
|
|
twice = False
|
|
if "--twice" in sys.argv:
|
|
sys.argv.remove("--twice")
|
|
twice = True
|
|
headeronly = False
|
|
if "--headeronly" in sys.argv:
|
|
sys.argv.remove("--headeronly")
|
|
headeronly = True
|
|
formatjson = False
|
|
if "--json" in sys.argv:
|
|
sys.argv.remove("--json")
|
|
formatjson = True
|
|
|
|
hgproto = None
|
|
if "--hgproto" in sys.argv:
|
|
idx = sys.argv.index("--hgproto")
|
|
hgproto = sys.argv[idx + 1]
|
|
sys.argv.pop(idx)
|
|
sys.argv.pop(idx)
|
|
|
|
tag = None
|
|
|
|
|
|
def request(host, path, show):
|
|
assert not path.startswith("/"), path
|
|
global tag
|
|
headers = {}
|
|
if tag:
|
|
headers["If-None-Match"] = tag
|
|
if hgproto:
|
|
headers["X-HgProto-1"] = hgproto
|
|
|
|
conn = httplib.HTTPConnection(host)
|
|
conn.request("GET", "/" + path, None, headers)
|
|
response = conn.getresponse()
|
|
print(response.status, response.reason)
|
|
if show[:1] == ["-"]:
|
|
show = sorted(h for h, v in response.getheaders() if h.lower() not in show)
|
|
for h in [h.lower() for h in show]:
|
|
if response.getheader(h, None) is not None:
|
|
print("%s: %s" % (h, response.getheader(h)))
|
|
if not headeronly:
|
|
print()
|
|
data = response.read()
|
|
|
|
# Pretty print JSON. This also has the beneficial side-effect
|
|
# of verifying emitted JSON is well-formed.
|
|
if formatjson:
|
|
# json.dumps() will print trailing newlines. Eliminate them
|
|
# to make tests easier to write.
|
|
data = json.loads(data)
|
|
lines = json.dumps(data, sort_keys=True, indent=2).splitlines()
|
|
for line in lines:
|
|
print(line.rstrip())
|
|
else:
|
|
sys.stdout.write(data)
|
|
|
|
if twice and response.getheader("ETag", None):
|
|
tag = response.getheader("ETag")
|
|
|
|
return response.status
|
|
|
|
|
|
status = request(sys.argv[1], sys.argv[2], sys.argv[3:])
|
|
if twice:
|
|
status = request(sys.argv[1], sys.argv[2], sys.argv[3:])
|
|
|
|
if 200 <= status <= 305:
|
|
sys.exit(0)
|
|
sys.exit(1)
|