diff options
Diffstat (limited to 'bot.py')
| -rw-r--r-- | bot.py | 44 |
1 files changed, 29 insertions, 15 deletions
@@ -1,5 +1,6 @@ #!/usr/bin/env python3 import os +import re # thank you claude import asyncio import json import time @@ -157,6 +158,14 @@ def steal_media(text: str): with MEDIA_FILE.open("a", encoding="utf-8") as f: f.write(text + "\n") +def resolve_mentions(content: str, guild) -> str: # thank you claude + if not guild: + return content + def replace(m): + member = guild.get_member(int(m.group(1))) + return f"@{member.display_name}" if member else m.group(0) + return re.sub(r'<@!?(\d+)>', replace, content) + def is_member_in_server(member: discord.Member) -> bool: guild = client.get_guild(server) return guild and guild.get_member(member.id) is not None @@ -316,23 +325,31 @@ async def on_message(message): if str(message.author.id) in block_id: return + # drop guild messages outside allowed channels before doing anything else, thank you claude + if message.guild and message.channel.name not in ["general", "announcements"]: + return + + content = resolve_mentions(message.content, message.guild) + # always detect replies replied_message = None replied_to_bot = False + replied_content = "" if message.reference: try: replied_message = message.reference.resolved or await message.channel.fetch_message(message.reference.message_id) if replied_message: if replied_message.author == client.user: replied_to_bot = True - log_message(f'[{timestamp()}] {message.author} replied to {replied_message.author} saying: "{replied_message.content}" with reply: "{message.content}"') + replied_content = resolve_mentions(replied_message.content, message.guild) + log_message(f'[{timestamp()}] {message.author} replied to {replied_message.author} saying: "{replied_content}" with reply: "{content}"') else: - log_message(f'[{timestamp()}] {message.author} said: {message.content}') + log_message(f'[{timestamp()}] {message.author} said: {content}') except Exception as e: log_error(f"[{timestamp()}] failed to resolve reply: {e}") - log_message(f'[{timestamp()}] {message.author} said: {message.content}') + log_message(f'[{timestamp()}] {message.author} said: {content}') else: - log_message(f'[{timestamp()}] {message.author} said: {message.content}') + log_message(f'[{timestamp()}] {message.author} said: {content}') interaction_count += 1 rebuild = (interaction_count % 150 == 0) @@ -342,9 +359,9 @@ async def on_message(message): return dm_sent = False if replied_message: - user_input = f'{message.author} replied to {replied_message.author} saying: "{replied_message.content}" with reply: "{message.content}"' + user_input = f'{message.author} replied to {replied_message.author} saying: "{replied_content}" with reply: "{content}"' else: - user_input = f"{message.author} said in DM: {message.content}" + user_input = f"{message.author} said in DM: {content}" payload = build_payload(system_prompt, user_input, rebuild) stop_typing = asyncio.Event() asyncio.create_task(keep_typing(message.channel, stop_typing)) @@ -355,16 +372,13 @@ async def on_message(message): return try: await message.reply(result) - print(f"[{timestamp()}] DM from {message.author}: {message.content} Bot replied in DM: {result}") - log_message(f"[{timestamp()}] DM from {message.author}: {message.content}") + print(f"[{timestamp()}] DM from {message.author}: {content} Bot replied in DM: {result}") + log_message(f"[{timestamp()}] DM from {message.author}: {content}") log_message(f"[{timestamp()}] Faggot-chan replied in DM: {result}") except Exception as e: await handle_error(f"[{timestamp()}] failed to send DM reply: {e}", notify_user=message.author) return - if message.channel.name not in ["general", "announcements"]: - return - mentioned = client.user.mentioned_in(message) # if it replied to bot, treat it as a mention automatically @@ -375,9 +389,9 @@ async def on_message(message): return if replied_message: - user_input = f'{message.author} replied to {replied_message.author} saying: "{replied_message.content}" with reply: "{message.content}"' + user_input = f'{message.author} replied to {replied_message.author} saying: "{replied_content}" with reply: "{content}"' else: - user_input = f"{message.author} said: {message.content}" + user_input = f"{message.author} said: {content}" payload = build_payload(system_prompt, user_input, rebuild) stop_typing = asyncio.Event() @@ -390,8 +404,8 @@ async def on_message(message): try: await message.reply(result) - print(f"[{timestamp()}] {message.author} said: {message.content}\n[{timestamp()}] Faggot-chan said: {result}") - log_message(f"[{timestamp()}] {message.author} said: {message.content}\n[{timestamp()}] Faggot-chan said: {result}") + print(f"[{timestamp()}] {message.author} said: {content}\n[{timestamp()}] Faggot-chan said: {result}") + log_message(f"[{timestamp()}] {message.author} said: {content}\n[{timestamp()}] Faggot-chan said: {result}") except Exception as e: await handle_error(f"[{timestamp()}] failed to send reply: {e}", notify_channel=message.channel) |