main.py

import logging
import asyncio
import json
import sys
from pathlib import Path
from aiogram import Bot, Dispatcher, types
from aiogram.filters import Command
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup
from aiogram.fsm.storage.memory import MemoryStorage

# Добавляем корневую папку проекта в путь для импортов
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from config import BOT_TOKEN, ADMIN_TELEGRAM_ID
from database.db import init_db, get_or_create_user, save_transaction, get_report_data

# Включаем логирование
logging.basicConfig(level=logging.INFO)

# Инициализация бота и диспетчера
bot = Bot(token=BOT_TOKEN)
storage = MemoryStorage()
dp = Dispatcher(storage=storage)

# Загружаем правила из rules.json
RULES_PATH = Path(__file__).parent / "rules.json"
with open(RULES_PATH, "r", encoding="utf-8") as f:
    RULES = json.load(f)


# FSM состояния
class TransactionStates(StatesGroup):
    waiting_for_amount = State()
    waiting_for_category = State()
    waiting_for_note = State()


class FeedbackStates(StatesGroup):
    waiting_for_feedback = State()


def classify_expense(category: str) -> tuple[str, str]:
    """
    Классифицирует расход по правилам из rules.json
    Возвращает (status, comment)
    """
    category_lower = category.lower()
    
    # Ищем совпадение ключевых слов
    for rule_name, rule_data in RULES.items():
        keywords = rule_data.get("keywords", [])
        for keyword in keywords:
            if keyword.lower() in category_lower:
                status = rule_data.get("status", "doubtful")
                comment = rule_data.get("comment", "Требуется дополнительный анализ.")
                return status, comment
    
    # Если правило не найдено
    return "doubtful", "Категория не найдена в правилах. Требуется дополнительный анализ."


@dp.message(Command("start"))
async def start_handler(message: types.Message):
    text = (
        "Ассаляму алейкум.\n\n"
        "Этот бот помогает вести финансовый учёт и смотреть на траты с точки зрения дозволенного, сомнительного и запретного.\n\n"
        "Бот не является учёным и не выносит религиозных решений — он лишь инструмент осознанности.\n\n"
        "❌ Не подходит тем, кто не хочет вести учёт\n"
        "❌ Не подходит тем, кто не считает религию важной\n\n"
        "✅ Подходит тем, кто хочет следить за своим ризком\n\n"
        "Доступные команды:\n"
        "/income — добавить доход\n"
        "/expense — добавить расход\n"
        "/report — отчёт за 7 дней\n"
        "/feedback — оставить отзыв о боте"
    )
    await message.answer(text)


@dp.message(Command("income"))
async def income_command(message: types.Message, state: FSMContext):
    """Начало процесса добавления дохода"""
    await message.answer("Введите сумму дохода:")
    await state.set_state(TransactionStates.waiting_for_amount)
    await state.update_data(transaction_type="income")


@dp.message(Command("expense"))
async def expense_command(message: types.Message, state: FSMContext):
    """Начало процесса добавления расхода"""
    await message.answer("Введите сумму расхода:")
    await state.set_state(TransactionStates.waiting_for_amount)
    await state.update_data(transaction_type="expense")


@dp.message(Command("report"))
async def report_command(message: types.Message):
    """Генерация отчёта за последние 7 дней"""
    # Получаем или создаём пользователя
    user_id = get_or_create_user(message.from_user.id)
    
    # Получаем данные за 7 дней
    data = get_report_data(user_id, days=7)
    
    # Форматируем суммы (округляем до целых)
    total_income = int(round(data["total_income"]))
    total_expense = int(round(data["total_expense"]))
    
    # Формируем топ категорий
    top_categories_text = ""
    top_categories = data["top_categories"]
    if top_categories:
        for i, row in enumerate(top_categories[:3], 1):
            category = row["category"]
            amount = int(round(row["total"]))
            top_categories_text += f"{i}. {category} — {amount} ₼\n"
    if not top_categories_text:
        top_categories_text = "Нет данных\n"
    
    # Формируем распределение по статусу
    status_dist = data["status_distribution"]
    ok_sum = 0
    doubtful_sum = 0
    haram_sum = 0
    
    for row in status_dist:
        status = row["status"]
        amount = row["total"]
        if status == "ok":
            ok_sum += amount
        elif status == "doubtful":
            doubtful_sum += amount
        elif status == "haram":
            haram_sum += amount
    
    ok_sum = int(round(ok_sum))
    doubtful_sum = int(round(doubtful_sum))
    haram_sum = int(round(haram_sum))
    
    # Формируем отчёт
    report = (
        f"📊 Отчёт за 7 дней\n\n"
        f"Доход: {total_income} ₼\n"
        f"Расход: {total_expense} ₼\n\n"
        f"Топ расходов:\n{top_categories_text}\n"
        f"По статусу:\n"
        f"✅ дозволено — {ok_sum} ₼\n"
        f"⚠️ сомнительно — {doubtful_sum} ₼\n"
        f"⛔ запретно — {haram_sum} ₼\n\n"
        f"«И будет спрошен человек о своём имуществе — откуда он его приобрёл и на что расходовал» (ат-Тирмизи)"
    )
    
    await message.answer(report)


