disco/bot.py

from os import getenv
from datetime import datetime, timedelta
from zoneinfo import ZoneInfo
from typing import Iterator, Optional
from dotenv import load_dotenv
from asyncache import cached
from cachetools import TTLCache
from requests import get, post

import discord
from discord import app_commands
from discord.ext import tasks

from helper import nth_element

load_dotenv()

GUILD_ID = int(getenv("GUILD_ID"))
TOKEN = getenv("DISCORD_TOKEN")
WEATHER_API_KEY = getenv("WEATHER_API_KEY")
GENERAL_CHANNEL_ID = int(getenv("GENERAL_CHANNEL_ID"))

MY_GUILD = discord.Object(id=GUILD_ID)


tt = datetime.now(ZoneInfo("Europe/Paris"))
md = tt.replace(hour=0, minute=0, second=0, microsecond=0) + timedelta(days=1)
# remove 600ms to avoid the bot to send the message at 00:00:00.600
md = md - timedelta(milliseconds=600)
mid = md.timetz()


def embed_from_quote(my_quote: dict):
    embed = discord.Embed(title="Quote")
    embed.description = my_quote["content"]
    embed.set_author(name=my_quote["author"])
    return embed


class Dropdown(discord.ui.Select):
    def __init__(self):
        # Set the options that will be presented inside the dropdown
        options = [
            discord.SelectOption(
                label="Red", description="Your favourite colour is red", emoji="🟥"
            ),
            discord.SelectOption(
                label="Green", description="Your favourite colour is green", emoji="🟩"
            ),
            discord.SelectOption(
                label="Blue", description="Your favourite colour is blue", emoji="🟦"
            ),
        ]

        # The placeholder is what will be shown when no option is chosen
        # The min and max values indicate we can only pick one of the three options
        # The options parameter defines the dropdown options. We defined this above
        super().__init__(
            placeholder="Choose your favourite colour...",
            min_values=1,
            max_values=1,
            options=options,
        )

    async def callback(self, interaction: discord.Interaction):
        # Use the interaction object to send a response message containing
        # the user's favourite colour or choice. The self object refers to the
        # Select object, and the values attribute gets a list of the user's
        # selected options. We only want the first one.
        await interaction.response.send_message(
            f"Your favourite colour is {self.values[0]}"
        )


class DropdownView(discord.ui.View):
    def __init__(self):
        super().__init__()

        # Adds the dropdown to our view object.
        self.add_item(Dropdown())


class MyClient(discord.Client):
    def __init__(self, *, client_intents: discord.Intents):
        super().__init__(intents=client_intents)
        self.tree = app_commands.CommandTree(self)

    async def setup_hook(self):
        self.tree.copy_global_to(guild=MY_GUILD)
        self.quote_task.start()
        await self.tree.sync(guild=MY_GUILD)

    @tasks.loop(time=mid)
    async def quote_task(self):
        channel = self.get_channel(GENERAL_CHANNEL_ID)
        response = get("https://api.quotable.io/random?maxLength=230", verify=False)
        my_quote = response.json()
        await channel.send(embed=embed_from_quote(my_quote))

    @quote_task.before_loop
    async def before_quote_task(self):
        await self.wait_until_ready()


intents = discord.Intents.default()
client = MyClient(client_intents=intents)


@client.event
async def on_ready():
    print(f"Logged in as {client.user} (ID: {client.user.id})")
    print("------")


@client.tree.command()
async def hello(interaction: discord.Interaction):
    """Says hello!"""
    await interaction.response.send_message(f"Hi, {interaction.user.mention}")


@client.tree.command()
@app_commands.describe(
    first_value="The first value you want to add something to",
    second_value="The value you want to add to the first value",
)
async def add(interaction: discord.Interaction, first_value: int, second_value: int):
    """Adds two numbers together."""
    await interaction.response.send_message(
        f"{first_value} + {second_value} = {first_value + second_value}"
    )


@client.tree.command()
@app_commands.rename(text_to_send="text")
@app_commands.describe(text_to_send="Text to send in the current channel")
async def send(interaction: discord.Interaction, text_to_send: str):
    """Sends the text into the current channel."""
    await interaction.response.send_message(text_to_send)


@client.tree.command()
@app_commands.describe(
    member="The member you want to get the joined date from; defaults to the user who uses the "
    "command"
)
async def joined(
    interaction: discord.Interaction, member: Optional[discord.Member] = None
):
    """Says when a member joined."""
    member = member or interaction.user
    await interaction.response.send_message(
        f"{member} joined {discord.utils.format_dt(member.joined_at)}", ephemeral=True
    )


