script.py

import argparse
import asyncio
import re
import sys

from telethon import TelegramClient

from utils.chat_utils import get_all_chats
from utils.config import API_HASH, API_ID, GROUP_ID
from utils.copy_private_group import export_private_group_messages
from utils.delete_messages_utils import delete_my_messages
from utils.export_unread_preview import export_unread_messages
from utils.export_utils import export_recent_actions
from utils.session_utils import create_or_check_session
from utils.story_privacy_utils import hide_my_stories_from_user


def _normalize_chat_id_argv(argv: list[str]) -> list[str]:
    """
    argparse не всегда корректно съедает значение --chat-id вида -100...:...
    (считает, что это новый флаг). Нормализуем к --chat-id=<value>.
    """
    normalized = list(argv)
    for idx, token in enumerate(normalized[:-1]):
        if token == "--chat-id":
            next_token = normalized[idx + 1]
            if re.match(r"^-\d+(?:[:/]\d+){1,2}$", next_token):
                normalized[idx] = f"--chat-id={next_token}"
                normalized[idx + 1] = ""
    return [part for part in normalized if part != ""]


def build_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        description="CLI для работы с Telegram-утилитами проекта."
    )
    parser.add_argument(
        "--session",
        default="session",
        help="Имя Telethon session-файла (по умолчанию: session)",
    )

    subparsers = parser.add_subparsers(dest="command", required=True)

    subparsers.add_parser(
        "create-session",
        help="Создать/проверить сессию (логин через телефон/код)",
    )

    list_chats_parser = subparsers.add_parser(
        "list-chats",
        help="Показать список доступных чатов и их ID (с экспортом в HTML)",
    )
    list_chats_parser.add_argument(
        "--items-per-page",
        type=int,
        default=100,
        help="Количество чатов на страницу HTML",
    )
    list_chats_parser.add_argument(
        "--no-open-browser",
        action="store_true",
        help="Не открывать HTML автоматически в браузере",
    )

    delete_parser = subparsers.add_parser(
        "delete-my-messages",
        help="Удалить все ваши сообщения в выбранном чате",
    )
    delete_parser.add_argument(
        "--chat-id",
        type=int,
        default=GROUP_ID,
        help="ID чата. По умолчанию GROUP_ID из utils.config",
    )
    delete_parser.add_argument(
        "--batch-size",
        type=int,
        default=50,
        help="Размер батча удаления (по умолчанию 50)",
    )
    delete_parser.add_argument(
        "--delay",
        type=int,
        default=2,
        help="Пауза между батчами, сек (по умолчанию 2)",
    )

    export_history_parser = subparsers.add_parser(
        "export-history",
        help="Экспорт recent actions (включая удаленные сообщения) в HTML",
    )
    export_history_parser.add_argument(
        "--chat-id",
        type=int,
        default=GROUP_ID,
        help="ID чата. По умолчанию GROUP_ID из utils.config",
    )
    export_history_parser.add_argument(
        "--limit",
        type=int,
        default=100,
        help="Количество элементов на страницу HTML",
    )

    export_unread_parser = subparsers.add_parser(
        "export-unread",
        help="Экспорт непрочитанных сообщений в HTML без read receipts",
    )
    export_unread_parser.add_argument(
        "--chat-id",
        type=int,
        default=GROUP_ID,
        help="ID чата. По умолчанию GROUP_ID из utils.config",
    )
    export_unread_parser.add_argument(
        "--limit",
        type=int,
        default=None,
        help="Ограничить количество непрочитанных сообщений",
    )
    export_unread_parser.add_argument(
        "--items-per-page",
        type=int,
        default=100,
        help="Сообщений на страницу HTML",
    )
    export_unread_parser.add_argument(
        "--no-open-browser",
        action="store_true",
        help="Не открывать HTML автоматически в браузере после экспорта",
    )

    copy_private_parser = subparsers.add_parser(
        "copy-private-group",
        help="Экспорт сообщений из чата в HTML (best-effort по доступности Telegram API)",
    )
    copy_private_parser.add_argument(
        "--chat-id",
        type=str,
        default=str(GROUP_ID),
        help=(
            "ID чата или составной ID раздела. Примеры: -1001234567890 "
            "или -1001234567890:42, -1001234567890:42:128353, "
            "либо ссылка t.me/c/.../msg или t.me/c/.../topic/msg"
        ),
    )
    copy_private_parser.add_argument(
        "--limit",
        type=int,
        default=300,
        help="Сколько последних сообщений выгружать (по умолчанию 300)",
    )
    copy_private_parser.add_argument(
        "--items-per-page",
        type=int,
        default=100,
        help="Сообщений на страницу HTML",
    )
    copy_private_parser.add_argument(
        "--message-id",
        type=int,
        default=None,
        help=(
            "Необязательный верхний предел message_id. "
            "Если задан, выгрузка ограничивается сообщениями с id <= message_id"
        ),
    )
    copy_private_parser.add_argument(
        "--no-open-browser",
        action="store_true",
        help="Не открывать HTML автоматически в браузере после экспорта",
    )

    hide_stories_parser = subparsers.add_parser(
        "hide-stories",
        help="Скрыть ваши активные сторис от пользователя по username (через privacy rules)",
    )
    hide_stories_parser.add_argument(
        "--username",
        required=True,
        help="Username пользователя, например: some_user или @some_user",
    )

    return parser


async def run_cli() -> None:
    argv = _normalize_chat_id_argv(sys.argv[1:])
    args = build_parser().parse_args(argv)

    if args.command == "create-session":
        await create_or_check_session(session_name=args.session)
        return

    async with TelegramClient(args.session, API_ID, API_HASH) as client:
        if args.command == "list-chats":
            await get_all_chats(
                client,
                items_per_page=args.items_per_page,
                open_browser=not args.no_open_browser,
            )
            return

        if args.command == "delete-my-messages":
            await delete_my_messages(
                client=client,
                chat_id=args.chat_id,
                batch_size=args.batch_size,
                delay=args.delay,
            )
            return

        if args.command == "export-history":
            await export_recent_actions(client, args.chat_id, limit=args.limit)
            return

        if args.command == "export-unread":
            await export_unread_messages(
                client=client,
                chat_id=args.chat_id,
                limit=args.limit,
                items_per_page=args.items_per_page,
                open_browser=not args.no_open_browser,
            )
            return

        if args.command == "copy-private-group":
            await export_private_group_messages(
                client=client,
                chat_ref=args.chat_id,
                limit=args.limit,
                items_per_page=args.items_per_page,
                message_id=args.message_id,
                open_browser=not args.no_open_browser,
            )
            return

        if args.command == "hide-stories":
            result = await hide_my_stories_from_user(client=client, username=args.username)
            print(
                "Готово: обновлена приватность сторис для "
                f"{result.target}. Активных сторис: {result.active_stories_count}, "
                f"обновлено: {len(result.updated_story_ids)}, уже скрыто: {len(result.already_hidden_story_ids)}, "
                f"пропущено: {len(result.skipped_story_ids)}."
            )
            return


if __name__ == "__main__":
    asyncio.run(run_cli())