Upload main.py

main.py

@@ -2,6 +2,7 @@ import logging
 import os
 import asyncio
 import threading
+import re
 from fastapi import FastAPI
 import uvicorn
 import redis.asyncio as redis
@@ -23,10 +24,12 @@ from telegram.ext import (
 
 # আপনার প্রজেক্টের ফাইলগুলো থেকে ইম্পোর্ট
 from credentials import BOT_TOKEN, BOT_OWNER_ID, REPORT_CHANNEL_ID, REDIS_URL_1, REDIS_URL_2
+import config
 from config import logger, BOT_VERSION, TELEGRAM_API_CONCURRENCY_LIMIT
 from db import create_db_pool_with_retry, setup_database
 import db 
 from bot_handlers.batch import BatchDeleter 
+from bot_handlers import utils
 
 # --- ১. Hugging Face Health Check Server (Port 7860) ---
 app_web = FastAPI()
@@ -44,6 +47,7 @@ async def global_error_handler(update: object, context: ContextTypes.DEFAULT_TYP
     logger.error("Exception while handling an update:", exc_info=context.error)
 
 # --- ৩. ব্যাকগ্রাউন্ড টাস্কসমূহ ---
+
 async def user_stats_flush_loop(app):
     while True:
         await asyncio.sleep(300) # প্রতি ৫ মিনিট পর DB-তে ডাটা ফ্লাশ হবে
@@ -68,31 +72,92 @@ async def check_expired_trials_loop(app):
                 await conn.execute("DELETE FROM vip_trials WHERE expiry_date < $1", now_utc)
         except Exception as e:
             logger.error(f"❌ VIP Monitor Error: {e}")
-            
 
 async def recovery_traffic_locks(app):
     """বট স্টার্ট হওয়ার সময় চেক করবে কোনো গ্রুপ লক হয়ে আছে কি না"""
     pool = app.bot_data.get('db_pool')
     if not pool: return
-    
-    # সব গ্রুপ আইডি আনা (db.py থেকে)
     group_ids = await db.get_all_group_ids(pool)
     for chat_id in group_ids:
         from bot_handlers.redis_manager import get_redis_client_for_chat
-        redis = get_redis_client_for_chat(app, chat_id)
-        if redis:
-            backup = await redis.get(f"backup_perms:{chat_id}")
+        redis_client = get_redis_client_for_chat(app, chat_id)
+        if redis_client:
+            backup = await redis_client.get(f"backup_perms:{chat_id}")
             if backup:
                 from bot_handlers.utils import restore_group_permissions
                 await restore_group_permissions(app, chat_id, backup)
+
+# 🔥 নতুন: ব্যাকগ্রাউন্ড বায়ো চেকার টাস্ক (একই প্রসেসের ভেতরে চলবে)
+async def background_bio_task(app):
+    """এটি মেইন বটের ভেতরেই কিউ থেকে আইডি নিয়ে এক এক করে প্রোফাইল চেক করবে"""
+    while True:
+        try:
+            # লাইন (Queue) থেকে ডাটা নিবে
+            data = await app.bot_data['bio_queue'].get()
+            chat_id, user_id, msg_id, settings, user_obj = data
+            
+            # রেট লিমিট এড়াতে ০.১ সেকেন্ড বিরতি
+            await asyncio.sleep(0.1)
+            
+            try:
+                full_user = await app.bot.get_chat(user_id)
+                bio_text = getattr(full_user, 'bio', "") or ""
+                first_name = full_user.first_name or ""
+                last_name = full_user.last_name or ""
+                personal_chat = getattr(full_user, 'personal_chat', None)
+                
+                full_profile_text = f"{first_name} {last_name} {bio_text}".strip()
+                is_bad = False
+                reason_text = "Bad Profile (Auto-detected)"
+                
+                if personal_chat:
+                    is_bad = True
+                    reason_text = "Personal Channel Linked"
+                elif full_profile_text:
+                    if utils.has_link_in_text(full_profile_text) or config.SPAM_PHRASES_COMPILED.search(full_profile_text):
+                        is_bad = True
+                        reason_text = "Link or Spam words in Bio/Name"
+
+                redis_client = app.bot_data.get('redis_clients')[0] if app.bot_data.get('redis_clients') else None
+                
+                if is_bad:
+                    if redis_client: await redis_client.set(f"bio_status:{user_id}", "bad", ex=1800)
+                    
+                    # মেসেজ ডিলিট
+                    try: await app.bot.delete_message(chat_id, msg_id)
+                    except: pass
+                    
+                    # ডিলেশন রিপোর্ট পাঠানো (Optional)
+                    try:
+                        from bot_handlers.core import _background_processing
+                        class DummyContext:
+                            def __init__(self, bot, bot_data):
+                                self.bot = bot
+                                self.bot_data = bot_data
+                        dummy_ctx = DummyContext(app.bot, app.bot_data)
+                        chat_obj = await app.bot.get_chat(chat_id)
+                        asyncio.create_task(_background_processing(dummy_ctx, user_obj, chat_obj, settings, ['censor_reason_bio_link'], reason_text))
+                    except: pass
+                else:
+                    if redis_client: await redis_client.set(f"bio_status:{user_id}", "safe", ex=1600)
+                    
+            except Exception:
+                pass
+            finally:
+                # কাজ শেষ হলে ইউজার লক রিলিজ করা
+                redis_client = app.bot_data.get('redis_clients')[0] if app.bot_data.get('redis_clients') else None
+                if redis_client: await redis_client.delete(f"bio_lock:{user_id}")
             
-# --- ৪. জয়েন রিকোয়েস্ট হ্যান্ডেলার (সাইলেন্ট মোড) ---
+            app.bot_data['bio_queue'].task_done()
+        except Exception as e:
+            logger.error(f"Bio Task Error: {e}")
+
+# --- ৪. জয়েন রিকোয়েস্ট হ্যান্ডেলার ---
 async def handle_join_request(update: Update, context: ContextTypes.DEFAULT_TYPE):
     req = update.chat_join_request
     user, chat = req.from_user, req.chat
     pool = context.bot_data.get('db_pool')
     
-    # ব্যাকগ্রাউন্ডে রেডিস আপডেট (যাতে /grant কমান্ড ইউজ করা যায়)
     if user.username:
         clean_username = user.username.lower().replace("@", "")
         clients = context.bot_data.get('redis_clients')
@@ -100,7 +165,6 @@ async def handle_join_request(update: Update, context: ContextTypes.DEFAULT_TYPE
             try: await clients[0].set(f"join_cache:{clean_username}", user.id, ex=2592000) 
             except: pass
 
-    # ডাটাবেসে টেম্পোরারি সেভ করা
     if pool:
         try:
             temp_expiry = datetime.now(timezone.utc) + timedelta(minutes=10)
@@ -120,12 +184,9 @@ async def w_msg(u, c): from bot_handlers.core import handle_message; await handl
 async def w_edit_msg(u, c): from bot_handlers.core import handle_edited_message; await handle_edited_message(u, c)
 async def w_callback(u, c): from bot_handlers import callbacks; await callbacks.w_callback(u, c)
 
-# main.py এর safe_start_command অংশটি রিপ্লেস করুন
-async def safe_start_command(update, context):
+async def safe_start_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
     user = update.effective_user
     if update.effective_chat.type != 'private': return
-
-    # ইউজার ডাটাবেসে সেভ (ট্র্যাকিংয়ের জন্য)
     pool = context.bot_data.get('db_pool')
     if pool:
         asyncio.create_task(pool.execute("""
@@ -139,11 +200,7 @@ async def safe_start_command(update, context):
          InlineKeyboardButton("বাংলা 🇧🇩", callback_data='lang_bn')], 
         [InlineKeyboardButton("हिन्दी 🇮🇳", callback_data='lang_hi')]
     ])
-    
-    await update.message.reply_html(
-        f"👋 <b>Hello {user.first_name}!</b>\n\nWelcome to Management Bot. Please choose your language to continue:", 
-        reply_markup=keyboard
-    )
+    await update.message.reply_html(f"👋 <b>Hello {user.first_name}!</b>\n\nPlease choose language:", reply_markup=keyboard)
 
 # --- ৬. POST INIT & STOP ---
 async def post_init(app: Application):
@@ -167,9 +224,13 @@ async def post_init(app: Application):
     app.bot_data['BOT_OWNER_ID'] = int(BOT_OWNER_ID) if BOT_OWNER_ID else None
     app.bot_data['REPORT_CHANNEL_ID'] = int(REPORT_CHANNEL_ID) if REPORT_CHANNEL_ID else None
     
+    # 🔥 কিউ তৈরি এবং ব্যাকগ্রাউন্ড টাস্ক চালু করা
+    app.bot_data['bio_queue'] = asyncio.Queue()
+    asyncio.create_task(background_bio_task(app))
+    
+    # বাকি লুপগুলো শুরু
     asyncio.create_task(check_expired_trials_loop(app))
     asyncio.create_task(user_stats_flush_loop(app))
-# ট্রাফিক লক রিকভারি টাস্ক শুরু করা
     asyncio.create_task(recovery_traffic_locks(app))
     
 async def post_stop(app: Application):
@@ -177,18 +238,16 @@ async def post_stop(app: Application):
     if p := app.bot_data.get('db_pool'): await p.close()
     for r in app.bot_data.get('redis_clients', []): await r.close()
 
-# --- ৭. প্রধান ফাংশন (All Handlers Registred) ---
+# --- ৭. প্রধান ফাংশন ---
 def main():
     if not BOT_TOKEN: return
     app = Application.builder().token(BOT_TOKEN).post_init(post_init).post_stop(post_stop).build()
     app.add_error_handler(global_error_handler)
     
-    # মডিউল ইম্পোর্ট
     from bot_handlers import admin_commands as ac
     from bot_handlers import extra_commands as ec
     from bot_handlers import owner_commands as oc
 
-    # --- ১. সাধারণ ও অ্যাডমিন কমান্ড ---
     app.add_handler(CommandHandler("start", safe_start_command))
     app.add_handler(CommandHandler("grant", ac.vip_trial_command))
     app.add_handler(CommandHandler("whois", ac.whois_command))
@@ -198,8 +257,6 @@ def main():
     app.add_handler(CommandHandler("delword", ac.del_word_command))
     app.add_handler(CommandHandler("wordlist", ac.wordlist_command))
     app.add_handler(CommandHandler("resetsettings", ac.reset_settings_command))
-
-    # --- ২. এক্সট্রা ও সেটিংস কমান্ড ---
     app.add_handler(CommandHandler("filter", ec.filter_command))
     app.add_handler(CommandHandler("stop", ec.stop_filter_command))
     app.add_handler(CommandHandler("save", ec.save_note_command))
@@ -222,8 +279,6 @@ def main():
     app.add_handler(CommandHandler("ticket", ec.ticket_command))
     app.add_handler(CommandHandler("trustscore", ec.trustscore_command))
     app.add_handler(CommandHandler("fixdb", ec.fix_database_command))
-
-    # --- ৩. ওনার ও টুলস কমান্ড ---
     app.add_handler(CommandHandler("broadcast", oc.broadcast_command))
     app.add_handler(CommandHandler("bc", oc.broadcast_groups_command))
     app.add_handler(CommandHandler("stats", oc.stats_command))
@@ -235,7 +290,6 @@ def main():
     app.add_handler(CommandHandler("sdel", oc.sdel_command))
     app.add_handler(CommandHandler("smute", oc.smute_command))
 
-    # --- ৪. কোর মেসেজ ও কলব্যাক হ্যান্ডেলার ---
     app.add_handler(CallbackQueryHandler(w_callback))
     app.add_handler(ChatJoinRequestHandler(handle_join_request))
     app.add_handler(MessageHandler(filters.ALL & ~filters.COMMAND, w_msg))
@@ -243,10 +297,7 @@ def main():
     app.add_handler(ChatMemberHandler(w_bot_status, ChatMemberHandler.MY_CHAT_MEMBER))
     app.add_handler(ChatMemberHandler(w_member_status, ChatMemberHandler.CHAT_MEMBER))
 
-    # ওয়েব সার্ভার চালু করা (Hugging Face Health Check-এর জন্য)
     threading.Thread(target=run_web_server, daemon=True).start()
-
-    # পোলিং মোডে বট চালু করা
     logger.info("Bot is polling silently...")
     app.run_polling(allowed_updates=["message", "callback_query", "chat_member", "my_chat_member", "chat_join_request", "edited_message"])