Syntrel/cogs/media/tweety.py

# The API used in the tweety command is made by me and can be found here:
# https://github.com/neoarz/tweety-api
# I made this out of spite since i couldnt find any free APIs for this
# Its serverless and hosted on Vercel :)

import os
import tempfile
import discord
from discord.ext import commands
import aiohttp
from datetime import datetime
from typing import Optional
import pytz


class TweetyHelpView(discord.ui.View):
    """This is the help view for the slash command cuz only the pinging and prefix versions can use this command"""
    
    def __init__(self, user_id: int, bot):
        super().__init__(timeout=180)
        self.user_id = user_id
        self.bot = bot
        self.current_page = 0
        self.pages = [
            {
                "title": "Tweety (Method 1)",
                "description": "Use the prefix command `.media tweety` while replying to a message.",
                "gif_url": "https://yes.nighty.works/raw/VrKX1L.gif", 
                "fields": [
                    {"name": "How to use", "value": "1. Reply to any message\n2. Type `.media tweety`\n3. Use the buttons to customize!", "inline": False},
                ]
            },
            {
                "title": "Tweety (Method 2)",
                "description": f"Mention <@{bot.user.id}> with `tweety` while replying to a message.",
                "gif_url": "https://yes.nighty.works/raw/9XEe9j.gif", 
                "fields": [
                    {"name": "How to use", "value": f"1. Reply to any message\n2. Type `<@{bot.user.id}> tweety`\n3. Use the buttons to customize!", "inline": False},
                ]
            }
        ]
        self.update_buttons()
    
    def create_embed(self):
        page = self.pages[self.current_page]
        embed = discord.Embed(
            title=page["title"],
            description=page["description"],
            color=0x7289DA,
        )
        embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
        
        for field in page["fields"]:
            embed.add_field(name=field["name"], value=field["value"], inline=field["inline"])
        
        embed.set_image(url=page["gif_url"])
        embed.set_footer(text=f"Page {self.current_page + 1}/{len(self.pages)}")
        
        return embed
    
    def update_buttons(self):
        self.clear_items()
        
        prev_button = discord.ui.Button(
            label="Prev",
            style=discord.ButtonStyle.secondary,
            emoji=discord.PartialEmoji(name="left", id=1420240344926126090),
            disabled=self.current_page == 0
        )
        prev_button.callback = self.previous_page
        self.add_item(prev_button)
        
        next_button = discord.ui.Button(
            label="Next",
            style=discord.ButtonStyle.secondary,
            emoji=discord.PartialEmoji(name="right", id=1420240334100627456),
            disabled=self.current_page == len(self.pages) - 1
        )
        next_button.callback = self.next_page
        self.add_item(next_button)
    
    async def previous_page(self, interaction: discord.Interaction):
        if interaction.user.id != self.user_id:
            await interaction.response.send_message("You can't control someone else's help menu!", ephemeral=True)
            return
        
        if self.current_page > 0:
            self.current_page -= 1
            self.update_buttons()
            await interaction.response.edit_message(embed=self.create_embed(), view=self)
        else:
            await interaction.response.defer()
    
    async def next_page(self, interaction: discord.Interaction):
        if interaction.user.id != self.user_id:
            await interaction.response.send_message("You can't control someone else's help menu!", ephemeral=True)
            return
        
        if self.current_page < len(self.pages) - 1:
            self.current_page += 1
            self.update_buttons()
            await interaction.response.edit_message(embed=self.create_embed(), view=self)
        else:
            await interaction.response.defer()
    
    async def on_timeout(self):
        for item in self.children:
            item.disabled = True


