# handlers/add_channel.py
from telegram import Update
from telegram.ext import (
    ContextTypes, ConversationHandler, CommandHandler, MessageHandler, filters
)
from sqlalchemy import select

from db import AsyncSessionLocal
from models import Channel, Admin

INPUT_NAME, INPUT_LINK = range(2)

async def add_channel_start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    if context.user_data is None:
        context.user_data = {}
    if update.message:
        await update.message.reply_text('Введите имя канала:')
    return INPUT_NAME

async def input_channel_name(update: Update, context: ContextTypes.DEFAULT_TYPE):
    if not update.message:
        return ConversationHandler.END
    name = (update.message.text or "").strip()
    if not name:
        await update.message.reply_text("Имя не может быть пустым. Введите имя канала:")
        return INPUT_NAME
    context.user_data["channel_name"] = name
    await update.message.reply_text('Отправьте ссылку на канал (формат "@username" или "-100..."):')
    return INPUT_LINK

async def _get_or_create_admin(session, tg_id: int) -> Admin:
    res = await session.execute(select(Admin).where(Admin.tg_id == tg_id))
    admin = res.scalar_one_or_none()
    if not admin:
        admin = Admin(tg_id=tg_id)
        session.add(admin)
        await session.flush()
    return admin

async def input_channel_link(update: Update, context: ContextTypes.DEFAULT_TYPE):
    if not update.message:
        return ConversationHandler.END

    link = (update.message.text or "").strip()
    if not (link.startswith("@") or link.startswith("-100")):
        await update.message.reply_text('Неверный формат. Укажите "@username" или "-100...".')
        return INPUT_LINK

    name = (context.user_data or {}).get("channel_name", "").strip()
    if not name:
        await update.message.reply_text("Не найдено имя. Начните заново: /add_channel")
        return ConversationHandler.END

    user = update.effective_user
    if not user:
        await update.message.reply_text("Не удалось определить администратора.")
        return ConversationHandler.END

    async with AsyncSessionLocal() as session:
        admin = await _get_or_create_admin(session, user.id)

        # если канал уже есть — обновим имя и владельца
        existing_q = await session.execute(select(Channel).where(Channel.link == link))
        existing = existing_q.scalar_one_or_none()
        if existing:
            existing.name = name
            existing.admin_id = admin.id
            await session.commit()
            await update.message.reply_text(f'Канал "{name}" уже был — обновил владельца и имя.')
        else:
            channel = Channel(name=name, link=link, admin_id=admin.id)
            session.add(channel)
            await session.commit()
            await update.message.reply_text(f'Канал "{name}" добавлен и привязан к вашему админ-аккаунту.')
    return ConversationHandler.END

add_channel_conv = ConversationHandler(
    entry_points=[CommandHandler('add_channel', add_channel_start)],
    states={
        INPUT_NAME: [MessageHandler(filters.TEXT & ~filters.COMMAND, input_channel_name)],
        INPUT_LINK: [MessageHandler(filters.TEXT & ~filters.COMMAND, input_channel_link)],
    },
    fallbacks=[]
)