import discord from discord.ext import commands from discord import app_commands import math import re def safe_eval(expression: str) -> str: allowed = re.compile(r'^[\d\s\+\-\*\/\.\%\^sqrtpiloge,]+$', re.IGNORECASE) clean = expression.strip() if not allowed.match(clean): raise ValueError("Invalid characters in expression.") clean = clean.replace('^', '**') safe_globals = { "__builtins__": {}, "sqrt": math.sqrt, "pi": math.pi, "e": math.e, "log": math.log, "sin": math.sin, "cos": math.cos, "tan": math.tan, "abs": abs, "round": round, "pow": pow, } result = eval(clean, safe_globals) if isinstance(result, float) and result.is_integer(): return str(int(result)) return str(round(result, 10)).rstrip('0').rstrip('.') def build_calc_embed(expression: str, result: str) -> discord.Embed: embed = discord.Embed(title="🧮 Calculator", color=0x2ECC71) embed.add_field(name="Expression", value=f"`{expression}`", inline=False) embed.add_field(name="Result", value=f"```{result}```", inline=False) return embed def build_error_embed(msg: str) -> discord.Embed: return discord.Embed(description=f"❌ {msg}", color=0xE74C3C) class Utility(commands.Cog): def __init__(self, bot: commands.Bot): self.bot = bot @commands.command(name="calc", aliases=["calculate", "math"]) async def calc_prefix(self, ctx: commands.Context, *, expression: str): try: result = safe_eval(expression) await ctx.send(embed=build_calc_embed(expression, result)) except ZeroDivisionError: await ctx.send(embed=build_error_embed("Division by zero is not allowed.")) except Exception: await ctx.send(embed=build_error_embed("Invalid expression. Use numbers and operators like `+`, `-`, `*`, `/`, `^`, `sqrt()`, `log()`, etc.")) @app_commands.command(name="calc", description="Solve a math expression instantly.") @app_commands.describe(expression="The math expression to evaluate (e.g. 2+2, sqrt(16), 5^3)") async def calc_slash(self, interaction: discord.Interaction, expression: str): try: result = safe_eval(expression) await interaction.response.send_message(embed=build_calc_embed(expression, result)) except ZeroDivisionError: await interaction.response.send_message(embed=build_error_embed("Division by zero is not allowed."), ephemeral=True) except Exception: await interaction.response.send_message(embed=build_error_embed("Invalid expression. Use numbers and operators like `+`, `-`, `*`, `/`, `^`, `sqrt()`, `log()`, etc."), ephemeral=True) @commands.command(name="help") async def help_prefix(self, ctx: commands.Context): await ctx.send(embed=build_help_embed()) @app_commands.command(name="help", description="Show all available bot commands.") async def help_slash(self, interaction: discord.Interaction): await interaction.response.send_message(embed=build_help_embed()) def build_help_embed() -> discord.Embed: embed = discord.Embed( title="📖 HelperBot — Command List", description="All commands work as both `/command` and `*command`.", color=0x3498DB ) embed.add_field( name="🧮 Utility", value=( "`calc ` — Solve a math expression\n" "`help` — Show this help menu" ), inline=False ) embed.add_field( name="🔨 Moderation", value=( "`ban [reason]` — Ban a member\n" "`kick [reason]` — Kick a member\n" "`mute [reason]` — Mute a member (remove roles)\n" "`timeout [reason]` — Timeout a member\n" "`warn ` — Warn a member\n" "`modlogs ` — View moderation logs for a member\n" "`clearwarns ` — Clear all warnings for a member" ), inline=False ) embed.add_field( name="⚙️ Warn System", value=( "`warnsystem enable` — Enable the auto-timeout warn system\n" "`warnsystem disable` — Disable the auto-timeout warn system\n" "`warnsystem status` — Check if the warn system is active\n" "`warnings ` — View warnings for a member" ), inline=False ) embed.set_footer(text="HelperBot • Prefix: * | Slash: /") return embed async def setup(bot: commands.Bot): await bot.add_cog(Utility(bot))