@whyshu/argent-engine 0.0.0-stage → 0.6.0

package/engine/scripts/wenyao_recon.py

@@ -0,0 +1,230 @@
+#!/usr/bin/env python3
+"""wenyao_recon.py — 登录态下的**只读**界面侦察（2026-09-10）
+
+安全边界（务必遵守）：
+  · **只点开弹窗看结构，绝不提交**（不选文件、不点「确定/导入」、不点导出/取消订单）；
+  · 打开弹窗后一律用「取消 / × / Esc」关闭；
+  · 不下载任何文件（下载会改对方系统的导出任务记录）。
+
+用途：为上海文鳐 workflow 的 ④⑤⑥⑦ 步找稳定选择器，并找《上海文鳐发票模版》的下载入口
+（用户提示：登录后在系统里可能能下到模板 —— 通常在「导入订单」弹窗里）。
+
+用法（需先 bash scripts/wenyao-browser.sh start 且已登录）：
+  python3 scripts/wenyao_recon.py --toolbar        # 工具栏按钮选择器 + 状态页签
+  python3 scripts/wenyao_recon.py --import-dialog  # 点开「导入订单」看弹窗（含模板下载链接）后关闭
+  python3 scripts/wenyao_recon.py --menus          # 顶部菜单/个人中心条目
+"""
+import argparse
+import json
+import os
+import sys
+from pathlib import Path
+from playwright.sync_api import sync_playwright
+
+HOME = Path.home()
+# 数据目录：**唯一判据**在 `scripts/lib/argent_datadir.py`（三级链，与 JS 内核同语义）
+sys.path.insert(0, str(Path(__file__).resolve().parent / 'lib'))
+from argent_datadir import resolve_data_dir  # noqa: E402  —— 勿在此复制判据
+DATA = Path(resolve_data_dir()['dir'])
+OUT = DATA / "wenyao-probe"
+CDP = os.environ.get("WENYAO_CDP", "http://127.0.0.1:9333")
+
+
+def shot(page, name):
+    OUT.mkdir(parents=True, exist_ok=True)
+    try:
+        page.screenshot(path=str(OUT / f"{name}.png"), full_page=True)
+    except Exception:
+        pass
+
+
+def dump_buttons(page):
+    js = """() => Array.from(document.querySelectorAll('button,a,[role=button],.next-btn')).map(e => {
+      const r = e.getBoundingClientRect();
+      return {t:(e.innerText||'').trim().slice(0,20), cls:(e.className&&e.className.toString()||'').slice(0,80),
+              href:e.getAttribute && (e.getAttribute('href')||''), x:Math.round(r.x), y:Math.round(r.y),
+              w:Math.round(r.width), h:Math.round(r.height), vis: !!(r.width&&r.height)};
+    }).filter(x => x.vis && x.t).slice(0,60)"""
+    try:
+        return page.evaluate(js)
+    except Exception as e:
+        return [{"err": str(e)[:120]}]
+
+
+def dump_modal(page):
+    js = """() => {
+      const out = {modals: [], links: [], inputs: [], texts: []};
+      document.querySelectorAll('.next-dialog, .next-overlay-inner, [class*=modal], [class*=dialog]').forEach(m => {
+        const r = m.getBoundingClientRect();
+        if (!r.width || !r.height) return;
+        out.modals.push({cls:(m.className||'').toString().slice(0,80), x:Math.round(r.x), y:Math.round(r.y),
+                         w:Math.round(r.width), h:Math.round(r.height), text:(m.innerText||'').replace(/\\s+/g,' ').slice(0,600)});
+        m.querySelectorAll('a').forEach(a => out.links.push({t:(a.innerText||'').trim().slice(0,40), href:a.getAttribute('href')||'', down:a.getAttribute('download')||''}));
+        m.querySelectorAll('input').forEach(i => out.inputs.push({type:i.type||'', accept:i.getAttribute('accept')||'', name:i.name||'', id:i.id||''}));
+        m.querySelectorAll('button,.next-btn').forEach(b => out.texts.push({btn:(b.innerText||'').trim().slice(0,20), cls:(b.className||'').toString().slice(0,70)}));
+      });
+      // 全页搜索"模板/模版/下载"关键字
+      out.template_hits = Array.from(document.querySelectorAll('*')).map(e => (e.innerText||'').trim())
+        .filter(t => t && t.length < 40 && /模板|模版|下载|示例|样例/.test(t)).filter((v,i,a)=>a.indexOf(v)===i).slice(0,20);
+      return out;
+    }"""
+    try:
+        return page.evaluate(js)
+    except Exception as e:
+        return {"err": str(e)[:150]}
+
+
+def dump_menu(page):
+    """导出类按钮的下拉菜单：**只 hover 不点击**（点击可能真触发导出，属写操作）。"""
+    js = """() => {
+      const out = {items: [], opened: false};
+      document.querySelectorAll('.next-menu, .next-overlay-inner, [class*=menu-popup], [role=menu]').forEach(m => {
+        const r = m.getBoundingClientRect();
+        if (!r.width || !r.height) return;
+        out.opened = true;
+        m.querySelectorAll('li,.next-menu-item,a,[role=menuitem]').forEach(li => {
+          const t = (li.innerText||'').trim().slice(0,30);
+          if (t) out.items.push({t, cls:(li.className||'').toString().slice(0,70)});
+        });
+      });
+      out.items = out.items.filter((v,i,a)=>a.findIndex(x=>x.t===v.t)===i);
+      return out;
+    }"""
+    try:
+        return page.evaluate(js)
+    except Exception as e:
+        return {"err": str(e)[:150]}
+
+
+def dump_grid(page):
+    """列表/查询栏/状态页签的结构（纯 DOM 读取）。
+
+    注意：列表是 **react-base-table 虚拟化表格**（`.BaseTable__*`），不是原生 <table>，
+    所以 tbody/tr 选择器拿不到行，必须用 `.BaseTable__row` / `.BaseTable__row-cell`。
+    """
+    js = """() => {
+      const out = {tableKind: null, tabs: [], queryFields: [], columns: [], firstRow: {cells: [], checkbox: null, detailBtn: null}};
+      out.tableKind = document.querySelector('.BaseTable__table') ? 'react-base-table' : (document.querySelector('table') ? 'html-table' : 'unknown');
+      document.querySelectorAll('[class*=tab]').forEach(e => {
+        const t = (e.innerText||'').trim(); const r = e.getBoundingClientRect();
+        if (t && r.width && t.length < 24 && e.children.length <= 3) out.tabs.push({t, cls:(e.className||'').toString().slice(0,70)});
+      });
+      out.tabs = out.tabs.filter((v,i,a)=>a.findIndex(x=>x.t===v.t)===i).slice(0,20);
+      // 查询栏：label + 最近输入框
+      document.querySelectorAll('label').forEach(l => {
+        const t = (l.innerText||'').trim(); if (!t || t.length > 16) return;
+        let box = l.closest('[class*=form-item],[class*=query],[class*=field]') || l.parentElement;
+        const inp = box ? box.querySelector('input') : null;
+        const r = (box||l).getBoundingClientRect(); if (!r.width) return;
+        out.queryFields.push({label:t, ph:inp?inp.placeholder||'':'', id:inp?inp.id||'':'',
+                              cls:box?(box.className||'').toString().slice(0,60):''});
+      });
+      out.queryFields = out.queryFields.filter((v,i,a)=>a.findIndex(x=>x.label===v.label)===i).slice(0,25);
+      // 表头 / 行（兼容两种表）
+      document.querySelectorAll('.BaseTable__header-cell, thead th, .next-table-header th').forEach(th => {
+        const t=(th.innerText||'').replace(/\\s+/g,' ').trim(); if (t) out.columns.push(t.slice(0,24));
+      });
+      out.columns = out.columns.slice(0, 20);
+      const row = document.querySelector('.BaseTable__row, tbody tr, .next-table-row');
+      if (row) {
+        const sel = row.querySelectorAll('.BaseTable__row-cell, td, .next-table-cell').length
+          ? '.BaseTable__row-cell, td, .next-table-cell' : '*';
+        row.querySelectorAll(sel).forEach(td => {
+          const t=(td.innerText||'').replace(/\\s+/g,' ').trim();
+          if (t) out.firstRow.cells.push(t.slice(0,50));
+        });
+        const cb = row.querySelector('input[type=checkbox],.next-checkbox,.BaseTable__row-checkbox');
+        out.firstRow.checkbox = cb ? {cls:(cb.className||'').toString().slice(0,70), tag:cb.tagName} : null;
+        const btn = Array.from(row.querySelectorAll('button,.next-btn')).find(b => (b.innerText||'').includes('详情'));
+        out.firstRow.detailBtn = btn ? {t:(btn.innerText||'').trim(), cls:(btn.className||'').toString().slice(0,70)} : null;
+        out.firstRow.rowCls = (row.className||'').toString().slice(0,80);
+      }
+      return out;
+    }"""
+    try:
+        return page.evaluate(js)
+    except Exception as e:
+        return {"err": str(e)[:150]}
+
+
+def main():
+    ap = argparse.ArgumentParser()
+    ap.add_argument("--toolbar", action="store_true")
+    ap.add_argument("--import-dialog", action="store_true")
+    ap.add_argument("--menus", action="store_true")
+    ap.add_argument("--grid", action="store_true", help="列表/查询栏/状态页签结构（只读）")
+    ap.add_argument("--export-menu", action="store_true", help="只看「导出进仓单」下拉（hover，不点击）")
+    a = ap.parse_args()
+    res = {}
+    with sync_playwright() as p:
+        browser = p.chromium.connect_over_cdp(CDP, timeout=8000)
+        pages = [pg for c in browser.contexts for pg in c.pages]
+        page = next((pg for pg in pages if "ffl-tech" in (pg.url or "")), pages[0])
+        res["url"] = page.url
+        if a.grid:
+            res["grid"] = dump_grid(page)
+            shot(page, "65-grid")
+        if a.export_menu:
+            el = None
+            for sel in ["button:has-text('导出进仓单')", ".next-menu-btn:has-text('导出进仓单')", "text=导出进仓单"]:
+                try:
+                    el = page.query_selector(sel)
+                    if el and el.is_visible():
+                        break
+                except Exception:
+                    el = None
+            res["export_btn_found"] = bool(el)
+            if el:
+                try:
+                    el.hover()          # ← 只 hover；绝不 click（点击可能真触发导出）
+                    page.wait_for_timeout(1200)
+                except Exception as e:
+                    res["hover_err"] = str(e)[:120]
+            res["menu"] = dump_menu(page)
+            shot(page, "66-export-menu")
+            try:
+                page.keyboard.press("Escape")
+            except Exception:
+                pass
+        if a.toolbar:
+            res["buttons"] = dump_buttons(page)
+            res["tabs"] = [b["t"] for b in res["buttons"] if b["t"] in ("全部", "待拼箱", "已完成", "待送达集货仓")]
+            shot(page, "60-toolbar")
+        if a.menus:
+            res["menus"] = dump_buttons(page)
+        if a.import_dialog:
+            opened = False
+            for sel in ["button:has-text('导入订单')", "text=导入订单"]:
+                try:
+                    el = page.query_selector(sel)
+                    if el and el.is_visible():
+                        el.click(); opened = True; break
+                except Exception:
+                    continue
+            page.wait_for_timeout(2500)
+            res["import_dialog_opened"] = opened
+            res["modal"] = dump_modal(page)
+            shot(page, "61-import-dialog")
+            # 关闭（只关不提交）
+            closed = False
+            for sel in ["button:has-text('取消')", ".next-dialog-close", "text=取消"]:
+                try:
+                    el = page.query_selector(sel)
+                    if el and el.is_visible():
+                        el.click(); closed = True; break
+                except Exception:
+                    continue
+            if not closed:
+                try:
+                    page.keyboard.press("Escape")
+                except Exception:
+                    pass
+            page.wait_for_timeout(1200)
+            res["closed"] = closed
+            res["url_after"] = page.url
+        print(json.dumps(res, ensure_ascii=False, indent=1))
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())