166 lines
6.3 KiB
Python
166 lines
6.3 KiB
Python
from garfpy import logger
|
|
import discord
|
|
import json
|
|
import os
|
|
import re
|
|
|
|
|
|
class GarfbotRespond:
|
|
def __init__(self):
|
|
self.guild_responses = {}
|
|
self.responses_file = "responses.json"
|
|
|
|
def load_responses(self):
|
|
if os.path.exists(self.responses_file):
|
|
try:
|
|
with open(self.responses_file, "r", encoding="utf-8") as f:
|
|
self.guild_responses = json.load(f)
|
|
self.guild_responses = {
|
|
int(k): v for k, v in self.guild_responses.items()
|
|
}
|
|
total_responses = sum(
|
|
len(responses) for responses in self.guild_responses.values()
|
|
)
|
|
logger.info(
|
|
f"Loaded responses for {len(self.guild_responses)} server(s), ({total_responses} total responses)"
|
|
)
|
|
except Exception as e:
|
|
logger.info(f"Error loading responses: {e}")
|
|
self.guild_responses = {}
|
|
else:
|
|
self.guild_responses = {}
|
|
|
|
def save_responses(self):
|
|
try:
|
|
save_data = {str(k): v for k, v in self.guild_responses.items()}
|
|
with open(self.responses_file, "w", encoding="utf-8") as f:
|
|
json.dump(save_data, f, indent=2, ensure_ascii=False)
|
|
total_responses = sum(
|
|
len(responses) for responses in self.guild_responses.values()
|
|
)
|
|
logger.info(
|
|
f"Saved responses for {len(self.guild_responses)} servers ({total_responses} total responses)"
|
|
)
|
|
except Exception as e:
|
|
logger.info(f"Error saving responses: {e}")
|
|
|
|
def get_responses(self, guild_id):
|
|
if guild_id not in self.guild_responses:
|
|
self.guild_responses[guild_id] = {}
|
|
return self.guild_responses[guild_id]
|
|
|
|
async def garfbot_response(self, message, content):
|
|
guild_id = message.guild.id
|
|
|
|
logger.info(message.content)
|
|
|
|
match = re.search(r'garfbot response add "(.+)" "(.+)"', content, re.IGNORECASE)
|
|
if match:
|
|
trigger = match.group(1)
|
|
response_text = match.group(2)
|
|
await self.add_response(message, guild_id, trigger, response_text)
|
|
return
|
|
|
|
match = re.search(r"garfbot response add (\S+) (.+)", content, re.IGNORECASE)
|
|
if match:
|
|
trigger = match.group(1)
|
|
response_text = match.group(2)
|
|
await self.add_response(message, guild_id, trigger, response_text)
|
|
return
|
|
|
|
match = re.search(r"garfbot\s+response\s+remove\s+(.+)", content, re.IGNORECASE)
|
|
if match:
|
|
trigger = match.group(1).strip()
|
|
await self.remove_response(message, guild_id, trigger)
|
|
return
|
|
|
|
if content.lower() == "garfbot response list":
|
|
await self.list_responses(message, guild_id)
|
|
return
|
|
|
|
await message.channel.send(
|
|
"**Garfbot Auto-Response Commands:**\n"
|
|
'`garfbot response add "trigger" "response"`\n'
|
|
'`garfbot response remove "trigger"`\n'
|
|
"`garfbot response list`\n\n"
|
|
"**Examples:**\n"
|
|
'`garfbot response add "hi" "Hello there!"`\n'
|
|
'`garfbot response add "that\'s what" "That\'s what she said!"`\n'
|
|
'`garfbot response remove "hi"`'
|
|
)
|
|
|
|
async def add_response(self, message, guild_id, trigger, response_text):
|
|
if not response_text or not trigger:
|
|
await message.channel.send("❌ Trigger and response must not be null.")
|
|
return
|
|
|
|
responses = self.get_responses(guild_id)
|
|
responses[trigger] = response_text
|
|
self.guild_responses[guild_id] = responses
|
|
self.save_responses()
|
|
|
|
embed = discord.Embed(title="✅ Auto-response Added.", color=0x00FF00)
|
|
embed.add_field(name="Trigger", value=f"`{trigger}`", inline=True)
|
|
embed.add_field(name="Response", value=f"`{response_text}`", inline=True)
|
|
embed.set_footer(text=f"Server: {message.guild.name}")
|
|
|
|
await message.channel.send(embed=embed)
|
|
|
|
async def remove_response(self, message, guild_id, trigger):
|
|
responses = self.get_responses(guild_id)
|
|
|
|
if trigger in responses:
|
|
removed_response = responses[trigger]
|
|
del responses[trigger]
|
|
self.guild_responses[guild_id] = responses
|
|
self.save_responses()
|
|
|
|
embed = discord.Embed(title="✅ Auto-response Removed.", color=0xFF6B6B)
|
|
embed.add_field(name="Trigger", value=f"`{trigger}`", inline=True)
|
|
embed.add_field(name="Response", value=f"`{removed_response}`", inline=True)
|
|
embed.set_footer(text=f"Server: {message.guild.name}")
|
|
|
|
await message.channel.send(embed=embed)
|
|
return
|
|
|
|
for key in responses.keys():
|
|
if key.lower() == trigger.lower():
|
|
removed_response = responses[key]
|
|
del responses[key]
|
|
self.guild_responses[guild_id] = responses
|
|
self.save_responses()
|
|
|
|
embed = discord.Embed(title="✅ Auto-response Removed.", color=0xFF6B6B)
|
|
embed.add_field(name="Trigger", value=f"`{key}`", inline=True)
|
|
embed.add_field(
|
|
name="Response", value=f"`{removed_response}`", inline=True
|
|
)
|
|
embed.set_footer(text=f"Server: {message.guild.name}")
|
|
|
|
await message.channel.send(embed=embed)
|
|
return
|
|
|
|
await message.channel.send(
|
|
f"❌ No auto-response found for trigger: `{trigger}`"
|
|
)
|
|
|
|
async def list_responses(self, message, guild_id):
|
|
responses = self.get_responses(guild_id)
|
|
|
|
if not responses:
|
|
await message.channel.send("No auto-responses configured for this server.")
|
|
return
|
|
|
|
embed = discord.Embed(
|
|
title=f"Auto-Responses for {message.guild.name}", color=0x3498DB
|
|
)
|
|
|
|
for i, (trigger, response) in enumerate(responses.items(), 1):
|
|
display_response = response[:50] + "..." if len(response) > 50 else response
|
|
embed.add_field(
|
|
name=f"{i}. `{trigger}`", value=f"→ {display_response}", inline=False
|
|
)
|
|
|
|
embed.set_footer(text=f"Total responses: {len(responses)}")
|
|
await message.channel.send(embed=embed)
|