replace-webkit-additions-includes.py [plain text]
import os
import re
import sys
def read_content_from_webkit_additions(built_products_directory, sdk_root_directory, filename):
additions_path = os.path.join("usr/local/include/WebKitAdditions", filename)
try:
file_in_build_directory = open(os.path.join(built_products_directory, additions_path), "r")
return file_in_build_directory.read()
except:
try:
file_in_sdk_root = open(os.path.join(sdk_root_directory, additions_path), "r")
return file_in_sdk_root.read()
except:
return ""
def main(argv=None):
if not argv:
argv = sys.argv
if len(argv) != 4:
print("Usage: replace-webkit-additions-includes.py <header_path> <built_products_directory> <sdk_root_directory>")
return 1
header_path = argv[1]
built_products_directory = argv[2]
sdk_root_directory = argv[3]
if not len(header_path):
print("(%s): header path unspecified" % argv[0])
return 1
if not len(built_products_directory):
print("(%s): built products directory unspecified" % argv[0])
return 1
if not len(sdk_root_directory):
print("(%s): SDK root directory unspecified" % argv[0])
return 1
additions_import_pattern = re.compile(r"\#if USE\(APPLE_INTERNAL_SDK\)\n#import <WebKitAdditions/(.*)>\n#endif")
try:
with open(header_path, "r") as header:
header_contents = header.read()
match = additions_import_pattern.search(header_contents)
while match:
header_contents = header_contents[:match.start()] + read_content_from_webkit_additions(built_products_directory, sdk_root_directory, match.groups()[0]) + header_contents[match.end():]
match = additions_import_pattern.search(header_contents)
try:
with open(header_path, "w") as header:
header.write(header_contents)
except:
print("(%s): failed to write to file: %s" % (argv[0], header_path))
return 1
return 0
except:
print("(%s): failed to read file: %s" % (argv[0], header_path))
return 1
if __name__ == "__main__":
sys.exit(main(sys.argv))