206 lines
9.1 KiB
Python
206 lines
9.1 KiB
Python
# this is where telegram-to-discord forwarding lives
|
||
import io
|
||
import json
|
||
import aiohttp
|
||
from telegram.ext import Application, MessageHandler, filters
|
||
|
||
class TelegramForwarder:
|
||
def __init__(self, token):
|
||
self.token = token
|
||
self.routes = {}
|
||
self._app = None
|
||
|
||
def add_route(self, group_id, webhook_url):
|
||
self.routes[group_id] = webhook_url
|
||
|
||
async def start(self):
|
||
self._app = Application.builder().token(self.token).build()
|
||
self._app.add_handler(MessageHandler(filters.ALL, self._on_message))
|
||
|
||
await self._app.initialize()
|
||
await self._app.start()
|
||
await self._app.updater.start_polling(drop_pending_updates=True)
|
||
print(f'✅ telegram forwarder started — watching {len(self.routes)} groups')
|
||
|
||
async def stop(self):
|
||
if self._app:
|
||
await self._app.updater.stop()
|
||
await self._app.stop()
|
||
await self._app.shutdown()
|
||
print('telegram forwarder stopped')
|
||
|
||
# internal helpers
|
||
def _sender_name(self, update):
|
||
user = update.effective_user
|
||
if user:
|
||
name = user.first_name or ''
|
||
if user.last_name:
|
||
name += f' {user.last_name}'
|
||
return name.strip() or user.username or 'unknown'
|
||
|
||
# channel posts have no user — use chat title
|
||
chat = update.effective_chat
|
||
if chat and chat.title:
|
||
return chat.title
|
||
return 'unknown'
|
||
|
||
def _forward_prefix(self, message):
|
||
if message.forward_origin:
|
||
origin = message.forward_origin
|
||
|
||
# user forward
|
||
if hasattr(origin, 'sender_user') and origin.sender_user:
|
||
u = origin.sender_user
|
||
name = u.first_name or ''
|
||
if u.last_name:
|
||
name += f' {u.last_name}'
|
||
return f'↩️ Forwarded from **{name.strip() or u.username or "unknown"}**:\n'
|
||
|
||
# channel forward
|
||
if hasattr(origin, 'chat') and origin.chat:
|
||
return f'↩️ Forwarded from **{origin.chat.title or "unknown channel"}**:\n'
|
||
|
||
# hidden user
|
||
if hasattr(origin, 'sender_user_name') and origin.sender_user_name:
|
||
return f'↩️ Forwarded from **{origin.sender_user_name}**:\n'
|
||
|
||
return '↩️ Forwarded:\n'
|
||
return ''
|
||
|
||
async def _download_file(self, file_id):
|
||
tg_file = await self._app.bot.get_file(file_id)
|
||
buf = io.BytesIO()
|
||
await tg_file.download_to_memory(buf)
|
||
buf.seek(0)
|
||
filename = tg_file.file_path.split('/')[-1] if tg_file.file_path else 'file'
|
||
return buf.read(), filename
|
||
|
||
async def _send_to_webhook(self, webhook_url, username, content='', file_data=None, filename=None):
|
||
async with aiohttp.ClientSession() as session:
|
||
data = aiohttp.FormData()
|
||
payload = {'username': username}
|
||
if content:
|
||
payload['content'] = content[:2000]
|
||
|
||
if file_data and filename:
|
||
data.add_field('payload_json', json.dumps(payload), content_type='application/json')
|
||
data.add_field('file', file_data, filename=filename)
|
||
else:
|
||
async with session.post(webhook_url, json=payload) as resp:
|
||
if resp.status not in (200, 204):
|
||
print(f'webhook post failed: {resp.status} {await resp.text()}')
|
||
return
|
||
|
||
async with session.post(webhook_url, data=data) as resp:
|
||
if resp.status not in (200, 204):
|
||
print(f'webhook post failed: {resp.status} {await resp.text()}')
|
||
|
||
# main message handler
|
||
async def _on_message(self, update, context):
|
||
message = update.effective_message
|
||
if not message:
|
||
return
|
||
|
||
chat_id = update.effective_chat.id if update.effective_chat else None
|
||
if chat_id not in self.routes:
|
||
return
|
||
|
||
webhook_url = self.routes[chat_id]
|
||
sender = self._sender_name(update)
|
||
prefix = self._forward_prefix(message)
|
||
caption = message.caption or ''
|
||
|
||
try:
|
||
if message.photo:
|
||
photo = message.photo[-1]
|
||
file_data, filename = await self._download_file(photo.file_id)
|
||
text = f'{prefix}{caption}'.strip()
|
||
await self._send_to_webhook(webhook_url, sender, text, file_data, filename)
|
||
|
||
elif message.video:
|
||
file_data, filename = await self._download_file(message.video.file_id)
|
||
text = f'{prefix}{caption}'.strip()
|
||
await self._send_to_webhook(webhook_url, sender, text, file_data, filename)
|
||
|
||
elif message.document:
|
||
file_data, filename = await self._download_file(message.document.file_id)
|
||
text = f'{prefix}{caption}'.strip()
|
||
await self._send_to_webhook(webhook_url, sender, text, file_data, filename)
|
||
|
||
elif message.animation:
|
||
file_data, filename = await self._download_file(message.animation.file_id)
|
||
|
||
# convert mp4 animation to true gif so discord auto-loops it
|
||
try:
|
||
import asyncio
|
||
import tempfile
|
||
import os
|
||
|
||
with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_mp4:
|
||
temp_mp4.write(file_data)
|
||
temp_mp4_path = temp_mp4.name
|
||
|
||
temp_gif_path = temp_mp4_path + '.gif'
|
||
|
||
# Use a palette to preserve quality, but drop fps to 15 and scale width to 320 to keep file size low.
|
||
# Discord webhooks reject files > 25MB (sometimes > 8MB), and GIFs are massively unoptimized compared to MP4s.
|
||
process = await asyncio.create_subprocess_exec(
|
||
'ffmpeg', '-y', '-i', temp_mp4_path,
|
||
'-filter_complex', 'fps=15,scale=320:-1:flags=lanczos,split[s0][s1];[s0]palettegen[p];[s1][p]paletteuse',
|
||
'-loop', '0', temp_gif_path,
|
||
stdout=asyncio.subprocess.DEVNULL,
|
||
stderr=asyncio.subprocess.PIPE
|
||
)
|
||
|
||
# Add a 30 second timeout so the bot doesn't hang if ffmpeg gets stuck
|
||
try:
|
||
_, stderr_data = await asyncio.wait_for(process.communicate(), timeout=30.0)
|
||
except asyncio.TimeoutError:
|
||
process.kill()
|
||
_, stderr_data = await process.communicate()
|
||
print("ffmpeg gif conversion timed out after 30 seconds.", flush=True)
|
||
|
||
if process.returncode == 0 and os.path.exists(temp_gif_path):
|
||
gif_size = os.path.getsize(temp_gif_path)
|
||
# Fallback to MP4 if GIF is > 10 MiB, to prevent Discord from rejecting the webhook post
|
||
if gif_size < 10 * 1024 * 1024:
|
||
with open(temp_gif_path, 'rb') as f:
|
||
file_data = f.read()
|
||
filename = filename.rsplit('.', 1)[0] + '.gif'
|
||
print(f"ffmpeg conversion succeeded: GIF size {gif_size / 1024 / 1024:.2f} MiB", flush=True)
|
||
else:
|
||
print(f"GIF was too large for Discord webhook ({gif_size / 1024 / 1024:.2f} MiB). Falling back to MP4.", flush=True)
|
||
else:
|
||
print(f"ffmpeg failed with code {process.returncode}. Stderr: {stderr_data.decode('utf-8', errors='ignore')}", flush=True)
|
||
|
||
try:
|
||
if os.path.exists(temp_mp4_path): os.remove(temp_mp4_path)
|
||
if os.path.exists(temp_gif_path): os.remove(temp_gif_path)
|
||
except Exception:
|
||
pass
|
||
except Exception as e:
|
||
print(f'ffmpeg gif conversion threw an exception: {e}', flush=True)
|
||
|
||
text = f'{prefix}{caption}'.strip()
|
||
await self._send_to_webhook(webhook_url, sender, text, file_data, filename)
|
||
|
||
elif message.sticker:
|
||
sticker_text = message.sticker.emoji or '(sticker)'
|
||
text = f'{prefix}{sticker_text}'
|
||
try:
|
||
file_data, filename = await self._download_file(message.sticker.file_id)
|
||
await self._send_to_webhook(webhook_url, sender, text, file_data, filename)
|
||
except Exception:
|
||
await self._send_to_webhook(webhook_url, sender, text)
|
||
|
||
elif message.text:
|
||
text = f'{prefix}{message.text}'
|
||
await self._send_to_webhook(webhook_url, sender, text)
|
||
|
||
else:
|
||
text = f'{prefix}*(unsupported message type)*'
|
||
await self._send_to_webhook(webhook_url, sender, text)
|
||
|
||
except Exception as e:
|
||
print(f'failed to forward message from {sender}: {e}')
|