aboutsummaryrefslogtreecommitdiffstats
path: root/build/scripts/collect_java_srcs.py
blob: 457791b9a6ccd4931ff70547c91eb8745a886152 (plain) (blame)
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
51
import os
import sys
import contextlib
import tarfile
import zipfile


if __name__ == '__main__':
    build_root = sys.argv[1]
    root = os.path.normpath(sys.argv[2])
    dest = os.path.normpath(sys.argv[3])
    srcs = sys.argv[4:]

    for src in srcs:
        src = os.path.normpath(src)
        if src.endswith('.java') or src.endswith('.kt'): 
            src_rel_path = os.path.relpath(src, root)

            if os.path.join(root, src_rel_path) == src:
                # Inside root
                dst = os.path.join(dest, src_rel_path)

            else:
                # Outside root
                print>>sys.stderr, 'External src file "{}" is outside of srcdir {}, ignore'.format(
                    os.path.relpath(src, build_root),
                    os.path.relpath(root, build_root),
                )
                continue

            if os.path.exists(dst):
                print>>sys.stderr, 'Duplicate external src file {}, choice is undefined'.format(
                    os.path.relpath(dst, root)
                )

            else:
                destdir = os.path.dirname(dst)
                if destdir and not os.path.exists(destdir):
                    os.makedirs(destdir)
                os.rename(src, dst)

        elif src.endswith('.jsr'):
            with contextlib.closing(tarfile.open(src, 'r')) as tf:
                tf.extractall(dst)

        elif src.endswith('-sources.jar'):
            with zipfile.ZipFile(src) as zf:
                zf.extractall(dst)

        else:
            print>>sys.stderr, 'Unrecognized file type', os.path.relpath(src, build_root)