import requests
import json

def get_vehicle_info(vehicle_number):
    vehicle_number = vehicle_number.upper()

    url = (
        "https://www.carinfo.app/_next/data/"
        "lEH1JDnqa1BNfE88p-0Q9/"
        f"rto-vehicle-registration-detail/rto-details/{vehicle_number}.json"
    )

    headers = {
        "User-Agent": "Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36",
        "Accept-Encoding": "gzip, deflate, br, zstd",
        "x-nextjs-data": "1",
        "sec-ch-ua-platform": "\"Android\"",
        "sec-ch-ua": "\"Google Chrome\";v=\"143\", \"Chromium\";v=\"143\", \"Not A(Brand\";v=\"24\"",
        "sec-ch-ua-mobile": "?0",
        "sec-fetch-site": "same-origin",
        "sec-fetch-mode": "cors",
        "sec-fetch-dest": "empty",
        "referer": "https://www.carinfo.app/rto-vehicle-registration-detail",
        "accept-language": "en-IN,en-GB;q=0.9,en-US;q=0.8,en;q=0.7",
        "priority": "u=1, i",
        "Cookie": "ABBASJWE8EB3IDJWINDNDOE93NR"
    }

    response = requests.get(url, headers=headers, timeout=20)

    # ---- Safety checks ----
    if response.status_code != 200:
        return {"error": "HTTP error", "status": response.status_code}

    if "application/json" not in response.headers.get("Content-Type", ""):
        return {
            "error": "Blocked or session expired",
            "preview": response.text[:200]
        }

    try:
        data = response.json()
    except Exception as e:
        return {"error": "JSON decode failed", "message": str(e)}

    # ---- Extract only real data ----
    section = data["pageProps"]["rtoDetailsReponse"]["webSections"][0]
    messages = section.get("messages", [])
    vehicle_msg = section.get("message", {})

    info = {m["title"]: m["subtitle"] for m in messages}
    registered_rto = info.get("Registered RTO", "")

    return {
        "vehicle_number": data["pageProps"].get("vehicalNum", ""),
        "registration_state": info.get("State", ""),
        "owner_name": vehicle_msg.get("title", ""),
        "model": vehicle_msg.get("subtitle", ""),
        "vehicle_type": "Two Wheeler",
        "rto_code": info.get("Number", ""),
        "rto_name": registered_rto.split(",")[0] if registered_rto else "",
        "state": info.get("State", ""),
        "rto_image": section.get("imageUrl", "")
    }


# ---- Run ----
if __name__ == "__main__":
    result = get_vehicle_info("mh14kk9159")
    print(json.dumps(result, indent=2, ensure_ascii=False))