usage-lib 6.8.0

use crate::docs::markdown::tera::TERA;
use crate::docs::models::Spec;
use crate::error::UsageErr;
use itertools::Itertools;
use regex::Regex;
use std::sync::LazyLock;

/// One of the templates used to generate Markdown documentation.
///
/// A renderer starts with a complete built-in template set. Replacing one member keeps the
/// others available, including through Tera's `{% include %}` directive. This lets an adopter
/// change the document shell or the presentation of one kind of item without copying the whole
/// theme.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum MarkdownTemplate {
    /// A single-file document containing the root and every subcommand.
    Spec,
    /// The landing page generated in multi-file mode.
    Index,
    /// A command and its arguments, flags, outputs, exits, and examples.
    Command,
    /// The details beneath one positional argument.
    Argument,
    /// The details beneath one flag.
    Flag,
    /// The configuration reference appended to a spec or written in multi-file mode.
    Config,
}

/// The built-in presentation used for generated Markdown.
#[derive(Debug, Default, Clone, Copy, PartialEq, Eq)]
pub enum MarkdownTheme {
    /// Dense grouped lists intended for scanning a large command reference.
    #[default]
    Compact,
    /// Give every argument and flag its own addressable heading and detail block.
    Detailed,
}

impl MarkdownTemplate {
    fn name(self) -> &'static str {
        match self {
            Self::Spec => "spec_template.md.tera",
            Self::Index => "index_template.md.tera",
            Self::Command => "cmd_template.md.tera",
            Self::Argument => "arg_template.md.tera",
            Self::Flag => "flag_template.md.tera",
            Self::Config => "config_template.md.tera",
        }
    }
}

/// A backtick span, or a bare `<` outside one.
static CODE_SPAN_OR_LT: LazyLock<Regex> = LazyLock::new(|| Regex::new(r"(`[^`]*`)|(<)").unwrap());

fn escape_md_with_indent(value: &str, html_encode: bool, indent: bool) -> String {
    let mut fence: Option<(char, usize)> = None;
    // Help text is allowed to contain terminal styling. clap-era applications commonly build
    // their examples with `color_print::cstr!`, which embeds SGR sequences even when color is
    // disabled at runtime. Terminal styling has no meaning in generated Markdown, and leaving
    // it here publishes literal escape bytes in docs and downstream static sites.
    let value = crate::docs::strip_ansi(value);

    value
        .lines()
        .enumerate()
        .map(|(index, line)| {
            let line = if !html_encode {
                line.to_string()
            } else {
                // Preserve indented code, including code nested within list items.
                if line.starts_with("    ") {
                    line.to_string()
                } else if let Some((marker, length)) = fence {
                    let trimmed = line.trim();
                    if trimmed.len() >= length && trimmed.chars().all(|c| c == marker) {
                        fence = None;
                    }
                    line.to_string()
                // Converted blocks may require longer fences to contain literal backticks.
                } else if line.trim_start().starts_with("```")
                    || line.trim_start().starts_with("~~~")
                {
                    let trimmed = line.trim_start();
                    let marker = trimmed.chars().next().unwrap();
                    fence = Some((marker, trimmed.chars().take_while(|c| *c == marker).count()));
                    line.to_string()
                } else {
                    // replace '<' with '&lt;' but not inside code blocks
                    CODE_SPAN_OR_LT
                        .replace_all(line, |caps: &regex::Captures| {
                            if caps.get(1).is_some() {
                                caps.get(1).unwrap().as_str().to_string()
                            } else {
                                "&lt;".to_string()
                            }
                        })
                        .to_string()
                }
            };
            if indent && index > 0 && !line.is_empty() {
                format!("  {line}")
            } else {
                line
            }
        })
        .join("\n")
}

fn escape_md(value: &str, html_encode: bool) -> String {
    escape_md_with_indent(value, html_encode, false)
}

#[derive(Debug, Clone)]
pub struct MarkdownRenderer {
    /// The spec as its author wrote it, before any rendering.
    ///
    /// Kept because rendering depends on the builder options, and the builders run *after*
    /// [`Self::new`]. Rendering eagerly in `new` read options nobody had set yet, and because
    /// rendering marks each item done, the later pass with the real options no-opped — which
    /// is how `--indented-blocks-to-code-fences` came to do nothing at all on single-file
    /// output. The rendered model is derived from this on first use instead, so it always sees
    /// the options the caller actually asked for.
    raw: crate::Spec,
    /// The rendered docs model, derived from [`Self::raw`] on first use by [`Self::spec`].
    ///
    /// Every builder clears this, so an option set after a render still takes effect.
    ///
    /// A `OnceLock` rather than a `OnceCell` so the renderer stays `Sync` and `RefUnwindSafe`
    /// for callers that hold one across threads — a `OnceCell` here would take those away.
    spec: std::sync::OnceLock<Spec>,
    pub(crate) header_level: usize,
    pub(crate) multi: bool,
    url_prefix: Option<String>,
    link_extension: String,
    html_encode: bool,
    indented_blocks_to_code_fences: bool,
    theme: MarkdownTheme,
    templates: Vec<(MarkdownTemplate, String)>,
}

