inkhaven 1.3.3

//! 1.2.10+ — schema model for the standalone TUI
//! configuration editor.
//!
//! Phase 1: types + a tree builder that walks
//! `Config::default()` serialised to `serde_json::Value`
//! to derive every leaf path + default + type.  Metadata
//! (enum variants, min/max ranges, help anchor) lives in
//! a small hand-rolled table in `metadata.rs` and is
//! layered on top of the auto-derived tree at build time.
//!
//! Read-only in Phase 1 — no widgets that mutate, no
//! save, no backup.  Just a *config explorer*.

use serde_json::Value;
use std::collections::BTreeMap;

/// Type of a config leaf (or `Stanza` for branches).
/// Phase 1 inferred shape from the JSON value alone;
/// Phase 6+ also consults a metadata table
/// (`refined_type`) that narrows generic `String`s
/// into `Color` / `Path` / `Enum` based on the path.
#[derive(Debug, Clone)]
pub enum ConfigType {
    Bool,
    Int,
    Float,
    String,
    /// 1.2.10+ — `#RRGGBB` hex color.  Routed when
    /// the path lives under `theme.*` and ends in
    /// `_bg` / `_fg` / `_border` / similar.
    Color,
    /// 1.2.10+ — filesystem path (file or
    /// directory).  Routed for paths whose key
    /// ends in `_dir` / `_directory` / `_path` /
    /// `_file`.
    Path,
    /// 1.2.10+ — fixed-set enum string.  Carries
    /// the allowed values for the picker UI.
    Enum(Vec<&'static str>),
    /// Array of strings — common shape for the
    /// `extra_words` / `*_stop_words` / per-language
    /// lists.
    StringList,
    /// Any array we can't narrow further.  Renders as
    /// JSON.
    Array,
    /// A nested stanza — has children, not a leaf.
    Stanza,
    /// A JSON object that isn't a `Stanza` (i.e. a map
    /// of dynamic keys, like `llm.providers`).  Phase 2
    /// gives it a dedicated map widget.
    #[allow(dead_code)]
    Map,
    /// Anything else — JSON null, mixed, etc.
    Unknown,
}

impl ConfigType {
    pub fn infer(value: &Value) -> Self {
        match value {
            Value::Null => Self::Unknown,
            Value::Bool(_) => Self::Bool,
            Value::Number(n) => {
                if n.is_i64() || n.is_u64() {
                    Self::Int
                } else {
                    Self::Float
                }
            }
            Value::String(_) => Self::String,
            Value::Array(arr) => {
                if arr.iter().all(|v| v.is_string()) {
                    Self::StringList
                } else {
                    Self::Array
                }
            }
            Value::Object(_) => Self::Stanza,
        }
    }

    pub fn label(&self) -> &'static str {
        match self {
            Self::Bool => "bool",
            Self::Int => "int",
            Self::Float => "float",
            Self::String => "string",
            Self::Color => "color (#RRGGBB)",
            Self::Path => "filesystem path",
            Self::Enum(_) => "enum",
            Self::StringList => "list of strings",
            Self::Array => "array",
            Self::Stanza => "stanza",
            Self::Map => "map",
            Self::Unknown => "unknown",
        }
    }
}

/// 1.2.10+ — narrow a JSON-inferred type using
/// path-shape heuristics + a small hand-rolled
/// metadata table.  Called by the tree builder so
/// each leaf carries the most specific type the
/// widget catalog has a picker for.
///
/// Heuristic precedence:
///   1. Exact-path lookup in the enum table.
///   2. Suffix match: `_bg` / `_fg` / `_border`
///      under `theme.*` → Color.
///   3. Suffix match: `_dir` / `_directory` /
///      `_path` / `_file` → Path.
///   4. Fall back to whatever JSON-inferred type
///      the caller already computed.
pub fn refined_type(path: &str, inferred: &ConfigType) -> ConfigType {
    // Enum overrides land first.
    if let Some(variants) = enum_variants_for(path) {
        return ConfigType::Enum(variants);
    }
    // Color overrides — only apply to fields that
    // were inferred as `String` to begin with.
    if matches!(inferred, ConfigType::String) {
        if path.starts_with("theme.")
            && (path.ends_with("_bg")
                || path.ends_with("_fg")
                || path.ends_with("_border"))
        {
            return ConfigType::Color;
        }
        if path.ends_with("_directory")
            || path.ends_with("_dir")
            || path.ends_with("_path")
            || path.ends_with("_file")
        {
            return ConfigType::Path;
        }
    }
    inferred.clone()
}

