add auto responses
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 16s

This commit is contained in:
crate 2025-06-03 20:33:55 -05:00
parent 9d5765c492
commit b4ca6b778d
4 changed files with 195 additions and 5 deletions

3
.gitignore vendored
View File

@ -6,3 +6,6 @@ garfpy/__pycache__/
*.old
*.log*
meows.py
meow_counts.json
user_stats.json
responses.json

View File

@ -1,3 +1,5 @@
import os
import json
import config
import asyncio
import discord
@ -7,7 +9,7 @@ from garfpy import(
logger, is_private,
kroger_token, find_store, search_product,
garfpic, process_image_requests, generate_chat,
aod_message, wikisum, generate_qr)
aod_message, wikisum, generate_qr, GarfbotRespond)
gapikey = config.GIF_TOKEN
@ -21,16 +23,17 @@ intents.messages = True
intents.message_content = True
garfbot = discord.Client(intents=intents)
garf_respond = GarfbotRespond
@garfbot.event
async def on_ready():
try:
asyncio.create_task(process_image_requests())
garf_respond.load_responses()
logger.info(f"Logged in as {garfbot.user.name} running {txtmodel} and {imgmodel}.")
except Exception as e:
logger.error(e)
@garfbot.event
async def on_message(message):
if message.author == garfbot.user:
@ -60,7 +63,7 @@ async def on_message(message):
if message.content.lower().startswith('garfqr '):
text = message.content[7:]
if len(text) > 1000:
await mesage.channel.send("❌ Text too long! Maximum 1000 characters.")
await message.channel.send("❌ Text too long! Maximum 1000 characters.")
else:
try:
qr_code = await generate_qr(text)
@ -143,9 +146,23 @@ async def on_message(message):
# Army of Dawn Server only!!
if message.guild and message.guild.id == 719605634772893757:
await aod_message(garfbot, message)
# Auto-responses
guild_id = message.guild.id
content = message.content.strip()
content_lower = content.lower()
responses = garf_respond.get_responses(guild_id)
if content.lower().startswith('garfbot response'):
await garf_respond.garfbot_response(message, content)
return
for trigger, response in responses.items():
if trigger.lower() in content_lower:
await message.channel.send(response)
break
async def garfbot_connect():
while True:

View File

@ -1,4 +1,5 @@
# garfpy/__init__.py
from .log import logger
from .kroger import(
kroger_token, find_store, search_product
@ -11,4 +12,5 @@ from .garfai import(
from .iputils import is_private
from .aod import aod_message
from .wiki import wikisum
from .qr import generate_qr
from .qr import generate_qr
from .respond import GarfbotRespond

168
garfpy/respond.py Normal file
View File

@ -0,0 +1,168 @@
from garfpy import logger
import discord
import json
import os
import re
class GarfbotRespond:
def __init__(self):
self.garfbot_guild_responses = {}
self.responses_file = 'responses.json'
def load_responses(self):
global garfbot_guild_responses
if os.path.exists(self.responses_file):
try:
with open(self.responses_file, 'r', encoding='utf-8') as f:
garfbot_guild_responses = json.load(f)
garfbot_guild_responses = {int(k): v for k, v in garfbot_guild_responses.items()}
total_responses = sum(len(responses) for responses in garfbot_guild_responses.values())
logger.info(f"Loaded responses for {len(garfbot_guild_responses)} server(s), ({total_responses} total responses)")
except Exception as e:
logger.info(f"Error loading responses: {e}")
garfbot_guild_responses = {}
else:
garfbot_guild_responses = {}
def save_responses(self):
try:
save_data = {str(k): v for k, v in garfbot_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 garfbot_guild_responses.values())
logger.info(f"Saved responses for {len(garfbot_guild_responses)} servers ({total_responses} total responses)")
except Exception as e:
logger.info(f"Error saving responses: {e}")
def get_responses(guild_id):
if guild_id not in garfbot_guild_responses:
garfbot_guild_responses[guild_id] = {}
return garfbot_guild_responses[guild_id]
async def garfbot_response(self, message, content):
guild_id = message.guild.id
add_pattern = r'garfbot\s+response\s+add\s+["\']([^"\']+)["\']\s+["\']([^"\']+)["\']'
add_match = re.search(add_pattern, content, re.IGNORECASE)
if add_match:
trigger = add_match.group(1)
response_text = add_match.group(2)
await self.add_response(message, guild_id, trigger, response_text)
return
add_simple_pattern = r'garfbot\s+response\s+add\s+(\S+)\s+(\S+)'
add_simple_match = re.search(add_simple_pattern, content, re.IGNORECASE)
if add_simple_match:
trigger = add_simple_match.group(1)
response_text = add_simple_match.group(2)
await self.add_response(message, guild_id, trigger, response_text)
return
remove_pattern = r'garfbot\s+response\s+remove\s+(\S+)'
remove_match = re.search(remove_pattern, content, re.IGNORECASE)
if remove_match:
trigger = remove_match.group(1).strip()
await self.remove_response(message, guild_id, trigger)
return
if re.search(r'garfbot\s+response\s+list', content, re.IGNORECASE):
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
garfbot_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]
garfbot_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]
garfbot_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)