main.py

import logging
from pyrogram import Client, enums, filters
from pyrogram.enums import ChatAction
from pyrogram.types import (
    Message, ReplyKeyboardMarkup, InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery, InputMediaPhoto)
from assortment import ASSORTMENT
from config import SESSION_NAME, BOT_TOKEN, TELEGRAM_API_ID, TELEGRAM_API_HASH


logging.basicConfig(level=logging.INFO)

# --- Экземпляр бота ---
bot = Client(
    SESSION_NAME,
    api_id=TELEGRAM_API_ID,
    api_hash=TELEGRAM_API_HASH,
    bot_token=BOT_TOKEN,
    parse_mode=enums.ParseMode.HTML
)

# --- Импорт датасета ---
categories_dict = ASSORTMENT.get("categories")
display_name_categories = [v.get("display_name") for k, v in categories_dict.items()]

# --- Клавиатуры ---
reply_keyboard_category = ReplyKeyboardMarkup(
    keyboard=[
        [display_name_categories[0], display_name_categories[1]],
        [display_name_categories[2]],
        [display_name_categories[3]]
    ],
    resize_keyboard=True,
    one_time_keyboard=False,
    placeholder="Выберите категорию..."
)

# --- Фильтры ---
async def special_word(_, __, message):
    return message.text in display_name_categories


words_filters = filters.create(special_word)

"""
--- АЛГОРИТМ РАБОТЫ ---
U> Пользователь
B> Бот
i: Информация


1. Начало:
    U> Пользователь отправляет команду /start
    B> Бот формирует reply-клавиатуру и отправляет пользователю приветственное сообщение.

2. Открытие категорий:
    U> Пользователь нажимает reply-кнопку с категорией ИЛИ inline-кнопку "Назад к категориям".
    B> Бот выполняет поиск по словарю и выдает список подкатегорий с Inline-кнопками:
        ASSORTMENT -> categories -> {subcategories} 
                                        ||
                                    {subcategories}: 
                                        * display_name => [РЕЗУЛЬТАТ] 
                                          
    i: Структура Inline-кнопки для подкатегории:  `subcategory:{category_key}:{subcategory_key}#0`
        `{category_key}` - ключ категории
        `{subcategory_key}` - ключ подкатегории
        `#0` - текущая страница (по умолчанию 0, так как это первое открытие подкатегории)
    i: Структура Inline-кнопки для возврата к категориям: `category:{category_key}:{subcategory_key}#{page}`
        (Используется для возврата из подкатегорий)
                      
3. Открытие подкатегорий (список товаров с пагинацией):
    U> Пользователь жмет Inline-кнопку подкатегории ИЛИ кнопки навигации (⬅️, ➡️) ИЛИ кнопку "Назад к каталогу".
    B> Бот выполняет поиск по словарю и выдает список продуктов с Inline-кнопками, разбитый на страницы (по 10 товаров на страницу):
        ASSORTMENT -> categories -> subcategories -> products => [РЕЗУЛЬТАТ]
    
    i: Структура Inline-кнопки для продукта:  `open:{category_key}:{subcategory_key}:{product_id}`
        `{category_key}` - ключ категории
        `{subcategory_key}` - ключ подкатегории
        `{product_id}` - id продукта
    i: Кнопки навигации по страницам: `subcategory:{category_key}:{subcategory_key}#{page}`
        `{page}` - номер страницы (начинается с 0)
    i: Кнопка "Назад к категориям": `category:{category_key}:{subcategory_key}#{current_page}`
        (Возвращает к списку подкатегорий текущей категории)
        
4. Открытие продукта (карточка товара):
    U> Пользователь жмет Inline-кнопку продукта.
    B> Бот выполняет поиск по словарю, формирует ответ из фотографий (медиагруппа), отображаемого имени, описания с Inline-кнопками:
        ASSORTMENT -> categories -> subcategories -> products -> product_id => {РЕЗУЛЬТАТ}
        
    i: Структура Inline-кнопки для возврата к списку товаров подкатегории ("Назад к каталогу"):  `subcategory:{category_key}:{subcategory_key}#{product_index // 10}`
        `{category_key}` - ключ категории
        `{subcategory_key}` - ключ подкатегории
        `{product_index // 10}` - вычисленная страница, на которой находился товар, для удобства возврата.
    i: Структура Inline-кнопки для закрытия карточки ("Закрыть карточку"): `close`
        (Удаляет сообщение с карточкой товара)
"""


