Add save subcommand

Author: ComputerDruid

CHANGELOG.md

@@ -1,5 +1,8 @@
 ## Unreleased
 
+### Added
+- New subcommand: `save`, which saves blueprints as a .json file, or as a directory of json files for blueprint books
+
 ### Changed
 - `upgrade-quality` works on combinators
 - `upgrade-quality` now errors when encountering something that looks like a quality that it doesn't know if it should upgrade.

Cargo.lock

@@ -10,6 +10,7 @@ base64 = "0.22.1"
 clap = { version = "4.5.42", features = ["derive"] }
 crossterm = { version = "0.29.0", features = ["osc52"] }
 flate2 = { version = "1.1.2", default-features = false, features = ["zlib-rs"] }
+sanitize-filename = "0.6.0"
 serde = { version = "1.0.219", features = ["derive"] }
 serde_json = "1.0.142"
 zlib-rs = "0.5.1"

Cargo.toml

@@ -10,6 +10,7 @@ Commands:
   unwrap           Unwraps a blueprint string to reveal the json representation
   wrap             Wraps json from stdin into a blueprint string
   upgrade-quality  Upgrades the quality of recipies/filters/conditions without upgrading entities/modules
+  save             Saves blueprint as a .json file, or as a directory of json files if it's a blueprint book
   help             Print this message or the help of the given subcommand(s)
 
 Options:

README.md

@@ -1,10 +1,10 @@
 mod blueprint;
 mod json_walk;
+mod save;
 
-use std::{
-    fmt::Write as _,
-    io::{Read, stderr, stdin},
-};
+use std::fmt::Write;
+use std::io::{Read, stderr, stdin};
+use std::str::FromStr;
 
 use blueprint::{blueprint_to_json, json_to_blueprint};
 use clap::{Parser, Subcommand};
@@ -43,6 +43,8 @@ enum Commands {
         to_clipboard: bool,
         blueprint_string: Option<String>,
     },
+    /// Saves blueprint as a .json file, or as a directory of json files if it's a blueprint book.
+    Save { blueprint_string: Option<String> },
 }
 
 mod terminal;
@@ -131,6 +133,13 @@ impl Commands {
                     println!("{bp}");
                 }
             }
+            Commands::Save { blueprint_string } => {
+                let blueprint_string = blueprint_string.unwrap_or_else(terminal::prompt_blueprint);
+                let json = blueprint_to_json(&blueprint_string);
+                let json = serde_json::Value::from_str(&json).expect("should contain valid json");
+
+                save::save(json, None);
+            }
         }
     }
 }

src/main.rs

@@ -0,0 +1,88 @@
+use std::{
+    fs::File,
+    io::{BufWriter, Write},
+    mem,
+    path::{Path, PathBuf},
+};
+
+pub fn save(mut json: serde_json::Value, dir: Option<&Path>) {
+    let mut name = if let Some(thing) = json.get("blueprint").or(json.get("blueprint_book"))
+        && let Some(name) = thing.get("label")
+        && let Some(name) = name.as_str()
+    {
+        name.to_owned()
+    } else if let Some(blueprint) = json.get("blueprint")
+        && let Some(icons) = blueprint.get("icons")
+        && let Some(icons) = icons.as_array()
+    {
+        let mut name = String::new();
+        for icon in icons {
+            if let Some(signal) = icon.get("signal")
+                && let Some(signal_name) = signal.get("name")
+                && let Some(signal_name) = signal_name.as_str()
+            {
+                if !name.is_empty() {
+                    name.push(' ');
+                }
+                name.push_str(signal_name);
+            }
+        }
+        if !name.is_empty() {
+            name
+        } else {
+            "Untitled".to_owned()
+        }
+    } else {
+        "Untitled".to_owned()
+    };
+
+    if let Some(index) = json.get_mut("index")
+        && let Some(index) = index.as_number()
+    {
+        name = format!("{index} {name}");
+    }
+
+    // replace problematic characters like "/"
+    name = sanitize_filename::sanitize_with_options(
+        name,
+        sanitize_filename::Options {
+            windows: true,
+            truncate: true,
+            replacement: "_",
+        },
+    );
+
+    let file_path: PathBuf = if let Some(blueprint_book) = json.get_mut("blueprint_book")
+        && let Some(blueprints) = blueprint_book.get_mut("blueprints")
+        && let Some(blueprints) = blueprints.as_array_mut()
+    {
+        let blueprints = mem::replace(blueprints, vec![]);
+        let path = if let Some(dir) = dir {
+            dir.join(&name)
+        } else {
+            name.clone().into()
+        };
+        std::fs::create_dir(&path)
+            .unwrap_or_else(|e| panic!("error creating book directory {path:?}: {e}"));
+        for blueprint in blueprints {
+            save(blueprint, Some(&path));
+        }
+        path.join("book.json")
+    } else {
+        let file = format!("{name}.json");
+        if let Some(dir) = dir {
+            dir.join(&file)
+        } else {
+            file.clone().into()
+        }
+    };
+
+    {
+        let out_file = File::create_new(&file_path)
+            .unwrap_or_else(|e| panic!("error creating file {file_path:?}: {e:?}"));
+        let mut writer = BufWriter::new(out_file);
+        serde_json::to_writer_pretty(&mut writer, &json).expect("error writing {file_path}");
+        writer.flush().expect("error writing {file_path}");
+    }
+    println!("{file_path:?} saved.");
+}