Handle unknown subcommands in binaries with Runtime::handle_unknown_subcommand

1 files changed, 6 insertions, 3 deletions

diff --git a/bin/domain/imag-habit/src/main.rs b/bin/domain/imag-habit/src/main.rs
index ccb8e71e..61b69bd6 100644
--- a/bin/domain/imag-habit/src/main.rs
+++ b/bin/domain/imag-habit/src/main.rs
@@ -90,9 +90,12 @@ fn main() {
                 "status" => today(&rt, true),
                 "show"   => show(&rt),
                 "done"   => done(&rt),
-                _        => {
-                    debug!("Unknown command"); // More error handling
-                    exit(1)
+                other    => {
+                    debug!("Unknown command");
+                    let _ = rt.handle_unknown_subcommand("imag-habit", other, rt.cli())
+                        .map_err_trace_exit_unwrap(1)
+                        .code()
+                        .map(::std::process::exit);
                 },
             }
         })