WIP

Author: dumbbell

deps/rabbit/Makefile

@@ -128,11 +128,13 @@ endef
 LOCAL_DEPS = sasl os_mon inets compiler public_key crypto ssl syntax_tools xmerl
 
 BUILD_DEPS = rabbitmq_cli
-DEPS = ranch rabbit_common amqp10_common rabbitmq_prelaunch ra sysmon_handler stdout_formatter recon redbug observer_cli osiris syslog systemd seshat khepri khepri_mnesia_migration cuttlefish gen_batch_server
+DEPS = ranch rabbit_common amqp10_common rabbitmq_prelaunch ra sysmon_handler stdout_formatter recon redbug observer_cli osiris syslog systemd seshat khepri khepri_mnesia_migration cuttlefish gen_batch_server host_triple terminfo
 TEST_DEPS = rabbitmq_ct_helpers rabbitmq_ct_client_helpers meck proper amqp_client rabbitmq_amqp_client rabbitmq_amqp1_0
 
 PLT_APPS += mnesia runtime_tools
 
+dep_host_triple = hex 0.1.0
+dep_terminfo = git https://github.com/rabbitmq/terminfo.git master
 dep_syslog = git https://github.com/schlagert/syslog 4.0.0
 
 define usage_xml_to_erl

deps/rabbit/src/rabbit_cli_cmd_list_exchanges.erl

@@ -2,6 +2,8 @@
 
 -export([cli/0]).
 
+-include("rabbit.hrl").
+
 cli() ->
     #{commands => #{
         "exchange" => #{
@@ -32,7 +34,27 @@ list_exchanges(Args) ->
                    rabbit_exchange,
                    info_all,
                    [VHost, InfoKeys]),
-    rabbit_cli_output:sync_notify(Ret).
+    case is_list(Ret) of
+        true ->
+            rabbit_cli_output:notify(
+              {info_table, #{keys => InfoKeys,
+                             rows => Ret,
+                             callbacks => #{
+                               info_key_to_col_title =>
+                               fun(S) -> string:titlecase(atom_to_list(S)) end,
+                               info_value_to_cell_content =>
+                               fun
+                                   ({name, #resource{name = <<>>}}) ->
+                                       "<empty>";
+                                   ({name, #resource{name = Name}}) ->
+                                       Name;
+                                   ({_, T}) ->
+                                       T
+                               end
+                              }}});
+        false ->
+            rabbit_cli_output:notify(Ret)
+    end.
 
 get_nodename(#{node := Nodename}) ->
     Nodename;

deps/rabbit/src/rabbit_cli_output_console.erl

@@ -10,12 +10,65 @@
          terminate/2,
          code_change/3]).
 
+-record(state, {
+          setupterm_done = false :: boolean(),
+          isatty = #{
+            stdin => false,
+            stdout => false,
+            stderr => false
+           } :: #{stdin => boolean(),
+                  stdout => boolean(),
+                  stderr => boolean()},
+          caps = #{} :: map()
+         }).
+
 init(_) ->
-    {ok, undefined}.
+    State = setup_term(#state{}),
+    {ok, State}.
 
 handle_call(_Request, State) ->
     {ok, ok, State}.
 
+handle_event(
+  {info_table, #{keys := Keys, rows := Rows0, callbacks := CBs}},
+  State)
+  when length(Keys) > 0 ->
+    UseColors = use_colors(stdout, State),
+    UseLines = use_lines(stdout, State),
+    Title = case UseColors of
+                true  -> #{title => true};
+                false -> #{}
+            end,
+    Bold = case UseColors of
+               true  -> #{fg => green};
+               false -> #{}
+           end,
+    Border = case UseLines of
+                 true  -> #{border_drawing => ansi};
+                 false -> #{border_drawing => ascii}
+             end,
+    CB1 = maps:get(info_key_to_col_title, CBs, fun(S) -> S end),
+    CB2 = maps:get(info_value_to_cell_content, CBs, fun(S) -> S end),
+    TableHeader = #row{cells = lists:map(
+                                 CB1,
+                                 Keys),
+                       props = Title},
+    Rows = lists:map(
+             fun(Entry) when length(Entry) =:= length(Keys) ->
+                     #row{
+                        cells = [
+                                 %% TODO: Format using callbacks provided by
+                                 %% the event emitter.
+                                 #paragraph{content = CB2(hd(Entry)),
+                                            props = Bold}
+                                 | lists:map(CB2, tl(Entry))
+                                ]
+                       }
+             end, Rows0),
+    stdout_formatter:display(
+      #table{rows = [TableHeader | Rows],
+             props = Border#{cell_padding => {0, 1}}}),
+    {ok, State};
 handle_event(Event, State) ->
     io:format("~p~n", [Event]),
     {ok, State}.
@@ -28,3 +81,45 @@ terminate(_Arg, _State) ->
 
 code_change(_OldVsn, State, _Extra) ->
     {ok, State}.
+
+setup_term(#state{setupterm_done = true} = State) ->
+    State;
+setup_term(State) ->
+    StdinIsTty = terminfo:isatty(stdin),
+    StdoutIsTty = terminfo:isatty(stdout),
+    StderrIsTty = terminfo:isatty(stderr),
+
+    Isatty = #{
+      stdin => StdinIsTty,
+      stdout => StdoutIsTty,
+      stderr => StderrIsTty
+     },
+
+    Caps = #{
+      stdout => query_term_caps(stdout, StdoutIsTty),
+      stderr => query_term_caps(stderr, StderrIsTty)
+     },
+
+    State#state{
+      setupterm_done = true,
+      isatty = Isatty,
+      caps = Caps
+     }.
+
+query_term_caps(FdName, true) ->
+    terminfo:setupterm(FdName),
+    terminfo:query_all_caps();
+query_term_caps(_FdName, false) ->
+    #{}.
+
+use_colors(FdName, #state{isatty = Isatty, caps = Caps}) ->
+    case maps:get(FdName, Isatty, false) of
+        true ->
+            FdCaps = maps:get(FdName, Caps, #{}),
+            maps:get(colors, FdCaps, 1) > 1;
+        false ->
+            false
+    end.
+
+use_lines(FdName, #state{isatty = Isatty}) ->
+    maps:get(FdName, Isatty, false).