import asyncio
import logging
from aiogram import Bot, Dispatcher, types
from aiogram.filters import CommandStart, Text
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton
import aiosqlite

API_TOKEN = "8600595898:AAFKtKLiicIP60NadX50TbhJYYYhVc1eLwU"

logging.basicConfig(level=logging.INFO)

bot = Bot(token=API_TOKEN)
dp = Dispatcher(bot)

# ------------------- ИНИЦИАЛИЗАЦИЯ БАЗЫ -------------------
async def init_db():
async with aiosqlite.connect("borrelbuddy.db") as db:
await db.execute("""
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
telegram_id INTEGER UNIQUE,
name TEXT,
age INTEGER,
city TEXT,
district TEXT,
drink TEXT,
bio TEXT
)
""")
await db.execute("""
CREATE TABLE IF NOT EXISTS likes (
from_user INTEGER,
to_user INTEGER
)
""")
await db.commit()

# ------------------- СТЕЙТЫ РЕГИСТРАЦИИ -------------------
user_states = {}

# ------------------- СТАРТ -------------------
@dp.message(CommandStart())
async def start(message: types.Message):
kb = ReplyKeyboardMarkup(
keyboard=[
[KeyboardButton("🍻 Создать профиль")],
[KeyboardButton("🔍 Найти компанию")],
[KeyboardButton("💬 Мои матчи")]
],
resize_keyboard=True
)
await message.answer("Привет! Добро пожаловать в BorrelBuddy 🍺\nНе пьём в одиночку!", reply_markup=kb)

# ------------------- СОЗДАНИЕ ПРОФИЛЯ -------------------
@dp.message(Text("🍻 Создать профиль"))
async def create_profile(message: types.Message):
user_states[message.from_user.id] = {"step": "name"}
await message.answer("Как тебя зовут?")

@dp.message()
async def register(message: types.Message):
user_id = message.from_user.id
if user_id not in user_states:
return
state = user_states[user_id]

step = state["step"]

if step == "name":
state["name"] = message.text
state["step"] = "age"
await message.answer("Сколько тебе лет?")
elif step == "age":
if not message.text.isdigit():
await message.answer("Введите число")
return
state["age"] = int(message.text)
state["step"] = "city"
await message.answer("В каком городе ты?")
elif step == "city":
state["city"] = message.text
state["step"] = "district"
await message.answer("В каком районе?")
elif step == "district":
state["district"] = message.text
state["step"] = "drink"
await message.answer("Что предпочитаешь пить? (пиво/вино/коктейль/всё равно)")
elif step == "drink":
state["drink"] = message.text
state["step"] = "bio"
await message.answer("Напиши коротко о себе (например: 'После работы ищу бар')")
elif step == "bio":
state["bio"] = message.text
# Сохраняем в базу
async with aiosqlite.connect("borrelbuddy.db") as db:
await db.execute("""
INSERT OR REPLACE INTO users
(telegram_id, name, age, city, district, drink, bio)
VALUES (?, ?, ?, ?, ?, ?, ?)
""", (
user_id,
state["name"],
state["age"],
state["city"],
state["district"],
state["drink"],
state["bio"]
))
await db.commit()
del user_states[user_id]
await message.answer("Анкета создана! 🔥\nТеперь можешь искать компанию.")

# ------------------- ПОИСК -------------------
@dp.message(Text("🔍 Найти компанию"))
async def search(message: types.Message):
user_id = message.from_user.id
async with aiosqlite.connect("borrelbuddy.db") as db:
async with db.execute("""
SELECT telegram_id, name, age, city, district, drink, bio
FROM users
WHERE telegram_id != ?
""", (user_id,)) as cursor:
users = await cursor.fetchall()

if not users:
await message.answer("Пока нет других участников в твоем районе 😢")
return

# Показываем случайного пользователя
target = users[0] # Можно random.choice(users) для случайного
user_states[user_id] = {"step": "like", "target": target[0]}

kb = ReplyKeyboardMarkup(
keyboard=[
[KeyboardButton("❤️ Погнали"), KeyboardButton("👎 Не сегодня")]
],
resize_keyboard=True
)

await message.answer(
f"{target[1]}, {target[2]} лет\n📍 {target[4]}, {target[3]}\n🍺 {target[5]}\n\n{target[6]}",
reply_markup=kb
)

# ------------------- ЛАЙК / ПРОПУСК -------------------
@dp.message(Text(["❤️ Погнали", "👎 Не сегодня"]))
async def like_skip(message: types.Message):
user_id = message.from_user.id
state = user_states.get(user_id)
if not state or state.get("step") != "like":
return

target_id = state["target"]

if message.text == "❤️ Погнали":
async with aiosqlite.connect("borrelbuddy.db") as db:
await db.execute("INSERT INTO likes VALUES (?, ?)", (user_id, target_id))
await db.commit()
# Проверяем взаимный лайк
async with db.execute("SELECT 1 FROM likes WHERE from_user=? AND to_user=?", (target_id, user_id)) as cursor:
match = await cursor.fetchone()
if match:
await message.answer("🔥 Отлично! Взаимный матч! Напиши друг другу.")
await bot.send_message(target_id, f"🔥 У вас взаимный матч с {user_id}!")

await message.answer("Продолжай поиск 🔍 / Найти компанию")

# ------------------- ЗАПУСК -------------------
async def main():
await init_db()
await dp.start_polling(bot)

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