import telebot, requests, time, threading

BOT_TOKEN = "YOUR_TELEGRAM_BOT_TOKEN"
API_KEY   = "YOUR_TAIKHOANXIN_API_KEY"
API_URL   = "https://taikhoanxin.site"

bot = telebot.TeleBot(BOT_TOKEN)
HEADERS = {"X-Api-Key": API_KEY, "Content-Type": "application/json"}
user_sessions = {}  # chat_id -> req_id

@bot.message_handler(commands=["start"])
def start(msg):
    bot.reply_to(msg,
        "🤖 Bot thuê số OTP - TaiKhoanXin\n\n"
        "📋 Lệnh:\n"
        "/sodu       - Xem số dư\n"
        "/danhsach   - Danh sách app & ID\n"
        "/thueso <ID> - Thuê số (VD: /thueso 1001)\n"
        "/huy        - Hủy yêu cầu đang chờ\n\n"
        "📱 App phổ biến:\n"
        "1001 FB | 1005 Google | 1006 Telegram\n"
        "1010 IG | 1032 TikTok | 1002 Shopee")

@bot.message_handler(commands=["sodu"])
def check_balance(msg):
    try:
        r = requests.get(f"{API_URL}/api/balance", headers=HEADERS, timeout=10)
        d = r.json()
        if d["success"]:
            bot.reply_to(msg, f"💰 Số dư: {d['data']['balance_display']}")
        else:
            bot.reply_to(msg, f"❌ Lỗi: {d['error']}")
    except Exception as e:
        bot.reply_to(msg, f"❌ Lỗi kết nối: {e}")

@bot.message_handler(commands=["danhsach"])
def list_apps(msg):
    try:
        r = requests.get(f"{API_URL}/api/apps", headers=HEADERS, timeout=10)
        d = r.json()
        if not d["success"]:
            bot.reply_to(msg, "❌ Lỗi lấy danh sách"); return
        text = "📱 Danh sách app OTP:\n\n"
        for a in d["data"]["apps"]:
            text += f"• ID {a['id']} | {a['name']} | {a['cost_display']}\n"
        text += "\nDùng: /thueso <ID>"
        bot.reply_to(msg, text)
    except Exception as e:
        bot.reply_to(msg, f"❌ Lỗi: {e}")

# --- Hàm chờ OTP chạy trong thread riêng (không làm đơ bot) ---
def poll_otp_thread(chat_id, req_id, phone, app_name, price_display):
    try:
        for _ in range(42):  # tối đa 42 x 7s ≈ 5 phút
            time.sleep(7)
            r2 = requests.get(f"{API_URL}/api/otp/{req_id}",
                              headers=HEADERS, timeout=10)
            d2 = r2.json()
            status = d2.get("data", {}).get("status")

            if status == "received":
                code = d2["data"]["code"]
                note = d2["data"].get("message", "")
                bot.send_message(chat_id,
                    f"🎉 *Mã OTP:* `{code}`\n"
                    f"📱 App: {app_name}\n"
                    f"📞 Số: `{phone}`"
                    + (f"\n📨 {note}" if note else ""),
                    parse_mode="Markdown")
                user_sessions.pop(chat_id, None)
                return

        # Hết giờ → tự động hoàn tiền
        requests.post(f"{API_URL}/api/cancel/{req_id}",
                      headers=HEADERS, timeout=10)
        bot.send_message(chat_id,
            f"⏰ Hết giờ chờ OTP!\n"
            f"📞 Số: `{phone}` | {app_name}\n"
            f"💰 Đã hoàn tiền: {price_display}",
            parse_mode="Markdown")
        user_sessions.pop(chat_id, None)
    except Exception as e:
        bot.send_message(chat_id, f"⚠️ Lỗi chờ OTP: {e}")
        user_sessions.pop(chat_id, None)

@bot.message_handler(commands=["thueso"])
def rent_number(msg):
    parts = msg.text.split()
    if len(parts) < 2:
        bot.reply_to(msg,
            "❌ Cú pháp: /thueso <app_id>\n"
            "VD: /thueso 1001\n"
            "Gõ /danhsach để xem danh sách ID")
        return
    try:
        app_id = int(parts[1])
    except ValueError:
        bot.reply_to(msg, "❌ App ID phải là số. VD: /thueso 1001"); return

    wait_msg = bot.reply_to(msg, "⏳ Đang lấy số...")
    try:
        r = requests.post(f"{API_URL}/api/rent",
            headers=HEADERS, json={"app_id": app_id}, timeout=15)
        d = r.json()
    except Exception as e:
        bot.edit_message_text(f"❌ Lỗi kết nối: {e}",
            msg.chat.id, wait_msg.message_id); return

    if not d["success"]:
        bot.edit_message_text(
            f"❌ {d.get('error', 'Lỗi không xác định')}\n"
            "💳 Gõ /sodu để kiểm tra số dư",
            msg.chat.id, wait_msg.message_id); return

    data     = d["data"]
    req_id   = data["request_id"]
    phone    = data["phone"]
    app_name = data["app_name"]
    price    = data["price_display"]
    bal_rem  = data.get("balance_remaining", 0)

    user_sessions[msg.chat.id] = req_id

    bot.edit_message_text(
        f"✅ *Đã lấy số thành công!*\n"
        f"📞 Số: `{phone}`\n"
        f"📱 App: {app_name}\n"
        f"💰 Đã trừ: {price} | Còn: {bal_rem:,}đ\n\n"
        f"🕒 Đang chờ OTP... (tối đa 5 phút)\n"
        f"_Dùng /huy để hủy & hoàn tiền_",
        msg.chat.id, wait_msg.message_id, parse_mode="Markdown")

    # ⭐ Chạy poll OTP trong thread riêng → bot tiếp tục nhận lệnh mới
    threading.Thread(
        target=poll_otp_thread,
        args=(msg.chat.id, req_id, phone, app_name, price),
        daemon=True).start()