impl MarkdownRenderer {
    pub fn new(spec: crate::Spec) -> Self {
        Self {
            raw: spec,
            spec: std::sync::OnceLock::new(),
            header_level: 1,
            multi: false,
            url_prefix: None,
            link_extension: ".md".into(),
            html_encode: true,
            indented_blocks_to_code_fences: false,
            theme: MarkdownTheme::default(),
            templates: Vec::new(),
        }
    }

    /// The rendered docs model, built on first use from the options set by then.
    pub(crate) fn spec(&self) -> &Spec {
        self.spec.get_or_init(|| {
            // By reference: `From<&Spec>` already clones internally, so handing it an owned
            // spec clones the whole tree twice.
            let mut spec = Spec::from(&self.raw);
            spec.render_md(self);
            spec
        })
    }

    /// Apply a builder option and drop any model rendered under the old options.
    ///
    /// Every `with_*` goes through this. A builder that set its field directly would leave a
    /// model rendered without its option behind, which is the bug this type already had once.
    fn with(mut self, set: impl FnOnce(&mut Self)) -> Self {
        set(&mut self);
        self.spec = std::sync::OnceLock::new();
        self
    }

    /// The file name the settings page gets in `--multi` mode.
    ///
    /// One name, always. `settings.md` was the obvious choice and the wrong one: mise has a
    /// `settings` command, whose own page lands there, and config is written second — so for
    /// the very CLI this feature is aimed at, the command's page was silently replaced and the
    /// index linked both entries to the same file.
    ///
    /// Choosing between two names by whether a command is in the way would fix that and
    /// introduce something worse: the name would move when a command is added or removed
    /// between runs, leaving the abandoned one behind as a stale page nothing links but the
    /// docs site still serves. A fixed name cannot do that, and `configuration` is a name CLIs
    /// give to a *file*, not to a command — the command is called `config`.
    pub fn config_page(&self) -> &'static str {
        "configuration.md"
    }

    /// A visible top-level command whose own page would be written to [`Self::config_page`].
    ///
    /// Vanishingly unlikely, and silence is what made the `settings.md` collision hard to see,
    /// so the one case left is reported rather than guessed at.
    pub fn config_page_collision(&self) -> Option<&str> {
        let stem = self.config_page().trim_end_matches(".md");
        self.spec()
            .cmd
            .subcommands
            .values()
            .find(|cmd| !cmd.hide && cmd.full_cmd == [stem])
            .map(|cmd| cmd.name.as_str())
    }

    pub fn with_header_level(self, header_level: usize) -> Self {
        self.with(|r| r.header_level = header_level)
    }

    pub fn with_multi(self, index: bool) -> Self {
        self.with(|r| r.multi = index)
    }

    pub fn with_url_prefix<S: Into<String>>(self, url_prefix: S) -> Self {
        self.with(|r| r.url_prefix = Some(url_prefix.into()))
    }

    pub fn with_html_encode(self, html_encode: bool) -> Self {
        self.with(|r| r.html_encode = html_encode)
    }

    /// Extension for generated page links, including the dot (default `.md`).
    /// Use `.html` for rendered websites or an empty string for extensionless URLs.
    /// This does not change the names of generated Markdown files.
    pub fn with_link_extension(self, extension: impl Into<String>) -> Self {
        self.with(|r| r.link_extension = extension.into())
    }

    /// Turn four-space indented blocks in help text into fenced code blocks.
    pub fn with_indented_blocks_to_code_fences(self, indented_blocks_to_code_fences: bool) -> Self {
        self.with(|r| r.indented_blocks_to_code_fences = indented_blocks_to_code_fences)
    }

    /// The former name of [`Self::with_indented_blocks_to_code_fences`]. Prefer that one.
    ///
    /// A misnomer from the start: no `<pre>` tag was ever involved. Kept, rather than renamed
    /// out from under callers, so an existing docs build kept compiling — and left in the
    /// public API rather than `#[doc(hidden)]`, because hiding it is itself the kind of
    /// removal that costs downstreams a major version.
    pub fn with_replace_pre_with_code_fences(self, indented_blocks_to_code_fences: bool) -> Self {
        self.with_indented_blocks_to_code_fences(indented_blocks_to_code_fences)
    }

    /// Select a built-in Markdown presentation.
    pub fn with_theme(self, theme: MarkdownTheme) -> Self {
        self.with(|r| r.theme = theme)
    }

    /// Replace one built-in Markdown template.
    ///
    /// Templates use [Tera](https://keats.github.io/tera/). A replacement may include any of the
    /// templates it did not replace; for example, a custom [`MarkdownTemplate::Spec`] can still
    /// contain `{% include "cmd_template.md.tera" %}`. Replacing the same member more than once
    /// uses the last value. Syntax and include errors are returned by the render method.
    pub fn with_template(self, template: MarkdownTemplate, source: impl Into<String>) -> Self {
        let source = source.into();
        self.with(|r| {
            if let Some((_, current)) = r
                .templates
                .iter_mut()
                .find(|(current, _)| *current == template)
            {
                *current = source;
            } else {
                r.templates.push((template, source));
            }
        })
    }

    fn tera_ctx(&self) -> tera::Context {
        let mut ctx = tera::Context::new();
        ctx.insert("spec", self.spec());
        ctx.insert("header_level", &self.header_level);
        ctx.insert("multi", &self.multi);
        ctx.insert("url_prefix", &self.url_prefix);
        ctx.insert("link_extension", &self.link_extension);
        ctx.insert(
            "config_link",
            &format!("configuration{}", self.link_extension),
        );
        ctx.insert("html_encode", &self.html_encode);
        ctx
    }

    /// Render with values that belong only to this page.
    ///
    /// A page used to clone the whole renderer — including the complete command tree — merely
    /// to insert one local into its stored context. Multi-page output paid that deep clone once
    /// per command. The context already has to be materialized for Tera, so enrich that directly.
    pub(crate) fn render_with(
        &self,
        template_name: &str,
        enrich: impl FnOnce(&mut tera::Context),
    ) -> Result<String, UsageErr> {
        let mut tera = match self.theme {
            MarkdownTheme::Compact => TERA.clone(),
            MarkdownTheme::Detailed => crate::docs::markdown::tera::DETAILED_TERA.clone(),
        };

        for (template, source) in &self.templates {
            tera.add_raw_template(template.name(), source)?;
        }

        let html_encode = self.html_encode;
        tera.register_filter(
            "escape_md",
            move |value: &tera::Value,
                  _: tera::Kwargs,
                  _: &tera::State|
                  -> tera::TeraResult<String> {
                let value = value.as_str().unwrap();
                let value = escape_md(value, html_encode);
                Ok(value)
            },
        );
        tera.register_filter(
            "escape_md_indented",
            move |value: &tera::Value,
                  _: tera::Kwargs,
                  _: &tera::State|
                  -> tera::TeraResult<String> {
                let value = value.as_str().unwrap();
                Ok(escape_md_with_indent(value, html_encode, true))
            },
        );

        let mut ctx = self.tera_ctx();
        enrich(&mut ctx);
        Ok(tera.render(template_name, &ctx)?)
    }

    /// Rewrite four-space indented blocks as fenced ones, when the caller asked for it.
    pub(crate) fn fence_indented_blocks(&self, md: String) -> String {
        if !self.indented_blocks_to_code_fences {
            return md;
        }
        use pulldown_cmark::{CodeBlockKind, Event, Parser, Tag, TagEnd};
        let mut edits = Vec::new();
        let mut block = None;
        for (event, range) in Parser::new(&md).into_offset_iter() {
            match event {
                Event::Start(Tag::CodeBlock(CodeBlockKind::Indented)) => {
                    block = Some((range, String::new()));
                }
                Event::Text(text) => {
                    if let Some((_, content)) = &mut block {
                        content.push_str(&text);
                    }
                }
                Event::End(TagEnd::CodeBlock) => {
                    if let Some((range, content)) = block.take() {
                        let start = md[..range.start].rfind('\n').map_or(0, |i| i + 1);
                        // A same-line list marker or blockquote prefix belongs to its
                        // container. Keep these blocks indented rather than replacing
                        // the container along with the code block.
                        if md[start..range.start].chars().any(|c| !c.is_whitespace()) {
                            continue;
                        }
                        let source_line = md[start..].lines().next().unwrap_or_default();
                        let source_indent =
                            source_line.len() - source_line.trim_start_matches(' ').len();
                        let content_line = content
                            .lines()
                            .find(|line| !line.trim().is_empty())
                            .unwrap_or_default();
                        let content_indent =
                            content_line.len() - content_line.trim_start_matches(' ').len();
                        let prefix = " ".repeat(source_indent.saturating_sub(4 + content_indent));
                        let fence = "`".repeat(
                            content
                                .split(|c| c != '`')
                                .map(str::len)
                                .max()
                                .unwrap_or(0)
                                .max(2)
                                + 1,
                        );
                        let mut replacement = format!("{prefix}{fence}\n");
                        for line in content.lines() {
                            replacement.push_str(&format!("{prefix}{line}\n"));
                        }
                        replacement.push_str(&format!("{prefix}{fence}\n"));
                        edits.push((start..range.end, replacement));
                    }
                }
                _ => {}
            }
        }
        let mut result = md;
        for (range, replacement) in edits.into_iter().rev() {
            result.replace_range(range, &replacement);
        }
        result
    }
}

