import asyncio
import httpx
from sqlalchemy import select, delete
from sqlalchemy.orm import selectinload
from app.db.session import AsyncSessionLocal
from app.models.identity import User
from app.models.marketplace.organization import Organization, Branch, OrganizationMember
from app.models.vehicle.asset import Asset
from app.models.vehicle.vehicle_definitions import VehicleModelDefinition
from app.core.security import create_tokens

async def setup_test_data():
    async with AsyncSessionLocal() as db:
        # Just find any 2 users that belong to ANY organization
        stmt = select(OrganizationMember).limit(2)
        members = (await db.execute(stmt)).scalars().all()
        if not members:
            print("No org members found")
            return None
        
        user1_id = members[0].user_id
        org_id = members[0].organization_id
        
        user2_id = members[1].user_id if len(members)>1 else user1_id

        # Find or create branch
        branch_stmt = select(Branch).where(Branch.organization_id == org_id, Branch.is_main == True)
        branch = (await db.execute(branch_stmt)).scalar()
        if not branch:
            branch = Branch(organization_id=org_id, name="Test Main Branch", is_main=True)
            db.add(branch)
            await db.flush()

        # Clean old test data
        await db.execute(delete(Asset).where(Asset.license_plate.in_(["TEST-123", "DRAFT-456", "OTHER-123"])))
        await db.execute(delete(VehicleModelDefinition).where(VehicleModelDefinition.make == "Ford", VehicleModelDefinition.marketing_name == "Focus"))
        await db.commit()

        model_def = VehicleModelDefinition(
            make="Ford",
            marketing_name="Focus",
            normalized_name="ford_focus",
            year_from=2015,
            year_to=2020,
            power_kw=92,
            data_status="verified"
        )
        db.add(model_def)
        await db.commit()

        token1, _ = create_tokens(data={"sub": str(user1_id)})
        token2, _ = create_tokens(data={"sub": str(user2_id)})
        
        return token1, token2, org_id, branch.id, model_def.id

async def run_tests():
    print("--- SETUP ---")
    result = await setup_test_data()
    if not result:
        return
    token1, token2, org_id, branch_id, model_id = result
    print("Test data created successfully.")

    headers1 = {"Authorization": f"Bearer {token1}"}
    headers2 = {"Authorization": f"Bearer {token2}"}

    async with httpx.AsyncClient(base_url="http://127.0.0.1:8000") as client:
        print("\n--- TEST 1: CATALOG MATCH & GARAGE TEST ---")
        payload1 = {
            "vin": "TESTVIN1234567890",
            "license_plate": "TEST-123",
            "brand": "Ford",
            "model": "Focus",
            "year_of_manufacture": 2018,
            "organization_id": org_id,
            "vehicle_class": "car",
            "fuel_type": "petrol"
        }
        resp1 = await client.post("/api/v1/vehicles", json=payload1, headers=headers1)
        print(f"Status: {resp1.status_code}")
        try:
            data1 = resp1.json()
            print(f"Response: {data1}")
            if resp1.status_code == 201:
                assert data1["catalog_id"] == model_id, f"Matcher failed: expected {model_id}, got {data1.get('catalog_id')}"
                assert data1["branch_id"] == str(branch_id), f"Branch failed: expected {branch_id}, got {data1.get('branch_id')}"
                print("-> SUCCESS: Matcher assigned catalog and Main Branch assigned.")
        except Exception as e:
            print(f"-> FAILED: {e}")

        print("\n--- TEST 2: DRAFT / INCOMPLETE TEST ---")
        payload2 = {
            "license_plate": "DRAFT-456",
            "brand": "Ismeretlen",
            "organization_id": org_id
        }
        resp2 = await client.post("/api/v1/vehicles", json=payload2, headers=headers1)
        print(f"Status: {resp2.status_code}")
        try:
            data2 = resp2.json()
            print(f"Response: {data2}")
            if resp2.status_code == 201:
                assert data2["status"] == "draft", f"Status failed: expected 'draft', got {data2.get('status')}"
                assert data2["branch_id"] == str(branch_id), f"Branch failed: expected {branch_id}, got {data2.get('branch_id')}"
                print("-> SUCCESS: Draft created and Main Branch assigned.")
        except Exception as e:
            print(f"-> FAILED: {e}")

        print("\n--- TEST 3: OWNERSHIP TRANSFER TEST ---")
        payload3 = {
            "vin": "TESTVIN1234567890",
            "license_plate": "OTHER-123",
            "brand": "Ford",
            "model": "Focus",
            "organization_id": org_id
        }
        resp3 = await client.post("/api/v1/vehicles", json=payload3, headers=headers2)
        print(f"Status: {resp3.status_code}")
        try:
            data3 = resp3.json()
            print(f"Response: {data3}")
            if resp3.status_code == 202:
                print("-> SUCCESS: VIN collision detected (transfer_pending).")
        except Exception as e:
            print(f"-> FAILED: {e}")

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