from telethon import TelegramClient, events
import asyncio
import logging

# Configuración de logging
logging.basicConfig(
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    level=logging.INFO
)
logger = logging.getLogger(__name__)

# ===================== CONFIGURACIÓN =====================
# Obtén estos valores de https://my.telegram.org
API_ID = 'TU_API_ID'
API_HASH = 'TU_API_HASH'
BOT_TOKEN = 'TU_BOT_TOKEN'  # Obtén esto de @BotFather

# IDs de los canales (pueden ser username o ID numérico)
SOURCE_CHANNEL = '@canal_origen'  # O -1001234567890
TARGET_CHANNEL = '@canal_destino'  # O -1001234567890

# Extensiones de archivo a repostear
ALLOWED_EXTENSIONS = ('.zip', '.rar', '.7z', '.tar', '.gz', '.3mf', '.stl')
# =========================================================


# Crear el cliente del bot
bot = TelegramClient('bot_session', API_ID, API_HASH)


@bot.on(events.NewMessage(chats=SOURCE_CHANNEL))
async def repost_handler(event):
    """
    Maneja los nuevos mensajes del canal origen y los repostea al destino
    """
    try:
        message = event.message
        
        # Verificar si el mensaje tiene un archivo
        if message.media and message.document:
            # Obtener el nombre del archivo
            file_name = None
            for attribute in message.document.attributes:
                if hasattr(attribute, 'file_name'):
                    file_name = attribute.file_name
                    break
            
            # Verificar si es una extensión permitida
            if file_name and any(file_name.lower().endswith(ext) for ext in ALLOWED_EXTENSIONS):
                logger.info(f"📥 Reposteando archivo: {file_name}")
                
                # Repostear el mensaje completo (archivo + caption si existe)
                await bot.forward_messages(
                    entity=TARGET_CHANNEL,
                    messages=message.id,
                    from_peer=SOURCE_CHANNEL
                )
                
                logger.info(f"✅ Archivo reposteado exitosamente: {file_name}")
            else:
                logger.debug(f"⏭️ Archivo ignorado (extensión no permitida): {file_name}")
        else:
            logger.debug("⏭️ Mensaje sin archivo, ignorado")
            
    except Exception as e:
        logger.error(f"❌ Error al repostear mensaje: {e}")


async def main():
    """
    Función principal para iniciar el bot
    """
    logger.info("🤖 Iniciando bot...")
    
    # Conectar con el token del bot
    await bot.start(bot_token=BOT_TOKEN)
    
    # Obtener información del bot
    me = await bot.get_me()
    logger.info(f"✅ Bot conectado: @{me.username}")
    logger.info(f"📡 Escuchando mensajes en: {SOURCE_CHANNEL}")
    logger.info(f"📤 Reposteando a: {TARGET_CHANNEL}")
    logger.info(f"📁 Extensiones permitidas: {', '.join(ALLOWED_EXTENSIONS)}")
    logger.info("🟢 Bot funcionando. Presiona Ctrl+C para detener.")
    
    # Mantener el bot corriendo
    await bot.run_until_disconnected()


if __name__ == '__main__':
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        logger.info("🛑 Bot detenido por el usuario")