#[cfg(test)]
mod tests {
    use super::{escape_md, MarkdownRenderer, MarkdownTemplate};
    use pretty_assertions::assert_eq;

    #[test]
    fn conversion_preserves_markdown_structure_and_code_contents() {
        use pulldown_cmark::{CodeBlockKind, Event, Parser, Tag};
        let renderer = MarkdownRenderer::new("bin ex".parse().unwrap())
            .with_indented_blocks_to_code_fences(true);
        for source in [
            "- outer\n  - inner\n\n        first\n\n          indented\n\n    prose\n",
            "```json\n{\n    \"key\": 1\n}\n```\n",
            "    [tools]\n    node = \"20\"\n\n    ```literal```\n",
            "    first\n\n      second\n",
            ">     quoted code\n>\n>       indented\n",
            "-     same-line list code\n",
            "Examples:\n    paragraph continuation\n\n    actual code\n",
        ] {
            let normalized = |text: &str| {
                pulldown_cmark::TextMergeStream::new(Parser::new(text))
                    .map(|event| match event {
                        Event::Start(Tag::CodeBlock(_)) => {
                            Event::Start(Tag::CodeBlock(CodeBlockKind::Indented))
                        }
                        event => event,
                    })
                    .map(Event::into_static)
                    .collect::<Vec<_>>()
            };
            let output = renderer.fence_indented_blocks(source.into());
            assert_eq!(normalized(source), normalized(&output), "{output}");
            assert_eq!(renderer.fence_indented_blocks(output.clone()), output);
        }
    }

