summaryrefslogtreecommitdiff
path: root/llvm/utils/gn/build/symlink_or_copy.py
blob: cbc559a6778f930260fa34b4b950320751788400 (plain)
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
#!/usr/bin/env python3

"""Symlinks, or on Windows copies, an existing file to a second location.

Overwrites the target location if it exists.
Updates the mtime on a stamp file when done."""

import argparse
import errno
import os
import sys


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument(
        "--stamp", required=True, help="name of a file whose mtime is updated on run"
    )
    parser.add_argument("source")
    parser.add_argument("output")
    args = parser.parse_args()

    # FIXME: This should not check the host platform but the target platform
    # (which needs to be passed in as an arg), for cross builds.
    if sys.platform != "win32":
        try:
            os.makedirs(os.path.dirname(args.output))
        except OSError as e:
            if e.errno != errno.EEXIST:
                raise
        try:
            os.symlink(args.source, args.output)
        except OSError as e:
            if e.errno == errno.EEXIST:
                os.remove(args.output)
                os.symlink(args.source, args.output)
            else:
                raise
    else:
        import shutil

        output = args.output + ".exe"
        source = args.source + ".exe"
        shutil.copyfile(os.path.join(os.path.dirname(output), source), output)

    open(args.stamp, "w")  # Update mtime on stamp file.


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