@client.tree.context_menu(name="Show Join Date")
async def show_join_date(interaction: discord.Interaction, member: discord.Member):
    await interaction.response.send_message(
        f"{member} joined at {discord.utils.format_dt(member.joined_at)}",
        ephemeral=True,
    )


@client.tree.context_menu(name="Translate")
async def translate(interaction: discord.Interaction, message: discord.Message):
    await interaction.response.defer(ephemeral=True)
    """Translate text to another language."""
    try:
        response = post(
            "https://translate.argosopentech.com/translate",
            json={"q": message.content, "source": "fr", "target": "en"},
            timeout=30,
        )
        response.raise_for_status()
    except Exception as e:
        await interaction.followup.send(f"An error occurred: {e}", ephemeral=True)
        return
    try:
        json = response.json()
        translated_text = json["translatedText"]
    except Exception as e:
        await interaction.followup.send(f"An error occurred: {e}", ephemeral=True)
        return
    await interaction.followup.send(
        f"Translated text: {translated_text}", ephemeral=True
    )


@client.tree.command()
async def quote(interaction: discord.Interaction):
    """Sends a random quote."""
    response = get("https://api.quotable.io/random?maxLength=230", verify=False)
    my_quote = response.json()
    await interaction.response.send_message(embed=embed_from_quote(my_quote))


@cached(cache=TTLCache(maxsize=1024, ttl=600))
async def get_weather_json(city: str):
    try:
        weather_response = get(
            f"https://api.weatherapi.com/v1/current.json?key={WEATHER_API_KEY}&q={city}&aqi=no"
        )
        weather_response.raise_for_status()
        forecast_response = get(
            f"https://api.weatherapi.com/v1/forecast.json?key={WEATHER_API_KEY}&q={city}&days=1&aqi=no&alerts=no"
        )
        forecast_response.raise_for_status()
    except Exception as e:
        raise e
    return weather_response.json(), forecast_response.json()


@client.tree.command()
@app_commands.describe(city="The city you want to get the weather for")
async def weather(interaction: discord.Interaction, city: str):
    """Get the current weather"""
    try:
        weather_response, forecast_response = await get_weather_json(city)
    except Exception as _e:
        await interaction.response.send_message(
            f'Error: the city "{city}" was not found', ephemeral=True
        )
        return
    current_temp = weather_response["current"]["temp_c"]
    forecast = forecast_response["forecast"]["forecastday"][0]
    forecast_min = forecast["day"]["mintemp_c"]
    forecast_max = forecast["day"]["maxtemp_c"]
    embed = discord.Embed(title="Weather")
    embed.description = (
        f"Current temperature: {current_temp}°C\n"
        f"Forecast: Min: {forecast_min}°C, Max: {forecast_max}°C "
    )
    await interaction.response.send_message(embed=embed)


# A command tha tell the time until midnight
@client.tree.command()
async def midnight(interaction: discord.Interaction):
    """Sends the time until midnight."""
    paris_dt = datetime.now(ZoneInfo("Europe/Paris"))
    midnight = paris_dt.replace(hour=0, minute=0, second=0, microsecond=0) + timedelta(
        days=1
    )
    time_until_midnight = midnight - paris_dt
    await interaction.response.send_message(
        f"Time until midnight: {time_until_midnight}"
    )


# A command that say what the best programming language is
@client.tree.command()
async def best_language(interaction: discord.Interaction):
    """Sends the best programming language."""
    await interaction.response.send_message("The best programming language is Rust")


@client.tree.command()
async def color(interaction: discord.Interaction):
    """Sends a random color."""
    await interaction.response.send_message(view=DropdownView())


def fibo_gen() -> Iterator[int]:
    a, b = 0, 1
    while True:
        yield a
        a, b = b, a + b


def fib(n: int) -> int:
    return nth_element(fibo_gen(), n)


# @client.tree.command()
# async def fibonacci(interaction: discord.Interaction, n: int):
#     """Sends the nth fibonacci number."""
#     await interaction.response.send_message(f'{n}th fibonacci number is {fib(n)}')


@client.event
async def on_member_join(member: discord.Member):
    guild = member.guild
    channel = client.get_channel(GENERAL_CHANNEL_ID)
    await channel.send(f"Welcome {member.mention} to {guild.name}!")


if __name__ == "__main__":
    client.run(TOKEN)