inkhaven 1.3.11

//! 1.2.11+ — `inkhaven show-dont-tell …` subcommand
//! family.  Currently hosts `bootstrap`, which queries
//! the configured LLM for the four per-language word
//! lists the show-don't-tell overlay consumes and emits
//! a copy-paste-ready HJSON snippet on stdout.
//!
//! Design choices:
//!
//!   * **Stdout-only.**  Mirrors `doctor
//!     --filter-words-snippet`.  Never touches the
//!     user's `inkhaven.hjson` — the author reviews
//!     and pastes what they like.
//!   * **Strict JSON contract with the model.**  The
//!     prompt instructs the LLM to emit a single JSON
//!     object with four arrays.  We extract the JSON
//!     from the streamed response (tolerating
//!     surrounding chatter or a fenced code block) and
//!     deserialise; on any parse failure we dump the
//!     raw response on stderr and exit non-zero so the
//!     author can debug + retry.
//!   * **No detector at runtime.**  The overlay stays
//!     regex + wordlist (instant, deterministic,
//!     offline, free).  The LLM is used once-per-
//!     language as a vocabulary curator, not as an
//!     always-on classifier.  Rationale: an always-on
//!     LLM overlay would burn tokens on every
//!     keystroke and lose the offline guarantee.

use std::path::Path;

use crate::ai::AiClient;
use crate::config::Config;
use crate::error::Result;
use crate::project::ProjectLayout;

use super::ShowDontTellCommand;

pub fn run(project: &Path, cmd: ShowDontTellCommand) -> Result<()> {
    match cmd {
        ShowDontTellCommand::Bootstrap {
            language,
            genre,
            provider,
            update,
        } => bootstrap(
            project,
            &language,
            genre.as_deref(),
            provider.as_deref(),
            update,
        ),
    }
}

fn bootstrap(
    project: &Path,
    language: &str,
    genre: Option<&str>,
    provider: Option<&str>,
    update: bool,
) -> Result<()> {
    let layout = ProjectLayout::new(project);
    layout.require_initialized()?;

    let cfg = Config::load_layered(&layout.config_path())?;
    let ai = AiClient::from_config(&cfg.llm)?;
    let (model, _env_var) = ai.resolve_provider(&cfg.llm, provider)?;

    let prompt = build_prompt(language, genre);
    eprintln!(
        "inkhaven show-dont-tell bootstrap · language: {language}{genre_tail} · model: {model}",
        genre_tail = genre
            .map(|g| format!(" · genre: {g}"))
            .unwrap_or_default(),
    );

    let raw = match crate::ai::stream::collect_blocking(
        ai.client.clone(),
        model.to_string(),
        Some(SYSTEM_PROMPT.to_string()),
        prompt,
    ) {
        Ok(raw) => raw,
        Err(e) => {
            eprintln!();
            eprintln!("inference error: {e}");
            return Ok(());
        }
    };
    eprintln!();

    let lists = match parse_lists(&raw) {
        Ok(l) => l,
        Err(why) => {
            eprintln!("could not parse model response: {why}");
            eprintln!("---- raw response ----");
            eprintln!("{raw}");
            eprintln!("---- end ----");
            return Ok(());
        }
    };

    // 1.2.11+ — `--update` path.  Merge the LLM lists
    // with what's already in the user's HJSON (union,
    // case-insensitive dedup, existing entries first
    // so the user's hand-edits keep their position) and
    // apply the merged lists in place via the shared
    // `config_tui::apply_in_place_edits` helper —
    // versioned backup + atomic write + comment
    // preservation come for free.  Then also print the
    // merged snippet to stdout so the user sees what
    // landed.
    let final_lists = if update {
        let existing = load_existing(&cfg, language);
        let merged = Lists {
            linking_verbs: merge(&existing.linking_verbs, &lists.linking_verbs),
            emotion_adjectives: merge(
                &existing.emotion_adjectives,
                &lists.emotion_adjectives,
            ),
            manner_adverbs: merge(
                &existing.manner_adverbs,
                &lists.manner_adverbs,
            ),
            cognition_verbs: merge(
                &existing.cognition_verbs,
                &lists.cognition_verbs,
            ),
        };
        let updates = build_updates(language, &merged);
        match crate::config_tui::apply_in_place_edits(project, &updates) {
            Ok(outcome) => {
                eprintln!(
                    "patched {} (pre-patch backup: {})",
                    outcome.config_path.display(),
                    outcome.backup.display(),
                );
            }
            Err(e) => {
                eprintln!("in-place update failed: {e}");
                eprintln!(
                    "(nothing was written to inkhaven.hjson; pasting the snippet below by hand still works)"
                );
                print_snippet(language, &merged);
                return Ok(());
            }
        }
        merged
    } else {
        lists
    };

    print_snippet(language, &final_lists);
    Ok(())
}

