93 lines
4.0 KiB
Python
93 lines
4.0 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
import aiohttp
|
|
from typing import Union
|
|
|
|
class TetoImageView(discord.ui.View):
|
|
def __init__(self, cog):
|
|
super().__init__(timeout=180.0)
|
|
self.cog = cog
|
|
|
|
@discord.ui.button(label='Show Another Image', style=discord.ButtonStyle.primary)
|
|
async def show_another_image(self, interaction: discord.Interaction, _: discord.ui.Button):
|
|
await self.cog.get_teto_image(interaction)
|
|
|
|
class TetoImageCog(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.teto_url = "https://slipstreamm.dev/teto"
|
|
self.footer_url = "https://slipstreamm.dev/teto-web"
|
|
|
|
async def fetch_teto_image(self):
|
|
"""Fetches a random Teto image and returns the URL."""
|
|
async with aiohttp.ClientSession() as session:
|
|
headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'}
|
|
async with session.get(self.teto_url, headers=headers, allow_redirects=False) as response:
|
|
if response.status == 302:
|
|
image_path = response.headers.get('Location')
|
|
if image_path:
|
|
return f"https://slipstreamm.dev{image_path}"
|
|
return None
|
|
|
|
@commands.hybrid_command(name='tetoimage', description='Get a random image of Kasane Teto.')
|
|
async def get_teto_image(self, ctx_or_interaction: Union[commands.Context, discord.Interaction]):
|
|
"""Gets a random image of Kasane Teto."""
|
|
# Determine if this is a Context or Interaction
|
|
is_interaction = isinstance(ctx_or_interaction, discord.Interaction)
|
|
|
|
# Handle typing indicator differently based on object type
|
|
if is_interaction:
|
|
await ctx_or_interaction.response.defer(thinking=True)
|
|
else:
|
|
async with ctx_or_interaction.typing():
|
|
pass
|
|
|
|
try:
|
|
image_url = await self.fetch_teto_image()
|
|
|
|
if image_url:
|
|
embed = discord.Embed(
|
|
title="Random Teto Image",
|
|
description=f"Website: {self.footer_url}",
|
|
color=discord.Color.red()
|
|
)
|
|
|
|
if image_url.startswith("http") or image_url.startswith("https"):
|
|
embed.set_image(url=image_url)
|
|
view = TetoImageView(self)
|
|
|
|
# Send response differently based on object type
|
|
if is_interaction:
|
|
if ctx_or_interaction.response.is_done():
|
|
await ctx_or_interaction.followup.send(embed=embed, view=view)
|
|
else:
|
|
await ctx_or_interaction.response.send_message(embed=embed, view=view)
|
|
else:
|
|
await ctx_or_interaction.send(embed=embed, view=view)
|
|
else:
|
|
error_msg = "The retrieved URL was not valid."
|
|
await self._send_error(ctx_or_interaction, error_msg)
|
|
else:
|
|
error_msg = "Could not get the image URL from the redirect."
|
|
await self._send_error(ctx_or_interaction, error_msg)
|
|
|
|
except aiohttp.ClientConnectorError as e:
|
|
error_msg = f"Could not connect to the image service: {e}"
|
|
await self._send_error(ctx_or_interaction, error_msg)
|
|
except Exception as e:
|
|
error_msg = f"An error occurred: {e}"
|
|
await self._send_error(ctx_or_interaction, error_msg)
|
|
|
|
async def _send_error(self, ctx_or_interaction, error_msg):
|
|
"""Helper method to send error messages based on context type."""
|
|
if isinstance(ctx_or_interaction, discord.Interaction):
|
|
if ctx_or_interaction.response.is_done():
|
|
await ctx_or_interaction.followup.send(error_msg)
|
|
else:
|
|
await ctx_or_interaction.response.send_message(error_msg)
|
|
else:
|
|
await ctx_or_interaction.send(error_msg)
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(TetoImageCog(bot))
|