fix(bot): resolve required channel join link
This commit is contained in:
@@ -77,14 +77,20 @@ SETTINGS_MANIFEST: List[SettingField] = [
|
||||
"int",
|
||||
"general",
|
||||
"ID обязательного канала",
|
||||
"Telegram ID канала, в котором нужно состоять.",
|
||||
(
|
||||
"Telegram ID канала для проверки подписки. Если бот видит канал, "
|
||||
"ссылка кнопки будет получена автоматически."
|
||||
),
|
||||
),
|
||||
SettingField(
|
||||
"REQUIRED_CHANNEL_LINK",
|
||||
"string",
|
||||
"general",
|
||||
"Ссылка на канал",
|
||||
"Имя пользователя или invite-link.",
|
||||
(
|
||||
"Необязательно: публичный @username или invite-link, "
|
||||
"если ссылку нельзя получить по ID канала."
|
||||
),
|
||||
),
|
||||
SettingField(
|
||||
"PANEL_API_URL",
|
||||
|
||||
@@ -27,6 +27,7 @@ from bot.utils.callback_answer import safe_answer_callback
|
||||
from bot.utils.channel_subscription import (
|
||||
is_required_channel_access_error,
|
||||
normalize_required_channel_id,
|
||||
resolve_required_channel_link,
|
||||
)
|
||||
from bot.utils.install_links import (
|
||||
append_install_share_link_text,
|
||||
@@ -376,11 +377,12 @@ async def ensure_required_channel_subscription(
|
||||
)
|
||||
return True
|
||||
|
||||
keyboard = (
|
||||
get_channel_subscription_keyboard(current_lang, i18n, settings.REQUIRED_CHANNEL_LINK)
|
||||
if i18n
|
||||
else None
|
||||
channel_link = await resolve_required_channel_link(
|
||||
bot_instance,
|
||||
required_channel_id,
|
||||
settings.REQUIRED_CHANNEL_LINK,
|
||||
)
|
||||
keyboard = get_channel_subscription_keyboard(current_lang, i18n, channel_link) if i18n else None
|
||||
|
||||
prompt_text = translate("channel_subscription_required")
|
||||
|
||||
|
||||
@@ -4,6 +4,7 @@ from aiogram.types import InlineKeyboardMarkup, WebAppInfo
|
||||
from aiogram.utils.keyboard import InlineKeyboardBuilder, InlineKeyboardButton
|
||||
|
||||
from bot.middlewares.i18n import locale_language_options
|
||||
from bot.utils.channel_subscription import normalize_required_channel_link
|
||||
from bot.utils.install_links import bot_install_guide_url
|
||||
from bot.utils.mini_app_url import subscription_mini_app_trial_url
|
||||
from config.settings import Settings
|
||||
@@ -718,10 +719,11 @@ def get_channel_subscription_keyboard(
|
||||
|
||||
has_buttons = False
|
||||
|
||||
if channel_link:
|
||||
channel_url = normalize_required_channel_link(channel_link)
|
||||
if channel_url:
|
||||
builder.button(
|
||||
text=_(key="channel_subscription_join_button"),
|
||||
url=channel_link,
|
||||
url=channel_url,
|
||||
)
|
||||
has_buttons = True
|
||||
|
||||
|
||||
@@ -11,7 +11,10 @@ from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from bot.keyboards.inline.user_keyboards import get_channel_subscription_keyboard
|
||||
from bot.middlewares.i18n import JsonI18n
|
||||
from bot.utils.channel_subscription import normalize_required_channel_id
|
||||
from bot.utils.channel_subscription import (
|
||||
normalize_required_channel_id,
|
||||
resolve_required_channel_link,
|
||||
)
|
||||
from config.settings import Settings
|
||||
from db.dal import user_dal
|
||||
|
||||
@@ -86,10 +89,14 @@ class ChannelSubscriptionMiddleware(BaseMiddleware):
|
||||
return i18n_instance.gettext(current_lang, key)
|
||||
return key
|
||||
|
||||
bot_instance = data.get("bot") or data.get("bot_instance")
|
||||
channel_link = await resolve_required_channel_link(
|
||||
bot_instance,
|
||||
required_channel_id,
|
||||
self.settings.REQUIRED_CHANNEL_LINK,
|
||||
)
|
||||
keyboard = (
|
||||
get_channel_subscription_keyboard(
|
||||
current_lang, i18n_instance, self.settings.REQUIRED_CHANNEL_LINK
|
||||
)
|
||||
get_channel_subscription_keyboard(current_lang, i18n_instance, channel_link)
|
||||
if i18n_instance
|
||||
else None
|
||||
)
|
||||
|
||||
@@ -1,4 +1,9 @@
|
||||
from typing import Optional
|
||||
import logging
|
||||
import re
|
||||
from typing import Any, Optional
|
||||
|
||||
_TELEGRAM_LINK_RE = re.compile(r"^(?:https?://|tg://)", re.IGNORECASE)
|
||||
_TELEGRAM_USERNAME_RE = re.compile(r"^[A-Za-z0-9_]{5,64}$")
|
||||
|
||||
|
||||
def normalize_required_channel_id(value: object) -> Optional[int]:
|
||||
@@ -28,6 +33,66 @@ def normalize_required_channel_id(value: object) -> Optional[int]:
|
||||
return -int(f"100{raw_abs}")
|
||||
|
||||
|
||||
def normalize_required_channel_link(value: object) -> Optional[str]:
|
||||
if value is None:
|
||||
return None
|
||||
|
||||
raw = str(value).strip()
|
||||
if not raw:
|
||||
return None
|
||||
|
||||
if _TELEGRAM_LINK_RE.match(raw):
|
||||
return raw
|
||||
|
||||
raw = raw.lstrip("@").strip()
|
||||
if not raw or re.search(r"\s", raw):
|
||||
return None
|
||||
|
||||
if raw.startswith(("t.me/", "telegram.me/")):
|
||||
return f"https://{raw}"
|
||||
|
||||
if raw.startswith(("+", "joinchat/", "c/")):
|
||||
return f"https://t.me/{raw}"
|
||||
|
||||
if _TELEGRAM_USERNAME_RE.fullmatch(raw):
|
||||
return f"https://t.me/{raw}"
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def _required_channel_link_from_chat(chat: Any) -> Optional[str]:
|
||||
username = str(getattr(chat, "username", "") or "").strip().lstrip("@")
|
||||
if username:
|
||||
return f"https://t.me/{username}"
|
||||
|
||||
invite_link = normalize_required_channel_link(getattr(chat, "invite_link", None))
|
||||
if invite_link:
|
||||
return invite_link
|
||||
|
||||
return None
|
||||
|
||||
|
||||
async def resolve_required_channel_link(
|
||||
bot: Any,
|
||||
required_channel_id: Optional[int],
|
||||
configured_link: object,
|
||||
) -> Optional[str]:
|
||||
if bot is not None and required_channel_id:
|
||||
try:
|
||||
chat = await bot.get_chat(required_channel_id)
|
||||
resolved_link = _required_channel_link_from_chat(chat)
|
||||
if resolved_link:
|
||||
return resolved_link
|
||||
except Exception as error:
|
||||
logging.warning(
|
||||
"Failed to resolve required channel link from chat %s: %s",
|
||||
required_channel_id,
|
||||
error,
|
||||
)
|
||||
|
||||
return normalize_required_channel_link(configured_link)
|
||||
|
||||
|
||||
def is_required_channel_access_error(error: BaseException) -> bool:
|
||||
message = str(error).lower()
|
||||
configuration_markers = (
|
||||
|
||||
Reference in New Issue
Block a user