Update main.py

main.py

@@ -93,7 +93,7 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: دیتابیس SQLite (نسخه مجهز به موتور تعمیر خودکار و استخراج داده)
+# 🟢 پارت 4: دیتابیس SQLite (نسخه فوق‌سریع - فقط RAM برای پیام‌ها + پاکسازی هارد)
 # ==============================================================================
 import os
 import sqlite3
@@ -102,6 +102,7 @@ import copy
 import threading
 
 DB_FILE = "/data/users_v3.db" 
+CORRUPT_FILE = "/data/users_db.db" 
 BACKUP_JSON = "/data/users_db.json.bak" 
 
 last_saved_state = {}
@@ -111,72 +112,32 @@ recent_messages_dict = {}
 db_lock = threading.Lock()
 msg_cache_lock = threading.Lock()
 
-def auto_repair_database():
-    print("🛠 در حال بازسازی فایل دیتابیس آسیب‌دیده...")
-    corrupt_backup = DB_FILE + ".corrupt"
-    
-    # 1. تغییر نام فایل خراب به فایل بکاپ
-    for ext in["", "-wal", "-shm"]:
-        if os.path.exists(DB_FILE + ext):
-            try:
-                if ext == "":
-                    if os.path.exists(corrupt_backup): os.remove(corrupt_backup)
-                    os.rename(DB_FILE, corrupt_backup)
-                else:
-                    os.remove(DB_FILE + ext)
-            except Exception: pass
-            
-    # 2. ساخت دیتابیس تمیز و جدید
-    conn = sqlite3.connect(DB_FILE, timeout=60.0)
-    conn.execute('PRAGMA journal_mode=WAL;')
-    conn.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
-    
-    # 3. تلاش برای استخراج تمام داده‌های سالم از فایل خراب به دیتابیس جدید
-    recovered = 0
-    if os.path.exists(corrupt_backup):
-        try:
-            conn.execute(f"ATTACH DATABASE '{corrupt_backup}' AS old_db")
-            conn.execute("INSERT OR IGNORE INTO users SELECT * FROM old_db.users")
-            conn.commit()
-            cursor = conn.cursor()
-            cursor.execute("SELECT COUNT(*) FROM users")
-            recovered = cursor.fetchone()[0]
-            conn.execute("DETACH DATABASE old_db")
-        except Exception as e:
-            print(f"⚠️ استخراج با خطا مواجه شد، اما داده‌های سالم نجات یافتند: {e}")
-            
-    conn.close()
-    print(f"✅ فایل بازسازی شد! تعداد {recovered} کاربر با موفقیت از فایل آسیب‌دیده استخراج شدند.")
-    
-    # 4. در نهایت از فایل متنی JSON هم بازیابی می‌کنیم تا هیچکس جا نماند
-    restore_users_from_json()
-
 def init_sqlite_db():
     try:
         os.makedirs(os.path.dirname(DB_FILE), exist_ok=True)
         is_first_run = not os.path.exists(DB_FILE)
 
         conn = sqlite3.connect(DB_FILE, timeout=60.0)
-        try:
-            # 🚀 تنظیمات سرعت دیتابیس
-            conn.execute('PRAGMA journal_mode=WAL;') 
-            conn.execute('PRAGMA synchronous=NORMAL;')
-            conn.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
-            conn.execute('DROP TABLE IF EXISTS processed_messages')
-            conn.commit()
-            conn.execute('VACUUM')
-        except sqlite3.DatabaseError as e:
-            if "malformed" in str(e).lower():
-                # اگر دیتابیس خراب بود، موتور تعمیر خودکار روشن میشود
-                conn.close()
-                auto_repair_database()
-                return
-        except Exception: pass
-        finally:
-            try: conn.close()
-            except: pass
+        cursor = conn.cursor()
+        
+        # 🚀 تنظیمات سرعت دیتابیس
+        cursor.execute('PRAGMA journal_mode=WAL;') 
+        cursor.execute('PRAGMA synchronous=NORMAL;')
+        
+        # 1️⃣ اطمینان از وجود و سلامت جدول کاربران (اطلاعات اصلی که نباید پاک شود)
+        cursor.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
+        
+        # 2️⃣ حذف جدول شناسه‌های قدیمی پیام‌ها برای همیشه
+        cursor.execute('DROP TABLE IF EXISTS processed_messages')
+        conn.commit()
+
+        # 3️⃣ آزادسازی فضای هارد دیسک و کاهش حجم فایل دیتابیس (حذف فیزیکی شناسه‌های قدیمی)
+        conn.execute('VACUUM')
+        
+        conn.close()
 
         if is_first_run:
+            print("🚨 عملیات نجات آغاز شد...")
             restore_users_from_json()
 
     except Exception as e:
@@ -184,7 +145,7 @@ def init_sqlite_db():
 
 def restore_users_from_json():
     if os.path.exists(BACKUP_JSON):
