Delete bot.py

bot.py

@@ -1,195 +0,0 @@
-import os
-import io
-import sys
-import zipfile
-import base64
-import random
-import shutil
-import subprocess
-import tempfile
-from pathlib import Path
-
-from fastapi import FastAPI, Request
-from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
-from telegram.ext import (
-    Application,
-    CommandHandler,
-    MessageHandler,
-    filters,
-    ContextTypes,
-)
-
-from motor.motor_asyncio import AsyncIOMotorClient
-import uvicorn
-from config import BOT_TOKEN, ADMIN_ID, MONGO_URI, DB_NAME
-
-# -------------------- MongoDB Setup --------------------
-mongo_client = AsyncIOMotorClient(MONGO_URI)
-db = mongo_client[DB_NAME]
-users_col = db["users"]
-bans_col = db["banned_users"]
-stats_col = db["stats"]
-
-async def init_stats():
-    if not await stats_col.find_one({"_id": "encryption_count"}):
-        await stats_col.insert_one({"_id": "encryption_count", "count": 0})
-
-# -------------------- Helpers --------------------
-def make_zip_bytes(filename: str, content: bytes) -> bytes:
-    bio = io.BytesIO()
-    with zipfile.ZipFile(bio, "w", compression=zipfile.ZIP_DEFLATED) as zf:
-        zf.writestr(filename, content)
-    return bio.getvalue()
-
-def rand_bytes(n): 
-    return os.urandom(n)
-
-def bytes_to_c_array_literal(b: bytes) -> str:
-    return ",".join(str(x) for x in b)
-
-def xor_string(s: str, key: int = None) -> str:
-    if key is None:
-        key = random.randint(1, 255)
-    arr = [ord(c) ^ key for c in s]
-    return f"(lambda s:''.join(chr(B^{key}) for B in s))([{','.join(str(x) for x in arr)}])"
-
-def generate_junk_data(min_kb=50, max_kb=70):
-    junk_size = random.randint(min_kb * 1024, max_kb * 1024)
-    chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_()-./,;:!?"
-    return ''.join(random.choice(chars) for _ in range(junk_size))
-
-# -------------------- C Template --------------------
-C_TEMPLATE = """/*KEY_BYTES*/"""
-SETUP_PY = """# Your setup.py content"""
-
-def compile_c_extension(build_dir: Path):
-    (build_dir / "setup.py").write_text(SETUP_PY)
-    proc = subprocess.run(
-        [sys.executable, "setup.py", "build_ext", "--inplace"],
-        cwd=str(build_dir),
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        timeout=30,
-    )
-    candidates = list(build_dir.glob("aes_helper*.so")) + list(build_dir.glob("build/lib.*/*.so"))
-    if not candidates:
-        raise RuntimeError("Compile fail:\n" + proc.stdout.decode())
-    return candidates[0]
-
-def build_wrapper(j_b64, so_b64, so_filename):
-    lines = [
-        "wulcan = '.WulcanPy'",
-        "import os as O, sys as S, base64 as B, tempfile as T",
-        f"WL='{j_b64}'",
-        f"M={repr(so_b64)}",
-        f"D=O.path.join(T.gettempdir(),{xor_string('ninja_tmp')});O.makedirs(D,exist_ok=True)",
-        f"P=O.path.join(D,{xor_string(so_filename)})",
-        "open(P,'wb').write(B.b64decode(M))",
-        "import importlib.machinery as L, importlib.util as U",
-        "loader=L.ExtensionFileLoader('aes_helper',P);spec=U.spec_from_loader(loader.name,loader);mod=U.module_from_spec(spec);loader.exec_module(mod)",
-        "import sys, io, zipfile, runpy",
-        "try:",
-        "\tc=B.b64decode(WL);z=mod.decrypt(c)",
-        "\tt=T.mkdtemp(prefix='nin_');zipfile.ZipFile(io.BytesIO(z)).extractall(t)",
-        "\tS.path.insert(0,t)",
-        "\te=[f for f in O.listdir(t) if f.endswith('.py')][0]",
-        "\tep=O.path.join(t,e)",
-        "\tS.argv=[ep]+S.argv[1:]",
-        "\trunpy.run_path(ep,run_name='__main__')",
-        "except Exception as E: print('ninja_error',E)",
-        "finally: O.remove(P) if O.path.exists(P) else None",
-        f"JUNK={repr(generate_junk_data())}"
-    ]
-    return "\n".join(lines)
-
-# -------------------- File Processing --------------------
-async def process_file(input_file: Path, update: Update, context: ContextTypes.DEFAULT_TYPE):
-    tmpdir = Path(tempfile.mkdtemp())
-    tmp_file = tmpdir / input_file.name
-    tmp_file.write_bytes(input_file.read_bytes())
-    py_bytes = tmp_file.read_bytes()
-    zip_b = make_zip_bytes(input_file.name, py_bytes)
-    key, iv = rand_bytes(32), rand_bytes(16)
-    c_src = C_TEMPLATE.replace("/*KEY_BYTES*/", bytes_to_c_array_literal(key)).replace("/*IV_BYTES*/", bytes_to_c_array_literal(iv))
-    (tmpdir / "aes_helper.c").write_text(c_src)
-    so_path = compile_c_extension(tmpdir)
-    import importlib.util
-    spec = importlib.util.spec_from_file_location("aes_helper", so_path)
-    mod = importlib.util.module_from_spec(spec)
-    spec.loader.exec_module(mod)
-    cipher = mod.encrypt(zip_b)
-    cipher_b64 = base64.b64encode(cipher).decode()
-    so_b64 = base64.b64encode(so_path.read_bytes()).decode()
-    wrapper = build_wrapper(cipher_b64, so_b64, so_path.name)
-    out_name = input_file.stem + "_enc.py"
-    out_path = tmpdir / out_name
-    out_path.write_text(wrapper)
-    await context.bot.send_document(chat_id=update.effective_chat.id, document=open(out_path, "rb"), filename=out_name)
-    await stats_col.update_one({"_id": "encryption_count"}, {"$inc": {"count": 1}})
-    shutil.rmtree(tmpdir, ignore_errors=True)
-
-# -------------------- Handlers --------------------
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user = update.effective_user
-    if not await users_col.find_one({"user_id": user.id}):
-        await users_col.insert_one({"user_id": user.id, "name": user.full_name,
-                                    "username": user.username or "None",
-                                    "language": user.language_code or "Unknown"})
-    keyboard = [[InlineKeyboardButton("▶ Join Channel", url="https://t.me/kuttuxd17")]]
-    await update.message.reply_text("Welcome to Secure File Encryptor Bot!", reply_markup=InlineKeyboardMarkup(keyboard))
-
-async def encode(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user = update.effective_user
-    if await bans_col.find_one({"user_id": user.id}):
-        await update.message.reply_text("You are banned.")
-        return
-    await update.message.reply_text("Please upload a .py file.")
-
-async def handle_document(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    document = update.message.document
-    tmpdir = Path(tempfile.mkdtemp())
-    tmp_path = tmpdir / document.file_name
-    file = await document.get_file()
-    await file.download_to_drive(str(tmp_path))
-    await process_file(tmp_path, update, context)
-
-# -------------------- Admin Commands --------------------
-async def total_users(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if update.effective_user.id != ADMIN_ID: return
-    count = await users_col.count_documents({})
-    await update.message.reply_text(f"Total users: {count}")
-
-async def total_enc(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    if update.effective_user.id != ADMIN_ID: return
-    stats = await stats_col.find_one({"_id": "encryption_count"})
-    await update.message.reply_text(f"Total encryptions: {stats['count'] if stats else 0}")
-
-# -------------------- FastAPI App --------------------
-app = FastAPI()
-bot_app = Application.builder().token(BOT_TOKEN).build()
-
-# Register Handlers
-bot_app.add_handler(CommandHandler("start", start))
-bot_app.add_handler(CommandHandler("encode", encode))
-bot_app.add_handler(MessageHandler(filters.Document.ALL, handle_document))
-bot_app.add_handler(CommandHandler("totalusers", total_users))
-bot_app.add_handler(CommandHandler("totalenc", total_enc))
-
-@app.post("/webhook")
-async def webhook_endpoint(request: Request):
-    update_data = await request.json()
-    update = Update.update_from_dict(update_data)
-    await bot_app.process_update(update)
-    return {"ok": True}
-
-# -------------------- Run --------------------
-if __name__ == "__main__":
-    import asyncio
-
-    async def main():
-        await init_stats()
-        await bot_app.initialize()
-        await bot_app.start()
-        uvicorn.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
-
-    asyncio.run(main())