| import asyncio
|
| import os
|
| import tempfile
|
| from pyrogram import Client, filters
|
| from pyrogram.types import Message
|
| from bot.config import Config
|
|
|
|
|
| async def background_wait(process, message: Message):
|
| stdout, stderr = await process.communicate()
|
| output = (stdout or stderr).decode(errors="ignore") or "Proses selesai tanpa output."
|
|
|
| if len(output) < 3500:
|
| await message.reply_text(
|
| f"✅ Proses selesai\n\n{output}",
|
| parse_mode=None,
|
| reply_to_message_id=message.id
|
| )
|
| else:
|
| filename = None
|
| try:
|
| with tempfile.NamedTemporaryFile(
|
| mode="w+",
|
| suffix=".txt",
|
| delete=False
|
| ) as f:
|
| f.write(output)
|
| filename = f.name
|
|
|
| await message.reply_document(
|
| filename,
|
| caption="📄 Proses selesai (output lengkap)",
|
| reply_to_message_id=message.id
|
| )
|
| finally:
|
| if filename and os.path.exists(filename):
|
| os.remove(filename)
|
|
|
|
|
| @Client.on_message(
|
| filters.command("shell") &
|
| filters.user(Config.OWNER_ID)
|
| )
|
| async def shell_handler(client: Client, message: Message):
|
| if len(message.command) < 2:
|
| await message.reply_text(
|
| "❌ Gunakan: /shell <command>",
|
| reply_to_message_id=message.id
|
| )
|
| return
|
|
|
| command = message.text.split(" ", 1)[1]
|
|
|
| process = await asyncio.create_subprocess_shell(
|
| command,
|
| stdout=asyncio.subprocess.PIPE,
|
| stderr=asyncio.subprocess.PIPE
|
| )
|
|
|
| try:
|
| stdout, stderr = await asyncio.wait_for(
|
| process.communicate(),
|
| timeout=3
|
| )
|
|
|
| output = (stdout or stderr).decode(errors="ignore") or "Tidak ada output."
|
|
|
| if len(output) < 3500:
|
| await message.reply_text(
|
| output,
|
| parse_mode=None,
|
| reply_to_message_id=message.id
|
| )
|
| else:
|
| filename = None
|
| try:
|
| with tempfile.NamedTemporaryFile(
|
| mode="w+",
|
| suffix=".txt",
|
| delete=False
|
| ) as f:
|
| f.write(output)
|
| filename = f.name
|
|
|
| await message.reply_document(
|
| filename,
|
| caption="📄 Output terlalu panjang",
|
| reply_to_message_id=message.id
|
| )
|
| finally:
|
| if filename and os.path.exists(filename):
|
| os.remove(filename)
|
|
|
| except asyncio.TimeoutError:
|
| asyncio.create_task(
|
| background_wait(process, message)
|
| )
|
|
|
| await message.reply_text(
|
| "⏳ Proses berat terdeteksi (mis. ffmpeg)\n"
|
| "Berjalan di background.\n"
|
| "Output akan dikirim otomatis saat selesai.",
|
| parse_mode=None,
|
| reply_to_message_id=message.id
|
| ) |