/// 1.2.11+ — pull the live per-language lists out of
/// the loaded `Config`.  Used by `--update` to compute
/// the merge baseline.  An empty configured list means
/// "user is using the built-in default" — we treat it
/// as empty (the union will then be just the LLM
/// output, which is exactly what we want: graduate
/// from built-in to a curated, persisted list).
fn load_existing(cfg: &Config, language: &str) -> Lists {
    let sdt = &cfg.editor.style_warnings.show_dont_tell;
    match language.to_lowercase().as_str() {
        "russian" => Lists {
            linking_verbs: sdt.russian_linking_verbs.clone(),
            emotion_adjectives: sdt.russian_emotion_adjectives.clone(),
            manner_adverbs: sdt.russian_manner_adverbs.clone(),
            cognition_verbs: sdt.russian_cognition_verbs.clone(),
        },
        "french" => Lists {
            linking_verbs: sdt.french_linking_verbs.clone(),
            emotion_adjectives: sdt.french_emotion_adjectives.clone(),
            manner_adverbs: sdt.french_manner_adverbs.clone(),
            cognition_verbs: sdt.french_cognition_verbs.clone(),
        },
        "german" => Lists {
            linking_verbs: sdt.german_linking_verbs.clone(),
            emotion_adjectives: sdt.german_emotion_adjectives.clone(),
            manner_adverbs: sdt.german_manner_adverbs.clone(),
            cognition_verbs: sdt.german_cognition_verbs.clone(),
        },
        "spanish" => Lists {
            linking_verbs: sdt.spanish_linking_verbs.clone(),
            emotion_adjectives: sdt.spanish_emotion_adjectives.clone(),
            manner_adverbs: sdt.spanish_manner_adverbs.clone(),
            cognition_verbs: sdt.spanish_cognition_verbs.clone(),
        },
        _ => Lists {
            linking_verbs: sdt.english_linking_verbs.clone(),
            emotion_adjectives: sdt.english_emotion_adjectives.clone(),
            manner_adverbs: sdt.english_manner_adverbs.clone(),
            cognition_verbs: sdt.english_cognition_verbs.clone(),
        },
    }
}

/// 1.2.11+ — union with case-insensitive dedup,
/// preserving insertion order: existing entries first
/// (so the user's hand-tuned ordering survives), then
/// new arrivals.  Empty / whitespace-only entries are
/// dropped silently.
fn merge(existing: &[String], new: &[String]) -> Vec<String> {
    use std::collections::HashSet;
    let mut seen: HashSet<String> = HashSet::new();
    let mut out: Vec<String> = Vec::with_capacity(existing.len() + new.len());
    let push = |w: &str, out: &mut Vec<String>, seen: &mut HashSet<String>| {
        let trimmed = w.trim();
        if trimmed.is_empty() {
            return;
        }
        let key = trimmed.to_lowercase();
        if seen.insert(key) {
            out.push(trimmed.to_string());
        }
    };
    for w in existing {
        push(w, &mut out, &mut seen);
    }
    for w in new {
        push(w, &mut out, &mut seen);
    }
    out
}

/// 1.2.11+ — build the dotted-path update tuples for
/// `config_tui::apply_in_place_edits`.  Path prefix
/// follows the `serde_hjson` derived shape of
/// `Config::editor::style_warnings::show_dont_tell::<lang>_*`.
fn build_updates(
    language: &str,
    lists: &Lists,
) -> Vec<(String, serde_json::Value)> {
    let lang = language.to_lowercase();
    let prefix = "editor.style_warnings.show_dont_tell";
    let to_value = |words: &Vec<String>| -> serde_json::Value {
        serde_json::Value::Array(
            words
                .iter()
                .map(|w| serde_json::Value::String(w.clone()))
                .collect(),
        )
    };
    vec![
        (
            format!("{prefix}.{lang}_linking_verbs"),
            to_value(&lists.linking_verbs),
        ),
        (
            format!("{prefix}.{lang}_emotion_adjectives"),
            to_value(&lists.emotion_adjectives),
        ),
        (
            format!("{prefix}.{lang}_manner_adverbs"),
            to_value(&lists.manner_adverbs),
        ),
        (
            format!("{prefix}.{lang}_cognition_verbs"),
            to_value(&lists.cognition_verbs),
        ),
    ]
}

