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
|
# 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:
[Optional[action]] 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_arg(action, arg_name, style = "trim"):
"""Get a named arg from a specific action
Args:
action: [Optional[Action]]
arg_name: [string]
style: ["trim"|"next"|"list"] The style of commandline arg
Returns:
[Optional[string]] The arg value, or None if it couldn't be found
"""
if not action:
return None
args = action.argv
matches = [(i, a) for (i, a) in enumerate(args) if a.startswith(arg_name)]
if len(matches) == 0:
return None
elif len(matches) > 1:
fail("Expected a single '%s' arg" % arg_name)
(index, arg) = matches[0]
if style == "trim":
return arg[len(arg_name):]
elif style == "next":
return args[index + 1]
elif style == "list":
result = []
for i in range(index + 1, len(args)):
if args[i].startswith("--"):
break
result.append(args[i])
return result
else:
fail("Unrecognized arg style '%s" % style)
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_arg = _get_arg,
# go/keep-sorted end
)
|
