Add modular Roblox buyer bot and documentation
This commit is contained in:
@@ -0,0 +1,459 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from aiogram import F, Router
|
||||
from aiogram.filters import Command
|
||||
from aiogram.fsm.context import FSMContext
|
||||
from aiogram.types import (
|
||||
CallbackQuery,
|
||||
FSInputFile,
|
||||
InlineKeyboardButton,
|
||||
InlineKeyboardMarkup,
|
||||
Message,
|
||||
)
|
||||
|
||||
from config import ADMIN_ID, ROBSEC_FILE
|
||||
from db.storage import load_config
|
||||
from db.users import get
|
||||
from keyboards.admin import admin_menu_kb
|
||||
from keyboards.user import back_kb
|
||||
from services.admin import (
|
||||
add_balance,
|
||||
add_treasury,
|
||||
ban_user,
|
||||
broadcast,
|
||||
clear_robsec,
|
||||
find_user,
|
||||
is_admin,
|
||||
robsec_info,
|
||||
sub_balance,
|
||||
toggle_bot,
|
||||
toggle_shop,
|
||||
toggle_user_admin,
|
||||
update_min_withdraw,
|
||||
update_rate,
|
||||
)
|
||||
from services.withdrawals import check_treasury_invoice, create_treasury_invoice
|
||||
from states.admin import Form
|
||||
from runtime import get_bot
|
||||
from utils.logging import safe_edit
|
||||
|
||||
router = Router(name="admin")
|
||||
|
||||
|
||||
def _admin(user_id: int) -> bool:
|
||||
return user_id == ADMIN_ID or is_admin(user_id)
|
||||
|
||||
|
||||
def _user_card(uid: int) -> tuple[str, InlineKeyboardMarkup] | None:
|
||||
profile = get(uid)
|
||||
if not profile:
|
||||
return None
|
||||
text = (
|
||||
"<b>User</b>\n\n"
|
||||
f"ID: <code>{uid}</code>\n"
|
||||
f"Username: @{profile.username or '-'}\n"
|
||||
f"Registered: {profile.registered or '-'}\n"
|
||||
f"Balance: {profile.balance:.2f} RUB\n"
|
||||
f"Total earned: {profile.total_earned:.2f} RUB\n"
|
||||
f"Cookies: {profile.cookies_loaded}\n"
|
||||
f"Referrals: {len(profile.referrals)}\n"
|
||||
f"Admin: {'yes' if profile.is_admin else 'no'}\n"
|
||||
f"Banned: {'yes: ' + profile.ban_reason if profile.is_banned else 'no'}"
|
||||
)
|
||||
keyboard = InlineKeyboardMarkup(
|
||||
inline_keyboard=[
|
||||
[
|
||||
InlineKeyboardButton(text="Add balance", callback_data=f"u_bal_add_{uid}"),
|
||||
InlineKeyboardButton(text="Subtract balance", callback_data=f"u_bal_sub_{uid}"),
|
||||
],
|
||||
[InlineKeyboardButton(
|
||||
text="Unban" if profile.is_banned else "Ban",
|
||||
callback_data=f"u_ban_{uid}",
|
||||
)],
|
||||
[InlineKeyboardButton(
|
||||
text="Remove admin" if profile.is_admin else "Grant admin",
|
||||
callback_data=f"u_admin_{uid}",
|
||||
)],
|
||||
[InlineKeyboardButton(text="Back", callback_data="admin_back")],
|
||||
]
|
||||
)
|
||||
return text, keyboard
|
||||
|
||||
|
||||
@router.message(Command("admin"))
|
||||
async def cmd_admin(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
await state.clear()
|
||||
await message.answer("<b>Admin panel</b>", parse_mode="HTML", reply_markup=admin_menu_kb())
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_close")
|
||||
async def admin_close(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
await state.clear()
|
||||
try:
|
||||
await callback.message.delete()
|
||||
except Exception:
|
||||
pass
|
||||
await callback.answer()
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_back")
|
||||
async def admin_back(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
await state.clear()
|
||||
await safe_edit(callback, "<b>Admin panel</b>", reply_markup=admin_menu_kb())
|
||||
await callback.answer()
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_toggle_shop")
|
||||
async def admin_toggle_shop(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
enabled = toggle_shop()
|
||||
await callback.message.edit_reply_markup(reply_markup=admin_menu_kb())
|
||||
await callback.answer(f"Shop: {'ON' if enabled else 'OFF'}")
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_toggle_bot")
|
||||
async def admin_toggle_bot(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
enabled = toggle_bot()
|
||||
await callback.message.edit_reply_markup(reply_markup=admin_menu_kb())
|
||||
await callback.answer(f"Bot: {'ON' if enabled else 'OFF'}")
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_robsec_get")
|
||||
async def admin_robsec_get(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
lines, size = robsec_info()
|
||||
if not size:
|
||||
await callback.answer("robsec.txt is empty", show_alert=True)
|
||||
return
|
||||
try:
|
||||
bot = get_bot() or callback.message.bot
|
||||
await bot.send_document(
|
||||
callback.from_user.id,
|
||||
FSInputFile(ROBSEC_FILE),
|
||||
caption=f"robsec.txt\nLines: {lines}\nSize: {size} bytes",
|
||||
)
|
||||
await callback.answer("Sent")
|
||||
except Exception as exc:
|
||||
await callback.answer(f"Error: {exc}", show_alert=True)
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_robsec_clear")
|
||||
async def admin_robsec_clear_prompt(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
lines, _ = robsec_info()
|
||||
keyboard = InlineKeyboardMarkup(
|
||||
inline_keyboard=[
|
||||
[InlineKeyboardButton(text="Yes, clear", callback_data="admin_robsec_clear_yes")],
|
||||
[InlineKeyboardButton(text="Cancel", callback_data="admin_back")],
|
||||
]
|
||||
)
|
||||
await safe_edit(callback, f"Clear robsec.txt? Current lines: <b>{lines}</b>", reply_markup=keyboard)
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_robsec_clear_yes")
|
||||
async def admin_robsec_clear_yes(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
clear_robsec()
|
||||
await safe_edit(callback, "<b>robsec.txt cleared</b>", reply_markup=admin_menu_kb())
|
||||
await callback.answer("Cleared")
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_rates")
|
||||
async def admin_rates(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
cfg = load_config()
|
||||
rows = [
|
||||
[InlineKeyboardButton(text=f"{rate.name}: {rate.price:.2f} RUB", callback_data=f"admin_rate_{key}")]
|
||||
for key, rate in cfg.rates.items()
|
||||
]
|
||||
rows.append([InlineKeyboardButton(text="Back", callback_data="admin_back")])
|
||||
await safe_edit(
|
||||
callback,
|
||||
"<b>Rate management</b>\nSelect a rate:",
|
||||
reply_markup=InlineKeyboardMarkup(inline_keyboard=rows),
|
||||
)
|
||||
await callback.answer()
|
||||
|
||||
|
||||
@router.callback_query(F.data.startswith("admin_rate_"))
|
||||
async def admin_rate_select(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
key = callback.data.removeprefix("admin_rate_")
|
||||
cfg = load_config()
|
||||
rate = cfg.rates.get(key)
|
||||
if not rate:
|
||||
await callback.answer("Rate not found", show_alert=True)
|
||||
return
|
||||
await state.update_data(rate_key=key)
|
||||
await state.set_state(Form.admin_edit_rate)
|
||||
await safe_edit(
|
||||
callback,
|
||||
f"Rate: <b>{rate.name}</b>\nCurrent price: {rate.price:.2f} RUB\n\nSend a new price:",
|
||||
reply_markup=back_kb("admin_back"),
|
||||
)
|
||||
|
||||
|
||||
@router.message(Form.admin_edit_rate)
|
||||
async def admin_edit_rate_save(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
try:
|
||||
price = float((message.text or "").replace(",", "."))
|
||||
if price < 0:
|
||||
raise ValueError
|
||||
except ValueError:
|
||||
await message.answer("Send a non-negative number.")
|
||||
return
|
||||
data = await state.get_data()
|
||||
key = data.get("rate_key")
|
||||
cfg = load_config()
|
||||
if key not in cfg.rates:
|
||||
await state.clear()
|
||||
return
|
||||
update_rate(key, price)
|
||||
await state.clear()
|
||||
await message.answer(f"Rate updated: {cfg.rates[key].name} = {price:.2f} RUB", reply_markup=admin_menu_kb())
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_min_withdraw")
|
||||
async def admin_min_withdraw(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
await state.set_state(Form.admin_min_withdraw)
|
||||
await safe_edit(
|
||||
callback,
|
||||
f"Current minimum withdrawal: {load_config().min_withdraw:.2f} RUB\nSend a new value:",
|
||||
reply_markup=back_kb("admin_back"),
|
||||
)
|
||||
|
||||
|
||||
@router.message(Form.admin_min_withdraw)
|
||||
async def admin_min_withdraw_save(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
try:
|
||||
value = float((message.text or "").replace(",", "."))
|
||||
if value <= 0:
|
||||
raise ValueError
|
||||
except ValueError:
|
||||
await message.answer("Send a positive number.")
|
||||
return
|
||||
update_min_withdraw(value)
|
||||
await state.clear()
|
||||
await message.answer(f"Minimum withdrawal updated to {value:.2f} RUB", reply_markup=admin_menu_kb())
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_treasury")
|
||||
async def admin_treasury(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
await state.set_state(Form.admin_treasury_amount)
|
||||
await safe_edit(callback, "How much RUB should be added to the treasury?", reply_markup=back_kb("admin_back"))
|
||||
|
||||
|
||||
@router.message(Form.admin_treasury_amount)
|
||||
async def admin_treasury_amount(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
try:
|
||||
amount = float((message.text or "").replace(",", "."))
|
||||
if amount <= 0:
|
||||
raise ValueError
|
||||
except ValueError:
|
||||
await message.answer("Send a positive number.")
|
||||
return
|
||||
await state.clear()
|
||||
invoice = await create_treasury_invoice(amount)
|
||||
if not invoice:
|
||||
await message.answer("CryptoBot invoice creation failed.", reply_markup=admin_menu_kb())
|
||||
return
|
||||
pay_url, invoice_id, amount_usdt = invoice
|
||||
keyboard = InlineKeyboardMarkup(
|
||||
inline_keyboard=[
|
||||
[InlineKeyboardButton(text="Pay", url=pay_url)],
|
||||
[InlineKeyboardButton(text="Check payment", callback_data=f"treasury_check_{invoice_id}_{amount}")],
|
||||
]
|
||||
)
|
||||
await message.answer(
|
||||
f"Treasury invoice: <b>{amount:.2f} RUB</b> (~{amount_usdt} USDT)\nPay it and check the status.",
|
||||
parse_mode="HTML",
|
||||
reply_markup=keyboard,
|
||||
)
|
||||
|
||||
|
||||
@router.callback_query(F.data.startswith("treasury_check_"))
|
||||
async def treasury_check(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
_, _, invoice_id, amount = callback.data.split("_", 3)
|
||||
try:
|
||||
paid = await check_treasury_invoice(int(invoice_id))
|
||||
amount_rub = float(amount)
|
||||
except (ValueError, TypeError):
|
||||
paid = False
|
||||
amount_rub = 0.0
|
||||
if not paid:
|
||||
await callback.answer("Payment has not arrived yet", show_alert=True)
|
||||
return
|
||||
balance = add_treasury(amount_rub)
|
||||
await safe_edit(callback, f"Treasury topped up by {amount_rub:.2f} RUB. Total: {balance:.2f} RUB", reply_markup=admin_menu_kb())
|
||||
await callback.answer("Paid")
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_broadcast")
|
||||
async def admin_broadcast_prompt(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
await state.set_state(Form.admin_broadcast)
|
||||
await safe_edit(callback, "Send the broadcast text or a photo with caption.", reply_markup=back_kb("admin_back"))
|
||||
|
||||
|
||||
@router.message(Form.admin_broadcast)
|
||||
async def admin_broadcast_send(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
await state.clear()
|
||||
sent, failed = await broadcast(
|
||||
message.bot,
|
||||
text=message.text,
|
||||
photo_id=message.photo[-1].file_id if message.photo else None,
|
||||
caption=message.caption,
|
||||
)
|
||||
await message.answer(f"Broadcast complete. Sent: {sent}, failed: {failed}.", reply_markup=admin_menu_kb())
|
||||
|
||||
|
||||
@router.callback_query(F.data == "admin_users")
|
||||
async def admin_users(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
await state.set_state(Form.admin_find_user)
|
||||
await safe_edit(callback, "Send a user ID or @username.", reply_markup=back_kb("admin_back"))
|
||||
|
||||
|
||||
@router.message(Form.admin_find_user)
|
||||
async def admin_find_user(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
uid = find_user((message.text or "").strip())
|
||||
await state.clear()
|
||||
if uid is None:
|
||||
await message.answer("User not found.", reply_markup=admin_menu_kb())
|
||||
return
|
||||
card = _user_card(uid)
|
||||
if card:
|
||||
await message.answer(card[0], parse_mode="HTML", reply_markup=card[1])
|
||||
|
||||
|
||||
@router.callback_query(F.data.startswith("u_bal_add_"))
|
||||
async def u_bal_add(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
uid = int(callback.data.removeprefix("u_bal_add_"))
|
||||
await state.update_data(target_uid=uid)
|
||||
await state.set_state(Form.admin_user_balance_add)
|
||||
await safe_edit(callback, "Send the amount to add:", reply_markup=back_kb("admin_back"))
|
||||
|
||||
|
||||
@router.message(Form.admin_user_balance_add)
|
||||
async def u_bal_add_save(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
try:
|
||||
amount = float((message.text or "").replace(",", "."))
|
||||
if amount <= 0:
|
||||
raise ValueError
|
||||
except ValueError:
|
||||
await message.answer("Send a positive number.")
|
||||
return
|
||||
uid = (await state.get_data()).get("target_uid")
|
||||
add_balance(uid, amount)
|
||||
await state.clear()
|
||||
await message.answer(f"Added {amount:.2f} RUB to {uid}.", reply_markup=admin_menu_kb())
|
||||
|
||||
|
||||
@router.callback_query(F.data.startswith("u_bal_sub_"))
|
||||
async def u_bal_sub(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
uid = int(callback.data.removeprefix("u_bal_sub_"))
|
||||
await state.update_data(target_uid=uid)
|
||||
await state.set_state(Form.admin_user_balance_sub)
|
||||
await safe_edit(callback, "Send the amount to subtract:", reply_markup=back_kb("admin_back"))
|
||||
|
||||
|
||||
@router.message(Form.admin_user_balance_sub)
|
||||
async def u_bal_sub_save(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
try:
|
||||
amount = float((message.text or "").replace(",", "."))
|
||||
if amount <= 0:
|
||||
raise ValueError
|
||||
except ValueError:
|
||||
await message.answer("Send a positive number.")
|
||||
return
|
||||
uid = (await state.get_data()).get("target_uid")
|
||||
sub_balance(uid, amount)
|
||||
await state.clear()
|
||||
await message.answer(f"Subtracted {amount:.2f} RUB from {uid}.", reply_markup=admin_menu_kb())
|
||||
|
||||
|
||||
@router.callback_query(F.data.startswith("u_ban_"))
|
||||
async def u_ban(callback: CallbackQuery, state: FSMContext):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
uid = int(callback.data.removeprefix("u_ban_"))
|
||||
profile = get(uid)
|
||||
if not profile:
|
||||
return
|
||||
if profile.is_banned:
|
||||
from services.admin import unban_user
|
||||
|
||||
unban_user(uid)
|
||||
await callback.answer("Unbanned")
|
||||
else:
|
||||
await state.update_data(target_uid=uid)
|
||||
await state.set_state(Form.admin_user_ban_reason)
|
||||
await safe_edit(callback, "Send the ban reason:", reply_markup=back_kb("admin_back"))
|
||||
return
|
||||
card = _user_card(uid)
|
||||
if card:
|
||||
await safe_edit(callback, card[0], reply_markup=card[1])
|
||||
|
||||
|
||||
@router.message(Form.admin_user_ban_reason)
|
||||
async def u_ban_reason(message: Message, state: FSMContext):
|
||||
if not _admin(message.from_user.id):
|
||||
return
|
||||
uid = (await state.get_data()).get("target_uid")
|
||||
ban_user(uid, (message.text or "No reason").strip())
|
||||
await state.clear()
|
||||
card = _user_card(uid)
|
||||
if card:
|
||||
await message.answer(card[0], parse_mode="HTML", reply_markup=card[1])
|
||||
|
||||
|
||||
@router.callback_query(F.data.startswith("u_admin_"))
|
||||
async def u_admin(callback: CallbackQuery):
|
||||
if not _admin(callback.from_user.id):
|
||||
return
|
||||
uid = int(callback.data.removeprefix("u_admin_"))
|
||||
enabled = toggle_user_admin(uid)
|
||||
await callback.answer(f"Admin: {'ON' if enabled else 'OFF'}")
|
||||
card = _user_card(uid)
|
||||
if card:
|
||||
await safe_edit(callback, card[0], reply_markup=card[1])
|
||||
Reference in New Issue
Block a user