Compare commits

...

31 Commits

Author SHA1 Message Date
39b821ff7f fix regex hopefully
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 5s
2025-06-04 03:14:25 -05:00
d9ac0cc36e some cleanup etc
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 15s
2025-06-04 02:55:49 -05:00
d707398c26 some cleanup on main
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 20s
2025-06-04 02:33:37 -05:00
0cb5f8fa38 implement auto response
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 15s
2025-06-04 02:22:19 -05:00
d70584d4f8 log test
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 15s
2025-06-03 20:57:39 -05:00
49831aeb47 response log
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 15s
2025-06-03 20:55:41 -05:00
4226ce86c8 fix GarfbotRespond instance
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 14s
2025-06-03 20:41:49 -05:00
b4ca6b778d add auto responses
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 16s
2025-06-03 20:33:55 -05:00
9d5765c492 fix gitea deploy pipeline
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 16s
2025-06-02 17:59:32 -05:00
8216af5f2a deploy test
Some checks failed
Garfbot CI/CD Deployment / Deploy (push) Failing after 3s
2025-06-02 13:52:24 -05:00
219b1745da deploy test
Some checks failed
Garfbot CI/CD Deployment / Deploy (push) Failing after 2s
2025-06-02 13:51:46 -05:00
d570069e6b test
Some checks failed
Garfbot CI/CD Deployment / Deploy (push) Failing after 3s
2025-06-02 13:51:06 -05:00
f267526411 deploy test
Some checks failed
Garfbot CI/CD Deployment / Deploy (push) Failing after 3s
2025-06-02 13:49:49 -05:00
86fa47d259 remove docker socket mount
Some checks failed
Garfbot CI/CD Deployment / Deploy (push) Failing after 5s
2025-06-02 13:47:57 -05:00
888e647e1f add gitea deploy script
Some checks failed
Garfbot CI/CD Deployment / Deploy (push) Failing after 1s
2025-06-02 13:06:23 -05:00
c1beb8374e fix qr sendfile 2025-05-25 04:15:08 -05:00
48a7033ba6 add pillow 2025-05-25 04:11:57 -05:00
9e436d8d78 fix qrcode call async 2025-05-25 04:08:18 -05:00
d8c286bb2d add qrcode 2025-05-25 04:04:32 -05:00
a081edfa97 add qr code gen 2025-05-25 04:02:34 -05:00
716e30e2b1 fix chat gen 2025-05-23 20:46:26 -05:00
73499a9586 oops 2025-05-23 13:35:19 -05:00
e3a762fe10 await chat gen 2025-05-23 13:33:02 -05:00
842b592bfc garfsum 2025-05-23 13:30:36 -05:00
f9f88be8af wiki except 2025-05-23 11:41:26 -05:00
5c8696216e fix wiki init 2025-05-22 17:12:56 -05:00
aa16064d17 fix wikisum import 2025-05-22 17:11:39 -05:00
ffd642fd2a add wiki stuff 2025-05-22 17:09:48 -05:00
3945eb6763 add wiki 2025-05-22 16:53:37 -05:00
0438b55ce6 remove tenor gif bs 2025-05-22 16:46:22 -05:00
8fb4e089be Update README.md 2024-12-28 05:08:50 +00:00
11 changed files with 343 additions and 62 deletions

View File

@ -0,0 +1,28 @@
name: Garfbot CI/CD Deployment
on:
push:
branches: [ main ]
jobs:
Deploy:
container:
volumes:
- /home/crate/garfbot:/workspace/crate/garfbot/deploy
steps:
- name: Pull Garfbot and restart container
run: |
cd /workspace/crate/garfbot/deploy
git pull origin main
CHANGED=$(git diff --name-only HEAD~1 HEAD)
if echo "$CHANGED" | grep -qE "(Dockerfile|requirements\.txt|docker-compose\.yml)"; then
docker stop garfbot
docker rm garfbot
docker build -t git.crate.zip/crate/garfbot:latest .
docker run -d --restart always -v $PWD:/usr/src/app --name garfbot git.crate.zip/crate/garfbot:latest
else
docker restart garfbot
fi

3
.gitignore vendored
View File

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

View File

