wip

Author: dumbbell

deps/rabbit/Makefile

@@ -158,7 +158,7 @@ ebin/$(PROJECT).app:: $(CLI_SCRIPTS)
 
 $(CLI_SCRIPTS): ebin/rabbit_cli.beam
 	$(gen_verbose) echo '#!/usr/bin/env escript' > "$@"
-	$(verbose) echo '%%! -start_epmd false' >> "$@"
+	$(verbose) echo '%%! +sbtu +A1 -start_epmd false' >> "$@"
 	$(verbose) cat $^ >> "$@"
 	$(verbose) chmod a+x "$@"
 

deps/rabbit/src/rabbit_cli.erl

@@ -15,20 +15,29 @@ main(Args) ->
 
 run_ci(Args) ->
     maybe
-        {ok, _RemainingArgs, Nodename} ?= determine_rabbitmq_nodename(Args),
+        io:format("Args = ~p~n", [Args]),
+        {ok, RemainingArgs, Nodename} ?= lookup_rabbitmq_nodename(Args),
+        io:format("Args = ~p~n", [RemainingArgs]),
         io:format("Nodename = ~p~n", [Nodename]),
-        {ok, CommandMap} ?= determine_command_map(Nodename),
-        io:format("CommandMap= ~p~n", [CommandMap]),
+        {ok, _} ?= net_kernel:start(undefined, #{name_domain => shortnames}),
+        true ?= net_kernel:connect_node(Nodename),
+        io:format("This node = ~p~n", [node()]),
+        %{ok, CommandMap} ?= lookup_command_map(Nodename),
+        %io:format("CommandMap= ~p~n", [CommandMap]),
+        Ret ?= run_command(Nodename,  RemainingArgs),
+        io:format("Ret = ~p~n", [Ret]),
         ok
     end.
 
-determine_rabbitmq_nodename(Args) ->
+lookup_rabbitmq_nodename(Args) ->
     case get_rabbitmq_nodename_from_args(Args) of
         {ok, Args1} ->
-            GuessedNodename = guess_rabbitmq_nodename(),
-            {ok, Args1, GuessedNodename};
-        {ok, Args1, Nodename} ->
-            {ok, Args1, Nodename};
+            GuessedNodename0 = guess_rabbitmq_nodename(),
+            GuessedNodename1 = complete_nodename(GuessedNodename0),
+            {ok, Args1, GuessedNodename1};
+        {ok, Args1, Nodename0} ->
+            Nodename1 = complete_nodename(Nodename0),
+            {ok, Args1, Nodename1};
         {error, _} = Error ->
             Error
     end.
@@ -74,15 +83,65 @@ guess_rabbitmq_nodename() ->
             "rabbit"
     end.
 
-determine_command_map(Nodename) ->
-    case is_node_local(Nodename) of
-        true ->
-            %% File or RPC.
-            {ok, #{}};
-        false ->
-            %% RPC.
-            {ok, #{}}
+complete_nodename(Nodename) ->
+    case re:run(Nodename, "@", [{capture, none}]) of
+        nomatch ->
+            {ok, ThisHost} = inet:gethostname(),
+            list_to_atom(Nodename ++ "@" ++ ThisHost);
+        match ->
+            list_to_atom(Nodename)
     end.
 
-is_node_local(Nodename) ->
-    Nodename =:= node().
+%lookup_command_map(Nodename) ->
+%    %% Order of operations:
+%    %% 1. refresh the cached copy:
+%    %%     a. is the node running?
+%    %%        yes -> query its uptime
+%    %%        no  -> local or remote?
+%    %%            local  -> list beam files + file size + last modified date
+%    %%            remote -> no refresh possible
+%    %%     b. compare uptime to cache date
+%    %%        or
+%    %%        compare files list to cached files list
+%    %%     c. if refresh needed, query command map or (local only) extract it from files list
+%    %%
+%    %% 2. use cached command map; error out if none
+%    %% 
+%    %% Or:
+%    %% 0. node extracts the command map on startup and stores it persistent_term
+%    %% 1. query node for command map with a very short timeout + store result locally
+%    %% 2. read local copy
+%    %% + we skip the query if `--help` or completion and if there is a local copy
+%    case is_node_local(Nodename) of
+%        true ->
+%            %% Generated local command map.
+%            lookup_local_command_map(),
+%            {ok, #{}};
+%        false ->
+%            %% Cache or RPC.
+%            {ok, #{}}
+%    end.
+%
+%is_node_local(Nodename) ->
+%    case re:run(Nodename, "@(.+)", [{capture, all_but_first, list}]) of
+%        {match, ["localhost"]} ->
+%            true;
+%        {match, [HostPart]} ->
+%            ThisHost = inet:gethostname(),
+%            HostPart =:= ThisHost;
+%        nomatch ->
+%            true
+%    end.
+%
+%lookup_local_command_map() ->
+%    ScriptDir = filename:dirname(escript:script_name()),
+%    io:format("Script = ~p~n", [ScriptDir]),
+%    ok.
+
+run_command(Nodename, Args) ->
+    try
+        erpc:call(Nodename, rabbit_cli_commands, run_command, [Args])
+    catch
+        error:{erpc, Reason} ->
+            {error, Reason}
+    end.

deps/rabbit/src/rabbit_cli_commands.erl

@@ -0,0 +1,44 @@
+-module(rabbit_cli_commands).
+
+-include_lib("kernel/include/logger.hrl").
+
+-export([commands/0, run_command/1]).
+-export([list_queues/1]).
+
+commands() ->
+    #{
+      arguments =>
+      [
+       #{name => verbose,
+         short => $v,
+         action => count,
+         help =>
+         "Be verbose; can be specified multiple times to increase "
+         "verbosity"}
+      ],
+      commands =>
+      #{
+       "list" =>
+       #{commands =>
+         #{"queues" =>
+           #{handler => {?MODULE, list_queues}}
+          }
+        }
+      }
+     }.
+
+run_command(Args) ->
+    Definition = commands(),
+    case argparse:parse(Args, Definition, #{}) of
+        {ok, ArgMap, CmdPath, #{handler := {Mod, Fun}} = Command} ->
+            ?LOG_ALERT("ArgMap=~0p~nCmdPath=~0p~nCommand=~0p", [ArgMap, CmdPath, Command]),
+            Mod:Fun(ArgMap),
+            ok;
+        {error, Reason} ->
+            ?LOG_ALERT("Reason=~0p", [Reason]),
+            ok
+    end.
+
+list_queues(ArgMap) ->
+    ?LOG_ALERT("Listing queues, ~0p", [ArgMap]),
+    ok.