From 8fea6dee76e6da904668d8e70cce06f6d3c02824 Mon Sep 17 00:00:00 2001
From: Josh Durgin <josh.durgin@inktank.com>
Date: Wed, 2 Jan 2013 10:52:27 -0800
Subject: [PATCH] rbd: add --pretty-format option

This is the same option the rados and radosgw-admin tool use for more
human-readable json/xml.

Signed-off-by: Josh Durgin <josh.durgin@inktank.com>
---
 doc/man/8/rbd.rst |  4 ++++
 src/rbd.cc        | 18 ++++++++++++++----
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/doc/man/8/rbd.rst b/doc/man/8/rbd.rst
index 344636463535f..b7d48ca60cf2f 100644
--- a/doc/man/8/rbd.rst
+++ b/doc/man/8/rbd.rst
@@ -104,6 +104,10 @@ Parameters
 
    Specifies output formatting (default: plain, json, xml)
 
+.. option:: --pretty-format
+
+   Make json or xml formatted output more human-readable.
+
 
 Commands
 ========
diff --git a/src/rbd.cc b/src/rbd.cc
index 85f5d26dd0fe8..05e5df3e11c90 100644
--- a/src/rbd.cc
+++ b/src/rbd.cc
@@ -133,7 +133,8 @@ void usage()
 "                                     authenticate as\n"
 "  --keyfile <path>                   file containing secret key for use with cephx\n"
 "  --shared <tag>                     take a shared (rather than exclusive) lock\n"
-"  --format <output-format>           output format (default: plain, json, xml)\n";
+"  --format <output-format>           output format (default: plain, json, xml)\n"
+"  --pretty-format                    make json or xml output more readable\n";
 }
 
 static string feature_str(uint64_t feature)
@@ -199,12 +200,13 @@ struct MyProgressContext : public librbd::ProgressContext {
 };
 
 static int get_outfmt(const char *output_format,
+		      bool pretty,
 		      boost::scoped_ptr<Formatter> *f)
 {
   if (!strcmp(output_format, "json")) {
-    f->reset(new JSONFormatter(false));
+    f->reset(new JSONFormatter(pretty));
   } else if (!strcmp(output_format, "xml")) {
-    f->reset(new XMLFormatter(false));
+    f->reset(new XMLFormatter(pretty));
   } else if (strcmp(output_format, "plain")) {
     cerr << "rbd: unknown format '" << output_format << "'" << std::endl;
     return -EINVAL;
@@ -1792,6 +1794,7 @@ int main(int argc, const char **argv)
     *devpath = NULL, *lock_cookie = NULL, *lock_client = NULL,
     *lock_tag = NULL, *output_format = "plain";
   bool lflag = false;
+  int pretty_format = 0;
   long long stripe_unit = 0, stripe_count = 0;
   long long bench_io_size = 4096, bench_io_threads = 16, bench_bytes = 1 << 30;
 
@@ -1871,6 +1874,7 @@ int main(int argc, const char **argv)
 	output_format = strdup(val.c_str());
 	output_format_specified = true;
       }
+    } else if (ceph_argparse_binary_flag(args, i, &pretty_format, NULL, "--pretty-format", (char*)NULL)) {
     } else {
       ++i;
     }
@@ -1977,6 +1981,12 @@ if (!set_conf_param(v, p1, p2, p3)) { \
     return EXIT_FAILURE;
   }
 
+  if (pretty_format && !strcmp(output_format, "plain")) {
+    cerr << "rbd: --pretty-format only works when --format is json or xml"
+	 << std::endl;
+    return EXIT_FAILURE;
+  }
+
   boost::scoped_ptr<Formatter> formatter;
   if (output_format_specified && opt_cmd != OPT_SHOWMAPPED &&
       opt_cmd != OPT_INFO && opt_cmd != OPT_LIST && opt_cmd != OPT_SNAP_LIST &&
@@ -1984,7 +1994,7 @@ if (!set_conf_param(v, p1, p2, p3)) { \
     cerr << "rbd: command doesn't use output formatting"
 	 << std::endl;
     return EXIT_FAILURE;
-  } else if (get_outfmt(output_format, &formatter) < 0) {
+  } else if (get_outfmt(output_format, pretty_format, &formatter) < 0) {
     return EXIT_FAILURE;
   }
 
-- 
2.39.5