1 files changed, 76 insertions, 0 deletions
diff --git a/ipfs-cli/src/command/diag.rs b/ipfs-cli/src/command/diag.rs
new file mode 100644
index 0000000..24f65aa
--- /dev/null
+++ b/ipfs-cli/src/command/diag.rs
@@ -0,0 +1,76 @@
+// Copyright 2017 rust-ipfs-api Developers
+//
+// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
+// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
+// http://opensource.org/licenses/MIT>, at your option. This file may not be
+// copied, modified, or distributed except according to those terms.
+//
+
+use clap::{App, Arg, ArgMatches, SubCommand};
+use command::EXPECTED_API;
+use ipfs_api::IpfsClient;
+use tokio_core::reactor::Core;
+
+
+pub fn signature<'a, 'b>() -> App<'a, 'b> {
+    // The clap macro does not allow hyphenated command names,
+    // so the `set-time` command has to be manually instantiated.
+    //
+    let set_time_command = SubCommand::with_name("set-time")
+        .about("Set how long to keep inactive requests in the log")
+        .arg(Arg::with_name("TIME").required(true).index(1).help(
+            "Time to keep inactive requests in the log",
+        ));
+
+    clap_app!(
+        @subcommand diag =>
+            (@setting SubcommandRequiredElseHelp)
+            (@subcommand cmds =>
+                (@setting SubcommandRequiredElseHelp)
+                (@subcommand clear =>
+                    (about: "Clear inactive requests from the log")
+                )
+                (subcommand: set_time_command)
+            )
+            (@subcommand sys =>
+                (about: "Print system diagnostic information")
+            )
+    )
+}
+
+
+pub fn handle(core: &mut Core, client: &IpfsClient, args: &ArgMatches) {
+    match args.subcommand() {
+        ("cmds", Some(args)) => {
+            match args.subcommand() {
+                ("clear", _) => {
+                    core.run(client.diag_cmds_clear()).expect(EXPECTED_API);
+
+                    println!("");
+                    println!("  OK");
+                    println!("");
+                }
+                ("set-time", Some(args)) => {
+                    let time = args.value_of("TIME").unwrap();
+
+                    core.run(client.diag_cmds_set_time(&time)).expect(
+                        EXPECTED_API,
+                    );
+
+                    println!("");
+                    println!("  OK");
+                    println!("");
+                }
+                _ => unreachable!(),
+            }
+        }
+        ("sys", _) => {
+            let sys = core.run(client.diag_sys()).expect(EXPECTED_API);
+
+            println!("");
+            println!("  {}", sys);
+            println!("");
+        }
+        _ => unreachable!(),
+    }
+}