Test interactive UI

Author: dumbbell

deps/rabbit/src/rabbit_cli_commands.erl

@@ -7,7 +7,8 @@
 
 -export([argparse_def/0, run_command/1, do_run_command/1]).
 -export([cmd_list_exchanges/1,
-         cmd_import_definitions/1]).
+         cmd_import_definitions/1,
+         cmd_top/1]).
 
 -rabbitmq_command(
    {#{cli => ["declare", "exchange"],
@@ -38,6 +39,11 @@
      #{help => "Import definitions",
        handler => {?MODULE, cmd_import_definitions}}]}).
 
+-rabbitmq_command(
+   {#{cli => ["top"]},
+    [#{help => "Top-like interactive view",
+       handler => {?MODULE, cmd_top}}]}).
+
 argparse_def() ->
     #{argparse_def := ArgparseDef} = get_discovered_commands(),
     ArgparseDef.
@@ -186,3 +192,24 @@ cmd_import_definitions(#{arg_map := ArgMap, io := IO}) ->
         {error, _} = Error ->
             Error
     end.
+
+cmd_top(#{io := IO} = Context) ->
+    Top = spawn_link(fun() -> run_top(IO) end),
+    wait_quit(Context, Top).
+
+run_top(IO) ->
+    receive
+        quit ->
+            ok
+    after 1000 ->
+              rabbit_cli_io:format(IO, "Refresh~n", []),
+              run_top(IO)
+    end.
+
+wait_quit(#{arg_map := _ArgMap, io := _IO}, Top) ->
+    receive
+        {keypress, _} ->
+            erlang:unlink(Top),
+            Top ! quit,
+            ok
+    end.

deps/rabbit/src/rabbit_cli_curses.erl

@@ -0,0 +1,6 @@
+-module(rabbit_cli_curses).
+
+-export([init/0]).
+
+init() ->
+    window.

deps/rabbit/src/rabbit_cli_io.erl

@@ -12,6 +12,7 @@
          start_record_stream/4,
          push_new_record/3,
          end_record_stream/2,
+         send_keyboard_input/3,
          read_file/2]).
 -export([init/1,
          handle_call/3,
@@ -21,7 +22,9 @@
          code_change/3]).
 
 -record(?MODULE, {progname,
-                  record_streams = #{}}).
+                  record_streams = #{},
+                  kbd_reader = undefined,
+                  kbd_subscribers = []}).
 
 start_link(Progname) ->
     gen_server:start_link(rabbit_cli_io, #{progname => Progname}, []).
@@ -92,6 +95,14 @@ end_record_stream({transport, Transport}, #{name := Name}) ->
 end_record_stream(IO, #{name := Name}) ->
     gen_server:cast(IO, {?FUNCTION_NAME, Name}).
 
+send_keyboard_input({transport, Transport}, ArgMap, Subscriber) ->
+    Transport ! {io_call, self(), {?FUNCTION_NAME, ArgMap, Subscriber}},
+    receive Ret -> Ret end;
+send_keyboard_input(IO, ArgMap, Subscriber)
+  when is_pid(IO) andalso
+       is_map(ArgMap) ->
+    gen_server:call(IO, {?FUNCTION_NAME, ArgMap, Subscriber}).
+
 read_file({transport, Transport}, ArgMap) ->
     Transport ! {io_call, self(), {?FUNCTION_NAME, ArgMap}},
     receive Ret -> Ret end;
@@ -116,14 +127,21 @@ handle_call(
 
     {ok, State2} = format_record_stream_start(Name, State1),
 
-    {noreply, State2};
+    {noreply, State2, compute_timeout(State2)};
+handle_call(
+  {send_keyboard_input, _ArgMap, Subscriber},
+  _From,
+  #?MODULE{kbd_subscribers = Subscribers} = State) ->
+    Subscribers1 = [Subscriber | Subscribers],
+    State1 = State#?MODULE{kbd_subscribers = Subscribers1},
+    {reply, ok, State1, compute_timeout(State1)};
 handle_call({read_file, ArgMap}, From, State) ->
     {ok, State1} = do_read_file(ArgMap, From, State),
-    {noreply, State1};
+    {noreply, State1, compute_timeout(State1)};
 handle_call(stop, _From, State) ->
     {stop, normal, ok, State};
 handle_call(_Request, _From, State) ->
-    {reply, ok, State}.
+    {reply, ok, State, compute_timeout(State)}.
 
 handle_cast(
   {display_help, #{cmd_path := CmdPath, argparse_def := ArgparseDef}},
@@ -134,28 +152,51 @@ handle_cast(
                 command => tl(CmdPath)},
     Help = argparse:help(ArgparseDef, Options),
     io:format("~s~n", [Help]),
-    {noreply, State};
+    {noreply, State, compute_timeout(State)};
 handle_cast({format, Format, Args}, State) ->
     io:format(Format, Args),
-    {noreply, State};
+    {noreply, State, compute_timeout(State)};
 handle_cast({push_new_record, Name, Record}, State) ->
     {ok, State1} = format_record(Name, Record, State),
-    {noreply, State1};
+    {noreply, State1, compute_timeout(State1)};
 handle_cast({end_record_stream, Name}, State) ->
     {ok, State1} = format_record_stream_end(Name, State),
-    {noreply, State1};
+    {noreply, State1, compute_timeout(State1)};
 handle_cast(_Request, State) ->
-    {noreply, State}.
+    {noreply, State, compute_timeout(State)}.
 
+handle_info(timeout, #?MODULE{kbd_reader = Reader} = State)
+  when is_pid(Reader) ->
+    {noreply, State};
+handle_info(timeout, #?MODULE{kbd_subscribers = []} = State) ->
+    {noreply, State};
+handle_info(timeout, #?MODULE{kbd_subscribers = Subscribers} = State) ->
+    Parent = self(),
+    Reader = spawn_link(
+               fun() ->
+                       Ret = io:read(""),
+                       lists:foreach(
+                         fun(Sub) ->
+                                 Sub ! {keypress, Ret}
+                         end, Subscribers),
+                       erlang:unlink(Parent)
+               end, Subscribers),
+    State1 = State#?MODULE{kbd_reader = Reader},
+    {noreply, State1, compute_timeout(State1)};
 handle_info(_Info, State) ->
-    {noreply, State}.
+    {noreply, State, compute_timeout(State)}.
 
 terminate(_Reason, _State) ->
     ok.
 
 code_change(_OldVsn, State, _Extra) ->
     {ok, State}.
 
+compute_timeout(#?MODULE{kbd_subscribers = []}) ->
+    infinity;
+compute_timeout(#?MODULE{kbd_subscribers = _}) ->
+    0.
+
 format_record_stream_start(
   Name,
   #?MODULE{record_streams = Streams} = State) ->

deps/rabbit/src/rabbit_khepri.erl

@@ -260,7 +260,8 @@ setup(_) ->
                       {default_ra_system, ?RA_SYSTEM}]}],
            [{persistent, true}]),
     RaServerConfig = #{cluster_name => ?RA_CLUSTER_NAME,
-                       friendly_name => ?RA_FRIENDLY_NAME},
+                       friendly_name => ?RA_FRIENDLY_NAME,
+                       machine_config => #{snapshot_interval => 60000}},
     case khepri:start(?RA_SYSTEM, RaServerConfig) of
         {ok, ?STORE_ID} ->
             RetryTimeout = retry_timeout(),