    #[test]
    fn legacy_examples_with_markdown_separation_form_one_code_block() {
        let renderer = MarkdownRenderer::new("bin ex".parse().unwrap())
            .with_indented_blocks_to_code_fences(true);
        let source = "Examples:\n\n    # first\n    ex first\n\n    # second\n    ex second\n";
        assert_eq!(
            renderer.fence_indented_blocks(source.into()),
            "Examples:\n\n```\n# first\nex first\n\n# second\nex second\n```\n"
        );
        let spec =
            crate::Spec::parse_file(std::path::Path::new("../benches/mise.usage.kdl")).unwrap();
        let page = MarkdownRenderer::new(spec.clone())
            .with_indented_blocks_to_code_fences(true)
            .render_cmd(&spec.cmd.subcommands["settings"])
            .unwrap();
        assert!(
            page.contains("Examples:\n\n```\n# list all settings"),
            "{page}"
        );
    }

    #[test]
    fn visible_children_get_a_heading_and_index_has_one_synopsis() {
        let spec: crate::Spec = "bin ex\ncmd a hide=#true\ncmd b\n".parse().unwrap();
        for theme in [
            super::MarkdownTheme::Compact,
            super::MarkdownTheme::Detailed,
        ] {
            let renderer = MarkdownRenderer::new(spec.clone())
                .with_theme(theme)
                .with_multi(true);
            assert!(renderer
                .render_cmd(&spec.cmd)
                .unwrap()
                .contains("## Subcommands"));
            assert_eq!(
                renderer
                    .render_index()
                    .unwrap()
                    .matches("**Usage:**")
                    .count(),
                1
            );
            let hidden: crate::Spec = "bin ex\ncmd a hide=#true\n".parse().unwrap();
            assert!(!MarkdownRenderer::new(hidden.clone())
                .with_theme(theme)
                .with_multi(true)
                .render_cmd(&hidden.cmd)
                .unwrap()
                .contains("## Subcommands"));
        }
    }

