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
|
## This Source Code Form is subject to the terms of the Mozilla Public
## License, v. 2.0. If a copy of the MPL was not distributed with this
## file, You can obtain one at https://mozilla.org/MPL/2.0/.
##
## Copyright (c) 2007-2020 VMware, Inc. or its affiliates. All rights reserved.
defmodule RabbitMQ.CLI.Core.CodePath do
alias RabbitMQ.CLI.Core.{Config, Paths, Platform}
def add_plugins_to_load_path(opts) do
with {:ok, plugins_dir} <- Paths.plugins_dir(opts) do
String.split(to_string(plugins_dir), Platform.path_separator())
|> Enum.map(&add_directory_plugins_to_load_path/1)
:ok
end
end
def add_directory_plugins_to_load_path(directory_with_plugins_inside_it) do
with {:ok, files} <- File.ls(directory_with_plugins_inside_it) do
Enum.map(
files,
fn filename ->
cond do
String.ends_with?(filename, [".ez"]) ->
Path.join([directory_with_plugins_inside_it, filename])
|> String.to_charlist()
|> add_archive_code_path()
File.dir?(filename) ->
Path.join([directory_with_plugins_inside_it, filename])
|> add_dir_code_path()
true ->
{:error, {:not_a_plugin, filename}}
end
end
)
end
end
defp add_archive_code_path(ez_dir) do
case :erl_prim_loader.list_dir(ez_dir) do
{:ok, [app_dir]} ->
app_in_ez = :filename.join(ez_dir, app_dir)
add_dir_code_path(app_in_ez)
_ ->
{:error, :no_app_dir}
end
end
defp add_dir_code_path(app_dir_0) do
app_dir = to_charlist(app_dir_0)
case :erl_prim_loader.list_dir(app_dir) do
{:ok, list} ->
case Enum.member?(list, 'ebin') do
true ->
ebin_dir = :filename.join(app_dir, 'ebin')
Code.append_path(ebin_dir)
false ->
{:error, :no_ebin}
end
_ ->
{:error, :app_dir_empty}
end
end
def require_rabbit_and_plugins(_, opts) do
require_rabbit_and_plugins(opts)
end
def require_rabbit_and_plugins(opts) do
with :ok <- require_rabbit(opts),
:ok <- add_plugins_to_load_path(opts),
do: :ok
end
def require_rabbit(_, opts) do
require_rabbit(opts)
end
def require_rabbit(opts) do
home = Config.get_option(:rabbitmq_home, opts)
case home do
nil ->
{:error, {:unable_to_load_rabbit, :rabbitmq_home_is_undefined}}
_ ->
case Application.load(:rabbit) do
:ok ->
Code.ensure_loaded(:rabbit_plugins)
:ok
{:error, {:already_loaded, :rabbit}} ->
Code.ensure_loaded(:rabbit_plugins)
:ok
{:error, err} ->
{:error, {:unable_to_load_rabbit, err}}
end
end
end
end
|
