90 lines
4.2 KiB
Python
90 lines
4.2 KiB
Python
import discord
|
|
import html
|
|
import re
|
|
from botdiscord.translate import get_lang_mapping, get_flag_mapping, get_name_to_code, translate_text
|
|
from botdiscord.database import get_message, save_translation, get_cached_translation
|
|
|
|
class TranslationButton(discord.ui.Button):
|
|
def __init__(self, lang_name: str, lang_code: str, flag: str):
|
|
label = flag if flag else lang_name
|
|
# custom_id estático: esto es lo que Discord usa para vincular el clic
|
|
custom_id = f"btn_trans_{lang_code}"
|
|
super().__init__(label=label, style=discord.ButtonStyle.primary, custom_id=custom_id)
|
|
self.lang_code = lang_code
|
|
|
|
async def callback(self, interaction: discord.Interaction):
|
|
# Evita el "Interacción fallida" por timeout de 3s
|
|
await interaction.response.defer()
|
|
|
|
try:
|
|
if not interaction.message.reference:
|
|
await interaction.followup.send("⚠️ No se pudo encontrar el mensaje original.", ephemeral=True)
|
|
return
|
|
|
|
original_msg_id = interaction.message.reference.message_id
|
|
db_msg = get_message(original_msg_id)
|
|
|
|
if not db_msg:
|
|
await interaction.followup.send("⚠️ Mensaje no encontrado en la base de datos.", ephemeral=True)
|
|
return
|
|
|
|
text = db_msg['content']
|
|
mentions_map = db_msg['mentions_map']
|
|
|
|
# Recuperar o traducir
|
|
cached = get_cached_translation(original_msg_id, self.lang_code)
|
|
if cached:
|
|
translated = cached
|
|
else:
|
|
translated = await translate_text(text, self.lang_code)
|
|
save_translation(original_msg_id, self.lang_code, translated)
|
|
|
|
# Limpiar texto
|
|
translated = html.unescape(translated)
|
|
if mentions_map:
|
|
for placeholder, mention in mentions_map.items():
|
|
# El mapa ya viene con el formato <m0 />: "<@123>"
|
|
translated = translated.replace(placeholder, mention)
|
|
# Por si el traductor quitó el espacio: <m0/>
|
|
translated = translated.replace(placeholder.replace(" ", ""), mention)
|
|
|
|
# Editar el mensaje que contiene los botones
|
|
await interaction.edit_original_response(content=translated, view=self.view)
|
|
|
|
except Exception as e:
|
|
print(f"[ERROR UI] {e}")
|
|
await interaction.followup.send(f"❌ Error: {str(e)}", ephemeral=True)
|
|
|
|
class PersistentTranslationView(discord.ui.View):
|
|
def __init__(self):
|
|
super().__init__(timeout=None)
|
|
# Cargamos los botones para todos los idiomas habilitados en la config
|
|
# Esto asegura que cualquier botón de cualquier idioma sea reconocido
|
|
from botdiscord.config import get_languages
|
|
for lang in get_languages():
|
|
self.add_item(TranslationButton(lang['name'], lang['code'], lang.get('flag', '')))
|
|
|
|
class ConfigSelect(discord.ui.Select):
|
|
def __init__(self, guild_id: int, bot_type: str = "discord"):
|
|
from botdiscord.database import get_active_languages
|
|
lang_mapping = get_lang_mapping(bot_type)
|
|
flag_mapping = get_flag_mapping(bot_type)
|
|
active = get_active_languages(guild_id)
|
|
options = []
|
|
for name, code in lang_mapping.items():
|
|
flag = flag_mapping.get(code, "")
|
|
options.append(discord.SelectOption(label=f"{flag} {name}" if flag else name, value=name, default=(code in active)))
|
|
super().__init__(placeholder="Selecciona idiomas...", min_values=0, max_values=len(options), options=options)
|
|
|
|
async def callback(self, interaction: discord.Interaction):
|
|
from botdiscord.database import set_active_languages
|
|
lang_mapping = get_lang_mapping("discord")
|
|
selected_codes = [lang_mapping[val] for val in self.values]
|
|
set_active_languages(interaction.guild_id, selected_codes)
|
|
await interaction.response.send_message("Configuración actualizada.", ephemeral=True)
|
|
|
|
class ConfigView(discord.ui.View):
|
|
def __init__(self, guild_id: int, bot_type: str = "discord"):
|
|
super().__init__()
|
|
self.add_item(ConfigSelect(guild_id, bot_type))
|