curds/booly
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Bugfixs

Browse source
This commit is contained in:
Charlie 2026-01-31 18:04:52 +13:00
parent 0f369873ac
commit 7578a5be6d
1 changed files with 63 additions and 172 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

225
cogs/utility.py
View file

@ -1,182 +1,73 @@
import discord import discord
from discord.ext import commands from discord.ext import commands
import os from discord import app_commands
import json import re
import logging import random
from aiohttp import web import string
import asyncio
logging.basicConfig(level=logging.INFO) class Utility(commands.Cog):
logger = logging.getLogger('bot') def __init__(self, bot):
self.bot = bot
self.domain = "u.chers.moe"
CONFIG = { def is_valid_url(self, url):
'xp_min': 15, url_pattern = re.compile(
'xp_max': 25, r'^https?://'
'xp_cooldown': 60, r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,6}\.?|'
'xp_per_level': 100, r'localhost|'
'level_up_multiplier': 10, r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})'
'data_file': 'data.json', r'(?::\d+)?'
'video_check_interval': 300 r'(?:/?|[/?]\S+)$', re.IGNORECASE)
} return url_pattern.match(url) is not None
class SimpleDB: def generate_short_code(self, length=6):
def __init__(self, filename): chars = string.ascii_letters + string.digits
self.filename = filename return ''.join(random.choice(chars) for _ in range(length))
self.data = self.load()
def load(self): def get_guild_data(self, guild_id):
if os.path.exists(self.filename): if 'guilds' not in self.bot.db.data:
try: self.bot.db.data['guilds'] = {}
with open(self.filename, 'r') as f: if str(guild_id) not in self.bot.db.data['guilds']:
return json.load(f) self.bot.db.data['guilds'][str(guild_id)] = {'urls': {}}
except: if 'urls' not in self.bot.db.data['guilds'][str(guild_id)]:
return {'users': {}, 'guilds': {}} self.bot.db.data['guilds'][str(guild_id)]['urls'] = {}
return {'users': {}, 'guilds': {}} return self.bot.db.data['guilds'][str(guild_id)]
def save(self): @app_commands.command(name='shorten', description='Shorten a long URL')
with open(self.filename, 'w') as f: @app_commands.describe(url='The URL you want to shorten')
json.dump(self.data, f, indent=2) async def shorten_url(self, interaction: discord.Interaction, url: str):
if not self.is_valid_url(url):
await interaction.response.send_message(
'Please provide a valid URL',
ephemeral=True
)
return
def get_user(self, guild_id, user_id): guild_data = self.get_guild_data(interaction.guild_id)
key = f"{guild_id}_{user_id}"
if key not in self.data['users']:
self.data['users'][key] = {
'coins': 0,
'bank': 0,
'level': 1,
'xp': 0,
'last_message': 0,
'last_daily': 0,
'last_work': 0
}
return self.data['users'][key]
def set_user(self, guild_id, user_id, data): for code, stored_url in guild_data['urls'].items():
key = f"{guild_id}_{user_id}" if stored_url == url:
self.data['users'][key] = data shortened = f"https://{self.domain}/{code}"
self.save() embed = discord.Embed(title='URL Shortened', color=0x5865F2)
embed.add_field(name='Original', value=url[:100] + ('...' if len(url) > 100 else ''), inline=False)
embed.add_field(name='Shortened', value=shortened, inline=False)
await interaction.response.send_message(embed=embed)
return
def get_all_guild_users(self, guild_id): code = self.generate_short_code()
users = [] while code in guild_data['urls']:
for key, data in self.data['users'].items(): code = self.generate_short_code()
if key.startswith(f"{guild_id}_"):
user_id = key.split('_')[1]
users.append({'user_id': user_id, 'data': data})
users.sort(key=lambda x: (x['data']['level'], x['data']['xp']), reverse=True)
return users
class MyBot(commands.Bot): guild_data['urls'][code] = url
def __init__(self): self.bot.db.save()
intents = discord.Intents.default()
intents.message_content = True
intents.members = True
super().__init__(command_prefix='/', intents=intents) shortened = f"https://{self.domain}/{code}"
self.db = SimpleDB(CONFIG['data_file'])
self.config = CONFIG
async def setup_hook(self): embed = discord.Embed(title='URL Shortened', color=0x5865F2)
await self.load_extension('cogs.leveling') embed.add_field(name='Original', value=url[:100] + ('...' if len(url) > 100 else ''), inline=False)
await self.load_extension('cogs.system') embed.add_field(name='Shortened', value=shortened, inline=False)
await self.load_extension('cogs.economy')
await self.load_extension('cogs.fun')
await self.load_extension('cogs.utility')
await self.tree.sync() await interaction.response.send_message(embed=embed)
logger.info('All cogs loaded and commands synced!')
bot = MyBot() async def setup(bot):
await bot.add_cog(Utility(bot))
async def health_check(request):
return web.Response(text="Bot is running!")
async def redirect_handler(request):
code = request.match_info.get('code', '')
if os.path.exists(CONFIG['data_file']):
try:
with open(CONFIG['data_file'], 'r') as f:
data = json.load(f)
for guild_id, guild_data in data.get('guilds', {}).items():
if 'urls' in guild_data and code in guild_data['urls']:
return web.Response(
status=301,
headers={'Location': guild_data['urls'][code]}
)
except Exception as e:
logger.error(f'Error: {e}')
return web.Response(text='Not Found', status=404)
async def start_web_server():
app = web.Application()
app.router.add_get('/', health_check)
app.router.add_get('/health', health_check)
app.router.add_get('/{code}', redirect_handler)
runner = web.AppRunner(app)
await runner.setup()
port = int(os.getenv('PORT', 8080))
site = web.TCPSite(runner, '0.0.0.0', port)
await site.start()
logger.info(f'Web server running on port {port}')
@bot.event
async def on_ready():
asyncio.create_task(start_web_server())
logger.info(f'Logged in as {bot.user}')
logger.info(f'Connected to {len(bot.guilds)} guilds')
await bot.change_presence(activity=discord.Game(name="/help"))
logger.info('All systems operational!')
@bot.tree.command(name='help', description='Show all available commands')
async def help_command(interaction: discord.Interaction):
embed = discord.Embed(
title='Bot Commands',
description='Here are all the slash commands you can use!',
color=0x5865F2
)
embed.add_field(
name='Leveling & Economy',
value='`/rank` - View your rank\n`/leaderboard` - Top 10 users\n`/balance` - Check balance\n`/daily` - Daily reward\n`/work` - Work for coins',
inline=False
)
embed.add_field(
name='Fun',
value='`/8ball` - Magic 8ball\n`/roll` - Roll dice\n`/flip` - Flip coin\n`/cat` - Random cat\n`/dog` - Random dog',
inline=False
)
embed.add_field(
name='System & Moderation',
value='`/kick` - Kick member\n`/ban` - Ban member\n`/unban` - Unban user\n`/timeout` - Timeout member\n`/warn` - Warn member\n`/warnings` - View warnings\n`/clearwarnings` - Clear warnings\n`/purge` - Delete messages\n`/lock` - Lock channel\n`/unlock` - Unlock channel',
inline=False
)
embed.add_field(
name='Reaction Roles & YouTube',
value='`/reactionrole` - Create reaction role\n`/removereactionrole` - Remove reaction role\n`/listreactionroles` - List reaction roles\n`/createreactionpanel` - Create panel\n`/setupyoutube` - Setup YouTube\n`/toggleyoutube` - Toggle YouTube\n`/youtubestatus` - YouTube status\n`/testlastvideo` - Test video',
inline=False
)
embed.add_field(
name='Utility',
value='`/shorten` - Shorten a URL',
inline=False
)
embed.add_field(
name='Info',
value='`/ping` - Bot latency\n`/serverinfo` - Server info\n`/userinfo` - User info',
inline=False
)
await interaction.response.send_message(embed=embed)
if __name__ == '__main__':
token = os.getenv('DISCORD_TOKEN')
if not token:
logger.error('DISCORD_TOKEN not set!')
exit(1)
logger.info("everythings working")
logger.info('hello from chersbobers and booly co :3')
bot.run(token)