StringsExtract/UpdateLineAfterUpdated.py at main · UTCLC/StringsExtract · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import os
import re
import json
from difflib import SequenceMatcher

STRING_PATTERN = re.compile(r'"((?:\\"|\\\\|\\[^"]|[^"\\])*)"')

def calculate_line_mapping(original_lines, modified_lines):
    matcher = SequenceMatcher(None, original_lines, modified_lines)
    line_map = {}
    for op in matcher.get_opcodes():
        tag, i1, i2, j1, j2 = op
        if tag == 'equal':
            for delta in range(i2 - i1):
                line_map[i1 + delta] = j1 + delta
    return line_map

def write(original, after, jsonn, output):
    with open(jsonn, mode="r", encoding="utf-8") as f:
        old_mappings = json.load(f)

    print(f"\nProcessing file: {os.path.basename(after)}")

    with open(original, "r", encoding="utf-8") as f:
        orig_lines = f.readlines()
    with open(after, "r", encoding="utf-8") as f:
        after_lines = f.readlines()

    line_mapping = calculate_line_mapping(orig_lines, after_lines)
    new_mappings = {}

    for old_key, value in old_mappings.items():
        try:
            filename, orig_lineno, str_index = old_key.split(":")
            orig_lineno = int(orig_lineno)
            str_index = int(str_index)
        except:
            print(f"Invalid key format: {old_key}")
            continue

        if orig_lineno >= len(orig_lines):
            print(f"Original line number out of range: {old_key}")
            continue

        new_lineno = line_mapping.get(orig_lineno)
        if new_lineno is None:
            print(f"No matching line found for: {old_key}")
            continue

        try:
            orig_line = orig_lines[orig_lineno].strip()
            new_line = after_lines[new_lineno].strip()
        except IndexError:
            print(f"Line number out of bounds: {old_key}")
            continue

        orig_strings = STRING_PATTERN.findall(orig_line)
        if str_index >= len(orig_strings):
            print(f"String index out of range: {old_key}")
            continue

        target_str = orig_strings[str_index]
        print(f"Found string '{target_str}' at original line {orig_lineno} (mapped to new line {new_lineno})")

        new_strings = STRING_PATTERN.findall(new_line)
        if str_index >= len(new_strings):
            print(f"New line has fewer strings: {old_key}")
            continue

        if new_strings[str_index] != target_str:
            print(f"String content mismatch: {old_key}")
            continue

        new_key = f"{filename}:{new_lineno}:{str_index}"
        new_mappings[new_key] = value
        print(f"Mapped '{target_str}' to new position: {new_key}")

    os.makedirs(os.path.dirname(output), exist_ok=True)
    with open(output, "w", encoding="utf-8") as f:
        f.write(json.dumps(new_mappings, indent=4, separators=(',', ': '), ensure_ascii=False))
    print(f"\nSuccessfully generated: {output}")

original = input("Original file: ").replace("\\", "/")
after = input("Current file: ").replace("\\", "/")
jsonn = input("JSON file: ").replace("\\", "/")
output = input("Output file (default in Repacked folder next to JSON file): ").replace("\\", "/")
if not output:
    output = os.path.join(os.path.dirname(jsonn), "Repacked", os.path.basename(jsonn))

write(original, after, jsonn, output)