main.py

import random
import discord
import subprocess
import re
from discord import app_commands
from discord.ext import commands
import uuid
import requests
import shutil
import sys
import os
import sqlite3
from settings import CBotSettings as CBotSettings

# Check if SqLite db exists
print(f"[INIT] Checking for SqLite3 Database...")
if os.path.isfile(f"./{CBotSettings.DbName}"):
    print(f"[INIT] Database exists!")
    pass
else:
    print(f"[INIT WARN] Database does not exist, created it!")
    conn = sqlite3.connect(CBotSettings.DbName)
    c = conn.cursor()
    c.execute('CREATE TABLE ServerMembers (id INTEGER PRIMARY KEY, name TEXT, password TEXT, discordTag TEXT, verifyCode TEXT, isVerified BOOLEAN)')
    conn.commit()
    conn.close()
    pass

# Update checks

def CheckRemote():
    repo_url = "https://github.com/ImExiled/LxliBot.git"
    process = subprocess.Popen(["git", "ls-remote", repo_url], stdout=subprocess.PIPE)
    stdout, stderr = process.communicate()
    sha = re.split(r'\t+', stdout.decode('ascii'))[0]
    print(f"[UPDATE CHECK] Latest push hash is {sha}")
    return sha

def CheckLocal():
    process = subprocess.Popen(["git", "rev-parse", 'HEAD'], stdout=subprocess.PIPE)
    stdout, stderr = process.communicate()
    sha = re.split(r'\t+', stdout.decode('ascii'))[0]
    print(f"[UPDATE CHECK] Latest local hash is {sha}")
    return sha

def CheckForUpdate():
        print(f"[UPDATER] Checking for updates")
        update = subprocess.Popen(["git", "pull"], stdout=subprocess.PIPE)
        stdout, stderr = update.communicate()
        print(stdout)

CheckForUpdate()

# Init checks to ensure bot functionality.
if not os.path.isdir(CBotSettings.ArchiveRoot):
    print(f"The archive root ({CBotSettings.ArchiveRoot}) does not exist. Created it! :)")
    os.mkdir(CBotSettings.ArchiveRoot)
for folder in CBotSettings.ArchiveChannels:
    if not os.path.isdir(f"{CBotSettings.ArchiveRoot}/{folder}"):
        print(f"The archive category ({CBotSettings.ArchiveRoot}/{folder}) does not exist. Created it!")
        os.mkdir(f"{CBotSettings.ArchiveRoot}/{folder}")


intents = discord.Intents.all()
intents.message_content = True

client = commands.Bot(intents = intents, command_prefix=CBotSettings.CommandSymbol)

@client.event
async def on_ready():
    print(f"Bot authenticated as: {client.user}")
    print("Syncing commands...")
    try:
        synced = await client.tree.sync()
        print(f"Synced {len(synced)} command(s)!")
        print(f"[STATUS] Launched successfully!")
    except Exception as e:
        print(f"Error syncing commands: {e}")

@client.event
async def on_message(message):
    # Archiver
    if CBotSettings.ArchiveImages == True:
        # Archiver
        try:
            RandomFileName = str(uuid.uuid4())
            SourceChannel = str(message.channel)
            if message.attachments[0].content_type == "image/png":
                await message.attachments[0].save(RandomFileName + ".png")
                shutil.move(RandomFileName + ".png", CBotSettings.ArchiveRoot + "/" + SourceChannel)
            if message.attachments[0].content_type == "image/jpeg":
                await message.attachments[0].save(RandomFileName + ".jpeg")
                shutil.move(RandomFileName + ".jpeg", CBotSettings.ArchiveRoot + "/" + SourceChannel)
            if message.attachments[0].content_type == "image/jpg":
                await message.attachments[0].save(RandomFileName + ".jpg")
                shutil.move(RandomFileName + ".jpg", CBotSettings.ArchiveRoot + "/" + SourceChannel)
            if message.attachments[0].content_type == "image/webm":
                await message.attachments[0].save(RandomFileName + ".webm")
                shutil.move(RandomFileName + ".webm", CBotSettings.ArchiveRoot + "/" + SourceChannel)
            if message.attachments[0].content_type == "image/gif":
                await message.attachments[0].save(RandomFileName + ".gif")
                shutil.move(RandomFileName + ".gif", CBotSettings.ArchiveRoot + "/" + SourceChannel)
        except Exception as e:
            #print("An error occured: " + e)
            pass
    if message.author == client.user:
        return

### COMMANDS ###

@client.tree.command(name="hello", description="Say hello!")
async def hello(interaction: discord.Interaction):
    await interaction.response.send_message(f"Hey {interaction.user.mention}! Get to gooning already!~", ephemeral=True)

@client.tree.command(name="cunny", description="You like small girls I see~")
async def cunny(interaction: discord.Interaction, category: str):
    media = os.listdir(CBotSettings.ArchiveRoot + "/" + category)
    random_media = random.choice(media)
    random_media_str = os.path.basename(random_media)
    await interaction.response.send_message(f"Get nasty!~", file=discord.File(f"{CBotSettings.ArchiveRoot}/{category}/{random_media_str}"), ephemeral=True)

@client.tree.command(name="totals", description="How many are there?!~")
async def totals(interaction: discord.Interaction):
    cpt = sum([len(files) for r, d, files in os.walk(f"{CBotSettings.ArchiveRoot}")])
    await interaction.response.send_message(f"We've archived {cpt} delicious tiny ones in total~")

# Bug reports!
@client.tree.command(name="bug", description="Used for bugs and feature requests")
async def bug(interaction: discord.Interaction):
    issuesLink = "https://github.com/ImExiled/LxliBot/issues"
    await interaction.response.send_message(f"Hey there, {interaction.user.mention}! Found a bug, or wanna request a feature? You can do so here: https://github.com/ImExiled/LxliBot/issues")
# FortuneOne's dedicated command
#@client.tree.command(name="fo", description="FortuneOne's dedicated command~")
#async def fo(interaction: discord.Interaction):
#    category = "incest"
#    fortune_media = os.listdir(CBotSettings.ArchiveRoot + "/" + category)
#    await interaction.response.send_message(f"Incest for the wincest~", ephemeral=True, file=discord.File(f"{fortune_media}"))

# Run the bot!
client.run(f'{CBotSettings.TOKEN}')