Merge pull request #2083 from topecongiro/issue-2082

Use a correct span for fn with pub(crate) syntax

Author: nrc

src/items.rs

@@ -1771,12 +1771,18 @@ fn rewrite_fn_base(
         result.push(' ')
     }
 
+    // Skip `pub(crate)`.
+    let lo_after_visibility = if let ast::Visibility::Crate(s) = fn_sig.visibility {
+        context.codemap.span_after(mk_sp(s.hi(), span.hi()), ")")
+    } else {
+        span.lo()
+    };
     // A conservative estimation, to goal is to be over all parens in generics
     let args_start = fn_sig
         .generics
         .ty_params
         .last()
-        .map_or(span.lo(), |tp| end_typaram(tp));
+        .map_or(lo_after_visibility, |tp| end_typaram(tp));
     let args_end = if fd.inputs.is_empty() {
         context
             .codemap

tests/source/fn-simple.rs

@@ -59,3 +59,6 @@ mod foo {
         foo();
     }
 }
+
+// #2082
+pub(crate) fn init() {}

tests/target/fn-simple.rs

@@ -101,3 +101,6 @@ mod foo {
         foo();
     }
 }
+
+// #2082
+pub(crate) fn init() {}