反射与序列化代码如何划定 VMP 和 keep 边界

import sys
import re
import json


def load_expected_classes(config_path):
    """Load expected reflection classes from JSON config."""
    try:
        with open(config_path, 'r', encoding='utf-8') as f:
            data = json.load(f)
        return set(data.get('reflectionClasses', []))
    except FileNotFoundError:
        print(f"Error: Config file {config_path} not found.")
        sys.exit(2)
    except json.JSONDecodeError:
        print(f"Error: Invalid JSON in {config_path}.")
        sys.exit(2)


def load_kept_classes_from_mapping(mapping_path):
    """Parse mapping.txt to find kept classes (no obfuscation)."""
    kept = set()
    try:
        with open(mapping_path, 'r', encoding='utf-8') as f:
            for line in f:
                line = line.strip()
                if not line or line.startswith('#'):
                    continue
                # Match pattern: original.name -> shortened: {
                match = re.match(r'^([\w.$]+) -> [\w]+:$', line)
                if match:
                    original_name = match.group(1)
                    kept.add(original_name)
    except FileNotFoundError:
        print(f"Error: Mapping file {mapping_path} not found.")
        sys.exit(2)
    return kept


if __name__ == '__main__':
    if len(sys.argv) != 3:
        print("Usage: check_reflection_keep.py <mapping.txt> <config.json>")
        print("  mapping.txt: R8 mapping output file")
        print("  config.json: JSON file containing 'reflectionClasses' list")
        sys.exit(1)

    mapping_file = sys.argv[1]
    config_file = sys.argv[2]

    expected_classes = load_expected_classes(config_file)
    kept_classes = load_kept_classes_from_mapping(mapping_file)

    missing_classes = expected_classes - kept_classes

    if missing_classes:
        print("\n[FAIL] The following reflection classes are NOT kept (obfuscated or removed):")
        for cls in sorted(missing_classes):
            print(f"  - {cls}")
        print("\nAction: Add precise -keep rules for these classes.")
        sys.exit(1)
    else:
        print("\n[PASS] All expected reflection classes are correctly kept.")
        sys.exit(0)