bot.py

#!/usr/bin/env python3
"""
STP Proxy Bot – Telegram bot for selling residential proxies with crypto payments.
"""

import asyncio
import logging
import secrets
import time
from aiogram import Bot, Dispatcher, F, types
from aiogram.filters import Command
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton

from config import (
    TELEGRAM_TOKEN, OPERATOR_ID, NOWPAYMENTS_API_KEY,
    PUBLIC_DOMAIN, PUBLIC_PORT, PRICE_PER_PROXY,
    WEBHOOK_HOST, WEBHOOK_PORT, WEBHOOK_URL
)
from db import (
    init_db, get_available_child_for_country, set_child_credentials,
    add_issued_proxy, get_user_balance, update_balance, deduct_balance
)
from nowpayments import NowPayments

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

bot = Bot(token=TELEGRAM_TOKEN)
dp = Dispatcher()

# ---------- Платежи ----------
payments = NowPayments(api_key=NOWPAYMENTS_API_KEY, webhook_path="/webhook")

# ---------- Список стран (пример, можно менять) ----------
COUNTRIES = {
    "GB": "🇬🇧 United Kingdom",
    "DE": "🇩🇪 Germany",
    "ES": "🇪🇸 Spain",
    "FR": "🇫🇷 France",
    "IT": "🇮🇹 Italy",
    "US": "🇺🇸 USA",
}

# ---------- Колбэк при успешной оплате ----------
def on_payment_success(user_id: int, amount: float, invoice_id: str):
    logger.info(f"Payment success: user {user_id}, amount {amount}, invoice {invoice_id}")
    # Здесь можно добавить логику, если нужно уведомить оператора
    asyncio.create_task(
        bot.send_message(
            OPERATOR_ID,
            f"✅ Оплата получена!\nПользователь ID: {user_id}\nСумма: {amount} USD"
        )
    )

payments.set_on_payment_callback(on_payment_success)

# ---------- Клавиатуры ----------
def main_keyboard():
    return ReplyKeyboardMarkup(
        keyboard=[
            [KeyboardButton(text="📘 Инструкция"), KeyboardButton(text="📊 Тарифы")],
            [KeyboardButton(text="🚀 Купить"), KeyboardButton(text="🌟 О нас")],
        ],
        resize_keyboard=True
    )

def countries_keyboard():
    buttons = [[KeyboardButton(text=name)] for name in COUNTRIES.values()]
    buttons.append([KeyboardButton(text="« Назад")])
    return ReplyKeyboardMarkup(keyboard=buttons, resize_keyboard=True)

# ---------- Команды ----------
@dp.message(Command("start"))
async def start_cmd(message: types.Message):
    await message.answer(
        "🌟 Добро пожаловать в STP Proxy Bot!\n\n"
        "Я продаю резидентские прокси с оплатой в криптовалюте.\n"
        "Выберите действие ниже 👇",
        reply_markup=main_keyboard()
    )

@dp.message(F.text == "📘 Инструкция")
async def instruction_cmd(message: types.Message):
    await message.answer(
        "📘 **Как использовать прокси:**\n\n"
        "1. Пополните баланс через кнопку «Купить».\n"
        "2. Выберите страну.\n"
        "3. Получите прокси в формате:\n"
        "`host:port:login:password`\n\n"
        "Прокси выдаётся на 24 часа. Поддерживаются SOCKS5 и HTTP CONNECT.",
        parse_mode="Markdown"
    )

@dp.message(F.text == "📊 Тарифы")
async def tariffs_cmd(message: types.Message):
    await message.answer(
        "📊 **Тарифы:**\n\n"
        "🔹 1 день — 0.50 USD\n"
        "🔹 7 дней — 3.00 USD\n"
        "🔹 30 дней — 10.00 USD\n\n"
        "Выберите «Купить» для оплаты.",
        parse_mode="Markdown"
    )

@dp.message(F.text == "🌟 О нас")
async def about_cmd(message: types.Message):
    await message.answer(
        "🌟 **О нас:**\n\n"
        "Мы предоставляем надёжные резидентские прокси для обхода блокировок.\n"
        "Наши прокси работают с любыми приложениями и браузерами.\n"
        "Поддержка: @volttecc"
    )

@dp.message(F.text == "🚀 Купить")
async def buy_start(message: types.Message):
    await message.answer(
        "Выберите страну:",
        reply_markup=countries_keyboard()
    )

@dp.message(F.text.in_(COUNTRIES.values()))
async def country_selected(message: types.Message):
    user_id = message.from_user.id
    # Находим код страны по названию
    country_code = None
    for code, name in COUNTRIES.items():
        if name == message.text:
            country_code = code
            break
    if not country_code:
        await message.answer("❌ Страна не найдена.")
        return

    # Проверяем баланс
    balance = get_user_balance(user_id)
    if balance < PRICE_PER_PROXY:
        await message.answer(
            f"❌ Недостаточно средств. Нужно ${PRICE_PER_PROXY:.2f}, у вас ${balance:.2f}.\n"
            "Пополните баланс через кнопку «Купить»."
        )
        return

    # Ищем свободный канал
    child = get_available_child_for_country(country_code)
    if not child:
        await message.answer("❌ Нет свободных прокси для этой страны. Попробуйте позже.")
        return

    # Генерируем логин/пароль
    login = f"u{secrets.token_hex(4)}"
    password = secrets.token_hex(6)
    child_id = child["id"]
    set_child_credentials(child_id, login, password)

    # Записываем выдачу на 24 часа
    expires_at = int(time.time()) + 86400
    add_issued_proxy(user_id, child_id, expires_at)

    # Списываем средства
    deduct_balance(user_id, PRICE_PER_PROXY)

    # Формируем строку прокси
    proxy_str = f"{PUBLIC_DOMAIN}:{PUBLIC_PORT}:{login}:{password}"
    await message.answer(
        f"✅ Прокси для {COUNTRIES[country_code]} куплен!\n\n"
        f"📡 Данные:\n`{proxy_str}`\n\n"
        f"⏳ Действует до: {time.ctime(expires_at)}\n"
        f"💰 Остаток: ${get_user_balance(user_id):.2f}",
        parse_mode="Markdown",
        reply_markup=main_keyboard()
    )

@dp.message(F.text == "« Назад")
async def back_to_main(message: types.Message):
    await message.answer("Главное меню:", reply_markup=main_keyboard())

@dp.message()
async def unknown(message: types.Message):
    await message.answer("Используйте кнопки меню.", reply_markup=main_keyboard())

# ---------- Запуск вебхук-сервера ----------
async def run_webhook():
    await payments.start_webhook_server(host=WEBHOOK_HOST, port=WEBHOOK_PORT)

# ---------- Запуск бота ----------
async def main():
    init_db()  # создаём таблицы, если их нет
    asyncio.create_task(run_webhook())
    logger.info("STP Proxy Bot started")
    await dp.start_polling(bot)

if __name__ == "__main__":
    asyncio.run(main())