class TweetyView(discord.ui.View):
    """API for Tweety is hosted on Vercel and made by me :) github can be found here: https://github.com/neoarz/tweety-api"""

    def __init__(self, author_id: int, original_message, tweet_data: dict, api_url: str, image_message: Optional[discord.Message] = None):
        super().__init__(timeout=300)
        self.author_id = author_id
        self.original_message = original_message
        self.tweet_data = tweet_data
        self.api_url = api_url
        self.is_dark = tweet_data.get("dark", False)
        self.is_verified = tweet_data.get("verified", False)
        self.image_message = image_message

        self.update_button_styles()
    
    def update_button_styles(self):
        self.clear_items()
        
        dark_button = discord.ui.Button(
            label="Dark Mode" if self.is_dark else "Light Mode",
            style=discord.ButtonStyle.primary if self.is_dark else discord.ButtonStyle.secondary,
            emoji=discord.PartialEmoji(name="darkmode", id=1425165393751965884),
            custom_id="toggle_dark"
        )
        dark_button.callback = self.toggle_dark_callback
        self.add_item(dark_button)
        
        verified_button = discord.ui.Button(
            label="Verified",
            style=discord.ButtonStyle.primary if self.is_verified else discord.ButtonStyle.secondary,
            emoji=discord.PartialEmoji(name="TwitterVerifiedBadge", id=1425165432142172392),
            custom_id="toggle_verified"
        )
        verified_button.callback = self.toggle_verified_callback
        self.add_item(verified_button)
    
    async def regenerate_tweet(self, interaction: discord.Interaction):
        """Regenerate only the image message with current settings"""
        await interaction.response.defer()

        try:
            async with aiohttp.ClientSession() as session:
                async with session.post(
                    f"{self.api_url}/api/render",
                    json=self.tweet_data,
                    headers={"Content-Type": "application/json"}
                ) as response:
                    
                    if response.status != 200:
                        error_text = await response.text()
                        embed = discord.Embed(
                            title="Error",
                            description=f"API Error ({response.status}): {error_text}",
                            color=0xE02B2B,
                        )
                        embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                        await interaction.followup.send(embed=embed, ephemeral=True)
                        return
                    
                    image_data = await response.read()
                    
                    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
                        temp_file.write(image_data)
                        temp_file_path = temp_file.name
                    
                    with open(temp_file_path, 'rb') as f:
                        author_name = self.original_message.author.name
                        filename = f"tweet_{author_name}_{int(datetime.now().timestamp())}.png"
                        file = discord.File(
                            f, 
                            filename=filename
                        )

                        self.update_button_styles()

                        if self.image_message is not None:
                            await self.image_message.edit(attachments=[file], view=self)
                        else:
                            await interaction.followup.send(file=file, view=self)
                    
                    os.remove(temp_file_path)
                    
        except Exception as e:
            embed = discord.Embed(
                title="Error",
                description="Error regenerating tweet image",
                color=0xE02B2B,
            )
            embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
            await interaction.followup.send(embed=embed, ephemeral=True)
    
    async def _check_author(self, interaction: discord.Interaction) -> bool:
        """Check if user is authorized to modify the tweet"""
        if interaction.user.id != self.author_id:
            embed = discord.Embed(
                title="Error",
                description="You can't modify someone else's tweet!",
                color=0xE02B2B,
            )
            embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
            await interaction.response.send_message(embed=embed, ephemeral=True)
            return False
        return True
    
    async def toggle_dark_callback(self, interaction: discord.Interaction):
        """Handle dark mode toggle button click"""
        if not await self._check_author(interaction):
            return
        self.is_dark = not self.is_dark
        self.tweet_data["dark"] = self.is_dark
        await self.regenerate_tweet(interaction)
    
    async def toggle_verified_callback(self, interaction: discord.Interaction):
        """Handle verified toggle button click"""
        if not await self._check_author(interaction):
            return
        self.is_verified = not self.is_verified
        self.tweet_data["verified"] = self.is_verified
        await self.regenerate_tweet(interaction)
    
    async def on_timeout(self):
        """Disable buttons when view times out"""
        for item in self.children:
            item.disabled = True

