import os
import logging
import httpx
from aiogram import Bot, Dispatcher, types
from aiogram.filters import Command
from aiogram.client.default import DefaultBotProperties
from aiogram.enums import ParseMode

# Настройки
TOKEN = os.getenv("TELEGRAM_TOKEN")
API_URL = os.getenv("VNDB_API_URL", "https://api.vndb.org/kana")

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Исправленная инициализация для aiogram 3.7.0+
bot = Bot(
    token=TOKEN, 
    default=DefaultBotProperties(parse_mode=ParseMode.HTML)
)
dp = Dispatcher()

async def fetch_vndb(endpoint: str, filters: list, fields: str):
    """Универсальный и безопасный клиент для VNDB API"""
    payload = {"filters": filters, "fields": fields}
    async with httpx.AsyncClient(timeout=10.0) as client:
        try:
            response = await client.post(f"{API_URL}/{endpoint}", json=payload)
            if response.status_code == 200:
                return response.json().get("results", [])
            logger.error(f"VNDB API Error {response.status_code}: {response.text}")
            return None
        except Exception as e:
            logger.error(f"Request failed: {e}")
            return None

@dp.message(Command("start", "help"))
async def cmd_help(message: types.Message):
    help_text = (
        "🤖 <b>VNDB Bot v2.0</b>\n\n"
        "<b>Поиск:</b>\n"
        "/search &lt;name&gt; — поиск VN\n"
        "/char &lt;name&gt; — поиск персонажа\n"
        "/release &lt;name&gt; — поиск релиза\n\n"
        "<b>Инфо по ID:</b>\n"
        "/vn &lt;id&gt; (напр. v17)\n"
        "/char_id &lt;id&gt; (напр. c1)\n"
        "/rel_id &lt;id&gt; (напр. r1)"
    )
    await message.answer(help_text)

# --- SEARCH HANDLERS ---

@dp.message(Command("search"))
async def search_vn(message: types.Message):
    query = message.extract_args()
    if not query: return await message.answer("Пример: <code>/search Steins;Gate</code>")
    
    res = await fetch_vndb("vn", ["search", "=", query], "title, rating")
    if not res: return await message.answer("❌ Ничего не найдено.")
    
    out = ["🔍 <b>VN Results:</b>"]
    for i in res[:10]:
        out.append(f"• {i['title']} (ID: <code>{i['id']}</code>)")
    await message.answer("\n".join(out))

@dp.message(Command("char"))
async def search_char(message: types.Message):
    query = message.extract_args()
    if not query: return await message.answer("Пример: <code>/char Kurisu</code>")
    
    res = await fetch_vndb("character", ["search", "=", query], "name")
    if not res: return await message.answer("❌ Персонаж не найден.")
    
    out = ["👤 <b>Characters:</b>"]
    for i in res[:10]:
        out.append(f"• {i['name']} (ID: <code>{i['id']}</code>)")
    await message.answer("\n".join(out))

@dp.message(Command("release"))
async def search_release(message: types.Message):
    query = message.extract_args()
    if not query: return await message.answer("Пример: <code>/release Chaos;Head</code>")
    
    res = await fetch_vndb("release", ["search", "=", query], "title")
    if not res: return await message.answer("❌ Релиз не найден.")
    
    out = ["💿 <b>Releases:</b>"]
    for i in res[:10]:
        out.append(f"• {i['title']} (ID: <code>{i['id']}</code>)")
    await message.answer("\n".join(out))

# --- DETAIL HANDLERS ---

@dp.message(Command("vn"))
async def detail_vn(message: types.Message):
    vid = message.extract_args()
    if not vid: return await message.answer("Введите ID (например v17)")
    
    res = await fetch_vndb("vn", ["id", "=", vid], "id, title, original, released, rating, votecount")
    if not res: return await message.answer("❌ VN не найдена.")
    
    v = res[0]
    rating = f"{v['rating']/10} ⭐" if v.get('rating') else "N/A"
    text = (f"📖 <b>{v['title']}</b>\n"
            f"Original: {v.get('original', 'N/A')}\n"
            f"Released: {v.get('released', 'N/A')}\n"
            f"Rating: {rating} ({v.get('votecount', 0)} votes)\n"
            f"https://vndb.org/{v['id']}")
    await message.answer(text)

@dp.message(Command("char_id"))
async def detail_char(message: types.Message):
    cid = message.extract_args()
    if not cid: return await message.answer("Введите ID (например c1)")
    
    res = await fetch_vndb("character", ["id", "=", cid], "id, name, original")
    if not res: return await message.answer("❌ Персонаж не найден.")
    
    c = res[0]
    text = (f"👤 <b>{c['name']}</b>\n"
            f"Original: {c.get('original', 'N/A')}\n"
            f"https://vndb.org/{c['id']}")
    await message.answer(text)

@dp.message(Command("rel_id"))
async def detail_rel(message: types.Message):
    rid = message.extract_args()
    if not rid: return await message.answer("Введите ID (например r1)")
    
    res = await fetch_vndb("release", ["id", "=", rid], "id, title, released")
    if not res: return await message.answer("❌ Релиз не найден.")
    
    r = res[0]
    text = (f"💿 <b>{r['title']}</b>\n"
            f"Released: {r.get('released', 'N/A')}\n"
            f"https://vndb.org/{r['id']}")
    await message.answer(text)

if __name__ == "__main__":
    dp.run_polling(bot)