2024-08-19 18:57:10 +01:00
|
|
|
import urllib.parse
|
|
|
|
|
2024-08-03 01:27:44 +01:00
|
|
|
import niobot
|
2024-08-19 18:48:36 +01:00
|
|
|
import time
|
|
|
|
import httpx
|
2024-08-03 01:27:44 +01:00
|
|
|
|
|
|
|
|
|
|
|
class LatencyModule(niobot.Module):
|
|
|
|
@niobot.command("latency")
|
2024-08-19 18:57:10 +01:00
|
|
|
async def latency(self, ctx: niobot.Context, homeserver: str = None):
|
2024-08-03 01:27:44 +01:00
|
|
|
"""See the bot's latency."""
|
|
|
|
latency = ctx.latency
|
2024-08-19 18:48:36 +01:00
|
|
|
|
2024-08-19 18:57:10 +01:00
|
|
|
homeserver = homeserver or await niobot.resolve_homeserver(ctx.message.sender.split(":")[1])
|
|
|
|
if not homeserver.startswith("http"):
|
|
|
|
homeserver = f"https://{homeserver}"
|
|
|
|
|
|
|
|
parsed = urllib.parse.urlparse(homeserver)
|
|
|
|
loc = "https://" + parsed.netloc
|
2024-08-19 18:48:36 +01:00
|
|
|
async with httpx.AsyncClient(headers={"User-Agent": niobot.__user_agent__}) as client:
|
|
|
|
timings = []
|
|
|
|
for rnd in range(5):
|
|
|
|
start = time.time()
|
2024-08-19 19:00:19 +01:00
|
|
|
try:
|
|
|
|
res = await client.get(f"{loc}/_matrix/client/v1/ping")
|
|
|
|
res.raise_for_status()
|
|
|
|
except httpx.HTTPError:
|
|
|
|
timings.append(-1)
|
2024-08-19 18:48:36 +01:00
|
|
|
end = time.time()
|
|
|
|
timings.append(end - start)
|
|
|
|
fed_latency = sum(timings) / len(timings) * 1000
|
|
|
|
|
2024-08-19 18:57:10 +01:00
|
|
|
hs_target = parsed.hostname
|
2024-08-19 18:48:36 +01:00
|
|
|
return await ctx.respond(
|
2024-08-19 18:50:21 +01:00
|
|
|
"Latency: {:,.2f}ms (federation latency to {!r}: {:,.2f}ms)".format(latency, hs_target, fed_latency)
|
2024-08-19 18:48:36 +01:00
|
|
|
)
|