es_fluent_cli/commands/

tree.rs

//! Tree command for displaying FTL structure.
//!
//! This module provides functionality to display a tree view of FTL items
//! for each FTL file associated with a crate.

use crate::commands::{WorkspaceArgs, WorkspaceCrates};
use crate::core::CliError;
use crate::ftl::{LocaleContext, parse_ftl_file};
use crate::utils::{discover_ftl_files, ui};
use anyhow::Result;
use clap::Parser;
use colored::Colorize as _;
use fluent_syntax::ast;
use std::path::Path;
use treelog::Tree;

#[derive(Clone, Copy)]
struct TreeRenderer {
    show_attributes: bool,
    show_variables: bool,
}

impl TreeRenderer {
    fn new(show_attributes: bool, show_variables: bool) -> Self {
        Self {
            show_attributes,
            show_variables,
        }
    }

    /// Build a tree for a single FTL file.
    fn build_file_tree(&self, relative_path: &str, abs_path: &Path) -> Tree {
        let resource = match parse_ftl_file(abs_path) {
            Ok(res) => res,
            Err(_) => {
                return Tree::Node(
                    relative_path.yellow().to_string(),
                    vec![Tree::Leaf(vec!["<parse error>".red().to_string()])],
                );
            },
        };

        let entries: Vec<Tree> = resource
            .body
            .iter()
            .filter_map(|entry| match entry {
                ast::Entry::Message(msg) => Some(self.build_message_tree(&msg.id.name, msg)),
                ast::Entry::Term(term) => Some(self.build_term_tree(&term.id.name, term)),
                ast::Entry::Comment(_) => None,
                ast::Entry::GroupComment(_) => None,
                ast::Entry::ResourceComment(_) => None,
                ast::Entry::Junk { .. } => None,
            })
            .collect();

        Tree::Node(relative_path.yellow().to_string(), entries)
    }

    /// Build a tree for a message entry.
    fn build_message_tree(&self, id: &str, msg: &ast::Message<String>) -> Tree {
        let children = self.build_entry_children(&msg.attributes, msg.value.as_ref());

        if children.is_empty() {
            Tree::Leaf(vec![id.to_string()])
        } else {
            Tree::Node(id.to_string(), children)
        }
    }

    /// Build a tree for a term entry.
    fn build_term_tree(&self, id: &str, term: &ast::Term<String>) -> Tree {
        let children = self.build_entry_children(&term.attributes, Some(&term.value));
        let label = format!("-{}", id);

        if children.is_empty() {
            Tree::Leaf(vec![label.dimmed().to_string()])
        } else {
            Tree::Node(label.dimmed().to_string(), children)
        }
    }

    /// Build child nodes for an entry (attributes and variables).
    fn build_entry_children(
        &self,
        attributes: &[ast::Attribute<String>],
        value: Option<&ast::Pattern<String>>,
    ) -> Vec<Tree> {
        let mut children: Vec<Tree> = Vec::new();

        if self.show_attributes {
            for attr in attributes {
                let attr_label = format!("@{}", attr.id.name);
                children.push(Tree::Leaf(vec![attr_label.dimmed().to_string()]));
            }
        }

        if self.show_variables {
            let mut variables = Vec::new();
            if let Some(pattern) = value {
                extract_variables_from_pattern_into(pattern, &mut variables);
            }
            for attr in attributes {
                extract_variables_from_pattern_into(&attr.value, &mut variables);
            }

            if !variables.is_empty() {
                variables.sort();
                variables.dedup();
                let vars_str = variables
                    .iter()
                    .map(|v| format!("${}", v))
                    .collect::<Vec<_>>()
                    .join(", ");
                children.push(Tree::Leaf(vec![vars_str.magenta().to_string()]));
            }
        }

        children
    }
}

/// Arguments for the tree command.
#[derive(Debug, Parser)]
pub struct TreeArgs {
    #[command(flatten)]
    pub workspace: WorkspaceArgs,

    /// Show all locales, not just the fallback language.
    #[arg(long)]
    pub all: bool,

    /// Show attributes under each message.
    #[arg(long)]
    pub attributes: bool,

    /// Show variables used in each message.
    #[arg(long)]
    pub variables: bool,
}

