src/attr.rs

   1 // Copyright 2018 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 //! Format attributes and meta items.
  12
  13 use comment::{contains_comment, rewrite_doc_comment};
  14 use config::lists::*;
  15 use config::IndentStyle;
  16 use expr::rewrite_literal;
  17 use lists::{definitive_tactic, itemize_list, write_list, ListFormatting, Separator};
  18 use rewrite::{Rewrite, RewriteContext};
  19 use shape::Shape;
  20 use types::{rewrite_path, PathContext};
  21 use utils::{count_newlines, mk_sp};
  22
  23 use std::borrow::Cow;
  24 use syntax::ast;
  25 use syntax::codemap::{BytePos, Span, DUMMY_SP};
  26
  27 /// Returns attributes on the given statement.
  28 pub fn get_attrs_from_stmt(stmt: &ast::Stmt) -> &[ast::Attribute] {
  29     match stmt.node {
  30         ast::StmtKind::Local(ref local) => &local.attrs,
  31         ast::StmtKind::Item(ref item) => &item.attrs,
  32         ast::StmtKind::Expr(ref expr) | ast::StmtKind::Semi(ref expr) => &expr.attrs,
  33         ast::StmtKind::Mac(ref mac) => &mac.2,
  34     }
  35 }
  36
  37 /// Returns attributes that are within `outer_span`.
  38 pub fn filter_inline_attrs(attrs: &[ast::Attribute], outer_span: Span) -> Vec<ast::Attribute> {
  39     attrs
  40         .iter()
  41         .filter(|a| outer_span.lo() <= a.span.lo() && a.span.hi() <= outer_span.hi())
  42         .cloned()
  43         .collect()
  44 }
  45
  46 fn is_derive(attr: &ast::Attribute) -> bool {
  47     attr.check_name("derive")
  48 }
  49
  50 /// Returns the arguments of `#[derive(...)]`.
  51 fn get_derive_spans<'a>(attr: &ast::Attribute) -> Option<Vec<Span>> {
  52     attr.meta_item_list().map(|meta_item_list| {
  53         meta_item_list
  54             .iter()
  55             .map(|nested_meta_item| nested_meta_item.span)
  56             .collect()
  57     })
  58 }
  59
  60 // The shape of the arguments to a function-like attribute.
  61 fn argument_shape(
  62     left: usize,
  63     right: usize,
  64     combine: bool,
  65     shape: Shape,
  66     context: &RewriteContext,
  67 ) -> Option<Shape> {
  68     match context.config.indent_style() {
  69         IndentStyle::Block => {
  70             if combine {
  71                 shape.offset_left(left)
  72             } else {
  73                 Some(
  74                     shape
  75                         .block_indent(context.config.tab_spaces())
  76                         .with_max_width(context.config),
  77                 )
  78             }
  79         }
  80         IndentStyle::Visual => shape
  81             .visual_indent(0)
  82             .shrink_left(left)
  83             .and_then(|s| s.sub_width(right)),
  84     }
  85 }
  86
  87 fn format_derive(
  88     derive_args: &[Span],
  89     prefix: &str,
  90     shape: Shape,
  91     context: &RewriteContext,
  92 ) -> Option<String> {
  93     let mut result = String::with_capacity(128);
  94     result.push_str(prefix);
  95     result.push_str("[derive(");
  96
  97     let argument_shape = argument_shape(10 + prefix.len(), 2, false, shape, context)?;
  98     let item_str = format_arg_list(
  99         derive_args.iter(),
 100         |_| DUMMY_SP.lo(),
 101         |_| DUMMY_SP.hi(),
 102         |sp| Some(context.snippet(**sp).to_owned()),
 103         DUMMY_SP,
 104         context,
 105         argument_shape,
 106         // 10 = "[derive()]", 3 = "()" and "]"
 107         shape.offset_left(10 + prefix.len())?.sub_width(3)?,
 108         None,
 109         false,
 110     )?;
 111
 112     result.push_str(&item_str);
 113     if item_str.starts_with('\n') {
 114         result.push(',');
 115         result.push_str(&shape.indent.to_string_with_newline(context.config));
 116     }
 117     result.push_str(")]");
 118     Some(result)
 119 }
 120
 121 /// Returns the first group of attributes that fills the given predicate.
 122 /// We consider two doc comments are in different group if they are separated by normal comments.
 123 fn take_while_with_pred<'a, P>(
 124     context: &RewriteContext,
 125     attrs: &'a [ast::Attribute],
 126     pred: P,
 127 ) -> &'a [ast::Attribute]
 128 where
 129     P: Fn(&ast::Attribute) -> bool,
 130 {
 131     let mut len = 0;
 132     let mut iter = attrs.iter().peekable();
 133
 134     while let Some(attr) = iter.next() {
 135         if pred(attr) {
 136             len += 1;
 137         } else {
 138             break;
 139         }
 140         if let Some(next_attr) = iter.peek() {
 141             // Extract comments between two attributes.
 142             let span_between_attr = mk_sp(attr.span.hi(), next_attr.span.lo());
 143             let snippet = context.snippet(span_between_attr);
 144             if count_newlines(snippet) >= 2 || snippet.contains('/') {
 145                 break;
 146             }
 147         }
 148     }
 149
 150     &attrs[..len]
 151 }
 152
 153 /// Rewrite the any doc comments which come before any other attributes.
 154 fn rewrite_initial_doc_comments(
 155     context: &RewriteContext,
 156     attrs: &[ast::Attribute],
 157     shape: Shape,
 158 ) -> Option<(usize, Option<String>)> {
 159     if attrs.is_empty() {
 160         return Some((0, None));
 161     }
 162     // Rewrite doc comments
 163     let sugared_docs = take_while_with_pred(context, attrs, |a| a.is_sugared_doc);
 164     if !sugared_docs.is_empty() {
 165         let snippet = sugared_docs
 166             .iter()
 167             .map(|a| context.snippet(a.span))
 168             .collect::<Vec<_>>()
 169             .join("\n");
 170         return Some((
 171             sugared_docs.len(),
 172             Some(rewrite_doc_comment(
 173                 &snippet,
 174                 shape.comment(context.config),
 175                 context.config,
 176             )?),
 177         ));
 178     }
 179
 180     Some((0, None))
 181 }
 182
 183 impl Rewrite for ast::NestedMetaItem {
 184     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
 185         match self.node {
 186             ast::NestedMetaItemKind::MetaItem(ref meta_item) => meta_item.rewrite(context, shape),
 187             ast::NestedMetaItemKind::Literal(ref l) => rewrite_literal(context, l, shape),
 188         }
 189     }
 190 }
 191
 192 fn has_newlines_before_after_comment(comment: &str) -> (&str, &str) {
 193     // Look at before and after comment and see if there are any empty lines.
 194     let comment_begin = comment.find('/');
 195     let len = comment_begin.unwrap_or_else(|| comment.len());
 196     let mlb = count_newlines(&comment[..len]) > 1;
 197     let mla = if comment_begin.is_none() {
 198         mlb
 199     } else {
 200         comment
 201             .chars()
 202             .rev()
 203             .take_while(|c| c.is_whitespace())
 204             .filter(|&c| c == '\n')
 205             .count()
 206             > 1
 207     };
 208     (if mlb { "\n" } else { "" }, if mla { "\n" } else { "" })
 209 }
 210
 211 impl Rewrite for ast::MetaItem {
 212     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
 213         Some(match self.node {
 214             ast::MetaItemKind::Word => {
 215                 rewrite_path(context, PathContext::Type, None, &self.ident, shape)?
 216             }
 217             ast::MetaItemKind::List(ref list) => {
 218                 let path = rewrite_path(context, PathContext::Type, None, &self.ident, shape)?;
 219
 220                 let snippet = context.snippet(self.span);
 221                 // 2 = )] (this might go wrong if there is whitespace between the brackets, but
 222                 // it's close enough).
 223                 let snippet = snippet[..snippet.len() - 2].trim();
 224                 let trailing_comma = if snippet.ends_with(',') { "," } else { "" };
 225                 let combine = list.len() == 1 && match list[0].node {
 226                     ast::NestedMetaItemKind::Literal(..) => false,
 227                     ast::NestedMetaItemKind::MetaItem(ref inner_meta_item) => {
 228                         match inner_meta_item.node {
 229                             ast::MetaItemKind::List(..) => rewrite_path(
 230                                 context,
 231                                 PathContext::Type,
 232                                 None,
 233                                 &inner_meta_item.ident,
 234                                 shape,
 235                             ).map_or(false, |s| s.len() + path.len() + 2 <= shape.width),
 236                             _ => false,
 237                         }
 238                     }
 239                 };
 240
 241                 let argument_shape = argument_shape(
 242                     path.len() + 1,
 243                     2 + trailing_comma.len(),
 244                     combine,
 245                     shape,
 246                     context,
 247                 )?;
 248                 let item_str = format_arg_list(
 249                     list.iter(),
 250                     |nested_meta_item| nested_meta_item.span.lo(),
 251                     |nested_meta_item| nested_meta_item.span.hi(),
 252                     |nested_meta_item| nested_meta_item.rewrite(context, argument_shape),
 253                     self.span,
 254                     context,
 255                     argument_shape,
 256                     // 3 = "()" and "]"
 257                     shape
 258                         .offset_left(path.len())?
 259                         .sub_width(3 + trailing_comma.len())?,
 260                     Some(context.config.width_heuristics().fn_call_width),
 261                     combine,
 262                 )?;
 263
 264                 let indent = if item_str.starts_with('\n') {
 265                     shape.indent.to_string_with_newline(context.config)
 266                 } else {
 267                     Cow::Borrowed("")
 268                 };
 269
 270                 format!("{}({}{}{})", path, item_str, trailing_comma, indent)
 271             }
 272             ast::MetaItemKind::NameValue(ref literal) => {
 273                 let path = rewrite_path(context, PathContext::Type, None, &self.ident, shape)?;
 274                 // 3 = ` = `
 275                 let lit_shape = shape.shrink_left(path.len() + 3)?;
 276                 // `rewrite_literal` returns `None` when `literal` exceeds max
 277                 // width. Since a literal is basically unformattable unless it
 278                 // is a string literal (and only if `format_strings` is set),
 279                 // we might be better off ignoring the fact that the attribute
 280                 // is longer than the max width and contiue on formatting.
 281                 // See #2479 for example.
 282                 let value = rewrite_literal(context, literal, lit_shape)
 283                     .unwrap_or_else(|| context.snippet(literal.span).to_owned());
 284                 format!("{} = {}", path, value)
 285             }
 286         })
 287     }
 288 }
 289
 290 fn format_arg_list<I, T, F1, F2, F3>(
 291     list: I,
 292     get_lo: F1,
 293     get_hi: F2,
 294     get_item_string: F3,
 295     span: Span,
 296     context: &RewriteContext,
 297     shape: Shape,
 298     one_line_shape: Shape,
 299     one_line_limit: Option<usize>,
 300     combine: bool,
 301 ) -> Option<String>
 302 where
 303     I: Iterator<Item = T>,
 304     F1: Fn(&T) -> BytePos,
 305     F2: Fn(&T) -> BytePos,
 306     F3: Fn(&T) -> Option<String>,
 307 {
 308     let items = itemize_list(
 309         context.snippet_provider,
 310         list,
 311         ")",
 312         ",",
 313         get_lo,
 314         get_hi,
 315         get_item_string,
 316         span.lo(),
 317         span.hi(),
 318         false,
 319     );
 320     let item_vec = items.collect::<Vec<_>>();
 321     let tactic = if let Some(limit) = one_line_limit {
 322         ListTactic::LimitedHorizontalVertical(limit)
 323     } else {
 324         ListTactic::HorizontalVertical
 325     };
 326
 327     let tactic = definitive_tactic(&item_vec, tactic, Separator::Comma, shape.width);
 328     let fmt = ListFormatting::new(shape, context.config)
 329         .tactic(tactic)
 330         .ends_with_newline(false);
 331     let item_str = write_list(&item_vec, &fmt)?;
 332
 333     let one_line_budget = one_line_shape.width;
 334     if context.config.indent_style() == IndentStyle::Visual
 335         || combine
 336         || (!item_str.contains('\n') && item_str.len() <= one_line_budget)
 337     {
 338         Some(item_str)
 339     } else {
 340         let nested_indent = shape.indent.to_string_with_newline(context.config);
 341         Some(format!("{}{}", nested_indent, item_str))
 342     }
 343 }
 344
 345 impl Rewrite for ast::Attribute {
 346     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
 347         let snippet = context.snippet(self.span);
 348         if self.is_sugared_doc {
 349             rewrite_doc_comment(snippet, shape.comment(context.config), context.config)
 350         } else {
 351             let prefix = attr_prefix(self);
 352
 353             if contains_comment(snippet) {
 354                 return Some(snippet.to_owned());
 355             }
 356             // 1 = `[`
 357             let shape = shape.offset_left(prefix.len() + 1)?;
 358             Some(
 359                 self.meta()
 360                     .and_then(|meta| meta.rewrite(context, shape))
 361                     .map_or_else(|| snippet.to_owned(), |rw| format!("{}[{}]", prefix, rw)),
 362             )
 363         }
 364     }
 365 }
 366
 367 impl<'a> Rewrite for [ast::Attribute] {
 368     fn rewrite(&self, context: &RewriteContext, shape: Shape) -> Option<String> {
 369         if self.is_empty() {
 370             return Some(String::new());
 371         }
 372
 373         // The current remaining attributes.
 374         let mut attrs = self;
 375         let mut result = String::new();
 376
 377         // This is not just a simple map because we need to handle doc comments
 378         // (where we take as many doc comment attributes as possible) and possibly
 379         // merging derives into a single attribute.
 380         loop {
 381             if attrs.is_empty() {
 382                 return Some(result);
 383             }
 384
 385             // Handle doc comments.
 386             let (doc_comment_len, doc_comment_str) =
 387                 rewrite_initial_doc_comments(context, attrs, shape)?;
 388             if doc_comment_len > 0 {
 389                 let doc_comment_str = doc_comment_str.expect("doc comments, but no result");
 390                 result.push_str(&doc_comment_str);
 391
 392                 let missing_span = attrs
 393                     .get(doc_comment_len)
 394                     .map(|next| mk_sp(attrs[doc_comment_len - 1].span.hi(), next.span.lo()));
 395                 if let Some(missing_span) = missing_span {
 396                     let snippet = context.snippet(missing_span);
 397                     let (mla, mlb) = has_newlines_before_after_comment(snippet);
 398                     let comment = ::comment::recover_missing_comment_in_span(
 399                         missing_span,
 400                         shape.with_max_width(context.config),
 401                         context,
 402                         0,
 403                     )?;
 404                     let comment = if comment.is_empty() {
 405                         format!("\n{}", mlb)
 406                     } else {
 407                         format!("{}{}\n{}", mla, comment, mlb)
 408                     };
 409                     result.push_str(&comment);
 410                     result.push_str(&shape.indent.to_string(context.config));
 411                 }
 412
 413                 attrs = &attrs[doc_comment_len..];
 414
 415                 continue;
 416             }
 417
 418             // Handle derives if we will merge them.
 419             if context.config.merge_derives() && is_derive(&attrs[0]) {
 420                 let derives = take_while_with_pred(context, attrs, is_derive);
 421                 let mut derive_spans = vec![];
 422                 for derive in derives {
 423                     derive_spans.append(&mut get_derive_spans(derive)?);
 424                 }
 425                 let derive_str =
 426                     format_derive(&derive_spans, attr_prefix(&attrs[0]), shape, context)?;
 427                 result.push_str(&derive_str);
 428
 429                 let missing_span = attrs
 430                     .get(derives.len())
 431                     .map(|next| mk_sp(attrs[derives.len() - 1].span.hi(), next.span.lo()));
 432                 if let Some(missing_span) = missing_span {
 433                     let comment = ::comment::recover_missing_comment_in_span(
 434                         missing_span,
 435                         shape.with_max_width(context.config),
 436                         context,
 437                         0,
 438                     )?;
 439                     result.push_str(&comment);
 440                     if let Some(next) = attrs.get(derives.len()) {
 441                         if next.is_sugared_doc {
 442                             let snippet = context.snippet(missing_span);
 443                             let (_, mlb) = has_newlines_before_after_comment(snippet);
 444                             result.push_str(&mlb);
 445                         }
 446                     }
 447                     result.push('\n');
 448                     result.push_str(&shape.indent.to_string(context.config));
 449                 }
 450
 451                 attrs = &attrs[derives.len()..];
 452
 453                 continue;
 454             }
 455
 456             // If we get here, then we have a regular attribute, just handle one
 457             // at a time.
 458
 459             let formatted_attr = attrs[0].rewrite(context, shape)?;
 460             result.push_str(&formatted_attr);
 461
 462             let missing_span = attrs
 463                 .get(1)
 464                 .map(|next| mk_sp(attrs[0].span.hi(), next.span.lo()));
 465             if let Some(missing_span) = missing_span {
 466                 let comment = ::comment::recover_missing_comment_in_span(
 467                     missing_span,
 468                     shape.with_max_width(context.config),
 469                     context,
 470                     0,
 471                 )?;
 472                 result.push_str(&comment);
 473                 if let Some(next) = attrs.get(1) {
 474                     if next.is_sugared_doc {
 475                         let snippet = context.snippet(missing_span);
 476                         let (_, mlb) = has_newlines_before_after_comment(snippet);
 477                         result.push_str(&mlb);
 478                     }
 479                 }
 480                 result.push('\n');
 481                 result.push_str(&shape.indent.to_string(context.config));
 482             }
 483
 484             attrs = &attrs[1..];
 485         }
 486     }
 487 }
 488
 489 fn attr_prefix(attr: &ast::Attribute) -> &'static str {
 490     match attr.style {
 491         ast::AttrStyle::Inner => "#!",
 492         ast::AttrStyle::Outer => "#",
 493     }
 494 }