import os
import uuid

import pytest
import pytest_asyncio
from dotenv import load_dotenv

load_dotenv()

from httpx import AsyncClient, ASGITransport
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker

from src.database.config import get_db
from src.database.models import Base, Tenant, User
from src.auth.password import hash_password

TEST_DATABASE_URL = os.getenv(
    "TEST_DATABASE_URL",
    "postgresql+asyncpg://factoryops:factoryops@localhost:5432/factoryops_v2_test",
)


@pytest_asyncio.fixture(scope="function")
async def db_session():
    engine = create_async_engine(TEST_DATABASE_URL, echo=False, pool_size=5)

    async with engine.begin() as conn:
        await conn.run_sync(Base.metadata.create_all)

    session_factory = async_sessionmaker(
        engine, class_=AsyncSession, expire_on_commit=False
    )

    async with session_factory() as session:
        yield session

    async with engine.begin() as conn:
        await conn.run_sync(Base.metadata.drop_all)

    await engine.dispose()


@pytest_asyncio.fixture(scope="function")
async def client(db_session: AsyncSession):
    from main import app

    async def override_get_db():
        yield db_session

    app.dependency_overrides[get_db] = override_get_db

    transport = ASGITransport(app=app)
    async with AsyncClient(transport=transport, base_url="http://test") as ac:
        yield ac

    app.dependency_overrides.clear()


@pytest_asyncio.fixture(scope="function")
async def seeded_db(db_session: AsyncSession):
    db_session.add(
        Tenant(id="test-co", name="Test Company", industry_type="manufacturing")
    )
    db_session.add(Tenant(id="other-co", name="Other Company", industry_type="general"))
    await db_session.commit()

    db_session.add(
        User(
            id=uuid.uuid4(),
            email="super@test.com",
            password_hash=hash_password("pass1234"),
            name="Super Admin",
            role="superadmin",
            tenant_id=None,
        )
    )
    db_session.add(
        User(
            id=uuid.uuid4(),
            email="admin@test-co.com",
            password_hash=hash_password("pass1234"),
            name="Tenant Admin",
            role="tenant_admin",
            tenant_id="test-co",
        )
    )
    await db_session.commit()
    return db_session


async def get_auth_headers(
    client: AsyncClient, email: str = "super@test.com", password: str = "pass1234"
) -> dict:
    resp = await client.post(
        "/api/auth/login", json={"email": email, "password": password}
    )
    token = resp.json()["access_token"]
    return {"Authorization": f"Bearer {token}"}