mirror of
https://github.com/facebook/sapling.git
synced 2024-10-16 19:57:18 +03:00
dfda82e492
Summary: Mostly empty lines removed and added. A few bugfixes on excessive line splitting. Reviewed By: quark-zju Differential Revision: D8199128 fbshipit-source-id: 90c1616061bfd7cfbba0b75f03f89683340374d5
196 lines
5.3 KiB
Python
Executable File
196 lines
5.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""
|
|
HTTP server for use in conduit tests.
|
|
"""
|
|
|
|
import BaseHTTPServer
|
|
import json
|
|
import signal
|
|
import sys
|
|
import urlparse
|
|
|
|
# no-check-code
|
|
from optparse import OptionParser
|
|
from StringIO import StringIO
|
|
|
|
|
|
try:
|
|
from mercurial.server import runservice
|
|
except ImportError:
|
|
from mercurial.cmdutil import service as runservice
|
|
|
|
known_translations = {}
|
|
next_error_message = []
|
|
|
|
|
|
class RequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
|
|
def handle_request(self, param):
|
|
from_repo = param["from_repo"]
|
|
from_scm = param["from_scm"]
|
|
to_repo = param["to_repo"]
|
|
to_scm = param["to_scm"]
|
|
revs = param["revs"]
|
|
|
|
if next_error_message:
|
|
self.send_response(500, next_error_message[0])
|
|
self.end_headers()
|
|
del next_error_message[0]
|
|
|
|
translations = known_translations.get(
|
|
(from_repo, from_scm, to_repo, to_scm), {}
|
|
)
|
|
translated_revs = {}
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
|
|
f = StringIO()
|
|
f.write("for(;;);")
|
|
|
|
response = {}
|
|
|
|
for rev in revs:
|
|
if rev in translations:
|
|
translated_revs[rev] = translations[rev]
|
|
else:
|
|
translated_revs[rev] = ""
|
|
else:
|
|
response["result"] = translated_revs
|
|
response["error_code"] = None
|
|
response["error_info"] = None
|
|
|
|
f.write(json.dumps(response))
|
|
self.wfile.write(f.getvalue())
|
|
|
|
def do_POST(self):
|
|
content_len = int(self.headers.getheader("content-length", 0))
|
|
data = self.rfile.read(content_len)
|
|
params = urlparse.parse_qs(data)
|
|
if self.path.startswith("/intern/conduit/scmquery.get.mirrored.revs"):
|
|
param = json.loads(params["params"][0])
|
|
self.handle_request(param)
|
|
return
|
|
|
|
self.send_response(500)
|
|
self.end_headers()
|
|
|
|
def get_path_comps(self):
|
|
assert self.path.startswith("/")
|
|
return self.path[1:].split("/")
|
|
|
|
def update(self, cmd, comps):
|
|
(from_repo, from_scm, to_repo, to_scm, from_rev, to_rev) = comps
|
|
key = (from_repo, from_scm, to_repo, to_scm)
|
|
translations = known_translations.setdefault(key, {})
|
|
|
|
if cmd == "PUT":
|
|
translations[from_rev] = to_rev
|
|
self.send_response(201)
|
|
self.end_headers()
|
|
elif cmd == "DELETE":
|
|
translations.pop(from_rev, None)
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
|
|
def do_PUT(self):
|
|
path_comps = self.get_path_comps()
|
|
self.log_message("%s", path_comps)
|
|
if len(path_comps) == 6:
|
|
self.update("PUT", path_comps)
|
|
return
|
|
elif len(path_comps) == 2 and path_comps[0] == "fail_next":
|
|
# This allows tests to ask us to fail the next HTTP request
|
|
next_error_message.append(path_comps[1])
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
return
|
|
|
|
self.send_response(500)
|
|
self.end_headers()
|
|
|
|
def do_DELETE(self):
|
|
path_comps = self.get_path_comps()
|
|
if len(path_comps) == 6:
|
|
self.update("DELETE", path_comps)
|
|
return
|
|
|
|
self.send_response(500)
|
|
self.end_headers()
|
|
|
|
def do_GET(self):
|
|
self.send_response(200)
|
|
self.end_headers()
|
|
self.wfile.write(known_translations)
|
|
|
|
|
|
class simplehttpservice(object):
|
|
def __init__(self, host, port, port_file):
|
|
self.address = (host, port)
|
|
self.port_file = port_file
|
|
|
|
def init(self):
|
|
self.httpd = BaseHTTPServer.HTTPServer(self.address, RequestHandler)
|
|
if self.port_file:
|
|
with open(self.port_file, "w") as f:
|
|
f.write("%d\n" % self.httpd.server_port)
|
|
|
|
def run(self):
|
|
self.httpd.serve_forever()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = OptionParser()
|
|
parser.add_option(
|
|
"-p",
|
|
"--port",
|
|
dest="port",
|
|
type="int",
|
|
default=0,
|
|
help="TCP port to listen on",
|
|
metavar="PORT",
|
|
)
|
|
parser.add_option(
|
|
"--port-file",
|
|
dest="port_file",
|
|
help="file name where the server port should be written",
|
|
)
|
|
parser.add_option(
|
|
"-H",
|
|
"--host",
|
|
dest="host",
|
|
default="localhost",
|
|
help="hostname or IP to listen on",
|
|
metavar="HOST",
|
|
)
|
|
parser.add_option(
|
|
"--pid", dest="pid", help="file name where the PID of the server is stored"
|
|
)
|
|
parser.add_option(
|
|
"-f",
|
|
"--foreground",
|
|
dest="foreground",
|
|
action="store_true",
|
|
help="do not start the HTTP server in the background",
|
|
)
|
|
parser.add_option("--daemon-postexec", action="append")
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
signal.signal(signal.SIGTERM, lambda x, y: sys.exit(0))
|
|
|
|
if options.foreground and options.pid:
|
|
parser.error("options --pid and --foreground are mutually exclusive")
|
|
|
|
opts = {
|
|
"pid_file": options.pid,
|
|
"daemon": not options.foreground,
|
|
"daemon_postexec": options.daemon_postexec,
|
|
}
|
|
service = simplehttpservice(options.host, options.port, options.port_file)
|
|
runservice(
|
|
opts,
|
|
initfn=service.init,
|
|
runfn=service.run,
|
|
runargs=[sys.executable, __file__] + sys.argv[1:],
|
|
)
|