const SYSTEM_PROMPT: &str = "\
You are a precise lexicographer assisting an author with their writing \
craft tooling.  Reply with a SINGLE JSON object and nothing else — no \
prose, no preamble, no markdown fences.  Every word you produce must be \
in the requested language (a real word native speakers use), lowercased, \
in its dictionary form (lemma).  Avoid genre-specific slang unless \
explicitly asked.";

fn build_prompt(language: &str, genre: Option<&str>) -> String {
    let genre_line = match genre {
        Some(g) => format!(
            "Bias the vocabulary toward the {g} register — pick words an \
             author working in that mode would actually reach for.\n"
        ),
        None => String::new(),
    };
    format!(
        "Produce vocabulary lists for a show-don't-tell style detector \
         in {language}.  The detector flags four kinds of \"telling\" \
         constructions:\n\n\
         1. `linking_verbs` — copula and quasi-copula verbs used to \
            assert internal state (English equivalents: be, seem, feel, \
            look, appear, become).  Provide ~10-20 lemmas.\n\
         2. `emotion_adjectives` — adjectives that name an emotion \
            outright (English equivalents: angry, sad, afraid, happy, \
            tired, surprised, embarrassed, proud, jealous, lonely, \
            bored, excited, hopeless).  Cover the major emotion \
            families.  Provide ~30-60 lemmas.\n\
         3. `manner_adverbs` — emotion-labelling adverbs (English \
            equivalents: angrily, sadly, nervously, happily).  Provide \
            ~15-30 lemmas.\n\
         4. `cognition_verbs` — verbs that narrate thought instead of \
            showing it (English equivalents: realised, knew, \
            understood, wondered, decided, believed).  Provide ~10-20 \
            lemmas.\n\n\
         {genre_line}\
         Reply with EXACTLY this JSON shape and no other text:\n\n\
         {{\n  \"linking_verbs\":       [\"…\", \"…\"],\n  \
         \"emotion_adjectives\":  [\"…\", \"…\"],\n  \
         \"manner_adverbs\":      [\"…\", \"…\"],\n  \
         \"cognition_verbs\":     [\"…\", \"…\"]\n}}\n"
    )
}

#[derive(Debug, serde::Deserialize)]
struct Lists {
    #[serde(default)]
    linking_verbs: Vec<String>,
    #[serde(default)]
    emotion_adjectives: Vec<String>,
    #[serde(default)]
    manner_adverbs: Vec<String>,
    #[serde(default)]
    cognition_verbs: Vec<String>,
}

fn parse_lists(raw: &str) -> std::result::Result<Lists, String> {
    let trimmed = raw.trim();
    // Strip a leading ``` or ```json fence if the model
    // wrapped the JSON in a code block despite the
    // instruction.  Strip the trailing fence too.
    let stripped = strip_code_fence(trimmed);
    // Pluck the first balanced { … } substring so any
    // stray prose around the JSON doesn't kill the
    // parse.  Brace-balanced rather than first-/last-
    // index so an example object embedded in the
    // model's explanation doesn't break us.
    let json = match extract_json_object(stripped) {
        Some(j) => j,
        None => {
            return Err("no JSON object found in response".to_string());
        }
    };
    serde_json::from_str::<Lists>(json).map_err(|e| format!("invalid JSON: {e}"))
}

fn strip_code_fence(s: &str) -> &str {
    let s = s.trim();
    let s = s.strip_prefix("```json").unwrap_or(s);
    let s = s.strip_prefix("```").unwrap_or(s);
    s.strip_suffix("```").unwrap_or(s).trim()
}

fn extract_json_object(s: &str) -> Option<&str> {
    let bytes = s.as_bytes();
    let start = bytes.iter().position(|&b| b == b'{')?;
    let mut depth: i32 = 0;
    let mut in_string = false;
    let mut escape = false;
    for (i, &b) in bytes.iter().enumerate().skip(start) {
        if in_string {
            if escape {
                escape = false;
            } else if b == b'\\' {
                escape = true;
            } else if b == b'"' {
                in_string = false;
            }
            continue;
        }
        match b {
            b'"' => in_string = true,
            b'{' => depth += 1,
            b'}' => {
                depth -= 1;
                if depth == 0 {
                    return Some(&s[start..=i]);
                }
            }
            _ => {}
        }
    }
    None
}