    #[test]
    fn page_links_follow_extension_without_renaming_files() {
        let spec: crate::Spec = "bin ex\ncmd go\nconfig {\n prop jobs type=\"uint\"\n}\n"
            .parse()
            .unwrap();
        for extension in [".md", ".html", ""] {
            let renderer = MarkdownRenderer::new(spec.clone())
                .with_url_prefix("/cli")
                .with_link_extension(extension);
            let index = renderer.render_index().unwrap();
            assert!(index.contains(&format!("(/cli/go{extension})")), "{index}");
            assert!(
                index.contains(&format!("(/cli/configuration{extension})")),
                "{index}"
            );
            assert_eq!(renderer.config_page(), "configuration.md");
            let custom = renderer.with_template(MarkdownTemplate::Index, "{{ link_extension }}");
            assert_eq!(custom.render_index().unwrap(), extension);
        }
    }

    #[test]
    fn escapes_html_around_fenced_code_blocks() {
        let input = "before <\n```\ninside <\n```  \nafter <";
        let expected = "before &lt;\n```\ninside <\n```  \nafter &lt;";

        assert_eq!(escape_md(input, true), expected);
    }

    #[test]
    fn supports_fence_info_strings() {
        let input = "```bash\necho <value>\n```\nafter <";
        let expected = "```bash\necho <value>\n```\nafter &lt;";

        assert_eq!(escape_md(input, true), expected);
    }

    #[test]
    fn leaves_unclosed_fences_unescaped() {
        let input = "```\necho <value>";

        assert_eq!(escape_md(input, true), input);
    }

    #[test]
    fn handles_longer_fences_and_indented_code() {
        let input = "    ```\nindented <\n````\nlonger <";
        let expected = "    ```\nindented &lt;\n````\nlonger <";

        assert_eq!(escape_md(input, true), expected);
    }

    #[test]
    fn leaves_markdown_unchanged_when_html_encoding_is_disabled() {
        let input = "before <\n```\ninside <\n```\nafter <";

        assert_eq!(escape_md(input, false), input);
    }

    #[test]
    fn strips_terminal_styling_from_generated_markdown() {
        let input =
            "\u{1b}[1m\u{1b}[4mExamples:\u{1b}[22m\u{1b}[24m\n\n    \u{1b}[1mmise run\u{1b}[22m";
        let expected = "Examples:\n\n    mise run";

        assert_eq!(escape_md(input, true), expected);
        assert_eq!(escape_md(input, false), expected);
    }

    #[test]
    fn one_template_can_be_replaced_without_copying_its_includes() {
        let spec = "bin \"ex\"\nflag \"--force\" help=\"Do it anyway\"\n"
            .parse()
            .unwrap();
        let page = MarkdownRenderer::new(spec)
            .with_template(
                MarkdownTemplate::Spec,
                "# Custom {{ spec.bin }}\n{% set cmd = spec.cmd %}\n{% include \"cmd_template.md.tera\" %}",
            )
            .render_spec()
            .unwrap();

        assert!(page.starts_with("# Custom ex\n"), "{page}");
        assert!(page.contains("- **`--force`**"), "{page}");
    }

    #[test]
    fn the_last_replacement_of_a_template_wins() {
        let spec = "bin \"ex\"\n".parse().unwrap();
        let page = MarkdownRenderer::new(spec)
            .with_template(MarkdownTemplate::Spec, "{{")
            .with_template(MarkdownTemplate::Spec, "second")
            .render_spec()
            .unwrap();

        assert_eq!(page, "second");
    }

    #[test]
    fn a_bad_custom_template_is_a_render_error() {
        let spec = "bin \"ex\"\n".parse().unwrap();
        let err = MarkdownRenderer::new(spec)
            .with_template(MarkdownTemplate::Spec, "{{")
            .render_spec()
            .unwrap_err();

        assert!(err.to_string().contains("template"), "{err}");
    }

    #[test]
    fn the_detailed_theme_keeps_addressable_entry_headings() {
        let spec = "bin \"ex\"\nflag \"--force\" help=\"Do it anyway\"\n"
            .parse()
            .unwrap();
        let page = MarkdownRenderer::new(spec)
            .with_theme(super::MarkdownTheme::Detailed)
            .render_spec()
            .unwrap();

        assert!(page.contains("### `--force`"), "{page}");
    }

    #[test]
    fn entry_template_overrides_apply_to_the_compact_theme() {
        let spec = "bin \"ex\"\narg \"<file>\"\nflag \"--force\"\n"
            .parse()
            .unwrap();
        let page = MarkdownRenderer::new(spec)
            .with_template(MarkdownTemplate::Argument, "argument: {{ arg.usage }}")
            .with_template(MarkdownTemplate::Flag, "flag: {{ flag.usage }}")
            .render_spec()
            .unwrap();

        assert!(page.contains("argument: <file>"), "{page}");
        assert!(page.contains("flag: --force"), "{page}");
    }
}