]>
Commit | Line | Data |
---|---|---|
e9174d1e SL |
1 | // Copyright 2015 The Rust Project Developers. See the COPYRIGHT |
2 | // file at the top-level directory of this distribution and at | |
3 | // http://rust-lang.org/COPYRIGHT. | |
4 | // | |
5 | // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or | |
6 | // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license | |
7 | // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your | |
8 | // option. This file may not be copied, modified, or distributed | |
9 | // except according to those terms. | |
10 | ||
11 | pub use self::AnnNode::*; | |
12 | ||
7453a54e | 13 | use syntax::abi::Abi; |
e9174d1e | 14 | use syntax::ast; |
3157f602 | 15 | use syntax::codemap::{CodeMap, Spanned}; |
32a655c1 | 16 | use syntax::parse::ParseSess; |
e9174d1e | 17 | use syntax::parse::lexer::comments; |
041b39d2 | 18 | use syntax::print::pp::{self, Breaks}; |
e9174d1e | 19 | use syntax::print::pp::Breaks::{Consistent, Inconsistent}; |
041b39d2 | 20 | use syntax::print::pprust::PrintState; |
e9174d1e | 21 | use syntax::ptr::P; |
476ff2be | 22 | use syntax::symbol::keywords; |
ea8adc8c | 23 | use syntax::util::parser::{self, AssocOp, Fixity}; |
3157f602 | 24 | use syntax_pos::{self, BytePos}; |
e9174d1e SL |
25 | |
26 | use hir; | |
32a655c1 | 27 | use hir::{PatKind, RegionTyParamBound, TraitTyParamBound, TraitBoundModifier, RangeEnd}; |
e9174d1e | 28 | |
32a655c1 | 29 | use std::cell::Cell; |
e9174d1e | 30 | use std::io::{self, Write, Read}; |
041b39d2 XL |
31 | use std::iter::Peekable; |
32 | use std::vec; | |
e9174d1e SL |
33 | |
34 | pub enum AnnNode<'a> { | |
e9174d1e SL |
35 | NodeName(&'a ast::Name), |
36 | NodeBlock(&'a hir::Block), | |
37 | NodeItem(&'a hir::Item), | |
38 | NodeSubItem(ast::NodeId), | |
39 | NodeExpr(&'a hir::Expr), | |
40 | NodePat(&'a hir::Pat), | |
41 | } | |
42 | ||
32a655c1 SL |
43 | pub enum Nested { |
44 | Item(hir::ItemId), | |
45 | TraitItem(hir::TraitItemId), | |
46 | ImplItem(hir::ImplItemId), | |
47 | Body(hir::BodyId), | |
48 | BodyArgPat(hir::BodyId, usize) | |
49 | } | |
50 | ||
e9174d1e | 51 | pub trait PpAnn { |
32a655c1 SL |
52 | fn nested(&self, _state: &mut State, _nested: Nested) -> io::Result<()> { |
53 | Ok(()) | |
54 | } | |
b039eaaf SL |
55 | fn pre(&self, _state: &mut State, _node: AnnNode) -> io::Result<()> { |
56 | Ok(()) | |
57 | } | |
58 | fn post(&self, _state: &mut State, _node: AnnNode) -> io::Result<()> { | |
59 | Ok(()) | |
60 | } | |
e9174d1e SL |
61 | } |
62 | ||
e9174d1e | 63 | pub struct NoAnn; |
e9174d1e | 64 | impl PpAnn for NoAnn {} |
32a655c1 | 65 | pub const NO_ANN: &'static PpAnn = &NoAnn; |
e9174d1e | 66 | |
32a655c1 SL |
67 | impl PpAnn for hir::Crate { |
68 | fn nested(&self, state: &mut State, nested: Nested) -> io::Result<()> { | |
69 | match nested { | |
70 | Nested::Item(id) => state.print_item(self.item(id.id)), | |
71 | Nested::TraitItem(id) => state.print_trait_item(self.trait_item(id)), | |
72 | Nested::ImplItem(id) => state.print_impl_item(self.impl_item(id)), | |
73 | Nested::Body(id) => state.print_expr(&self.body(id).value), | |
74 | Nested::BodyArgPat(id, i) => state.print_pat(&self.body(id).arguments[i].pat) | |
75 | } | |
76 | } | |
77 | } | |
e9174d1e SL |
78 | |
79 | pub struct State<'a> { | |
80 | pub s: pp::Printer<'a>, | |
81 | cm: Option<&'a CodeMap>, | |
b039eaaf | 82 | comments: Option<Vec<comments::Comment>>, |
041b39d2 XL |
83 | literals: Peekable<vec::IntoIter<comments::Literal>>, |
84 | cur_cmnt: usize, | |
e9174d1e | 85 | boxes: Vec<pp::Breaks>, |
b039eaaf | 86 | ann: &'a (PpAnn + 'a), |
e9174d1e SL |
87 | } |
88 | ||
b039eaaf SL |
89 | impl<'a> PrintState<'a> for State<'a> { |
90 | fn writer(&mut self) -> &mut pp::Printer<'a> { | |
91 | &mut self.s | |
92 | } | |
93 | ||
94 | fn boxes(&mut self) -> &mut Vec<pp::Breaks> { | |
95 | &mut self.boxes | |
96 | } | |
97 | ||
98 | fn comments(&mut self) -> &mut Option<Vec<comments::Comment>> { | |
99 | &mut self.comments | |
100 | } | |
101 | ||
041b39d2 XL |
102 | fn cur_cmnt(&mut self) -> &mut usize { |
103 | &mut self.cur_cmnt | |
b039eaaf SL |
104 | } |
105 | ||
041b39d2 XL |
106 | fn cur_lit(&mut self) -> Option<&comments::Literal> { |
107 | self.literals.peek() | |
108 | } | |
109 | ||
110 | fn bump_lit(&mut self) -> Option<comments::Literal> { | |
111 | self.literals.next() | |
b039eaaf SL |
112 | } |
113 | } | |
114 | ||
e9174d1e SL |
115 | #[allow(non_upper_case_globals)] |
116 | pub const indent_unit: usize = 4; | |
117 | ||
118 | #[allow(non_upper_case_globals)] | |
119 | pub const default_columns: usize = 78; | |
120 | ||
121 | ||
122 | /// Requires you to pass an input filename and reader so that | |
123 | /// it can scan the input text for comments and literals to | |
124 | /// copy forward. | |
125 | pub fn print_crate<'a>(cm: &'a CodeMap, | |
32a655c1 | 126 | sess: &ParseSess, |
e9174d1e SL |
127 | krate: &hir::Crate, |
128 | filename: String, | |
129 | input: &mut Read, | |
b039eaaf | 130 | out: Box<Write + 'a>, |
e9174d1e | 131 | ann: &'a PpAnn, |
b039eaaf SL |
132 | is_expanded: bool) |
133 | -> io::Result<()> { | |
32a655c1 | 134 | let mut s = State::new_from_input(cm, sess, filename, input, out, ann, is_expanded); |
e9174d1e SL |
135 | |
136 | // When printing the AST, we sometimes need to inject `#[no_std]` here. | |
137 | // Since you can't compile the HIR, it's not necessary. | |
138 | ||
54a0048b SL |
139 | s.print_mod(&krate.module, &krate.attrs)?; |
140 | s.print_remaining_comments()?; | |
041b39d2 | 141 | s.s.eof() |
e9174d1e SL |
142 | } |
143 | ||
144 | impl<'a> State<'a> { | |
145 | pub fn new_from_input(cm: &'a CodeMap, | |
32a655c1 | 146 | sess: &ParseSess, |
e9174d1e SL |
147 | filename: String, |
148 | input: &mut Read, | |
b039eaaf | 149 | out: Box<Write + 'a>, |
e9174d1e | 150 | ann: &'a PpAnn, |
32a655c1 | 151 | is_expanded: bool) |
b039eaaf | 152 | -> State<'a> { |
32a655c1 | 153 | let (cmnts, lits) = comments::gather_comments_and_literals(sess, filename, input); |
b039eaaf SL |
154 | |
155 | State::new(cm, | |
156 | out, | |
157 | ann, | |
158 | Some(cmnts), | |
159 | // If the code is post expansion, don't use the table of | |
160 | // literals, since it doesn't correspond with the literals | |
161 | // in the AST anymore. | |
162 | if is_expanded { | |
163 | None | |
164 | } else { | |
165 | Some(lits) | |
32a655c1 | 166 | }) |
e9174d1e SL |
167 | } |
168 | ||
169 | pub fn new(cm: &'a CodeMap, | |
b039eaaf | 170 | out: Box<Write + 'a>, |
e9174d1e SL |
171 | ann: &'a PpAnn, |
172 | comments: Option<Vec<comments::Comment>>, | |
32a655c1 | 173 | literals: Option<Vec<comments::Literal>>) |
b039eaaf | 174 | -> State<'a> { |
e9174d1e SL |
175 | State { |
176 | s: pp::mk_printer(out, default_columns), | |
177 | cm: Some(cm), | |
b039eaaf | 178 | comments: comments.clone(), |
041b39d2 XL |
179 | literals: literals.unwrap_or_default().into_iter().peekable(), |
180 | cur_cmnt: 0, | |
e9174d1e | 181 | boxes: Vec::new(), |
041b39d2 | 182 | ann, |
e9174d1e SL |
183 | } |
184 | } | |
185 | } | |
186 | ||
32a655c1 | 187 | pub fn to_string<F>(ann: &PpAnn, f: F) -> String |
b039eaaf | 188 | where F: FnOnce(&mut State) -> io::Result<()> |
e9174d1e SL |
189 | { |
190 | let mut wr = Vec::new(); | |
191 | { | |
32a655c1 SL |
192 | let mut printer = State { |
193 | s: pp::mk_printer(Box::new(&mut wr), default_columns), | |
194 | cm: None, | |
195 | comments: None, | |
041b39d2 XL |
196 | literals: vec![].into_iter().peekable(), |
197 | cur_cmnt: 0, | |
32a655c1 | 198 | boxes: Vec::new(), |
041b39d2 | 199 | ann, |
32a655c1 | 200 | }; |
e9174d1e | 201 | f(&mut printer).unwrap(); |
041b39d2 | 202 | printer.s.eof().unwrap(); |
e9174d1e SL |
203 | } |
204 | String::from_utf8(wr).unwrap() | |
205 | } | |
206 | ||
32a655c1 SL |
207 | pub fn visibility_qualified(vis: &hir::Visibility, w: &str) -> String { |
208 | to_string(NO_ANN, |s| { | |
209 | s.print_visibility(vis)?; | |
041b39d2 | 210 | s.s.word(w) |
e9174d1e SL |
211 | }) |
212 | } | |
213 | ||
e9174d1e | 214 | impl<'a> State<'a> { |
e9174d1e SL |
215 | pub fn cbox(&mut self, u: usize) -> io::Result<()> { |
216 | self.boxes.push(pp::Breaks::Consistent); | |
041b39d2 | 217 | self.s.cbox(u) |
e9174d1e SL |
218 | } |
219 | ||
b039eaaf | 220 | pub fn nbsp(&mut self) -> io::Result<()> { |
041b39d2 | 221 | self.s.word(" ") |
e9174d1e SL |
222 | } |
223 | ||
e9174d1e | 224 | pub fn word_nbsp(&mut self, w: &str) -> io::Result<()> { |
041b39d2 | 225 | self.s.word(w)?; |
e9174d1e SL |
226 | self.nbsp() |
227 | } | |
228 | ||
e9174d1e SL |
229 | pub fn head(&mut self, w: &str) -> io::Result<()> { |
230 | // outer-box is consistent | |
54a0048b | 231 | self.cbox(indent_unit)?; |
e9174d1e | 232 | // head-box is inconsistent |
54a0048b | 233 | self.ibox(w.len() + 1)?; |
e9174d1e SL |
234 | // keyword that starts the head |
235 | if !w.is_empty() { | |
54a0048b | 236 | self.word_nbsp(w)?; |
e9174d1e SL |
237 | } |
238 | Ok(()) | |
239 | } | |
240 | ||
241 | pub fn bopen(&mut self) -> io::Result<()> { | |
041b39d2 | 242 | self.s.word("{")?; |
e9174d1e SL |
243 | self.end() // close the head-box |
244 | } | |
245 | ||
3157f602 | 246 | pub fn bclose_(&mut self, span: syntax_pos::Span, indented: usize) -> io::Result<()> { |
e9174d1e SL |
247 | self.bclose_maybe_open(span, indented, true) |
248 | } | |
b039eaaf | 249 | pub fn bclose_maybe_open(&mut self, |
3157f602 | 250 | span: syntax_pos::Span, |
b039eaaf SL |
251 | indented: usize, |
252 | close_box: bool) | |
253 | -> io::Result<()> { | |
ea8adc8c | 254 | self.maybe_print_comment(span.hi())?; |
54a0048b | 255 | self.break_offset_if_not_bol(1, -(indented as isize))?; |
041b39d2 | 256 | self.s.word("}")?; |
e9174d1e | 257 | if close_box { |
54a0048b | 258 | self.end()?; // close the outer-box |
e9174d1e SL |
259 | } |
260 | Ok(()) | |
261 | } | |
3157f602 | 262 | pub fn bclose(&mut self, span: syntax_pos::Span) -> io::Result<()> { |
e9174d1e SL |
263 | self.bclose_(span, indent_unit) |
264 | } | |
265 | ||
e9174d1e SL |
266 | pub fn in_cbox(&self) -> bool { |
267 | match self.boxes.last() { | |
268 | Some(&last_box) => last_box == pp::Breaks::Consistent, | |
b039eaaf | 269 | None => false, |
e9174d1e SL |
270 | } |
271 | } | |
b039eaaf | 272 | pub fn space_if_not_bol(&mut self) -> io::Result<()> { |
e9174d1e | 273 | if !self.is_bol() { |
041b39d2 | 274 | self.s.space()?; |
e9174d1e SL |
275 | } |
276 | Ok(()) | |
277 | } | |
b039eaaf | 278 | pub fn break_offset_if_not_bol(&mut self, n: usize, off: isize) -> io::Result<()> { |
e9174d1e | 279 | if !self.is_bol() { |
041b39d2 | 280 | self.s.break_offset(n, off) |
e9174d1e SL |
281 | } else { |
282 | if off != 0 && self.s.last_token().is_hardbreak_tok() { | |
283 | // We do something pretty sketchy here: tuck the nonzero | |
284 | // offset-adjustment we were going to deposit along with the | |
285 | // break into the previous hardbreak. | |
041b39d2 | 286 | self.s.replace_last_token(pp::Printer::hardbreak_tok_offset(off)); |
e9174d1e SL |
287 | } |
288 | Ok(()) | |
289 | } | |
290 | } | |
291 | ||
292 | // Synthesizes a comment that was not textually present in the original source | |
293 | // file. | |
294 | pub fn synth_comment(&mut self, text: String) -> io::Result<()> { | |
041b39d2 XL |
295 | self.s.word("/*")?; |
296 | self.s.space()?; | |
297 | self.s.word(&text[..])?; | |
298 | self.s.space()?; | |
299 | self.s.word("*/") | |
e9174d1e SL |
300 | } |
301 | ||
e9174d1e SL |
302 | |
303 | pub fn commasep_cmnt<T, F, G>(&mut self, | |
304 | b: Breaks, | |
305 | elts: &[T], | |
306 | mut op: F, | |
b039eaaf SL |
307 | mut get_span: G) |
308 | -> io::Result<()> | |
309 | where F: FnMut(&mut State, &T) -> io::Result<()>, | |
3157f602 | 310 | G: FnMut(&T) -> syntax_pos::Span |
e9174d1e | 311 | { |
54a0048b | 312 | self.rbox(0, b)?; |
e9174d1e SL |
313 | let len = elts.len(); |
314 | let mut i = 0; | |
315 | for elt in elts { | |
ea8adc8c | 316 | self.maybe_print_comment(get_span(elt).hi())?; |
54a0048b | 317 | op(self, elt)?; |
e9174d1e SL |
318 | i += 1; |
319 | if i < len { | |
041b39d2 | 320 | self.s.word(",")?; |
ea8adc8c | 321 | self.maybe_print_trailing_comment(get_span(elt), Some(get_span(&elts[i]).hi()))?; |
54a0048b | 322 | self.space_if_not_bol()?; |
e9174d1e SL |
323 | } |
324 | } | |
325 | self.end() | |
326 | } | |
327 | ||
476ff2be | 328 | pub fn commasep_exprs(&mut self, b: Breaks, exprs: &[hir::Expr]) -> io::Result<()> { |
7453a54e | 329 | self.commasep_cmnt(b, exprs, |s, e| s.print_expr(&e), |e| e.span) |
e9174d1e SL |
330 | } |
331 | ||
b039eaaf | 332 | pub fn print_mod(&mut self, _mod: &hir::Mod, attrs: &[ast::Attribute]) -> io::Result<()> { |
54a0048b | 333 | self.print_inner_attributes(attrs)?; |
32a655c1 SL |
334 | for &item_id in &_mod.item_ids { |
335 | self.ann.nested(self, Nested::Item(item_id))?; | |
e9174d1e SL |
336 | } |
337 | Ok(()) | |
338 | } | |
339 | ||
b039eaaf SL |
340 | pub fn print_foreign_mod(&mut self, |
341 | nmod: &hir::ForeignMod, | |
342 | attrs: &[ast::Attribute]) | |
343 | -> io::Result<()> { | |
54a0048b | 344 | self.print_inner_attributes(attrs)?; |
e9174d1e | 345 | for item in &nmod.items { |
54a0048b | 346 | self.print_foreign_item(item)?; |
e9174d1e SL |
347 | } |
348 | Ok(()) | |
349 | } | |
350 | ||
32a655c1 SL |
351 | pub fn print_opt_lifetime(&mut self, lifetime: &hir::Lifetime) -> io::Result<()> { |
352 | if !lifetime.is_elided() { | |
353 | self.print_lifetime(lifetime)?; | |
54a0048b | 354 | self.nbsp()?; |
e9174d1e SL |
355 | } |
356 | Ok(()) | |
357 | } | |
358 | ||
359 | pub fn print_type(&mut self, ty: &hir::Ty) -> io::Result<()> { | |
ea8adc8c | 360 | self.maybe_print_comment(ty.span.lo())?; |
54a0048b | 361 | self.ibox(0)?; |
e9174d1e | 362 | match ty.node { |
c30ab7b3 | 363 | hir::TySlice(ref ty) => { |
041b39d2 | 364 | self.s.word("[")?; |
54a0048b | 365 | self.print_type(&ty)?; |
041b39d2 | 366 | self.s.word("]")?; |
e9174d1e SL |
367 | } |
368 | hir::TyPtr(ref mt) => { | |
041b39d2 | 369 | self.s.word("*")?; |
e9174d1e | 370 | match mt.mutbl { |
54a0048b SL |
371 | hir::MutMutable => self.word_nbsp("mut")?, |
372 | hir::MutImmutable => self.word_nbsp("const")?, | |
e9174d1e | 373 | } |
54a0048b | 374 | self.print_type(&mt.ty)?; |
e9174d1e SL |
375 | } |
376 | hir::TyRptr(ref lifetime, ref mt) => { | |
041b39d2 | 377 | self.s.word("&")?; |
54a0048b SL |
378 | self.print_opt_lifetime(lifetime)?; |
379 | self.print_mt(mt)?; | |
e9174d1e | 380 | } |
5bcae85e | 381 | hir::TyNever => { |
041b39d2 | 382 | self.s.word("!")?; |
5bcae85e | 383 | }, |
e9174d1e | 384 | hir::TyTup(ref elts) => { |
54a0048b SL |
385 | self.popen()?; |
386 | self.commasep(Inconsistent, &elts[..], |s, ty| s.print_type(&ty))?; | |
e9174d1e | 387 | if elts.len() == 1 { |
041b39d2 | 388 | self.s.word(",")?; |
e9174d1e | 389 | } |
54a0048b | 390 | self.pclose()?; |
e9174d1e | 391 | } |
e9174d1e SL |
392 | hir::TyBareFn(ref f) => { |
393 | let generics = hir::Generics { | |
394 | lifetimes: f.lifetimes.clone(), | |
9cc50fc6 | 395 | ty_params: hir::HirVec::new(), |
e9174d1e SL |
396 | where_clause: hir::WhereClause { |
397 | id: ast::DUMMY_NODE_ID, | |
9cc50fc6 | 398 | predicates: hir::HirVec::new(), |
e9174d1e | 399 | }, |
9e0c209e | 400 | span: syntax_pos::DUMMY_SP, |
e9174d1e | 401 | }; |
ea8adc8c XL |
402 | self.print_ty_fn(f.abi, f.unsafety, &f.decl, None, &generics, |
403 | &f.arg_names[..])?; | |
e9174d1e | 404 | } |
476ff2be SL |
405 | hir::TyPath(ref qpath) => { |
406 | self.print_qpath(qpath, false)? | |
e9174d1e | 407 | } |
32a655c1 SL |
408 | hir::TyTraitObject(ref bounds, ref lifetime) => { |
409 | let mut first = true; | |
410 | for bound in bounds { | |
411 | self.nbsp()?; | |
412 | if first { | |
413 | first = false; | |
414 | } else { | |
415 | self.word_space("+")?; | |
416 | } | |
417 | self.print_poly_trait_ref(bound)?; | |
418 | } | |
419 | if !lifetime.is_elided() { | |
420 | self.word_space("+")?; | |
421 | self.print_lifetime(lifetime)?; | |
422 | } | |
e9174d1e | 423 | } |
5bcae85e SL |
424 | hir::TyImplTrait(ref bounds) => { |
425 | self.print_bounds("impl ", &bounds[..])?; | |
426 | } | |
32a655c1 | 427 | hir::TyArray(ref ty, v) => { |
041b39d2 | 428 | self.s.word("[")?; |
54a0048b | 429 | self.print_type(&ty)?; |
041b39d2 | 430 | self.s.word("; ")?; |
32a655c1 | 431 | self.ann.nested(self, Nested::Body(v))?; |
041b39d2 | 432 | self.s.word("]")?; |
e9174d1e | 433 | } |
32a655c1 | 434 | hir::TyTypeof(e) => { |
041b39d2 | 435 | self.s.word("typeof(")?; |
32a655c1 | 436 | self.ann.nested(self, Nested::Body(e))?; |
041b39d2 | 437 | self.s.word(")")?; |
e9174d1e SL |
438 | } |
439 | hir::TyInfer => { | |
041b39d2 | 440 | self.s.word("_")?; |
e9174d1e | 441 | } |
cc61c64b | 442 | hir::TyErr => { |
041b39d2 | 443 | self.s.word("?")?; |
cc61c64b | 444 | } |
e9174d1e SL |
445 | } |
446 | self.end() | |
447 | } | |
448 | ||
b039eaaf | 449 | pub fn print_foreign_item(&mut self, item: &hir::ForeignItem) -> io::Result<()> { |
54a0048b | 450 | self.hardbreak_if_not_bol()?; |
ea8adc8c | 451 | self.maybe_print_comment(item.span.lo())?; |
54a0048b | 452 | self.print_outer_attributes(&item.attrs)?; |
e9174d1e | 453 | match item.node { |
32a655c1 | 454 | hir::ForeignItemFn(ref decl, ref arg_names, ref generics) => { |
54a0048b SL |
455 | self.head("")?; |
456 | self.print_fn(decl, | |
457 | hir::Unsafety::Normal, | |
458 | hir::Constness::NotConst, | |
459 | Abi::Rust, | |
460 | Some(item.name), | |
461 | generics, | |
32a655c1 SL |
462 | &item.vis, |
463 | arg_names, | |
464 | None)?; | |
54a0048b | 465 | self.end()?; // end head-ibox |
041b39d2 | 466 | self.s.word(";")?; |
e9174d1e SL |
467 | self.end() // end the outer fn box |
468 | } | |
469 | hir::ForeignItemStatic(ref t, m) => { | |
54a0048b | 470 | self.head(&visibility_qualified(&item.vis, "static"))?; |
e9174d1e | 471 | if m { |
54a0048b | 472 | self.word_space("mut")?; |
e9174d1e | 473 | } |
54a0048b SL |
474 | self.print_name(item.name)?; |
475 | self.word_space(":")?; | |
476 | self.print_type(&t)?; | |
041b39d2 | 477 | self.s.word(";")?; |
54a0048b | 478 | self.end()?; // end the head-ibox |
e9174d1e SL |
479 | self.end() // end the outer cbox |
480 | } | |
481 | } | |
482 | } | |
483 | ||
484 | fn print_associated_const(&mut self, | |
b039eaaf | 485 | name: ast::Name, |
e9174d1e | 486 | ty: &hir::Ty, |
32a655c1 | 487 | default: Option<hir::BodyId>, |
54a0048b | 488 | vis: &hir::Visibility) |
b039eaaf | 489 | -> io::Result<()> { |
041b39d2 | 490 | self.s.word(&visibility_qualified(vis, ""))?; |
54a0048b SL |
491 | self.word_space("const")?; |
492 | self.print_name(name)?; | |
493 | self.word_space(":")?; | |
494 | self.print_type(ty)?; | |
e9174d1e | 495 | if let Some(expr) = default { |
041b39d2 | 496 | self.s.space()?; |
54a0048b | 497 | self.word_space("=")?; |
32a655c1 | 498 | self.ann.nested(self, Nested::Body(expr))?; |
e9174d1e | 499 | } |
041b39d2 | 500 | self.s.word(";") |
e9174d1e SL |
501 | } |
502 | ||
503 | fn print_associated_type(&mut self, | |
b039eaaf | 504 | name: ast::Name, |
e9174d1e SL |
505 | bounds: Option<&hir::TyParamBounds>, |
506 | ty: Option<&hir::Ty>) | |
507 | -> io::Result<()> { | |
54a0048b SL |
508 | self.word_space("type")?; |
509 | self.print_name(name)?; | |
e9174d1e | 510 | if let Some(bounds) = bounds { |
54a0048b | 511 | self.print_bounds(":", bounds)?; |
e9174d1e SL |
512 | } |
513 | if let Some(ty) = ty { | |
041b39d2 | 514 | self.s.space()?; |
54a0048b SL |
515 | self.word_space("=")?; |
516 | self.print_type(ty)?; | |
e9174d1e | 517 | } |
041b39d2 | 518 | self.s.word(";") |
e9174d1e SL |
519 | } |
520 | ||
521 | /// Pretty-print an item | |
522 | pub fn print_item(&mut self, item: &hir::Item) -> io::Result<()> { | |
54a0048b | 523 | self.hardbreak_if_not_bol()?; |
ea8adc8c | 524 | self.maybe_print_comment(item.span.lo())?; |
54a0048b SL |
525 | self.print_outer_attributes(&item.attrs)?; |
526 | self.ann.pre(self, NodeItem(item))?; | |
e9174d1e SL |
527 | match item.node { |
528 | hir::ItemExternCrate(ref optional_path) => { | |
54a0048b | 529 | self.head(&visibility_qualified(&item.vis, "extern crate"))?; |
e9174d1e SL |
530 | if let Some(p) = *optional_path { |
531 | let val = p.as_str(); | |
532 | if val.contains("-") { | |
54a0048b | 533 | self.print_string(&val, ast::StrStyle::Cooked)?; |
e9174d1e | 534 | } else { |
54a0048b | 535 | self.print_name(p)?; |
e9174d1e | 536 | } |
041b39d2 XL |
537 | self.s.space()?; |
538 | self.s.word("as")?; | |
539 | self.s.space()?; | |
e9174d1e | 540 | } |
54a0048b | 541 | self.print_name(item.name)?; |
041b39d2 | 542 | self.s.word(";")?; |
54a0048b SL |
543 | self.end()?; // end inner head-block |
544 | self.end()?; // end outer head-block | |
e9174d1e | 545 | } |
476ff2be | 546 | hir::ItemUse(ref path, kind) => { |
54a0048b | 547 | self.head(&visibility_qualified(&item.vis, "use"))?; |
476ff2be SL |
548 | self.print_path(path, false)?; |
549 | ||
550 | match kind { | |
551 | hir::UseKind::Single => { | |
552 | if path.segments.last().unwrap().name != item.name { | |
041b39d2 | 553 | self.s.space()?; |
476ff2be SL |
554 | self.word_space("as")?; |
555 | self.print_name(item.name)?; | |
556 | } | |
041b39d2 | 557 | self.s.word(";")?; |
476ff2be | 558 | } |
041b39d2 XL |
559 | hir::UseKind::Glob => self.s.word("::*;")?, |
560 | hir::UseKind::ListStem => self.s.word("::{};")? | |
476ff2be | 561 | } |
54a0048b SL |
562 | self.end()?; // end inner head-block |
563 | self.end()?; // end outer head-block | |
e9174d1e | 564 | } |
32a655c1 | 565 | hir::ItemStatic(ref ty, m, expr) => { |
54a0048b | 566 | self.head(&visibility_qualified(&item.vis, "static"))?; |
e9174d1e | 567 | if m == hir::MutMutable { |
54a0048b | 568 | self.word_space("mut")?; |
e9174d1e | 569 | } |
54a0048b SL |
570 | self.print_name(item.name)?; |
571 | self.word_space(":")?; | |
572 | self.print_type(&ty)?; | |
041b39d2 | 573 | self.s.space()?; |
54a0048b | 574 | self.end()?; // end the head-ibox |
e9174d1e | 575 | |
54a0048b | 576 | self.word_space("=")?; |
32a655c1 | 577 | self.ann.nested(self, Nested::Body(expr))?; |
041b39d2 | 578 | self.s.word(";")?; |
54a0048b | 579 | self.end()?; // end the outer cbox |
e9174d1e | 580 | } |
32a655c1 | 581 | hir::ItemConst(ref ty, expr) => { |
54a0048b SL |
582 | self.head(&visibility_qualified(&item.vis, "const"))?; |
583 | self.print_name(item.name)?; | |
584 | self.word_space(":")?; | |
585 | self.print_type(&ty)?; | |
041b39d2 | 586 | self.s.space()?; |
54a0048b SL |
587 | self.end()?; // end the head-ibox |
588 | ||
589 | self.word_space("=")?; | |
32a655c1 | 590 | self.ann.nested(self, Nested::Body(expr))?; |
041b39d2 | 591 | self.s.word(";")?; |
54a0048b | 592 | self.end()?; // end the outer cbox |
e9174d1e | 593 | } |
32a655c1 | 594 | hir::ItemFn(ref decl, unsafety, constness, abi, ref typarams, body) => { |
54a0048b SL |
595 | self.head("")?; |
596 | self.print_fn(decl, | |
597 | unsafety, | |
598 | constness, | |
599 | abi, | |
600 | Some(item.name), | |
601 | typarams, | |
32a655c1 SL |
602 | &item.vis, |
603 | &[], | |
604 | Some(body))?; | |
041b39d2 | 605 | self.s.word(" ")?; |
476ff2be SL |
606 | self.end()?; // need to close a box |
607 | self.end()?; // need to close a box | |
32a655c1 | 608 | self.ann.nested(self, Nested::Body(body))?; |
e9174d1e SL |
609 | } |
610 | hir::ItemMod(ref _mod) => { | |
54a0048b SL |
611 | self.head(&visibility_qualified(&item.vis, "mod"))?; |
612 | self.print_name(item.name)?; | |
613 | self.nbsp()?; | |
614 | self.bopen()?; | |
615 | self.print_mod(_mod, &item.attrs)?; | |
616 | self.bclose(item.span)?; | |
e9174d1e SL |
617 | } |
618 | hir::ItemForeignMod(ref nmod) => { | |
54a0048b SL |
619 | self.head("extern")?; |
620 | self.word_nbsp(&nmod.abi.to_string())?; | |
621 | self.bopen()?; | |
622 | self.print_foreign_mod(nmod, &item.attrs)?; | |
623 | self.bclose(item.span)?; | |
e9174d1e | 624 | } |
cc61c64b XL |
625 | hir::ItemGlobalAsm(ref ga) => { |
626 | self.head(&visibility_qualified(&item.vis, "global asm"))?; | |
041b39d2 | 627 | self.s.word(&ga.asm.as_str())?; |
cc61c64b XL |
628 | self.end()? |
629 | } | |
e9174d1e | 630 | hir::ItemTy(ref ty, ref params) => { |
54a0048b SL |
631 | self.ibox(indent_unit)?; |
632 | self.ibox(0)?; | |
633 | self.word_nbsp(&visibility_qualified(&item.vis, "type"))?; | |
634 | self.print_name(item.name)?; | |
635 | self.print_generics(params)?; | |
636 | self.end()?; // end the inner ibox | |
637 | ||
638 | self.print_where_clause(¶ms.where_clause)?; | |
041b39d2 | 639 | self.s.space()?; |
54a0048b SL |
640 | self.word_space("=")?; |
641 | self.print_type(&ty)?; | |
041b39d2 | 642 | self.s.word(";")?; |
54a0048b | 643 | self.end()?; // end the outer ibox |
e9174d1e SL |
644 | } |
645 | hir::ItemEnum(ref enum_definition, ref params) => { | |
54a0048b | 646 | self.print_enum_def(enum_definition, params, item.name, item.span, &item.vis)?; |
e9174d1e SL |
647 | } |
648 | hir::ItemStruct(ref struct_def, ref generics) => { | |
54a0048b SL |
649 | self.head(&visibility_qualified(&item.vis, "struct"))?; |
650 | self.print_struct(struct_def, generics, item.name, item.span, true)?; | |
e9174d1e | 651 | } |
9e0c209e SL |
652 | hir::ItemUnion(ref struct_def, ref generics) => { |
653 | self.head(&visibility_qualified(&item.vis, "union"))?; | |
654 | self.print_struct(struct_def, generics, item.name, item.span, true)?; | |
655 | } | |
e9174d1e | 656 | hir::ItemDefaultImpl(unsafety, ref trait_ref) => { |
54a0048b SL |
657 | self.head("")?; |
658 | self.print_visibility(&item.vis)?; | |
659 | self.print_unsafety(unsafety)?; | |
660 | self.word_nbsp("impl")?; | |
661 | self.print_trait_ref(trait_ref)?; | |
041b39d2 | 662 | self.s.space()?; |
54a0048b SL |
663 | self.word_space("for")?; |
664 | self.word_space("..")?; | |
665 | self.bopen()?; | |
666 | self.bclose(item.span)?; | |
e9174d1e SL |
667 | } |
668 | hir::ItemImpl(unsafety, | |
669 | polarity, | |
7cac9316 | 670 | defaultness, |
e9174d1e SL |
671 | ref generics, |
672 | ref opt_trait, | |
673 | ref ty, | |
674 | ref impl_items) => { | |
54a0048b SL |
675 | self.head("")?; |
676 | self.print_visibility(&item.vis)?; | |
7cac9316 | 677 | self.print_defaultness(defaultness)?; |
54a0048b SL |
678 | self.print_unsafety(unsafety)?; |
679 | self.word_nbsp("impl")?; | |
e9174d1e SL |
680 | |
681 | if generics.is_parameterized() { | |
54a0048b | 682 | self.print_generics(generics)?; |
041b39d2 | 683 | self.s.space()?; |
e9174d1e SL |
684 | } |
685 | ||
686 | match polarity { | |
687 | hir::ImplPolarity::Negative => { | |
041b39d2 | 688 | self.s.word("!")?; |
b039eaaf | 689 | } |
e9174d1e SL |
690 | _ => {} |
691 | } | |
692 | ||
693 | match opt_trait { | |
694 | &Some(ref t) => { | |
54a0048b | 695 | self.print_trait_ref(t)?; |
041b39d2 | 696 | self.s.space()?; |
54a0048b | 697 | self.word_space("for")?; |
e9174d1e SL |
698 | } |
699 | &None => {} | |
700 | } | |
701 | ||
54a0048b SL |
702 | self.print_type(&ty)?; |
703 | self.print_where_clause(&generics.where_clause)?; | |
e9174d1e | 704 | |
041b39d2 | 705 | self.s.space()?; |
54a0048b SL |
706 | self.bopen()?; |
707 | self.print_inner_attributes(&item.attrs)?; | |
e9174d1e | 708 | for impl_item in impl_items { |
32a655c1 | 709 | self.ann.nested(self, Nested::ImplItem(impl_item.id))?; |
e9174d1e | 710 | } |
54a0048b | 711 | self.bclose(item.span)?; |
e9174d1e SL |
712 | } |
713 | hir::ItemTrait(unsafety, ref generics, ref bounds, ref trait_items) => { | |
54a0048b SL |
714 | self.head("")?; |
715 | self.print_visibility(&item.vis)?; | |
716 | self.print_unsafety(unsafety)?; | |
717 | self.word_nbsp("trait")?; | |
718 | self.print_name(item.name)?; | |
719 | self.print_generics(generics)?; | |
e9174d1e SL |
720 | let mut real_bounds = Vec::with_capacity(bounds.len()); |
721 | for b in bounds.iter() { | |
722 | if let TraitTyParamBound(ref ptr, hir::TraitBoundModifier::Maybe) = *b { | |
041b39d2 | 723 | self.s.space()?; |
54a0048b SL |
724 | self.word_space("for ?")?; |
725 | self.print_trait_ref(&ptr.trait_ref)?; | |
e9174d1e SL |
726 | } else { |
727 | real_bounds.push(b.clone()); | |
728 | } | |
729 | } | |
54a0048b SL |
730 | self.print_bounds(":", &real_bounds[..])?; |
731 | self.print_where_clause(&generics.where_clause)?; | |
041b39d2 | 732 | self.s.word(" ")?; |
54a0048b | 733 | self.bopen()?; |
e9174d1e | 734 | for trait_item in trait_items { |
32a655c1 | 735 | self.ann.nested(self, Nested::TraitItem(trait_item.id))?; |
e9174d1e | 736 | } |
54a0048b | 737 | self.bclose(item.span)?; |
e9174d1e SL |
738 | } |
739 | } | |
740 | self.ann.post(self, NodeItem(item)) | |
741 | } | |
742 | ||
476ff2be SL |
743 | pub fn print_trait_ref(&mut self, t: &hir::TraitRef) -> io::Result<()> { |
744 | self.print_path(&t.path, false) | |
e9174d1e SL |
745 | } |
746 | ||
747 | fn print_formal_lifetime_list(&mut self, lifetimes: &[hir::LifetimeDef]) -> io::Result<()> { | |
748 | if !lifetimes.is_empty() { | |
041b39d2 | 749 | self.s.word("for<")?; |
e9174d1e SL |
750 | let mut comma = false; |
751 | for lifetime_def in lifetimes { | |
752 | if comma { | |
54a0048b | 753 | self.word_space(",")? |
e9174d1e | 754 | } |
54a0048b | 755 | self.print_lifetime_def(lifetime_def)?; |
e9174d1e SL |
756 | comma = true; |
757 | } | |
041b39d2 | 758 | self.s.word(">")?; |
e9174d1e SL |
759 | } |
760 | Ok(()) | |
761 | } | |
762 | ||
763 | fn print_poly_trait_ref(&mut self, t: &hir::PolyTraitRef) -> io::Result<()> { | |
54a0048b | 764 | self.print_formal_lifetime_list(&t.bound_lifetimes)?; |
e9174d1e SL |
765 | self.print_trait_ref(&t.trait_ref) |
766 | } | |
767 | ||
b039eaaf SL |
768 | pub fn print_enum_def(&mut self, |
769 | enum_definition: &hir::EnumDef, | |
770 | generics: &hir::Generics, | |
771 | name: ast::Name, | |
3157f602 | 772 | span: syntax_pos::Span, |
54a0048b | 773 | visibility: &hir::Visibility) |
b039eaaf | 774 | -> io::Result<()> { |
54a0048b SL |
775 | self.head(&visibility_qualified(visibility, "enum"))?; |
776 | self.print_name(name)?; | |
777 | self.print_generics(generics)?; | |
778 | self.print_where_clause(&generics.where_clause)?; | |
041b39d2 | 779 | self.s.space()?; |
e9174d1e SL |
780 | self.print_variants(&enum_definition.variants, span) |
781 | } | |
782 | ||
783 | pub fn print_variants(&mut self, | |
92a42be0 | 784 | variants: &[hir::Variant], |
3157f602 | 785 | span: syntax_pos::Span) |
b039eaaf | 786 | -> io::Result<()> { |
54a0048b | 787 | self.bopen()?; |
e9174d1e | 788 | for v in variants { |
54a0048b | 789 | self.space_if_not_bol()?; |
ea8adc8c | 790 | self.maybe_print_comment(v.span.lo())?; |
54a0048b SL |
791 | self.print_outer_attributes(&v.node.attrs)?; |
792 | self.ibox(indent_unit)?; | |
793 | self.print_variant(v)?; | |
041b39d2 | 794 | self.s.word(",")?; |
54a0048b SL |
795 | self.end()?; |
796 | self.maybe_print_trailing_comment(v.span, None)?; | |
e9174d1e SL |
797 | } |
798 | self.bclose(span) | |
799 | } | |
800 | ||
54a0048b SL |
801 | pub fn print_visibility(&mut self, vis: &hir::Visibility) -> io::Result<()> { |
802 | match *vis { | |
e9174d1e | 803 | hir::Public => self.word_nbsp("pub"), |
a7813a04 | 804 | hir::Visibility::Crate => self.word_nbsp("pub(crate)"), |
32a655c1 | 805 | hir::Visibility::Restricted { ref path, .. } => { |
041b39d2 | 806 | self.s.word("pub(")?; |
32a655c1 SL |
807 | self.print_path(path, false)?; |
808 | self.word_nbsp(")") | |
809 | } | |
b039eaaf | 810 | hir::Inherited => Ok(()), |
e9174d1e SL |
811 | } |
812 | } | |
813 | ||
7cac9316 XL |
814 | pub fn print_defaultness(&mut self, defaultness: hir::Defaultness) -> io::Result<()> { |
815 | match defaultness { | |
816 | hir::Defaultness::Default { .. } => self.word_nbsp("default")?, | |
817 | hir::Defaultness::Final => (), | |
818 | } | |
819 | Ok(()) | |
820 | } | |
821 | ||
e9174d1e | 822 | pub fn print_struct(&mut self, |
b039eaaf | 823 | struct_def: &hir::VariantData, |
e9174d1e | 824 | generics: &hir::Generics, |
b039eaaf | 825 | name: ast::Name, |
3157f602 | 826 | span: syntax_pos::Span, |
b039eaaf SL |
827 | print_finalizer: bool) |
828 | -> io::Result<()> { | |
54a0048b SL |
829 | self.print_name(name)?; |
830 | self.print_generics(generics)?; | |
b039eaaf SL |
831 | if !struct_def.is_struct() { |
832 | if struct_def.is_tuple() { | |
54a0048b SL |
833 | self.popen()?; |
834 | self.commasep(Inconsistent, struct_def.fields(), |s, field| { | |
ea8adc8c | 835 | s.maybe_print_comment(field.span.lo())?; |
3157f602 XL |
836 | s.print_outer_attributes(&field.attrs)?; |
837 | s.print_visibility(&field.vis)?; | |
54a0048b SL |
838 | s.print_type(&field.ty) |
839 | })?; | |
840 | self.pclose()?; | |
841 | } | |
842 | self.print_where_clause(&generics.where_clause)?; | |
b039eaaf | 843 | if print_finalizer { |
041b39d2 | 844 | self.s.word(";")?; |
b039eaaf | 845 | } |
54a0048b | 846 | self.end()?; |
e9174d1e SL |
847 | self.end() // close the outer-box |
848 | } else { | |
54a0048b SL |
849 | self.print_where_clause(&generics.where_clause)?; |
850 | self.nbsp()?; | |
851 | self.bopen()?; | |
852 | self.hardbreak_if_not_bol()?; | |
e9174d1e | 853 | |
b039eaaf | 854 | for field in struct_def.fields() { |
54a0048b | 855 | self.hardbreak_if_not_bol()?; |
ea8adc8c | 856 | self.maybe_print_comment(field.span.lo())?; |
54a0048b SL |
857 | self.print_outer_attributes(&field.attrs)?; |
858 | self.print_visibility(&field.vis)?; | |
859 | self.print_name(field.name)?; | |
860 | self.word_nbsp(":")?; | |
861 | self.print_type(&field.ty)?; | |
041b39d2 | 862 | self.s.word(",")?; |
e9174d1e SL |
863 | } |
864 | ||
865 | self.bclose(span) | |
866 | } | |
867 | } | |
868 | ||
869 | pub fn print_variant(&mut self, v: &hir::Variant) -> io::Result<()> { | |
54a0048b SL |
870 | self.head("")?; |
871 | let generics = hir::Generics::empty(); | |
872 | self.print_struct(&v.node.data, &generics, v.node.name, v.span, false)?; | |
32a655c1 | 873 | if let Some(d) = v.node.disr_expr { |
041b39d2 | 874 | self.s.space()?; |
32a655c1 SL |
875 | self.word_space("=")?; |
876 | self.ann.nested(self, Nested::Body(d))?; | |
e9174d1e | 877 | } |
32a655c1 | 878 | Ok(()) |
e9174d1e | 879 | } |
e9174d1e | 880 | pub fn print_method_sig(&mut self, |
b039eaaf | 881 | name: ast::Name, |
e9174d1e | 882 | m: &hir::MethodSig, |
32a655c1 SL |
883 | vis: &hir::Visibility, |
884 | arg_names: &[Spanned<ast::Name>], | |
885 | body_id: Option<hir::BodyId>) | |
e9174d1e SL |
886 | -> io::Result<()> { |
887 | self.print_fn(&m.decl, | |
888 | m.unsafety, | |
889 | m.constness, | |
890 | m.abi, | |
b039eaaf | 891 | Some(name), |
e9174d1e | 892 | &m.generics, |
32a655c1 SL |
893 | vis, |
894 | arg_names, | |
895 | body_id) | |
e9174d1e SL |
896 | } |
897 | ||
b039eaaf | 898 | pub fn print_trait_item(&mut self, ti: &hir::TraitItem) -> io::Result<()> { |
54a0048b SL |
899 | self.ann.pre(self, NodeSubItem(ti.id))?; |
900 | self.hardbreak_if_not_bol()?; | |
ea8adc8c | 901 | self.maybe_print_comment(ti.span.lo())?; |
54a0048b | 902 | self.print_outer_attributes(&ti.attrs)?; |
e9174d1e | 903 | match ti.node { |
32a655c1 SL |
904 | hir::TraitItemKind::Const(ref ty, default) => { |
905 | self.print_associated_const(ti.name, &ty, default, &hir::Inherited)?; | |
e9174d1e | 906 | } |
32a655c1 SL |
907 | hir::TraitItemKind::Method(ref sig, hir::TraitMethod::Required(ref arg_names)) => { |
908 | self.print_method_sig(ti.name, sig, &hir::Inherited, arg_names, None)?; | |
041b39d2 | 909 | self.s.word(";")?; |
32a655c1 SL |
910 | } |
911 | hir::TraitItemKind::Method(ref sig, hir::TraitMethod::Provided(body)) => { | |
912 | self.head("")?; | |
913 | self.print_method_sig(ti.name, sig, &hir::Inherited, &[], Some(body))?; | |
914 | self.nbsp()?; | |
915 | self.end()?; // need to close a box | |
916 | self.end()?; // need to close a box | |
917 | self.ann.nested(self, Nested::Body(body))?; | |
918 | } | |
919 | hir::TraitItemKind::Type(ref bounds, ref default) => { | |
54a0048b SL |
920 | self.print_associated_type(ti.name, |
921 | Some(bounds), | |
922 | default.as_ref().map(|ty| &**ty))?; | |
e9174d1e SL |
923 | } |
924 | } | |
925 | self.ann.post(self, NodeSubItem(ti.id)) | |
926 | } | |
927 | ||
928 | pub fn print_impl_item(&mut self, ii: &hir::ImplItem) -> io::Result<()> { | |
54a0048b SL |
929 | self.ann.pre(self, NodeSubItem(ii.id))?; |
930 | self.hardbreak_if_not_bol()?; | |
ea8adc8c | 931 | self.maybe_print_comment(ii.span.lo())?; |
54a0048b | 932 | self.print_outer_attributes(&ii.attrs)?; |
7cac9316 | 933 | self.print_defaultness(ii.defaultness)?; |
54a0048b | 934 | |
e9174d1e | 935 | match ii.node { |
32a655c1 SL |
936 | hir::ImplItemKind::Const(ref ty, expr) => { |
937 | self.print_associated_const(ii.name, &ty, Some(expr), &ii.vis)?; | |
e9174d1e | 938 | } |
32a655c1 | 939 | hir::ImplItemKind::Method(ref sig, body) => { |
54a0048b | 940 | self.head("")?; |
32a655c1 | 941 | self.print_method_sig(ii.name, sig, &ii.vis, &[], Some(body))?; |
54a0048b | 942 | self.nbsp()?; |
476ff2be SL |
943 | self.end()?; // need to close a box |
944 | self.end()?; // need to close a box | |
32a655c1 | 945 | self.ann.nested(self, Nested::Body(body))?; |
e9174d1e | 946 | } |
92a42be0 | 947 | hir::ImplItemKind::Type(ref ty) => { |
54a0048b | 948 | self.print_associated_type(ii.name, None, Some(ty))?; |
e9174d1e SL |
949 | } |
950 | } | |
951 | self.ann.post(self, NodeSubItem(ii.id)) | |
952 | } | |
953 | ||
e9174d1e | 954 | pub fn print_stmt(&mut self, st: &hir::Stmt) -> io::Result<()> { |
ea8adc8c | 955 | self.maybe_print_comment(st.span.lo())?; |
e9174d1e SL |
956 | match st.node { |
957 | hir::StmtDecl(ref decl, _) => { | |
54a0048b | 958 | self.print_decl(&decl)?; |
e9174d1e SL |
959 | } |
960 | hir::StmtExpr(ref expr, _) => { | |
54a0048b SL |
961 | self.space_if_not_bol()?; |
962 | self.print_expr(&expr)?; | |
e9174d1e SL |
963 | } |
964 | hir::StmtSemi(ref expr, _) => { | |
54a0048b SL |
965 | self.space_if_not_bol()?; |
966 | self.print_expr(&expr)?; | |
041b39d2 | 967 | self.s.word(";")?; |
e9174d1e SL |
968 | } |
969 | } | |
970 | if stmt_ends_with_semi(&st.node) { | |
041b39d2 | 971 | self.s.word(";")?; |
e9174d1e SL |
972 | } |
973 | self.maybe_print_trailing_comment(st.span, None) | |
974 | } | |
975 | ||
976 | pub fn print_block(&mut self, blk: &hir::Block) -> io::Result<()> { | |
977 | self.print_block_with_attrs(blk, &[]) | |
978 | } | |
979 | ||
980 | pub fn print_block_unclosed(&mut self, blk: &hir::Block) -> io::Result<()> { | |
981 | self.print_block_unclosed_indent(blk, indent_unit) | |
982 | } | |
983 | ||
b039eaaf SL |
984 | pub fn print_block_unclosed_indent(&mut self, |
985 | blk: &hir::Block, | |
986 | indented: usize) | |
987 | -> io::Result<()> { | |
e9174d1e SL |
988 | self.print_block_maybe_unclosed(blk, indented, &[], false) |
989 | } | |
990 | ||
991 | pub fn print_block_with_attrs(&mut self, | |
992 | blk: &hir::Block, | |
b039eaaf SL |
993 | attrs: &[ast::Attribute]) |
994 | -> io::Result<()> { | |
e9174d1e SL |
995 | self.print_block_maybe_unclosed(blk, indent_unit, attrs, true) |
996 | } | |
997 | ||
998 | pub fn print_block_maybe_unclosed(&mut self, | |
999 | blk: &hir::Block, | |
1000 | indented: usize, | |
b039eaaf SL |
1001 | attrs: &[ast::Attribute], |
1002 | close_box: bool) | |
1003 | -> io::Result<()> { | |
e9174d1e | 1004 | match blk.rules { |
54a0048b SL |
1005 | hir::UnsafeBlock(..) => self.word_space("unsafe")?, |
1006 | hir::PushUnsafeBlock(..) => self.word_space("push_unsafe")?, | |
1007 | hir::PopUnsafeBlock(..) => self.word_space("pop_unsafe")?, | |
b039eaaf | 1008 | hir::DefaultBlock => (), |
e9174d1e | 1009 | } |
ea8adc8c | 1010 | self.maybe_print_comment(blk.span.lo())?; |
54a0048b SL |
1011 | self.ann.pre(self, NodeBlock(blk))?; |
1012 | self.bopen()?; | |
e9174d1e | 1013 | |
54a0048b | 1014 | self.print_inner_attributes(attrs)?; |
e9174d1e SL |
1015 | |
1016 | for st in &blk.stmts { | |
54a0048b | 1017 | self.print_stmt(st)?; |
e9174d1e SL |
1018 | } |
1019 | match blk.expr { | |
1020 | Some(ref expr) => { | |
54a0048b SL |
1021 | self.space_if_not_bol()?; |
1022 | self.print_expr(&expr)?; | |
ea8adc8c | 1023 | self.maybe_print_trailing_comment(expr.span, Some(blk.span.hi()))?; |
e9174d1e | 1024 | } |
b039eaaf | 1025 | _ => (), |
e9174d1e | 1026 | } |
54a0048b | 1027 | self.bclose_maybe_open(blk.span, indented, close_box)?; |
e9174d1e SL |
1028 | self.ann.post(self, NodeBlock(blk)) |
1029 | } | |
1030 | ||
1031 | fn print_else(&mut self, els: Option<&hir::Expr>) -> io::Result<()> { | |
1032 | match els { | |
1033 | Some(_else) => { | |
1034 | match _else.node { | |
1035 | // "another else-if" | |
1036 | hir::ExprIf(ref i, ref then, ref e) => { | |
54a0048b SL |
1037 | self.cbox(indent_unit - 1)?; |
1038 | self.ibox(0)?; | |
041b39d2 | 1039 | self.s.word(" else if ")?; |
ea8adc8c | 1040 | self.print_expr_as_cond(&i)?; |
041b39d2 | 1041 | self.s.space()?; |
cc61c64b | 1042 | self.print_expr(&then)?; |
e9174d1e SL |
1043 | self.print_else(e.as_ref().map(|e| &**e)) |
1044 | } | |
1045 | // "final else" | |
1046 | hir::ExprBlock(ref b) => { | |
54a0048b SL |
1047 | self.cbox(indent_unit - 1)?; |
1048 | self.ibox(0)?; | |
041b39d2 | 1049 | self.s.word(" else ")?; |
7453a54e | 1050 | self.print_block(&b) |
e9174d1e SL |
1051 | } |
1052 | // BLEAH, constraints would be great here | |
1053 | _ => { | |
1054 | panic!("print_if saw if with weird alternative"); | |
1055 | } | |
1056 | } | |
1057 | } | |
b039eaaf | 1058 | _ => Ok(()), |
e9174d1e SL |
1059 | } |
1060 | } | |
1061 | ||
b039eaaf SL |
1062 | pub fn print_if(&mut self, |
1063 | test: &hir::Expr, | |
cc61c64b | 1064 | blk: &hir::Expr, |
b039eaaf SL |
1065 | elseopt: Option<&hir::Expr>) |
1066 | -> io::Result<()> { | |
54a0048b | 1067 | self.head("if")?; |
ea8adc8c | 1068 | self.print_expr_as_cond(test)?; |
041b39d2 | 1069 | self.s.space()?; |
cc61c64b | 1070 | self.print_expr(blk)?; |
e9174d1e SL |
1071 | self.print_else(elseopt) |
1072 | } | |
1073 | ||
b039eaaf SL |
1074 | pub fn print_if_let(&mut self, |
1075 | pat: &hir::Pat, | |
1076 | expr: &hir::Expr, | |
1077 | blk: &hir::Block, | |
1078 | elseopt: Option<&hir::Expr>) | |
1079 | -> io::Result<()> { | |
54a0048b SL |
1080 | self.head("if let")?; |
1081 | self.print_pat(pat)?; | |
041b39d2 | 1082 | self.s.space()?; |
54a0048b | 1083 | self.word_space("=")?; |
ea8adc8c | 1084 | self.print_expr_as_cond(expr)?; |
041b39d2 | 1085 | self.s.space()?; |
54a0048b | 1086 | self.print_block(blk)?; |
e9174d1e SL |
1087 | self.print_else(elseopt) |
1088 | } | |
1089 | ||
1090 | ||
476ff2be | 1091 | fn print_call_post(&mut self, args: &[hir::Expr]) -> io::Result<()> { |
54a0048b SL |
1092 | self.popen()?; |
1093 | self.commasep_exprs(Inconsistent, args)?; | |
e9174d1e SL |
1094 | self.pclose() |
1095 | } | |
1096 | ||
ea8adc8c XL |
1097 | pub fn print_expr_maybe_paren(&mut self, expr: &hir::Expr, prec: i8) -> io::Result<()> { |
1098 | let needs_par = expr_precedence(expr) < prec; | |
1099 | if needs_par { | |
1100 | self.popen()?; | |
1101 | } | |
1102 | self.print_expr(expr)?; | |
1103 | if needs_par { | |
1104 | self.pclose()?; | |
1105 | } | |
1106 | Ok(()) | |
1107 | } | |
1108 | ||
1109 | /// Print an expr using syntax that's acceptable in a condition position, such as the `cond` in | |
1110 | /// `if cond { ... }`. | |
1111 | pub fn print_expr_as_cond(&mut self, expr: &hir::Expr) -> io::Result<()> { | |
1112 | let needs_par = match expr.node { | |
1113 | // These cases need parens due to the parse error observed in #26461: `if return {}` | |
1114 | // parses as the erroneous construct `if (return {})`, not `if (return) {}`. | |
1115 | hir::ExprClosure(..) | | |
1116 | hir::ExprRet(..) | | |
1117 | hir::ExprBreak(..) => true, | |
1118 | ||
1119 | _ => contains_exterior_struct_lit(expr), | |
1120 | }; | |
1121 | ||
e9174d1e | 1122 | if needs_par { |
54a0048b | 1123 | self.popen()?; |
e9174d1e | 1124 | } |
54a0048b | 1125 | self.print_expr(expr)?; |
e9174d1e | 1126 | if needs_par { |
54a0048b | 1127 | self.pclose()?; |
e9174d1e SL |
1128 | } |
1129 | Ok(()) | |
1130 | } | |
1131 | ||
476ff2be | 1132 | fn print_expr_vec(&mut self, exprs: &[hir::Expr]) -> io::Result<()> { |
54a0048b | 1133 | self.ibox(indent_unit)?; |
041b39d2 | 1134 | self.s.word("[")?; |
476ff2be | 1135 | self.commasep_exprs(Inconsistent, exprs)?; |
041b39d2 | 1136 | self.s.word("]")?; |
e9174d1e SL |
1137 | self.end() |
1138 | } | |
1139 | ||
32a655c1 | 1140 | fn print_expr_repeat(&mut self, element: &hir::Expr, count: hir::BodyId) -> io::Result<()> { |
54a0048b | 1141 | self.ibox(indent_unit)?; |
041b39d2 | 1142 | self.s.word("[")?; |
54a0048b SL |
1143 | self.print_expr(element)?; |
1144 | self.word_space(";")?; | |
32a655c1 | 1145 | self.ann.nested(self, Nested::Body(count))?; |
041b39d2 | 1146 | self.s.word("]")?; |
e9174d1e SL |
1147 | self.end() |
1148 | } | |
1149 | ||
1150 | fn print_expr_struct(&mut self, | |
476ff2be | 1151 | qpath: &hir::QPath, |
e9174d1e | 1152 | fields: &[hir::Field], |
b039eaaf SL |
1153 | wth: &Option<P<hir::Expr>>) |
1154 | -> io::Result<()> { | |
476ff2be | 1155 | self.print_qpath(qpath, true)?; |
041b39d2 | 1156 | self.s.word("{")?; |
54a0048b SL |
1157 | self.commasep_cmnt(Consistent, |
1158 | &fields[..], | |
1159 | |s, field| { | |
1160 | s.ibox(indent_unit)?; | |
c30ab7b3 SL |
1161 | if !field.is_shorthand { |
1162 | s.print_name(field.name.node)?; | |
1163 | s.word_space(":")?; | |
1164 | } | |
54a0048b SL |
1165 | s.print_expr(&field.expr)?; |
1166 | s.end() | |
1167 | }, | |
1168 | |f| f.span)?; | |
b039eaaf SL |
1169 | match *wth { |
1170 | Some(ref expr) => { | |
54a0048b | 1171 | self.ibox(indent_unit)?; |
b039eaaf | 1172 | if !fields.is_empty() { |
041b39d2 XL |
1173 | self.s.word(",")?; |
1174 | self.s.space()?; | |
e9174d1e | 1175 | } |
041b39d2 | 1176 | self.s.word("..")?; |
54a0048b SL |
1177 | self.print_expr(&expr)?; |
1178 | self.end()?; | |
e9174d1e | 1179 | } |
b039eaaf | 1180 | _ => if !fields.is_empty() { |
041b39d2 | 1181 | self.s.word(",")? |
b039eaaf | 1182 | }, |
e9174d1e | 1183 | } |
041b39d2 | 1184 | self.s.word("}")?; |
e9174d1e SL |
1185 | Ok(()) |
1186 | } | |
1187 | ||
476ff2be | 1188 | fn print_expr_tup(&mut self, exprs: &[hir::Expr]) -> io::Result<()> { |
54a0048b | 1189 | self.popen()?; |
476ff2be | 1190 | self.commasep_exprs(Inconsistent, exprs)?; |
e9174d1e | 1191 | if exprs.len() == 1 { |
041b39d2 | 1192 | self.s.word(",")?; |
e9174d1e SL |
1193 | } |
1194 | self.pclose() | |
1195 | } | |
1196 | ||
476ff2be | 1197 | fn print_expr_call(&mut self, func: &hir::Expr, args: &[hir::Expr]) -> io::Result<()> { |
ea8adc8c XL |
1198 | let prec = |
1199 | match func.node { | |
1200 | hir::ExprField(..) | | |
1201 | hir::ExprTupField(..) => parser::PREC_FORCE_PAREN, | |
1202 | _ => parser::PREC_POSTFIX, | |
1203 | }; | |
1204 | ||
1205 | self.print_expr_maybe_paren(func, prec)?; | |
e9174d1e SL |
1206 | self.print_call_post(args) |
1207 | } | |
1208 | ||
1209 | fn print_expr_method_call(&mut self, | |
041b39d2 | 1210 | segment: &hir::PathSegment, |
476ff2be | 1211 | args: &[hir::Expr]) |
b039eaaf | 1212 | -> io::Result<()> { |
e9174d1e | 1213 | let base_args = &args[1..]; |
ea8adc8c | 1214 | self.print_expr_maybe_paren(&args[0], parser::PREC_POSTFIX)?; |
041b39d2 XL |
1215 | self.s.word(".")?; |
1216 | self.print_name(segment.name)?; | |
ea8adc8c XL |
1217 | |
1218 | segment.with_parameters(|parameters| { | |
1219 | if !parameters.lifetimes.is_empty() || | |
1220 | !parameters.types.is_empty() || | |
1221 | !parameters.bindings.is_empty() | |
1222 | { | |
1223 | self.print_path_parameters(¶meters, segment.infer_types, true) | |
1224 | } else { | |
1225 | Ok(()) | |
1226 | } | |
1227 | })?; | |
e9174d1e SL |
1228 | self.print_call_post(base_args) |
1229 | } | |
1230 | ||
1231 | fn print_expr_binary(&mut self, | |
1232 | op: hir::BinOp, | |
1233 | lhs: &hir::Expr, | |
b039eaaf SL |
1234 | rhs: &hir::Expr) |
1235 | -> io::Result<()> { | |
ea8adc8c XL |
1236 | let assoc_op = bin_op_to_assoc_op(op.node); |
1237 | let prec = assoc_op.precedence() as i8; | |
1238 | let fixity = assoc_op.fixity(); | |
1239 | ||
1240 | let (left_prec, right_prec) = match fixity { | |
1241 | Fixity::Left => (prec, prec + 1), | |
1242 | Fixity::Right => (prec + 1, prec), | |
1243 | Fixity::None => (prec + 1, prec + 1), | |
1244 | }; | |
1245 | ||
1246 | self.print_expr_maybe_paren(lhs, left_prec)?; | |
041b39d2 | 1247 | self.s.space()?; |
54a0048b | 1248 | self.word_space(op.node.as_str())?; |
ea8adc8c | 1249 | self.print_expr_maybe_paren(rhs, right_prec) |
e9174d1e SL |
1250 | } |
1251 | ||
b039eaaf | 1252 | fn print_expr_unary(&mut self, op: hir::UnOp, expr: &hir::Expr) -> io::Result<()> { |
041b39d2 | 1253 | self.s.word(op.as_str())?; |
ea8adc8c | 1254 | self.print_expr_maybe_paren(expr, parser::PREC_PREFIX) |
e9174d1e SL |
1255 | } |
1256 | ||
1257 | fn print_expr_addr_of(&mut self, | |
1258 | mutability: hir::Mutability, | |
b039eaaf SL |
1259 | expr: &hir::Expr) |
1260 | -> io::Result<()> { | |
041b39d2 | 1261 | self.s.word("&")?; |
54a0048b | 1262 | self.print_mutability(mutability)?; |
ea8adc8c | 1263 | self.print_expr_maybe_paren(expr, parser::PREC_PREFIX) |
e9174d1e SL |
1264 | } |
1265 | ||
1266 | pub fn print_expr(&mut self, expr: &hir::Expr) -> io::Result<()> { | |
ea8adc8c | 1267 | self.maybe_print_comment(expr.span.lo())?; |
32a655c1 | 1268 | self.print_outer_attributes(&expr.attrs)?; |
54a0048b SL |
1269 | self.ibox(indent_unit)?; |
1270 | self.ann.pre(self, NodeExpr(expr))?; | |
e9174d1e | 1271 | match expr.node { |
b039eaaf | 1272 | hir::ExprBox(ref expr) => { |
54a0048b | 1273 | self.word_space("box")?; |
ea8adc8c | 1274 | self.print_expr_maybe_paren(expr, parser::PREC_PREFIX)?; |
e9174d1e | 1275 | } |
c30ab7b3 | 1276 | hir::ExprArray(ref exprs) => { |
476ff2be | 1277 | self.print_expr_vec(exprs)?; |
e9174d1e | 1278 | } |
32a655c1 SL |
1279 | hir::ExprRepeat(ref element, count) => { |
1280 | self.print_expr_repeat(&element, count)?; | |
e9174d1e | 1281 | } |
476ff2be SL |
1282 | hir::ExprStruct(ref qpath, ref fields, ref wth) => { |
1283 | self.print_expr_struct(qpath, &fields[..], wth)?; | |
e9174d1e SL |
1284 | } |
1285 | hir::ExprTup(ref exprs) => { | |
476ff2be | 1286 | self.print_expr_tup(exprs)?; |
e9174d1e SL |
1287 | } |
1288 | hir::ExprCall(ref func, ref args) => { | |
476ff2be | 1289 | self.print_expr_call(&func, args)?; |
e9174d1e | 1290 | } |
041b39d2 XL |
1291 | hir::ExprMethodCall(ref segment, _, ref args) => { |
1292 | self.print_expr_method_call(segment, args)?; | |
e9174d1e SL |
1293 | } |
1294 | hir::ExprBinary(op, ref lhs, ref rhs) => { | |
54a0048b | 1295 | self.print_expr_binary(op, &lhs, &rhs)?; |
e9174d1e SL |
1296 | } |
1297 | hir::ExprUnary(op, ref expr) => { | |
54a0048b | 1298 | self.print_expr_unary(op, &expr)?; |
e9174d1e SL |
1299 | } |
1300 | hir::ExprAddrOf(m, ref expr) => { | |
54a0048b | 1301 | self.print_expr_addr_of(m, &expr)?; |
e9174d1e SL |
1302 | } |
1303 | hir::ExprLit(ref lit) => { | |
54a0048b | 1304 | self.print_literal(&lit)?; |
e9174d1e SL |
1305 | } |
1306 | hir::ExprCast(ref expr, ref ty) => { | |
ea8adc8c XL |
1307 | let prec = AssocOp::As.precedence() as i8; |
1308 | self.print_expr_maybe_paren(&expr, prec)?; | |
041b39d2 | 1309 | self.s.space()?; |
54a0048b SL |
1310 | self.word_space("as")?; |
1311 | self.print_type(&ty)?; | |
e9174d1e | 1312 | } |
9cc50fc6 | 1313 | hir::ExprType(ref expr, ref ty) => { |
ea8adc8c XL |
1314 | let prec = AssocOp::Colon.precedence() as i8; |
1315 | self.print_expr_maybe_paren(&expr, prec)?; | |
54a0048b SL |
1316 | self.word_space(":")?; |
1317 | self.print_type(&ty)?; | |
9cc50fc6 | 1318 | } |
e9174d1e | 1319 | hir::ExprIf(ref test, ref blk, ref elseopt) => { |
54a0048b | 1320 | self.print_if(&test, &blk, elseopt.as_ref().map(|e| &**e))?; |
e9174d1e | 1321 | } |
3157f602 XL |
1322 | hir::ExprWhile(ref test, ref blk, opt_sp_name) => { |
1323 | if let Some(sp_name) = opt_sp_name { | |
1324 | self.print_name(sp_name.node)?; | |
54a0048b | 1325 | self.word_space(":")?; |
e9174d1e | 1326 | } |
54a0048b | 1327 | self.head("while")?; |
ea8adc8c | 1328 | self.print_expr_as_cond(&test)?; |
041b39d2 | 1329 | self.s.space()?; |
54a0048b | 1330 | self.print_block(&blk)?; |
e9174d1e | 1331 | } |
476ff2be | 1332 | hir::ExprLoop(ref blk, opt_sp_name, _) => { |
3157f602 XL |
1333 | if let Some(sp_name) = opt_sp_name { |
1334 | self.print_name(sp_name.node)?; | |
54a0048b | 1335 | self.word_space(":")?; |
e9174d1e | 1336 | } |
54a0048b | 1337 | self.head("loop")?; |
041b39d2 | 1338 | self.s.space()?; |
54a0048b | 1339 | self.print_block(&blk)?; |
e9174d1e SL |
1340 | } |
1341 | hir::ExprMatch(ref expr, ref arms, _) => { | |
54a0048b SL |
1342 | self.cbox(indent_unit)?; |
1343 | self.ibox(4)?; | |
1344 | self.word_nbsp("match")?; | |
ea8adc8c | 1345 | self.print_expr_as_cond(&expr)?; |
041b39d2 | 1346 | self.s.space()?; |
54a0048b | 1347 | self.bopen()?; |
e9174d1e | 1348 | for arm in arms { |
54a0048b | 1349 | self.print_arm(arm)?; |
e9174d1e | 1350 | } |
54a0048b | 1351 | self.bclose_(expr.span, indent_unit)?; |
e9174d1e | 1352 | } |
ea8adc8c | 1353 | hir::ExprClosure(capture_clause, ref decl, body, _fn_decl_span, _gen) => { |
54a0048b | 1354 | self.print_capture_clause(capture_clause)?; |
e9174d1e | 1355 | |
32a655c1 | 1356 | self.print_closure_args(&decl, body)?; |
041b39d2 | 1357 | self.s.space()?; |
e9174d1e | 1358 | |
476ff2be | 1359 | // this is a bare expression |
32a655c1 | 1360 | self.ann.nested(self, Nested::Body(body))?; |
476ff2be | 1361 | self.end()?; // need to close a box |
e9174d1e | 1362 | |
e9174d1e SL |
1363 | // a box will be closed by print_expr, but we didn't want an overall |
1364 | // wrapper so we closed the corresponding opening. so create an | |
1365 | // empty box to satisfy the close. | |
54a0048b | 1366 | self.ibox(0)?; |
e9174d1e SL |
1367 | } |
1368 | hir::ExprBlock(ref blk) => { | |
1369 | // containing cbox, will be closed by print-block at } | |
54a0048b | 1370 | self.cbox(indent_unit)?; |
e9174d1e | 1371 | // head-box, will be closed by print-block after { |
54a0048b SL |
1372 | self.ibox(0)?; |
1373 | self.print_block(&blk)?; | |
e9174d1e SL |
1374 | } |
1375 | hir::ExprAssign(ref lhs, ref rhs) => { | |
ea8adc8c XL |
1376 | let prec = AssocOp::Assign.precedence() as i8; |
1377 | self.print_expr_maybe_paren(&lhs, prec + 1)?; | |
041b39d2 | 1378 | self.s.space()?; |
54a0048b | 1379 | self.word_space("=")?; |
ea8adc8c | 1380 | self.print_expr_maybe_paren(&rhs, prec)?; |
e9174d1e SL |
1381 | } |
1382 | hir::ExprAssignOp(op, ref lhs, ref rhs) => { | |
ea8adc8c XL |
1383 | let prec = AssocOp::Assign.precedence() as i8; |
1384 | self.print_expr_maybe_paren(&lhs, prec + 1)?; | |
041b39d2 XL |
1385 | self.s.space()?; |
1386 | self.s.word(op.node.as_str())?; | |
54a0048b | 1387 | self.word_space("=")?; |
ea8adc8c | 1388 | self.print_expr_maybe_paren(&rhs, prec)?; |
e9174d1e | 1389 | } |
b039eaaf | 1390 | hir::ExprField(ref expr, name) => { |
ea8adc8c | 1391 | self.print_expr_maybe_paren(expr, parser::PREC_POSTFIX)?; |
041b39d2 | 1392 | self.s.word(".")?; |
54a0048b | 1393 | self.print_name(name.node)?; |
e9174d1e SL |
1394 | } |
1395 | hir::ExprTupField(ref expr, id) => { | |
ea8adc8c | 1396 | self.print_expr_maybe_paren(&expr, parser::PREC_POSTFIX)?; |
041b39d2 | 1397 | self.s.word(".")?; |
54a0048b | 1398 | self.print_usize(id.node)?; |
e9174d1e SL |
1399 | } |
1400 | hir::ExprIndex(ref expr, ref index) => { | |
ea8adc8c | 1401 | self.print_expr_maybe_paren(&expr, parser::PREC_POSTFIX)?; |
041b39d2 | 1402 | self.s.word("[")?; |
54a0048b | 1403 | self.print_expr(&index)?; |
041b39d2 | 1404 | self.s.word("]")?; |
e9174d1e | 1405 | } |
476ff2be SL |
1406 | hir::ExprPath(ref qpath) => { |
1407 | self.print_qpath(qpath, true)? | |
e9174d1e | 1408 | } |
8bb4bdeb | 1409 | hir::ExprBreak(label, ref opt_expr) => { |
041b39d2 XL |
1410 | self.s.word("break")?; |
1411 | self.s.space()?; | |
8bb4bdeb XL |
1412 | if let Some(label_ident) = label.ident { |
1413 | self.print_name(label_ident.node.name)?; | |
041b39d2 | 1414 | self.s.space()?; |
476ff2be SL |
1415 | } |
1416 | if let Some(ref expr) = *opt_expr { | |
ea8adc8c | 1417 | self.print_expr_maybe_paren(expr, parser::PREC_JUMP)?; |
041b39d2 | 1418 | self.s.space()?; |
e9174d1e SL |
1419 | } |
1420 | } | |
8bb4bdeb | 1421 | hir::ExprAgain(label) => { |
041b39d2 XL |
1422 | self.s.word("continue")?; |
1423 | self.s.space()?; | |
8bb4bdeb XL |
1424 | if let Some(label_ident) = label.ident { |
1425 | self.print_name(label_ident.node.name)?; | |
041b39d2 | 1426 | self.s.space()? |
e9174d1e SL |
1427 | } |
1428 | } | |
1429 | hir::ExprRet(ref result) => { | |
041b39d2 | 1430 | self.s.word("return")?; |
e9174d1e SL |
1431 | match *result { |
1432 | Some(ref expr) => { | |
041b39d2 | 1433 | self.s.word(" ")?; |
ea8adc8c | 1434 | self.print_expr_maybe_paren(&expr, parser::PREC_JUMP)?; |
e9174d1e | 1435 | } |
b039eaaf | 1436 | _ => (), |
e9174d1e SL |
1437 | } |
1438 | } | |
54a0048b | 1439 | hir::ExprInlineAsm(ref a, ref outputs, ref inputs) => { |
041b39d2 | 1440 | self.s.word("asm!")?; |
54a0048b | 1441 | self.popen()?; |
476ff2be | 1442 | self.print_string(&a.asm.as_str(), a.asm_str_style)?; |
54a0048b SL |
1443 | self.word_space(":")?; |
1444 | ||
1445 | let mut out_idx = 0; | |
1446 | self.commasep(Inconsistent, &a.outputs, |s, out| { | |
476ff2be SL |
1447 | let constraint = out.constraint.as_str(); |
1448 | let mut ch = constraint.chars(); | |
54a0048b SL |
1449 | match ch.next() { |
1450 | Some('=') if out.is_rw => { | |
1451 | s.print_string(&format!("+{}", ch.as_str()), | |
1452 | ast::StrStyle::Cooked)? | |
92a42be0 | 1453 | } |
476ff2be | 1454 | _ => s.print_string(&constraint, ast::StrStyle::Cooked)?, |
92a42be0 | 1455 | } |
54a0048b SL |
1456 | s.popen()?; |
1457 | s.print_expr(&outputs[out_idx])?; | |
1458 | s.pclose()?; | |
1459 | out_idx += 1; | |
92a42be0 | 1460 | Ok(()) |
54a0048b | 1461 | })?; |
041b39d2 | 1462 | self.s.space()?; |
54a0048b SL |
1463 | self.word_space(":")?; |
1464 | ||
1465 | let mut in_idx = 0; | |
1466 | self.commasep(Inconsistent, &a.inputs, |s, co| { | |
476ff2be | 1467 | s.print_string(&co.as_str(), ast::StrStyle::Cooked)?; |
54a0048b SL |
1468 | s.popen()?; |
1469 | s.print_expr(&inputs[in_idx])?; | |
1470 | s.pclose()?; | |
1471 | in_idx += 1; | |
92a42be0 | 1472 | Ok(()) |
54a0048b | 1473 | })?; |
041b39d2 | 1474 | self.s.space()?; |
54a0048b | 1475 | self.word_space(":")?; |
e9174d1e | 1476 | |
54a0048b | 1477 | self.commasep(Inconsistent, &a.clobbers, |s, co| { |
476ff2be | 1478 | s.print_string(&co.as_str(), ast::StrStyle::Cooked)?; |
92a42be0 | 1479 | Ok(()) |
54a0048b | 1480 | })?; |
e9174d1e | 1481 | |
92a42be0 | 1482 | let mut options = vec![]; |
e9174d1e SL |
1483 | if a.volatile { |
1484 | options.push("volatile"); | |
1485 | } | |
1486 | if a.alignstack { | |
1487 | options.push("alignstack"); | |
1488 | } | |
b039eaaf | 1489 | if a.dialect == ast::AsmDialect::Intel { |
e9174d1e SL |
1490 | options.push("intel"); |
1491 | } | |
1492 | ||
1493 | if !options.is_empty() { | |
041b39d2 | 1494 | self.s.space()?; |
54a0048b SL |
1495 | self.word_space(":")?; |
1496 | self.commasep(Inconsistent, &options, |s, &co| { | |
1497 | s.print_string(co, ast::StrStyle::Cooked)?; | |
92a42be0 | 1498 | Ok(()) |
54a0048b | 1499 | })?; |
e9174d1e SL |
1500 | } |
1501 | ||
54a0048b | 1502 | self.pclose()?; |
e9174d1e | 1503 | } |
ea8adc8c XL |
1504 | hir::ExprYield(ref expr) => { |
1505 | self.word_space("yield")?; | |
1506 | self.print_expr_maybe_paren(&expr, parser::PREC_JUMP)?; | |
1507 | } | |
e9174d1e | 1508 | } |
54a0048b | 1509 | self.ann.post(self, NodeExpr(expr))?; |
e9174d1e SL |
1510 | self.end() |
1511 | } | |
1512 | ||
1513 | pub fn print_local_decl(&mut self, loc: &hir::Local) -> io::Result<()> { | |
54a0048b | 1514 | self.print_pat(&loc.pat)?; |
e9174d1e | 1515 | if let Some(ref ty) = loc.ty { |
54a0048b SL |
1516 | self.word_space(":")?; |
1517 | self.print_type(&ty)?; | |
e9174d1e SL |
1518 | } |
1519 | Ok(()) | |
1520 | } | |
1521 | ||
1522 | pub fn print_decl(&mut self, decl: &hir::Decl) -> io::Result<()> { | |
ea8adc8c | 1523 | self.maybe_print_comment(decl.span.lo())?; |
e9174d1e SL |
1524 | match decl.node { |
1525 | hir::DeclLocal(ref loc) => { | |
54a0048b SL |
1526 | self.space_if_not_bol()?; |
1527 | self.ibox(indent_unit)?; | |
1528 | self.word_nbsp("let")?; | |
e9174d1e | 1529 | |
54a0048b SL |
1530 | self.ibox(indent_unit)?; |
1531 | self.print_local_decl(&loc)?; | |
1532 | self.end()?; | |
e9174d1e | 1533 | if let Some(ref init) = loc.init { |
54a0048b SL |
1534 | self.nbsp()?; |
1535 | self.word_space("=")?; | |
1536 | self.print_expr(&init)?; | |
e9174d1e SL |
1537 | } |
1538 | self.end() | |
1539 | } | |
32a655c1 SL |
1540 | hir::DeclItem(item) => { |
1541 | self.ann.nested(self, Nested::Item(item)) | |
92a42be0 | 1542 | } |
e9174d1e SL |
1543 | } |
1544 | } | |
1545 | ||
e9174d1e | 1546 | pub fn print_usize(&mut self, i: usize) -> io::Result<()> { |
041b39d2 | 1547 | self.s.word(&i.to_string()) |
e9174d1e SL |
1548 | } |
1549 | ||
1550 | pub fn print_name(&mut self, name: ast::Name) -> io::Result<()> { | |
041b39d2 | 1551 | self.s.word(&name.as_str())?; |
e9174d1e SL |
1552 | self.ann.post(self, NodeName(&name)) |
1553 | } | |
1554 | ||
b039eaaf | 1555 | pub fn print_for_decl(&mut self, loc: &hir::Local, coll: &hir::Expr) -> io::Result<()> { |
54a0048b | 1556 | self.print_local_decl(loc)?; |
041b39d2 | 1557 | self.s.space()?; |
54a0048b | 1558 | self.word_space("in")?; |
e9174d1e SL |
1559 | self.print_expr(coll) |
1560 | } | |
1561 | ||
32a655c1 SL |
1562 | pub fn print_path(&mut self, |
1563 | path: &hir::Path, | |
1564 | colons_before_params: bool) | |
1565 | -> io::Result<()> { | |
ea8adc8c | 1566 | self.maybe_print_comment(path.span.lo())?; |
e9174d1e | 1567 | |
32a655c1 SL |
1568 | for (i, segment) in path.segments.iter().enumerate() { |
1569 | if i > 0 { | |
041b39d2 | 1570 | self.s.word("::")? |
e9174d1e | 1571 | } |
041b39d2 XL |
1572 | if segment.name != keywords::CrateRoot.name() && |
1573 | segment.name != keywords::DollarCrate.name() { | |
ea8adc8c XL |
1574 | self.print_name(segment.name)?; |
1575 | segment.with_parameters(|parameters| { | |
1576 | self.print_path_parameters(parameters, | |
1577 | segment.infer_types, | |
1578 | colons_before_params) | |
1579 | })?; | |
32a655c1 | 1580 | } |
e9174d1e SL |
1581 | } |
1582 | ||
1583 | Ok(()) | |
1584 | } | |
1585 | ||
32a655c1 SL |
1586 | pub fn print_qpath(&mut self, |
1587 | qpath: &hir::QPath, | |
1588 | colons_before_params: bool) | |
1589 | -> io::Result<()> { | |
476ff2be SL |
1590 | match *qpath { |
1591 | hir::QPath::Resolved(None, ref path) => { | |
1592 | self.print_path(path, colons_before_params) | |
1593 | } | |
1594 | hir::QPath::Resolved(Some(ref qself), ref path) => { | |
041b39d2 | 1595 | self.s.word("<")?; |
476ff2be | 1596 | self.print_type(qself)?; |
041b39d2 | 1597 | self.s.space()?; |
476ff2be SL |
1598 | self.word_space("as")?; |
1599 | ||
32a655c1 SL |
1600 | for (i, segment) in path.segments[..path.segments.len() - 1].iter().enumerate() { |
1601 | if i > 0 { | |
041b39d2 | 1602 | self.s.word("::")? |
476ff2be | 1603 | } |
041b39d2 XL |
1604 | if segment.name != keywords::CrateRoot.name() && |
1605 | segment.name != keywords::DollarCrate.name() { | |
32a655c1 | 1606 | self.print_name(segment.name)?; |
ea8adc8c XL |
1607 | segment.with_parameters(|parameters| { |
1608 | self.print_path_parameters(parameters, | |
1609 | segment.infer_types, | |
1610 | colons_before_params) | |
1611 | })?; | |
32a655c1 | 1612 | } |
476ff2be SL |
1613 | } |
1614 | ||
041b39d2 XL |
1615 | self.s.word(">")?; |
1616 | self.s.word("::")?; | |
476ff2be SL |
1617 | let item_segment = path.segments.last().unwrap(); |
1618 | self.print_name(item_segment.name)?; | |
ea8adc8c XL |
1619 | item_segment.with_parameters(|parameters| { |
1620 | self.print_path_parameters(parameters, | |
1621 | item_segment.infer_types, | |
1622 | colons_before_params) | |
1623 | }) | |
476ff2be SL |
1624 | } |
1625 | hir::QPath::TypeRelative(ref qself, ref item_segment) => { | |
041b39d2 | 1626 | self.s.word("<")?; |
476ff2be | 1627 | self.print_type(qself)?; |
041b39d2 XL |
1628 | self.s.word(">")?; |
1629 | self.s.word("::")?; | |
476ff2be | 1630 | self.print_name(item_segment.name)?; |
ea8adc8c XL |
1631 | item_segment.with_parameters(|parameters| { |
1632 | self.print_path_parameters(parameters, | |
1633 | item_segment.infer_types, | |
1634 | colons_before_params) | |
1635 | }) | |
476ff2be | 1636 | } |
e9174d1e | 1637 | } |
e9174d1e SL |
1638 | } |
1639 | ||
1640 | fn print_path_parameters(&mut self, | |
1641 | parameters: &hir::PathParameters, | |
ea8adc8c | 1642 | infer_types: bool, |
e9174d1e | 1643 | colons_before_params: bool) |
b039eaaf | 1644 | -> io::Result<()> { |
3b2f2976 XL |
1645 | if parameters.parenthesized { |
1646 | self.s.word("(")?; | |
1647 | self.commasep(Inconsistent, parameters.inputs(), |s, ty| s.print_type(&ty))?; | |
1648 | self.s.word(")")?; | |
e9174d1e | 1649 | |
3b2f2976 XL |
1650 | self.space_if_not_bol()?; |
1651 | self.word_space("->")?; | |
1652 | self.print_type(¶meters.bindings[0].ty)?; | |
1653 | } else { | |
1654 | let start = if colons_before_params { "::<" } else { "<" }; | |
1655 | let empty = Cell::new(true); | |
1656 | let start_or_comma = |this: &mut Self| { | |
1657 | if empty.get() { | |
1658 | empty.set(false); | |
1659 | this.s.word(start) | |
1660 | } else { | |
1661 | this.word_space(",") | |
e9174d1e | 1662 | } |
3b2f2976 | 1663 | }; |
e9174d1e | 1664 | |
3b2f2976 XL |
1665 | if !parameters.lifetimes.iter().all(|lt| lt.is_elided()) { |
1666 | for lifetime in ¶meters.lifetimes { | |
32a655c1 | 1667 | start_or_comma(self)?; |
3b2f2976 | 1668 | self.print_lifetime(lifetime)?; |
476ff2be | 1669 | } |
3b2f2976 | 1670 | } |
476ff2be | 1671 | |
3b2f2976 XL |
1672 | if !parameters.types.is_empty() { |
1673 | start_or_comma(self)?; | |
1674 | self.commasep(Inconsistent, ¶meters.types, |s, ty| s.print_type(&ty))?; | |
1675 | } | |
e9174d1e | 1676 | |
3b2f2976 XL |
1677 | // FIXME(eddyb) This would leak into error messages, e.g.: |
1678 | // "non-exhaustive patterns: `Some::<..>(_)` not covered". | |
ea8adc8c | 1679 | if infer_types && false { |
3b2f2976 XL |
1680 | start_or_comma(self)?; |
1681 | self.s.word("..")?; | |
e9174d1e SL |
1682 | } |
1683 | ||
3b2f2976 XL |
1684 | for binding in parameters.bindings.iter() { |
1685 | start_or_comma(self)?; | |
1686 | self.print_name(binding.name)?; | |
1687 | self.s.space()?; | |
1688 | self.word_space("=")?; | |
1689 | self.print_type(&binding.ty)?; | |
1690 | } | |
e9174d1e | 1691 | |
3b2f2976 XL |
1692 | if !empty.get() { |
1693 | self.s.word(">")? | |
e9174d1e SL |
1694 | } |
1695 | } | |
1696 | ||
1697 | Ok(()) | |
1698 | } | |
1699 | ||
1700 | pub fn print_pat(&mut self, pat: &hir::Pat) -> io::Result<()> { | |
ea8adc8c | 1701 | self.maybe_print_comment(pat.span.lo())?; |
54a0048b | 1702 | self.ann.pre(self, NodePat(pat))?; |
92a42be0 SL |
1703 | // Pat isn't normalized, but the beauty of it |
1704 | // is that it doesn't matter | |
e9174d1e | 1705 | match pat.node { |
041b39d2 | 1706 | PatKind::Wild => self.s.word("_")?, |
476ff2be | 1707 | PatKind::Binding(binding_mode, _, ref path1, ref sub) => { |
e9174d1e | 1708 | match binding_mode { |
3b2f2976 XL |
1709 | hir::BindingAnnotation::Ref => { |
1710 | self.word_nbsp("ref")?; | |
1711 | self.print_mutability(hir::MutImmutable)?; | |
1712 | } | |
1713 | hir::BindingAnnotation::RefMut => { | |
54a0048b | 1714 | self.word_nbsp("ref")?; |
3b2f2976 | 1715 | self.print_mutability(hir::MutMutable)?; |
e9174d1e | 1716 | } |
3b2f2976 XL |
1717 | hir::BindingAnnotation::Unannotated => {} |
1718 | hir::BindingAnnotation::Mutable => { | |
54a0048b | 1719 | self.word_nbsp("mut")?; |
e9174d1e SL |
1720 | } |
1721 | } | |
a7813a04 | 1722 | self.print_name(path1.node)?; |
3157f602 | 1723 | if let Some(ref p) = *sub { |
041b39d2 | 1724 | self.s.word("@")?; |
3157f602 | 1725 | self.print_pat(&p)?; |
e9174d1e SL |
1726 | } |
1727 | } | |
476ff2be SL |
1728 | PatKind::TupleStruct(ref qpath, ref elts, ddpos) => { |
1729 | self.print_qpath(qpath, true)?; | |
3157f602 XL |
1730 | self.popen()?; |
1731 | if let Some(ddpos) = ddpos { | |
1732 | self.commasep(Inconsistent, &elts[..ddpos], |s, p| s.print_pat(&p))?; | |
1733 | if ddpos != 0 { | |
1734 | self.word_space(",")?; | |
e9174d1e | 1735 | } |
041b39d2 | 1736 | self.s.word("..")?; |
3157f602 | 1737 | if ddpos != elts.len() { |
041b39d2 | 1738 | self.s.word(",")?; |
3157f602 XL |
1739 | self.commasep(Inconsistent, &elts[ddpos..], |s, p| s.print_pat(&p))?; |
1740 | } | |
1741 | } else { | |
9e0c209e | 1742 | self.commasep(Inconsistent, &elts[..], |s, p| s.print_pat(&p))?; |
e9174d1e | 1743 | } |
9e0c209e | 1744 | self.pclose()?; |
e9174d1e | 1745 | } |
476ff2be SL |
1746 | PatKind::Path(ref qpath) => { |
1747 | self.print_qpath(qpath, true)?; | |
e9174d1e | 1748 | } |
476ff2be SL |
1749 | PatKind::Struct(ref qpath, ref fields, etc) => { |
1750 | self.print_qpath(qpath, true)?; | |
54a0048b SL |
1751 | self.nbsp()?; |
1752 | self.word_space("{")?; | |
1753 | self.commasep_cmnt(Consistent, | |
1754 | &fields[..], | |
1755 | |s, f| { | |
1756 | s.cbox(indent_unit)?; | |
1757 | if !f.node.is_shorthand { | |
1758 | s.print_name(f.node.name)?; | |
1759 | s.word_nbsp(":")?; | |
1760 | } | |
1761 | s.print_pat(&f.node.pat)?; | |
1762 | s.end() | |
1763 | }, | |
1764 | |f| f.node.pat.span)?; | |
e9174d1e | 1765 | if etc { |
b039eaaf | 1766 | if !fields.is_empty() { |
54a0048b | 1767 | self.word_space(",")?; |
b039eaaf | 1768 | } |
041b39d2 | 1769 | self.s.word("..")?; |
e9174d1e | 1770 | } |
041b39d2 XL |
1771 | self.s.space()?; |
1772 | self.s.word("}")?; | |
e9174d1e | 1773 | } |
3157f602 | 1774 | PatKind::Tuple(ref elts, ddpos) => { |
54a0048b | 1775 | self.popen()?; |
3157f602 XL |
1776 | if let Some(ddpos) = ddpos { |
1777 | self.commasep(Inconsistent, &elts[..ddpos], |s, p| s.print_pat(&p))?; | |
1778 | if ddpos != 0 { | |
1779 | self.word_space(",")?; | |
1780 | } | |
041b39d2 | 1781 | self.s.word("..")?; |
3157f602 | 1782 | if ddpos != elts.len() { |
041b39d2 | 1783 | self.s.word(",")?; |
3157f602 XL |
1784 | self.commasep(Inconsistent, &elts[ddpos..], |s, p| s.print_pat(&p))?; |
1785 | } | |
1786 | } else { | |
1787 | self.commasep(Inconsistent, &elts[..], |s, p| s.print_pat(&p))?; | |
1788 | if elts.len() == 1 { | |
041b39d2 | 1789 | self.s.word(",")?; |
3157f602 | 1790 | } |
e9174d1e | 1791 | } |
54a0048b | 1792 | self.pclose()?; |
e9174d1e | 1793 | } |
7453a54e | 1794 | PatKind::Box(ref inner) => { |
041b39d2 | 1795 | self.s.word("box ")?; |
54a0048b | 1796 | self.print_pat(&inner)?; |
e9174d1e | 1797 | } |
7453a54e | 1798 | PatKind::Ref(ref inner, mutbl) => { |
041b39d2 | 1799 | self.s.word("&")?; |
e9174d1e | 1800 | if mutbl == hir::MutMutable { |
041b39d2 | 1801 | self.s.word("mut ")?; |
e9174d1e | 1802 | } |
54a0048b | 1803 | self.print_pat(&inner)?; |
e9174d1e | 1804 | } |
54a0048b | 1805 | PatKind::Lit(ref e) => self.print_expr(&e)?, |
32a655c1 | 1806 | PatKind::Range(ref begin, ref end, ref end_kind) => { |
54a0048b | 1807 | self.print_expr(&begin)?; |
041b39d2 | 1808 | self.s.space()?; |
32a655c1 | 1809 | match *end_kind { |
041b39d2 XL |
1810 | RangeEnd::Included => self.s.word("...")?, |
1811 | RangeEnd::Excluded => self.s.word("..")?, | |
32a655c1 | 1812 | } |
54a0048b | 1813 | self.print_expr(&end)?; |
e9174d1e | 1814 | } |
c30ab7b3 | 1815 | PatKind::Slice(ref before, ref slice, ref after) => { |
041b39d2 | 1816 | self.s.word("[")?; |
54a0048b | 1817 | self.commasep(Inconsistent, &before[..], |s, p| s.print_pat(&p))?; |
e9174d1e | 1818 | if let Some(ref p) = *slice { |
b039eaaf | 1819 | if !before.is_empty() { |
54a0048b | 1820 | self.word_space(",")?; |
b039eaaf | 1821 | } |
7453a54e | 1822 | if p.node != PatKind::Wild { |
54a0048b | 1823 | self.print_pat(&p)?; |
e9174d1e | 1824 | } |
041b39d2 | 1825 | self.s.word("..")?; |
b039eaaf | 1826 | if !after.is_empty() { |
54a0048b | 1827 | self.word_space(",")?; |
b039eaaf | 1828 | } |
e9174d1e | 1829 | } |
54a0048b | 1830 | self.commasep(Inconsistent, &after[..], |s, p| s.print_pat(&p))?; |
041b39d2 | 1831 | self.s.word("]")?; |
e9174d1e SL |
1832 | } |
1833 | } | |
1834 | self.ann.post(self, NodePat(pat)) | |
1835 | } | |
1836 | ||
1837 | fn print_arm(&mut self, arm: &hir::Arm) -> io::Result<()> { | |
1838 | // I have no idea why this check is necessary, but here it | |
1839 | // is :( | |
1840 | if arm.attrs.is_empty() { | |
041b39d2 | 1841 | self.s.space()?; |
e9174d1e | 1842 | } |
54a0048b SL |
1843 | self.cbox(indent_unit)?; |
1844 | self.ibox(0)?; | |
1845 | self.print_outer_attributes(&arm.attrs)?; | |
e9174d1e SL |
1846 | let mut first = true; |
1847 | for p in &arm.pats { | |
1848 | if first { | |
1849 | first = false; | |
1850 | } else { | |
041b39d2 | 1851 | self.s.space()?; |
54a0048b | 1852 | self.word_space("|")?; |
e9174d1e | 1853 | } |
54a0048b | 1854 | self.print_pat(&p)?; |
e9174d1e | 1855 | } |
041b39d2 | 1856 | self.s.space()?; |
e9174d1e | 1857 | if let Some(ref e) = arm.guard { |
54a0048b SL |
1858 | self.word_space("if")?; |
1859 | self.print_expr(&e)?; | |
041b39d2 | 1860 | self.s.space()?; |
e9174d1e | 1861 | } |
54a0048b | 1862 | self.word_space("=>")?; |
e9174d1e SL |
1863 | |
1864 | match arm.body.node { | |
1865 | hir::ExprBlock(ref blk) => { | |
1866 | // the block will close the pattern's ibox | |
54a0048b | 1867 | self.print_block_unclosed_indent(&blk, indent_unit)?; |
e9174d1e SL |
1868 | |
1869 | // If it is a user-provided unsafe block, print a comma after it | |
1870 | if let hir::UnsafeBlock(hir::UserProvided) = blk.rules { | |
041b39d2 | 1871 | self.s.word(",")?; |
e9174d1e SL |
1872 | } |
1873 | } | |
1874 | _ => { | |
54a0048b SL |
1875 | self.end()?; // close the ibox for the pattern |
1876 | self.print_expr(&arm.body)?; | |
041b39d2 | 1877 | self.s.word(",")?; |
e9174d1e SL |
1878 | } |
1879 | } | |
1880 | self.end() // close enclosing cbox | |
1881 | } | |
1882 | ||
e9174d1e SL |
1883 | pub fn print_fn(&mut self, |
1884 | decl: &hir::FnDecl, | |
1885 | unsafety: hir::Unsafety, | |
1886 | constness: hir::Constness, | |
7453a54e | 1887 | abi: Abi, |
b039eaaf | 1888 | name: Option<ast::Name>, |
e9174d1e | 1889 | generics: &hir::Generics, |
32a655c1 SL |
1890 | vis: &hir::Visibility, |
1891 | arg_names: &[Spanned<ast::Name>], | |
1892 | body_id: Option<hir::BodyId>) | |
b039eaaf | 1893 | -> io::Result<()> { |
54a0048b | 1894 | self.print_fn_header_info(unsafety, constness, abi, vis)?; |
e9174d1e SL |
1895 | |
1896 | if let Some(name) = name { | |
54a0048b SL |
1897 | self.nbsp()?; |
1898 | self.print_name(name)?; | |
e9174d1e | 1899 | } |
54a0048b | 1900 | self.print_generics(generics)?; |
e9174d1e | 1901 | |
54a0048b | 1902 | self.popen()?; |
32a655c1 SL |
1903 | let mut i = 0; |
1904 | // Make sure we aren't supplied *both* `arg_names` and `body_id`. | |
1905 | assert!(arg_names.is_empty() || body_id.is_none()); | |
1906 | self.commasep(Inconsistent, &decl.inputs, |s, ty| { | |
1907 | s.ibox(indent_unit)?; | |
1908 | if let Some(name) = arg_names.get(i) { | |
041b39d2 XL |
1909 | s.s.word(&name.node.as_str())?; |
1910 | s.s.word(":")?; | |
1911 | s.s.space()?; | |
32a655c1 SL |
1912 | } else if let Some(body_id) = body_id { |
1913 | s.ann.nested(s, Nested::BodyArgPat(body_id, i))?; | |
041b39d2 XL |
1914 | s.s.word(":")?; |
1915 | s.s.space()?; | |
32a655c1 SL |
1916 | } |
1917 | i += 1; | |
1918 | s.print_type(ty)?; | |
1919 | s.end() | |
1920 | })?; | |
e9174d1e | 1921 | if decl.variadic { |
041b39d2 | 1922 | self.s.word(", ...")?; |
e9174d1e | 1923 | } |
54a0048b | 1924 | self.pclose()?; |
e9174d1e | 1925 | |
32a655c1 SL |
1926 | self.print_fn_output(decl)?; |
1927 | self.print_where_clause(&generics.where_clause) | |
e9174d1e SL |
1928 | } |
1929 | ||
32a655c1 | 1930 | fn print_closure_args(&mut self, decl: &hir::FnDecl, body_id: hir::BodyId) -> io::Result<()> { |
041b39d2 | 1931 | self.s.word("|")?; |
32a655c1 SL |
1932 | let mut i = 0; |
1933 | self.commasep(Inconsistent, &decl.inputs, |s, ty| { | |
1934 | s.ibox(indent_unit)?; | |
1935 | ||
1936 | s.ann.nested(s, Nested::BodyArgPat(body_id, i))?; | |
1937 | i += 1; | |
1938 | ||
1939 | if ty.node != hir::TyInfer { | |
041b39d2 XL |
1940 | s.s.word(":")?; |
1941 | s.s.space()?; | |
32a655c1 SL |
1942 | s.print_type(ty)?; |
1943 | } | |
1944 | s.end() | |
1945 | })?; | |
041b39d2 | 1946 | self.s.word("|")?; |
e9174d1e SL |
1947 | |
1948 | if let hir::DefaultReturn(..) = decl.output { | |
1949 | return Ok(()); | |
1950 | } | |
1951 | ||
54a0048b SL |
1952 | self.space_if_not_bol()?; |
1953 | self.word_space("->")?; | |
e9174d1e SL |
1954 | match decl.output { |
1955 | hir::Return(ref ty) => { | |
54a0048b | 1956 | self.print_type(&ty)?; |
ea8adc8c | 1957 | self.maybe_print_comment(ty.span.lo()) |
e9174d1e SL |
1958 | } |
1959 | hir::DefaultReturn(..) => unreachable!(), | |
e9174d1e SL |
1960 | } |
1961 | } | |
1962 | ||
b039eaaf | 1963 | pub fn print_capture_clause(&mut self, capture_clause: hir::CaptureClause) -> io::Result<()> { |
e9174d1e SL |
1964 | match capture_clause { |
1965 | hir::CaptureByValue => self.word_space("move"), | |
1966 | hir::CaptureByRef => Ok(()), | |
1967 | } | |
1968 | } | |
1969 | ||
b039eaaf | 1970 | pub fn print_bounds(&mut self, prefix: &str, bounds: &[hir::TyParamBound]) -> io::Result<()> { |
e9174d1e | 1971 | if !bounds.is_empty() { |
041b39d2 | 1972 | self.s.word(prefix)?; |
e9174d1e SL |
1973 | let mut first = true; |
1974 | for bound in bounds { | |
54a0048b | 1975 | self.nbsp()?; |
e9174d1e SL |
1976 | if first { |
1977 | first = false; | |
1978 | } else { | |
54a0048b | 1979 | self.word_space("+")?; |
e9174d1e SL |
1980 | } |
1981 | ||
54a0048b | 1982 | match *bound { |
e9174d1e SL |
1983 | TraitTyParamBound(ref tref, TraitBoundModifier::None) => { |
1984 | self.print_poly_trait_ref(tref) | |
1985 | } | |
1986 | TraitTyParamBound(ref tref, TraitBoundModifier::Maybe) => { | |
041b39d2 | 1987 | self.s.word("?")?; |
e9174d1e SL |
1988 | self.print_poly_trait_ref(tref) |
1989 | } | |
1990 | RegionTyParamBound(ref lt) => { | |
1991 | self.print_lifetime(lt) | |
1992 | } | |
54a0048b | 1993 | }? |
e9174d1e SL |
1994 | } |
1995 | Ok(()) | |
1996 | } else { | |
1997 | Ok(()) | |
1998 | } | |
1999 | } | |
2000 | ||
b039eaaf | 2001 | pub fn print_lifetime(&mut self, lifetime: &hir::Lifetime) -> io::Result<()> { |
ea8adc8c | 2002 | self.print_name(lifetime.name.name()) |
e9174d1e SL |
2003 | } |
2004 | ||
b039eaaf | 2005 | pub fn print_lifetime_def(&mut self, lifetime: &hir::LifetimeDef) -> io::Result<()> { |
54a0048b | 2006 | self.print_lifetime(&lifetime.lifetime)?; |
e9174d1e SL |
2007 | let mut sep = ":"; |
2008 | for v in &lifetime.bounds { | |
041b39d2 | 2009 | self.s.word(sep)?; |
54a0048b | 2010 | self.print_lifetime(v)?; |
e9174d1e SL |
2011 | sep = "+"; |
2012 | } | |
2013 | Ok(()) | |
2014 | } | |
2015 | ||
b039eaaf | 2016 | pub fn print_generics(&mut self, generics: &hir::Generics) -> io::Result<()> { |
e9174d1e SL |
2017 | let total = generics.lifetimes.len() + generics.ty_params.len(); |
2018 | if total == 0 { | |
2019 | return Ok(()); | |
2020 | } | |
2021 | ||
041b39d2 | 2022 | self.s.word("<")?; |
e9174d1e SL |
2023 | |
2024 | let mut ints = Vec::new(); | |
2025 | for i in 0..total { | |
2026 | ints.push(i); | |
2027 | } | |
2028 | ||
54a0048b | 2029 | self.commasep(Inconsistent, &ints[..], |s, &idx| { |
92a42be0 SL |
2030 | if idx < generics.lifetimes.len() { |
2031 | let lifetime = &generics.lifetimes[idx]; | |
2032 | s.print_lifetime_def(lifetime) | |
2033 | } else { | |
2034 | let idx = idx - generics.lifetimes.len(); | |
2035 | let param = &generics.ty_params[idx]; | |
2036 | s.print_ty_param(param) | |
2037 | } | |
54a0048b | 2038 | })?; |
e9174d1e | 2039 | |
041b39d2 | 2040 | self.s.word(">")?; |
e9174d1e SL |
2041 | Ok(()) |
2042 | } | |
2043 | ||
2044 | pub fn print_ty_param(&mut self, param: &hir::TyParam) -> io::Result<()> { | |
54a0048b SL |
2045 | self.print_name(param.name)?; |
2046 | self.print_bounds(":", ¶m.bounds)?; | |
e9174d1e SL |
2047 | match param.default { |
2048 | Some(ref default) => { | |
041b39d2 | 2049 | self.s.space()?; |
54a0048b | 2050 | self.word_space("=")?; |
7453a54e | 2051 | self.print_type(&default) |
e9174d1e | 2052 | } |
b039eaaf | 2053 | _ => Ok(()), |
e9174d1e SL |
2054 | } |
2055 | } | |
2056 | ||
b039eaaf | 2057 | pub fn print_where_clause(&mut self, where_clause: &hir::WhereClause) -> io::Result<()> { |
e9174d1e | 2058 | if where_clause.predicates.is_empty() { |
92a42be0 | 2059 | return Ok(()); |
e9174d1e SL |
2060 | } |
2061 | ||
041b39d2 | 2062 | self.s.space()?; |
54a0048b | 2063 | self.word_space("where")?; |
e9174d1e SL |
2064 | |
2065 | for (i, predicate) in where_clause.predicates.iter().enumerate() { | |
2066 | if i != 0 { | |
54a0048b | 2067 | self.word_space(",")?; |
e9174d1e SL |
2068 | } |
2069 | ||
2070 | match predicate { | |
2071 | &hir::WherePredicate::BoundPredicate(hir::WhereBoundPredicate{ref bound_lifetimes, | |
2072 | ref bounded_ty, | |
2073 | ref bounds, | |
2074 | ..}) => { | |
54a0048b SL |
2075 | self.print_formal_lifetime_list(bound_lifetimes)?; |
2076 | self.print_type(&bounded_ty)?; | |
2077 | self.print_bounds(":", bounds)?; | |
e9174d1e SL |
2078 | } |
2079 | &hir::WherePredicate::RegionPredicate(hir::WhereRegionPredicate{ref lifetime, | |
2080 | ref bounds, | |
2081 | ..}) => { | |
54a0048b | 2082 | self.print_lifetime(lifetime)?; |
041b39d2 | 2083 | self.s.word(":")?; |
e9174d1e SL |
2084 | |
2085 | for (i, bound) in bounds.iter().enumerate() { | |
54a0048b | 2086 | self.print_lifetime(bound)?; |
e9174d1e SL |
2087 | |
2088 | if i != 0 { | |
041b39d2 | 2089 | self.s.word(":")?; |
e9174d1e SL |
2090 | } |
2091 | } | |
2092 | } | |
32a655c1 SL |
2093 | &hir::WherePredicate::EqPredicate(hir::WhereEqPredicate{ref lhs_ty, |
2094 | ref rhs_ty, | |
2095 | ..}) => { | |
2096 | self.print_type(lhs_ty)?; | |
041b39d2 | 2097 | self.s.space()?; |
54a0048b | 2098 | self.word_space("=")?; |
32a655c1 | 2099 | self.print_type(rhs_ty)?; |
e9174d1e SL |
2100 | } |
2101 | } | |
2102 | } | |
2103 | ||
2104 | Ok(()) | |
2105 | } | |
2106 | ||
b039eaaf | 2107 | pub fn print_mutability(&mut self, mutbl: hir::Mutability) -> io::Result<()> { |
e9174d1e SL |
2108 | match mutbl { |
2109 | hir::MutMutable => self.word_nbsp("mut"), | |
2110 | hir::MutImmutable => Ok(()), | |
2111 | } | |
2112 | } | |
2113 | ||
2114 | pub fn print_mt(&mut self, mt: &hir::MutTy) -> io::Result<()> { | |
54a0048b | 2115 | self.print_mutability(mt.mutbl)?; |
7453a54e | 2116 | self.print_type(&mt.ty) |
e9174d1e SL |
2117 | } |
2118 | ||
e9174d1e SL |
2119 | pub fn print_fn_output(&mut self, decl: &hir::FnDecl) -> io::Result<()> { |
2120 | if let hir::DefaultReturn(..) = decl.output { | |
2121 | return Ok(()); | |
2122 | } | |
2123 | ||
54a0048b SL |
2124 | self.space_if_not_bol()?; |
2125 | self.ibox(indent_unit)?; | |
2126 | self.word_space("->")?; | |
e9174d1e | 2127 | match decl.output { |
e9174d1e | 2128 | hir::DefaultReturn(..) => unreachable!(), |
54a0048b | 2129 | hir::Return(ref ty) => self.print_type(&ty)?, |
e9174d1e | 2130 | } |
54a0048b | 2131 | self.end()?; |
e9174d1e SL |
2132 | |
2133 | match decl.output { | |
ea8adc8c | 2134 | hir::Return(ref output) => self.maybe_print_comment(output.span.lo()), |
b039eaaf | 2135 | _ => Ok(()), |
e9174d1e SL |
2136 | } |
2137 | } | |
2138 | ||
2139 | pub fn print_ty_fn(&mut self, | |
7453a54e | 2140 | abi: Abi, |
e9174d1e SL |
2141 | unsafety: hir::Unsafety, |
2142 | decl: &hir::FnDecl, | |
b039eaaf | 2143 | name: Option<ast::Name>, |
ea8adc8c XL |
2144 | generics: &hir::Generics, |
2145 | arg_names: &[Spanned<ast::Name>]) | |
e9174d1e | 2146 | -> io::Result<()> { |
54a0048b | 2147 | self.ibox(indent_unit)?; |
e9174d1e | 2148 | if !generics.lifetimes.is_empty() || !generics.ty_params.is_empty() { |
041b39d2 | 2149 | self.s.word("for")?; |
54a0048b | 2150 | self.print_generics(generics)?; |
e9174d1e SL |
2151 | } |
2152 | let generics = hir::Generics { | |
9cc50fc6 SL |
2153 | lifetimes: hir::HirVec::new(), |
2154 | ty_params: hir::HirVec::new(), | |
e9174d1e SL |
2155 | where_clause: hir::WhereClause { |
2156 | id: ast::DUMMY_NODE_ID, | |
9cc50fc6 | 2157 | predicates: hir::HirVec::new(), |
e9174d1e | 2158 | }, |
9e0c209e | 2159 | span: syntax_pos::DUMMY_SP, |
e9174d1e | 2160 | }; |
54a0048b SL |
2161 | self.print_fn(decl, |
2162 | unsafety, | |
2163 | hir::Constness::NotConst, | |
2164 | abi, | |
2165 | name, | |
2166 | &generics, | |
32a655c1 | 2167 | &hir::Inherited, |
ea8adc8c | 2168 | arg_names, |
32a655c1 | 2169 | None)?; |
e9174d1e SL |
2170 | self.end() |
2171 | } | |
2172 | ||
b039eaaf | 2173 | pub fn maybe_print_trailing_comment(&mut self, |
3157f602 | 2174 | span: syntax_pos::Span, |
e9174d1e | 2175 | next_pos: Option<BytePos>) |
b039eaaf | 2176 | -> io::Result<()> { |
e9174d1e SL |
2177 | let cm = match self.cm { |
2178 | Some(cm) => cm, | |
b039eaaf | 2179 | _ => return Ok(()), |
e9174d1e | 2180 | }; |
3157f602 XL |
2181 | if let Some(ref cmnt) = self.next_comment() { |
2182 | if (*cmnt).style != comments::Trailing { | |
2183 | return Ok(()); | |
2184 | } | |
ea8adc8c | 2185 | let span_line = cm.lookup_char_pos(span.hi()); |
3157f602 XL |
2186 | let comment_line = cm.lookup_char_pos((*cmnt).pos); |
2187 | let mut next = (*cmnt).pos + BytePos(1); | |
2188 | if let Some(p) = next_pos { | |
2189 | next = p; | |
2190 | } | |
ea8adc8c | 2191 | if span.hi() < (*cmnt).pos && (*cmnt).pos < next && |
3157f602 XL |
2192 | span_line.line == comment_line.line { |
2193 | self.print_comment(cmnt)?; | |
e9174d1e | 2194 | } |
e9174d1e SL |
2195 | } |
2196 | Ok(()) | |
2197 | } | |
2198 | ||
2199 | pub fn print_remaining_comments(&mut self) -> io::Result<()> { | |
2200 | // If there aren't any remaining comments, then we need to manually | |
2201 | // make sure there is a line break at the end. | |
2202 | if self.next_comment().is_none() { | |
041b39d2 | 2203 | self.s.hardbreak()?; |
e9174d1e SL |
2204 | } |
2205 | loop { | |
2206 | match self.next_comment() { | |
2207 | Some(ref cmnt) => { | |
54a0048b | 2208 | self.print_comment(cmnt)?; |
e9174d1e | 2209 | } |
b039eaaf | 2210 | _ => break, |
e9174d1e SL |
2211 | } |
2212 | } | |
2213 | Ok(()) | |
2214 | } | |
2215 | ||
e9174d1e | 2216 | pub fn print_opt_abi_and_extern_if_nondefault(&mut self, |
7453a54e | 2217 | opt_abi: Option<Abi>) |
b039eaaf | 2218 | -> io::Result<()> { |
e9174d1e | 2219 | match opt_abi { |
7453a54e | 2220 | Some(Abi::Rust) => Ok(()), |
e9174d1e | 2221 | Some(abi) => { |
54a0048b | 2222 | self.word_nbsp("extern")?; |
e9174d1e SL |
2223 | self.word_nbsp(&abi.to_string()) |
2224 | } | |
b039eaaf | 2225 | None => Ok(()), |
e9174d1e SL |
2226 | } |
2227 | } | |
2228 | ||
7453a54e | 2229 | pub fn print_extern_opt_abi(&mut self, opt_abi: Option<Abi>) -> io::Result<()> { |
e9174d1e SL |
2230 | match opt_abi { |
2231 | Some(abi) => { | |
54a0048b | 2232 | self.word_nbsp("extern")?; |
e9174d1e SL |
2233 | self.word_nbsp(&abi.to_string()) |
2234 | } | |
b039eaaf | 2235 | None => Ok(()), |
e9174d1e SL |
2236 | } |
2237 | } | |
2238 | ||
2239 | pub fn print_fn_header_info(&mut self, | |
2240 | unsafety: hir::Unsafety, | |
2241 | constness: hir::Constness, | |
7453a54e | 2242 | abi: Abi, |
54a0048b | 2243 | vis: &hir::Visibility) |
b039eaaf | 2244 | -> io::Result<()> { |
041b39d2 | 2245 | self.s.word(&visibility_qualified(vis, ""))?; |
54a0048b | 2246 | self.print_unsafety(unsafety)?; |
e9174d1e SL |
2247 | |
2248 | match constness { | |
2249 | hir::Constness::NotConst => {} | |
54a0048b | 2250 | hir::Constness::Const => self.word_nbsp("const")?, |
e9174d1e SL |
2251 | } |
2252 | ||
7453a54e | 2253 | if abi != Abi::Rust { |
54a0048b SL |
2254 | self.word_nbsp("extern")?; |
2255 | self.word_nbsp(&abi.to_string())?; | |
e9174d1e SL |
2256 | } |
2257 | ||
041b39d2 | 2258 | self.s.word("fn") |
e9174d1e SL |
2259 | } |
2260 | ||
2261 | pub fn print_unsafety(&mut self, s: hir::Unsafety) -> io::Result<()> { | |
2262 | match s { | |
2263 | hir::Unsafety::Normal => Ok(()), | |
2264 | hir::Unsafety::Unsafe => self.word_nbsp("unsafe"), | |
2265 | } | |
2266 | } | |
2267 | } | |
2268 | ||
e9174d1e SL |
2269 | // Dup'ed from parse::classify, but adapted for the HIR. |
2270 | /// Does this expression require a semicolon to be treated | |
2271 | /// as a statement? The negation of this: 'can this expression | |
2272 | /// be used as a statement without a semicolon' -- is used | |
2273 | /// as an early-bail-out in the parser so that, for instance, | |
2274 | /// if true {...} else {...} | |
2275 | /// |x| 5 | |
2276 | /// isn't parsed as (if true {...} else {...} | x) | 5 | |
2277 | fn expr_requires_semi_to_be_stmt(e: &hir::Expr) -> bool { | |
2278 | match e.node { | |
b039eaaf SL |
2279 | hir::ExprIf(..) | |
2280 | hir::ExprMatch(..) | | |
2281 | hir::ExprBlock(_) | | |
2282 | hir::ExprWhile(..) | | |
2283 | hir::ExprLoop(..) => false, | |
2284 | _ => true, | |
e9174d1e SL |
2285 | } |
2286 | } | |
2287 | ||
2288 | /// this statement requires a semicolon after it. | |
2289 | /// note that in one case (stmt_semi), we've already | |
2290 | /// seen the semicolon, and thus don't need another. | |
2291 | fn stmt_ends_with_semi(stmt: &hir::Stmt_) -> bool { | |
2292 | match *stmt { | |
2293 | hir::StmtDecl(ref d, _) => { | |
2294 | match d.node { | |
2295 | hir::DeclLocal(_) => true, | |
b039eaaf | 2296 | hir::DeclItem(_) => false, |
e9174d1e SL |
2297 | } |
2298 | } | |
b039eaaf | 2299 | hir::StmtExpr(ref e, _) => { |
7453a54e | 2300 | expr_requires_semi_to_be_stmt(&e) |
b039eaaf SL |
2301 | } |
2302 | hir::StmtSemi(..) => { | |
2303 | false | |
2304 | } | |
e9174d1e SL |
2305 | } |
2306 | } | |
ea8adc8c XL |
2307 | |
2308 | ||
2309 | fn expr_precedence(expr: &hir::Expr) -> i8 { | |
2310 | use syntax::util::parser::*; | |
2311 | ||
2312 | match expr.node { | |
2313 | hir::ExprClosure(..) => PREC_CLOSURE, | |
2314 | ||
2315 | hir::ExprBreak(..) | | |
2316 | hir::ExprAgain(..) | | |
2317 | hir::ExprRet(..) | | |
2318 | hir::ExprYield(..) => PREC_JUMP, | |
2319 | ||
2320 | // Binop-like expr kinds, handled by `AssocOp`. | |
2321 | hir::ExprBinary(op, _, _) => bin_op_to_assoc_op(op.node).precedence() as i8, | |
2322 | ||
2323 | hir::ExprCast(..) => AssocOp::As.precedence() as i8, | |
2324 | hir::ExprType(..) => AssocOp::Colon.precedence() as i8, | |
2325 | ||
2326 | hir::ExprAssign(..) | | |
2327 | hir::ExprAssignOp(..) => AssocOp::Assign.precedence() as i8, | |
2328 | ||
2329 | // Unary, prefix | |
2330 | hir::ExprBox(..) | | |
2331 | hir::ExprAddrOf(..) | | |
2332 | hir::ExprUnary(..) => PREC_PREFIX, | |
2333 | ||
2334 | // Unary, postfix | |
2335 | hir::ExprCall(..) | | |
2336 | hir::ExprMethodCall(..) | | |
2337 | hir::ExprField(..) | | |
2338 | hir::ExprTupField(..) | | |
2339 | hir::ExprIndex(..) | | |
2340 | hir::ExprInlineAsm(..) => PREC_POSTFIX, | |
2341 | ||
2342 | // Never need parens | |
2343 | hir::ExprArray(..) | | |
2344 | hir::ExprRepeat(..) | | |
2345 | hir::ExprTup(..) | | |
2346 | hir::ExprLit(..) | | |
2347 | hir::ExprPath(..) | | |
2348 | hir::ExprIf(..) | | |
2349 | hir::ExprWhile(..) | | |
2350 | hir::ExprLoop(..) | | |
2351 | hir::ExprMatch(..) | | |
2352 | hir::ExprBlock(..) | | |
2353 | hir::ExprStruct(..) => PREC_PAREN, | |
2354 | } | |
2355 | } | |
2356 | ||
2357 | fn bin_op_to_assoc_op(op: hir::BinOp_) -> AssocOp { | |
2358 | use hir::BinOp_::*; | |
2359 | match op { | |
2360 | BiAdd => AssocOp::Add, | |
2361 | BiSub => AssocOp::Subtract, | |
2362 | BiMul => AssocOp::Multiply, | |
2363 | BiDiv => AssocOp::Divide, | |
2364 | BiRem => AssocOp::Modulus, | |
2365 | ||
2366 | BiAnd => AssocOp::LAnd, | |
2367 | BiOr => AssocOp::LOr, | |
2368 | ||
2369 | BiBitXor => AssocOp::BitXor, | |
2370 | BiBitAnd => AssocOp::BitAnd, | |
2371 | BiBitOr => AssocOp::BitOr, | |
2372 | BiShl => AssocOp::ShiftLeft, | |
2373 | BiShr => AssocOp::ShiftRight, | |
2374 | ||
2375 | BiEq => AssocOp::Equal, | |
2376 | BiLt => AssocOp::Less, | |
2377 | BiLe => AssocOp::LessEqual, | |
2378 | BiNe => AssocOp::NotEqual, | |
2379 | BiGe => AssocOp::GreaterEqual, | |
2380 | BiGt => AssocOp::Greater, | |
2381 | } | |
2382 | } | |
2383 | ||
2384 | /// Expressions that syntactically contain an "exterior" struct literal i.e. not surrounded by any | |
2385 | /// parens or other delimiters, e.g. `X { y: 1 }`, `X { y: 1 }.method()`, `foo == X { y: 1 }` and | |
2386 | /// `X { y: 1 } == foo` all do, but `(X { y: 1 }) == foo` does not. | |
2387 | fn contains_exterior_struct_lit(value: &hir::Expr) -> bool { | |
2388 | match value.node { | |
2389 | hir::ExprStruct(..) => true, | |
2390 | ||
2391 | hir::ExprAssign(ref lhs, ref rhs) | | |
2392 | hir::ExprAssignOp(_, ref lhs, ref rhs) | | |
2393 | hir::ExprBinary(_, ref lhs, ref rhs) => { | |
2394 | // X { y: 1 } + X { y: 2 } | |
2395 | contains_exterior_struct_lit(&lhs) || contains_exterior_struct_lit(&rhs) | |
2396 | } | |
2397 | hir::ExprUnary(_, ref x) | | |
2398 | hir::ExprCast(ref x, _) | | |
2399 | hir::ExprType(ref x, _) | | |
2400 | hir::ExprField(ref x, _) | | |
2401 | hir::ExprTupField(ref x, _) | | |
2402 | hir::ExprIndex(ref x, _) => { | |
2403 | // &X { y: 1 }, X { y: 1 }.y | |
2404 | contains_exterior_struct_lit(&x) | |
2405 | } | |
2406 | ||
2407 | hir::ExprMethodCall(.., ref exprs) => { | |
2408 | // X { y: 1 }.bar(...) | |
2409 | contains_exterior_struct_lit(&exprs[0]) | |
2410 | } | |
2411 | ||
2412 | _ => false, | |
2413 | } | |
2414 | } |