from handlers.share_bot import share_bot_handler
from handlers.invite_admin import invite_admin_handler
import sys
import asyncio
if sys.platform.startswith('win'):
    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
import logging
import os
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import Application, CommandHandler, MessageHandler, filters, CallbackQueryHandler, ConversationHandler, ContextTypes
from dotenv import load_dotenv
from db import AsyncSessionLocal, init_db
from models import Admin, Channel, Group, Button
from asyncio import run as sync_to_async
from handlers.admin_panel import admin_panel_conv
load_dotenv()
TELEGRAM_TOKEN = os.getenv('TELEGRAM_TOKEN')

logging.basicConfig(level=logging.INFO)
logging.getLogger("httpx").setLevel(logging.WARNING)
logger = logging.getLogger(__name__)

import asyncio


async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    session = AsyncSessionLocal()
    user_id = update.effective_user.id if update.effective_user else None
    result = await session.execute(Admin.__table__.select().where(Admin.tg_id == user_id))
    admin = result.first() if user_id else None
    if not admin and user_id:
        await session.execute(Admin.__table__.insert().values(tg_id=user_id))
        await session.commit()
        if update.message:
            await update.message.reply_text('Вы зарегистрированы как админ.')
    else:
        if update.message:
            await update.message.reply_text('Вы уже зарегистрированы.')
    await session.close()

async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
    help_text = (
        "<b>Справка по командам бота:</b>\n\n"
        "<b>/start</b> — регистрация пользователя как администратора.\n"
        "<b>/help</b> — показать это подробное описание команд.\n\n"
        "<b>/add_channel &lt;название&gt; &lt;ссылка&gt;</b> — добавить канал для публикаций. Пример: /add_channel MyChannel @my_channel\n"
        "<b>/add_group &lt;название&gt; &lt;ссылка&gt;</b> — добавить группу для публикаций. Пример: /add_group MyGroup @my_group\n"
        "<b>/add_button</b> — добавить кнопку к выбранному каналу или группе. Запустит диалог выбора и добавления.\n"
        "<b>/edit_button &lt;название&gt; &lt;новое_название&gt; &lt;новая_ссылка&gt;</b> — изменить кнопку. Пример: /edit_button old new https://site.ru\n"
        "<b>/del_button &lt;название&gt;</b> — удалить кнопку по названию.\n"
        "<b>/new_post</b> — создать новый пост (картинка + текст + выбор канала/группы + кнопки).\n"
        "<b>/group_buttons</b> — показать и настроить кнопки для группы.\n"
        "<b>/channel_buttons</b> — показать и настроить кнопки для канала.\n\n"
        "<b>Примеры использования:</b>\n"
        "- /add_channel Новости @news_channel\n"
        "- /add_group Чат @chat_group\n"
        "- /edit_button Подписка Subscribe https://subscribe.ru\n"
        "- /del_button Subscribe\n\n"
        "<b>Порядок работы:</b>\n"
        "1. Добавьте канал или группу.\n"
        "2. Добавьте кнопки для них.\n"
        "3. Создайте пост через /new_post и выберите, куда отправить.\n"
        "4. Управляйте кнопками через /group_buttons и /channel_buttons.\n\n"
        "Если возникли вопросы — используйте /help или обратитесь к администратору."
    )
    if update.message:
        await update.message.reply_text(help_text, parse_mode='HTML')

# Импорт обработчиков
from handlers.add_channel import add_channel_conv
from handlers.add_group import add_group_conv
from handlers.add_button import add_button_conv
from handlers.new_post import new_post_conv
from handlers.group_buttons import group_buttons_conv
from handlers.channel_buttons import channel_buttons_conv
from handlers.edit_button import edit_button
from handlers.del_button import del_button



def main():
    if not TELEGRAM_TOKEN:
        print("Ошибка: TELEGRAM_TOKEN не найден в переменных окружения.")
        return
    sync_to_async(init_db())
    application = Application.builder().token(TELEGRAM_TOKEN).build()
    application.add_handler(CommandHandler('start', start))
    application.add_handler(CommandHandler('help', help_command))
    application.add_handler(add_channel_conv)
    application.add_handler(add_group_conv)
    application.add_handler(add_button_conv)
    application.add_handler(new_post_conv)
    application.add_handler(group_buttons_conv)
    application.add_handler(channel_buttons_conv)
    application.add_handler(CommandHandler('edit_button', edit_button))
    application.add_handler(CommandHandler('del_button', del_button))
    application.add_handler(admin_panel_conv)
    application.add_handler(share_bot_handler)
    application.add_handler(invite_admin_handler)
    import sys
    import asyncio
    if sys.platform.startswith('win'):
        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
    try:
        loop = asyncio.get_event_loop()
    except RuntimeError:
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
    application.run_polling()

if __name__ == "__main__":
    main()