Initial commit

src/main.rs

@@ -0,0 +1,218 @@
+use anyhow::Context;
+use clap::Parser;
+use regex::Regex;
+use serde::Deserialize;
+use std::fs;
+use std::io::{Read, stdin};
+use std::path::PathBuf;
+
+/// Convert statblocks from Archives of Nethys into markdownish-blocks compatible with pf2e-stats obsidian plugin.
+#[derive(Parser, Debug)]
+#[command(author, version, about, long_about = None)]
+struct Args {
+    /// Path to the JSON file. If none, the JSON will be read from stdin.
+    path: Option<PathBuf>,
+}
+
+#[allow(dead_code)]
+#[derive(Deserialize, Debug)]
+#[serde(rename_all = "snake_case")]
+struct CharacterSb {
+    name: String,
+    #[serde(rename = "type")]
+    kind: String,
+    level: serde_json::Value,
+    #[serde(rename = "trait")]
+    traits: Vec<String>,
+    skill_markdown: Option<String>,
+    hp_raw: String,
+    perception: i32,
+    sense_markdown: String,
+    strength: i32,
+    dexterity: i32,
+    constitution: i32,
+    intelligence: i32,
+    wisdom: i32,
+    charisma: i32,
+    ac: i32,
+    fortitude_save: i32,
+    reflex_save: i32,
+    will_save: i32,
+    immunity_markdown: Option<String>,
+    resistance_markdown: Option<String>,
+    weakness_markdown: Option<String>,
+    speed_markdown: String,
+    markdown: String,
+    /// If this is a legacy creature that has been remastered, this will be `Some`
+    remaster_id: Option<Vec<String>>,
+    /// If this is a remastered creature, this will be `Some`
+    legacy_id: Option<Vec<String>>,
+}
+
+fn format_val(v: &serde_json::Value) -> String {
+    if v.is_string() {
+        v.as_str().unwrap_or("").to_string()
+    } else {
+        v.to_string().replace('\"', "")
+    }
+}
+
+fn fix_action_icons(input: String) -> String {
+    input
+        .replace(r#"<actions string="Free Action" />"#, "`[free-action]`")
+        .replace(r#"<actions string="Single Action" />"#, "`[one-action]`")
+        .replace(r#"<actions string="Two Actions" />"#, "`[two-actions]`")
+        .replace(r#"<actions string="Three Actions" />"#, "`[three-actions]`")
+        .replace(r#"<actions string="Reaction" />"#, "`[reaction]`")
+}
+
+fn parse_ability_section(
+    markdown: &str,
+    mut section_start: impl FnMut(&str) -> bool,
+    mut section_end: impl FnMut(&str) -> bool,
+) -> String {
+    let re_split = Regex::new(r"\n\s*\n").unwrap();
+    let sections: Vec<&str> = re_split.split(markdown).collect();
+
+    let relevant_sections = sections
+        .into_iter()
+        .map(|s| s.trim_start())
+        .filter(|s| s.starts_with("**"))
+        .skip_while(|s| !section_start(s))
+        .skip(1)
+        .take_while(|s| !section_end(s));
+
+    let processed: Vec<String> = relevant_sections
+        .map(|s| s.replace("\r", ""))
+        .map(|s| fix_action_icons(s.to_string()))
+        .map(|s| s.replace('\n', " "))
+        .collect();
+
+    processed.join("\n")
+}
+
+fn parse_offensive_abilities(markdown: &str) -> String {
+    parse_ability_section(markdown, |s| s.starts_with("**Speed**"), |_| false)
+}
+
+fn parse_defensive_abilities(markdown: &str) -> String {
+    let re_split = Regex::new(r"\n\s*\n").unwrap();
+    let sections: Vec<&str> = re_split.split(markdown).collect();
+
+    let relevant_sections = sections
+        .into_iter()
+        .map(|s| s.trim_start())
+        .map(|s| dbg!(s))
+        .filter(|s| s.starts_with("**"))
+        .skip_while(|s| !s.starts_with("**HP**"))
+        .skip(1)
+        .skip_while(|s| {
+            s.starts_with("**Immunities**")
+                || s.starts_with("**Resistances**")
+                || s.starts_with("**Weaknesses**")
+        })
+        .take_while(|s| !s.starts_with("**Speed**"));
+
+    let processed: Vec<String> = relevant_sections
+        .map(|s| s.replace("\r", ""))
+        .map(|s| fix_action_icons(s.to_string()))
+        .map(|s| s.replace('\n', " "))
+        .collect();
+
+    processed.join("\n")
+}
+
+fn parse_general_abilities(markdown: &str) -> String {
+    parse_ability_section(
+        markdown,
+        |s| s.starts_with("**Cha**"),
+        |s| s.starts_with("**Fort**"),
+    )
+}
+
+fn main() -> anyhow::Result<()> {
+    let args = Args::parse();
+    let content = if let Some(path) = &args.path {
+        fs::read_to_string(path).context("Failed to read JSON file")?
+    } else {
+        let mut s = String::new();
+        stdin()
+            .read_to_string(&mut s)
+            .context("Failed to read from stdin")?;
+        s
+    };
+    let sb: CharacterSb = serde_json::from_str(&content).context("Failed to parse JSON file")?;
+
+    let traits_str = sb
+        .traits
+        .iter()
+        .map(|t| format!("=={}==", t))
+        .collect::<Vec<_>>()
+        .join(" ");
+
+    let hp_text = format!("**HP** {}", &sb.hp_raw);
+    let hp_stuff = [
+        ("Immunities", &sb.immunity_markdown),
+        ("Resistances", &sb.resistance_markdown),
+        ("Weaknesses", &sb.weakness_markdown),
+    ]
+    .into_iter()
+    .filter_map(|(prefix, list)| Some(prefix).zip(list.as_deref()))
+    .map(|(prefix, list)| format!("**{prefix}** {list}"))
+    .fold(hp_text, |b, s| b + "; " + &s);
+
+    let general_abilities = parse_general_abilities(&sb.markdown);
+    let defensive_abilities = parse_defensive_abilities(&sb.markdown);
+    let offensive_abilities = parse_offensive_abilities(&sb.markdown);
+
+    println!("```pf2e-stats");
+    println!("# {}", sb.name);
+    println!("## {} {}", sb.kind, format_val(&sb.level));
+    println!("---");
+    println!("\n{}\n", traits_str);
+    println!(
+        "\n**Perception** {}; {}\n",
+        plussed(sb.perception),
+        &sb.sense_markdown,
+    );
+    println!(
+        "\n**Skills**: {}\n",
+        sb.skill_markdown.as_deref().unwrap_or_default() // TODO
+    );
+    println!(
+        "\n**Str** {}, **Dex** {}, **Con** {}, **Int** {}, **Wis** {}, **Cha** {}",
+        plussed(sb.strength),
+        plussed(sb.dexterity),
+        plussed(sb.constitution),
+        plussed(sb.intelligence),
+        plussed(sb.wisdom),
+        plussed(sb.charisma),
+    );
+    println!("{general_abilities}");
+    println!();
+    println!("---");
+    println!(
+        "\n**AC** {}; **Fort** {}, **Ref** {} **Will** {}",
+        &sb.ac,
+        plussed(sb.fortitude_save),
+        plussed(sb.reflex_save),
+        plussed(sb.will_save),
+    );
+    println!("{}", hp_stuff);
+    println!("{defensive_abilities}");
+    println!();
+    println!("\n---");
+    println!("\n**Speed**: {}\n", &sb.speed_markdown);
+    println!("{}", offensive_abilities);
+    println!("\n```");
+
+    Ok(())
+}
+
+fn plussed(number: i32) -> String {
+    if number >= 0 {
+        format!("+{number}")
+    } else {
+        format!("{number}")
+    }
+}