use crate::TextRange; use ra_syntax::{ algo::visit::{visitor, Visitor}, ast::{self, AttrsOwner, NameOwner, TypeAscriptionOwner, TypeParamsOwner}, AstNode, SourceFile, SyntaxKind, SyntaxNode, WalkEvent, }; #[derive(Debug, Clone)] pub struct StructureNode { pub parent: Option, pub label: String, pub navigation_range: TextRange, pub node_range: TextRange, pub kind: SyntaxKind, pub detail: Option, pub deprecated: bool, } pub fn file_structure(file: &SourceFile) -> Vec { let mut res = Vec::new(); let mut stack = Vec::new(); for event in file.syntax().preorder() { match event { WalkEvent::Enter(node) => { if let Some(mut symbol) = structure_node(&node) { symbol.parent = stack.last().copied(); stack.push(res.len()); res.push(symbol); } } WalkEvent::Leave(node) => { if structure_node(&node).is_some() { stack.pop().unwrap(); } } } } res } fn structure_node(node: &SyntaxNode) -> Option { fn decl(node: N) -> Option { decl_with_detail(node, None) } fn decl_with_ascription( node: N, ) -> Option { let ty = node.ascribed_type(); decl_with_type_ref(node, ty) } fn decl_with_type_ref( node: N, type_ref: Option, ) -> Option { let detail = type_ref.map(|type_ref| { let mut detail = String::new(); collapse_ws(type_ref.syntax(), &mut detail); detail }); decl_with_detail(node, detail) } fn decl_with_detail( node: N, detail: Option, ) -> Option { let name = node.name()?; Some(StructureNode { parent: None, label: name.text().to_string(), navigation_range: name.syntax().text_range(), node_range: node.syntax().text_range(), kind: node.syntax().kind(), detail, deprecated: node.attrs().filter_map(|x| x.as_named()).any(|x| x == "deprecated"), }) } fn collapse_ws(node: &SyntaxNode, output: &mut String) { let mut can_insert_ws = false; node.text().for_each_chunk(|chunk| { for line in chunk.lines() { let line = line.trim(); if line.is_empty() { if can_insert_ws { output.push(' '); can_insert_ws = false; } } else { output.push_str(line); can_insert_ws = true; } } }) } visitor() .visit(|fn_def: ast::FnDef| { let mut detail = String::from("fn"); if let Some(type_param_list) = fn_def.type_param_list() { collapse_ws(type_param_list.syntax(), &mut detail); } if let Some(param_list) = fn_def.param_list() { collapse_ws(param_list.syntax(), &mut detail); } if let Some(ret_type) = fn_def.ret_type() { detail.push_str(" "); collapse_ws(ret_type.syntax(), &mut detail); } decl_with_detail(fn_def, Some(detail)) }) .visit(decl::) .visit(decl::) .visit(decl::) .visit(decl::) .visit(decl::) .visit(|td: ast::TypeAliasDef| { let ty = td.type_ref(); decl_with_type_ref(td, ty) }) .visit(decl_with_ascription::) .visit(decl_with_ascription::) .visit(decl_with_ascription::) .visit(|im: ast::ImplBlock| { let target_type = im.target_type()?; let target_trait = im.target_trait(); let label = match target_trait { None => format!("impl {}", target_type.syntax().text()), Some(t) => { format!("impl {} for {}", t.syntax().text(), target_type.syntax().text(),) } }; let node = StructureNode { parent: None, label, navigation_range: target_type.syntax().text_range(), node_range: im.syntax().text_range(), kind: im.syntax().kind(), detail: None, deprecated: false, }; Some(node) }) .visit(|mc: ast::MacroCall| { let first_token = mc.syntax().first_token().unwrap(); if first_token.text().as_str() != "macro_rules" { return None; } decl(mc) }) .accept(&node)? } #[cfg(test)] mod tests { use super::*; use insta::assert_debug_snapshot; #[test] fn test_file_structure() { let file = SourceFile::parse( r#" struct Foo { x: i32 } mod m { fn bar1() {} fn bar2(t: T) -> T {} fn bar3(a: A, b: B) -> Vec< u32 > {} } enum E { X, Y(i32) } type T = (); static S: i32 = 92; const C: i32 = 92; impl E {} impl fmt::Debug for E {} macro_rules! mc { () => {} } #[deprecated] fn obsolete() {} #[deprecated(note = "for awhile")] fn very_obsolete() {} "#, ) .ok() .unwrap(); let structure = file_structure(&file); assert_debug_snapshot!(structure, @r#"[ StructureNode { parent: None, label: "Foo", navigation_range: [8; 11), node_range: [1; 26), kind: STRUCT_DEF, detail: None, deprecated: false, }, StructureNode { parent: Some( 0, ), label: "x", navigation_range: [18; 19), node_range: [18; 24), kind: RECORD_FIELD_DEF, detail: Some( "i32", ), deprecated: false, }, StructureNode { parent: None, label: "m", navigation_range: [32; 33), node_range: [28; 158), kind: MODULE, detail: None, deprecated: false, }, StructureNode { parent: Some( 2, ), label: "bar1", navigation_range: [43; 47), node_range: [40; 52), kind: FN_DEF, detail: Some( "fn()", ), deprecated: false, }, StructureNode { parent: Some( 2, ), label: "bar2", navigation_range: [60; 64), node_range: [57; 81), kind: FN_DEF, detail: Some( "fn(t: T) -> T", ), deprecated: false, }, StructureNode { parent: Some( 2, ), label: "bar3", navigation_range: [89; 93), node_range: [86; 156), kind: FN_DEF, detail: Some( "fn(a: A, b: B) -> Vec< u32 >", ), deprecated: false, }, StructureNode { parent: None, label: "E", navigation_range: [165; 166), node_range: [160; 180), kind: ENUM_DEF, detail: None, deprecated: false, }, StructureNode { parent: Some( 6, ), label: "X", navigation_range: [169; 170), node_range: [169; 170), kind: ENUM_VARIANT, detail: None, deprecated: false, }, StructureNode { parent: Some( 6, ), label: "Y", navigation_range: [172; 173), node_range: [172; 178), kind: ENUM_VARIANT, detail: None, deprecated: false, }, StructureNode { parent: None, label: "T", navigation_range: [186; 187), node_range: [181; 193), kind: TYPE_ALIAS_DEF, detail: Some( "()", ), deprecated: false, }, StructureNode { parent: None, label: "S", navigation_range: [201; 202), node_range: [194; 213), kind: STATIC_DEF, detail: Some( "i32", ), deprecated: false, }, StructureNode { parent: None, label: "C", navigation_range: [220; 221), node_range: [214; 232), kind: CONST_DEF, detail: Some( "i32", ), deprecated: false, }, StructureNode { parent: None, label: "impl E", navigation_range: [239; 240), node_range: [234; 243), kind: IMPL_BLOCK, detail: None, deprecated: false, }, StructureNode { parent: None, label: "impl fmt::Debug for E", navigation_range: [265; 266), node_range: [245; 269), kind: IMPL_BLOCK, detail: None, deprecated: false, }, StructureNode { parent: None, label: "mc", navigation_range: [284; 286), node_range: [271; 303), kind: MACRO_CALL, detail: None, deprecated: false, }, StructureNode { parent: None, label: "obsolete", navigation_range: [322; 330), node_range: [305; 335), kind: FN_DEF, detail: Some( "fn()", ), deprecated: true, }, StructureNode { parent: None, label: "very_obsolete", navigation_range: [375; 388), node_range: [337; 393), kind: FN_DEF, detail: Some( "fn()", ), deprecated: true, }, ]"# ); } }