import os
import discord
from discord.ext import commands
from git import Repo
import datetime
from typing import Set

# Bot setup
intents = discord.Intents.default()
intents.message_content = True
bot = commands.Bot(command_prefix='!', intents=intents)

# Initialize repo and tracked messages
REPO_PATH = os.getenv('REPO_PATH', '.')  # Path to your git repository
repo = None
tracked_messages: Set[int] = set()

@bot.event
async def on_ready():
    global repo
    print(f'{bot.user} has connected to Discord!')
    try:
        repo = Repo(REPO_PATH)
        print(f'Connected to git repository at {REPO_PATH}')
    except Exception as e:
        print(f'Error connecting to git repository: {e}')

@bot.command(name='track')
async def track_message(ctx, message_id: str):
    try:
        # Convert message_id to integer
        msg_id = int(message_id)
        tracked_messages.add(msg_id)
        await ctx.send(f'Now tracking message with ID: {msg_id}')
    except ValueError:
        await ctx.send('Please provide a valid message ID')

@bot.command(name='untrack')
async def untrack_message(ctx, message_id: str):
    try:
        # Convert message_id to integer
        msg_id = int(message_id)
        if msg_id in tracked_messages:
            tracked_messages.remove(msg_id)
            await ctx.send(f'Stopped tracking message with ID: {msg_id}')
        else:
            await ctx.send('This message was not being tracked')
    except ValueError:
        await ctx.send('Please provide a valid message ID')

@bot.command(name='list-tracked')
async def list_tracked_messages(ctx):
    if tracked_messages:
        message = 'Currently tracking these message IDs:\n' + '\n'.join(str(mid) for mid in tracked_messages)
    else:
        message = 'No messages are currently being tracked'
    await ctx.send(message)

@bot.event
async def on_message_edit(before, after):
    if before.author.bot or before.id not in tracked_messages:
        return

    try:
        # Create a new branch for the edit
        current = repo.active_branch
        timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
        branch_name = f'msg_edit_{timestamp}'
        
        # Create and checkout new branch
        new_branch = repo.create_head(branch_name)
        new_branch.checkout()

        # Create or update the message file
        message_file = os.path.join(REPO_PATH, f'message_{before.id}.txt')
        with open(message_file, 'w') as f:
            f.write(f'Message ID: {before.id}\n')
            f.write(f'Channel: {before.channel.name}\n')
            f.write(f'Author: {before.author.name}\n')
            f.write(f'Last Updated: {timestamp}\n\n')
            f.write(after.content)

        # Stage and commit the changes
        repo.index.add([message_file])
        repo.index.commit(f'Update message {before.id} - {timestamp}')

        # Switch back to the original branch
        current.checkout()

        await after.channel.send(f'Changes committed to branch: `{branch_name}`')

    except Exception as e:
        await after.channel.send(f'Error committing changes: {e}')

# Replace 'YOUR_TOKEN' with your bot token
bot.run(os.getenv('DISCORD_TOKEN'))