96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
from aiogram import F, Router
|
|
from aiogram.filters import CommandStart
|
|
from aiogram.fsm.context import FSMContext
|
|
from aiogram.types import CallbackQuery, Message
|
|
|
|
from config import ADMIN_ID, BOT_USERNAME, SHOP_NAME
|
|
from db.storage import load_config
|
|
from db.users import get, register
|
|
from keyboards.user import main_menu_kb
|
|
from services.admin import is_admin
|
|
from runtime import get_bot, get_log_bot
|
|
from utils.logging import log_admin
|
|
|
|
router = Router(name="start")
|
|
|
|
|
|
def _welcome_text(cfg) -> str:
|
|
text = (
|
|
f"Welcome to <b>{SHOP_NAME}</b>!\n\n"
|
|
"You can sell cookies at these rates:\n"
|
|
)
|
|
for rate in cfg.rates.values():
|
|
text += f"- {rate.name}: {rate.price:.2f} RUB\n"
|
|
return text + (
|
|
f"\nAVG pricing applies to {cfg.avg_min_cookies}+ cookies with donations.\n"
|
|
f"The AVG price is capped at {cfg.avg_min_price:.2f}-{cfg.avg_max_price:.2f} RUB per cookie.\n\n"
|
|
"Send a .txt file or paste your cookie text."
|
|
)
|
|
|
|
|
|
async def _show_start(
|
|
message: Message,
|
|
state: FSMContext,
|
|
user_id: int,
|
|
username: str | None,
|
|
referrer: int | None = None,
|
|
) -> None:
|
|
await state.clear()
|
|
is_new = register(user_id, username, referrer)
|
|
profile = get(user_id)
|
|
cfg = load_config()
|
|
if profile and profile.is_banned:
|
|
await message.answer(f"You are banned. Reason: {profile.ban_reason or '-'}")
|
|
return
|
|
if not cfg.bot_enabled and user_id != ADMIN_ID:
|
|
await message.answer("The bot is temporarily disabled by an administrator.")
|
|
return
|
|
if is_new:
|
|
bot = get_bot() or message.bot
|
|
await log_admin(
|
|
bot,
|
|
ADMIN_ID,
|
|
f"<b>New user</b>\nID: <code>{user_id}</code>\n"
|
|
f"@{username or '-'}\nReferrer: {referrer or '-'}",
|
|
get_log_bot(),
|
|
)
|
|
|
|
text = _welcome_text(cfg)
|
|
if is_admin(user_id):
|
|
text += "\n\n<b>Admin accounts cannot sell cookies. Use /admin.</b>"
|
|
await message.answer(text, parse_mode="HTML", reply_markup=main_menu_kb())
|
|
|
|
|
|
@router.message(CommandStart())
|
|
async def cmd_start(message: Message, state: FSMContext):
|
|
argument = (message.text or "").split(maxsplit=1)
|
|
referrer = None
|
|
if len(argument) > 1 and argument[1].startswith("ref_"):
|
|
try:
|
|
referrer = int(argument[1][4:])
|
|
except ValueError:
|
|
referrer = None
|
|
await _show_start(
|
|
message,
|
|
state,
|
|
message.from_user.id,
|
|
message.from_user.username,
|
|
referrer,
|
|
)
|
|
|
|
|
|
@router.callback_query(F.data == "back_main")
|
|
async def cb_back_main(callback: CallbackQuery, state: FSMContext):
|
|
await state.clear()
|
|
try:
|
|
await callback.message.delete()
|
|
except Exception:
|
|
pass
|
|
await _show_start(
|
|
callback.message,
|
|
state,
|
|
callback.from_user.id,
|
|
callback.from_user.username,
|
|
)
|
|
await callback.answer()
|