#!/usr/bin/env python3
"""DAS: uji apakah kanonik + noindex Yoast bisa ditulis lewat XML-RPC (pakai 1 artikel percobaan)."""
import base64, html, json, re, time, urllib.request, urllib.error, importlib.util
from xml.sax.saxutils import escape

WS = "/home/mantapman/.hermes/profiles/atun/workspace"
spec = importlib.util.spec_from_file_location("cfg_das", f"{WS}/cfg_das.py")
m = importlib.util.module_from_spec(spec); spec.loader.exec_module(m)
CFG = m.CFG
DOM, USER, PASS = CFG["domain"], CFG["user"], CFG["pass"]
AUTH = base64.b64encode(f"{USER}:{PASS}".encode()).decode()
UA = {"User-Agent": "Mozilla/5.0"}
UJI = 6932          # artikel Innova yang jadi calon kalah
MENANG = "https://dasindonesia.com/das-bengkel-spesialis-innova/"


def rpc(method, params=()):
    def val(v):
        if isinstance(v, int):
            return f"<value><int>{v}</int></value>"
        if isinstance(v, bool):
            return f"<value><boolean>{1 if v else 0}</boolean></value>"
        if isinstance(v, list):
            return "<value><array><data>" + "".join(val(x) for x in v) + "</data></array></value>"
        if isinstance(v, dict):
            return "<value><struct>" + "".join(f"<member><name>{escape(str(k))}</name>{val(x)}</member>"
                                               for k, x in v.items()) + "</struct></value>"
        return f"<value><string>{escape(str(v))}</string></value>"
    body = ("<?xml version=\"1.0\"?><methodCall><methodName>" + method + "</methodName><params>" +
            "".join(f"<param>{val(p)}</param>" for p in params) + "</params></methodCall>").encode()
    r = urllib.request.Request(f"https://www.{DOM}/xmlrpc.php", data=body,
                               headers={"Content-Type": "text/xml", "User-Agent": "Mozilla/5.0",
                                        "Authorization": f"Basic {AUTH}"}, method="POST")
    try:
        with urllib.request.urlopen(r, timeout=180) as x:
            return x.read().decode("utf-8", "replace")
    except Exception as e:
        return f"ERR {str(e)[:100]}"


def baca_meta(pid):
    isi = rpc("wp.getPost", [1, USER, PASS, pid])
    hasil = {}
    for mm in re.finditer(r"<member><name>key</name><value><string>(_[^<]+)</string></value></member>\s*"
                          r"<member><name>value</name><value><string>(.*?)</string>", isi, re.S):
        hasil.setdefault(mm.group(1), html.unescape(mm.group(2)))
    return hasil


print("=== sebelum ===")
m0 = baca_meta(UJI)
for k in ("_yoast_wpseo_canonical", "_yoast_wpseo_meta-robots-noindex", "_yoast_wpseo_focuskw"):
    print(f"   {k:<36} {m0.get(k) or '(kosong)'}")

print("\n=== tulis kanonik + noindex ===")
isi = rpc("wp.editPost", [1, USER, PASS, UJI, {"custom_fields": [
    {"key": "_yoast_wpseo_canonical", "value": MENANG},
    {"key": "_yoast_wpseo_meta-robots-noindex", "value": "1"}]}])
print("   balasan:", re.sub(r"\s+", " ", isi)[:180])

print("\n=== sesudah (baca dari API) ===")
m1 = baca_meta(UJI)
for k in ("_yoast_wpseo_canonical", "_yoast_wpseo_meta-robots-noindex"):
    print(f"   {k:<36} {m1.get(k) or '(kosong)'}")

print("\n=== cek halaman live ===")
try:
    h = urllib.request.urlopen(urllib.request.Request("https://dasindonesia.com/8-layanan-lengkap-das-indonesia-untuk-innova-fortuner-di-surabaya/?v=%d" % int(time.time()), headers=UA), timeout=120).read().decode("utf-8", "replace")
    rob = re.search(r'<meta name="robots" content="([^"]+)"', h)
    can = re.findall(r'<link rel="canonical" href="([^"]+)"', h)
    print("   meta robots :", rob.group(1) if rob else "(tidak ada)")
    print("   canonical   :", can)
except Exception as e:
    print("   gagal:", str(e)[:80])