import base64
import json
import secrets
import time
import uuid
import requests
from Crypto.Cipher import AES, PKCS1_v1_5
from Crypto.PublicKey import RSA
from Crypto.Util.Padding import pad, unpad
API = "https://api.yallabaloot.com/account/v2/passwordLogin"
IV = b"8460257940005478"
PUCK3 = (
    "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAw4tQ9I7GINCnxfxoHF4vLe15fyYzGI/mNoXTRG3P"
    "xOtcHZ0oNW5oGcGqCBw4Zw8e/bcgfhn5tWTmfEJPnVny+hmF+Jj6KHpD4mbefHXCnrDs6Zeu/Nh3gjj/M/LXt"
    "9z1zs5Rt6WG1XQw00fJ9xmm3dBvTRn7f7S4kXLyhhwu3GW0iWdCK2Srrw6KlJ5kOrV9/dEFgF/OYCowU0w46"
    "SVki8VCbS7UZb0yIZAcpHOSXlU3UAKKuUDgldNfaMYtInmSqhAAghHptRZziLhMmnUS+gqtdw8nDItwQQEsv"
    "FLe0sftAo/iJPHdIY3RS6sWpHvy/dBhyzUIzEtN6CvKw6hoewIDAQAB"
)

phone = input("Phone : ").strip()
password = input("Password: ").strip()

try:
    device_id = open(".yalla_device").read().strip()
except Exception:
    device_id = uuid.uuid4().hex
    open(".yalla_device", "w").write(device_id)

uu = "".join(secrets.choice("0123456789abcdef") for _ in range(4)) * 8
key_bytes = uu.encode("utf-8")

body_obj = {"phone": phone, "password": password, "nationalityId": 8}
head = {
    "deviceId": device_id,
    "deviceCode": "BRP-N21",
    "osType": "6",
    "osVersion": "1.5.2.1",
    "osVersionCode": "15200",
    "userAgent": "edge",
    "platform": "6",
    "timestamp": str(int(time.time() * 1000)),
    "uuId": uu,
    "osInfo": "16",
    "sign": "sign",
    "netWork": "WIFI",
}

c1 = AES.new(key_bytes, AES.MODE_CBC, IV)
data_b64 = base64.b64encode(c1.encrypt(pad(json.dumps(body_obj, separators=(",", ":")).encode(), 16))).decode()

c2 = AES.new(key_bytes, AES.MODE_CBC, IV)
head_b64 = base64.b64encode(c2.encrypt(pad(json.dumps(head, separators=(",", ":")).encode(), 16))).decode()

pub = RSA.import_key(base64.b64decode(PUCK3))
key_b64 = base64.b64encode(PKCS1_v1_5.new(pub).encrypt(key_bytes)).decode()

payload = {"data": data_b64}
headers = {
    "User-Agent": "Dalvik/2.1.0 (Linux; U; Android 16; BRP-NX1 Build/HONORBRP-N21)",
    "Connection": "Keep-Alive",
    "Accept-Encoding": "identity",
    "HeadContent": head_b64,
    "Key": key_b64,
    "ApiVersion": "",
    "Content-Type": "application/json; charset=UTF-8;",
}
r = requests.post(API, json=payload, headers=headers, timeout=25)

rkey = r.headers.get("key") or r.headers.get("Key")
if not rkey:
    print(r.text[:2000])
else:
    j = json.loads(r.text)
    if j.get("data") and j["data"] != "{}":
        c3 = AES.new(rkey.encode("utf-8"), AES.MODE_CBC, IV)
        plain = unpad(c3.decrypt(base64.b64decode(j["data"])), 16).decode("utf-8", "replace")
        j["data"] = json.loads(plain)
    print(json.dumps(j, indent=2, ensure_ascii=False))