file output for all command

Author: cqb13

src/commands/all.rs

@@ -1,4 +1,4 @@
-use crate::utils::{pretty_dates, request};
+use crate::utils::{pretty_dates, request, write_json_to_file};
 use serde_json::Value;
 use std::path::PathBuf;
 
@@ -20,15 +20,24 @@ pub fn all(
     if !display {
         println!("{:#?}", json);
     } else {
-        simplify_and_display_json(json)
+        simplify_and_display_json(&json)
     }
 
-    
+    match output {
+        Some(path) => {
+            let result = write_json_to_file(json, path);
+            match result {
+                Ok(_) => {}
+                Err(err) => println!("{}", err),
+            }
+        }
+        None => {}
+    }
 
     Ok(())
 }
 
-fn simplify_and_display_json(json: Value) {
+fn simplify_and_display_json(json: &Value) {
     // General Info
     let name = &json["name"].as_str().unwrap_or("None");
     let description = &json["description"].as_str().unwrap_or("None");

src/main.rs

@@ -2,8 +2,6 @@ pub mod cli;
 pub mod commands;
 pub mod utils;
 
-use std::process::Output;
-
 use crate::cli::{Arg, Cli, Command};
 use crate::commands::all::all;
 use crate::utils::validate_and_convert_path;

src/utils.rs

@@ -1,5 +1,7 @@
 use reqwest::header::{HeaderMap, ACCEPT, USER_AGENT};
 use serde_json::Value;
+use std::fs::{create_dir_all, remove_file, File};
+use std::io::{BufWriter, Write};
 use std::path::{Path, PathBuf};
 
 pub fn request(url: String) -> Result<Value, Box<dyn std::error::Error>> {
@@ -28,15 +30,42 @@ pub fn pretty_dates(date: &str) -> String {
 pub fn validate_and_convert_path(path: String) -> Result<PathBuf, String> {
     let real_path = Path::new(&path);
 
+    if real_path.extension().is_some() && real_path.extension().unwrap() != "json" {
+        return Err("The file must be a json file.".to_owned());
+    }
+
     if real_path.is_file() {
         println!("A file was found at the path: \"{}\"", path);
         println!("Would you like to clear the file and continue? [y/N]");
         let mut input = String::new();
         std::io::stdin().read_line(&mut input).unwrap();
         if input.trim().to_lowercase() != "y" {
             return Err("A file already exists at the path.".to_owned());
+        } else {
+            remove_file(&real_path).expect("Failed to delete existing directory");
         }
     }
 
     Ok(real_path.to_owned())
 }
+
+pub fn write_json_to_file(json: Value, mut path: PathBuf) -> Result<(), String> {
+    if path.extension().is_none() {
+        path.push("gstats-output.json");
+    }
+
+    println!("Saving the json to: {:?}", path);
+    create_dir_all(path.parent().unwrap()).unwrap();
+    let file = match File::create(&path) {
+        Ok(file) => file,
+        Err(_) => {
+            return Err("Failed to create the file.".to_owned());
+        }
+    };
+
+    let mut writer = BufWriter::new(file);
+    let json_string = serde_json::to_string_pretty(&json).unwrap();
+    writer.write_all(json_string.as_bytes()).unwrap();
+
+    Ok(())
+}