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
|
load("@bazel-erlang//:bazel_erlang_lib.bzl", "ErlangLibInfo", "path_join")
RabbitmqHomeInfo = provider(
doc = "An assembled RABBITMQ_HOME dir",
fields = {
'sbin': 'Files making up the sbin dir',
'escript': 'Files making up the escript dir',
'plugins': 'Files making up the plugins dir',
'erlang_version': 'Version of the Erlang compiler used',
},
)
def _copy_script(ctx, script):
dest = ctx.actions.declare_file(path_join(ctx.label.name, "sbin", script.basename))
args = ctx.actions.args()
args.add_all([script, dest])
ctx.actions.run(
inputs = [script],
outputs = [dest],
executable = "cp",
arguments = [args],
)
return dest
def _link_escript(ctx, escript):
e = escript.files_to_run.executable
s = ctx.actions.declare_file(path_join(ctx.label.name, "escript", e.basename))
ctx.actions.symlink(
output = s,
target_file = e,
)
return s
def _plugins_dir_links(ctx, plugin):
lib_info = plugin[ErlangLibInfo]
plugin_path = path_join(
ctx.label.name,
"plugins",
"{}-{}".format(lib_info.lib_name, lib_info.lib_version),
)
links = []
for f in lib_info.include:
o = ctx.actions.declare_file(path_join(plugin_path, "include", f.basename))
ctx.actions.symlink(
output = o,
target_file = f
)
links.append(o)
for f in lib_info.beam:
o = ctx.actions.declare_file(path_join(plugin_path, "ebin", f.basename))
ctx.actions.symlink(
output = o,
target_file = f
)
links.append(o)
for f in lib_info.priv:
p = f.short_path.replace(plugin.label.package + "/", "")
o = ctx.actions.declare_file(path_join(plugin_path, p))
ctx.actions.symlink(
output = o,
target_file = f
)
links.append(o)
return links
def _unique_versions(plugins):
erlang_versions = []
for plugin in plugins:
erlang_version = plugin[ErlangLibInfo].erlang_version
if not erlang_version in erlang_versions:
erlang_versions.append(erlang_version)
return erlang_versions
def _flatten(list_of_lists):
return [item for sublist in list_of_lists for item in sublist]
def _impl(ctx):
erlang_versions = _unique_versions(ctx.attr.plugins)
if len(erlang_versions) > 1:
fail("plugins do not have a unified erlang version", erlang_versions)
scripts = [_copy_script(ctx, script) for script in ctx.files._scripts]
escripts = [_link_escript(ctx, escript) for escript in ctx.attr.escripts]
plugins = _flatten([_plugins_dir_links(ctx, plugin) for plugin in ctx.attr.plugins])
return [
RabbitmqHomeInfo(
sbin = scripts,
escript = escripts,
plugins = plugins,
erlang_version = erlang_versions[0],
),
DefaultInfo(
files = depset(scripts + escripts + plugins),
),
]
rabbitmq_home = rule(
implementation = _impl,
attrs = {
"_scripts": attr.label_list(
default = [
"//deps/rabbit:scripts/rabbitmq-env",
"//deps/rabbit:scripts/rabbitmq-defaults",
"//deps/rabbit:scripts/rabbitmq-server",
"//deps/rabbit:scripts/rabbitmqctl",
],
allow_files = True,
),
"_erlang_version": attr.label(default = "@bazel-erlang//:erlang_version"),
"escripts": attr.label_list(),
# Maybe we should not have to declare the deps here that rabbit/rabbit_common declare
"plugins": attr.label_list(),
},
)
|
