میدلورها (Middleware)

from rubka.middleware import MiddlewarePriority

@bot.before_middleware(priority=MiddlewarePriority.HIGHEST)
async def first_middleware(bot, message, context):
    print("اولین میدلور اجرا می‌شود")
    return True

@bot.middleware(priority=MiddlewarePriority.HIGH)
async def log_middleware(bot, message, next_handler, metadata):
    print("بل از اجرا")
    result = await next_handler(bot, message)
    print("بعد از اجرا")
    return result

@bot.before_middleware(priority=MiddlewarePriority.HIGHEST)
async def check_user(bot, message, context):
    if message.sender_id not in allowed_users:
        await message.reply("شما دسترسی ندارید!")
        return False  # جلوگیری از ادامه پردازش
    return True  # ادامه پردازش

@bot.after_middleware()
async def send_log(bot, message, result, context):
    print(f"پیام {message.message_id} پردازش شد")
    # می‌توانید result را تغییر دهید
    return result

@bot.error_middleware()
async def handle_error(bot, message, error, context):
    print(f"❌ خطا: {error}")
    await message.reply("متأسفانه خطایی رخ داد!")
    return True  # خطا مدیریت شد

from rubka.middleware import MiddlewareManager, MiddlewarePriority

manager = MiddlewareManager()

# افزودن میدلور
manager.add_middleware(log_middleware, MiddlewarePriority.NORMAL)

# اجرا
result = await manager.run_middlewares(bot, message, handler, {})

from rubka import Robot
from rubka.context import Message
from rubka.middleware import MiddlewarePriority

bot = Robot(token="YOUR_TOKEN")

# قبل از همه میدلورها
@bot.before_middleware(priority=MiddlewarePriority.HIGHEST)
async def before_all(bot, message, context):
    print("⚡ قبل از همه چیز")
    return True

# میدلور اصلی برای لاگینگ
@bot.middleware()
async def log_middleware(bot, message, next_handler, metadata):
    print("1️⃣ قبل از اجرا")
    result = await next_handler(bot, message)
    print("3️⃣ بعد از اجرا")
    return result

# بعد از میدلورها
@bot.after_middleware()
async def after_all(bot, message, result, context):
    if hasattr(message, 'reply'):
        await message.reply("پیام از after_middleware ارسال شد!")
    return result

# هندلر اصلی
@bot.on_message(commands=["start"])
async def start(bot: Robot, message: Message):
    print("داخل دستور start")
    await message.reply("سلام!")

bot.run()

# محدودیت نرخ ارسال (Rate Limiting)
@bot.before_middleware()
async def rate_limit(bot, message, context):
    if is_rate_limited(message.sender_id):
        await message.reply("⏳ لطفاً کمی صبر کنید!")
        return False
    return True

#⃣ بررسی عضویت در کانال
@bot.before_middleware()
async def check_membership(bot, message, context):
    is_member = await bot.check_join("channel_id", message.chat_id)
    if not is_member:
        await message.reply("لطفاً ابتدا در کانال عضو شوید!")
        return False
    return True

# اندازه‌گیری زمان پاسخ
@bot.middleware()
async def measure_time(bot, message, next_handler, metadata):
    start = time.time()
    result = await next_handler(bot, message)
    duration = time.time() - start
    print(f"⏱️ زمان پاسخ: {duration:.2f} ثانیه")
    return result

#ذخیره خودکار پیام‌ها در دیتابیس
@bot.after_middleware()
async def save_to_db(bot, message, result, context):
    await bot.save_message(message)
    return result