From 659c7d4f8e1041d905acb5fd03edb3fb74c5db2d Mon Sep 17 00:00:00 2001
From: Patrick Auernig <patrick.auernig@mykolab.com>
Date: Tue, 8 Aug 2023 00:24:40 +0200
Subject: [PATCH] Add edit subcommand

---
 crates/wayfarer/src/edit.rs | 82 +++++++++++++++++++++++++++++++++++++
 crates/wayfarer/src/main.rs |  7 ++++
 2 files changed, 89 insertions(+)
 create mode 100644 crates/wayfarer/src/edit.rs

diff --git a/crates/wayfarer/src/edit.rs b/crates/wayfarer/src/edit.rs
new file mode 100644
index 0000000..3605490
--- /dev/null
+++ b/crates/wayfarer/src/edit.rs
@@ -0,0 +1,82 @@
+use std::fs::{self, File};
+use std::path::PathBuf;
+
+use anyhow::Result;
+use clap::builder::PossibleValuesParser;
+use clap::{value_parser, Parser as ArgParser};
+use jrny_save::{RobeColor, Savefile, LEVEL_NAMES};
+
+use crate::Args as AppArgs;
+
+
+#[derive(Debug, Clone, ArgParser)]
+pub(crate) struct Args {
+    in_path: PathBuf,
+    out_path: PathBuf,
+
+    #[arg(long, value_parser = value_parser!(u32).range(1..=32))]
+    scarf_length: Option<u32>,
+
+    #[arg(long, value_parser = PossibleValuesParser::new(LEVEL_NAMES))]
+    current_level: Option<String>,
+
+    #[arg(long, value_parser = value_parser!(u32).range(0..=21))]
+    symbol: Option<u32>,
+
+    #[arg(long, value_parser = PossibleValuesParser::new(["red", "white"]))]
+    robe_color: Option<String>,
+
+    /// Sets the robe tier from 1 to 4, white robe always has a minimum of 2
+    #[arg(long, value_parser = value_parser!(u32).range(1..=4))]
+    robe_tier: Option<u32>,
+}
+
+
+pub(crate) fn execute(_app_args: &AppArgs, sub_args: &Args) -> Result<()> {
+    let in_file = File::open(&sub_args.in_path)?;
+
+    let savefile = Savefile::from_reader(in_file)?;
+
+    let new_savefile = edit_file(&savefile, &sub_args)?;
+
+    let out_file = fs::OpenOptions::new()
+        .write(true)
+        .create(true)
+        .truncate(true)
+        .open(&sub_args.out_path)?;
+
+    new_savefile.write(out_file)?;
+
+    Ok(())
+}
+
+
+fn edit_file(cur_savefile: &Savefile, args: &Args) -> Result<Savefile> {
+    let mut savefile = cur_savefile.clone();
+
+    if let Some(val) = args.scarf_length {
+        savefile.scarf_length = val;
+    }
+
+    if let Some(val) = &args.current_level {
+        savefile.current_level = LEVEL_NAMES.iter().position(|&v| v == val).unwrap() as u64;
+    }
+
+    if let Some(val) = args.symbol {
+        savefile.symbol = val;
+    }
+
+    if let Some(color) = &args.robe_color {
+        match color.as_ref() {
+            "red" => savefile.set_robe_color(RobeColor::Red),
+            "white" => savefile.set_robe_color(RobeColor::White),
+            _ => (),
+        }
+    }
+
+    if let Some(tier) = args.robe_tier {
+        savefile.set_robe_tier(tier);
+    }
+
+    Ok(savefile)
+}
diff --git a/crates/wayfarer/src/main.rs b/crates/wayfarer/src/main.rs
index 9ffc8bd..176c61f 100644
--- a/crates/wayfarer/src/main.rs
+++ b/crates/wayfarer/src/main.rs
@@ -1,3 +1,4 @@
+mod edit;
 mod show;
 
 
@@ -6,6 +7,7 @@ use clap::Parser as ArgParser;
 
 
 #[derive(Debug, ArgParser)]
+#[command(author, version, about)]
 pub(crate) struct Args {
     #[command(subcommand)]
     command: CommandArgs,
@@ -14,7 +16,11 @@ pub(crate) struct Args {
 
 #[derive(Debug, Clone, clap::Subcommand)]
 pub(crate) enum CommandArgs {
+    /// Display info about save files
     Show(show::Args),
+
+    /// Edit verious aspect of save files
+    Edit(edit::Args),
 }
 
 
@@ -23,6 +29,7 @@ fn main() -> Result<()> {
 
     match &args.command {
         CommandArgs::Show(sub_args) => show::execute(&args, sub_args)?,
+        CommandArgs::Edit(sub_args) => edit::execute(&args, sub_args)?,
     }
 
     Ok(())