@ -12,5 +12,8 @@ RUN pip3 install discord
RUN pip3 install openai RUN pip3 install openai
RUN pip3 install aiohttp RUN pip3 install aiohttp
RUN pip3 install requests RUN pip3 install requests
RUN pip3 install wikipedia
RUN pip3 install pillow
RUN pip3 install qrcode
CMD [ "python", "garfmain.py" ] CMD [ "python", "garfmain.py" ]

View File

@ -16,7 +16,7 @@ Responds with iputils-ping result from target.
`garfpic {target}` `garfpic {target}`
Responds with dns lookup result from target. Responds with dns lookup result from target.
`garfpic {target}` `garfhack {target}`
Responds with nmap scan result from target. Responds with nmap scan result from target.
`garfshop {item} {zip}` `garfshop {item} {zip}`
@ -36,7 +36,6 @@ Add your various API tokens:
```python ```python
GARFBOT_TOKEN = "Discord API token" GARFBOT_TOKEN = "Discord API token"
OPENAI_TOKEN = "OpenAI API token" OPENAI_TOKEN = "OpenAI API token"
GIF_TOKEN = "tenor.com API token"
``` ```
I recommend building a docker image using the included DockerFile as a template. Run the container binding /usr/src/app to GarfBot's CWD: I recommend building a docker image using the included DockerFile as a template. Run the container binding /usr/src/app to GarfBot's CWD:

View File

