mirror of
https://github.com/XRPLF/xrpl-dev-portal.git
synced 2025-11-20 19:55:54 +00:00
90 lines
2.9 KiB
Python
Executable File
90 lines
2.9 KiB
Python
Executable File
#!/bin/env python
|
|
|
|
"""
|
|
Tool to parse a markdown file for WebSocket examples,
|
|
reformat those examples as JSON-RPC, and test them against
|
|
a public WebSocket server.
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
import re, json, sys, warnings, argparse
|
|
|
|
if sys.version_info[:2] <= (2,7):
|
|
import httplib
|
|
#gotta define this
|
|
class FileNotFoundError(IOError):
|
|
pass
|
|
else:
|
|
import http.client as httplib
|
|
|
|
RIPPLED_RPC_HOST = "s1.ripple.com"
|
|
RIPPLED_RPC_PORT = 51234
|
|
|
|
def ws2rpc(s):
|
|
ws_j = json.loads(s)
|
|
|
|
if "command" not in ws_j:
|
|
#apparently not a WebSocket request after all?
|
|
warnings.warn("Incorrect match? "+s)
|
|
return ""
|
|
|
|
rpc_j = {"params":[{}]}
|
|
for key,val in ws_j.items():
|
|
if key == "id":
|
|
continue
|
|
elif key == "command":
|
|
rpc_j["method"] = val
|
|
else:
|
|
rpc_j["params"][0][key] = val
|
|
|
|
return json.dumps(rpc_j, sort_keys=True, indent=4, separators=(',', ': '))
|
|
|
|
def do_JSONRPC(req, host, port):
|
|
conn = httplib.HTTPConnection(host, port)
|
|
conn.request("POST", "/", js)
|
|
response = conn.getresponse()
|
|
|
|
s = response.read()
|
|
header = "%s %s" % (response.status, response.reason)
|
|
return (header, s)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Convert WebSocket examples to JSON-RPC with responses from the live rippled network")
|
|
parser.add_argument("inputfile")
|
|
parser.add_argument("--offline", action="store_true", help="don't connect to the network to generate responses")
|
|
parser.add_argument("--rippled_host", help="hostname of a rippled server to use", type=str, default=RIPPLED_RPC_HOST)
|
|
parser.add_argument("--rippled_port", help="port number of a rippled server to use", type=int, default=RIPPLED_RPC_PORT)
|
|
|
|
args = parser.parse_args()
|
|
|
|
WS_REGEX = r"\*WebSocket\*\s*```\s*(?P<json>\{[^`]*\})\s*```"
|
|
|
|
try:
|
|
f = open(args.inputfile, 'r')
|
|
matches = re.findall(WS_REGEX, f.read())
|
|
f.close()
|
|
except (FileNotFoundError, IOError, OSError):
|
|
exit("Couldn't read file %s" % args.inputfile)
|
|
|
|
for s in matches:
|
|
js = ws2rpc(s)
|
|
if js:
|
|
print("Request:\n*JSON-RPC*\n\n```\n"+js+"\n```\n\n")
|
|
|
|
if args.offline:
|
|
continue
|
|
|
|
header, response_body = do_JSONRPC(js, args.rippled_host, args.rippled_port)
|
|
|
|
try:
|
|
#parse & pretty-print response JSON if valid
|
|
response_json = json.loads(response_body.decode("utf-8"))
|
|
response_body = json.dumps(response_json, sort_keys=True, indent=4, separators=(',', ': '))
|
|
except ValueError:
|
|
#invalid JSON; leave response as-is
|
|
pass
|
|
|
|
print("Response:\n*JSON-RPC*\n\n```\n%s\n%s\n```\n\n" % (
|
|
header, response_body) )
|