recipes/goma_toolchain.resources/preprocess_clang.py

#!/usr/bin/env python
# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Minimize Fuchsia clang toochains.

Usage:
  python preprocess_clang.py --src=SRC_PATH --dst=dst_PATH --platforms='linux_amd64, mac_amd64'
"""

import argparse
import os
import os.path
import sys
import shutil


def preprocess_clang(src_dir, dst_dir):
    # remove unnecessary clang libs from toolchain images.
    bin_root = os.path.join(dst_dir, "bin")
    os.makedirs(bin_root)
    # read link target of 'clang'
    src_clang = os.path.join(
        src_dir, "bin", os.readlink(os.path.join(src_dir, "bin", "clang"))
    )
    dst_clang = os.path.join(bin_root, os.path.basename(src_clang))
    shutil.copy2(src_clang, dst_clang)

    cwd = os.getcwd()
    try:
        os.chdir(bin_root)
        clang = os.path.basename(src_clang)
        os.symlink(clang, "clang")
        os.symlink(clang, "clang++")
        os.symlink(clang, "clang-cl")
    finally:
        os.chdir(cwd)
    shutil.copytree(os.path.join(src_dir, "include"), os.path.join(dst_dir, "include"))
    shutil.copy2(os.path.join(src_dir, "LICENSE"), os.path.join(dst_dir, "LICENSE"))


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("src", help="source toolchain path")
    parser.add_argument("dst", help="target toolchain path")
    parser.add_argument("--platform", action="append", help="target platform name")
    args = parser.parse_args()
    for platform in args.platform:
        preprocess_clang(
            os.path.join(args.src, platform), os.path.join(args.dst, platform)
        )


if __name__ == "__main__":
    sys.exit(main())