@ -1,5 +1,4 @@
import config import config
# import random
import asyncio import asyncio
import discord import discord
import subprocess import subprocess
@ -7,8 +6,8 @@ import subprocess
from garfpy import( from garfpy import(
logger, is_private, logger, is_private,
kroger_token, find_store, search_product, kroger_token, find_store, search_product,
picture_time, process_image_requests, generate_chat, garfpic, process_image_requests, generate_chat,
aod_message) aod_message, wikisum, generate_qr, GarfbotRespond)
gapikey = config.GIF_TOKEN gapikey = config.GIF_TOKEN
@ -22,11 +21,14 @@ intents.messages = True
intents.message_content = True intents.message_content = True
garfbot = discord.Client(intents=intents) garfbot = discord.Client(intents=intents)
garf_respond = GarfbotRespond()
@garfbot.event @garfbot.event
async def on_ready(): async def on_ready():
try: try:
asyncio.create_task(process_image_requests()) asyncio.create_task(process_image_requests())
garf_respond.load_responses()
logger.info(f"Logged in as {garfbot.user.name} running {txtmodel} and {imgmodel}.") logger.info(f"Logged in as {garfbot.user.name} running {txtmodel} and {imgmodel}.")
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
@ -34,39 +36,56 @@ async def on_ready():
@garfbot.event @garfbot.event
async def on_message(message): async def on_message(message):
content = message.content.strip()
lower = content.lower()
user = message.author.name
guild = message.guild.name if message.guild else "Direct Message"
guild_id = message.guild.id
# Chats & pics
if message.author == garfbot.user: if message.author == garfbot.user:
return return
if message.content.lower().startswith("hey garfield") or isinstance(message.channel, discord.DMChannel): if lower.startswith("hey garfield") or isinstance(message.channel, discord.DMChannel):
question = message.content[12:] if message.content.lower().startswith("hey garfield") else message.content question = content[12:] if lower.startswith("hey garfield") else message.content
answer = await generate_chat(question) answer = await generate_chat(question)
logger.info(f"Chat Request - User: {user}, Server: {guild}, Prompt: {question}")
await message.channel.send(answer) await message.channel.send(answer)
if message.content.lower().startswith('garfpic '): if lower.startswith('garfpic '):
user = message.author.name prompt = content[8:]
server = message.guild.name if message.guild else "Direct Message" logger.info(f"Image Request - User: {user}, Server: {guild}, Prompt: {prompt}")
prompt = message.content[8:]
logger.info(f"Image Request - User: {user}, Server: {server}, Prompt: {prompt}")
await message.channel.send(f"`Please wait... image generation queued: {prompt}`") await message.channel.send(f"`Please wait... image generation queued: {prompt}`")
await picture_time(message, prompt) await garfpic(message, prompt)
# if message.content.lower() == "lasagna": # Wikipedia
# await send_gif(message, "garfield lasagna") if lower.startswith('garfwiki '):
search_term = message.content[9:]
summary = await wikisum(search_term)
await message.channel.send(summary)
# if message.content.lower() == "monday": # QR codes
# await send_gif(message, "garfield monday") if lower.startswith('garfqr '):
text = message.content[7:]
# if message.content.lower().startswith("garfgif "): if len(text) > 1000:
# search_term = message.content[8:] await message.channel.send("❌ Text too long! Maximum 1000 characters.")
# await send_gif(message, search_term) else:
if message.content.lower().startswith("garfping "):
try: try:
qr_code = await generate_qr(text)
sendfile = discord.File(fp=qr_code, filename="qrcode.png")
await message.channel.send(file=sendfile)
except Exception as e:
logger.error(e)
await message.channel.send(e)
# IP utils
query = message.content.split() query = message.content.split()
user = message.author.name
server = message.guild.name if message.guild else "Direct Message"
target = query[-1] target = query[-1]
logger.info(f"Ping Request - User: {user}, Server: {server}, Target: {target}")
if lower.startswith("garfping "):
try:
logger.info(f"Ping Request - User: {user}, Server: {guild}, Target: {target}")
if is_private(target): if is_private(target):
rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.") rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
await message.channel.send(rejection) await message.channel.send(rejection)
@ -76,13 +95,9 @@ async def on_message(message):
except Exception as e: except Exception as e:
await message.channel.send(f"`GarfBot Error: {str(e)}`") await message.channel.send(f"`GarfBot Error: {str(e)}`")
if message.content.lower().startswith("garfdns "): if lower.startswith("garfdns "):
try: try:
query = message.content.split() logger.info(f"NSLookup Request - User: {user}, Server: {guild}, Target: {target}")
user = message.author.name
server = message.guild.name if message.guild else "Direct Message"
target = query[-1]
logger.info(f"NSLookup Request - User: {user}, Server: {server}, Target: {target}")
if is_private(target): if is_private(target):
rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.") rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
await message.channel.send(rejection) await message.channel.send(rejection)
@ -92,13 +107,9 @@ async def on_message(message):
except Exception as e: except Exception as e:
await message.channel.send(f"`GarfBot Error: {str(e)}`") await message.channel.send(f"`GarfBot Error: {str(e)}`")
if message.content.lower().startswith("garfhack "): if lower.startswith("garfhack "):
try: try:
query = message.content.split() logger.info(f"Nmap Request - User: {user}, Server: {guild}, Target: {target}")
user = message.author.name
server = message.guild.name if message.guild else "Direct Message"
target = query[-1]
logger.info(f"Nmap Request - User: {user}, Server: {server}, Target: {target}")
if is_private(target): if is_private(target):
rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.") rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
await message.channel.send(rejection) await message.channel.send(rejection)
@ -110,7 +121,7 @@ async def on_message(message):
await message.channel.send(f"`GarfBot Error: {str(e)}`") await message.channel.send(f"`GarfBot Error: {str(e)}`")
# Kroger Shopping # Kroger Shopping
if message.content.lower().startswith("garfshop "): if lower.startswith("garfshop "):
try: try:
kroken = kroger_token() kroken = kroger_token()
kroger_query = message.content.split() kroger_query = message.content.split()
@ -133,29 +144,22 @@ async def on_message(message):
# Army of Dawn Server only!! # Army of Dawn Server only!!
if message.guild and message.guild.id == 719605634772893757: if message.guild and message.guild.id == 719605634772893757:
await aod_message(garfbot, message) await aod_message(garfbot, message)
# Auto-responses
if message.guild:
responses = garf_respond.get_responses(guild_id)
# # GarfGifs if lower.startswith('garfbot response '):
# @garfbot.event await garf_respond.garfbot_response(message, content)
# async def send_gif(message, search_term): return
# lmt = 50
# ckey = "garfbot"
# r = requests.get(f"https://tenor.googleapis.com/v2/search?q={search_term}&key={gapikey}&client_key={ckey}&limit={lmt}")
# if r.status_code == 200:
# top_50gifs = json.loads(r.content)
# gif_url = random.choice(top_50gifs["results"])["itemurl"]
# logger.info(gif_url)
# # logger.info(gif_url)
# try:
# await message.channel.send(gif_url)
# except KeyError:
# await message.channel.send("Oops, something went wrong.")
# else:
# await message.channel.send(f"`Oops, something went wrong. Error code: {r.status_code}`")
for trigger, response in responses.items():
if trigger.lower() in lower:
await message.channel.send(response)
break
# Run Garfbot
async def garfbot_connect(): async def garfbot_connect():
while True: while True:
try: try:

