1 use crate::utils::span_lint;
2 use itertools::Itertools;
5 use rustc::impl_lint_pass;
6 use rustc::lint::{LateContext, LateLintPass, LintArray, LintPass};
7 use rustc_data_structures::fx::FxHashSet;
8 use rustc_session::declare_tool_lint;
10 use syntax::ast::{AttrKind, Attribute};
11 use syntax::source_map::{BytePos, Span};
15 declare_clippy_lint! {
16 /// **What it does:** Checks for the presence of `_`, `::` or camel-case words
17 /// outside ticks in documentation.
19 /// **Why is this bad?** *Rustdoc* supports markdown formatting, `_`, `::` and
20 /// camel-case probably indicates some code which should be included between
21 /// ticks. `_` can also be used for emphasis in markdown, this lint tries to
24 /// **Known problems:** Lots of bad docs won’t be fixed, what the lint checks
25 /// for is limited, and there are still false positives.
29 /// /// Do something with the foo_bar parameter. See also
30 /// /// that::other::module::foo.
31 /// // ^ `foo_bar` and `that::other::module::foo` should be ticked.
32 /// fn doit(foo_bar: usize) {}
36 "presence of `_`, `::` or camel-case outside backticks in documentation"
39 declare_clippy_lint! {
40 /// **What it does:** Checks for the doc comments of publicly visible
41 /// unsafe functions and warns if there is no `# Safety` section.
43 /// **Why is this bad?** Unsafe functions should document their safety
44 /// preconditions, so that users can be sure they are using them safely.
46 /// **Known problems:** None.
50 ///# type Universe = ();
51 /// /// This function should really be documented
52 /// pub unsafe fn start_apocalypse(u: &mut Universe) {
57 /// At least write a line about safety:
60 ///# type Universe = ();
63 /// /// This function should not be called before the horsemen are ready.
64 /// pub unsafe fn start_apocalypse(u: &mut Universe) {
68 pub MISSING_SAFETY_DOC,
70 "`pub unsafe fn` without `# Safety` docs"
73 declare_clippy_lint! {
74 /// **What it does:** Checks for `fn main() { .. }` in doctests
76 /// **Why is this bad?** The test can be shorter (and likely more readable)
77 /// if the `fn main()` is left implicit.
79 /// **Known problems:** None.
83 /// /// An example of a doctest with a `main()` function
89 /// /// // this needs not be in an `fn`
92 /// fn needless_main() {
96 pub NEEDLESS_DOCTEST_MAIN,
98 "presence of `fn main() {` in code examples"
101 #[allow(clippy::module_name_repetitions)]
103 pub struct DocMarkdown {
104 valid_idents: FxHashSet<String>,
109 pub fn new(valid_idents: FxHashSet<String>) -> Self {
112 in_trait_impl: false,
117 impl_lint_pass!(DocMarkdown => [DOC_MARKDOWN, MISSING_SAFETY_DOC, NEEDLESS_DOCTEST_MAIN]);
119 impl<'a, 'tcx> LateLintPass<'a, 'tcx> for DocMarkdown {
120 fn check_crate(&mut self, cx: &LateContext<'a, 'tcx>, krate: &'tcx hir::Crate) {
121 check_attrs(cx, &self.valid_idents, &krate.attrs);
124 fn check_item(&mut self, cx: &LateContext<'a, 'tcx>, item: &'tcx hir::Item) {
125 if check_attrs(cx, &self.valid_idents, &item.attrs) {
130 hir::ItemKind::Fn(ref sig, ..) => {
131 if cx.access_levels.is_exported(item.hir_id) && sig.header.unsafety == hir::Unsafety::Unsafe {
136 "unsafe function's docs miss `# Safety` section",
140 hir::ItemKind::Impl(_, _, _, _, ref trait_ref, ..) => {
141 self.in_trait_impl = trait_ref.is_some();
147 fn check_item_post(&mut self, _cx: &LateContext<'a, 'tcx>, item: &'tcx hir::Item) {
148 if let hir::ItemKind::Impl(..) = item.kind {
149 self.in_trait_impl = false;
153 fn check_trait_item(&mut self, cx: &LateContext<'a, 'tcx>, item: &'tcx hir::TraitItem) {
154 if check_attrs(cx, &self.valid_idents, &item.attrs) {
158 if let hir::TraitItemKind::Method(ref sig, ..) = item.kind {
159 if cx.access_levels.is_exported(item.hir_id) && sig.header.unsafety == hir::Unsafety::Unsafe {
164 "unsafe function's docs miss `# Safety` section",
170 fn check_impl_item(&mut self, cx: &LateContext<'a, 'tcx>, item: &'tcx hir::ImplItem) {
171 if check_attrs(cx, &self.valid_idents, &item.attrs) || self.in_trait_impl {
175 if let hir::ImplItemKind::Method(ref sig, ..) = item.kind {
176 if cx.access_levels.is_exported(item.hir_id) && sig.header.unsafety == hir::Unsafety::Unsafe {
181 "unsafe function's docs miss `# Safety` section",
188 /// Cleanup documentation decoration (`///` and such).
190 /// We can't use `syntax::attr::AttributeMethods::with_desugared_doc` or
191 /// `syntax::parse::lexer::comments::strip_doc_comment_decoration` because we
192 /// need to keep track of
193 /// the spans but this function is inspired from the later.
194 #[allow(clippy::cast_possible_truncation)]
196 pub fn strip_doc_comment_decoration(comment: &str, span: Span) -> (String, Vec<(usize, Span)>) {
197 // one-line comments lose their prefix
198 const ONELINERS: &[&str] = &["///!", "///", "//!", "//"];
199 for prefix in ONELINERS {
200 if comment.starts_with(*prefix) {
201 let doc = &comment[prefix.len()..];
202 let mut doc = doc.to_owned();
206 vec![(doc.len(), span.with_lo(span.lo() + BytePos(prefix.len() as u32)))],
211 if comment.starts_with("/*") {
212 let doc = &comment[3..comment.len() - 2];
213 let mut sizes = vec![];
214 let mut contains_initial_stars = false;
215 for line in doc.lines() {
216 let offset = line.as_ptr() as usize - comment.as_ptr() as usize;
217 debug_assert_eq!(offset as u32 as usize, offset);
218 contains_initial_stars |= line.trim_start().starts_with('*');
219 // +1 for the newline
220 sizes.push((line.len() + 1, span.with_lo(span.lo() + BytePos(offset as u32))));
222 if !contains_initial_stars {
223 return (doc.to_string(), sizes);
225 // remove the initial '*'s if any
226 let mut no_stars = String::with_capacity(doc.len());
227 for line in doc.lines() {
228 let mut chars = line.chars();
229 while let Some(c) = chars.next() {
230 if c.is_whitespace() {
233 no_stars.push(if c == '*' { ' ' } else { c });
237 no_stars.push_str(chars.as_str());
240 return (no_stars, sizes);
243 panic!("not a doc-comment: {}", comment);
246 pub fn check_attrs<'a>(cx: &LateContext<'_, '_>, valid_idents: &FxHashSet<String>, attrs: &'a [Attribute]) -> bool {
247 let mut doc = String::new();
248 let mut spans = vec![];
251 if let AttrKind::DocComment(ref comment) = attr.kind {
252 let comment = comment.to_string();
253 let (comment, current_spans) = strip_doc_comment_decoration(&comment, attr.span);
254 spans.extend_from_slice(¤t_spans);
255 doc.push_str(&comment);
256 } else if attr.check_name(sym!(doc)) {
257 // ignore mix of sugared and non-sugared doc
258 return true; // don't trigger the safety check
263 for &mut (ref mut offset, _) in &mut spans {
264 let offset_copy = *offset;
266 current += offset_copy;
273 let parser = pulldown_cmark::Parser::new(&doc).into_offset_iter();
274 // Iterate over all `Events` and combine consecutive events into one
275 let events = parser.coalesce(|previous, current| {
276 use pulldown_cmark::Event::*;
278 let previous_range = previous.1;
279 let current_range = current.1;
281 match (previous.0, current.0) {
282 (Text(previous), Text(current)) => {
283 let mut previous = previous.to_string();
284 previous.push_str(¤t);
285 Ok((Text(previous.into()), previous_range))
287 (previous, current) => Err(((previous, previous_range), (current, current_range))),
290 check_doc(cx, valid_idents, events, &spans)
293 fn check_doc<'a, Events: Iterator<Item = (pulldown_cmark::Event<'a>, Range<usize>)>>(
294 cx: &LateContext<'_, '_>,
295 valid_idents: &FxHashSet<String>,
297 spans: &[(usize, Span)],
299 // true if a safety header was found
300 use pulldown_cmark::Event::*;
301 use pulldown_cmark::Tag::*;
303 let mut safety_header = false;
304 let mut in_code = false;
305 let mut in_link = None;
306 let mut in_heading = false;
308 for (event, range) in events {
310 Start(CodeBlock(_)) => in_code = true,
311 End(CodeBlock(_)) => in_code = false,
312 Start(Link(_, url, _)) => in_link = Some(url),
313 End(Link(..)) => in_link = None,
314 Start(Heading(_)) => in_heading = true,
315 End(Heading(_)) => in_heading = false,
316 Start(_tag) | End(_tag) => (), // We don't care about other tags
317 Html(_html) => (), // HTML is weird, just ignore it
318 SoftBreak | HardBreak | TaskListMarker(_) | Code(_) | Rule => (),
319 FootnoteReference(text) | Text(text) => {
320 if Some(&text) == in_link.as_ref() {
321 // Probably a link of the form `<http://example.com>`
322 // Which are represented as a link to "http://example.com" with
323 // text "http://example.com" by pulldown-cmark
326 safety_header |= in_heading && text.trim() == "Safety";
327 let index = match spans.binary_search_by(|c| c.0.cmp(&range.start)) {
331 let (begin, span) = spans[index];
333 check_code(cx, &text, span);
335 // Adjust for the beginning of the current `Event`
336 let span = span.with_lo(span.lo() + BytePos::from_usize(range.start - begin));
338 check_text(cx, valid_idents, &text, span);
346 fn check_code(cx: &LateContext<'_, '_>, text: &str, span: Span) {
347 if text.contains("fn main() {") && !(text.contains("static") || text.contains("fn main() {}")) {
348 span_lint(cx, NEEDLESS_DOCTEST_MAIN, span, "needless `fn main` in doctest");
352 fn check_text(cx: &LateContext<'_, '_>, valid_idents: &FxHashSet<String>, text: &str, span: Span) {
353 for word in text.split(|c: char| c.is_whitespace() || c == '\'') {
354 // Trim punctuation as in `some comment (see foo::bar).`
356 // Or even as in `_foo bar_` which is emphasized.
357 let word = word.trim_matches(|c: char| !c.is_alphanumeric());
359 if valid_idents.contains(word) {
363 // Adjust for the current word
364 let offset = word.as_ptr() as usize - text.as_ptr() as usize;
365 let span = Span::new(
366 span.lo() + BytePos::from_usize(offset),
367 span.lo() + BytePos::from_usize(offset + word.len()),
371 check_word(cx, word, span);
375 fn check_word(cx: &LateContext<'_, '_>, word: &str, span: Span) {
376 /// Checks if a string is camel-case, i.e., contains at least two uppercase
377 /// letters (`Clippy` is ok) and one lower-case letter (`NASA` is ok).
378 /// Plurals are also excluded (`IDs` is ok).
379 fn is_camel_case(s: &str) -> bool {
380 if s.starts_with(|c: char| c.is_digit(10)) {
384 let s = if s.ends_with('s') { &s[..s.len() - 1] } else { s };
386 s.chars().all(char::is_alphanumeric)
387 && s.chars().filter(|&c| c.is_uppercase()).take(2).count() > 1
388 && s.chars().filter(|&c| c.is_lowercase()).take(1).count() > 0
391 fn has_underscore(s: &str) -> bool {
392 s != "_" && !s.contains("\\_") && s.contains('_')
395 fn has_hyphen(s: &str) -> bool {
396 s != "-" && s.contains('-')
399 if let Ok(url) = Url::parse(word) {
400 // try to get around the fact that `foo::bar` parses as a valid URL
401 if !url.cannot_be_a_base() {
406 "you should put bare URLs between `<`/`>` or make a proper Markdown link",
413 // We assume that mixed-case words are not meant to be put inside bacticks. (Issue #2343)
414 if has_underscore(word) && has_hyphen(word) {
418 if has_underscore(word) || word.contains("::") || is_camel_case(word) {
423 &format!("you should put `{}` between ticks in the documentation", word),