/// Run the tree command.
pub fn run_tree(args: TreeArgs) -> Result<(), CliError> {
    let workspace = WorkspaceCrates::discover(args.workspace)?;

    ui::print_tree_header();

    if workspace.crates.is_empty() {
        ui::print_no_crates_found();
        return Ok(());
    }

    for krate in &workspace.crates {
        print_crate_tree(krate, args.all, args.attributes, args.variables)?;
    }

    Ok(())
}

/// Print the tree for a single crate.
fn print_crate_tree(
    krate: &crate::core::CrateInfo,
    all_locales: bool,
    show_attributes: bool,
    show_variables: bool,
) -> Result<()> {
    let ctx = LocaleContext::from_crate(krate, all_locales)?;
    let renderer = TreeRenderer::new(show_attributes, show_variables);

    let mut locale_trees: Vec<Tree> = Vec::new();

    for locale in &ctx.locales {
        let locale_dir = ctx.locale_dir(locale);
        if !locale_dir.exists() {
            continue;
        }

        let ftl_files = discover_ftl_files(&ctx.assets_dir, locale, &ctx.crate_name)?;

        if ftl_files.is_empty() {
            continue;
        }

        let file_trees: Vec<Tree> = ftl_files
            .iter()
            .map(|file_info| {
                renderer.build_file_tree(
                    &file_info.relative_path.display().to_string(),
                    &file_info.abs_path,
                )
            })
            .collect();

        locale_trees.push(Tree::Node(locale.green().to_string(), file_trees));
    }

    let tree = Tree::Node(krate.name.bold().cyan().to_string(), locale_trees);
    println!("{}", tree.render_to_string());

    Ok(())
}

/// Extract variable names from a pattern into a vector.
fn extract_variables_from_pattern_into(
    pattern: &ast::Pattern<String>,
    variables: &mut Vec<String>,
) {
    for element in &pattern.elements {
        if let ast::PatternElement::Placeable { expression } = element {
            extract_variables_from_expression(expression, variables);
        }
    }
}

/// Extract variable names from an expression.
fn extract_variables_from_expression(
    expression: &ast::Expression<String>,
    variables: &mut Vec<String>,
) {
    match expression {
        ast::Expression::Inline(inline) => {
            extract_variables_from_inline(inline, variables);
        },
        ast::Expression::Select { selector, variants } => {
            extract_variables_from_inline(selector, variables);
            for variant in variants {
                extract_variables_from_pattern_into(&variant.value, variables);
            }
        },
    }
}

