clippy_utils/src/source.rs

   1 //! Utils for extracting, inspecting or transforming source code
   2
   3 #![allow(clippy::module_name_repetitions)]
   4
   5 use crate::line_span;
   6 use rustc_errors::Applicability;
   7 use rustc_hir::{Expr, ExprKind};
   8 use rustc_lint::{LateContext, LintContext};
   9 use rustc_span::hygiene;
  10 use rustc_span::{BytePos, Pos, Span, SyntaxContext};
  11 use std::borrow::Cow;
  12
  13 /// Like `snippet_block`, but add braces if the expr is not an `ExprKind::Block`.
  14 /// Also takes an `Option<String>` which can be put inside the braces.
  15 pub fn expr_block<'a, T: LintContext>(
  16     cx: &T,
  17     expr: &Expr<'_>,
  18     option: Option<String>,
  19     default: &'a str,
  20     indent_relative_to: Option<Span>,
  21 ) -> Cow<'a, str> {
  22     let code = snippet_block(cx, expr.span, default, indent_relative_to);
  23     let string = option.unwrap_or_default();
  24     if expr.span.from_expansion() {
  25         Cow::Owned(format!("{{ {} }}", snippet_with_macro_callsite(cx, expr.span, default)))
  26     } else if let ExprKind::Block(_, _) = expr.kind {
  27         Cow::Owned(format!("{}{}", code, string))
  28     } else if string.is_empty() {
  29         Cow::Owned(format!("{{ {} }}", code))
  30     } else {
  31         Cow::Owned(format!("{{\n{};\n{}\n}}", code, string))
  32     }
  33 }
  34
  35 /// Returns a new Span that extends the original Span to the first non-whitespace char of the first
  36 /// line.
  37 ///
  38 /// ```rust,ignore
  39 ///     let x = ();
  40 /// //          ^^
  41 /// // will be converted to
  42 ///     let x = ();
  43 /// //  ^^^^^^^^^^
  44 /// ```
  45 pub fn first_line_of_span<T: LintContext>(cx: &T, span: Span) -> Span {
  46     first_char_in_first_line(cx, span).map_or(span, |first_char_pos| span.with_lo(first_char_pos))
  47 }
  48
  49 fn first_char_in_first_line<T: LintContext>(cx: &T, span: Span) -> Option<BytePos> {
  50     let line_span = line_span(cx, span);
  51     snippet_opt(cx, line_span).and_then(|snip| {
  52         snip.find(|c: char| !c.is_whitespace())
  53             .map(|pos| line_span.lo() + BytePos::from_usize(pos))
  54     })
  55 }
  56
  57 /// Returns the indentation of the line of a span
  58 ///
  59 /// ```rust,ignore
  60 /// let x = ();
  61 /// //      ^^ -- will return 0
  62 ///     let x = ();
  63 /// //          ^^ -- will return 4
  64 /// ```
  65 pub fn indent_of<T: LintContext>(cx: &T, span: Span) -> Option<usize> {
  66     snippet_opt(cx, line_span(cx, span)).and_then(|snip| snip.find(|c: char| !c.is_whitespace()))
  67 }
  68
  69 /// Gets a snippet of the indentation of the line of a span
  70 pub fn snippet_indent<T: LintContext>(cx: &T, span: Span) -> Option<String> {
  71     snippet_opt(cx, line_span(cx, span)).map(|mut s| {
  72         let len = s.len() - s.trim_start().len();
  73         s.truncate(len);
  74         s
  75     })
  76 }
  77
  78 // If the snippet is empty, it's an attribute that was inserted during macro
  79 // expansion and we want to ignore those, because they could come from external
  80 // sources that the user has no control over.
  81 // For some reason these attributes don't have any expansion info on them, so
  82 // we have to check it this way until there is a better way.
  83 pub fn is_present_in_source<T: LintContext>(cx: &T, span: Span) -> bool {
  84     if let Some(snippet) = snippet_opt(cx, span) {
  85         if snippet.is_empty() {
  86             return false;
  87         }
  88     }
  89     true
  90 }
  91
  92 /// Returns the positon just before rarrow
  93 ///
  94 /// ```rust,ignore
  95 /// fn into(self) -> () {}
  96 ///              ^
  97 /// // in case of unformatted code
  98 /// fn into2(self)-> () {}
  99 ///               ^
 100 /// fn into3(self)   -> () {}
 101 ///               ^
 102 /// ```
 103 pub fn position_before_rarrow(s: &str) -> Option<usize> {
 104     s.rfind("->").map(|rpos| {
 105         let mut rpos = rpos;
 106         let chars: Vec<char> = s.chars().collect();
 107         while rpos > 1 {
 108             if let Some(c) = chars.get(rpos - 1) {
 109                 if c.is_whitespace() {
 110                     rpos -= 1;
 111                     continue;
 112                 }
 113             }
 114             break;
 115         }
 116         rpos
 117     })
 118 }
 119
 120 /// Reindent a multiline string with possibility of ignoring the first line.
 121 #[allow(clippy::needless_pass_by_value)]
 122 pub fn reindent_multiline(s: Cow<'_, str>, ignore_first: bool, indent: Option<usize>) -> Cow<'_, str> {
 123     let s_space = reindent_multiline_inner(&s, ignore_first, indent, ' ');
 124     let s_tab = reindent_multiline_inner(&s_space, ignore_first, indent, '\t');
 125     reindent_multiline_inner(&s_tab, ignore_first, indent, ' ').into()
 126 }
 127
 128 fn reindent_multiline_inner(s: &str, ignore_first: bool, indent: Option<usize>, ch: char) -> String {
 129     let x = s
 130         .lines()
 131         .skip(ignore_first as usize)
 132         .filter_map(|l| {
 133             if l.is_empty() {
 134                 None
 135             } else {
 136                 // ignore empty lines
 137                 Some(l.char_indices().find(|&(_, x)| x != ch).unwrap_or((l.len(), ch)).0)
 138             }
 139         })
 140         .min()
 141         .unwrap_or(0);
 142     let indent = indent.unwrap_or(0);
 143     s.lines()
 144         .enumerate()
 145         .map(|(i, l)| {
 146             if (ignore_first && i == 0) || l.is_empty() {
 147                 l.to_owned()
 148             } else if x > indent {
 149                 l.split_at(x - indent).1.to_owned()
 150             } else {
 151                 " ".repeat(indent - x) + l
 152             }
 153         })
 154         .collect::<Vec<String>>()
 155         .join("\n")
 156 }
 157
 158 /// Converts a span to a code snippet if available, otherwise use default.
 159 ///
 160 /// This is useful if you want to provide suggestions for your lint or more generally, if you want
 161 /// to convert a given `Span` to a `str`.
 162 ///
 163 /// # Example
 164 /// ```rust,ignore
 165 /// snippet(cx, expr.span, "..")
 166 /// ```
 167 pub fn snippet<'a, T: LintContext>(cx: &T, span: Span, default: &'a str) -> Cow<'a, str> {
 168     snippet_opt(cx, span).map_or_else(|| Cow::Borrowed(default), From::from)
 169 }
 170
 171 /// Same as `snippet`, but it adapts the applicability level by following rules:
 172 ///
 173 /// - Applicability level `Unspecified` will never be changed.
 174 /// - If the span is inside a macro, change the applicability level to `MaybeIncorrect`.
 175 /// - If the default value is used and the applicability level is `MachineApplicable`, change it to
 176 /// `HasPlaceholders`
 177 pub fn snippet_with_applicability<'a, T: LintContext>(
 178     cx: &T,
 179     span: Span,
 180     default: &'a str,
 181     applicability: &mut Applicability,
 182 ) -> Cow<'a, str> {
 183     if *applicability != Applicability::Unspecified && span.from_expansion() {
 184         *applicability = Applicability::MaybeIncorrect;
 185     }
 186     snippet_opt(cx, span).map_or_else(
 187         || {
 188             if *applicability == Applicability::MachineApplicable {
 189                 *applicability = Applicability::HasPlaceholders;
 190             }
 191             Cow::Borrowed(default)
 192         },
 193         From::from,
 194     )
 195 }
 196
 197 /// Same as `snippet`, but should only be used when it's clear that the input span is
 198 /// not a macro argument.
 199 pub fn snippet_with_macro_callsite<'a, T: LintContext>(cx: &T, span: Span, default: &'a str) -> Cow<'a, str> {
 200     snippet(cx, span.source_callsite(), default)
 201 }
 202
 203 /// Converts a span to a code snippet. Returns `None` if not available.
 204 pub fn snippet_opt<T: LintContext>(cx: &T, span: Span) -> Option<String> {
 205     cx.sess().source_map().span_to_snippet(span).ok()
 206 }
 207
 208 /// Converts a span (from a block) to a code snippet if available, otherwise use default.
 209 ///
 210 /// This trims the code of indentation, except for the first line. Use it for blocks or block-like
 211 /// things which need to be printed as such.
 212 ///
 213 /// The `indent_relative_to` arg can be used, to provide a span, where the indentation of the
 214 /// resulting snippet of the given span.
 215 ///
 216 /// # Example
 217 ///
 218 /// ```rust,ignore
 219 /// snippet_block(cx, block.span, "..", None)
 220 /// // where, `block` is the block of the if expr
 221 ///     if x {
 222 ///         y;
 223 ///     }
 224 /// // will return the snippet
 225 /// {
 226 ///     y;
 227 /// }
 228 /// ```
 229 ///
 230 /// ```rust,ignore
 231 /// snippet_block(cx, block.span, "..", Some(if_expr.span))
 232 /// // where, `block` is the block of the if expr
 233 ///     if x {
 234 ///         y;
 235 ///     }
 236 /// // will return the snippet
 237 /// {
 238 ///         y;
 239 ///     } // aligned with `if`
 240 /// ```
 241 /// Note that the first line of the snippet always has 0 indentation.
 242 pub fn snippet_block<'a, T: LintContext>(
 243     cx: &T,
 244     span: Span,
 245     default: &'a str,
 246     indent_relative_to: Option<Span>,
 247 ) -> Cow<'a, str> {
 248     let snip = snippet(cx, span, default);
 249     let indent = indent_relative_to.and_then(|s| indent_of(cx, s));
 250     reindent_multiline(snip, true, indent)
 251 }
 252
 253 /// Same as `snippet_block`, but adapts the applicability level by the rules of
 254 /// `snippet_with_applicability`.
 255 pub fn snippet_block_with_applicability<'a, T: LintContext>(
 256     cx: &T,
 257     span: Span,
 258     default: &'a str,
 259     indent_relative_to: Option<Span>,
 260     applicability: &mut Applicability,
 261 ) -> Cow<'a, str> {
 262     let snip = snippet_with_applicability(cx, span, default, applicability);
 263     let indent = indent_relative_to.and_then(|s| indent_of(cx, s));
 264     reindent_multiline(snip, true, indent)
 265 }
 266
 267 /// Same as `snippet_with_applicability`, but first walks the span up to the given context. This
 268 /// will result in the macro call, rather then the expansion, if the span is from a child context.
 269 /// If the span is not from a child context, it will be used directly instead.
 270 ///
 271 /// e.g. Given the expression `&vec![]`, getting a snippet from the span for `vec![]` as a HIR node
 272 /// would result in `box []`. If given the context of the address of expression, this function will
 273 /// correctly get a snippet of `vec![]`.
 274 ///
 275 /// This will also return whether or not the snippet is a macro call.
 276 pub fn snippet_with_context(
 277     cx: &LateContext<'_>,
 278     span: Span,
 279     outer: SyntaxContext,
 280     default: &'a str,
 281     applicability: &mut Applicability,
 282 ) -> (Cow<'a, str>, bool) {
 283     let (span, is_macro_call) = walk_span_to_context(span, outer).map_or_else(
 284         || {
 285             // The span is from a macro argument, and the outer context is the macro using the argument
 286             if *applicability != Applicability::Unspecified {
 287                 *applicability = Applicability::MaybeIncorrect;
 288             }
 289             // TODO: get the argument span.
 290             (span, false)
 291         },
 292         |outer_span| (outer_span, span.ctxt() != outer),
 293     );
 294
 295     (
 296         snippet_with_applicability(cx, span, default, applicability),
 297         is_macro_call,
 298     )
 299 }
 300
 301 /// Walks the span up to the target context, thereby returning the macro call site if the span is
 302 /// inside a macro expansion, or the original span if it is not. Note this will return `None` in the
 303 /// case of the span being in a macro expansion, but the target context is from expanding a macro
 304 /// argument.
 305 pub fn walk_span_to_context(span: Span, outer: SyntaxContext) -> Option<Span> {
 306     let outer_span = hygiene::walk_chain(span, outer);
 307     (outer_span.ctxt() == outer).then(|| outer_span)
 308 }
 309
 310 /// Removes block comments from the given `Vec` of lines.
 311 ///
 312 /// # Examples
 313 ///
 314 /// ```rust,ignore
 315 /// without_block_comments(vec!["/*", "foo", "*/"]);
 316 /// // => vec![]
 317 ///
 318 /// without_block_comments(vec!["bar", "/*", "foo", "*/"]);
 319 /// // => vec!["bar"]
 320 /// ```
 321 pub fn without_block_comments(lines: Vec<&str>) -> Vec<&str> {
 322     let mut without = vec![];
 323
 324     let mut nest_level = 0;
 325
 326     for line in lines {
 327         if line.contains("/*") {
 328             nest_level += 1;
 329             continue;
 330         } else if line.contains("*/") {
 331             nest_level -= 1;
 332             continue;
 333         }
 334
 335         if nest_level == 0 {
 336             without.push(line);
 337         }
 338     }
 339
 340     without
 341 }
 342
 343 #[cfg(test)]
 344 mod test {
 345     use super::{reindent_multiline, without_block_comments};
 346
 347     #[test]
 348     fn test_reindent_multiline_single_line() {
 349         assert_eq!("", reindent_multiline("".into(), false, None));
 350         assert_eq!("...", reindent_multiline("...".into(), false, None));
 351         assert_eq!("...", reindent_multiline("    ...".into(), false, None));
 352         assert_eq!("...", reindent_multiline("\t...".into(), false, None));
 353         assert_eq!("...", reindent_multiline("\t\t...".into(), false, None));
 354     }
 355
 356     #[test]
 357     #[rustfmt::skip]
 358     fn test_reindent_multiline_block() {
 359         assert_eq!("\
 360     if x {
 361         y
 362     } else {
 363         z
 364     }", reindent_multiline("    if x {
 365             y
 366         } else {
 367             z
 368         }".into(), false, None));
 369         assert_eq!("\
 370     if x {
 371     \ty
 372     } else {
 373     \tz
 374     }", reindent_multiline("    if x {
 375         \ty
 376         } else {
 377         \tz
 378         }".into(), false, None));
 379     }
 380
 381     #[test]
 382     #[rustfmt::skip]
 383     fn test_reindent_multiline_empty_line() {
 384         assert_eq!("\
 385     if x {
 386         y
 387
 388     } else {
 389         z
 390     }", reindent_multiline("    if x {
 391             y
 392
 393         } else {
 394             z
 395         }".into(), false, None));
 396     }
 397
 398     #[test]
 399     #[rustfmt::skip]
 400     fn test_reindent_multiline_lines_deeper() {
 401         assert_eq!("\
 402         if x {
 403             y
 404         } else {
 405             z
 406         }", reindent_multiline("\
 407     if x {
 408         y
 409     } else {
 410         z
 411     }".into(), true, Some(8)));
 412     }
 413
 414     #[test]
 415     fn test_without_block_comments_lines_without_block_comments() {
 416         let result = without_block_comments(vec!["/*", "", "*/"]);
 417         println!("result: {:?}", result);
 418         assert!(result.is_empty());
 419
 420         let result = without_block_comments(vec!["", "/*", "", "*/", "#[crate_type = \"lib\"]", "/*", "", "*/", ""]);
 421         assert_eq!(result, vec!["", "#[crate_type = \"lib\"]", ""]);
 422
 423         let result = without_block_comments(vec!["/* rust", "", "*/"]);
 424         assert!(result.is_empty());
 425
 426         let result = without_block_comments(vec!["/* one-line comment */"]);
 427         assert!(result.is_empty());
 428
 429         let result = without_block_comments(vec!["/* nested", "/* multi-line", "comment", "*/", "test", "*/"]);
 430         assert!(result.is_empty());
 431
 432         let result = without_block_comments(vec!["/* nested /* inline /* comment */ test */ */"]);
 433         assert!(result.is_empty());
 434
 435         let result = without_block_comments(vec!["foo", "bar", "baz"]);
 436         assert_eq!(result, vec!["foo", "bar", "baz"]);
 437     }
 438 }