Implement Rewrite for ast::Stmt

[rust.git] / src / visitor.rs

diff --git a/src/visitor.rs b/src/visitor.rs
index 534e8b112e4e340a336477dd0f71a00fe79f675b..6b98fd276b61debb30a68b166d84a374aa0ca96b 100644 (file)
--- a/src/visitor.rs
+++ b/src/visitor.rs
@@ -41,12 +41,18 @@ fn visit_stmt(&mut self, stmt: &ast::Stmt) {
                 if let ast::Decl_::DeclItem(ref item) = decl.node {
                     self.visit_item(item);
                 } else {
-                    let rewrite = self.rewrite_stmt(stmt);
+                    let rewrite = stmt.rewrite(&self.get_context(),
+                                               self.config.max_width - self.block_indent.width(),
+                                               self.block_indent);
+
                     self.push_rewrite(stmt.span, rewrite);
                 }
             }
             ast::Stmt_::StmtExpr(..) | ast::Stmt_::StmtSemi(..) => {
-                let rewrite = self.rewrite_stmt(stmt);
+                let rewrite = stmt.rewrite(&self.get_context(),
+                                           self.config.max_width - self.block_indent.width(),
+                                           self.block_indent);
+
                 self.push_rewrite(stmt.span, rewrite);
             }
             ast::Stmt_::StmtMac(ref mac, _macro_style) => {