import os
import uuid
import pytest
from sqlalchemy import create_engine, text
from sqlalchemy.orm import sessionmaker
from app.db import Base, database
from app.main import app
from app import models as m
from app.config import settings
from fastapi.testclient import TestClient


@pytest.fixture(scope='session')
def engine():
    url = os.environ.get('OMOTERRA_TEST_DATABASE_URL')
    if not url:
        pytest.skip('Set OMOTERRA_TEST_DATABASE_URL to a disposable PostgreSQL database')
    engine = create_engine(url)
    # Dedicated schema keeps other applications and tables untouched.
    schema = 'omoterra_tests_' + uuid.uuid4().hex
    with engine.begin() as db:
        db.execute(text(f'CREATE SCHEMA {schema}'))
    engine.dispose()
    engine = create_engine(url, connect_args={'options': f'-csearch_path={schema}'})
    Base.metadata.create_all(engine)
    from app.migrate_inventory import install_history_guards
    install_history_guards(engine)
    yield engine
    with engine.begin() as db:
        db.execute(text(f'DROP SCHEMA {schema} CASCADE'))
    engine.dispose()


@pytest.fixture
def sessions(engine):
    with engine.begin() as connection:
        for table in reversed(Base.metadata.sorted_tables):
            connection.execute(text(f'TRUNCATE "{table.name}" CASCADE'))
    return sessionmaker(engine, expire_on_commit=False)


@pytest.fixture
def seeded(sessions):
    from datetime import timedelta
    with sessions.begin() as db:
        buyer = m.User(phone='+255712345678', roles=['buyer'], name='Buyer Test', region='Dar')
        other = m.User(phone='+255712345679', roles=['buyer'], name='Other Buyer', region='Dar')
        supplier = m.User(phone='+255712345680', roles=['supplier'], name='Private Supplier', region='Pwani')
        db.add_all([buyer, other, supplier]); db.flush()
        db.add(m.SupplierProfile(user_id=supplier.id, legal_name='Secret legal name', internal_pickup_address='Secret farm address', public_alias='Green Pastures', alias_approved=True))
        listing = m.Listing(supplier_id=supplier.id, category='broilers', unit_type='bird', specs={'avg_weight_kg': '2', 'breed_type': 'Broiler', 'age_weeks': '6', 'live_or_dressed': 'live', 'ready_date': '2027-01-01'}, region='Pwani', quantity_total=10, farmer_asking_price_per_unit=10000, supplier_payout_price_per_unit=9000, buyer_price_per_unit=12000, listing_status='live', last_confirmed_at=m.now(), confirmation_due_at=m.now() + timedelta(hours=48), approved_at=m.now())
        address = m.Address(user_id=buyer.id, label='Home', recipient_name='Private Buyer', phone=buyer.phone, region='Dar', district_area='Kinondoni', address_text='Secret buyer address')
        db.add_all([listing, address]); db.flush()
        return {'buyer': buyer.id, 'other': other.id, 'supplier': supplier.id, 'listing': listing.id, 'address': address.id}


@pytest.fixture
def client(sessions, seeded):
    from datetime import timedelta
    from app.auth import digest
    def override():
        with sessions() as db:
            with db.begin():
                yield db
    app.dependency_overrides[database] = override
    settings().ops_token = 'test-operator-secret'
    with sessions.begin() as db:
        for role in ['buyer', 'supplier', 'other']:
            db.add(m.AuthSession(user_id=seeded[role], token_hash=digest(role), expires_at=m.now() + timedelta(days=1)))
    with TestClient(app) as client:
        yield client
    app.dependency_overrides.clear()