bot.py

import asyncio
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup, ReplyKeyboardMarkup, KeyboardButton
from telegram.ext import Application, CommandHandler, ContextTypes, CallbackQueryHandler
import database as db

# Импортируем токен из config.py
try:
    from config import BOT_TOKEN
except ImportError:
    print("❌ ОШИБКА: Файл config.py не найден!")
    print("Создайте файл config.py на основе config.example.py")
    print("И добавьте туда токен вашего бота от @BotFather")
    exit(1)


# ============= КОМАНДЫ ДЛЯ СОТРУДНИКОВ =============

async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /start - приветствие с кнопками"""
    if not update.message:
        return

    # Проверяем является ли пользователь администратором
    is_user_admin = db.is_admin(update.message.chat_id)

    # Создаем клавиатуру с кнопками
    keyboard = [
        [KeyboardButton("📰 Новости"), KeyboardButton("👥 Сотрудники")],
    ]

    if is_user_admin:
        keyboard.append([KeyboardButton("➕ Добавить новость"), KeyboardButton("📢 Рассылка")])

    keyboard.append([KeyboardButton("ℹ️ Помощь")])

    reply_markup = ReplyKeyboardMarkup(keyboard, resize_keyboard=True)

    welcome_text = "👋 *Добро пожаловать в систему HR-уведомлений!*\n\n"
    if is_user_admin:
        welcome_text += "🔑 Вы вошли как *администратор HR*\n\n"

    welcome_text += "Используйте кнопки ниже для быстрого доступа к функциям или команду /help"

    await update.message.reply_text(
        welcome_text,
        reply_markup=reply_markup,
        parse_mode='Markdown'
    )


async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /help - список всех команд"""
    if not update.message:
        return

    is_user_admin = db.is_admin(update.message.chat_id)

    help_text = "📋 *Доступные команды:*\n\n"
    help_text += "👥 *Для всех сотрудников:*\n"
    help_text += "• `/news` - Показать последние новости\n"
    help_text += "• `/company` - Список всех сотрудников\n"
    help_text += "• `/help` - Показать это сообщение\n\n"

    if is_user_admin:
        help_text += "🔧 *Для HR-администраторов:*\n"
        help_text += "• `/add_news <текст>` - Добавить новость\n"
        help_text += "• `/broadcast <текст>` - Рассылка всем сотрудникам\n"
        help_text += "• `/send <username> <текст>` - Личное сообщение\n\n"
        help_text += "📝 *Примеры:*\n"
        help_text += "`/add_news Завтра выходной`\n"
        help_text += "`/broadcast Всем! Встреча в 15:00`\n"
        help_text += "`/send mark Марк, зайди в HR`\n"

    await update.message.reply_text(help_text, parse_mode='Markdown')


