Vladless
319 lines
12 KiB
Python
319 lines
12 KiB
Python
import hashlib
|
||
|
||
from aiogram import F, Router, types
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
|
||
from aiogram.utils.keyboard import InlineKeyboardBuilder
|
||
from aiohttp import web
|
||
from loguru import logger
|
||
from robokassa import HashAlgorithm, Robokassa
|
||
|
||
from bot import bot
|
||
from config import ROBOKASSA_ENABLE, ROBOKASSA_LOGIN, ROBOKASSA_PASSWORD1, ROBOKASSA_PASSWORD2, ROBOKASSA_TEST_MODE
|
||
from database import add_connection, check_connection_exists, get_key_count, update_balance
|
||
from handlers.texts import PAYMENT_OPTIONS
|
||
|
||
router = Router()
|
||
|
||
|
||
class ReplenishBalanceState(StatesGroup):
|
||
choosing_amount_robokassa = State()
|
||
waiting_for_payment_confirmation_robokassa = State()
|
||
|
||
|
||
if ROBOKASSA_ENABLE:
|
||
robokassa = Robokassa(
|
||
merchant_login=ROBOKASSA_LOGIN,
|
||
password1=ROBOKASSA_PASSWORD1,
|
||
password2=ROBOKASSA_PASSWORD2,
|
||
algorithm=HashAlgorithm.md5,
|
||
is_test=ROBOKASSA_TEST_MODE,
|
||
)
|
||
|
||
logger.info("Robokassa initialized with login: {}", ROBOKASSA_LOGIN)
|
||
|
||
|
||
def generate_payment_link(amount, inv_id, description, tg_id):
|
||
"""Генерация ссылки на оплату."""
|
||
logger.debug(
|
||
f"Generating payment link for amount: {amount}, inv_id: {inv_id}, description: {description}"
|
||
)
|
||
payment_link = robokassa._payment.link.generate_by_script(
|
||
out_sum=amount, inv_id=inv_id, description="пополнение баланса", id=f"{tg_id}"
|
||
)
|
||
logger.info(f"Generated payment link: {payment_link}")
|
||
return payment_link
|
||
|
||
|
||
async def send_message_with_deletion(
|
||
chat_id, text, reply_markup=None, state=None, message_key="last_message_id"
|
||
):
|
||
if state:
|
||
try:
|
||
state_data = await state.get_data()
|
||
previous_message_id = state_data.get(message_key)
|
||
|
||
if previous_message_id:
|
||
logger.debug(
|
||
f"Deleting previous message with ID: {previous_message_id}"
|
||
)
|
||
await bot.delete_message(
|
||
chat_id=chat_id, message_id=previous_message_id
|
||
)
|
||
|
||
sent_message = await bot.send_message(
|
||
chat_id=chat_id, text=text, reply_markup=reply_markup
|
||
)
|
||
await state.update_data({message_key: sent_message.message_id})
|
||
|
||
logger.debug(f"Sent new message with ID: {sent_message.message_id}")
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при удалении/отправке сообщения: {e}")
|
||
return None
|
||
|
||
return sent_message
|
||
|
||
|
||
@router.callback_query(F.data == "pay_robokassa")
|
||
async def process_callback_pay_robokassa(
|
||
callback_query: types.CallbackQuery, state: FSMContext
|
||
):
|
||
tg_id = callback_query.from_user.id
|
||
logger.info(f"User {tg_id} initiated Robokassa payment.")
|
||
|
||
builder = InlineKeyboardBuilder()
|
||
for i in range(0, len(PAYMENT_OPTIONS), 2):
|
||
if i + 1 < len(PAYMENT_OPTIONS):
|
||
builder.row(
|
||
InlineKeyboardButton(
|
||
text=PAYMENT_OPTIONS[i]["text"],
|
||
callback_data=f'robokassa_amount|{PAYMENT_OPTIONS[i]["callback_data"]}',
|
||
),
|
||
InlineKeyboardButton(
|
||
text=PAYMENT_OPTIONS[i + 1]["text"],
|
||
callback_data=f'robokassa_amount|{PAYMENT_OPTIONS[i + 1]["callback_data"]}',
|
||
),
|
||
)
|
||
else:
|
||
builder.row(
|
||
InlineKeyboardButton(
|
||
text=PAYMENT_OPTIONS[i]["text"],
|
||
callback_data=f'robokassa_amount|{PAYMENT_OPTIONS[i]["callback_data"]}',
|
||
)
|
||
)
|
||
builder.row(
|
||
InlineKeyboardButton(
|
||
text="💰 Ввести свою сумму", callback_data="enter_custom_amount_robokassa"
|
||
)
|
||
)
|
||
builder.row(InlineKeyboardButton(text="⬅️ Назад", callback_data="back_to_profile"))
|
||
|
||
key_count = await get_key_count(tg_id)
|
||
|
||
if key_count == 0:
|
||
exists = await check_connection_exists(tg_id)
|
||
if not exists:
|
||
await add_connection(tg_id, balance=0.0, trial=0)
|
||
logger.info(f"Created new connection for user {tg_id} with balance 0.0.")
|
||
|
||
try:
|
||
await bot.delete_message(
|
||
chat_id=tg_id, message_id=callback_query.message.message_id
|
||
)
|
||
logger.debug(f"Deleted message with ID: {callback_query.message.message_id}")
|
||
except Exception as e:
|
||
logger.error(f"Не удалось удалить сообщение: {e}")
|
||
|
||
await bot.send_message(
|
||
chat_id=tg_id,
|
||
text="Выберите сумму пополнения:",
|
||
reply_markup=builder.as_markup(),
|
||
)
|
||
await state.set_state(ReplenishBalanceState.choosing_amount_robokassa)
|
||
logger.info(f"Displayed amount selection for user {tg_id}.")
|
||
await callback_query.answer()
|
||
|
||
|
||
@router.callback_query(F.data.startswith("robokassa_amount|"))
|
||
async def process_amount_selection(
|
||
callback_query: types.CallbackQuery, state: FSMContext
|
||
):
|
||
logger.info(f"Получены данные callback_data: {callback_query.data}")
|
||
|
||
data = callback_query.data.split("|")
|
||
if len(data) != 3 or data[1] != "amount":
|
||
logger.error("Ошибка: callback_data не соответствует формату.")
|
||
await send_message_with_deletion(
|
||
chat_id=callback_query.from_user.id,
|
||
text="Неверные данные для выбора суммы.",
|
||
state=state,
|
||
)
|
||
await callback_query.answer("Ошибка: данные повреждены.")
|
||
return
|
||
|
||
amount_str = data[2]
|
||
try:
|
||
amount = int(amount_str)
|
||
if amount <= 0:
|
||
raise ValueError("Сумма должна быть положительным числом.")
|
||
except ValueError as e:
|
||
logger.error(f"Некорректное значение суммы: {amount_str}. Ошибка: {e}")
|
||
await send_message_with_deletion(
|
||
chat_id=callback_query.from_user.id,
|
||
text="Некорректная сумма. Попробуйте снова.",
|
||
state=state,
|
||
)
|
||
await callback_query.answer("Некорректная сумма.")
|
||
return
|
||
|
||
await state.update_data(amount=amount)
|
||
logger.info(f"User {callback_query.from_user.id} selected amount: {amount}.")
|
||
inv_id = 0
|
||
|
||
tg_id = callback_query.from_user.id
|
||
payment_url = generate_payment_link(amount, inv_id, "Пополнение баланса", tg_id)
|
||
|
||
logger.info(f"Payment URL for user {callback_query.from_user.id}: {payment_url}")
|
||
|
||
confirm_keyboard = InlineKeyboardMarkup(
|
||
inline_keyboard=[
|
||
[InlineKeyboardButton(text="Оплатить", url=payment_url)],
|
||
[InlineKeyboardButton(text="⬅️ Назад", callback_data="pay_robokassa")],
|
||
]
|
||
)
|
||
|
||
await callback_query.message.edit_text(
|
||
text=f"Вы выбрали пополнение на {amount} рублей. Для оплаты перейдите по ссылке ниже:",
|
||
reply_markup=confirm_keyboard,
|
||
)
|
||
logger.info(f"Payment link sent to user {callback_query.from_user.id}.")
|
||
await callback_query.answer()
|
||
|
||
|
||
async def robokassa_webhook(request):
|
||
"""Обработка webhook-уведомлений от Robokassa с учетом shp_id."""
|
||
try:
|
||
params = await request.post()
|
||
|
||
logger.info(f"Received webhook params: {params}")
|
||
|
||
amount = params.get("OutSum")
|
||
inv_id = params.get("InvId")
|
||
shp_id = params.get("shp_id")
|
||
signature_value = params.get("SignatureValue")
|
||
|
||
logger.info(
|
||
f"OutSum: {amount}, InvId: {inv_id}, shp_id: {shp_id}, SignatureValue: {signature_value}"
|
||
)
|
||
|
||
if not check_payment_signature(params):
|
||
logger.error("Неверная подпись или данные запроса.")
|
||
return web.Response(status=400)
|
||
|
||
if not amount or not inv_id or not shp_id:
|
||
logger.error("Отсутствуют обязательные параметры.")
|
||
return web.Response(status=400)
|
||
|
||
tg_id = shp_id
|
||
|
||
logger.info(f"Processing payment for user {tg_id} with amount {amount}.")
|
||
|
||
await update_balance(int(tg_id), float(amount))
|
||
await send_payment_success_notification(tg_id, float(amount))
|
||
|
||
logger.info(f"Payment successful. Balance updated for user {tg_id}.")
|
||
|
||
return web.Response(text=f"OK{inv_id}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"Error processing webhook: {e}")
|
||
return web.Response(status=500)
|
||
|
||
|
||
def check_payment_signature(params):
|
||
"""Проверка подписи запроса от Robokassa с учетом shp_id."""
|
||
out_sum = params.get("OutSum")
|
||
inv_id = params.get("InvId")
|
||
signature_value = params.get("SignatureValue")
|
||
shp_id = params.get("shp_id")
|
||
|
||
signature_string = f"{out_sum}:{inv_id}:{ROBOKASSA_PASSWORD2}:shp_id={shp_id}"
|
||
|
||
logger.info(f"Signature string before hashing: {signature_string}")
|
||
|
||
expected_signature = (
|
||
hashlib.md5(signature_string.encode("utf-8")).hexdigest().upper()
|
||
)
|
||
|
||
logger.info(f"Expected signature: {expected_signature}")
|
||
logger.info(f"Received signature: {signature_value}")
|
||
|
||
return signature_value.upper() == expected_signature.upper()
|
||
|
||
|
||
async def send_payment_success_notification(user_id: int, amount: float):
|
||
builder = InlineKeyboardBuilder()
|
||
builder.row(
|
||
InlineKeyboardButton(text="Перейти в профиль", callback_data="view_profile")
|
||
)
|
||
|
||
await bot.send_message(
|
||
chat_id=user_id,
|
||
text=f"Ваш баланс успешно пополнен на {amount} рублей. Спасибо за оплату!",
|
||
reply_markup=builder.as_markup(),
|
||
)
|
||
logger.info(f"Sent payment success notification to user {user_id}.")
|
||
|
||
|
||
@router.callback_query(F.data == "enter_custom_amount_robokassa")
|
||
async def process_custom_amount_selection(
|
||
callback_query: types.CallbackQuery, state: FSMContext
|
||
):
|
||
tg_id = callback_query.from_user.id
|
||
logger.info(f"User {tg_id} chose to enter a custom amount.")
|
||
|
||
await callback_query.message.edit_text(
|
||
text="Пожалуйста, введите сумму пополнения в рублях (например, 150):"
|
||
)
|
||
await state.set_state(
|
||
ReplenishBalanceState.waiting_for_payment_confirmation_robokassa
|
||
)
|
||
await callback_query.answer()
|
||
|
||
|
||
@router.message(ReplenishBalanceState.waiting_for_payment_confirmation_robokassa)
|
||
async def handle_custom_amount_input(message: types.Message, state: FSMContext):
|
||
tg_id = message.from_user.id
|
||
logger.info(f"User {tg_id} entered custom amount: {message.text}")
|
||
inv_id = 0
|
||
|
||
try:
|
||
amount = int(message.text)
|
||
if amount <= 0:
|
||
raise ValueError("Сумма должна быть положительным числом.")
|
||
|
||
await state.update_data(amount=amount)
|
||
|
||
payment_url = generate_payment_link(amount, inv_id, "Пополнение баланса", tg_id)
|
||
|
||
logger.info(f"Generated payment link for user {tg_id}: {payment_url}")
|
||
|
||
confirm_keyboard = InlineKeyboardMarkup(
|
||
inline_keyboard=[
|
||
[InlineKeyboardButton(text="Оплатить", url=payment_url)],
|
||
[InlineKeyboardButton(text="⬅️ Назад", callback_data="pay_robokassa")],
|
||
]
|
||
)
|
||
|
||
await message.answer(
|
||
text=f"Вы выбрали пополнение на {amount} рублей. Для оплаты перейдите по ссылке ниже:",
|
||
reply_markup=confirm_keyboard,
|
||
)
|
||
await state.clear()
|
||
except ValueError as e:
|
||
logger.error(f"Некорректная сумма от пользователя {tg_id}: {e}")
|
||
await message.answer(
|
||
text="Введите корректную сумму в рублях (целое положительное число)."
|
||
)
|