Change syntax::ast_util::stmt_id to not panic on macros

This enables the Debug trait to work on syntax::ast::Stmt
This commit is contained in:
Jethro Beekman
2015-09-21 16:10:38 -07:00
parent 7e8d19b24d
commit 0a2ffa0835
2 changed files with 10 additions and 7 deletions

View File

@@ -65,6 +65,7 @@ use ptr::P;
use std::fmt;
use std::rc::Rc;
use std::borrow::Cow;
use serialize::{Encodable, Decodable, Encoder, Decoder};
// FIXME #6993: in librustc, uses of "ident" should be replaced
@@ -685,7 +686,8 @@ pub type Stmt = Spanned<Stmt_>;
impl fmt::Debug for Stmt {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "stmt({}: {})",
ast_util::stmt_id(self),
ast_util::stmt_id(self)
.map_or(Cow::Borrowed("<macro>"),|id|Cow::Owned(id.to_string())),
pprust::stmt_to_string(self))
}
}

View File

@@ -28,12 +28,12 @@ pub fn path_name_i(idents: &[Ident]) -> String {
idents.iter().map(|i| i.to_string()).collect::<Vec<String>>().join("::")
}
pub fn stmt_id(s: &Stmt) -> NodeId {
pub fn stmt_id(s: &Stmt) -> Option<NodeId> {
match s.node {
StmtDecl(_, id) => id,
StmtExpr(_, id) => id,
StmtSemi(_, id) => id,
StmtMac(..) => panic!("attempted to analyze unexpanded stmt")
StmtDecl(_, id) => Some(id),
StmtExpr(_, id) => Some(id),
StmtSemi(_, id) => Some(id),
StmtMac(..) => None,
}
}
@@ -385,7 +385,8 @@ impl<'a, 'v, O: IdVisitingOperation> Visitor<'v> for IdVisitor<'a, O> {
}
fn visit_stmt(&mut self, statement: &Stmt) {
self.operation.visit_id(ast_util::stmt_id(statement));
self.operation
.visit_id(ast_util::stmt_id(statement).expect("attempted to visit unexpanded stmt"));
visit::walk_stmt(self, statement)
}