124 lines
4.3 KiB
Python
124 lines
4.3 KiB
Python
from telegram import Update
|
||
from telegram.constants import ParseMode, ChatType
|
||
from telegram.ext import ContextTypes
|
||
from telegram.error import BadRequest
|
||
|
||
from app.bot.messages import (
|
||
ASK_ADD_GROUP,
|
||
GROUP_BOUND,
|
||
NEED_ADD_FIRST,
|
||
NO_RIGHTS_CHANNEL,
|
||
NO_RIGHTS_GROUP,
|
||
)
|
||
from app.db.session import get_session
|
||
from app.db.models import User, Chat
|
||
from .utils import parse_chat_id, verify_and_fetch_chat
|
||
|
||
STATE_KEY = "await_chat_id"
|
||
|
||
|
||
def _get_forwarded_chat_id(message) -> int | None:
|
||
"""
|
||
Универсально достаём chat_id источника пересылки.
|
||
Поддерживает старые поля (forward_from_chat) и новые (forward_origin.chat).
|
||
"""
|
||
if not message:
|
||
return None
|
||
|
||
# Старое поле (иногда ещё присутствует)
|
||
fwd_chat = getattr(message, "forward_from_chat", None)
|
||
if fwd_chat:
|
||
return fwd_chat.id
|
||
|
||
# Новая схема Bot API: forward_origin с типами MessageOrigin*
|
||
origin = getattr(message, "forward_origin", None)
|
||
if origin:
|
||
chat = getattr(origin, "chat", None) # у MessageOriginChat/Channel есть .chat
|
||
if chat:
|
||
return chat.id
|
||
|
||
return None
|
||
|
||
|
||
async def add_group_cmd(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
|
||
# ensure user exists
|
||
with get_session() as s:
|
||
u = s.query(User).filter_by(tg_id=update.effective_user.id).first()
|
||
if not u:
|
||
u = User(tg_id=update.effective_user.id, name=update.effective_user.full_name)
|
||
s.add(u)
|
||
s.commit()
|
||
ctx.user_data[STATE_KEY] = True
|
||
await update.effective_message.reply_text(ASK_ADD_GROUP, parse_mode=ParseMode.MARKDOWN)
|
||
|
||
|
||
async def add_group_capture(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
|
||
if update.effective_chat.type != ChatType.PRIVATE:
|
||
return
|
||
if not ctx.user_data.get(STATE_KEY):
|
||
return
|
||
|
||
msg = update.effective_message
|
||
raw = None
|
||
|
||
# 1) Пересланное сообщение из чата/канала
|
||
fwd_cid = _get_forwarded_chat_id(msg)
|
||
if fwd_cid:
|
||
raw = fwd_cid
|
||
else:
|
||
# 2) Ввод в виде текста или подписи (caption)
|
||
txt = (getattr(msg, "text", None) or getattr(msg, "caption", None) or "").strip()
|
||
if not txt:
|
||
await msg.reply_text("Вставьте chat_id или перешлите сообщение из группы/канала.")
|
||
return
|
||
|
||
if txt.startswith("@") or "t.me/" in txt:
|
||
raw = txt
|
||
else:
|
||
cid = parse_chat_id(txt)
|
||
if cid is None:
|
||
await msg.reply_text("Не удалось распознать chat_id. Пример: -1001234567890 или @username.")
|
||
return
|
||
raw = cid
|
||
|
||
# Проверяем, что бот в чате и что у него есть право постить (если нужно)
|
||
try:
|
||
chat, member, can_post = await verify_and_fetch_chat(ctx, raw)
|
||
except BadRequest:
|
||
await msg.reply_text("Такого чата не существует или у меня нет доступа. Проверьте chat_id/username.")
|
||
return
|
||
|
||
if member is None:
|
||
await msg.reply_text(NEED_ADD_FIRST.format(title_or_id=(chat.title or chat.id)))
|
||
return
|
||
|
||
# Сохраняем привязку
|
||
with get_session() as s:
|
||
me = s.query(User).filter_by(tg_id=update.effective_user.id).first()
|
||
row = s.query(Chat).filter_by(chat_id=chat.id).first()
|
||
if not row:
|
||
row = Chat(
|
||
chat_id=chat.id,
|
||
type=chat.type,
|
||
title=chat.title,
|
||
owner_user_id=me.id,
|
||
can_post=can_post,
|
||
)
|
||
s.add(row)
|
||
else:
|
||
row.title = chat.title
|
||
row.type = chat.type
|
||
if row.owner_user_id is None:
|
||
row.owner_user_id = me.id
|
||
row.can_post = can_post
|
||
s.commit()
|
||
|
||
ctx.user_data.pop(STATE_KEY, None)
|
||
|
||
rights = (
|
||
"✅ У меня есть право публиковать."
|
||
if can_post
|
||
else (NO_RIGHTS_CHANNEL if chat.type == ChatType.CHANNEL else NO_RIGHTS_GROUP)
|
||
)
|
||
await msg.reply_text(GROUP_BOUND.format(title_or_id=(chat.title or chat.id), rights=rights))
|