goober/assets/cogs/internal/base_commands.py

import os
import platform
import subprocess
from typing import Dict, List
import discord
from discord import Colour
from discord.ext import commands
import discord.ext
import discord.ext.commands
from modules.globalvars import local_version
from modules.volta.main import _ , set_language
from modules.permission import requires_admin
from modules.sentenceprocessing import send_message
from modules.settings import instance as settings_manager
from modules.version import check_for_update

import requests

settings = settings_manager.settings

def get_git_origin_raw():
    try:
        result = subprocess.run(
            ["git", "config", "--get", "remote.origin.url"],
            capture_output=True, text=True, check=True
        )
        return result.stdout.strip()
    except Exception:
        return "Failed to get git origin" 

class BaseCommands(commands.Cog):
    def __init__(self, bot):
        self.bot: discord.ext.commands.Bot = bot

    def get_git_origin_raw():
        try:
            result = subprocess.run(
                ["git", "config", "--get", "remote.origin.url"],
                capture_output=True, text=True, check=True
            )
            return result.stdout.strip()
        except Exception:
            return "https://forgejo.expect.ovh/gooberinc/goober"  # fallback if git fails

    @commands.command()
    async def help(self, ctx: commands.Context) -> None:
        embed: discord.Embed = discord.Embed(
            title=f"{_('command_help_embed_title')}",
            description=f"{_('command_help_embed_desc')}",
            color=discord.Colour(0x000000),
        )

        command_categories = {
            f"{_('command_help_categories_general')}": [
                "mem",
                "talk",
                "about",
                "ping",
                "impact",
                "demotivator",
                "help",
            ],
            f"{_('command_help_categories_admin')}": ["stats", "retrain", "setlanguage"],
        }

        custom_commands: List[str] = []
        for cog_name, cog in self.bot.cogs.items():
            for command in cog.get_commands():
                if (
                    command.name
                    not in command_categories[f"{_('command_help_categories_general')}"]
                    and command.name
                    not in command_categories[f"{_('command_help_categories_admin')}"]
                ):
                    custom_commands.append(command.name)

        if custom_commands:
            embed.add_field(
                name=_('command_help_categories_custom'),
                value="\n".join(
                    [
                        f"{settings['bot']['prefix']}{command}"
                        for command in custom_commands
                    ]
                ),
                inline=False,
            )

        for category, commands_list in command_categories.items():
            commands_in_category: str = "\n".join(
                [f"{settings['bot']['prefix']}{command}" for command in commands_list]
            )
            embed.add_field(name=category, value=commands_in_category, inline=False)


        await send_message(ctx, embed=embed)

    @requires_admin()
    @commands.command()
    async def setlanguage(self, ctx: commands.Context, locale: str) -> None:
        await ctx.defer()
        set_language(locale)
        await ctx.send(":thumbsup:")

    @commands.command()
    async def ping(self, ctx: commands.Context) -> None:
        await ctx.defer()
        latency: int = round(self.bot.latency * 1000)

        embed: discord.Embed = discord.Embed(
            title="Pong!!",
            description=(
                settings["bot"]["misc"]["ping_line"],
                f"`{_('command_ping_embed_desc')}: {latency}ms`\n",
            ),
            color=discord.Colour(0x000000),
        )
        embed.set_footer(
            text=f"{_('command_ping_footer')} {ctx.author.name}",
            icon_url=ctx.author.display_avatar.url,
        )

        await ctx.send(embed=embed)

    @commands.command()
    async def about(self, ctx: commands.Context) -> None:
        latest_version: str = check_for_update(slient=True)
        embed: discord.Embed = discord.Embed(title=f"{(_('command_about_embed_title'))}", description="", color=Colour(0x000000))
        embed.add_field(name=f"{(_('command_about_embed_field1'))}", value=f"{settings['name']}", inline=False)
        embed.add_field(name=f"{(_('command_about_embed_field2name'))}", value=f"{(_('command_about_embed_field2value')).format(local_version=local_version, latest_version=latest_version)}", inline=False)
        embed.add_field(name=f"Git", value=get_git_origin_raw())
        embed.add_field(name=f"OS", value=platform.platform())
        
        await send_message(ctx, embed=embed)

    @commands.command()
    async def stats(self, ctx: commands.Context) -> None:
        memory_file: str = "memory.json"
        file_size: int = os.path.getsize(memory_file)

        with open(memory_file, "r") as file:
            line_count: int = sum(1 for _ in file)

        embed: discord.Embed = discord.Embed(
            title=f"{_('command_stats_embed_title')}",
            description=f"{_('command_stats_embed_desc')}",
            color=discord.Colour(0x000000),
        )
        embed.add_field(
            name=f"{_('command_stats_embed_field1name')}",
            value=f"placeholder",
            inline=False,
        )

        with open(settings["splash_text_loc"], "r") as f:
            splash_text = "".join(f.readlines())

        embed.add_field(
            name=_('command_stats_embed_field3name'),
            value=_('command_stats_embed_field3value').format(
                NAME=settings["name"],
                PREFIX=settings["bot"]["prefix"],
                ownerid=settings["bot"]["owner_ids"][0],
                PING_LINE=settings["bot"]["misc"]["ping_line"],
                showmemenabled=settings["bot"]["allow_show_mem_command"],
                USERTRAIN_ENABLED=settings["bot"]["user_training"],
                song=settings["bot"]["misc"]["active_song"],
                splashtext=splash_text
            ),
            inline=False,
        )


        await send_message(ctx, embed=embed)

    @commands.command()
    async def mem(self, ctx: commands.Context) -> None:
        if not settings["bot"]["allow_show_mem_command"]:
            return

        with open(settings["bot"]["active_memory"], "rb") as f:
            data: bytes = f.read()

        response = requests.post(
            "https://litterbox.catbox.moe/resources/internals/api.php",
            data={"reqtype": "fileupload", "time": "1h"},
            files={"fileToUpload": data},
        )

        await send_message(ctx, response.text)


async def setup(bot: discord.ext.commands.Bot):
    print("Setting up base_commands")
    bot.remove_command("help")
    await bot.add_cog(BaseCommands(bot))