From 8593e133b73d2c911f216a21996d05033cf81e7b Mon Sep 17 00:00:00 2001
From: Dominic <git@msrd0.de>
Date: Thu, 30 Apr 2020 16:48:50 +0200
Subject: [PATCH] also detect _state as state argument

---
 gotham_restful_derive/src/method.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/gotham_restful_derive/src/method.rs b/gotham_restful_derive/src/method.rs
index 95e9221..db0b8b8 100644
--- a/gotham_restful_derive/src/method.rs
+++ b/gotham_restful_derive/src/method.rs
@@ -181,7 +181,8 @@ fn interpret_arg_ty(attrs : &[Attribute], name : &str, ty : Type) -> Result<Meth
 		.find(|arg| arg.path.segments.iter().any(|path| &path.ident.to_string() == "rest_arg"))
 		.map(|arg| arg.tokens.to_string());
 	
-	if attr.as_deref() == Some("state") || (attr.is_none() && name == "state")
+	// TODO issue a warning for _state usage once diagnostics become stable
+	if attr.as_deref() == Some("state") || (attr.is_none() && (name == "state" || name == "_state"))
 	{
 		return match ty {
 			Type::Reference(ty) => Ok(if ty.mutability.is_none() { MethodArgumentType::StateRef } else { MethodArgumentType::StateMutRef }),