mirror of
https://github.com/darkrenaissance/darkfi.git
synced 2026-01-06 21:34:00 -05:00
135 lines
3.7 KiB
Python
Executable File
135 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# This file is part of DarkFi (https://dark.fi)
|
|
#
|
|
# Copyright (C) 2020-2026 Dyne.org foundation
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import asyncio, json, random, sys, time
|
|
|
|
class JsonRpc:
|
|
|
|
async def start(self, server, port):
|
|
reader, writer = await asyncio.open_connection(server, port)
|
|
self.reader = reader
|
|
self.writer = writer
|
|
|
|
async def stop(self):
|
|
self.writer.close()
|
|
await self.writer.wait_closed()
|
|
|
|
async def _make_request(self, method, params):
|
|
ident = random.randint(0, 2**16)
|
|
#print(ident)
|
|
request = {
|
|
"jsonrpc": "2.0",
|
|
"method": method,
|
|
"params": params,
|
|
"id": ident,
|
|
}
|
|
|
|
message = json.dumps(request) + "\n"
|
|
self.writer.write(message.encode())
|
|
await self.writer.drain()
|
|
|
|
data = await self.reader.readline()
|
|
message = data.decode().strip()
|
|
response = json.loads(message)
|
|
#print(response)
|
|
return response
|
|
|
|
async def _subscribe(self, method, params):
|
|
ident = random.randint(0, 2**16)
|
|
request = {
|
|
"jsonrpc": "2.0",
|
|
"method": method,
|
|
"params": params,
|
|
"id": ident,
|
|
}
|
|
|
|
message = json.dumps(request) + "\n"
|
|
self.writer.write(message.encode())
|
|
await self.writer.drain()
|
|
#print("Subscribed")
|
|
|
|
async def ping(self):
|
|
return await self._make_request("ping", [])
|
|
|
|
async def dnet_switch(self, state):
|
|
return await self._make_request("dnet.switch", [state])
|
|
|
|
async def dnet_subscribe_events(self):
|
|
return await self._subscribe("dnet.subscribe_events", [])
|
|
|
|
async def get_info(self):
|
|
return await self._make_request("p2p.get_info", [])
|
|
|
|
async def main(argv):
|
|
rpc = JsonRpc()
|
|
while True:
|
|
try:
|
|
await rpc.start("localhost", 26660)
|
|
break
|
|
except OSError:
|
|
pass
|
|
response = await rpc._make_request("p2p.get_info", [])
|
|
if "error" in response:
|
|
print("Error: ", response["error"])
|
|
await rpc.stop()
|
|
return
|
|
info = response["result"]
|
|
channels = info["channels"]
|
|
channel_lookup = {}
|
|
for channel in channels:
|
|
id = channel["id"]
|
|
channel_lookup[id] = channel
|
|
|
|
print("inbound:")
|
|
for channel in channels:
|
|
if channel["session"] != "inbound":
|
|
continue
|
|
url = channel["url"]
|
|
print(f" {url}")
|
|
|
|
print("outbound:")
|
|
for i, id in enumerate(info["outbound_slots"]):
|
|
if id == 0:
|
|
print(f" {i}: none")
|
|
continue
|
|
|
|
assert id in channel_lookup
|
|
url = channel_lookup[id]["url"]
|
|
print(f" {i}: {url}")
|
|
|
|
print("seed:")
|
|
for channel in channels:
|
|
if channel["session"] != "seed":
|
|
continue
|
|
url = channel["url"]
|
|
print(f" {url}")
|
|
|
|
print("manual:")
|
|
for channel in channels:
|
|
if channel["session"] != "manual":
|
|
continue
|
|
url = channel["url"]
|
|
print(f" {url}")
|
|
|
|
await rpc.stop()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main(sys.argv))
|
|
|