import random
import discord
import sys
import os
from discord.ext import commands
from discord.utils import get
from discord import SyncWebhook
import json

admins = [00000000000000] # User IDs of Admins
guild_id = 0000000000000000 # Discord Server ID
category_id = "DMs" # Discord Channel Category Name

# global replies

# if not os.path.isfile("replies.json"):
#     with open("replies.json", "w+") as g:
#         json.dump({}, g, indent=1)
#     g.close()

if not os.path.isfile("dm_ids.json"):
    with open("dm_ids.json", "w+") as g:
        json.dump({'': {'webhook_url': '', 'channel_id': ''}}, g, indent=1)
    g.close()


# def file_update():
#     global replies
#     with open("replies.json", "r") as g:
#         replies = json.load(g)
# 
#         # json.dump(ids, f, indent=1)
#     g.close()


def attachment_list(message):
    dm_attach = message.content
    for x in message.attachments:
        dm_attach = dm_attach + " " + x.url
    return dm_attach


# file_update()

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

client = commands.Bot(command_prefix='.', description="test", intents=intents)

with open("dm_ids.json", "r") as f:
    ids = json.load(f)
f.close()


@client.event
async def on_ready():
    print(f'We have logged in as {client.user}')


@client.event
async def on_message(message):
    global ids
    guild = client.get_guild(guild_id) 
    channelcategory = get(guild.channels, name=category_id) 
    if message.author == client.user or message.author.bot:
        return

    for userid in ids:
        if str(message.channel.id) == ids[userid]["channel_id"]:
            user = await client.fetch_user(userid)
            await user.send(attachment_list(message))

    if isinstance(message.channel, discord.DMChannel):
        try: # If user has already messaged the bot once
            SyncWebhook.from_url(ids[str(message.author.id)]["webhook_url"]).send(attachment_list(message),
                                                                                  username=message.author.name,
                                                                                  avatar_url=message.author.avatar)
        except KeyError: # If user has not messaged the bot at least once
            user = await client.fetch_user(message.author.id)
            await user.send("Warning: All messages sent to this bot are logged and readable by this bot's owners. By "
                            "continuing, you accept this. Your first message has not been logged. Your subsequent "
                            "messages will be logged.")
            dm_channel = await guild.create_text_channel(str(message.author).replace("#", "-"), overwrites={},
                                                         category=channelcategory)
            webhook = await dm_channel.create_webhook(name="user")
            # SyncWebhook.from_url(webhook.url).send(attachment_list(message),
            #                                        username=message.author.name, avatar_url=message.author.avatar)
            ids[str(message.author.id)] = {"webhook_url": str(webhook.url), "channel_id": str(dm_channel.id)}
            with open("dm_ids.json", "w") as f:
                json.dump(ids, f, indent=1)
                # ids = json.load(f)
            f.close()

    # if not message.content.startswith("."):
    #     for i in replies:
    #         if i in message.content.strip().lower():
    #             await message.reply(replies[i])
    #             return

    await client.process_commands(message)



# @client.command()
# async def upload(ctx):
#     if ctx.author.id in admins:
#         await ctx.reply(file=discord.File(r'replies.json'))


# @client.command()
# async def update(ctx):
#     if ctx.author.id in admins:
#         file_update()
#         await ctx.reply("updated!")


# @client.command()
# async def add(ctx, args, args2: str = None):
#     if ctx.author.id in admins:
#         global replies
#         if args in replies:
#             await ctx.reply(args + " is already added")
#         else:
#             if args2 is None:
#                 replies[args] = args
#             else:
#                 replies[args] = args2
#             with open("replies.json", "w") as g:
#                 json.dump(replies, g, indent=1)
#             g.close()
# 
#             await ctx.reply("added " + args)
#             file_update()


# @client.command()
# async def remove(ctx, args):
#     if ctx.author.id in admins:
#         try:
#             del replies[args]
#             with open("replies.json", "w") as g:
#                 json.dump(replies, g, indent=1)
#             g.close()
#             await ctx.reply(f"{args} has been removed")
#             file_update()
#         except KeyError:
#             await ctx.reply("command borke :( key no existo")


@client.group(pass_context=True)
async def send(ctx):
    if ctx.author.id in admins:
        if ctx.invoked_subcommand is None:
            await ctx.send('Parameters: `.send (channel/dm) ID "message contents"`')


@send.group(pass_context=True)
async def channel(ctx, uid: int, message):
    if ctx.author.id in admins:
        channelsend = await ctx.bot.fetch_channel(uid)
        await channelsend.send(message)


@channel.error
async def test_error(ctx):
    await ctx.send('Parameters: `.send channel ID "message contents"`')


@send.group(pass_context=True)
async def dm(ctx, uid: int, message):
    if ctx.author.id in admins:
        dmuser = await ctx.bot.fetch_user(uid)
        await dmuser.send(message)


@dm.error
async def test_error(ctx):
    await ctx.send('Parameters: `.send dm ID "message contents"`')


@client.command()
async def react(ctx):  # msg: discord.Message
    if ctx.author.id in admins:
        await ctx.message.add_reaction(":thumbs_up:")


# @react.error
# async def test_error(ctx):
#     await ctx.send('''Parameters: `.react Channel_ID "emoji"`
#                    Emoji example: `<:>`''')

client.run(sys.argv[1])