99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
import re
|
|
import logging
|
|
import niobot
|
|
import typing
|
|
|
|
|
|
if typing.TYPE_CHECKING:
|
|
from ..main import NonsenseBot
|
|
|
|
|
|
class MemeteraCounter(niobot.Module):
|
|
ROOM_IDS = {
|
|
"!gJtz8vGvObJTcs8eYT:puppygock.gay",
|
|
"!n8DKU1BeeJilOJXDPr:seirdy.one",
|
|
"!CvZqWrEc2aKtoCoZo4:puppygock.gay"
|
|
}
|
|
USER_IDS = {
|
|
"@june:awawawawawawawawawawawawawawawawawawawawawawawawawawawawawawaw.gay",
|
|
"@june:constellatory.net",
|
|
"@june:girlboss.ceo",
|
|
"@june:itzzen.net",
|
|
"@strawberry:transfem.dev",
|
|
"@strawberry:puppygock.gay",
|
|
"@girlbossceo:matrix.org",
|
|
"@strawberry:sulian.eu"
|
|
}
|
|
WORDS = {
|
|
"memetera": re.compile(r"^memetera$", re.IGNORECASE),
|
|
"comma": re.compile(r","),
|
|
"nyo": re.compile(r"nyo", re.IGNORECASE),
|
|
"bite": re.compile(r"bite", re.IGNORECASE)
|
|
}
|
|
bot: "NonsenseBot"
|
|
log = logging.getLogger(__name__)
|
|
|
|
@niobot.event("message")
|
|
async def on_message(self, room: niobot.MatrixRoom, event: niobot.RoomMessage):
|
|
if not isinstance(event, niobot.RoomMessageFormatted):
|
|
return
|
|
if self.bot.is_old(event):
|
|
self.log.debug("Ignoring old event")
|
|
return
|
|
if room.room_id not in self.ROOM_IDS:
|
|
self.log.debug("Ignoring non-interesting event in %s", room.room_id)
|
|
return
|
|
if event.sender not in self.USER_IDS:
|
|
self.log.debug("Ignoring uninteresting event by %s in %s", event.sender, room.room_id)
|
|
return
|
|
if not self.bot.redis:
|
|
self.log.warning("Redis is not configured!")
|
|
return
|
|
|
|
self.log.debug("Interested in event %r in %r by %r", event.event_id, room.room_id, event.sender)
|
|
|
|
counts = {}
|
|
lower_body = event.body.casefold()
|
|
for word, regex in self.WORDS.items():
|
|
word_count = len(regex.findall(lower_body))
|
|
self.log.debug("%d results for %r", word_count, word)
|
|
if word_count:
|
|
self.log.debug("June said %r %d times in %s!", word, word_count, event.event_id)
|
|
counts.setdefault(word, 0)
|
|
counts[word] += word_count
|
|
|
|
for word, count in counts.items():
|
|
key = "june_wordcount_%s" % word
|
|
self.log.debug("Fetching redis key %r", key)
|
|
stored = await self.bot.redis.get(key)
|
|
stored = int(stored) if stored else 0
|
|
stored += count
|
|
await self.bot.redis.set(key, count)
|
|
self.log.debug("June has now said %r %d times", word, stored)
|
|
|
|
@niobot.command(name="june-word-count", hidden=True)
|
|
async def june_word_count(
|
|
self,
|
|
ctx: niobot.Context,
|
|
raw: bool = False
|
|
):
|
|
keys = {}
|
|
lines = []
|
|
if raw:
|
|
lines.append("```md")
|
|
for key in await self.bot.redis.keys("june_wordcount_*"):
|
|
key = key.decode()
|
|
value = await self.bot.redis.get(key)
|
|
value = int(value)
|
|
if raw:
|
|
lines.append(f"* {key}: {value}")
|
|
else:
|
|
keys[key[15:]] = value
|
|
|
|
for key, count in keys.items():
|
|
lines.append(f"* {key}: {count:,}")
|
|
|
|
text = "\n".join(lines)
|
|
if raw:
|
|
text += "\n```"
|
|
return await ctx.respond(text)
|