From: MegaBrutal <megabrutal+github@megabrutal.com>
Date: Sun, 19 Feb 2023 03:15:45 +0000 (+0100)
Subject: Pretty print option
X-Git-Url: http://git.megabrutal.com/?p=hlquery.git;a=commitdiff_plain;h=2175495eeef66a37cb5dc34c1463c5cc4b2ad4bb

Pretty print option
---

diff --git a/src/main.rs b/src/main.rs
index f92d3f8..792b64c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -13,9 +13,14 @@ use crate::HLQueryError::{IOError,A2SError};
 #[command(version)]
 #[command(about = "Query Half-Life servers", long_about = None)]
 struct Cli {
+    /// Print output in JSON format
     #[arg(short, long)]
     json: bool,
 
+    /// Pretty-print JSON or Rust objects
+    #[arg(short, long)]
+    pretty: bool,
+
     addresses: Vec<String>
 }
 
@@ -114,9 +119,19 @@ fn main() {
         .collect();
 
     if cli.json {
-        println!("{}", serde_json::to_string_pretty(&query_results).unwrap());
+        if cli.pretty {
+            println!("{}", serde_json::to_string_pretty(&query_results).unwrap());
+        }
+        else {
+            println!("{}", serde_json::to_string(&query_results).unwrap());
+        }
     }
     else {
-        println!("{:?}", query_results);
+        if cli.pretty {
+            println!("{:#?}", query_results);
+        }
+        else {
+            println!("{:?}", query_results);
+        }
     }
 }