#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""干净验证：报表页点下载 -> 捕获CDN直链 -> 页面内同步XHR取字节 -> base64回传 -> 落盘。"""
import json, urllib.request, base64, time, os

D = "http://127.0.0.1:10086/command"; S = "2haohr"

def call(a, args, timeout=120):
    b = json.dumps({"action": a, "args": args, "session": S}).encode()
    r = urllib.request.Request(D, data=b, headers={"Content-Type": "application/json"})
    return json.loads(urllib.request.urlopen(r, timeout=timeout).read().decode())

def ev(code):
    return call("evaluate", {"code": code}).get("data", {}).get("value")

# 单一、幂等的 hook（navigate 后页面重载，hook 会被清）
HOOK = r"""(()=>{
  if(window.__H)return 'already';
  window.__H=1; window.__cap=[];
  const ac=HTMLAnchorElement.prototype.click;
  HTMLAnchorElement.prototype.click=function(){
    try{const h=this.href||'';const d=this.getAttribute('download')||'';
      if(h && (/prv-cdn|attname=|\.xlsx|\.xls|\.docx|\.doc|\.zip|\.pdf/i.test(h)||d))
        window.__cap.push(h);}catch(e){}
    return ac.apply(this,arguments);
  };
  return 'hooked';
})()"""

CLICK_DL = r"""(()=>{const cand=['直接下载','下载','生成excel','导出Excel','导出'];
  for(const t of cand){const b=[...document.querySelectorAll('button,a,[role=button]')]
    .find(x=>(x.innerText||'').trim()===t && x.getBoundingClientRect().width>0);
    if(b){b.click();return t;}}return 'no-btn';})()"""

# 弹窗里确认导出（选择部门后 生成excel/确定/下载）
CONFIRM = r"""(()=>{const cand=['生成excel','确定','下载','导出','确认'];
  const ms=[...document.querySelectorAll('[class*=modal],[class*=Modal],[role=dialog],[class*=dialog],[class*=drawer]')];
  for(const m of ms){const r=m.getBoundingClientRect();
    if(r.width>200&&r.height>100&&getComputedStyle(m).display!=='none'){
      for(const t of cand){const b=[...m.querySelectorAll('button,a')]
        .find(x=>(x.innerText||'').trim()===t);
        if(b){b.click();return 'modal:'+t;}}}}return 'no-modal';})()"""

FETCH_B64 = r"""(()=>{try{
  if(!window.__cap.length)return 'NO_HREF';
  const u=window.__cap[window.__cap.length-1];
  const x=new XMLHttpRequest();x.open('GET',u,false);
  x.overrideMimeType('text/plain;charset=x-user-defined');x.send();
  if(x.status!==200)return 'HTTP'+x.status;
  const s=x.responseText;let out='';const CH=8192;
  for(let i=0;i<s.length;i+=CH){let c='';const e=Math.min(i+CH,s.length);
    for(let j=i;j<e;j++)c+=String.fromCharCode(s.charCodeAt(j)&0xff);out+=c;}
  window.__b64=btoa(out);
  const m=u.match(/attname=([^&]+)/);let name=m?decodeURIComponent(m[1]):'file';
  return JSON.stringify({size:s.length,name:name,url:u.slice(0,60)});
}catch(e){return 'EX '+e.message;}})()"""

def main():
    call("find_tab", {"url": "2haohr.com"})
    call("navigate", {"url": "https://2haohr.com/personnel/employee/personnel-report/home", "newTab": False})
    # 轮询等下载按钮出现
    COUNT = r"""(()=>{let n=0;document.querySelectorAll('button,a,[role=button]').forEach(x=>{const t=(x.innerText||'').trim();if(['直接下载','下载','生成excel','导出Excel','导出'].includes(t)&&x.getBoundingClientRect().width>0)n++;});return n;})()"""
    for i in range(15):
        time.sleep(2)
        c = ev(COUNT)
        print("  wait btn count=", c)
        if isinstance(c, int) and c > 0:
            break
    print("hook:", ev(HOOK))
    print("click:", ev(CLICK_DL))
    time.sleep(2.5)
    print("confirm:", ev(CONFIRM))
    time.sleep(2.5)
    meta = ev(FETCH_B64)
    print("fetch:", meta)
    if isinstance(meta, str) and meta.startswith("{"):
        info = json.loads(meta)
        b64 = ev("window.__b64")
        data = base64.b64decode(b64)
        os.makedirs("downloads/_test", exist_ok=True)
        out = "downloads/_test/probe_" + info["name"]
        open(out, "wb").write(data)
        print("落盘:", out, len(data), "bytes")

if __name__ == "__main__":
    main()