@dp.message(TransactionStates.waiting_for_amount)
async def process_amount(message: types.Message, state: FSMContext):
    """Обработка суммы"""
    try:
        amount = float(message.text.replace(",", "."))
        if amount <= 0:
            await message.answer("Сумма должна быть положительным числом. Попробуйте снова:")
            return
        
        await state.update_data(amount=amount)
        await message.answer("Введите категорию:")
        await state.set_state(TransactionStates.waiting_for_category)
    except ValueError:
        await message.answer("Пожалуйста, введите корректное число (например: 1000 или 1000.50):")


@dp.message(TransactionStates.waiting_for_category)
async def process_category(message: types.Message, state: FSMContext):
    """Обработка категории"""
    category = message.text.strip()
    if not category:
        await message.answer("Категория не может быть пустой. Введите категорию:")
        return
    
    data = await state.get_data()
    transaction_type = data.get("transaction_type")
    
    # Для расходов определяем halal_status
    halal_status = None
    halal_comment = None
    if transaction_type == "expense":
        halal_status, halal_comment = classify_expense(category)
        await state.update_data(halal_status=halal_status, halal_comment=halal_comment)
    
    await state.update_data(category=category)
    await message.answer("Введите примечание (или отправьте /skip чтобы пропустить):")
    await state.set_state(TransactionStates.waiting_for_note)


@dp.message(Command("skip"), TransactionStates.waiting_for_note)
async def skip_note(message: types.Message, state: FSMContext):
    """Пропуск примечания"""
    await process_note_internal(message, state, note=None)


@dp.message(TransactionStates.waiting_for_note)
async def process_note(message: types.Message, state: FSMContext):
    """Обработка примечания и сохранение транзакции"""
    note = message.text.strip()
    await process_note_internal(message, state, note)


async def process_note_internal(message: types.Message, state: FSMContext, note: str = None):
    """Внутренняя функция для сохранения транзакции"""
    data = await state.get_data()
    transaction_type = data.get("transaction_type")
    amount = data.get("amount")
    category = data.get("category")
    halal_status = data.get("halal_status")
    halal_comment = data.get("halal_comment")
    
    # Получаем или создаём пользователя
    user_id = get_or_create_user(message.from_user.id)
    
    # Сохраняем транзакцию
    save_transaction(
        user_id=user_id,
        transaction_type=transaction_type,
        amount=amount,
        category=category,
        note=note,
        halal_status=halal_status
    )
    
    # Формируем ответ
    if transaction_type == "income":
        response = "Доход сохранён ✅"
    else:
        # Для расходов добавляем информацию о статусе
        status_text = {
            "ok": "✅ Разрешено",
            "doubtful": "⚠️ Сомнительно",
            "haram": "❌ Харам (Запрещено)"
        }.get(halal_status, "⚠️ Сомнительно")
        
        response = (
            f"Расход сохранён.\n"
            f"Статус: {status_text}\n"
            f"Причина: {halal_comment}\n\n"
            f"Бот не является учёным и не выносит религиозных решений."
        )
    
    await message.answer(response)
    await state.clear()


@dp.message(Command("feedback"))
async def feedback_command(message: types.Message, state: FSMContext):
    """Начало процесса отправки отзыва"""
    await message.answer(
        "Поделитесь своими мыслями о боте.\n"
        "Напишите ваш отзыв, предложения или замечания.\n\n"
        "Можете ответить на следующие вопросы:\n"
        "1️⃣ Было ли понятно, как пользоваться?\n"
        "2️⃣ Что было неудобно?\n"
        "3️⃣ Пользовался ли ты /report?\n"
        "4️⃣ Хотелось ли продолжить пользоваться?\n"
        "5️⃣ Что бы ты убрал?"
    )

    await state.set_state(FeedbackStates.waiting_for_feedback)


@dp.message(FeedbackStates.waiting_for_feedback)
async def process_feedback(message: types.Message, state: FSMContext):
    """Обработка отзыва и отправка администратору"""
    feedback_text = message.text.strip()
    
    if not feedback_text:
        await message.answer("Отзыв не может быть пустым. Пожалуйста, напишите ваш отзыв:")
        return
    
    # Получаем информацию о пользователе
    user = message.from_user
    user_info = f"@{user.username}" if user.username else f"ID: {user.id}"
    user_full_name = f"{user.first_name or ''} {user.last_name or ''}".strip() or "Не указано"
    
    # Формируем сообщение для администратора
    admin_message = (
        f"📝 Новый отзыв о боте\n\n"
        f"От: {user_full_name} ({user_info})\n"
        f"Отзыв:\n{feedback_text}"
    )
    
    try:
        # Отправляем отзыв администратору
        await bot.send_message(chat_id=ADMIN_TELEGRAM_ID, text=admin_message)
        
        # Подтверждаем пользователю
        await message.answer(
            "✅ Спасибо за ваш отзыв! Ваше мнение очень важно для нас."
        )
    except Exception as e:
        logging.error(f"Ошибка при отправке отзыва администратору: {e}")
        await message.answer(
            "❌ Произошла ошибка при отправке отзыва. Попробуйте позже."
        )
    
    await state.clear()


async def main():
    # Инициализируем БД
    init_db()
    await dp.start_polling(bot, skip_updates=True)


if __name__ == '__main__':
    asyncio.run(main())