Create bot.py

bot.py

@@ -0,0 +1,170 @@
+import os
+import re
+import time
+import io
+from telethon import events, Button
+import db
+
+def get_admins():
+    return {int(x) for x in os.environ.get("ADMIN_IDS", "").split(",") if x}
+
+def is_admin(event):
+    return event.sender_id in get_admins()
+
+def setup_handlers(client):
+    @client.on(events.NewMessage(pattern='/start'))
+    async def start_cmd(event):
+        if not is_admin(event):
+            await event.respond("⛔ Unauthorized.")
+            return
+        
+        buttons = [
+            [Button.inline("▶️ Start Scan", b"start_scan"), Button.inline("⏸ Pause", b"pause_scan")],
+            [Button.inline("⏹ Stop", b"stop_scan"), Button.inline("📊 Status", b"show_status")],
+            [Button.inline("⚡ Set Speed", b"set_speed"), Button.inline("📥 Export", b"export_files")],
+            [Button.inline("🔄 Load Words", b"load_words"), Button.inline("💣 Reset", b"reset_confirm")]
+        ]
+        await event.respond("👋 Fragment Scanner Bot", buttons=buttons)
+
+    @client.on(events.NewMessage(pattern='/load'))
+    async def load_cmd(event):
+        if not is_admin(event):
+            await event.respond("⛔ Unauthorized.")
+            return
+        await _load_words(event.respond)
+
+    async def _load_words(reply_func):
+        try:
+            with open("words.txt", "r", encoding="utf-8") as f:
+                words = [line.strip().lower() for line in f]
+        except Exception:
+            await reply_func("❌ words.txt not found.")
+            return
+
+        valid_words = [w for w in set(words) if re.match(r'^[a-z0-9_]{4,32}$', w)]
+        
+        r = await db.get_redis()
+        done = await r.smembers("frag:done")
+        
+        to_queue = [w for w in valid_words if w not in done]
+        
+        if to_queue:
+            pipe = r.pipeline()
+            for i in range(0, len(to_queue), 1000):
+                pipe.lpush("frag:queue", *to_queue[i:i+1000])
+            await pipe.execute()
+            
+        await db.set_state(total=len(valid_words))
+        skipped = len(valid_words) - len(to_queue)
+        await reply_func(f"📥 Loaded {len(to_queue)} words. Skipped {skipped} (already done).")
+
+    @client.on(events.CallbackQuery())
+    async def button_handler(event):
+        if not is_admin(event):
+            await event.answer("⛔ Unauthorized.", alert=True)
+            return
+
+        data = event.data.decode('utf-8')
+
+        if data == "load_words":
+            await event.answer()
+            await _load_words(event.reply)
+            
+        elif data == "start_scan":
+            await db.set_state(running="1", paused="0", start_time=str(time.time()))
+            qlen = await db.queue_size()
+            await event.answer()
+            await event.reply(f"🚀 Scan started! {qlen} words in queue.")
+            
+        elif data == "pause_scan":
+            state = await db.get_state()
+            if state.get("paused") == "1":
+                await db.set_state(paused="0")
+                await event.answer()
+                await event.reply("▶️ Scan resumed.")
+            else:
+                await db.set_state(paused="1")
+                await event.answer()
+                await event.reply("⏸ Scan paused.")
+                
+        elif data == "stop_scan":
+            await db.set_state(running="0", paused="0")
+            await event.answer()
+            await event.reply("⏹ Scan stopped. Progress saved.")
+            
+        elif data == "show_status":
+            state = await db.get_state()
+            counts = await db.get_counts()
+            concurrency = await db.get_concurrency()
+            qlen = await db.queue_size()
+            
+            total = int(state.get("total", 0))
+            processed = int(state.get("processed", 0))
+            pct = (processed / total * 100) if total > 0 else 0
+            run_str = "YES" if state.get("running") == "1" else "NO"
+            pause_str = "YES" if state.get("paused") == "1" else "NO"
+            
+            msg = (
+                f"**📊 SCANNER STATUS**\n"
+                f"━━━━━━━━━━━━━━━━━━━\n"
+                f"🟢 Running  : `{run_str:<10}`\n"
+                f"⏸ Paused   : `{pause_str:<10}`\n"
+                f"📦 Total    : `{total:<10,}`\n"
+                f"✅ Processed : `{processed:<10,}`\n"
+                f"📈 Progress : `{pct:.2f}%`\n"
+                f"━━━━━━━━━━━━━━━━━━━\n"
+                f"🔴 Taken    : `{counts['taken']:<10,}`\n"
+                f"🚫 Unavail  : `{counts['unavailable']:<10,}`\n"
+                f"💰 For Sale : `{counts['forsale']:<10,}`\n"
+                f"🔨 Auction  : `{counts['auction']:<10,}`\n"
+                f"━━━━━━━━━━━━━━━━━━━\n"
+                f"⚡ Speed    : `{concurrency:<10}`\n"
+                f"📋 Queue    : `{qlen:<10,}`\n"
+            )
+            await event.answer()
+            await event.reply(msg)
+            
+        elif data == "set_speed":
+            buttons = [
+                [Button.inline("🐢 10", b"spd_10"), Button.inline("🚶 20", b"spd_20"), Button.inline("🚗 30", b"spd_30")],
+                [Button.inline("🚀 50", b"spd_50"), Button.inline("⚡ 75", b"spd_75"), Button.inline("🔥 100", b"spd_100")]
+            ]
+            await event.answer()
+            await event.reply("Select Speed:", buttons=buttons)
+            
+        elif data.startswith("spd_"):
+            val = int(data.split("_")[1])
+            await db.set_concurrency(val)
+            await event.answer()
+            await event.reply(f"⚡ Speed set to {val} concurrent workers.")
+            
+        elif data == "export_files":
+            await event.answer()
+            taken = sorted(await db.get_all_taken())
+            unavail = sorted(await db.get_all_unavailable())
+            
+            f_taken = io.BytesIO("\n".join(taken).encode('utf-8'))
+            f_taken.name = "taken.txt"
+            await client.send_file(event.chat_id, file=f_taken)
+            
+            f_unavail = io.BytesIO("\n".join(unavail).encode('utf-8'))
+            f_unavail.name = "unavailable.txt"
+            await client.send_file(event.chat_id, file=f_unavail)
+            
+            await event.reply(f"✅ Exported {len(taken)} taken, {len(unavail)} unavailable.")
+            
+        elif data == "reset_confirm":
+            buttons = [
+                [Button.inline("✅ Yes, WIPE ALL DATA", b"reset_do"), Button.inline("❌ Cancel", b"reset_cancel")]
+            ]
+            await event.answer()
+            await event.reply("⚠️ Are you sure you want to completely wipe the database?", buttons=buttons)
+            
+        elif data == "reset_do":
+            await db.flush_all()
+            await event.answer()
+            await event.reply("💣 All data wiped. Use /load to reload words.")
+            
+        elif data == "reset_cancel":
+            await event.answer()
+            await event.reply("Cancelled.")