async def news(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /news - показать последние новости"""
    if not update.message:
        return

    latest_news = db.get_latest_news(limit=5)

    if not latest_news:
        await update.message.reply_text("📭 *Пока нет новостей*", parse_mode='Markdown')
        return

    response = "📰 *Последние новости компании*\n"
    response += "━━━━━━━━━━━━━━━━━━━━━━\n\n"

    for i, (text, created_at) in enumerate(latest_news, 1):
        # Форматируем дату
        date_str = created_at.split()[0]
        response += f"*{i}.* {text}\n"
        response += f"   📅 _{date_str}_\n\n"

    response += "━━━━━━━━━━━━━━━━━━━━━━"

    await update.message.reply_text(response, parse_mode='Markdown')


async def company(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /company - показать всех сотрудников"""
    if not update.message:
        return

    users = db.get_all_users_full()

    if not users:
        await update.message.reply_text("👥 *Нет зарегистрированных сотрудников*", parse_mode='Markdown')
        return

    response = "👥 *Сотрудники компании*\n"
    response += "━━━━━━━━━━━━━━━━━━━━━━\n\n"

    admin_count = 0
    employee_count = 0

    for username, full_name, is_admin, created_at in users:
        if is_admin:
            admin_count += 1
            response += f"🔑 *{full_name}*\n"
            response += f"   @{username} • _Администратор HR_\n\n"
        else:
            employee_count += 1
            response += f"👤 *{full_name}*\n"
            response += f"   @{username} • _Сотрудник_\n\n"

    response += "━━━━━━━━━━━━━━━━━━━━━━\n"
    response += f"📊 Всего: {len(users)} чел. "
    response += f"(👤 {employee_count} сотр., 🔑 {admin_count} адм.)"

    await update.message.reply_text(response, parse_mode='Markdown')


# ============= КОМАНДЫ ДЛЯ HR =============

async def add_news(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /add_news - добавить новость в базу"""
    if not update.message:
        return

    # Проверяем права администратора
    if not db.is_admin(update.message.chat_id):
        await update.message.reply_text(
            "❌ У вас нет прав для выполнения этой команды.\n"
            "Только администраторы HR могут добавлять новости."
        )
        return

    # Проверяем, что передан текст новости
    if not context.args:
        await update.message.reply_text(
            "❌ Использование: /add_news <текст новости>\n"
            "Пример: /add_news Завтра выходной день"
        )
        return

    # Собираем весь текст после команды
    news_text = ' '.join(context.args)

    # Добавляем в базу
    db.add_news(news_text)

    await update.message.reply_text(
        f"✅ *Новость успешно добавлена!*\n\n"
        f"📝 _{news_text}_\n\n"
        f"Все сотрудники увидят её в разделе Новости",
        parse_mode='Markdown'
    )


async def broadcast(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /broadcast - отправить сообщение всем пользователям"""
    if not update.message:
        return

    # Проверяем права администратора
    if not db.is_admin(update.message.chat_id):
        await update.message.reply_text(
            "❌ У вас нет прав для выполнения этой команды.\n"
            "Только администраторы HR могут отправлять массовые рассылки."
        )
        return

    # Проверяем, что передан текст сообщения
    if not context.args:
        await update.message.reply_text(
            "❌ Использование: /broadcast <текст сообщения>\n"
            "Пример: /broadcast Всем! Не забудьте про встречу в 15:00"
        )
        return

    # Собираем текст сообщения
    message_text = ' '.join(context.args)

    # Получаем всех пользователей
    users = db.get_all_users()

    if not users:
        await update.message.reply_text("❌ Нет зарегистрированных пользователей")
        return

    # Отправляем сообщение каждому пользователю
    sent_count = 0
    failed_count = 0

    await update.message.reply_text(
        f"📤 *Начинаю массовую рассылку...*\n"
        f"Получателей: {len(users)} чел.",
        parse_mode='Markdown'
    )

    for chat_id, username in users:
        try:
            # КЛЮЧЕВАЯ ФУНКЦИЯ: Бот САМ отправляет сообщение пользователю
            await context.bot.send_message(
                chat_id=chat_id,
                text=f"📢 *Сообщение от HR-отдела*\n"
                     f"━━━━━━━━━━━━━━━━━━━━━━\n\n"
                     f"{message_text}",
                parse_mode='Markdown'
            )
            sent_count += 1
        except Exception as e:
            failed_count += 1
            print(f"Ошибка отправки пользователю {username} ({chat_id}): {e}")

    # Отчет о рассылке
    if update.message:
        report = f"✅ *Рассылка завершена!*\n"
        report += f"━━━━━━━━━━━━━━━━━━━━━━\n\n"
        report += f"✉️ Отправлено: *{sent_count}* чел.\n"
        if failed_count > 0:
            report += f"❌ Ошибок: *{failed_count}*\n"
        report += f"\n📝 Текст рассылки:\n_{message_text}_"

        await update.message.reply_text(report, parse_mode='Markdown')


async def send_to_user(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /send - отправить сообщение конкретному пользователю"""
    if not update.message:
        return

    # Проверяем права администратора
    if not db.is_admin(update.message.chat_id):
        await update.message.reply_text(
            "❌ У вас нет прав для выполнения этой команды.\n"
            "Только администраторы HR могут отправлять личные сообщения сотрудникам."
        )
        return

    # Проверяем формат команды
    if len(context.args) < 2:
        await update.message.reply_text(
            "❌ Использование: /send <username> <текст сообщения>\n"
            "Пример: /send ivan_petrov Иван, зайдите пожалуйста в HR отдел"
        )
        return

    # Первый аргумент - username получателя
    target_username = context.args[0]

    # Остальное - текст сообщения
    message_text = ' '.join(context.args[1:])

    # Ищем пользователя в базе
    user = db.get_user_by_username(target_username)

    if not user:
        await update.message.reply_text(
            f"❌ Пользователь '{target_username}' не найден в базе\n\n"
            f"Используйте точный username из базы данных"
        )
        return

    chat_id, username = user

    # Отправляем личное сообщение
    try:
        await context.bot.send_message(
            chat_id=chat_id,
            text=f"💌 *Личное сообщение от HR-отдела*\n"
                 f"━━━━━━━━━━━━━━━━━━━━━━\n\n"
                 f"{message_text}",
            parse_mode='Markdown'
        )

        if update.message:
            await update.message.reply_text(
                f"✅ *Сообщение доставлено!*\n\n"
                f"👤 Получатель: @{username}\n"
                f"📝 Текст: _{message_text}_",
                parse_mode='Markdown'
            )
    except Exception as e:
        if update.message:
            await update.message.reply_text(
                f"❌ *Ошибка при отправке*\n\n{str(e)}",
                parse_mode='Markdown'
            )


# ============= ОБРАБОТЧИК КНОПОК =============

async def handle_button_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Обработчик текстовых сообщений от кнопок"""
    if not update.message or not update.message.text:
        return

    text = update.message.text

    if text == "📰 Новости":
        await news(update, context)
    elif text == "👥 Сотрудники":
        await company(update, context)
    elif text == "ℹ️ Помощь":
        await help_command(update, context)
    elif text == "➕ Добавить новость":
        await update.message.reply_text(
            "📝 Используйте команду:\n"
            "`/add_news <текст новости>`\n\n"
            "Пример:\n"
            "`/add_news Завтра выходной день`",
            parse_mode='Markdown'
        )
    elif text == "📢 Рассылка":
        await update.message.reply_text(
            "📢 Используйте команду:\n"
            "`/broadcast <текст сообщения>`\n\n"
            "Пример:\n"
            "`/broadcast Всем! Встреча в 15:00`",
            parse_mode='Markdown'
        )


# ============= ЗАПУСК БОТА =============

def main():
    """Главная функция - запуск бота"""

    print("🤖 Инициализация HR Notification Bot...")

    # Инициализируем базу данных
    db.init_db()
    db.seed_test_data()

    # Создаем приложение
    application = Application.builder().token(BOT_TOKEN).build()

    # Регистрируем обработчики команд
    application.add_handler(CommandHandler("start", start))
    application.add_handler(CommandHandler("help", help_command))
    application.add_handler(CommandHandler("news", news))
    application.add_handler(CommandHandler("company", company))
    application.add_handler(CommandHandler("add_news", add_news))
    application.add_handler(CommandHandler("broadcast", broadcast))
    application.add_handler(CommandHandler("send", send_to_user))

    # Обработчик текстовых сообщений (для кнопок)
    from telegram.ext import MessageHandler, filters
    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_button_text))

    print("✅ Бот запущен и готов к работе!")
    print("Нажмите Ctrl+C для остановки\n")

    # Запускаем бота
    application.run_polling(allowed_updates=Update.ALL_TYPES)


if __name__ == '__main__':
    main()