fn print_snippet(language: &str, lists: &Lists) {
    let lang = language.to_lowercase();
    println!("// Paste under editor.style_warnings.show_dont_tell:");
    println!(
        "// (existing values for other languages stay; empty arrays use built-in defaults.)"
    );
    println!();
    println!("show_dont_tell: {{");
    println!("  enabled: true");
    println!("  use_stemming: true");
    println!();
    print_list(&format!("{lang}_linking_verbs"), &lists.linking_verbs);
    print_list(
        &format!("{lang}_emotion_adjectives"),
        &lists.emotion_adjectives,
    );
    print_list(&format!("{lang}_manner_adverbs"), &lists.manner_adverbs);
    print_list(&format!("{lang}_cognition_verbs"), &lists.cognition_verbs);
    println!("}}");
}

fn print_list(field: &str, words: &[String]) {
    println!("  // Lemmas — Snowball stemming catches inflections.");
    println!("  {field}: [");
    let mut buf = String::from("    ");
    for (i, w) in words.iter().enumerate() {
        let w = w.trim();
        if w.is_empty() {
            continue;
        }
        if i > 0 {
            buf.push(' ');
        }
        buf.push('"');
        buf.push_str(&w.replace('"', "'"));
        buf.push('"');
        if buf.chars().count() > 64 {
            println!("{buf}");
            buf = String::from("    ");
        }
    }
    if buf.trim() != "" {
        println!("{buf}");
    }
    println!("  ]");
    println!();
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn extract_json_handles_chatter() {
        let raw = "Here is the JSON you asked for:\n\n{ \"linking_verbs\": [\"быть\"], \"emotion_adjectives\": [], \"manner_adverbs\": [], \"cognition_verbs\": [] }\n\nLet me know if you'd like more.";
        let json = extract_json_object(raw).unwrap();
        assert!(json.starts_with('{') && json.ends_with('}'));
        let lists: Lists = serde_json::from_str(json).unwrap();
        assert_eq!(lists.linking_verbs, vec!["быть".to_string()]);
    }

    #[test]
    fn extract_json_handles_nested_braces() {
        let raw = "{ \"linking_verbs\": [\"a\"], \"note\": \"object {x: 1}\", \"emotion_adjectives\": [], \"manner_adverbs\": [], \"cognition_verbs\": [] }";
        let json = extract_json_object(raw).unwrap();
        assert!(json.ends_with('}'));
    }

    #[test]
    fn strip_fence_removes_markdown_wrapping() {
        let raw = "```json\n{\"linking_verbs\": [], \"emotion_adjectives\": [], \"manner_adverbs\": [], \"cognition_verbs\": []}\n```";
        let stripped = strip_code_fence(raw);
        assert!(stripped.starts_with('{'));
        assert!(stripped.ends_with('}'));
    }

    #[test]
    fn parse_lists_rejects_garbage() {
        let err = parse_lists("not json at all").unwrap_err();
        assert!(err.contains("no JSON object"));
    }

    #[test]
    fn merge_preserves_existing_order_and_dedups_case_insensitively() {
        let existing = vec![
            "быть".to_string(),
            "казаться".to_string(),
            "выглядеть".to_string(),
        ];
        let new = vec![
            "Казаться".to_string(), // case-insensitive dup
            "оставаться".to_string(),
            "  ".to_string(),       // whitespace dropped
            "становиться".to_string(),
        ];
        let merged = merge(&existing, &new);
        // Existing order preserved at the head.
        assert_eq!(
            merged,
            vec![
                "быть".to_string(),
                "казаться".to_string(),
                "выглядеть".to_string(),
                "оставаться".to_string(),
                "становиться".to_string(),
            ]
        );
    }

    #[test]
    fn merge_handles_empty_existing() {
        // The user hasn't customised this language at
        // all — the merge should reduce to the LLM
        // output, with whitespace trimmed.
        let merged = merge(&[], &[
            "  ser  ".to_string(),
            "estar".to_string(),
        ]);
        assert_eq!(merged, vec!["ser".to_string(), "estar".to_string()]);
    }
}