@bot.on_callback_query()
async def open_category_callback(client: Client, callback: CallbackQuery):
    """Работа с Callback-вызовами."""
    callback_data = callback.data

    # --- Пункт 3: Открытие подкатегорий ---
    if "subcategory:" in callback_data:
        _, category_key, parts = callback_data.split(":", 2)
        subcategory_key, current_page = parts.split("#", 1)
        current_page = int(current_page)
        await open_subcategory(client, callback, category_key, subcategory_key, current_page)
        return

    if "category:" in callback_data:
        await open_category(client, callback)
        return

    # --- Пункт 4: Открытие продукта ---
    if "open:" in callback_data:
        _, category_key, subcategory_key, product_id = callback_data.split(":", 3)
        await open_card(client, callback, category_key, subcategory_key, product_id)

    if "close" in callback_data:
        await client.delete_messages(
            callback.message.chat.id,
            callback.message.id
        )



@bot.on_message(filters.command("start") & filters.private)  # --- Пункт 1 ---
async def start_command(client: Client, message: Message):
    """Отвечает на команду /start, предлагая выбрать категорию на клавиатуре."""
    welcome_text = "Привет! Я бот команды #НеЧайник.\nДля ознакомления с ассортиментом используйте клавиатуру ниже ⬇"
    await message.reply(welcome_text, reply_markup=reply_keyboard_category)


@bot.on_message(words_filters & filters.private)  # --- Пункт 2 ---
async def open_category_message(client: Client, message: Message):
    """Открывает выбранные категории по сообщению."""
    await open_category(client, message)


async def open_category(  # --- Пункт 2 ---
        client: Client, data: Message | CallbackQuery
):
    """Открывает категорию."""
    # --- Экземпляр клавиатуры ---
    markup = InlineKeyboardMarkup(inline_keyboard=[])

    # --- Получение категории, выборка кнопок ---
    if isinstance(data, Message):
        user_text = data.text
        category_data = next((k, v) for k, v in categories_dict.items() if v.get("display_name") == user_text)
    else:
        callback_data = data.data
        _, category_key, parts = callback_data.split(":", 2)
        category_data = next((k, v) for k, v in categories_dict.items() if k == category_key)

    category_key = category_data[0]
    category_values = category_data[1]
    display_current_category = category_values.get("display_name")
    subcategories = category_values.get("subcategories")

    # --- Заполнение кнопками товаров ---
    try:
        for subcategory_key, v in subcategories.items():
            display_subcategory = v.get("display_name")
            markup.inline_keyboard.append(
                [
                    InlineKeyboardButton(
                        display_subcategory,
                        callback_data=f'subcategory:{category_key}:{subcategory_key}#0'
                    )
                ]
            )
    except AttributeError:
        markup = None

    # --- Отправка результата пользователю ---
    message_text = f"Вы находитесь в категории: <b>{display_current_category}</b>"
    if markup:
        message_text += "\n<i>Пожалуйста, используйте кнопки ниже для навигации!</i>"
    else:
        message_text += "\n<i>Увы, эта категория пуста...</i>"

    if isinstance(data, Message):
        await data.reply(
            message_text,
            reply_markup=markup
        )
    else:
        await data.edit_message_text(
            message_text,
            reply_markup=markup
        )


