From a20e7fdbef88ae98944e9f96b26742fe075878ce Mon Sep 17 00:00:00 2001
From: jkkicks <99859706+jkkicks@users.noreply.github.com>
Date: Mon, 25 Mar 2024 17:30:05 -0700
Subject: [PATCH] Initial commit

---
 .gitignore        |   4 ++
 availcommands.txt |   3 ++
 main.py           | 111 ++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 118 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 availcommands.txt
 create mode 100644 main.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..324bc7d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,4 @@
+*.env
+/.venv
+member_data.db
+discord.log
\ No newline at end of file
diff --git a/availcommands.txt b/availcommands.txt
new file mode 100644
index 0000000..3b9eb8a
--- /dev/null
+++ b/availcommands.txt
@@ -0,0 +1,3 @@
+/help : to view all commands
+/nick : to view current nickname
+/setnick : to change nickname
\ No newline at end of file
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..5c861da
--- /dev/null
+++ b/main.py
@@ -0,0 +1,111 @@
+import discord
+from discord.ext import commands
+import sqlite3
+from dotenv import load_dotenv
+import os
+import logging
+from datetime import datetime
+
+# Set up logging handler. Passed in at the end in bot.run
+handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
+
+# Load ENV file
+load_dotenv()
+
+# Gather variables from .env file
+BOT_TOKEN = os.getenv('BOT_TOKEN')
+CHANNEL_ID = int(os.getenv('BOT_CHANNEL_ID'))
+botCommandsEnable = True
+
+if BOT_TOKEN is None:
+    print("BOT TOKEN NOT FOUND IN ENV")
+    exit()
+if CHANNEL_ID is None:
+    print("NO CHANNEL SELECTED, DISABLING BOT COMMANDS")
+    botCommandsEnable = False
+else:
+    print(f'Channel ID selected: {CHANNEL_ID}')
+
+
+intents = discord.Intents.all()
+intents.members = True  # Enable the member update intent
+intents.message_content = True  # Enable the Privileged Content intent
+
+# Set discord end bot commands to / prefix
+bot = commands.Bot(command_prefix='/', intents=intents)
+
+# Connect to SQLite
+conn = sqlite3.connect('member_data.db')
+cursor = conn.cursor()
+
+# Create Dataset if it doesn't exist already
+cursor.execute('''
+    CREATE TABLE IF NOT EXISTS members (
+        user_id INTEGER PRIMARY KEY,
+        username TEXT,
+        nickname TEXT,
+        join_date TEXT
+    )
+''')
+conn.commit()
+
+
+@bot.event
+async def on_ready():
+    # When API connects and logs in.
+    print(f'Bot logged in as {bot.user.name}')
+
+    # Send help command to bot-commands channel
+    # channel = bot.get_channel(int(CHANNEL_ID))
+    # await channel.send(f'Use /help to view available commands')
+
+
+@bot.event
+async def on_member_join(member):
+    # Print user's name and ID in the console on new join
+    print(f'New member joined: {member.name} (ID: {member.id}), joined at {member.joined_at}')
+    print(member)
+
+    # Store member data in the database
+    cursor.execute('INSERT INTO members (user_id, username, join_date) VALUES (?, ?, ?)',
+                   (member.id, member.name, datetime.now().strftime('%Y-%m-%d %H:%M:%S')))
+    conn.commit()
+
+
+@bot.command()
+async def commands(ctx):
+    avail_commands = open("availcommands.txt", "r")
+    await ctx.send(avail_commands.read())
+    avail_commands.close()
+
+
+@bot.command()
+async def setnick(ctx, new_nickname: str):
+
+        # Update user's nickname in the database
+        cursor.execute('UPDATE members SET nickname = ? WHERE user_id = ?', (new_nickname, ctx.author.id))
+        conn.commit()
+
+        # Retrieve the updated nickname from the database
+        cursor.execute('SELECT nickname FROM members WHERE user_id = ?', (ctx.author.id,))
+        updated_nickname = cursor.fetchone()[0]
+
+        # Print the updated nickname to the console
+        print(f'Nickname updated for {ctx.author.name} (ID: {ctx.author.id}): {updated_nickname}')
+
+        member = ctx.guild.get_member(ctx.author.id)
+        if member:
+            try:
+                await member.edit(nick=updated_nickname)
+                print(f'Nickname updated on the server for {ctx.author.name}')
+            except discord.Forbidden:
+                print(f'Failed to update nickname on the server for {ctx.author.name}: Missing permissions')
+            except discord.HTTPException as e:
+                print(f'Failed to update nickname on the server for {ctx.author.name}: {e}')
+
+# Close the database connection when the bot is stopped
+@bot.event
+async def on_disconnect():
+    conn.close()
+
+bot.run(BOT_TOKEN, log_handler=handler, log_level=logging.DEBUG)