nex/nonsensebot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix homeserver ping

Browse source
This commit is contained in:
Nexus 2024-08-19 19:02:45 +01:00
parent 775be84d26
commit 83e6f7108e
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
app/modules/latency.py
View file

@ -11,23 +11,25 @@ class LatencyModule(niobot.Module):
"""See the bot's latency."""
latency = ctx.latency
homeserver = homeserver or await niobot.resolve_homeserver(ctx.message.sender.split(":")[1])
homeserver = homeserver or ctx.message.sender.split(":")[1]
if not homeserver.startswith("http"):
homeserver = f"https://{homeserver}"
parsed = urllib.parse.urlparse(homeserver)
loc = "https://" + parsed.netloc
homeserver = await niobot.resolve_homeserver(parsed.netloc)
async with httpx.AsyncClient(headers={"User-Agent": niobot.__user_agent__}) as client:
timings = []
for rnd in range(5):
start = time.time()
try:
res = await client.get(f"{loc}/_matrix/client/v1/ping")
res = await client.get(f"{homeserver}/_matrix/client/v1/ping")
res.raise_for_status()
except httpx.HTTPError:
timings.append(-1)
end = time.time()
timings.append(end - start)
else:
end = time.time()
timings.append(end - start)
fed_latency = sum(timings) / len(timings) * 1000
hs_target = parsed.hostname