From d88b9d3ccbcccd4808b64960fcc28483339bc551 Mon Sep 17 00:00:00 2001 From: "Edward Z. Yang" Date: Thu, 19 Dec 2013 14:41:59 +0800 Subject: [PATCH] Generalize item_span into node_span, which works on more types. Signed-off-by: Edward Z. Yang --- src/libsyntax/ast_map.rs | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/src/libsyntax/ast_map.rs b/src/libsyntax/ast_map.rs index 961d8170fd35d..0c27a338321f6 100644 --- a/src/libsyntax/ast_map.rs +++ b/src/libsyntax/ast_map.rs @@ -501,3 +501,24 @@ pub fn item_span(items: map, } } } + +pub fn node_span(items: map, + id: ast::NodeId) + -> Span { + match items.find(&id) { + Some(&node_item(item, _)) => item.span, + Some(&node_foreign_item(foreign_item, _, _, _)) => foreign_item.span, + Some(&node_trait_method(@required(ref type_method), _, _)) => type_method.span, + Some(&node_trait_method(@provided(ref method), _, _)) => method.span, + Some(&node_method(method, _, _)) => method.span, + Some(&node_variant(variant, _, _)) => variant.span, + Some(&node_expr(expr)) => expr.span, + Some(&node_stmt(stmt)) => stmt.span, + Some(&node_arg(pat)) => pat.span, + Some(&node_local(_)) => fail!("node_local"), + Some(&node_block(block)) => block.span, + Some(&node_struct_ctor(_, item, _)) => item.span, + Some(&node_callee_scope(expr)) => expr.span, + None => fail!("None"), + } +}