%% -------------------------------------------------------------------
%%
%% Copyright (c) 2014 Basho Technologies, Inc.  All Rights Reserved.
%%
%% This file is provided to you 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.
%%
%% -------------------------------------------------------------------
-module(clique_nodes).

-export([init/0,
         safe_rpc/4,
         nodes/0,
         register/1]).

-ifdef(TEST).
-export([teardown/0]).
-endif.

-define(nodes_table, clique_nodes).

init() ->
    _ = ets:new(?nodes_table, [public, named_table]),
    ok.

-spec register(fun()) -> true.
register(Fun) ->
    ets:insert(?nodes_table, {nodes_fun, Fun}).

-spec nodes() -> [node()].
nodes() ->
    [{nodes_fun, Fun}] = ets:lookup(?nodes_table, nodes_fun),
    Fun().

%% @doc Wraps an rpc:call/4 in a try/catch to handle the case where the
%%      'rex' process is not running on the remote node. This is safe in
%%      the sense that it won't crash the calling process if the rex
%%      process is down.
-spec safe_rpc(Node :: node(), Module :: atom(), Function :: atom(), Args :: [any()]) -> {'badrpc', any()} | any().
safe_rpc(Node, Module, Function, Args) ->
    try rpc:call(Node, Module, Function, Args) of
        Result ->
            Result
    catch
        exit:{noproc, _NoProcDetails} ->
            {badrpc, rpc_process_down}
    end.

-ifdef(TEST).
teardown() ->
    ets:delete(?nodes_table).
-endif.