def tweety_command():
    @commands.hybrid_command(
        name="tweety",
        description="Convert a replied message to a tweet image."
    )
    @commands.cooldown(1, 10, commands.BucketType.user)
    async def tweety(self, context):
        if hasattr(context, "interaction") and context.interaction:
            view = TweetyHelpView(user_id=context.author.id, bot=self.bot)
            embed = view.create_embed()
            await context.send(embed=embed, view=view, ephemeral=True)
            return
        
        if not context.message.reference or not context.message.reference.message_id:
            embed = discord.Embed(
                title="Error",
                description="You must reply to a message to use this command!",
                color=0xE02B2B,
            )
            embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
            await context.send(embed=embed)
            return
        
        try:
            original_message = await context.channel.fetch_message(context.message.reference.message_id)
            
            if original_message.author.bot:
                embed = discord.Embed(
                    title="Error",
                    description="Cannot convert bot messages to tweets!",
                    color=0xE02B2B,
                )
                embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                await context.send(embed=embed)
                return

            processing_embed = discord.Embed(
                title="Tweet Generator (Processing)",
                description="<a:mariospin:1423677027013103709> Generating tweet... This may take a moment.",
                color=0x7289DA,
            )
            processing_embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
            processing_msg = await context.send(embed=processing_embed)

            author = original_message.author
            display_name = author.display_name or author.name
            username = f"@{author.name}"
            avatar_url = str(author.avatar.url) if author.avatar else str(author.default_avatar.url)
            message_text = original_message.content
            
            image_url = None
            if original_message.attachments:
                for attachment in original_message.attachments:
                    if any(attachment.filename.lower().endswith(ext) for ext in ['.png', '.jpg', '.jpeg', '.gif', '.webp']):
                        raw_url = attachment.url
                        if 'cdn.discordapp.com' in raw_url:
                            media_url = raw_url.replace('cdn.discordapp.com', 'media.discordapp.net')
                            if '?' not in media_url:
                                media_url += f"?width={attachment.width}&height={attachment.height}"
                            elif 'width=' not in media_url and attachment.width:
                                media_url += f"&width={attachment.width}&height={attachment.height}"
                            image_url = media_url
                        else:
                            image_url = raw_url
                        break
            
            if not message_text.strip() and not image_url:
                await processing_msg.delete()
                embed = discord.Embed(
                    title="Error",
                    description="Message must have either text content or an image/GIF to convert!",
                    color=0xE02B2B,
                )
                embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                await context.send(embed=embed)
                return
            
            ny_tz = pytz.timezone('America/New_York')
            msg_time_ny = original_message.created_at.astimezone(ny_tz)
            timestamp = msg_time_ny.strftime("%I:%M %p · %b %d, %Y").replace(" 0", " ")
            tweet_data = {
                "name": display_name[:50],
                "handle": username[:20],
                "text": message_text[:300],
                "avatar": avatar_url,
                "timestamp": timestamp,
                "verified": False,
                "dark": False
            }
            
            if image_url:
                tweet_data["image"] = image_url
            
            API_BASE_URL = "https://tweety-api.vercel.app"
            
            async with aiohttp.ClientSession() as session:
                try:
                    async with session.post(
                        f"{API_BASE_URL}/api/render",
                        json=tweet_data,
                        headers={"Content-Type": "application/json"}
                    ) as response:
                        
                        if response.status != 200:
                            await processing_msg.delete()
                            error_text = await response.text()
                            embed = discord.Embed(
                                title="Error",
                                description=f"API Error ({response.status}): {error_text}",
                                color=0xE02B2B,
                            )
                            embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                            await context.send(embed=embed)
                            return
                        
                        image_data = await response.read()
                        
                        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
                            temp_file.write(image_data)
                            temp_file_path = temp_file.name

                        await processing_msg.delete()
                        
                        with open(temp_file_path, 'rb') as f:
                            file = discord.File(f, filename=f"tweet_{author.name}_{int(datetime.now().timestamp())}.png")
                            embed = discord.Embed(
                                title="Tweet Generated",
                                color=0x7289DA,
                            )
                            embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                            embed.set_footer(
                                text=f"Requested by {context.author.name}",
                                icon_url=context.author.display_avatar.url,
                            )

                            view = TweetyView(
                                author_id=context.author.id,
                                original_message=original_message,
                                tweet_data=tweet_data,
                                api_url=API_BASE_URL
                            )

                            await context.send(embed=embed)
                            image_message = await context.send(file=file, view=view)
                            view.image_message = image_message

                        os.remove(temp_file_path)
                        
                except aiohttp.ClientError:
                    await processing_msg.delete()
                    embed = discord.Embed(
                        title="Error",
                        description=f"Connection error: Could not reach tweet API",
                        color=0xE02B2B,
                    )
                    embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                    await context.send(embed=embed)
                except Exception:
                    await processing_msg.delete()
                    embed = discord.Embed(
                        title="Error",
                        description="Error generating tweet image",
                        color=0xE02B2B,
                    )
                    embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
                    await context.send(embed=embed)

        except Exception:
            embed = discord.Embed(
                title="Error",
                description="Error processing the message!",
                color=0xE02B2B,
            )
            embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
            await context.send(embed=embed)
    
    return tweety