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 os
import argparse
import re
import sys
def list_all_files(directory, prefix='/', hidden_files=False):
result = []
if os.path.exists(directory):
for i in os.listdir(directory):
abs_path = os.path.join(directory, i)
result += (
list_all_files(os.path.join(directory, abs_path), prefix + i + '/', hidden_files)
if os.path.isdir(abs_path)
else ([prefix + i] if (hidden_files or not i.startswith('.')) else [])
)
return result
def pattern_to_regexp(p):
return (
'^'
+ ('/' if not p.startswith('**') else '')
+ re.escape(p)
.replace(r'\*\*/', '[_DIR_]')
.replace(r'\*', '[_FILE_]')
.replace('[_DIR_]', '(.*/)?')
.replace('[_FILE_]', '([^/]*)')
+ '$'
)
def resolve_java_srcs(srcdir, include_patterns, exclude_patterns, all_resources, resolve_kotlin=False):
result = {'java': [], 'not_java': [], 'kotlin': []}
include_patterns_normal, include_patterns_hidden, exclude_patterns_normal, exclude_patterns_hidden = [], [], [], []
for vis, hid, patterns in (
(include_patterns_normal, include_patterns_hidden, include_patterns),
(exclude_patterns_normal, exclude_patterns_hidden, exclude_patterns),
):
for pattern in patterns:
if (pattern if pattern.find('/') == -1 else pattern.rsplit('/', 1)[1]).startswith('.'):
hid.append(pattern)
else:
vis.append(pattern)
re_patterns = map(pattern_to_regexp, vis + hid)
if sys.platform in ('win32', 'darwin'):
re_patterns = [re.compile(i, re.IGNORECASE) for i in re_patterns]
else:
re_patterns = [re.compile(i) for i in re_patterns]
vis[:], hid[:] = re_patterns[: len(vis)], re_patterns[len(vis) :]
for inc_patterns, exc_patterns, with_hidden_files in (
(include_patterns_normal, exclude_patterns_normal, False),
(include_patterns_hidden, exclude_patterns_hidden, True),
):
for f in list_all_files(srcdir, hidden_files=with_hidden_files):
excluded = False
for exc_re in exc_patterns:
if exc_re.match(f):
excluded = True
break
if excluded:
continue
for inc_re in inc_patterns:
if inc_re.match(f):
s = os.path.normpath(f[1:])
if all_resources or not (f.endswith('.java') or f.endswith('.kt')):
result['not_java'].append(s)
elif f.endswith('.java'):
result['java'].append(os.path.join(srcdir, s))
elif f.endswith('.kt') and resolve_kotlin:
result['kotlin'].append(os.path.join(srcdir, s))
else:
result['not_java'].append(s)
break
return sorted(result['java']), sorted(result['not_java']), sorted(result['kotlin'])
def resolve_sources_and_fill_filelists(
directory,
sources_file,
resources_file,
kotlin_sources_file,
include_patterns,
exclude_patterns,
resolve_kotlin,
append,
all_resources,
):
j, r, k = resolve_java_srcs(directory, include_patterns, exclude_patterns, all_resources, resolve_kotlin)
mode = 'a' if append else 'w'
open(sources_file, mode).writelines(i + '\n' for i in j)
open(resources_file, mode).writelines(i + '\n' for i in r)
if kotlin_sources_file:
open(kotlin_sources_file, mode).writelines(i + '\n' for i in k + j)
def cli_main(argv, force_skip_source_jars=False):
parser = argparse.ArgumentParser()
parser.add_argument('-d', '--directory', required=True)
parser.add_argument('-s', '--sources-file', required=True)
parser.add_argument('-r', '--resources-file', required=True)
parser.add_argument('-k', '--kotlin-sources-file', default=None)
parser.add_argument('--append', action='store_true', default=False)
parser.add_argument('--all-resources', action='store_true', default=False)
parser.add_argument('--resolve-kotlin', action='store_true', default=False)
parser.add_argument('--include-patterns', nargs='*', default=[])
parser.add_argument('--exclude-patterns', nargs='*', default=[])
args = parser.parse_args(argv)
if force_skip_source_jars and args.all_resources:
return
resolve_sources_and_fill_filelists(**vars(args))
if __name__ == '__main__':
cli_main(sys.argv[1:])
|