fn enum_variants_for(path: &str) -> Option<Vec<&'static str>> {
    match path {
        "typst_compile.engine" => Some(vec!["external", "inprocess"]),
        "embeddings.model" => Some(vec![
            "MultilingualE5Small",
            "MultilingualE5Base",
            "MultilingualE5Large",
            "BGEM3",
        ]),
        // 1.2.12+ — prompt-language resolver mode.  See
        // `Documentation/PROPOSALS/MULTILINGUAL_PROMPTS.md` §5.
        "editor.prompt_language_mode" => {
            Some(vec!["book_defined", "paragraph_detected"])
        }
        // 1.2.11+ — project working language.  Drives
        // per-language stop-word + stemmer + SDT-list +
        // prompt-resolver paths across the binary.  The
        // five supported values match what inkhaven
        // actually ships plumbing for; any other value
        // silently falls back to English.
        "language" => Some(vec![
            "english",
            "russian",
            "french",
            "german",
            "spanish",
        ]),
        // 1.2.11+ — typst's `#set text(lang: ...)` for
        // hyphenation + smart-quote behaviour.  ISO 639-1
        // two-letter codes; the five here match
        // inkhaven's supported languages so the picker
        // stays small + consistent with the top-level
        // `language` field.  Users wanting an exotic
        // typst language (it/pt/ja/zh/…) can still set
        // it via `Ctrl+B 0` raw HJSON edit.
        "typst_page.language" => {
            Some(vec!["en", "ru", "fr", "de", "es"])
        }
        // 1.2.11+ — typst page-number format.  The set
        // covers every common form: empty (no numbers,
        // typst default), arabic, lowercase / uppercase
        // roman, lowercase / uppercase letter, and the
        // "1 of 1" with-total form.  Custom prefixes
        // (e.g. "Chapter 1.1") fall outside this set and
        // need the raw HJSON edit.
        "typst_page.page_numbering" => Some(vec![
            "",
            "1",
            "i",
            "I",
            "a",
            "A",
            "1 of 1",
        ]),
        _ => None,
    }
}

/// Source of a leaf's current value at file-load time.
///
///   * `Default`      — not present in `inkhaven.hjson`;
///                      the displayed value is the
///                      built-in default.
///   * `Configured`   — present in HJSON, value taken
///                      from the file.
///   * `Unknown`      — present in HJSON, but **not in
///                      the schema** — i.e. a user-added
///                      field outside the inkhaven
///                      schema.  Preserved on save (see
///                      proposal §6.6) but not editable.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ValueSource {
    Default,
    Configured,
    // Reserved for Phase 2 — the surgical-rewrite save
    // pipeline will tag values that came from outside
    // the schema.  Phase 1 routes those into the
    // `unknowns` Vec instead.
    #[allow(dead_code)]
    Unknown,
}

/// One node in the config tree.  Both leaves and
/// stanzas use the same shape; `children` is empty for
/// leaves.
#[derive(Debug, Clone)]
pub struct SchemaNode {
    /// Dotted path from the root of `Config`, e.g.
    /// `editor.style_warnings.enabled`.  Empty string
    /// for the synthetic root.
    pub path: String,
    /// Display name — the last path segment, or
    /// `"<root>"` for the root.
    pub display: String,
    /// Inferred type.  Stanzas have `Stanza`; leaves
    /// have their JSON-derived type.
    pub ty: ConfigType,
    /// Default value as JSON.  For stanzas: `Object`
    /// (recursive defaults of children).  For leaves:
    /// the scalar / array default.
    pub default: Value,
    /// Current effective value: configured value if
    /// present in HJSON, else default.
    pub current: Value,
    /// Where `current` came from.
    pub source: ValueSource,
    /// Child nodes for stanzas; empty for leaves.
    /// Sorted by display name.
    pub children: Vec<SchemaNode>,
}

impl SchemaNode {
    pub fn is_leaf(&self) -> bool {
        self.children.is_empty()
            && !matches!(self.ty, ConfigType::Stanza | ConfigType::Map)
    }

    /// Flatten the tree into `(path, depth, &node)` in
    /// pre-order, skipping subtrees whose path is in
    /// `collapsed`.  Used by the tree renderer.
    pub fn flatten<'a>(
        &'a self,
        collapsed: &std::collections::HashSet<String>,
        out: &mut Vec<(usize, &'a SchemaNode)>,
        depth: usize,
    ) {
        out.push((depth, self));
        if collapsed.contains(&self.path) {
            return;
        }
        for child in &self.children {
            child.flatten(collapsed, out, depth + 1);
        }
    }
}

