@harmonyos-arkts/d2h 0.1.11 → 0.1.13

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

@@ -1,232 +1,232 @@
-#!/usr/bin/env python3
-"""preinject_ids.py — ID 预置（铁律①的机械强制，先于脚本编写，不依赖 cases DSL）。
-
-输入 features.json（scan_features 产物），对全部 has_id=False 的交互点批量注入
-`.id('tc_<page>_<n>')`（资源文案/精确模糊文本/行号三路定位，属性链内插桩永远合法），
-输出 ids.json 映射（脚本编写只认此表与既有 ID）。注入后须 rebuild + uninstall&&install。
-
-用法:
-  python3 preinject_ids.py --features features.json [--out ids.json]
-"""
-import argparse
-import glob
-import json
-import os
-import re
-
-TEXT_PAT = re.compile(
-    r"^(\s*)(Text|Button|Toggle|Checkbox|Radio|MenuItem|Image|Row|Column)\("
-    r"[^\n]*?['\"](.*?)['\"][^\n]*\)?\s*$")
-
-
-def read(p):
-    return open(p, encoding='utf-8').read()
-
-
-def insert_after_component(lines, i):
-    depth = 0
-    started = False
-    for k in range(i, min(len(lines), i + 40)):
-        depth += lines[k].count('(') - lines[k].count(')')
-        started = True
-        if started and depth <= 0 and k > i:
-            return k + 1
-        if started and depth < 0:
-            return k
-    return min(len(lines), i + 12)
-
-
-def find_and_inject(page_file, target, seq):
-    """三路定位注入：资源文案 → 精确/模糊文本 → 行号。返回 (id or None, modified_src)。"""
-    src = read(page_file)
-    lines = src.split('\n')
-    tid = f"tc_{os.path.splitext(os.path.basename(page_file))[0].lower()}_{seq}"
-
-    if target and re.search(rf"\.id\(\s*'{re.escape(target)}'", src):
-        return target, src
-
-    if target and 'app.string.' in src:
-        modroot = page_file
-        for _ in range(4):
-            modroot = os.path.dirname(modroot)
-            cand = os.path.join(modroot, 'resources', 'base', 'element', 'string.json')
-            if os.path.exists(cand):
-                resname = None
-                try:
-                    for item in json.load(open(cand, encoding='utf-8')).get('string', []):
-                        if item.get('value') == target:
-                            resname = item.get('name')
-                            break
-                except Exception:
-                    pass
-                if resname:
-                    break
-        if resname:
-            rx = re.compile(r"[\$r]\(\s*'app\.string\." + re.escape(resname) + r"'")
-            for i, ln in enumerate(lines):
-                if rx.search(ln):
-                    for k in range(i, min(len(lines), i + 10)):
-                        if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
-                            indent = re.match(r'(\s*)', lines[k]).group(1)
-                            lines.insert(k, f"{indent}.id('{tid}')")
-                            return tid, '\n'.join(lines)
-                    indent = re.match(r'(\s*)', ln).group(1)
-                    lines.insert(i + 1, f"{indent}  .id('{tid}')")
-                    return tid, '\n'.join(lines)
-
-    if target:
-        for i, ln in enumerate(lines):
-            m = TEXT_PAT.match(ln)
-            if not m:
-                continue
-            if m.group(3) == target or (target in m.group(3) or m.group(3) in target):
-                seg = '\n'.join(lines[i:i + 25])
-                if re.search(r'\.id\(', seg):
-                    idm = re.search(r"\.id\(\s*'([^']+)'", seg)
-                    if idm:
-                        return idm.group(1), src
-                for k in range(i + 1, min(len(lines), i + 15)):
-                    if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
-                        indent = re.match(r'(\s*)', lines[k]).group(1)
-                        lines.insert(k, f"{indent}.id('{tid}')")
-                        return tid, '\n'.join(lines)
-                j = insert_after_component(lines, i)
-                lines.insert(j, f"{m.group(1)}  .id('{tid}')")
-                return tid, '\n'.join(lines)
-
-    tc = re.match(r'line(\d+):(\w+)', target or '')
-    if tc:
-        i = int(tc.group(1)) - 1
-        if 0 <= i < len(lines):
-            for k in range(i + 1, min(len(lines), i + 9)):
-                if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
-                    indent = re.match(r'(\s*)', lines[k]).group(1)
-                    lines.insert(k, f"{indent}.id('{tid}')")
-                    return tid, '\n'.join(lines)
-            for k in range(i, max(0, i - 8), -1):
-                if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
-                    indent = re.match(r'(\s*)', lines[k]).group(1)
-                    lines.insert(k + 1, f"{indent}.id('{tid}')")
-                    return tid, '\n'.join(lines)
-    return None, src
-
-
-def main():
-    ap = argparse.ArgumentParser()
-    ap.add_argument('--features', required=True)
-    ap.add_argument('--out', default=None)
-    args = ap.parse_args()
-    feats = json.load(open(args.features, encoding='utf-8'))
-    project = feats['project_dir']
-    out = args.out or os.path.join(os.path.dirname(args.features), 'ids.json')
-    mapping = {}
-    if os.path.exists(out):
-        try:
-            mapping = json.load(open(out, encoding='utf-8'))
-        except Exception:
-            mapping = {}
-    injected = existing = missed = 0
-    for pg in feats.get('pages', []):
-        fpath = pg.get('file')
-        if not fpath or not os.path.exists(fpath):
-            continue
-        seq = 0
-        for it in pg.get('interactions', []):
-            target = it.get('target')
-            desc = it.get('desc')
-            if not target:
-                continue
-            seq += 1
-            key = f"{pg['page']}:{desc or target}"
-            prev = mapping.get(key)
-            if prev and re.search(rf"\.id\(\s*'{re.escape(prev)}'", read(fpath)):
-                existing += 1
-                continue
-            if it.get('has_id'):
-                existing += 1
-                mapping[key] = target
-                continue
-            tid, new_src = find_and_inject(fpath, target, seq)
-            if not tid:
-                tid2, new_src2 = find_and_inject(fpath, f"line{it.get('line')}:{it.get('type')}", seq)
-                tid, new_src = tid2 or tid, new_src2 if tid2 else new_src
-            if tid:
-                disk = read(fpath)
-                if new_src != disk:
-                    open(fpath, 'w', encoding='utf-8').write(new_src)  # 逐次落盘（页尾统一写会丢同页其余注入）
-                    injected += 1
-                else:
-                    existing += 1
-                mapping[key] = tid
-            else:
-                missed += 1
-    for res_id in re.findall(r"\.id\('([a-z][a-z0-9_]+)'\)", '\n'.join(
-            read(p) for p in glob.glob(os.path.join(project, '**', '*.ets'), recursive=True)
-            if 'oh_modules' not in p))[:0]:
-        pass
-    # 连续 .id() 消除（TD-006）：注入点组件已有语义 id 时，删除本轮新插的机械 id，
-    # mapping 改指向已有 id——运行时链式 .id() 后者覆盖前者，机械 id 失效且污染源码
-    import glob as _g
-    fixed = 0
-    for pg in feats.get('pages', []):
-        fp = pg.get('file')
-        if not fp or not os.path.exists(fp):
-            continue
-        try:
-            src = open(fp, encoding='utf-8').read()
-        except Exception:
-            continue
-        new_src = src
-        for m in re.finditer(r"\.id\('(tc_[^']+)'\)\s*\.id\('([^']+)'\)", src):
-            dead, alive = m.group(1), m.group(2)
-            new_src = new_src.replace(f".id('{dead}')\n", '', 1).replace(f".id('{dead}')", '', 1)
-            for k, v in list(mapping.items()):
-                if v == dead:
-                    mapping[k] = alive
-            fixed += 1
-        if new_src != src:
-            open(fp, 'w', encoding='utf-8').write(new_src)
-    # ID 稳定性门禁（TD-013）：旧 ids.json 被用例引用的 id 若在新轮消失 → BLOCKER 阻断，
-    # 禁止静默让用例回退 text 定位（run 期间 id 漂移的机制化防线）
-    old_out = out + '.prev'
-    if os.path.exists(out):
-        import shutil as _sh
-        _sh.copy(out, old_out)
-        try:
-            old_map = json.load(open(old_out, encoding='utf-8'))
-        except Exception:
-            old_map = {}
-        gone = {k: v for k, v in old_map.items() if v not in set(mapping.values())
-                and v not in {m for m in mapping.values()}}
-        if gone:
-            print(f'✗ BLOCKER：上一轮 {len(gone)} 个 id 本轮消失（用例可能引用，禁静默回退 text）：')
-            for k, v in list(gone.items())[:6]:
-                print(f'    {k} → {v}')
-            print('   处置：核对注入冲突/页面改动，或显式重命名并同步全部用例后重跑。')
-    json.dump(mapping, open(out, 'w', encoding='utf-8'), ensure_ascii=False, indent=1)
-    print(f'ID 预置完成: 新注入 {injected} / 已有 {existing} / 未定位 {missed}'
-          f' / 连续id消除 {fixed} → {out}')
-
-    # 语法快速自检：检测连续 .id() 调用（最常见注入冲突）
-    import re as _re
-    _warn = 0
-    for pg in feats.get('pages', []):
-        fp = pg.get('file')
-        if not fp or not os.path.exists(fp):
-            continue
-        try:
-            src = open(fp, encoding='utf-8').read()
-            for m in _re.finditer(r"\.id\(\s*'[^']+'\s*\)\s*\.id\(", src):
-                print(f'  ⚠️ 语法风险: {os.path.basename(fp)} 存在连续 .id() 调用')
-                _warn += 1
-        except Exception:
-            pass
-    if _warn:
-        print(f'⚠️ 发现 {_warn} 处语法风险——建议先跑构建验证再使用 ids.json')
-    else:
-        print('✓ 语法快速检查通过')
-    print('⚠️ 注入后必须 rebuild + uninstall&&install 再执行脚本')
-
-
-if __name__ == '__main__':
-    main()
+#!/usr/bin/env python3
+"""preinject_ids.py — ID 预置（铁律①的机械强制，先于脚本编写，不依赖 cases DSL）。
+
+输入 features.json（scan_features 产物），对全部 has_id=False 的交互点批量注入
+`.id('tc_<page>_<n>')`（资源文案/精确模糊文本/行号三路定位，属性链内插桩永远合法），
+输出 ids.json 映射（脚本编写只认此表与既有 ID）。注入后须 rebuild + uninstall&&install。
+
+用法:
+  python3 preinject_ids.py --features features.json [--out ids.json]
+"""
+import argparse
+import glob
+import json
+import os
+import re
+
+TEXT_PAT = re.compile(
+    r"^(\s*)(Text|Button|Toggle|Checkbox|Radio|MenuItem|Image|Row|Column)\("
+    r"[^\n]*?['\"](.*?)['\"][^\n]*\)?\s*$")
+
+
+def read(p):
+    return open(p, encoding='utf-8').read()
+
+
+def insert_after_component(lines, i):
+    depth = 0
+    started = False
+    for k in range(i, min(len(lines), i + 40)):
+        depth += lines[k].count('(') - lines[k].count(')')
+        started = True
+        if started and depth <= 0 and k > i:
+            return k + 1
+        if started and depth < 0:
+            return k
+    return min(len(lines), i + 12)
+
+
+def find_and_inject(page_file, target, seq):
+    """三路定位注入：资源文案 → 精确/模糊文本 → 行号。返回 (id or None, modified_src)。"""
+    src = read(page_file)
+    lines = src.split('\n')
+    tid = f"tc_{os.path.splitext(os.path.basename(page_file))[0].lower()}_{seq}"
+
+    if target and re.search(rf"\.id\(\s*'{re.escape(target)}'", src):
+        return target, src
+
+    if target and 'app.string.' in src:
+        modroot = page_file
+        for _ in range(4):
+            modroot = os.path.dirname(modroot)
+            cand = os.path.join(modroot, 'resources', 'base', 'element', 'string.json')
+            if os.path.exists(cand):
+                resname = None
+                try:
+                    for item in json.load(open(cand, encoding='utf-8')).get('string', []):
+                        if item.get('value') == target:
+                            resname = item.get('name')
+                            break
+                except Exception:
+                    pass
+                if resname:
+                    break
+        if resname:
+            rx = re.compile(r"[\$r]\(\s*'app\.string\." + re.escape(resname) + r"'")
+            for i, ln in enumerate(lines):
+                if rx.search(ln):
+                    for k in range(i, min(len(lines), i + 10)):
+                        if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
+                            indent = re.match(r'(\s*)', lines[k]).group(1)
+                            lines.insert(k, f"{indent}.id('{tid}')")
+                            return tid, '\n'.join(lines)
+                    indent = re.match(r'(\s*)', ln).group(1)
+                    lines.insert(i + 1, f"{indent}  .id('{tid}')")
+                    return tid, '\n'.join(lines)
+
+    if target:
+        for i, ln in enumerate(lines):
+            m = TEXT_PAT.match(ln)
+            if not m:
+                continue
+            if m.group(3) == target or (target in m.group(3) or m.group(3) in target):
+                seg = '\n'.join(lines[i:i + 25])
+                if re.search(r'\.id\(', seg):
+                    idm = re.search(r"\.id\(\s*'([^']+)'", seg)
+                    if idm:
+                        return idm.group(1), src
+                for k in range(i + 1, min(len(lines), i + 15)):
+                    if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
+                        indent = re.match(r'(\s*)', lines[k]).group(1)
+                        lines.insert(k, f"{indent}.id('{tid}')")
+                        return tid, '\n'.join(lines)
+                j = insert_after_component(lines, i)
+                lines.insert(j, f"{m.group(1)}  .id('{tid}')")
+                return tid, '\n'.join(lines)
+
+    tc = re.match(r'line(\d+):(\w+)', target or '')
+    if tc:
+        i = int(tc.group(1)) - 1
+        if 0 <= i < len(lines):
+            for k in range(i + 1, min(len(lines), i + 9)):
+                if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
+                    indent = re.match(r'(\s*)', lines[k]).group(1)
+                    lines.insert(k, f"{indent}.id('{tid}')")
+                    return tid, '\n'.join(lines)
+            for k in range(i, max(0, i - 8), -1):
+                if re.match(r'^\s*\.[a-zA-Z]', lines[k]):
+                    indent = re.match(r'(\s*)', lines[k]).group(1)
+                    lines.insert(k + 1, f"{indent}.id('{tid}')")
+                    return tid, '\n'.join(lines)
+    return None, src
+
+
+def main():
+    ap = argparse.ArgumentParser()
+    ap.add_argument('--features', required=True)
+    ap.add_argument('--out', default=None)
+    args = ap.parse_args()
+    feats = json.load(open(args.features, encoding='utf-8'))
+    project = feats['project_dir']
+    out = args.out or os.path.join(os.path.dirname(args.features), 'ids.json')
+    mapping = {}
+    if os.path.exists(out):
+        try:
+            mapping = json.load(open(out, encoding='utf-8'))
+        except Exception:
+            mapping = {}
+    injected = existing = missed = 0
+    for pg in feats.get('pages', []):
+        fpath = pg.get('file')
+        if not fpath or not os.path.exists(fpath):
+            continue
+        seq = 0
+        for it in pg.get('interactions', []):
+            target = it.get('target')
+            desc = it.get('desc')
+            if not target:
+                continue
+            seq += 1
+            key = f"{pg['page']}:{desc or target}"
+            prev = mapping.get(key)
+            if prev and re.search(rf"\.id\(\s*'{re.escape(prev)}'", read(fpath)):
+                existing += 1
+                continue
+            if it.get('has_id'):
+                existing += 1
+                mapping[key] = target
+                continue
+            tid, new_src = find_and_inject(fpath, target, seq)
+            if not tid:
+                tid2, new_src2 = find_and_inject(fpath, f"line{it.get('line')}:{it.get('type')}", seq)
+                tid, new_src = tid2 or tid, new_src2 if tid2 else new_src
+            if tid:
+                disk = read(fpath)
+                if new_src != disk:
+                    open(fpath, 'w', encoding='utf-8').write(new_src)  # 逐次落盘（页尾统一写会丢同页其余注入）
+                    injected += 1
+                else:
+                    existing += 1
+                mapping[key] = tid
+            else:
+                missed += 1
+    for res_id in re.findall(r"\.id\('([a-z][a-z0-9_]+)'\)", '\n'.join(
+            read(p) for p in glob.glob(os.path.join(project, '**', '*.ets'), recursive=True)
+            if 'oh_modules' not in p))[:0]:
+        pass
+    # 连续 .id() 消除（TD-006）：注入点组件已有语义 id 时，删除本轮新插的机械 id，
+    # mapping 改指向已有 id——运行时链式 .id() 后者覆盖前者，机械 id 失效且污染源码
+    import glob as _g
+    fixed = 0
+    for pg in feats.get('pages', []):
+        fp = pg.get('file')
+        if not fp or not os.path.exists(fp):
+            continue
+        try:
+            src = open(fp, encoding='utf-8').read()
+        except Exception:
+            continue
+        new_src = src
+        for m in re.finditer(r"\.id\('(tc_[^']+)'\)\s*\.id\('([^']+)'\)", src):
+            dead, alive = m.group(1), m.group(2)
+            new_src = new_src.replace(f".id('{dead}')\n", '', 1).replace(f".id('{dead}')", '', 1)
+            for k, v in list(mapping.items()):
+                if v == dead:
+                    mapping[k] = alive
+            fixed += 1
+        if new_src != src:
+            open(fp, 'w', encoding='utf-8').write(new_src)
+    # ID 稳定性门禁（TD-013）：旧 ids.json 被用例引用的 id 若在新轮消失 → BLOCKER 阻断，
+    # 禁止静默让用例回退 text 定位（run 期间 id 漂移的机制化防线）
+    old_out = out + '.prev'
+    if os.path.exists(out):
+        import shutil as _sh
+        _sh.copy(out, old_out)
+        try:
+            old_map = json.load(open(old_out, encoding='utf-8'))
+        except Exception:
+            old_map = {}
+        gone = {k: v for k, v in old_map.items() if v not in set(mapping.values())
+                and v not in {m for m in mapping.values()}}
+        if gone:
+            print(f'✗ BLOCKER：上一轮 {len(gone)} 个 id 本轮消失（用例可能引用，禁静默回退 text）：')
+            for k, v in list(gone.items())[:6]:
+                print(f'    {k} → {v}')
+            print('   处置：核对注入冲突/页面改动，或显式重命名并同步全部用例后重跑。')
+    json.dump(mapping, open(out, 'w', encoding='utf-8'), ensure_ascii=False, indent=1)
+    print(f'ID 预置完成: 新注入 {injected} / 已有 {existing} / 未定位 {missed}'
+          f' / 连续id消除 {fixed} → {out}')
+
+    # 语法快速自检：检测连续 .id() 调用（最常见注入冲突）
+    import re as _re
+    _warn = 0
+    for pg in feats.get('pages', []):
+        fp = pg.get('file')
+        if not fp or not os.path.exists(fp):
+            continue
+        try:
+            src = open(fp, encoding='utf-8').read()
+            for m in _re.finditer(r"\.id\(\s*'[^']+'\s*\)\s*\.id\(", src):
+                print(f'  ⚠️ 语法风险: {os.path.basename(fp)} 存在连续 .id() 调用')
+                _warn += 1
+        except Exception:
+            pass
+    if _warn:
+        print(f'⚠️ 发现 {_warn} 处语法风险——建议先跑构建验证再使用 ids.json')
+    else:
+        print('✓ 语法快速检查通过')
+    print('⚠️ 注入后必须 rebuild + uninstall&&install 再执行脚本')
+
+
+if __name__ == '__main__':
+    main()