@harmonyos-arkts/d2h 0.1.5 → 0.1.7

package/skills/d2h-hypium-forge/scripts/probe.py

@@ -0,0 +1,197 @@
+#!/usr/bin/env python3
+"""probe.py — 单设备快速取证探针（替代手写 UiDriver 内联脚本）。
+
+背景（Grit 实证）：测试阶段 69 次手工探针 + 35 次单例调试里，agent 反复手写
+`python3 - <<EOF from hypium import ...` 取证——每次 10-20 行代码生成 + connect 样板，
+占探针时间的一半以上。本工具把高频取证动作固化为一条命令，输出紧凑可直接读。
+
+用法（--device 缺省取 env FORGE_DEVICE，再缺省 127.0.0.1:5555）:
+  python3 probe.py texts                    # 当前页可见 Text/Button 文案
+  python3 probe.py ids                      # 当前页全部 key/id
+  python3 probe.py find <id>                # 定位: 属性+bounds+可见性
+  python3 probe.py tree [--grep x] [--depth 6]   # 紧凑组件树（含 type/key/text/bounds）
+  python3 probe.py click <id|text>          # 点击: id 优先，文本兜底(uinput 坐标, Toggle 实证可靠)
+  python3 probe.py tap <x> <y>              # 原始坐标点击 (uinput -T -c)
+  python3 probe.py shot [out.jpeg]          # 截图
+  python3 probe.py hilog [tag] [n]          # hilog 过滤 (缺省 JSAPP/400)
+  python3 probe.py db-pull <package>        # 拉取应用 rdb 目录到 ./db_probe/
+  python3 probe.py db-sql <db> <sql>        # 本地 sqlite 查询（db-pull 之后）
+
+除 click/shot 外全部基于单次 uitest dumpLayout 本地解析（1 次 RPC 取全量证据）。
+"""
+import argparse
+import json
+import os
+import re
+import subprocess
+import sys
+import tempfile
+
+LAYOUT_REMOTE = '/data/local/tmp/_probe_layout.json'
+
+
+def hdc_bin():
+    return os.environ.get('FORGE_HDC', 'hdc')
+
+
+def hdc(dev, args, timeout=25):
+    r = subprocess.run([hdc_bin(), '-t', dev] + args,
+                       capture_output=True, timeout=timeout)
+    return (r.stdout + r.stderr).decode('utf-8', 'ignore')
+
+
+def dump_layout(dev, timeout=25):
+    """单次 RPC: uitest dumpLayout → 本地 JSON dict（失败返回 {}）。"""
+    local = os.path.join(tempfile.gettempdir(), '_probe_layout.json')
+    hdc(dev, ['shell', 'uitest', 'dumpLayout', '-p', LAYOUT_REMOTE], timeout)
+    hdc(dev, ['file', 'recv', LAYOUT_REMOTE, local], timeout)
+    try:
+        return json.load(open(local, encoding='utf-8'))
+    except Exception:
+        return {}
+
+
+def walk(node):
+    yield node
+    for ch in node.get('children') or []:
+        yield from walk(ch)
+
+
+def attrs(n):
+    a = n.get('attributes') or {}
+    return a.get('type', '?'), a.get('key', '') or '', a.get('text', '') or '', a.get('bounds', '')
+
+
+def cmd_texts(dev, _a):
+    seen, out = set(), []
+    for n in walk(dump_layout(dev)):
+        t, k, txt, b = attrs(n)
+        if txt and txt not in seen:
+            seen.add(txt)
+            out.append(f'{txt}  [{b}]')
+    print('\n'.join(out[:80]) or '(无文本)')
+
+
+def cmd_ids(dev, _a):
+    out = []
+    for n in walk(dump_layout(dev)):
+        t, k, txt, b = attrs(n)
+        if k:
+            out.append(f'{k}  ({t}{"/" + txt if txt else ""})  [{b}]')
+    print('\n'.join(out[:120]) or '(无 id)')
+
+
+def cmd_find(dev, a):
+    target = a.target
+    for n in walk(dump_layout(dev)):
+        t, k, txt, b = attrs(n)
+        if k == target or txt == target:
+            print(f'type={t} key={k} text={txt} bounds={b} visible={bool(b)}')
+            return
+    print(f'not found: {target}')
+
+
+def cmd_tree(dev, a):
+    root = dump_layout(dev)
+    lines = []
+
+    def rec(n, depth):
+        if depth > a.depth:
+            return
+        t, k, txt, b = attrs(n)
+        tag = t + (f'#{k}' if k else '') + (f' "{txt[:14]}"' if txt else '')
+        if (not a.grep or a.grep in k or a.grep in txt or a.grep in t):
+            lines.append('  ' * depth + f'{tag}  {b}')
+        for ch in n.get('children') or []:
+            rec(ch, depth + 1)
+
+    rec(root, 0)
+    print('\n'.join(lines[:200]) or '(空树)')
+
+
+def _bounds_center(b):
+    # uitest 布局 bounds 兼容两种格式: '[l,t][r,b]'（标准）与 '[l,t,r,b]'
+    nums = [int(x) for x in re.findall(r'-?\d+', str(b))]
+    if len(nums) != 4:
+        return None
+    l, t, r, bt = nums
+    return (l + r) // 2, (t + bt) // 2
+
+
+def cmd_click(dev, a):
+    # 1) id 命中 → 取中心坐标（uinput 通道，Toggle 等 hypium touch 失效场景实证可靠）
+    for n in walk(dump_layout(dev)):
+        t, k, txt, b = attrs(n)
+        if k == a.target or txt == a.target:
+            c = _bounds_center(b)
+            if c:
+                print(hdc(dev, ['shell', 'uinput', '-T', '-c', str(c[0]), str(c[1])]).strip() or
+                      f'clicked {a.target} @ {c}')
+                return
+    print(f'not found: {a.target}')
+
+
+def cmd_tap(dev, a):
+    print(hdc(dev, ['shell', 'uinput', '-T', '-c', str(a.x), str(a.y)]).strip() or f'tapped {a.x},{a.y}')
+
+
+def cmd_shot(dev, a):
+    out = a.path or 'probe_shot.jpeg'
+    try:
+        from hypium import UiDriver
+        UiDriver.connect(device_sn=dev).screenshot(out)
+        print(out)
+        return
+    except Exception:
+        pass
+    remote = '/data/local/tmp/_probe_shot.jpeg'
+    hdc(dev, ['shell', 'snapshot_display', '-f', remote])
+    hdc(dev, ['file', 'recv', remote, out])
+    hdc(dev, ['shell', 'rm', '-f', remote])
+    print(out)
+
+
+def cmd_hilog(dev, a):
+    print(hdc(dev, ['shell', 'hilog', '-z', str(a.n), '-T', a.tag])[:8000])
+
+
+def cmd_db_pull(dev, a):
+    outdir = 'db_probe'
+    os.makedirs(outdir, exist_ok=True)
+    listing = hdc(dev, ['shell', f'ls /data/app/el2/*/database/{a.package}/entry/rdb/ 2>/dev/null'])
+    got = []
+    for fn in [x.strip() for x in listing.splitlines() if x.strip().endswith(('.db', '.db-wal', '.db-shm'))]:
+        hdc(dev, ['file', 'recv', f'/data/app/el2/100/database/{a.package}/entry/rdb/{fn}',
+                  os.path.join(outdir, fn)])
+        got.append(fn)
+    print(f'pulled {len(got)} files -> {outdir}/: {" ".join(got) or "(空,检查包名)" }')
+
+
+def cmd_db_sql(_dev, a):
+    r = subprocess.run(['sqlite3', a.db, a.sql], capture_output=True, text=True)
+    print(r.stdout or r.stderr)
+
+
+def main():
+    ap = argparse.ArgumentParser(add_help=True)
+    ap.add_argument('--device', default=os.environ.get('FORGE_DEVICE', '127.0.0.1:5555'))
+    sub = ap.add_subparsers(dest='cmd', required=True)
+    sub.add_parser('texts')
+    sub.add_parser('ids')
+    p = sub.add_parser('find'); p.add_argument('target')
+    p = sub.add_parser('tree'); p.add_argument('--grep'); p.add_argument('--depth', type=int, default=6)
+    p = sub.add_parser('click'); p.add_argument('target')
+    p = sub.add_parser('tap'); p.add_argument('x', type=int); p.add_argument('y', type=int)
+    p = sub.add_parser('shot'); p.add_argument('path', nargs='?')
+    p = sub.add_parser('hilog'); p.add_argument('tag', nargs='?', default='JSAPP'); p.add_argument('n', nargs='?', type=int, default=400)
+    p = sub.add_parser('db-pull'); p.add_argument('package')
+    p = sub.add_parser('db-sql'); p.add_argument('db'); p.add_argument('sql')
+    a = ap.parse_args()
+    dev = a.device
+    {'texts': cmd_texts, 'ids': cmd_ids, 'find': cmd_find, 'tree': cmd_tree,
+     'click': cmd_click, 'tap': cmd_tap, 'shot': cmd_shot, 'hilog': cmd_hilog,
+     'db-pull': cmd_db_pull, 'db-sql': cmd_db_sql}[a.cmd](dev, a)
+
+
+if __name__ == '__main__':
+    main()