/// Build a schema tree by walking the JSON
/// representation of `Config::default()`.
///
/// `live` is the parsed HJSON from the project's
/// `inkhaven.hjson` (or `Value::Object(Map::new())` when
/// the file doesn't exist).  Each tree node's `source`
/// is set by checking whether the path is present in
/// `live`:
///   * Present + scalar match: `Configured` (value =
///     live value).
///   * Absent: `Default` (value = built-in default).
///   * Stanza: `source` rolls up via the children — a
///     stanza is `Configured` if any leaf descendant is.
///
/// Unknown fields (present in `live` but absent from the
/// defaults tree) are returned separately as a flat
/// `Vec<(path, Value)>` so the top-bar chip can show
/// the count; they do NOT appear in the schema tree
/// (see proposal §6.6).
pub fn build(
    defaults: &Value,
    live: &Value,
) -> (SchemaNode, Vec<(String, Value)>) {
    let mut unknowns: Vec<(String, Value)> = Vec::new();
    let root = build_node(
        "",
        "<root>",
        defaults,
        live,
        &mut unknowns,
    );
    (root, unknowns)
}

/// 1.2.10+ — known map-shaped config paths.  Children
/// of these paths use dynamic keys (HashMap<String, T>
/// in the Rust schema), so user-added keys in the
/// live HJSON are valid map entries — NOT unknown
/// fields.
///
/// Phase 4 ships a single entry: `llm.providers`.
/// Add new entries here as new `HashMap`-shaped
/// stanzas land in the `Config` struct.
const KNOWN_MAP_PATHS: &[&str] = &["llm.providers"];

pub fn is_known_map_path(path: &str) -> bool {
    KNOWN_MAP_PATHS.contains(&path)
}

/// Recursively force a sub-tree to
/// `ValueSource::Configured`.  Used for user-added
/// map entries — every leaf came from the live HJSON,
/// not from a built-in default, so the source rolls
/// up uniformly.
fn force_configured(node: &mut SchemaNode) {
    node.source = ValueSource::Configured;
    for child in &mut node.children {
        force_configured(child);
    }
}

fn build_node(
    path: &str,
    display: &str,
    default: &Value,
    live: &Value,
    unknowns: &mut Vec<(String, Value)>,
) -> SchemaNode {
    let ty = ConfigType::infer(default);
    match default {
        Value::Object(default_map) => {
            // Build children from the default map; walk
            // live in parallel to find unknown keys.
            let live_map = live.as_object();
            let mut children: Vec<SchemaNode> =
                Vec::with_capacity(default_map.len());
            let mut any_configured = false;
            for (key, child_default) in default_map {
                let child_path = if path.is_empty() {
                    key.clone()
                } else {
                    format!("{path}.{key}")
                };
                let child_live = live_map
                    .and_then(|m| m.get(key))
                    .cloned()
                    .unwrap_or_else(|| child_default.clone());
                let child = build_node(
                    &child_path,
                    key,
                    child_default,
                    &child_live,
                    unknowns,
                );
                if child.source == ValueSource::Configured {
                    any_configured = true;
                }
                children.push(child);
            }
            // Detect live keys that aren't in
            // defaults.  Two routes:
            //
            //   * At a **known map path** (e.g.
            //     `llm.providers`) — these are valid
            //     map entries with dynamic keys.
            //     Build them into the tree using any
            //     existing default entry as a
            //     template; force them to Configured.
            //   * Anywhere else — they're unknown
            //     user-added fields and get routed to
            //     `unknowns` (preserved on save,
            //     never edited).
            if let Some(map) = live_map {
                let map_here = is_known_map_path(path);
                for (key, value) in map {
                    if default_map.contains_key(key) {
                        continue;
                    }
                    let child_path = if path.is_empty() {
                        key.clone()
                    } else {
                        format!("{path}.{key}")
                    };
                    if map_here {
                        // Template = any default
                        // entry's shape.  Fall back
                        // to the live value's shape
                        // when the defaults are
                        // empty.
                        let template: Value = default_map
                            .values()
                            .next()
                            .cloned()
                            .unwrap_or_else(|| value.clone());
                        let mut child = build_node(
                            &child_path,
                            key,
                            &template,
                            value,
                            unknowns,
                        );
                        force_configured(&mut child);
                        any_configured = true;
                        children.push(child);
                    } else {
                        collect_unknowns(&child_path, value, unknowns);
                    }
                }
            }
            // Sort children deterministically — keep
            // serde_json's key order for now (HJSON
            // field order tends to match struct field
            // order).  Already sorted by BTreeMap or
            // declaration order.
            children.sort_by(|a, b| a.display.cmp(&b.display));
            let source = if any_configured {
                ValueSource::Configured
            } else {
                ValueSource::Default
            };
            SchemaNode {
                path: path.to_string(),
                display: display.to_string(),
                ty,
                default: default.clone(),
                current: live.clone(),
                source,
                children,
            }
        }
        _ => {
            // Leaf: classify by source.
            let source = if live == default {
                // Could be "user set it explicitly to
                // the same value as the default" — but
                // without the byte-range index (Phase 2)
                // we can't distinguish.  Phase 1
                // policy: equal-to-default counts as
                // Default.  Phase 2's surgical rewrite
                // tracks explicit presence.
                ValueSource::Default
            } else {
                ValueSource::Configured
            };
            // 1.2.10+ — narrow generic String leaves
            // into Color / Path / Enum where the
            // path-shape metadata table says so.
            let refined_ty = refined_type(path, &ty);
            SchemaNode {
                path: path.to_string(),
                display: display.to_string(),
                ty: refined_ty,
                default: default.clone(),
                current: live.clone(),
                source,
                children: Vec::new(),
            }
        }
    }
}

