@harmonyos-arkts/d2h 0.0.0-stage → 0.1.1

package/skills/arkts-icon-sizing/scripts/icon_fix.py

@@ -0,0 +1,119 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""第 3/3 步 —— 把还原出的尺寸插入 .ets 源码。
+
+实时重扫（因此对行号位移健壮），对 icon_plan.json 中存在的每个欠约束静态图标：
+  - 既无 width 又无 height  -> 插入  .width(W)  .height(H)   （固定标称 vp）
+  - 恰好只有一维            -> 插入  .aspectRatio(W/H)       （保留有意的那一维，
+        如满宽横幅的 width('100%')，只修比例）
+新修饰符插入在 `Image(...)` 开头之后、与原链同缩进，从而与 `.objectFit(...)` 等干净组合。
+
+动态源图片（Image(this.foo())、三元 Image(cond ? a : b)、Builder `Image(param)`）
+不自动改 —— 只报告，便于解析底层资源、用 icon_dims.py --icons 查尺寸后手工改。
+
+安全
+  默认 dry-run（打印将要做的改动）。加 --apply 才落盘。
+  调用前记录允许修改的文件范围；需要恢复时只撤销本次输出清单中的修改，
+  不得暂存、提交或整体回滚调用方已有改动。
+
+用法
+  python icon_fix.py --plan icon_plan.json --report icon_report.json            # dry-run
+  python icon_fix.py --plan icon_plan.json --report icon_report.json --apply    # 落盘
+"""
+import os, re, json, argparse
+
+
+def chain_end(lines, i):
+    base = len(lines[i]) - len(lines[i].lstrip())
+    j = i + 1
+    while j < len(lines):
+        if lines[j].strip() == '':
+            j += 1
+            continue
+        if (len(lines[j]) - len(lines[j].lstrip())) <= base:
+            break
+        j += 1
+    return j
+
+
+def apply_fixes(plan, report, apply=True):
+    """对 report 标记的文件实时重扫，按 plan 给欠约束静态图标插入尺寸。
+
+    被 icon_fix.main() 与 icon_autofix.py 共用。
+      plan   : {icon: {w,h,...}}（icon_dims.build_plan 产出）
+      report : icon_report.json 内容（取 flagged 的文件集合作为作用域）
+      apply  : True 落盘，False 仅计算（dry-run）
+    返回 (applied, dynamic)：applied 为修复描述行列表，dynamic 为动态源待手工项。
+    """
+    files = sorted({f['file'] for f in report['flagged']})  # 仅针对被标记的文件
+    applied, dynamic = [], []
+    for path in files:
+        if not os.path.exists(path):
+            continue
+        lines = open(path, encoding='utf-8').read().split('\n')
+        edits = []
+        for i, ln in enumerate(lines):
+            st = ln.lstrip()
+            if st.startswith('*') or st.startswith('//') or st.startswith('/*'):
+                continue
+            code = ln.split('//')[0]
+            if not re.search(r'\bImage\(', code):
+                continue
+            end = chain_end(lines, i)
+            win = '\n'.join(lines[i + 1:end])
+            has_w, has_h = '.width(' in win, '.height(' in win
+            has_size, has_aspect = '.size(' in win, '.aspectRatio(' in win
+            has_weight = '.layoutWeight(' in win
+            w100, h100 = ".width('100%')" in win, ".height('100%')" in win
+            ok = has_size or (has_w and has_h) or ((has_w ^ has_h) and (has_aspect or has_weight))
+            if ok or (w100 and h100):
+                continue
+            m = re.search(r"(?:app|sys)\.media\.([A-Za-z0-9_]+)", code)
+            if not m:
+                dynamic.append(f'{path}:{i + 1}  {st[:55]}')
+                continue
+            icon = m.group(1)
+            if icon not in plan:
+                continue
+            w, h = plan[icon]['w'], plan[icon]['h']
+            ind = ' ' * ((len(lines[i + 1]) - len(lines[i + 1].lstrip())) if i + 1 < end else (len(ln) - len(st)) + 2)
+            if not has_w and not has_h:
+                ins = [f'{ind}.width({w})', f'{ind}.height({h})']
+                kind = f'w/h {w}x{h}'
+            else:
+                ratio = round(w / h, 4)
+                ins = [f'{ind}.aspectRatio({ratio})']
+                kind = f'aspectRatio {ratio} ({"W" if has_w else "H"} set)'
+            edits.append((i, ins))
+            applied.append(f'{os.path.basename(path)}:{i + 1}  {icon:34} -> {kind}')
+        if edits and apply:
+            for i, ins in sorted(edits, reverse=True):
+                lines[i + 1:i + 1] = ins
+            open(path, 'w', encoding='utf-8').write('\n'.join(lines))
+    return applied, dynamic
+
+
+def main():
+    ap = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
+    ap.add_argument('--plan', default='icon_plan.json')
+    ap.add_argument('--report', default='icon_report.json')
+    ap.add_argument('--apply', action='store_true', help='落盘（默认 dry-run）')
+    a = ap.parse_args()
+
+    plan = json.load(open(a.plan, encoding='utf-8'))
+    rep = json.load(open(a.report, encoding='utf-8'))
+    applied, dynamic = apply_fixes(plan, rep, apply=a.apply)
+
+    verb = '已落盘' if a.apply else '将要修复（dry-run）'
+    print(f'== {verb}：{len(applied)} 处 ==')
+    for s in sorted(applied):
+        print('  ' + s)
+    print(f'\n== 动态源（手工解析，见 SKILL.md）：{len(dynamic)} ==')
+    for d in dynamic:
+        print('  ' + d)
+    if not a.apply:
+        print('\n仅 dry-run。加 --apply 落盘，然后重跑 icon_audit.py + 编译。')
+
+
+if __name__ == '__main__':
+    main()