CliRemote 1.4.3__py3-none-any.whl → 1.4.5__py3-none-any.whl

{cliremote-1.4.3.dist-info → cliremote-1.4.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: CliRemote
-Version: 1.4.3
+Version: 1.4.5
 Summary: Remote client framework for Telegram automation using Pyrogram
 Home-page: https://github.com/MohammadAhmadi-R/CliRemote
 Author: MrAhmadiRad

{cliremote-1.4.3.dist-info → cliremote-1.4.5.dist-info}/RECORD

@@ -1,6 +1,6 @@
-cliremote-1.4.3.dist-info/licenses/LICENSE,sha256=O-0zMbcEi6wXz1DiSdVgzMlQjJcNqNe5KDv08uYzqR0,1055
+cliremote-1.4.5.dist-info/licenses/LICENSE,sha256=O-0zMbcEi6wXz1DiSdVgzMlQjJcNqNe5KDv08uYzqR0,1055
 remote/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-remote/account_manager.py,sha256=gPgtEtJuH7-VhBl1xpOv3qaLmIiU3vOGj2rTAcgPRQ4,5707
+remote/account_manager.py,sha256=u8yqcyq7Vl8LZ1rNlrEfwlJEbJiBzoSAlL3z2hhTzyc,9829
 remote/account_viewer.py,sha256=MQoH5lOz24651EjhlzVwi6O5hVUAT7Q5fFU6ZHjBlsM,4243
 remote/admin_manager.py,sha256=WiUUVmSs5JTUdXeSry8PkK_3TRemAdSZjm0G1ilAA-A,3532
 remote/analytics_manager.py,sha256=cQ_uATzOK345U1J88Fki3gOfNedS4Ur5NpLjazsxQkg,3625
@@ -31,7 +31,7 @@ remote/speed_manager.py,sha256=fIWSQAP9qW8AHZtMZq0MrC4_nvxcTFU1SBU75kpRzB8,1115
 remote/stop_manager.py,sha256=UXzKJTblEyQqCjp7fenvQ51Q96Unx05WeOiuFMdj25M,1151
 remote/text_manager.py,sha256=C2wNSXPSCDu8NSD3RsfbKmUQMWOYd1B5N4tzy-Jsriw,2195
 remote/username_manager.py,sha256=nMNdke-2FIv86xR1Y6rR-43oUoQu_3Khw8wEo54noXI,3388
-cliremote-1.4.3.dist-info/METADATA,sha256=I9DYhIjUBPsrdH5eyBKpYd7hJzTtm0ShoD0zOzwUlfk,1202
-cliremote-1.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cliremote-1.4.3.dist-info/top_level.txt,sha256=yBZidJ6zCix_a2ubGlYaewvlzBFXWbckQt20dudxJ1E,7
-cliremote-1.4.3.dist-info/RECORD,,
+cliremote-1.4.5.dist-info/METADATA,sha256=KBdd5hKfr0fh7rPnySbjPklm4isIJGuHztut6Rnds_U,1202
+cliremote-1.4.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cliremote-1.4.5.dist-info/top_level.txt,sha256=yBZidJ6zCix_a2ubGlYaewvlzBFXWbckQt20dudxJ1E,7
+cliremote-1.4.5.dist-info/RECORD,,

remote/account_manager.py

@@ -1,161 +1,246 @@
-# antispam_core/account_manager.py
-import os, asyncio, json, logging
-from typing import Optional, Dict, List
+import os
+import json
+import asyncio
+import logging
+import random
+import traceback
+from typing import Optional, Dict, List, Set, Tuple
 from pyrogram import Client, errors
-from .client_manager import (
-    ACCOUNTS_FOLDER, 
-    get_account_data, 
-    save_account_data, 
-    stop_all_clients, 
-    accounts, 
-    client_pool, 
-    client_locks
-)
 
+# ============================================================
+# ⚙️ تنظیم لاگ دقیق برای دیباگ Pyrogram و SQLite
+# ============================================================
+os.makedirs("logs", exist_ok=True)
+log_file = "logs/client_debug_log.txt"
 logger = logging.getLogger(__name__)
-login = {}
+logger.setLevel(logging.DEBUG)
 
-# ==========================
-# افزودن اکانت جدید
-# ==========================
-async def add_account_cmd(message, get_app_info):
+if not any(isinstance(h, logging.FileHandler) and h.baseFilename.endswith(log_file) for h in logger.handlers):
+    fh = logging.FileHandler(log_file, encoding="utf-8")
+    fmt = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s", "%Y-%m-%d %H:%M:%S")
+    fh.setFormatter(fmt)
+    logger.addHandler(fh)
+
+logger.info("🧩 Client Manager started in DEBUG MODE.")
+
+# ============================================================
+# 🧱 ساختار داده‌ها
+# ============================================================
+client_pool: Dict[str, Client] = {}
+client_locks: Dict[str, asyncio.Lock] = {}
+
+ACCOUNTS_FOLDER = "acc"
+ACCOUNTS_DATA_FOLDER = "acc_data"
+os.makedirs(ACCOUNTS_FOLDER, exist_ok=True)
+os.makedirs(ACCOUNTS_DATA_FOLDER, exist_ok=True)
+
+
+# ============================================================
+# 🧠 ساخت یا دریافت کلاینت
+# ============================================================
+async def get_or_start_client(phone_number: str) -> Optional[Client]:
+    cli = client_pool.get(phone_number)
     try:
-        parts = message.text.split(' ', 1)
-        if len(parts) < 2:
-            await message.reply('مثال: `add +989123456789`')
-            return
-
-        phone_number = parts[1].strip()
-        session_file = os.path.join(ACCOUNTS_FOLDER, f'{phone_number}.session')
-
-        if os.path.exists(session_file):
-            await message.reply('این اکانت وجود دارد!')
-            return
-
-        global login
-        api = get_app_info()
-        if not api or len(api) < 2:
-            await message.reply('مشکل در API!')
-            return
-
-        login['id'] = int(api[1])
-        login['hash'] = api[0]
-        login['number'] = phone_number
-        login['api_data'] = {
-            'api_id': api[1],
-            'api_hash': api[0],
-            'phone_number': phone_number,
-            'session': phone_number,
-            '2fa_password': None
-        }
+        if cli is not None and getattr(cli, "is_connected", False):
+            logger.debug(f"{phone_number}: Already connected → {cli.session_name}")
+            return cli
+
+        cli = _make_client_from_json(phone_number)
+        if cli is None:
+            logger.error(f"{phone_number}: ❌ Could not build client (no JSON or invalid data)")
+            return None
+
+        session_db_path = f"{cli.session_name}.session"
+        logger.debug(f"{phone_number}: Session DB path → {session_db_path}")
+
+        if not os.path.exists(session_db_path):
+            logger.warning(f"{phone_number}: Session file not found → {session_db_path}")
+        else:
+            size = os.path.getsize(session_db_path)
+            logger.debug(f"{phone_number}: Session file exists ({size} bytes)")
+            if not os.access(session_db_path, os.R_OK | os.W_OK):
+                logger.warning(f"{phone_number}: ⚠️ No read/write permission for {session_db_path}")
 
         try:
-            login['client'] = Client(name=session_file.replace('.session', ''), api_id=login['id'], api_hash=login['hash'])
-            await login['client'].connect()
-            login['response'] = await login['client'].send_code(phone_number)
-            await message.reply(f'✅ کد تأیید به {phone_number} ارسال شد.\n`code 12345`')
-        except errors.BadRequest as e:
-            await message.reply(f'Bad request: {str(e)}')
-        except errors.FloodWait as e:
-            await message.reply(f'Flood wait: {e.value} sec')
+            await cli.start()
+            await asyncio.sleep(0.4)
+            logger.info(f"{phone_number}: ✅ Client started successfully.")
+        except errors.SessionPasswordNeeded:
+            twofa = getattr(cli, "_twofa_password", None)
+            if twofa:
+                await cli.check_password(twofa)
+                logger.info(f"{phone_number}: ✅ 2FA password applied.")
+            else:
+                logger.error(f"{phone_number}: ⚠️ 2FA required but missing.")
+                return None
+        except errors.AuthKeyDuplicated:
+            logger.error(f"{phone_number}: ❌ AuthKeyDuplicated (session invalid).")
+            return None
         except Exception as e:
-            await message.reply(f'Connection error: {str(e)}')
-    except Exception as e:
-        await message.reply(f'خطا: {str(e)}')
+            tb = traceback.format_exc(limit=3)
+            logger.error(f"{phone_number}: ❌ Start failed - {type(e).__name__}: {e}\n{tb}")
+            return None
 
+        client_pool[phone_number] = cli
+        client_locks.setdefault(phone_number, asyncio.Lock())
+        return cli
 
-# ==========================
-# تأیید کد ورود
-# ==========================
-async def set_code_cmd(message):
-    global login
-    parts = message.text.split(' ', 1)
-    if len(parts) < 2:
-        await message.reply('`code 12345`')
-        return
-    code = parts[1].strip()
+    except Exception as e:
+        tb = traceback.format_exc(limit=3)
+        logger.critical(f"{phone_number}: 💥 Fatal error in get_or_start_client - {type(e).__name__}: {e}\n{tb}")
+        return None
 
+
+# ============================================================
+# 🧩 ساخت کلاینت از JSON
+# ============================================================
+def _make_client_from_json(phone_number: str) -> Optional[Client]:
     try:
-        await login['client'].sign_in(login['number'], login['response'].phone_code_hash, code)
-        await login['client'].disconnect()
-        save_account_data(login['number'], login['api_data'])
-        await message.reply(f"✅ اکانت اضافه شد!\n├ شماره: {login['number']}")
-        login = {}
-    except errors.SessionPasswordNeeded:
-        await message.reply('🔒 لطفا رمز را با `pass your_password` بدهید')
-    except errors.BadRequest:
-        await message.reply('ورود با مشکل مواجه شد')
+        data_path = os.path.join(ACCOUNTS_DATA_FOLDER, f"{phone_number}.json")
+        if not os.path.exists(data_path):
+            logger.error(f"{phone_number}: ⚠️ Account JSON not found → {data_path}")
+            return None
+
+        with open(data_path, "r", encoding="utf-8") as f:
+            account_data = json.load(f)
+
+        session_base = account_data.get("session")
+        if not session_base:
+            logger.error(f"{phone_number}: Missing 'session' key in JSON → {data_path}")
+            return None
+
+        session_path = os.path.join(ACCOUNTS_FOLDER, session_base)
+        if not session_path.endswith(".session"):
+            session_path += ".session"
+
+        os.makedirs(os.path.dirname(session_path), exist_ok=True)
+
+        logger.debug(f"{phone_number}: Final session path → {session_path}")
+
+        api_id = account_data.get("api_id")
+        api_hash = account_data.get("api_hash")
+        if not api_id or not api_hash:
+            logger.error(f"{phone_number}: Missing API credentials in JSON → {data_path}")
+            return None
+
+        cli = Client(
+            name=session_path,
+            api_id=int(api_id),
+            api_hash=str(api_hash),
+            sleep_threshold=30,
+            workdir=os.path.join("acc_temp", phone_number),
+            no_updates=True,
+        )
+
+        if account_data.get("2fa_password"):
+            setattr(cli, "_twofa_password", account_data["2fa_password"])
+
+        return cli
+
     except Exception as e:
-        await message.reply(f'⚠️ خطا در ورود: {e}')
+        tb = traceback.format_exc(limit=3)
+        logger.critical(f"{phone_number}: 💥 Error creating client - {type(e).__name__}: {e}\n{tb}")
+        return None
 
 
-# ==========================
-# افزودن رمز دومرحله‌ای
-# ==========================
-async def set_2fa_cmd(message):
-    global login
-    parts = message.text.split(' ', 1)
-    if len(parts) < 2:
-        await message.reply('`pass my_password`')
+# ============================================================
+# 🚀 Preload با لاگ کامل
+# ============================================================
+async def preload_clients(limit: Optional[int] = None) -> None:
+    phones = list(get_active_accounts())
+    if limit is not None:
+        phones = phones[:max(0, int(limit))]
+
+    if not phones:
+        logger.info("⚙️ No accounts found for preload.")
         return
-    password = parts[1].strip()
-    try:
-        await login['client'].check_password(password)
-        login['api_data']['2fa_password'] = password
-        save_account_data(login['number'], login['api_data'])
-        await message.reply(f"✅ اکانت با موفقیت اضافه شد!\n├ شماره: {login['number']}")
-        await login['client'].disconnect()
-        login = {}
-    except errors.BadRequest:
-        await message.reply('رمز اشتباه است!')
-    except Exception as e:
-        await message.reply(f'⚠️ خطا در ثبت پسورد: {e}')
 
+    logger.info(f"🚀 Preloading {len(phones)} clients...")
+    ok, bad = 0, 0
 
-# ==========================
-# حذف یک اکانت خاص
-# ==========================
-def remove_client_from_pool(phone_number: str):
-    cli = client_pool.get(phone_number)
-    if cli:
+    for idx, phone in enumerate(phones, 1):
+        logger.info(f"🔹 [{idx}/{len(phones)}] Loading client {phone}")
         try:
-            asyncio.create_task(cli.stop())
-        except:
-            pass
-        client_pool.pop(phone_number, None)
-        client_locks.pop(phone_number, None)
+            cli = await get_or_start_client(phone)
+            if cli and getattr(cli, "is_connected", False):
+                ok += 1
+                logger.info(f"{phone}: ✅ Connected.")
+            else:
+                bad += 1
+                logger.warning(f"{phone}: ❌ Not connected after start().")
+        except Exception as e:
+            bad += 1
+            tb = traceback.format_exc(limit=3)
+            logger.error(f"{phone}: ❌ Exception during preload - {type(e).__name__}: {e}\n{tb}")
 
+        await asyncio.sleep(1.0)
 
-async def delete_account_cmd(message):
+    logger.info(f"🎯 Preload completed: OK={ok} | FAIL={bad}")
+
+
+# ============================================================
+# 🧹 توقف تمام کلاینت‌ها
+# ============================================================
+async def stop_all_clients() -> None:
+    logger.info("🧹 Stopping all clients...")
+    for phone, cli in list(client_pool.items()):
+        try:
+            await cli.stop()
+            logger.info(f"{phone}: 📴 Stopped successfully.")
+        except Exception as e:
+            tb = traceback.format_exc(limit=2)
+            logger.warning(f"{phone}: ⚠️ Error stopping client - {type(e).__name__}: {e}\n{tb}")
+        finally:
+            client_pool.pop(phone, None)
+            await asyncio.sleep(0.3)
+    logger.info("✅ All clients stopped cleanly.")
+
+
+# ============================================================
+# 📦 مدیریت JSON داده‌های اکانت
+# ============================================================
+def get_account_data(phone_number: str) -> Optional[Dict]:
+    """
+    خواندن داده‌های JSON اکانت از acc_data/{phone}.json
+    """
+    file_path = os.path.join(ACCOUNTS_DATA_FOLDER, f"{phone_number}.json")
+    if not os.path.exists(file_path):
+        logger.warning(f"{phone_number}: ⚠️ Account JSON not found at {file_path}")
+        return None
     try:
-        phone_number = message.text.split()[1]
-        main_path = f'{ACCOUNTS_FOLDER}/{phone_number}.session'
-        remove_client_from_pool(phone_number)
-        if os.path.isfile(main_path):
-            os.unlink(main_path)
-            await message.reply('<b>Account deleted successfully.</b>')
-        else:
-            await message.reply('<b>Account not found in database.</b>')
-    except IndexError:
-        await message.reply('Please enter phone number')
+        with open(file_path, "r", encoding="utf-8") as f:
+            return json.load(f)
     except Exception as e:
-        await message.reply(f'<b>Error deleting account: {str(e)}</b>')
+        logger.error(f"{phone_number}: ⚠️ Error reading JSON - {type(e).__name__}: {e}")
+        return None
 
 
-# ==========================
-# حذف تمام اکانت‌ها
-# ==========================
-async def delete_all_accounts_cmd(message):
+def save_account_data(phone_number: str, data: Dict) -> None:
+    """
+    ذخیره اطلاعات JSON اکانت در acc_data/{phone}.json
+    """
+    os.makedirs(ACCOUNTS_DATA_FOLDER, exist_ok=True)
+    file_path = os.path.join(ACCOUNTS_DATA_FOLDER, f"{phone_number}.json")
     try:
-        accountss = accounts()
-        count = len(accountss)
-        await stop_all_clients()
-        for session in accountss:
-            main_path = f'{ACCOUNTS_FOLDER}/{session}.session'
-            try:
-                os.unlink(main_path)
-            except Exception:
-                pass
-        await message.reply(f'<b>{count} accounts deleted.</b>')
+        with open(file_path, "w", encoding="utf-8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=4)
+        logger.info(f"{phone_number}: 💾 Account data saved successfully → {file_path}")
     except Exception as e:
-        await message.reply(f'<b>Error deleting all accounts: {str(e)}</b>')
+        logger.error(f"{phone_number}: ⚠️ Error saving JSON - {type(e).__name__}: {e}")
+
+
+# ============================================================
+# 📋 لیست اکانت‌های فعال
+# ============================================================
+def accounts() -> List[str]:
+    accs: Set[str] = set()
+    if not os.path.isdir(ACCOUNTS_FOLDER):
+        return []
+    for acc in os.listdir(ACCOUNTS_FOLDER):
+        if acc.endswith(".session"):
+            accs.add(acc.split(".")[0])
+    return list(accs)
+
+
+def get_active_accounts() -> Set[str]:
+    return set(accounts())