fn collect_unknowns(
    path: &str,
    value: &Value,
    unknowns: &mut Vec<(String, Value)>,
) {
    match value {
        Value::Object(map) => {
            for (key, child) in map {
                let child_path = format!("{path}.{key}");
                collect_unknowns(&child_path, child, unknowns);
            }
        }
        _ => unknowns.push((path.to_string(), value.clone())),
    }
}

/// Walk the schema tree and return a map `path → node`
/// for direct lookup (used by the help pane + future
/// surgical-rewrite save pipeline).  Skips the synthetic
/// root.  Unused in Phase 1; the help-pane lookup goes
/// through `App::current_node()`.  Kept public so Phase
/// 2's save pipeline can index without re-walking.
#[allow(dead_code)]
pub fn index_by_path(root: &SchemaNode) -> BTreeMap<String, &SchemaNode> {
    let mut out = BTreeMap::new();
    fn walk<'a>(
        node: &'a SchemaNode,
        out: &mut BTreeMap<String, &'a SchemaNode>,
    ) {
        if !node.path.is_empty() {
            out.insert(node.path.clone(), node);
        }
        for child in &node.children {
            walk(child, out);
        }
    }
    walk(root, &mut out);
    out
}

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

    #[test]
    fn empty_live_marks_everything_default() {
        let defaults = json!({
            "editor": {
                "autosave_seconds": 5,
                "wrap": true,
            },
        });
        let live = json!({});
        let (root, unknowns) = build(&defaults, &live);
        let editor = &root.children[0];
        assert_eq!(editor.path, "editor");
        assert_eq!(editor.source, ValueSource::Default);
        for leaf in &editor.children {
            assert_eq!(leaf.source, ValueSource::Default);
        }
        assert!(unknowns.is_empty());
    }

    #[test]
    fn configured_leaf_marked_configured() {
        let defaults = json!({
            "editor": { "autosave_seconds": 5 },
        });
        let live = json!({
            "editor": { "autosave_seconds": 30 },
        });
        let (root, _) = build(&defaults, &live);
        let editor = &root.children[0];
        assert_eq!(editor.source, ValueSource::Configured);
        let leaf = &editor.children[0];
        assert_eq!(leaf.source, ValueSource::Configured);
        assert_eq!(leaf.current, json!(30));
        assert_eq!(leaf.default, json!(5));
    }

    #[test]
    fn unknown_field_collected_not_in_tree() {
        let defaults = json!({ "editor": { "wrap": true } });
        let live = json!({
            "editor": { "wrap": false },
            "experimental": { "my_custom_flag": "yes" },
        });
        let (root, unknowns) = build(&defaults, &live);
        // Only "editor" in the tree.
        assert_eq!(root.children.len(), 1);
        assert_eq!(root.children[0].display, "editor");
        // The unknown field is reported separately.
        assert_eq!(unknowns.len(), 1);
        assert_eq!(unknowns[0].0, "experimental.my_custom_flag");
    }

    #[test]
    fn type_inference_recognises_string_list() {
        let defaults = json!({
            "editor": { "extra_words": ["just", "very"] },
        });
        let (root, _) = build(&defaults, &json!({}));
        let editor = &root.children[0];
        let leaf = &editor.children[0];
        assert!(matches!(leaf.ty, ConfigType::StringList));
    }

    #[test]
    fn flatten_skips_collapsed_subtree() {
        let defaults = json!({
            "a": { "b": 1, "c": 2 },
            "d": 3,
        });
        let (root, _) = build(&defaults, &json!({}));
        let mut out = Vec::new();
        let mut collapsed = std::collections::HashSet::new();
        collapsed.insert("a".to_string());
        root.flatten(&collapsed, &mut out, 0);
        // root + a + d (a's children skipped)
        let displays: Vec<&str> =
            out.iter().map(|(_, n)| n.display.as_str()).collect();
        assert_eq!(displays, vec!["<root>", "a", "d"]);
    }

    #[test]
    fn index_by_path_skips_root() {
        let defaults = json!({
            "editor": { "wrap": true },
        });
        let (root, _) = build(&defaults, &json!({}));
        let idx = index_by_path(&root);
        assert!(idx.contains_key("editor"));
        assert!(idx.contains_key("editor.wrap"));
        assert!(!idx.contains_key(""));
    }

    #[test]
    fn user_added_provider_appears_under_known_map_path() {
        // `llm.providers` is a known map path.  A
        // user-added provider name in the live HJSON
        // should appear in the schema tree, NOT in
        // `unknowns`.
        let defaults = json!({
            "llm": {
                "providers": {
                    "gemini": { "model": "gemini-2.5-pro", "api_key_env": "GEMINI_API_KEY" }
                }
            }
        });
        let live = json!({
            "llm": {
                "providers": {
                    "gemini": { "model": "gemini-2.5-pro", "api_key_env": "GEMINI_API_KEY" },
                    "ollama_remote": { "model": "llama3.2", "api_key_env": "OLLAMA_KEY" }
                }
            }
        });
        let (root, unknowns) = build(&defaults, &live);
        // The user-added provider must NOT be in unknowns.
        assert!(
            unknowns.iter().all(|(p, _)| !p.starts_with("llm.providers.ollama_remote")),
            "expected llm.providers.ollama_remote to live in the tree, not unknowns; got: {unknowns:?}"
        );
        // It must be reachable via index_by_path.
        let idx = index_by_path(&root);
        assert!(idx.contains_key("llm.providers.ollama_remote"));
        assert!(idx.contains_key("llm.providers.ollama_remote.model"));
        assert!(idx.contains_key("llm.providers.ollama_remote.api_key_env"));
    }

    #[test]
    fn user_added_provider_marked_configured() {
        let defaults = json!({
            "llm": {
                "providers": {
                    "gemini": { "model": "gemini-2.5-pro", "api_key_env": "GEMINI_API_KEY" }
                }
            }
        });
        let live = json!({
            "llm": {
                "providers": {
                    "gemini": { "model": "gemini-2.5-pro", "api_key_env": "GEMINI_API_KEY" },
                    "ollama_remote": { "model": "llama3.2", "api_key_env": "OLLAMA_KEY" }
                }
            }
        });
        let (root, _) = build(&defaults, &live);
        let idx = index_by_path(&root);
        let entry = idx.get("llm.providers.ollama_remote").unwrap();
        assert_eq!(entry.source, ValueSource::Configured);
        // Leaves under the user-added provider are
        // also forced to Configured.
        let model = idx.get("llm.providers.ollama_remote.model").unwrap();
        assert_eq!(model.source, ValueSource::Configured);
    }

    #[test]
    fn unknown_path_outside_map_still_collected() {
        // Sanity: only `llm.providers` is map-shaped;
        // a top-level unknown still goes to unknowns.
        let defaults = json!({ "editor": { "wrap": true } });
        let live = json!({
            "editor": { "wrap": true },
            "experimental": { "my_flag": "yes" }
        });
        let (_, unknowns) = build(&defaults, &live);
        assert!(unknowns.iter().any(|(p, _)| p == "experimental.my_flag"));
    }

    #[test]
    fn map_path_with_extra_field_in_entry_still_reports_unknown() {
        // Inside a map entry, fields that aren't in
        // the template are STILL unknown — the user
        // probably typo'd `api_token` for
        // `api_key_env`.
        let defaults = json!({
            "llm": {
                "providers": {
                    "gemini": { "model": "x", "api_key_env": "Y" }
                }
            }
        });
        let live = json!({
            "llm": {
                "providers": {
                    "gemini": { "model": "x", "api_key_env": "Y" },
                    "custom": { "model": "z", "api_token": "T" }
                }
            }
        });
        let (root, unknowns) = build(&defaults, &live);
        let idx = index_by_path(&root);
        // The provider exists in the tree.
        assert!(idx.contains_key("llm.providers.custom"));
        // But api_token (not in the template) is
        // routed to unknowns.
        assert!(
            unknowns.iter().any(|(p, _)| p == "llm.providers.custom.api_token"),
            "expected typo'd field to land in unknowns; got: {unknowns:?}"
        );
    }
}