/// Extract variable names from an inline expression.
fn extract_variables_from_inline(
    inline: &ast::InlineExpression<String>,
    variables: &mut Vec<String>,
) {
    match inline {
        ast::InlineExpression::VariableReference { id } => {
            variables.push(id.name.clone());
        },
        ast::InlineExpression::FunctionReference { arguments, .. } => {
            for arg in &arguments.positional {
                extract_variables_from_inline(arg, variables);
            }
            for arg in &arguments.named {
                extract_variables_from_inline(&arg.value, variables);
            }
        },
        ast::InlineExpression::Placeable { expression } => {
            extract_variables_from_expression(expression, variables);
        },
        _ => {},
    }
}

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

    fn parse_ftl(content: &str) -> ast::Resource<String> {
        parser::parse(content.to_string()).unwrap()
    }

    fn get_message<'a>(
        resource: &'a ast::Resource<String>,
        id: &str,
    ) -> Option<&'a ast::Message<String>> {
        resource.body.iter().find_map(|entry| {
            if let ast::Entry::Message(msg) = entry
                && msg.id.name == id
            {
                return Some(msg);
            }
            None
        })
    }

    fn renderer(show_attributes: bool, show_variables: bool) -> TreeRenderer {
        TreeRenderer::new(show_attributes, show_variables)
    }

    #[test]
    fn test_extract_variables_simple() {
        let content = "hello = Hello { $name }!";
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "hello").unwrap();

        let mut variables = Vec::new();
        extract_variables_from_pattern_into(msg.value.as_ref().unwrap(), &mut variables);

        assert_eq!(variables, vec!["name"]);
    }

    #[test]
    fn test_extract_variables_multiple() {
        let content = "greeting = Hello { $name }, you have { $count } messages";
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "greeting").unwrap();

        let mut variables = Vec::new();
        extract_variables_from_pattern_into(msg.value.as_ref().unwrap(), &mut variables);
        variables.sort();

        assert_eq!(variables, vec!["count", "name"]);
    }

    #[test]
    fn test_extract_variables_select() {
        let content = r#"count = { $num ->
    [one] One item
   *[other] { $num } items
}"#;
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "count").unwrap();

        let mut variables = Vec::new();
        extract_variables_from_pattern_into(msg.value.as_ref().unwrap(), &mut variables);
        variables.sort();
        variables.dedup();

        assert_eq!(variables, vec!["num"]);
    }

    #[test]
    fn test_extract_variables_nested() {
        let content = r#"message = Hello { $user }, today is { DATETIME($date) }"#;
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "message").unwrap();

        let mut variables = Vec::new();
        extract_variables_from_pattern_into(msg.value.as_ref().unwrap(), &mut variables);
        variables.sort();

        assert_eq!(variables, vec!["date", "user"]);
    }

    #[test]
    fn test_build_message_tree_simple() {
        let content = "hello = Hello World";
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "hello").unwrap();

        let tree = renderer(false, false).build_message_tree("hello", msg);

        match tree {
            Tree::Leaf(lines) => assert_eq!(lines, vec!["hello"]),
            _ => panic!("Expected leaf node"),
        }
    }

    #[test]
    fn test_build_message_tree_with_attributes() {
        let content = r#"button = Button
    .tooltip = Click me
    .aria-label = Submit"#;
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "button").unwrap();

        let tree = renderer(true, false).build_message_tree("button", msg);

        match tree {
            Tree::Node(label, children) => {
                assert_eq!(label, "button");
                assert_eq!(children.len(), 2);
            },
            _ => panic!("Expected node with children"),
        }
    }

    #[test]
    fn test_build_message_tree_with_variables() {
        let content = "greeting = Hello { $name }";
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "greeting").unwrap();

        let tree = renderer(false, true).build_message_tree("greeting", msg);

        match tree {
            Tree::Node(label, children) => {
                assert_eq!(label, "greeting");
                assert_eq!(children.len(), 1);
            },
            _ => panic!("Expected node with children"),
        }
    }

    #[test]
    fn test_build_entry_children_no_attributes_no_variables() {
        let children = renderer(false, false).build_entry_children(&[], None);
        assert!(children.is_empty());
    }

    #[test]
    fn test_build_entry_children_attributes_only() {
        let content = r#"button = Button
    .tooltip = Click me"#;
        let resource = parse_ftl(content);
        let msg = get_message(&resource, "button").unwrap();

        let children =
            renderer(true, false).build_entry_children(&msg.attributes, msg.value.as_ref());

        assert_eq!(children.len(), 1);
    }

    #[test]
    fn test_build_file_tree_nonexistent() {
        let tree =
            renderer(false, false).build_file_tree("test.ftl", Path::new("/nonexistent/path.ftl"));

        match tree {
            Tree::Node(label, children) => {
                assert!(label.contains("test.ftl"));
                assert!(
                    children.is_empty(),
                    "nonexistent file should produce empty tree"
                );
            },
            _ => panic!("Expected node"),
        }
    }

    #[test]
    fn test_tree_render_basic() {
        let tree = Tree::Node(
            "root".to_string(),
            vec![
                Tree::Leaf(vec!["item1".to_string()]),
                Tree::Leaf(vec!["item2".to_string()]),
            ],
        );

        let output = tree.render_to_string();
        assert!(output.contains("root"));
        assert!(output.contains("item1"));
        assert!(output.contains("item2"));
    }

    #[test]
    fn test_tree_render_nested() {
        let tree = Tree::Node(
            "crate".to_string(),
            vec![Tree::Node(
                "en".to_string(),
                vec![Tree::Leaf(vec!["message".to_string()])],
            )],
        );

        let output = tree.render_to_string();
        assert!(output.contains("crate"));
        assert!(output.contains("en"));
        assert!(output.contains("message"));
    }
}