async def open_subcategory(  # --- Пункт 3 ---
        client: Client, data: Message | CallbackQuery,
        category_key: str,
        subcategory_key: str,
        current_page: int | None
):
    """Открывает N страницу из подкатегорий."""
    # --- Первоначальные разметки страницы ---
    start_value_page = 0
    end_value_page = 10

    start_page = start_value_page + (current_page * 10)
    end_page = end_value_page + (current_page * 10)

    # --- Экземпляр клавиатуры ---
    markup = InlineKeyboardMarkup(inline_keyboard=[])

    # --- Получение категории, выборка кнопок ---
    categories_data = categories_dict.get(category_key)
    display_category = categories_data.get("display_name")
    subcategories_data = categories_data.get("subcategories").get(subcategory_key)
    display_subcategory = subcategories_data.get("display_name")
    total_products = subcategories_data.get("products")
    products = total_products[start_page:end_page]

    # --- Заполнение кнопками товаров ---
    for product in products:
        product_id = product.get("product_id")
        if product_id:
            display_name = product.get("display_name")
            markup.inline_keyboard.append(
                [
                    InlineKeyboardButton(
                        display_name,
                        callback_data=f'open:{category_key}:{subcategory_key}:{product_id}'
                    )
                ]
            )

    # --- Проверка на наличие товаров ---
    if markup.inline_keyboard:
        info_message = "<i>Пожалуйста, используйте кнопки ниже для навигации!</i>"
    else:
        info_message = "<i>Увы, эта подкатегория пуста...</i>"

    # --- Создание кнопок навигации ---
    navigation_buttons = []
    if current_page > start_value_page:
        navigation_buttons.append(
            InlineKeyboardButton(
                "⬅️",
                callback_data=f"subcategory:{category_key}:{subcategory_key}#{current_page - 1}"
            )
        )
    navigation_buttons.append(
        InlineKeyboardButton(
            f"Страница: {current_page + 1}",
            callback_data=f"ignore"
        )
    )

    if end_page < len(total_products) and len(markup.inline_keyboard) == end_value_page:
        navigation_buttons.append(
            InlineKeyboardButton(
                "️➡️",
                callback_data=f"subcategory:{category_key}:{subcategory_key}#{current_page + 1}"
            )
        )
    markup.inline_keyboard.append(navigation_buttons)
    markup.inline_keyboard.append(
        [
            InlineKeyboardButton(
                "↩️ Назад к категориям",
                f"category:{category_key}:{subcategory_key}#{current_page}"
            )
        ]
    )

    # --- Отправка результата пользователю ---
    message_text = (f"Вы находитесь в категории: <b>{display_category}</b>\n"
                    f"Подкатегория: <b>{display_subcategory}</b>\n"
                    f"{info_message}")

    if isinstance(data, Message):
        await data.reply(
            message_text,
            reply_markup=markup
        )
    else:
        if data.message.media:
            await client.delete_messages(
                data.message.chat.id,
                data.message.id
            )
            await client.send_message(
                data.message.chat.id,
                message_text,
                reply_markup=markup
            )
        else:
            await data.edit_message_text(
                message_text,
                reply_markup=markup
            )


async def open_card(  # --- Пункт 4 ---
        client: Client, callback: CallbackQuery,
        category_key: str, subcategory_key: str, product_id: str
):
    """Открывает карточку товара."""
    # --- Поиск карточки из каталога ---
    subcategory_data = categories_dict.get(category_key)
    subcategories = subcategory_data.get("subcategories")
    products_data = subcategories.get(subcategory_key)
    products = products_data.get("products")

    target_product = None
    for index, product in enumerate(products):
        if product.get("product_id") == product_id:
            target_product = product
            product_index = index
            break

    if not target_product:  # Выдать ошибку если поиск не удался
        await callback.answer("Без паники, вы обнаружили скрытую ошибку, которую мы не нашли!\nРасскажите о ней владельцу, чтобы мы смогли быстренько всё починить!)")

    # --- Получение основных сведений ---
    product_name = target_product.get("display_name")
    product_description = target_product.get("description")
    image_paths = target_product.get("image_path")

    # --- Подготовка данных, выдача результата ---
    message_text = f"<blockquote><b>{product_name}</b></blockquote>\n{product_description}"
    markup = InlineKeyboardMarkup(
        [
            [
                InlineKeyboardButton(
                    "⬅️ Назад к каталогу",
                    callback_data=f"subcategory:{category_key}:{subcategory_key}#{product_index // 10}"
                )
            ],
            [
                InlineKeyboardButton(
                    "❎ Закрыть карточку",
                    callback_data=f"close"
                )
            ]
        ]
    )

    chat_id = callback.message.chat.id

    await client.delete_messages(
        callback.message.chat.id,
        callback.message.id
    )
    await client.send_chat_action(chat_id, ChatAction.UPLOAD_PHOTO)
    try:
        # --- Получение медиагруппы и отправка сообщения ---
        media_group = []
        if not isinstance(image_paths, list):
            image_paths = [image_paths]

        for index, photo_path in enumerate(image_paths):
            media_group.append(
                InputMediaPhoto(
                    media=photo_path,
                    caption=message_text if index == 0 else ""
                )
            )

        message = await client.send_media_group(
            chat_id,
            media_group
        )

        await client.edit_message_reply_markup(
            chat_id,
            message[0].id,
            reply_markup=markup
        )
    except FileNotFoundError:
        await callback.answer(
            "Без паники, вы обнаружили скрытую ошибку, которую мы не нашли!\n"
            "Расскажите о ней владельцу, чтобы мы смогли быстренько всё починить!)"
        )
        return
    await callback.answer()

if __name__ == "__main__":
    logging.info("Бот успешно запущен!")
    try:
        bot.run()
    except (KeyboardInterrupt, SystemExit):
        pass
    logging.info("Бот остановлен")