blob: 83e3daa4407a24b8f60a7dfc91717dc2daa9d138 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
"""Return system information on our production 9front infrastructure."""
import random
from datetime import datetime
import aiohttp
from discord import Message
from discord.ext.commands import Cog
from loguru import logger
from arthur.apis.systems import lib9front
from arthur.bot import KingArthur
from arthur.config import CONFIG
BLOGCOM = "https://git.9front.org/plan9front/plan9front/HEAD/lib/blogcom/raw"
THRESHOLD = 0.01
MIN_MINUTES = 30
BLOG_ABOUT_IT_THRESHOLD = 1000
class SystemInformation(Cog):
"""Utilities for fetching system information from our 9front infrastructure."""
def __init__(self, bot: KingArthur) -> None:
self.bot = bot
self.cached_data = None
self.last_sent = None
async def fetch_blogcom(self) -> str:
"""Fetch the blogcom file from the upstream location, or return the cached copy."""
if not self.cached_data:
async with aiohttp.ClientSession() as session, session.get(BLOGCOM) as resp:
self.cached_data = await resp.text()
return self.cached_data
@Cog.listener()
async def on_message(self, msg: Message) -> None:
"""Handler for incoming messages, potentially returning system information."""
if not msg.guild:
logger.trace("Ignoring DM")
return
if msg.author.id == self.bot.user.id:
logger.trace("Ignoring own message")
return
if msg.channel.id != CONFIG.devops_channel_id:
logger.trace("Ignoring message outside of DevOps channel")
return
if self.last_sent:
if (datetime.utcnow() - self.last_sent).minutes < MIN_MINUTES:
logger.trace("Ignoring message as within cooldown")
return
msg_thresh = THRESHOLD
if CONFIG.devops_role not in (r.id for r in msg.author.roles):
logger.trace("Upping threshold due to non-DevOps member")
msg_thresh *= 10
if len(msg.content) > BLOG_ABOUT_IT_THRESHOLD:
msg_thresh += 0.10
if random.random() < msg_thresh:
logger.trace("Criteria hit, generating comment.")
blogcom = await self.fetch_blogcom()
comment = lib9front.generate_blog_comment(blogcom).strip()
await msg.reply(f"{comment} :smile:")
self.last_sent = datetime.utcnow()
async def setup(bot: KingArthur) -> None:
"""Add cog to bot."""
await bot.add_cog(SystemInformation(bot))
|
