#!/usr/bin/env python3
"""Uji cepat 2 fakta hukum yang dipakai di tambahan isi artikel 3234 (tslaw.id)."""
import json, urllib.request

tok = json.load(open("/home/mantapman/.hermes/profiles/atun/workspace/.serper_token.json"))
KEY = tok.get("api_key") or tok.get("key") or list(tok.values())[0]


def cari(q):
    req = urllib.request.Request(
        "https://google.serper.dev/search",
        data=json.dumps({"q": q, "gl": "id", "hl": "id", "num": 5}).encode(),
        headers={"X-API-KEY": KEY, "Content-Type": "application/json"})
    with urllib.request.urlopen(req, timeout=60) as r:
        d = json.load(r)
    out = []
    for o in d.get("organic", [])[:5]:
        out.append({"t": o.get("title"), "s": (o.get("snippet") or "")[:300],
                    "u": o.get("link")})
    return d.get("answerBox") or d.get("knowledgeGraph") or None, out


for q in ["apakah usaha mikro wajib lapor LKPM setiap triwulan OSS",
          "PT Perorangan didirikan tanpa akta notaris satu orang",
          "CV sekutu aktif tanggung jawab sampai harta pribadi"]:
    ab, org = cari(q)
    print("=" * 100)
    print("Q:", q)
    if ab:
        print("ANSWERBOX:", json.dumps(ab, ensure_ascii=False)[:600])
    for o in org:
        print(" -", o["t"], "|", o["u"])
        print("   ", o["s"])