]> git.lizzy.rs Git - rust.git/blob - src/types.rs
Merge pull request #3078 from YaLTeR/fix-closure-indentation
[rust.git] / src / types.rs
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 use std::iter::ExactSizeIterator;
12 use std::ops::Deref;
13
14 use config::lists::*;
15 use syntax::ast::{self, FunctionRetTy, Mutability};
16 use syntax::source_map::{self, BytePos, Span};
17 use syntax::symbol::keywords;
18
19 use config::{IndentStyle, TypeDensity};
20 use expr::{rewrite_assign_rhs, rewrite_tuple, rewrite_unary_prefix};
21 use lists::{definitive_tactic, itemize_list, write_list, ListFormatting, Separator};
22 use macros::{rewrite_macro, MacroPosition};
23 use overflow;
24 use pairs::{rewrite_pair, PairParts};
25 use rewrite::{Rewrite, RewriteContext};
26 use shape::Shape;
27 use source_map::SpanUtils;
28 use spanned::Spanned;
29 use utils::{
30     colon_spaces, extra_offset, first_line_width, format_abi, format_mutability,
31     last_line_extendable, last_line_width, mk_sp, rewrite_ident,
32 };
33
34 #[derive(Copy, Clone, Debug, Eq, PartialEq)]
35 pub enum PathContext {
36     Expr,
37     Type,
38     Import,
39 }
40
41 // Does not wrap on simple segments.
42 pub fn rewrite_path(
43     context: &RewriteContext,
44     path_context: PathContext,
45     qself: Option<&ast::QSelf>,
46     path: &ast::Path,
47     shape: Shape,
48 ) -> Option<String> {
49     let skip_count = qself.map_or(0, |x| x.position);
50
51     let mut result = if path.is_global() && qself.is_none() && path_context != PathContext::Import {
52         "::".to_owned()
53     } else {
54         String::new()
55     };
56
57     let mut span_lo = path.span.lo();
58
59     if let Some(qself) = qself {
60         result.push('<');
61
62         let fmt_ty = qself.ty.rewrite(context, shape)?;
63         result.push_str(&fmt_ty);
64
65         if skip_count > 0 {
66             result.push_str(" as ");
67             if path.is_global() && path_context != PathContext::Import {
68                 result.push_str("::");
69             }
70
71             // 3 = ">::".len()
72             let shape = shape.sub_width(3)?;
73
74             result = rewrite_path_segments(
75                 PathContext::Type,
76                 result,
77                 path.segments.iter().take(skip_count),
78                 span_lo,
79                 path.span.hi(),
80                 context,
81                 shape,
82             )?;
83         }
84
85         result.push_str(">::");
86         span_lo = qself.ty.span.hi() + BytePos(1);
87     }
88
89     rewrite_path_segments(
90         path_context,
91         result,
92         path.segments.iter().skip(skip_count),
93         span_lo,
94         path.span.hi(),
95         context,
96         shape,
97     )
98 }
99
100 fn rewrite_path_segments<'a, I>(
101     path_context: PathContext,
102     mut buffer: String,
103     iter: I,
104     mut span_lo: BytePos,
105     span_hi: BytePos,
106     context: &RewriteContext,
107     shape: Shape,
108 ) -> Option<String>
109 where
110     I: Iterator<Item = &'a ast::PathSegment>,
111 {
112     let mut first = true;
113     let shape = shape.visual_indent(0);
114
115     for segment in iter {
116         // Indicates a global path, shouldn't be rendered.
117         if segment.ident.name == keywords::CrateRoot.name() {
118             continue;
119         }
120         if first {
121             first = false;
122         } else {
123             buffer.push_str("::");
124         }
125
126         let extra_offset = extra_offset(&buffer, shape);
127         let new_shape = shape.shrink_left(extra_offset)?;
128         let segment_string = rewrite_segment(
129             path_context,
130             segment,
131             &mut span_lo,
132             span_hi,
133             context,
134             new_shape,
135         )?;
136
137         buffer.push_str(&segment_string);
138     }
139
140     Some(buffer)
141 }
142
143 #[derive(Debug)]
144 pub enum SegmentParam<'a> {
145     LifeTime(&'a ast::Lifetime),
146     Type(&'a ast::Ty),
147     Binding(&'a ast::TypeBinding),
148 }
149
150 impl<'a> SegmentParam<'a> {
151     fn from_generic_arg(arg: &ast::GenericArg) -> SegmentParam {
152         match arg {
153             ast::GenericArg::Lifetime(ref lt) => SegmentParam::LifeTime(lt),
154             ast::GenericArg::Type(ref ty) => SegmentParam::Type(ty),
155         }
156     }
157 }
158
159 impl<'a> Spanned for SegmentParam<'a> {
160     fn span(&self) -> Span {
161         match *self {
162             SegmentParam::LifeTime(lt) => lt.ident.span,
163             SegmentParam::Type(ty) => ty.span,
164             SegmentParam::Binding(binding) => binding.span,
165         }
166     }
167 }
168
169 impl<'a> Rewrite for SegmentParam<'a> {
170     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
171         match *self {
172             SegmentParam::LifeTime(lt) => lt.rewrite(context, shape),
173             SegmentParam::Type(ty) => ty.rewrite(context, shape),
174             SegmentParam::Binding(binding) => {
175                 let mut result = match context.config.type_punctuation_density() {
176                     TypeDensity::Wide => format!("{} = ", rewrite_ident(context, binding.ident)),
177                     TypeDensity::Compressed => {
178                         format!("{}=", rewrite_ident(context, binding.ident))
179                     }
180                 };
181                 let budget = shape.width.checked_sub(result.len())?;
182                 let rewrite = binding
183                     .ty
184                     .rewrite(context, Shape::legacy(budget, shape.indent + result.len()))?;
185                 result.push_str(&rewrite);
186                 Some(result)
187             }
188         }
189     }
190 }
191
192 // Formats a path segment. There are some hacks involved to correctly determine
193 // the segment's associated span since it's not part of the AST.
194 //
195 // The span_lo is assumed to be greater than the end of any previous segment's
196 // parameters and lesser or equal than the start of current segment.
197 //
198 // span_hi is assumed equal to the end of the entire path.
199 //
200 // When the segment contains a positive number of parameters, we update span_lo
201 // so that invariants described above will hold for the next segment.
202 fn rewrite_segment(
203     path_context: PathContext,
204     segment: &ast::PathSegment,
205     span_lo: &mut BytePos,
206     span_hi: BytePos,
207     context: &RewriteContext,
208     shape: Shape,
209 ) -> Option<String> {
210     let mut result = String::with_capacity(128);
211     result.push_str(rewrite_ident(context, segment.ident));
212
213     let ident_len = result.len();
214     let shape = if context.use_block_indent() {
215         shape.offset_left(ident_len)?
216     } else {
217         shape.shrink_left(ident_len)?
218     };
219
220     if let Some(ref args) = segment.args {
221         match **args {
222             ast::GenericArgs::AngleBracketed(ref data)
223                 if !data.args.is_empty() || !data.bindings.is_empty() =>
224             {
225                 let param_list = data
226                     .args
227                     .iter()
228                     .map(SegmentParam::from_generic_arg)
229                     .chain(data.bindings.iter().map(|x| SegmentParam::Binding(&*x)))
230                     .collect::<Vec<_>>();
231
232                 let separator = if path_context == PathContext::Expr {
233                     "::"
234                 } else {
235                     ""
236                 };
237                 result.push_str(separator);
238
239                 let generics_str = overflow::rewrite_with_angle_brackets(
240                     context,
241                     "",
242                     param_list.iter(),
243                     shape,
244                     mk_sp(*span_lo, span_hi),
245                 )?;
246
247                 // Update position of last bracket.
248                 *span_lo = context
249                     .snippet_provider
250                     .span_after(mk_sp(*span_lo, span_hi), "<");
251
252                 result.push_str(&generics_str)
253             }
254             ast::GenericArgs::Parenthesized(ref data) => {
255                 let output = match data.output {
256                     Some(ref ty) => FunctionRetTy::Ty(ty.clone()),
257                     None => FunctionRetTy::Default(source_map::DUMMY_SP),
258                 };
259                 result.push_str(&format_function_type(
260                     data.inputs.iter().map(|x| &**x),
261                     &output,
262                     false,
263                     data.span,
264                     context,
265                     shape,
266                 )?);
267             }
268             _ => (),
269         }
270     }
271
272     Some(result)
273 }
274
275 fn format_function_type<'a, I>(
276     inputs: I,
277     output: &FunctionRetTy,
278     variadic: bool,
279     span: Span,
280     context: &RewriteContext,
281     shape: Shape,
282 ) -> Option<String>
283 where
284     I: ExactSizeIterator,
285     <I as Iterator>::Item: Deref,
286     <I::Item as Deref>::Target: Rewrite + Spanned + 'a,
287 {
288     debug!("format_function_type {:#?}", shape);
289
290     // Code for handling variadics is somewhat duplicated for items, but they
291     // are different enough to need some serious refactoring to share code.
292     enum ArgumentKind<T>
293     where
294         T: Deref,
295         <T as Deref>::Target: Rewrite + Spanned,
296     {
297         Regular(T),
298         Variadic(BytePos),
299     }
300
301     let variadic_arg = if variadic {
302         let variadic_start = context.snippet_provider.span_before(span, "...");
303         Some(ArgumentKind::Variadic(variadic_start))
304     } else {
305         None
306     };
307
308     // 2 for ()
309     let budget = shape.width.checked_sub(2)?;
310     // 1 for (
311     let offset = match context.config.indent_style() {
312         IndentStyle::Block => {
313             shape
314                 .block()
315                 .block_indent(context.config.tab_spaces())
316                 .indent
317         }
318         IndentStyle::Visual => shape.indent + 1,
319     };
320     let list_shape = Shape::legacy(budget, offset);
321     let list_lo = context.snippet_provider.span_after(span, "(");
322     let items = itemize_list(
323         context.snippet_provider,
324         inputs.map(ArgumentKind::Regular).chain(variadic_arg),
325         ")",
326         ",",
327         |arg| match *arg {
328             ArgumentKind::Regular(ref ty) => ty.span().lo(),
329             ArgumentKind::Variadic(start) => start,
330         },
331         |arg| match *arg {
332             ArgumentKind::Regular(ref ty) => ty.span().hi(),
333             ArgumentKind::Variadic(start) => start + BytePos(3),
334         },
335         |arg| match *arg {
336             ArgumentKind::Regular(ref ty) => ty.rewrite(context, list_shape),
337             ArgumentKind::Variadic(_) => Some("...".to_owned()),
338         },
339         list_lo,
340         span.hi(),
341         false,
342     );
343
344     let item_vec: Vec<_> = items.collect();
345
346     let tactic = definitive_tactic(
347         &*item_vec,
348         ListTactic::HorizontalVertical,
349         Separator::Comma,
350         budget,
351     );
352     let trailing_separator = if !context.use_block_indent() || variadic {
353         SeparatorTactic::Never
354     } else {
355         context.config.trailing_comma()
356     };
357
358     let fmt = ListFormatting::new(list_shape, context.config)
359         .tactic(tactic)
360         .trailing_separator(trailing_separator)
361         .ends_with_newline(tactic.ends_with_newline(context.config.indent_style()))
362         .preserve_newline(true);
363     let list_str = write_list(&item_vec, &fmt)?;
364
365     let ty_shape = match context.config.indent_style() {
366         // 4 = " -> "
367         IndentStyle::Block => shape.offset_left(4)?,
368         IndentStyle::Visual => shape.block_left(4)?,
369     };
370     let output = match *output {
371         FunctionRetTy::Ty(ref ty) => {
372             let type_str = ty.rewrite(context, ty_shape)?;
373             format!(" -> {}", type_str)
374         }
375         FunctionRetTy::Default(..) => String::new(),
376     };
377
378     let args = if (!list_str.contains('\n') || list_str.is_empty()) && !output.contains('\n')
379         || !context.use_block_indent()
380     {
381         format!("({})", list_str)
382     } else {
383         format!(
384             "({}{}{})",
385             offset.to_string_with_newline(context.config),
386             list_str,
387             shape.block().indent.to_string_with_newline(context.config),
388         )
389     };
390     if output.is_empty() || last_line_width(&args) + first_line_width(&output) <= shape.width {
391         Some(format!("{}{}", args, output))
392     } else {
393         Some(format!(
394             "{}\n{}{}",
395             args,
396             offset.to_string(context.config),
397             output.trim_left()
398         ))
399     }
400 }
401
402 fn type_bound_colon(context: &RewriteContext) -> &'static str {
403     colon_spaces(
404         context.config.space_before_colon(),
405         context.config.space_after_colon(),
406     )
407 }
408
409 impl Rewrite for ast::WherePredicate {
410     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
411         // FIXME: dead spans?
412         let result = match *self {
413             ast::WherePredicate::BoundPredicate(ast::WhereBoundPredicate {
414                 ref bound_generic_params,
415                 ref bounded_ty,
416                 ref bounds,
417                 ..
418             }) => {
419                 let type_str = bounded_ty.rewrite(context, shape)?;
420                 let colon = type_bound_colon(context).trim_right();
421                 let lhs = if let Some(lifetime_str) =
422                     rewrite_lifetime_param(context, shape, bound_generic_params)
423                 {
424                     format!("for<{}> {}{}", lifetime_str, type_str, colon)
425                 } else {
426                     format!("{}{}", type_str, colon)
427                 };
428
429                 rewrite_assign_rhs(context, lhs, bounds, shape)?
430             }
431             ast::WherePredicate::RegionPredicate(ast::WhereRegionPredicate {
432                 ref lifetime,
433                 ref bounds,
434                 ..
435             }) => rewrite_bounded_lifetime(lifetime, bounds, context, shape)?,
436             ast::WherePredicate::EqPredicate(ast::WhereEqPredicate {
437                 ref lhs_ty,
438                 ref rhs_ty,
439                 ..
440             }) => {
441                 let lhs_ty_str = lhs_ty.rewrite(context, shape).map(|lhs| lhs + " =")?;
442                 rewrite_assign_rhs(context, lhs_ty_str, &**rhs_ty, shape)?
443             }
444         };
445
446         Some(result)
447     }
448 }
449
450 impl Rewrite for ast::GenericArg {
451     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
452         match *self {
453             ast::GenericArg::Lifetime(ref lt) => lt.rewrite(context, shape),
454             ast::GenericArg::Type(ref ty) => ty.rewrite(context, shape),
455         }
456     }
457 }
458
459 fn rewrite_bounded_lifetime(
460     lt: &ast::Lifetime,
461     bounds: &[ast::GenericBound],
462     context: &RewriteContext,
463     shape: Shape,
464 ) -> Option<String> {
465     let result = lt.rewrite(context, shape)?;
466
467     if bounds.is_empty() {
468         Some(result)
469     } else {
470         let colon = type_bound_colon(context);
471         let overhead = last_line_width(&result) + colon.len();
472         let result = format!(
473             "{}{}{}",
474             result,
475             colon,
476             join_bounds(context, shape.sub_width(overhead)?, bounds, true)?
477         );
478         Some(result)
479     }
480 }
481
482 impl Rewrite for ast::Lifetime {
483     fn rewrite(&self, context: &RewriteContext, _: Shape) -> Option<String> {
484         Some(rewrite_ident(context, self.ident).to_owned())
485     }
486 }
487
488 impl Rewrite for ast::GenericBound {
489     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
490         match *self {
491             ast::GenericBound::Trait(ref poly_trait_ref, trait_bound_modifier) => {
492                 match trait_bound_modifier {
493                     ast::TraitBoundModifier::None => poly_trait_ref.rewrite(context, shape),
494                     ast::TraitBoundModifier::Maybe => {
495                         let rw = poly_trait_ref.rewrite(context, shape.offset_left(1)?)?;
496                         Some(format!("?{}", rw))
497                     }
498                 }
499             }
500             ast::GenericBound::Outlives(ref lifetime) => lifetime.rewrite(context, shape),
501         }
502     }
503 }
504
505 impl Rewrite for ast::GenericBounds {
506     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
507         if self.is_empty() {
508             return Some(String::new());
509         }
510
511         let span = mk_sp(self.get(0)?.span().lo(), self.last()?.span().hi());
512         let has_paren = context.snippet(span).starts_with('(');
513         let bounds_shape = if has_paren {
514             shape.offset_left(1)?.sub_width(1)?
515         } else {
516             shape
517         };
518         join_bounds(context, bounds_shape, self, true).map(|s| {
519             if has_paren {
520                 format!("({})", s)
521             } else {
522                 s
523             }
524         })
525     }
526 }
527
528 impl Rewrite for ast::GenericParam {
529     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
530         let mut result = String::with_capacity(128);
531         // FIXME: If there are more than one attributes, this will force multiline.
532         match self.attrs.rewrite(context, shape) {
533             Some(ref rw) if !rw.is_empty() => result.push_str(&format!("{} ", rw)),
534             _ => (),
535         }
536         result.push_str(rewrite_ident(context, self.ident));
537         if !self.bounds.is_empty() {
538             result.push_str(type_bound_colon(context));
539             result.push_str(&self.bounds.rewrite(context, shape)?)
540         }
541         if let ast::GenericParamKind::Type {
542             default: Some(ref def),
543         } = self.kind
544         {
545             let eq_str = match context.config.type_punctuation_density() {
546                 TypeDensity::Compressed => "=",
547                 TypeDensity::Wide => " = ",
548             };
549             result.push_str(eq_str);
550             let budget = shape.width.checked_sub(result.len())?;
551             let rewrite =
552                 def.rewrite(context, Shape::legacy(budget, shape.indent + result.len()))?;
553             result.push_str(&rewrite);
554         }
555
556         Some(result)
557     }
558 }
559
560 impl Rewrite for ast::PolyTraitRef {
561     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
562         if let Some(lifetime_str) =
563             rewrite_lifetime_param(context, shape, &self.bound_generic_params)
564         {
565             // 6 is "for<> ".len()
566             let extra_offset = lifetime_str.len() + 6;
567             let path_str = self
568                 .trait_ref
569                 .rewrite(context, shape.offset_left(extra_offset)?)?;
570
571             Some(format!("for<{}> {}", lifetime_str, path_str))
572         } else {
573             self.trait_ref.rewrite(context, shape)
574         }
575     }
576 }
577
578 impl Rewrite for ast::TraitRef {
579     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
580         rewrite_path(context, PathContext::Type, None, &self.path, shape)
581     }
582 }
583
584 impl Rewrite for ast::Ty {
585     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
586         match self.node {
587             ast::TyKind::TraitObject(ref bounds, tobj_syntax) => {
588                 // we have to consider 'dyn' keyword is used or not!!!
589                 let is_dyn = tobj_syntax == ast::TraitObjectSyntax::Dyn;
590                 // 4 is length of 'dyn '
591                 let shape = if is_dyn { shape.offset_left(4)? } else { shape };
592                 let res = bounds.rewrite(context, shape)?;
593                 if is_dyn {
594                     Some(format!("dyn {}", res))
595                 } else {
596                     Some(res)
597                 }
598             }
599             ast::TyKind::Ptr(ref mt) => {
600                 let prefix = match mt.mutbl {
601                     Mutability::Mutable => "*mut ",
602                     Mutability::Immutable => "*const ",
603                 };
604
605                 rewrite_unary_prefix(context, prefix, &*mt.ty, shape)
606             }
607             ast::TyKind::Rptr(ref lifetime, ref mt) => {
608                 let mut_str = format_mutability(mt.mutbl);
609                 let mut_len = mut_str.len();
610                 Some(match *lifetime {
611                     Some(ref lifetime) => {
612                         let lt_budget = shape.width.checked_sub(2 + mut_len)?;
613                         let lt_str = lifetime.rewrite(
614                             context,
615                             Shape::legacy(lt_budget, shape.indent + 2 + mut_len),
616                         )?;
617                         let lt_len = lt_str.len();
618                         let budget = shape.width.checked_sub(2 + mut_len + lt_len)?;
619                         format!(
620                             "&{} {}{}",
621                             lt_str,
622                             mut_str,
623                             mt.ty.rewrite(
624                                 context,
625                                 Shape::legacy(budget, shape.indent + 2 + mut_len + lt_len)
626                             )?
627                         )
628                     }
629                     None => {
630                         let budget = shape.width.checked_sub(1 + mut_len)?;
631                         format!(
632                             "&{}{}",
633                             mut_str,
634                             mt.ty.rewrite(
635                                 context,
636                                 Shape::legacy(budget, shape.indent + 1 + mut_len)
637                             )?
638                         )
639                     }
640                 })
641             }
642             // FIXME: we drop any comments here, even though it's a silly place to put
643             // comments.
644             ast::TyKind::Paren(ref ty) => {
645                 let budget = shape.width.checked_sub(2)?;
646                 ty.rewrite(context, Shape::legacy(budget, shape.indent + 1))
647                     .map(|ty_str| format!("({})", ty_str))
648             }
649             ast::TyKind::Slice(ref ty) => {
650                 let budget = shape.width.checked_sub(4)?;
651                 ty.rewrite(context, Shape::legacy(budget, shape.indent + 1))
652                     .map(|ty_str| format!("[{}]", ty_str))
653             }
654             ast::TyKind::Tup(ref items) => {
655                 rewrite_tuple(context, items.iter(), self.span, shape, items.len() == 1)
656             }
657             ast::TyKind::Path(ref q_self, ref path) => {
658                 rewrite_path(context, PathContext::Type, q_self.as_ref(), path, shape)
659             }
660             ast::TyKind::Array(ref ty, ref repeats) => rewrite_pair(
661                 &**ty,
662                 &*repeats.value,
663                 PairParts::new("[", "; ", "]"),
664                 context,
665                 shape,
666                 SeparatorPlace::Back,
667             ),
668             ast::TyKind::Infer => {
669                 if shape.width >= 1 {
670                     Some("_".to_owned())
671                 } else {
672                     None
673                 }
674             }
675             ast::TyKind::BareFn(ref bare_fn) => rewrite_bare_fn(bare_fn, self.span, context, shape),
676             ast::TyKind::Never => Some(String::from("!")),
677             ast::TyKind::Mac(ref mac) => {
678                 rewrite_macro(mac, None, context, shape, MacroPosition::Expression)
679             }
680             ast::TyKind::ImplicitSelf => Some(String::from("")),
681             ast::TyKind::ImplTrait(_, ref it) => it
682                 .rewrite(context, shape)
683                 .map(|it_str| format!("impl {}", it_str)),
684             ast::TyKind::Err | ast::TyKind::Typeof(..) => unreachable!(),
685         }
686     }
687 }
688
689 fn rewrite_bare_fn(
690     bare_fn: &ast::BareFnTy,
691     span: Span,
692     context: &RewriteContext,
693     shape: Shape,
694 ) -> Option<String> {
695     debug!("rewrite_bare_fn {:#?}", shape);
696
697     let mut result = String::with_capacity(128);
698
699     if let Some(ref lifetime_str) = rewrite_lifetime_param(context, shape, &bare_fn.generic_params)
700     {
701         result.push_str("for<");
702         // 6 = "for<> ".len(), 4 = "for<".
703         // This doesn't work out so nicely for multiline situation with lots of
704         // rightward drift. If that is a problem, we could use the list stuff.
705         result.push_str(lifetime_str);
706         result.push_str("> ");
707     }
708
709     result.push_str(::utils::format_unsafety(bare_fn.unsafety));
710
711     result.push_str(&format_abi(
712         bare_fn.abi,
713         context.config.force_explicit_abi(),
714         false,
715     ));
716
717     result.push_str("fn");
718
719     let func_ty_shape = if context.use_block_indent() {
720         shape.offset_left(result.len())?
721     } else {
722         shape.visual_indent(result.len()).sub_width(result.len())?
723     };
724
725     let rewrite = format_function_type(
726         bare_fn.decl.inputs.iter(),
727         &bare_fn.decl.output,
728         bare_fn.decl.variadic,
729         span,
730         context,
731         func_ty_shape,
732     )?;
733
734     result.push_str(&rewrite);
735
736     Some(result)
737 }
738
739 fn is_generic_bounds_in_order(generic_bounds: &[ast::GenericBound]) -> bool {
740     let is_trait = |b: &ast::GenericBound| match b {
741         ast::GenericBound::Outlives(..) => false,
742         ast::GenericBound::Trait(..) => true,
743     };
744     let is_lifetime = |b: &ast::GenericBound| !is_trait(b);
745     let last_trait_index = generic_bounds.iter().rposition(is_trait);
746     let first_lifetime_index = generic_bounds.iter().position(is_lifetime);
747     match (last_trait_index, first_lifetime_index) {
748         (Some(last_trait_index), Some(first_lifetime_index)) => {
749             last_trait_index < first_lifetime_index
750         }
751         _ => true,
752     }
753 }
754
755 fn join_bounds(
756     context: &RewriteContext,
757     shape: Shape,
758     items: &[ast::GenericBound],
759     need_indent: bool,
760 ) -> Option<String> {
761     // Try to join types in a single line
762     let joiner = match context.config.type_punctuation_density() {
763         TypeDensity::Compressed => "+",
764         TypeDensity::Wide => " + ",
765     };
766     let type_strs = items
767         .iter()
768         .map(|item| item.rewrite(context, shape))
769         .collect::<Option<Vec<_>>>()?;
770     let result = type_strs.join(joiner);
771     if items.len() <= 1 || (!result.contains('\n') && result.len() <= shape.width) {
772         return Some(result);
773     }
774
775     // We need to use multiple lines.
776     let (type_strs, offset) = if need_indent {
777         // Rewrite with additional indentation.
778         let nested_shape = shape.block_indent(context.config.tab_spaces());
779         let type_strs = items
780             .iter()
781             .map(|item| item.rewrite(context, nested_shape))
782             .collect::<Option<Vec<_>>>()?;
783         (type_strs, nested_shape.indent)
784     } else {
785         (type_strs, shape.indent)
786     };
787
788     let is_bound_extendable = |s: &str, b: &ast::GenericBound| match b {
789         ast::GenericBound::Outlives(..) => true,
790         ast::GenericBound::Trait(..) => last_line_extendable(s),
791     };
792     let mut result = String::with_capacity(128);
793     result.push_str(&type_strs[0]);
794     let mut can_be_put_on_the_same_line = is_bound_extendable(&result, &items[0]);
795     let generic_bounds_in_order = is_generic_bounds_in_order(items);
796     for (bound, bound_str) in items[1..].iter().zip(type_strs[1..].iter()) {
797         if generic_bounds_in_order && can_be_put_on_the_same_line {
798             result.push_str(joiner);
799         } else {
800             result.push_str(&offset.to_string_with_newline(context.config));
801             result.push_str("+ ");
802         }
803         result.push_str(bound_str);
804         can_be_put_on_the_same_line = is_bound_extendable(bound_str, bound);
805     }
806
807     Some(result)
808 }
809
810 pub fn can_be_overflowed_type(context: &RewriteContext, ty: &ast::Ty, len: usize) -> bool {
811     match ty.node {
812         ast::TyKind::Tup(..) => context.use_block_indent() && len == 1,
813         ast::TyKind::Rptr(_, ref mutty) | ast::TyKind::Ptr(ref mutty) => {
814             can_be_overflowed_type(context, &*mutty.ty, len)
815         }
816         _ => false,
817     }
818 }
819
820 /// Returns `None` if there is no `LifetimeDef` in the given generic parameters.
821 fn rewrite_lifetime_param(
822     context: &RewriteContext,
823     shape: Shape,
824     generic_params: &[ast::GenericParam],
825 ) -> Option<String> {
826     let result = generic_params
827         .iter()
828         .filter(|p| match p.kind {
829             ast::GenericParamKind::Lifetime => true,
830             _ => false,
831         })
832         .map(|lt| lt.rewrite(context, shape))
833         .collect::<Option<Vec<_>>>()?
834         .join(", ");
835     if result.is_empty() {
836         None
837     } else {
838         Some(result)
839     }
840 }