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
|
# Copyright 2022 Google LLC. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the License);
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""kt_analysis"""
load("//:visibility.bzl", "RULES_KOTLIN")
visibility(RULES_KOTLIN)
def _get_action(actions, mnemonic):
"""Get a specific action
Args:
actions: [List[Action]]
mnemonic: [string] Identify the action whose args to search
Returns:
[Action|None] The arg value, or None if it couldn't be found
"""
menmonic_actions = [a for a in actions if a.mnemonic == mnemonic]
if len(menmonic_actions) == 0:
return None
elif len(menmonic_actions) > 1:
fail("Expected a single '%s' action" % mnemonic)
return menmonic_actions[0]
def _get_all_args(action, arg_name, style = "trim"):
"""Gets values for all instances of an arg name from a specific action.
Args:
action: [Action|None]
arg_name: [string]
style: ["trim"|"next"|"list"] The style of commandline arg
Returns:
[list[string]|list[list[string]]|None] The list of matching arg values
"""
if not action:
return []
args = action.argv
matches = [(i, a) for (i, a) in enumerate(args) if a.startswith(arg_name)]
result = []
for index, arg in matches:
if style == "trim":
result.append(arg[len(arg_name):])
elif style == "next":
result.append(args[index + 1])
elif style == "list":
sub_result = []
for i in range(index + 1, len(args)):
if args[i].startswith("--"):
break
sub_result.append(args[i])
result.append(sub_result)
else:
fail("Unrecognized arg style '%s" % style)
return result
def _get_arg(action, arg_name, style = "trim"):
"""Gets values for exactly one instance of an arg name from a specific action.
Args:
action: [Action|None]
arg_name: [string]
style: ["trim"|"next"|"list"] The style of commandline arg
Returns:
[string|list[string]|None] The arg value, or None if it couldn't be found
"""
results = _get_all_args(action, arg_name, style)
if len(results) == 0:
return None
elif len(results) == 1:
return results[0]
else:
fail("Expected a single '%s' arg" % arg_name)
def _check_endswith_test(ctx):
name = ctx.label.name
for i in range(0, 10):
# TODO: Remove support for suffix digits
if name.endswith(str(i)):
name = name.removesuffix(str(i))
break
if name.endswith("_test"):
return
fail("Analysis test names must end in '_test'")
kt_analysis = struct(
# go/keep-sorted start
DEFAULT_LIST = ["__default__"],
check_endswith_test = _check_endswith_test,
get_action = _get_action,
get_all_args = _get_all_args,
get_arg = _get_arg,
# go/keep-sorted end
)
|
