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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
import argparse
import contextlib
from distutils import dir_util
import os
import shutil
import subprocess as sp
import tarfile
import zipfile
import sys
def parse_args(args):
parser = argparse.ArgumentParser(description='Wrapper to invoke Java compilation from ya make build')
parser.add_argument('--javac-bin', help='path to javac')
parser.add_argument('--jar-bin', help='path to jar tool')
parser.add_argument('--java-bin', help='path to java binary')
parser.add_argument('--kotlin-compiler', help='path to kotlin compiler jar file')
parser.add_argument('--vcs-mf', help='path to VCS info manifest snippet')
parser.add_argument('--package-prefix', help='package prefix for resource files')
parser.add_argument('--jar-output', help='jar file with compiled classes destination path')
parser.add_argument('--srcs-jar-output', help='jar file with sources destination path')
parser.add_argument('srcs', nargs="*")
args = parser.parse_args(args)
return args, args.srcs
def mkdir_p(directory):
if not os.path.exists(directory):
os.makedirs(directory)
def split_cmd_by_delim(cmd, delim='DELIM'):
result = [[]]
for arg in cmd:
if arg == delim:
result.append([])
else:
result[-1].append(arg)
return result
def main():
cmd_parts = split_cmd_by_delim(sys.argv[1:])
assert len(cmd_parts) == 4
args, javac_opts, peers, ktc_opts = cmd_parts
opts, jsrcs = parse_args(args)
jsrcs += list(filter(lambda x: x.endswith('.jsrc'), peers))
peers = list(filter(lambda x: not x.endswith('.jsrc'), peers))
sources_dir = 'src'
mkdir_p(sources_dir)
for s in jsrcs:
if s.endswith('.jsrc'):
with contextlib.closing(tarfile.open(s, 'r')) as tf:
tf.extractall(sources_dir)
srcs = []
for r, _, files in os.walk(sources_dir):
for f in files:
srcs.append(os.path.join(r, f))
srcs += jsrcs
ktsrcs = list(filter(lambda x: x.endswith('.kt'), srcs))
srcs = list(filter(lambda x: x.endswith('.java'), srcs))
classes_dir = 'cls'
mkdir_p(classes_dir)
classpath = os.pathsep.join(peers)
if srcs:
temp_sources_file = 'temp.sources.list'
with open(temp_sources_file, 'w') as ts:
ts.write(' '.join(srcs))
if ktsrcs:
temp_kt_sources_file = 'temp.kt.sources.list'
with open(temp_kt_sources_file, 'w') as ts:
ts.write(' '.join(ktsrcs + srcs))
kt_classes_dir = 'kt_cls'
mkdir_p(kt_classes_dir)
sp.check_call([opts.java_bin, '-jar', opts.kotlin_compiler, '-classpath', classpath, '-d', kt_classes_dir] + ktc_opts + ['@' + temp_kt_sources_file])
classpath = os.pathsep.join([kt_classes_dir, classpath])
if srcs:
sp.check_call([opts.javac_bin, '-nowarn', '-g', '-classpath', classpath, '-encoding', 'UTF-8', '-d', classes_dir] + javac_opts + ['@' + temp_sources_file])
for s in jsrcs:
if s.endswith('-sources.jar'):
with zipfile.ZipFile(s) as zf:
zf.extractall(sources_dir)
elif s.endswith('.jar'):
with zipfile.ZipFile(s) as zf:
zf.extractall(classes_dir)
if ktsrcs:
dir_util.copy_tree(kt_classes_dir, classes_dir)
if opts.vcs_mf:
sp.check_call([opts.jar_bin, 'cfm', opts.jar_output, opts.vcs_mf, os.curdir], cwd=classes_dir)
else:
sp.check_call([opts.jar_bin, 'cfM', opts.jar_output, os.curdir], cwd=classes_dir)
if opts.srcs_jar_output:
for s in jsrcs:
if s.endswith('.java'):
if opts.package_prefix:
d = os.path.join(sources_dir, *(opts.package_prefix.split('.') + [os.path.basename(s)]))
else:
d = os.path.join(sources_dir, os.path.basename(s))
shutil.copyfile(s, d)
if opts.vcs_mf:
sp.check_call([opts.jar_bin, 'cfm', opts.srcs_jar_output, opts.vcs_mf, os.curdir], cwd=sources_dir)
else:
sp.check_call([opts.jar_bin, 'cfM', opts.srcs_jar_output, os.curdir], cwd=sources_dir)
if __name__ == '__main__':
main()
|