View File

@ -1,12 +1,16 @@
# garfpy/__init__.py # garfpy/__init__.py
from .log import logger from .log import logger
from .kroger import( from .kroger import(
kroger_token, find_store, search_product kroger_token, find_store, search_product
) )
from .garfai import( from .garfai import(
picture_time, garfpic,
process_image_requests, process_image_requests,
generate_chat generate_chat
) )
from .iputils import is_private from .iputils import is_private
from .aod import aod_message from .aod import aod_message
from .wiki import wikisum
from .qr import generate_qr
from .respond import GarfbotRespond

View File

@ -12,10 +12,9 @@ openaikey = config.OPENAI_TOKEN
txtmodel = config.TXT_MODEL txtmodel = config.TXT_MODEL
imgmodel = config.IMG_MODEL imgmodel = config.IMG_MODEL
# GarfPics
image_request_queue = asyncio.Queue() image_request_queue = asyncio.Queue()
async def picture_time(message, prompt): async def garfpic(message, prompt):
await image_request_queue.put({'message': message, 'prompt': prompt}) await image_request_queue.put({'message': message, 'prompt': prompt})
async def generate_image(prompt): async def generate_image(prompt):

63
garfpy/qr.py Normal file
View File

@ -0,0 +1,63 @@
import qrcode
from io import BytesIO
def calculate_qr_settings(text):
text_length = len(text)
if text_length <= 25:
version = 1
box_size = 12
elif text_length <= 47:
version = 2
box_size = 10
elif text_length <= 77:
version = 3
box_size = 8
elif text_length <= 114:
version = 4
box_size = 7
elif text_length <= 154:
version = 5
box_size = 6
elif text_length <= 195:
version = 6
box_size = 5
elif text_length <= 224:
version = 7
box_size = 5
elif text_length <= 279:
version = 8
box_size = 4
elif text_length <= 335:
version = 9
box_size = 4
elif text_length <= 395:
version = 10
box_size = 3
else:
version = None
box_size = 3
return version, box_size
async def generate_qr(text):
version, box_size = calculate_qr_settings(text)
qr = qrcode.QRCode(
version=version,
error_correction=qrcode.constants.ERROR_CORRECT_L,
box_size=box_size,
border=4,
)
qr.add_data(text)
qr.make(fit=True)
qr_image = qr.make_image(fill_color="black", back_color="white")
img_buffer = BytesIO()
qr_image.save(img_buffer, format='PNG')
img_buffer.seek(0)
return img_buffer

163
garfpy/respond.py Normal file
View File

@ -0,0 +1,163 @@
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):
if os.path.exists(self.responses_file):
try:
with open(self.responses_file, 'r', encoding='utf-8') as f:
self.garfbot_guild_responses = json.load(f)
self.garfbot_guild_responses = {int(k): v for k, v in self.garfbot_guild_responses.items()}
total_responses = sum(len(responses) for responses in self.garfbot_guild_responses.values())
logger.info(f"Loaded responses for {len(self.garfbot_guild_responses)} server(s), ({total_responses} total responses)")
except Exception as e:
logger.info(f"Error loading responses: {e}")
self.garfbot_guild_responses = {}
else:
self.garfbot_guild_responses = {}
def save_responses(self):
try:
save_data = {str(k): v for k, v in self.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 self.garfbot_guild_responses.values())
logger.info(f"Saved responses for {len(self.garfbot_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.garfbot_guild_responses:
self.garfbot_guild_responses[guild_id] = {}
return self.garfbot_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.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]
self.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]
self.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)

12
garfpy/wiki.py Normal file
View File

@ -0,0 +1,12 @@
import wikipedia
from garfpy import generate_chat
async def wikisum(search_term):
try:
summary = wikipedia.summary(search_term)
garfsum = await generate_chat(f"Please summarize in your own words: {summary}")
return garfsum
except Exception as e:
return e

View File

@ -2,3 +2,6 @@ discord.py
openai openai
aiohttp aiohttp
requests requests
wikipedia
pillow
qrcode