build-script: teach the script to install SwiftSyntax build.

Author: nkcsgexi

build-script.py

@@ -7,7 +7,8 @@
 import subprocess
 import sys
 import tempfile
-
+import errno
+import platform
 
 PACKAGE_DIR = os.path.dirname(os.path.realpath(__file__))
 WORKSPACE_DIR = os.path.realpath(PACKAGE_DIR + '/..')
@@ -22,6 +23,9 @@
 def printerr(message):
     print(message, file=sys.stderr)
 
+def note(message):
+    print("--- %s: note: %s" % (os.path.basename(sys.argv[0]), message))
+    sys.stdout.flush()
 
 def fatal_error(message):
     printerr(message)
@@ -286,6 +290,48 @@ def run_xctests(swift_test_exec, build_dir, release, swiftc_exec, verbose):
     return call(swiftpm_call, env=subenv, verbose=verbose) == 0
 
 
+def mkdir_p(path):
+    """
+    mkdir_p(path)
+    Create the given directory, if it does not exist.
+    """
+
+    try:
+        os.makedirs(path)
+    except OSError as e:
+        # Ignore EEXIST, which may occur during a race condition.
+        if e.errno != errno.EEXIST:
+            raise
+
+def change_id_rpath(rpath, binary):
+  if platform.system() == 'Darwin':
+      cmd = ["install_name_tool", "-id", rpath, binary]
+      note("changing id in %s: %s" % (binary, ' '.join(cmd)))
+      result = subprocess.call(cmd)
+      if result != 0:
+          fatal_error("command failed with exit status %d" % (result,))
+  else:
+      fatal_error("unable to invoke install_name_tool on this platform")
+
+def check_and_sync(file_path, install_path):
+  cmd = ["rsync", "-a", file_path, install_path]
+  note("installing %s: %s" % (os.path.basename(file_path), ' '.join(cmd)))
+  result = subprocess.check_call(cmd)
+  if result != 0:
+    fatal_error("install failed with exit status %d" % (result,))
+
+def install(build_dir, prefix):
+  dylibPath = build_dir + '/libSwiftSyntax.dylib'
+  modulePath = build_dir + '/SwiftSyntax.swiftmodule'
+  docPath = build_dir + '/SwiftSyntax.swiftdoc'
+  install_dir = prefix + '/lib/swift/swiftsyntax'
+  mkdir_p(install_dir)
+  change_id_rpath('@rpath/../swiftsyntax/libSwiftSyntax.dylib', dylibPath)
+  check_and_sync(file_path=dylibPath,install_path=install_dir)
+  check_and_sync(file_path=modulePath,install_path=install_dir)
+  check_and_sync(file_path=docPath,install_path=install_dir)
+  return
+
 ### Main
 
 def main():
@@ -324,7 +370,14 @@ def main():
                              help='''
       Insert ###sourceLocation comments in generated code for line-directive.
       ''')
-
+    basic_group.add_argument('--install', action='store_true',
+                             help='''
+      Install the build artifact to a specified toolchain directory.
+      ''')
+    basic_group.add_argument('--prefix',
+                             help='''
+      The directory to where the build artifact should be installed.
+      ''')
     testing_group = parser.add_argument_group('Testing')
     testing_group.add_argument('-t', '--test', action='store_true',
                                help='Run tests')
@@ -355,6 +408,13 @@ def main():
 
     args = parser.parse_args(sys.argv[1:])
 
+    if args.install:
+      if not args.prefix:
+        fatal_error('Must specify directory to install')
+      if not args.build_dir:
+        fatal_error('Must specify build directory to copy from')
+      install(build_dir=args.build_dir, prefix=args.prefix)
+      sys.exit(0)
 
     try:
         generate_gyb_files(verbose=args.verbose,