Update main.py

main.py

@@ -93,7 +93,7 @@ def gregorian_to_jalali(gy, gm, gd):
     return jy, jm, jd
 
 # ==============================================================================
-# 🟢 پارت 4: دیتابیس SQLite (نسخه فوق‌سریع v4 + سیستم نجات خودکار از خطای Malformed)
+# 🟢 پارت 4: دیتابیس SQLite (نسخه مجهز به موتور تعمیر خودکار و استخراج داده)
 # ==============================================================================
 import os
 import sqlite3
@@ -101,73 +101,90 @@ import json
 import copy
 import threading
 
-DB_FILE = "/data/users_v4.db"            # 🟢 دیتابیس کاملا جدید و سالم
-CORRUPT_V3_FILE = "/data/users_v3.db"    # فایل ۳۰ مگابایتی که فهرستش خراب شده
+DB_FILE = "/data/users_v3.db" 
 BACKUP_JSON = "/data/users_db.json.bak" 
 
 last_saved_state = {}
+# حافظه کش فقط در RAM (با ری‌استارت اسپیس پاک می‌شود)
 recent_messages_dict = {}
+# قفل‌ها برای جلوگیری از تداخل رشته‌ها (Threads)
 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)
-        cursor = conn.cursor()
-        
-        # 🚀 تنظیمات سرعت دیتابیس (بدون دستور خطرناک VACUUM)
-        cursor.execute('PRAGMA journal_mode=WAL;') 
-        cursor.execute('PRAGMA synchronous=NORMAL;')
-        cursor.execute('CREATE TABLE IF NOT EXISTS users (chat_id TEXT PRIMARY KEY, user_data TEXT)')
-        conn.commit()
-        conn.close()
+        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
 
         if is_first_run:
-            print("🚨 دیتابیس v4 ساخته شد. عملیات نجات داده‌ها آغاز شد...")
-            rescue_data_from_corrupt_db()
+            restore_users_from_json()
 
     except Exception as e:
         print(f"❌ خطا در راه اندازی دیتابیس: {e}")
 
-def rescue_data_from_corrupt_db():
-    """این تابع تلاش می‌کند داده‌ها را از فایل خراب v3 استخراج کرده و ب�� v4 بریزد"""
-    if os.path.exists(CORRUPT_V3_FILE):
-        print("🔍 در حال خواندن اطلاعات از فایل ۳۰ مگابایتی...")
-        try:
-            # اتصال به فایل سالم
-            conn_new = sqlite3.connect(DB_FILE, timeout=60.0)
-            conn_new.execute('PRAGMA journal_mode=WAL;')
-            
-            # اتصال موقت به فایل خراب
-            conn_old = sqlite3.connect(CORRUPT_V3_FILE, timeout=60.0)
-            cursor_old = conn_old.cursor()
-            
-            # بیرون کشیدن داده‌ها (حتی اگر دیتابیس اخطار بدهد، ردیف‌های سالم را می‌خواند)
-            cursor_old.execute("SELECT chat_id, user_data FROM users")
-            rows = cursor_old.fetchall()
-            
-            if rows:
-                conn_new.executemany("INSERT OR IGNORE INTO users VALUES (?, ?)", rows)
-                conn_new.commit()
-                print(f"✅ موفقیت‌آمیز! تعداد {len(rows)} کاربر از دیتابیس خراب نجات یافتند.")
-            
-            conn_old.close()
-            conn_new.close()
-            
-            # تغییر نام فایل خراب تا در ری‌استارت‌های بعدی دوباره پردازش نشود
-            os.rename(CORRUPT_V3_FILE, "/data/users_v3_corrupt.bak")
-            
-        except Exception as e:
-            print(f"⚠️ اخطار در حین نجات: {e}")
-            print("در حال تلاش برای بازیابی از فایل JSON...")
-            restore_users_from_json()
-
 def restore_users_from_json():
     if os.path.exists(BACKUP_JSON):
+        print("📥 در حال بازیابی اعتبار کاربران از فایل پشتیبان JSON...")
         try:
             with open(BACKUP_JSON, "r", encoding="utf-8") as f:
                 data = json.load(f)
@@ -178,9 +195,9 @@ def restore_users_from_json():
                     conn.execute("INSERT OR IGNORE INTO users VALUES (?, ?)", (str(cid), json.dumps(udata)))
                 conn.commit()
                 conn.close()
-                print(f"✅ اعتبار {len(data)} کاربر از JSON بازیابی شد.")
+                print(f"✅ اعتبار {len(data)} کاربر از فایل JSON بازیابی شد.")
         except Exception as e:
-            print(f"❌ خطا در بازیابی از JSON: {e}")
+            print(f"❌ خطا در بازیابی اعتبارها: {e}")
 
 def load_db():
     global last_saved_state, recent_messages_dict
@@ -191,16 +208,17 @@ 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])
             
         conn.close()
         last_saved_state = copy.deepcopy(db_dict)
-        print(f"🚀 دیتابیس آماده شد. تعداد کاربران: {len(db_dict)}")
+        print(f"🚀 دیتابیس آماده شد. تعداد کاربران: {len(db_dict)} | کش پیام‌ها پاکسازی شد.")
         return db_dict
     except Exception as e:
-        print(f"⚠️ ارور در لود نهایی دیتابیس: {e}")
+        print(f"⚠️ ارور در لود نهایی: {e}")
         return {}
 
 # --- سیستم ذخیره سازی کاربران در پس‌زمینه ---
@@ -213,8 +231,13 @@ 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:
-            pass # سکوت در برابر خطاهای جزئی پس‌زمینه
+            print(f"❌ خطا در ذخیره بک‌گراند: {e}")
 
 def save_db(db_data):
     global last_saved_state
@@ -225,6 +248,7 @@ def save_db(db_data):
             
     if not changed: return
     
+    # آپدیت وضعیت استیت محلی
     for cid, _ in changed:
         last_saved_state[cid] = copy.deepcopy(db_data[cid])
         
@@ -242,8 +266,6 @@ def mark_message_processed(message_id):
             oldest_key = next(iter(recent_messages_dict))
             del recent_messages_dict[oldest_key]
 
-user_credits_db = load_db()
-
 # ==============================================================================
 # 🟢 پارت 5: توابع کد معرف، مدیریت سهمیه حساب کاربری و تبدیل اعداد
 # ==============================================================================
@@ -2144,12 +2166,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", "")
@@ -2324,7 +2346,11 @@ 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:
-                    await client.send_message(chat_id, join_msg)
+                    # 🔴 رفع خطای مسدودیت: اگر کاربر ربات را بلاک کرده باشد و خطا دهد، ربات با آرامش عبور میکند.
+                    try:
+                        await client.send_message(chat_id, join_msg)
+                    except Exception:
+                        pass
                 return
 
             if user_text_lower.startswith("/start"):