@bot.message_handler(commands=["huy"])
def cancel(msg):
    req_id = user_sessions.get(msg.chat.id)
    if not req_id:
        bot.reply_to(msg, "ℹ️ Không có yêu cầu nào đang chờ."); return
    try:
        r = requests.post(f"{API_URL}/api/cancel/{req_id}",
                          headers=HEADERS, timeout=10)
        d = r.json()
        if d["success"]:
            refund = d["data"].get("refunded", 0)
            bot.reply_to(msg, f"✅ Đã hủy! Hoàn {refund:,}đ vào tài khoản.")
        else:
            bot.reply_to(msg, f"❌ {d.get('error')}")
    except Exception as e:
        bot.reply_to(msg, f"❌ Lỗi: {e}")
    user_sessions.pop(msg.chat.id, None)

print("🤖 Bot đang chạy...")
bot.infinity_polling(timeout=20, long_polling_timeout=15)

<!DOCTYPE html>
<html lang="vi">
<head>
  <meta charset="UTF-8">
  <title>Thuê Số OTP</title>
  <style>
    * { margin:0; padding:0; box-sizing:border-box; font-family:sans-serif; }
    body { background:#0a0e17; color:#fff; padding:20px; }
    .card { background:rgba(255,255,255,0.05); border-radius:12px;
            padding:20px; max-width:500px; margin:20px auto; }
    select,button { padding:10px 16px; border-radius:8px; border:none;
                    font-size:1rem; cursor:pointer; width:100%; margin-top:10px; }
    button { background:linear-gradient(135deg,#ff007f,#ff8c00); color:#fff; }
    #result { margin-top:16px; padding:14px; background:rgba(0,0,0,0.4);
              border-radius:8px; display:none; }
    .phone { font-size:1.5rem; font-weight:bold; color:#00f0ff; }
    .otp   { font-size:2rem; font-weight:bold; color:#10b981; }
  </style>
</head>
<body>
  <div class="card">
    <h2>🔐 Thuê Số OTP</h2>
    <select id="appSelect">
      <option value="">-- Đang tải danh sách --</option>
    </select>
    <button onclick="rentNumber()">Thuê Số Ngay</button>
    <div id="result">
      <p>Số điện thoại: <span class="phone" id="phone"></span></p>
      <p id="otp-status">⏳ Đang chờ mã OTP...</p>
      <p class="otp" id="otp-code"></p>
    </div>
  </div>
  <script>
    const API = "https://YOUR_API_URL";
    const KEY = "YOUR_API_KEY";
    const H   = {"X-Api-Key": KEY, "Content-Type":"application/json"};

    // Load danh sách app
    fetch(`${API}/api/apps`, {headers:H})
      .then(r=>r.json()).then(d=>{
        const sel = document.getElementById("appSelect");
        sel.innerHTML = d.data.apps.map(a=>
          `<option value="${a.id}">${a.name} - ${a.cost_display}</option>`
        ).join("");
      });

    async function rentNumber() {
      const appId = document.getElementById("appSelect").value;
      if (!appId) return alert("Chọn app trước!");
      const r = await fetch(`${API}/api/rent`, {
        method:"POST", headers:H,
        body: JSON.stringify({app_id: parseInt(appId)})
      });
      const d = await r.json();
      if (!d.success) return alert(d.error);
      document.getElementById("phone").textContent = d.data.phone;
      document.getElementById("result").style.display = "block";
      pollOTP(d.data.request_id);
    }

    async function pollOTP(reqId) {
      for (let i=0; i<42; i++) {
        await new Promise(r=>setTimeout(r,7000));
        const r = await fetch(`${API}/api/otp/${reqId}`, {headers:H});
        const d = await r.json();
        if (d.data.status === "received") {
          document.getElementById("otp-status").textContent = "✅ Mã OTP:";
          document.getElementById("otp-code").textContent = d.data.code;
          return;
        }
      }
      // Hủy & hoàn tiền
      await fetch(`${API}/api/cancel/${reqId}`, {method:"POST",headers:H});
      document.getElementById("otp-status").textContent = "⏰ Hết giờ, đã hoàn tiền.";
    }
  </script>
</body>
</html>