blob: adcc92fcb32c66c698687f12355898f5e2952b40 [file] [log] [blame]
#!/usr/bin/env python
# utils/ - Apply edits from .remap files -*- python -*-
# This source file is part of the open source project
# Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
# Licensed under Apache License v2.0 with Runtime Library Exception
# See for license information
# See for the list of Swift project authors
from __future__ import print_function
import argparse
import collections
import json
import os
import sys
def find_remap_files(path):
for root, dirs, files in os.walk(path):
for filename in files:
if not filename.endswith(".remap"):
yield os.path.join(root, filename)
def apply_edits(path):
remap_files = find_remap_files(path)
if not remap_files:
print("No remap files found")
return 1
edits_per_file = collections.defaultdict(list)
for remap_file in remap_files:
with open(remap_file) as f:
json_data =
json_data = json_data.replace(",\n }", "\n }")
json_data = json_data.replace(",\n]", "\n]")
curr_edits = json.loads(json_data)
for ed in curr_edits:
fname = ed["file"]
offset = ed["offset"]
length = ed.get("remove", 0)
text = ed.get("text", "")
edits_per_file[fname].append((offset, length, text))
for fname, edits in edits_per_file.items():
print('Updating', fname)
with open(fname) as f:
file_data =
for ed in edits:
offset, length, text = ed
file_data = file_data[:offset] + str(text) + \
file_data[offset + length:]
with open(fname, 'w') as f:
return 0
def main():
parser = argparse.ArgumentParser(
description="""Finds all .remap files in a directory and applies their
edits to the source files.""")
help="path to index info")
args = parser.parse_args()
return apply_edits(args.build_dir_path)
if __name__ == "__main__":