mirror of
https://github.com/neoarz/Syntrel.git
synced 2025-12-25 11:40:12 +01:00
7419ec5f59
Simplifies the tweety command by removing the 'verified' and 'theme' arguments from command signatures and related logic. Defaults for these options are now set internally, and toggling is handled via UI elements instead of command parameters.
439 lines
20 KiB
Python
439 lines
20 KiB
Python
import asyncio
|
|
import os
|
|
import tempfile
|
|
import discord
|
|
from discord.ext import commands
|
|
from discord import app_commands
|
|
import aiohttp
|
|
import io
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
|
|
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):
|
|
"""Update button styles to reflect current state"""
|
|
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 toggle_dark_callback(self, interaction: discord.Interaction):
|
|
"""Handle dark mode toggle button click"""
|
|
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
|
|
|
|
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 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
|
|
|
|
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
|
|
|
|
try:
|
|
pass
|
|
except:
|
|
pass
|
|
|
|
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):
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
try:
|
|
embed = discord.Embed(
|
|
title="Tweety",
|
|
description=(
|
|
"Use the prefix command: `.media tweety`\n"
|
|
f"Or reply to a message with: <@{self.bot.user.id}> tweety"
|
|
),
|
|
color=0x7289DA,
|
|
)
|
|
embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
else:
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
except Exception:
|
|
pass
|
|
return
|
|
|
|
# Default to light mode, non-verified (buttons will allow toggling)
|
|
verified_bool = False
|
|
theme_bool = False
|
|
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
else:
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await context.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
original_message = await context.channel.fetch_message(context.message.reference.message_id)
|
|
|
|
try:
|
|
if not original_message:
|
|
embed = discord.Embed(
|
|
title="Error",
|
|
description="Could not find the original message!",
|
|
color=0xE02B2B,
|
|
)
|
|
embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
else:
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await context.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
else:
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await context.send(embed=embed, ephemeral=True)
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(embed=processing_embed, ephemeral=True)
|
|
else:
|
|
await interaction.followup.send(embed=processing_embed, ephemeral=True)
|
|
else:
|
|
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:
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
try:
|
|
await interaction.delete_original_response()
|
|
except:
|
|
pass
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await processing_msg.delete()
|
|
await context.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
msg_time = original_message.created_at
|
|
timestamp = msg_time.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": verified_bool,
|
|
"dark": theme_bool
|
|
}
|
|
|
|
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:
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
try:
|
|
await interaction.delete_original_response()
|
|
except:
|
|
pass
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await processing_msg.delete()
|
|
await context.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:
|
|
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
|
|
)
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
embed_message = await context.channel.send(embed=embed)
|
|
image_message = await context.channel.send(file=file, view=view)
|
|
view.image_message = image_message
|
|
try:
|
|
await interaction.delete_original_response()
|
|
except:
|
|
pass
|
|
else:
|
|
await processing_msg.delete()
|
|
embed_message = await context.channel.send(embed=embed)
|
|
image_message = await context.channel.send(file=file, view=view)
|
|
view.image_message = image_message
|
|
|
|
os.remove(temp_file_path)
|
|
|
|
except aiohttp.ClientError as e:
|
|
embed = discord.Embed(
|
|
title="Error",
|
|
description=f"Connection error: Could not reach tweet API at {API_BASE_URL}",
|
|
color=0xE02B2B,
|
|
)
|
|
embed.set_author(name="Media", icon_url="https://yes.nighty.works/raw/y5SEZ9.webp")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
try:
|
|
await interaction.delete_original_response()
|
|
except:
|
|
pass
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await processing_msg.delete()
|
|
await context.send(embed=embed, ephemeral=True)
|
|
except Exception as e:
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
try:
|
|
await interaction.delete_original_response()
|
|
except:
|
|
pass
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await processing_msg.delete()
|
|
await context.send(embed=embed, ephemeral=True)
|
|
|
|
except Exception as e:
|
|
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")
|
|
|
|
interaction = getattr(context, "interaction", None)
|
|
if interaction is not None:
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
else:
|
|
await interaction.followup.send(embed=embed, ephemeral=True)
|
|
else:
|
|
await context.send(embed=embed, ephemeral=True)
|
|
|
|
return tweety
|