Format all Khepri timeout errors for ctl commands

Author: dcorbacho

deps/rabbitmq_cli/lib/rabbitmq/cli/ctl/commands/reset_command.ex

@@ -28,11 +28,6 @@ defmodule RabbitMQ.CLI.Ctl.Commands.ResetCommand do
      RabbitMQ.CLI.DefaultOutput.mnesia_running_error(node_name)}
   end
 
-  def output({:error, {:timeout, {:metadata_store, _}}}, %{node: node_name}) do
-    {:error, RabbitMQ.CLI.Core.ExitCodes.exit_tempfail(),
-     RabbitMQ.CLI.DefaultOutput.khepri_timeout_error(node_name)}
-  end
-
   use RabbitMQ.CLI.DefaultOutput
 
   def usage, do: "reset"

deps/rabbitmq_cli/lib/rabbitmq/cli/default_output.ex

@@ -17,7 +17,7 @@ defmodule RabbitMQ.CLI.DefaultOutput do
   end
 
   def output(result, opts \\ %{}) do
-    format_output(normalize_output(result, opts))
+    format_output(format_khepri_output(normalize_output(result, opts), opts))
   end
 
   def mnesia_running_error(node_name) do
@@ -68,6 +68,18 @@ defmodule RabbitMQ.CLI.DefaultOutput do
   defp normalize_output({unknown, _} = input, _opts) when is_atom(unknown), do: {:error, input}
   defp normalize_output(result, _opts) when not is_atom(result), do: {:ok, result}
 
+  defp format_khepri_output({:error, {:timeout, {:metadata_store, _}}}, %{node: node_name}) do
+    {:error, RabbitMQ.CLI.Core.ExitCodes.exit_tempfail(), khepri_timeout_error(node_name)}
+  end
+
+  defp format_khepri_output({:error, :timeout_waiting_for_leader}, %{node: node_name}) do
+    {:error, RabbitMQ.CLI.Core.ExitCodes.exit_tempfail(), khepri_timeout_error(node_name)}
+  end
+
+  defp format_khepri_output(result, _opts) do
+    result
+  end
+
   defp format_output({:error, _} = result) do
     result
   end