"""
title: Discord Reader
version: 0.0.1
changelog:
- 0.0.1 - Initial build.
"""

import requests
import json
import time
import os
import random
from typing import Awaitable, Callable
from pydantic import BaseModel, Field
from requests.models import Response


def retrieve_manymessages(channelid, multiples50=1):
    # this function is a stub. didnt add it in
    before = ""
    data = []
    for i in range(multiples50):

        r = retrieve_messages(channelid, before)
        json = r.json()
        data.append(json)
        if len(json) < 50:  # stop if we run out of messages
            break
        else:
            last_message = json[49]
            before = last_message["id"]

        time.sleep(random.uniform(1, 3))

    return data


class Tools:
    def __init__(self):
        pass

    class Valves(BaseModel):
        USER_AGENT: str = Field(
            default="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36",
            description="The user agent to use when making requests to Reddit.",
        )
        authorization: str = Field(
            default="",
            description="open discord in browser -> f12 -> network -> record -> visit channel -> find messages/?limit=50 -> find authentication field",
        )

    async def get_messages(
        self,
        channelid: int,
        __event_emitter__: Callable[[dict], Awaitable[None]],
        __user__: dict = {},
    ) -> str:
        """
        Get the latest messages from a discord channel, as an array of JSON objects with the following properties: 'content','id','author','timestamp','edited_timestamp'
        :param channelid: The channelid to get the latest messages from.
        :return: A list of messages with the previously mentioned properties, or an error message.
        """
        headers = {
            "User-Agent": __user__["valves"].USER_AGENT,
            "authorization": __user__["valves"].authorization,
            "Referer": "https://discord.com/channels/@me/",
            "accept-encoding": "gzip, deflate, br, zstd",
            "accept-language": "en-US,en;q=0.9",
            "referer": "https://discord.com/channels/@me/",
            "sec-ch-ua": '"Chromium";v="128", "Not;A=Brand";v="24", "Google Chrome";v="128"',
            "sec-ch-ua-mobile": "?0",
            "sec-ch-ua-platform": '"iOS"',
            "sec-fetch-mode": "navigate",
            "sec-fetch-site": "none",
            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/74.0.3729.169 Safari/537.3",
        }

        await __event_emitter__(
            {
                "data": {
                    "description": f"Starting retrieval for channelID {channelid}.",
                    "status": "in_progress",
                    "done": False,
                },
                "type": "status",
            }
        )

        if channelid == 0:
            await __event_emitter__(
                {
                    "data": {
                        "description": f"Error: No channelid provided.",
                        "status": "complete",
                        "done": True,
                    },
                    "type": "status",
                }
            )
            return "Error: No channelid provided"

        try:

            response = requests.get(
                f"https://discord.com/api/v9/channels/{channelid}/messages?limit=50",
                headers=headers,
            )

            if not response.ok:
                await __event_emitter__(
                    {
                        "data": {
                            "description": f"Error: Failed to retrieve messages: {response.status_code}.",
                            "status": "complete",
                            "done": True,
                        },
                        "type": "status",
                    }
                )
                return f"Error: {response.status_code}"
            else:
                output = response.json()
                await __event_emitter__(
                    {
                        "data": {
                            "description": f"Retrieved {len(output)} messages from {channelid}.",
                            "status": "complete",
                            "done": True,
                        },
                        "type": "status",
                    }
                )
                return json.dumps(output)
        except Exception as e:
            await __event_emitter__(
                {
                    "data": {
                        "description": f"Failed to retrieve any messages from {channelid} Exception is: {e}.",
                        "status": "complete",
                        "done": True,
                    },
                    "type": "status",
                }
            )
            return f"Error: {e}"