-        print("📥 در حال بازیابی اعتبار کاربران از فایل پشتیبان JSON...")
+        print("📥 در حال بازیابی اعتبار کاربران از فایل پشتیبان...")
         try:
             with open(BACKUP_JSON, "r", encoding="utf-8") as f:
                 data = json.load(f)
@@ -192,10 +153,10 @@ def restore_users_from_json():
                 conn = sqlite3.connect(DB_FILE, timeout=60.0)
                 conn.execute('PRAGMA journal_mode=WAL;')
                 for cid, udata in data.items():
-                    conn.execute("INSERT OR IGNORE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
+                    conn.execute("INSERT OR REPLACE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
                 conn.commit()
                 conn.close()
-                print(f"✅ اعتبار {len(data)} کاربر از فایل JSON بازیابی شد.")
+                print(f"✅ اعتبار {len(data)} کاربر با موفقیت بازیابی شد.")
         except Exception as e:
             print(f"❌ خطا در بازیابی اعتبارها: {e}")
 
@@ -208,7 +169,7 @@ def load_db():
         conn.execute('PRAGMA journal_mode=WAL;')
         cursor = conn.cursor()
         
-        # 🟢 فقط لود کردن اطلاعات کاربران
+        # 🟢 فقط لود کردن اطلاعات کاربران (بدون دست زدن به شناسه‌های پیام)
         cursor.execute("SELECT chat_id, user_data FROM users")
         for row in cursor.fetchall():
             db_dict[row[0]] = json.loads(row[1])
@@ -231,17 +192,12 @@ def background_save_worker(changed_data):
             conn.executemany("INSERT OR REPLACE INTO users (chat_id, user_data) VALUES (?, ?)", changed_data)
             conn.commit()
             conn.close()
-        except sqlite3.DatabaseError as e:
-            # مانیتورینگ زنده: اگر وسط کار فایل خراب شد سریعاً تعمیر می‌شود
-            if "malformed" in str(e).lower():
-                print("❌ دیتابیس هنگام ذخیره خراب شد! در حال تعمیر خودکار...")
-                auto_repair_database()
         except Exception as e:
             print(f"❌ خطا در ذخیره بک‌گراند: {e}")
 
 def save_db(db_data):
     global last_saved_state
-    changed =[]
+    changed = []
     for cid, data in db_data.items():
         if cid not in last_saved_state or last_saved_state[cid] != data:
             changed.append((str(cid), json.dumps(data, ensure_ascii=False)))
@@ -256,16 +212,22 @@ def save_db(db_data):
 
 # --- سیستم ضد تکرار پیام کاملاً منتقل شده به RAM ---
 def is_message_processed(message_id):
+    # چک کردن داخل RAM در کسری از ثانیه
     return str(message_id) in recent_messages_dict
 
 def mark_message_processed(message_id):
     msg_id_str = str(message_id)
     with msg_cache_lock:
+        # ثبت فقط در RAM
         recent_messages_dict[msg_id_str] = True
+        
+        # جلوگیری از پر شدن بیش از حد RAM (نگه داشتن فقط 15000 پیام آخر)
         if len(recent_messages_dict) > 15000:
             oldest_key = next(iter(recent_messages_dict))
             del recent_messages_dict[oldest_key]
 
+user_credits_db = load_db()
+
 # ==============================================================================
 # 🟢 پارت 5: توابع کد معرف، مدیریت سهمیه حساب کاربری و تبدیل اعداد
 # ==============================================================================
@@ -2166,12 +2128,12 @@ else:
                 except Exception:
                     pass
 
+                # پیام چه قدیمی باشد چه جدید، در دیتابیس ثبت می‌شود که دیگر خوانده نشود
+                mark_message_processed(str_msg_id)
+
                 if is_old:
-                    # پیام‌های تلنبار شده و قدیمی اصلا در دیتابیس و حافظه ذخیره نمی‌شوند و درجا دور ریخته می‌شوند
+                    # پیام تلنبار شده و قدیمی است، کاملا بی‌سروصدا دور ریخته می‌شود (بدون جواب دادن)
                     return
-
-                # ربات فقط شناسه‌های پیام‌های کاملاً جدید را در کش رم ذخیره می‌کند
-                mark_message_processed(str_msg_id)
             # ===================================================================
 
             user_text = getattr(update, "text", "") or getattr(msg_obj, "text", "")
@@ -2346,11 +2308,7 @@ else:
                     payload = {"chat_id": chat_id, "text": join_msg, "chat_keypad_type": "New", "chat_keypad": JOIN_KEYPAD_DICT}
                     await client._make_request("sendMessage", payload)
                 except Exception:
-                    # 🔴 رفع خطای مسدودیت: اگر کاربر ربات را بلاک کرده باشد و خطا دهد، ربات با آرامش عبور میکن��.
-                    try:
-                        await client.send_message(chat_id, join_msg)
-                    except Exception:
-                        pass
+                    await client.send_message(chat_id, join_msg)
                 return
 
             if user_text_lower.startswith("/start"):