about summary refs log tree commit diff
diff options
context:
space:
mode:
authorvarkor <github@varkor.com>2019-04-18 18:44:55 +0100
committervarkor <github@varkor.com>2019-04-20 22:03:39 +0100
commitc0ad4b0a96d039ef113a3ad5e422058e6f8fd2d7 (patch)
tree665c145503a37daeb9424142e430f5d72867896e
parent4530c528ba4eadd7623813316d75bd4b97012b6c (diff)
downloadrust-c0ad4b0a96d039ef113a3ad5e422058e6f8fd2d7.tar.gz
rust-c0ad4b0a96d039ef113a3ad5e422058e6f8fd2d7.zip
Add a `header` method to `FnKind`
-rw-r--r--src/librustc/hir/intravisit.rs8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/librustc/hir/intravisit.rs b/src/librustc/hir/intravisit.rs
index c2265eeb30d..007eaef74a7 100644
--- a/src/librustc/hir/intravisit.rs
+++ b/src/librustc/hir/intravisit.rs
@@ -57,6 +57,14 @@ impl<'a> FnKind<'a> {
             FnKind::Closure(attrs) => attrs,
         }
     }
+
+    pub fn header(&self) -> Option<FnHeader> {
+        match *self {
+            FnKind::ItemFn(_, _, header, _, _) => Some(header),
+            FnKind::Method(_, sig, _, _) => Some(sig.header),
+            FnKind::Closure(_) => None,
+        }
+    }
 }
 
 /// Specifies what nested things a visitor wants to visit. The most