limoka/archquise/H.Modules/IrisSimpleMod.py

# Proprietary License Agreement

# Copyright (c) 2024-29 CodWiz

# Permission is hereby granted to any person obtaining a copy of this software and associated documentation files (the "Software"), to use the Software for personal and non-commercial purposes, subject to the following conditions:

# 1. The Software may not be modified, altered, or otherwise changed in any way without the explicit written permission of the author.

# 2. Redistribution of the Software, in original or modified form, is strictly prohibited without the explicit written permission of the author.

# 3. The Software is provided "as is", without warranty of any kind, express or implied, including but not limited to the warranties of merchantability, fitness for a particular purpose, and non-infringement. In no event shall the author or copyright holder be liable for any claim, damages, or other liability, whether in an action of contract, tort, or otherwise, arising from, out of, or in connection with the Software or the use or other dealings in the Software.

# 4. Any use of the Software must include the above copyright notice and this permission notice in all copies or substantial portions of the Software.

# 5. By using the Software, you agree to be bound by the terms and conditions of this license.

# For any inquiries or requests for permissions, please contact codwiz@yandex.ru.

# ---------------------------------------------------------------------------------
# Name: IrisSimpleMod
# Description: Module for basic interaction with Iris.
# Author: @hikka_mods
# ---------------------------------------------------------------------------------
# meta developer: @hikka_mods
# scope: IrisSimpleMod
# scope: IrisSimpleMod 1.0.1
# ---------------------------------------------------------------------------------

import logging
from typing import Optional

from .. import loader, utils

__version__ = (1, 0, 1)

logger = logging.getLogger(__name__)


@loader.tds
class IrisSimpleMod(loader.Module):
    """Module for basic interaction with Iris bot"""

    strings = {
        "name": "IrisSimpleMod",
        "checking_bag": "<emoji document_id=5188311512791393083>🌎</emoji> Checking bag...",
        "bag_result": "<emoji document_id=5854762571659218443>✅</emoji> Your bag: <code>{}</code>",
        "farming": "<emoji document_id=5188311512791393083>🌎</emoji> Farming iris-coins...",
        "farm_result": "<emoji document_id=5854762571659218443>✅</emoji> Farm result: <code>{}</code>",
        "getting_stats": "<emoji document_id=5188311512791393083>🌎</emoji> Getting user stats...",
        "stats_result": "<emoji document_id=5854762571659218443>✅</emoji> User stats: <code>{}</code>",
        "bot_stats": "<emoji document_id=5188311512791393083>🌎</emoji> Getting bot stats...",
        "bot_stats_result": "<emoji document_id=5854762571659218443>✅</emoji> Bot stats: <code>{}</code>",
        "error_no_response": "<emoji document_id=5854929766146118183>❌</emoji> No response from bot. Please try again.",
        "error_timeout": "<emoji document_id=5854929766146118183>❌</emoji> Request timeout. Please try again.",
        "error_general": "<emoji document_id=5854929766146118183>❌</emoji> An error occurred: {error}",
    }

    strings_ru = {
        "checking_bag": "<emoji document_id=5188311512791393083>🌎</emoji> Проверка мешка...",
        "bag_result": "<emoji document_id=5854762571659218443>✅</emoji> Ваш мешок: <code>{}</code>",
        "farming": "<emoji document_id=5188311512791393083>🌎</emoji> Фарм ирис-коинов...",
        "farm_result": "<emoji document_id=5854762571659218443>✅</emoji> Результат фарма: <code>{}</code>",
        "getting_stats": "<emoji document_id=5188311512791393083>🌎</emoji> Получение статистики пользователя...",
        "stats_result": "<emoji document_id=5854762571659218443>✅</emoji> Статистика пользователя: <code>{}</code>",
        "bot_stats": "<emoji document_id=5188311512791393083>🌎</emoji> Получение статистики ботов...",
        "bot_stats_result": "<emoji document_id=5854762571659218443>✅</emoji> Статистика ботов: <code>{}</code>",
        "error_no_response": "<emoji document_id=5854929766146118183>❌</emoji> Нет ответа от бота. Попробуйте еще раз.",
        "error_timeout": "<emoji document_id=5854929766146118183>❌</emoji> Таймаут запроса. Попробуйте еще раз.",
        "error_general": "<emoji document_id=5854929766146118183>❌</emoji> Произошла ошибка: {error}",
    }

    async def _send_and_delete(
        self, message, command_message: str, response_timeout: int = 15
    ) -> Optional[str]:
        """Send command to Iris and get response with timeout"""
        try:
            async with self.client.conversation(
                self._iris_user_id, timeout=self._timeout
            ) as conv:
                await conv.send_message(command_message)
                await message.delete()

                response_msg = await conv.get_response()
                if response_msg:
                    await utils.answer(message, response_msg.text)
                    return response_msg.text
                else:
                    return None
        except Exception as e:
            logger.error(f"Error in conversation: {e}")
            await utils.answer(
                message, self.strings["error_general"].format(error=str(e))
            )
            return None

    @loader.command(
        ru_doc="Проверить мешок",
        en_doc="Check bag",
    )
    async def bag(self, message):
        """Check bag"""
        await utils.answer(message, self.strings["checking_bag"])

        result = await self._send_and_delete(message, "мешок", response_timeout=20)

        if result:
            await utils.answer(message, self.strings["bag_result"].format(result))

    @loader.command(
        ru_doc="Зафармить ирис-коины",
        en_doc="Farm iris-coins",
    )
    async def farm(self, message):
        """Farm iris-coins"""
        await utils.answer(message, self.strings["farming"])

        result = await self._send_and_delete(message, "ферма", response_timeout=25)

        if result:
            await utils.answer(message, self.strings["farm_result"].format(result))

    @loader.command(
        ru_doc="Вывести анкету",
        en_doc="Display user stats",
    )
    async def irisstats(self, message):
        """Display user stats"""
        await utils.answer(message, self.strings["getting_stats"])

        result = await self._send_and_delete(message, "анкета", response_timeout=20)

        if result:
            await utils.answer(message, self.strings["stats_result"].format(result))