Support changing state type in #[debug_handler] (#1271)

* support setting body type for #[debug_handler]

* Use lookahead1 to give better errors and detect duplicate arguments

* fix docs link

axum-macros/tests/debug_handler/fail/duplicate_args.rs

@@ -0,0 +1,9 @@
+use axum_macros::debug_handler;
+
+#[debug_handler(body = BoxBody, body = BoxBody)]
+async fn handler() {}
+
+#[debug_handler(state = (), state = ())]
+async fn handler_2() {}
+
+fn main() {}

axum-macros/tests/debug_handler/fail/duplicate_args.stderr

@@ -0,0 +1,11 @@
+error: `body` specified more than once
+ --> tests/debug_handler/fail/duplicate_args.rs:3:33
+  |
+3 | #[debug_handler(body = BoxBody, body = BoxBody)]
+  |                                 ^^^^
+
+error: `state` specified more than once
+ --> tests/debug_handler/fail/duplicate_args.rs:6:29
+  |
+6 | #[debug_handler(state = (), state = ())]
+  |                             ^^^^^

axum-macros/tests/debug_handler/fail/invalid_attrs.stderr

@@ -1,4 +1,4 @@
-error: unknown argument
+error: expected `body` or `state`
  --> tests/debug_handler/fail/invalid_attrs.rs:3:17
   |
 3 | #[debug_handler(foo)]

axum-macros/tests/debug_handler/pass/infer_state.rs

@@ -0,0 +1,31 @@
+use axum_macros::debug_handler;
+use axum::extract::State;
+
+#[debug_handler]
+async fn handler(_: State<AppState>) {}
+
+#[debug_handler]
+async fn handler_2(_: axum::extract::State<AppState>) {}
+
+#[debug_handler]
+async fn handler_3(
+    _: axum::extract::State<AppState>,
+    _: axum::extract::State<AppState>,
+) {}
+
+#[debug_handler]
+async fn handler_4(
+    _: State<AppState>,
+    _: State<AppState>,
+) {}
+
+#[debug_handler]
+async fn handler_5(
+    _: axum::extract::State<AppState>,
+    _: State<AppState>,
+) {}
+
+#[derive(Clone)]
+struct AppState;
+
+fn main() {}

axum-macros/tests/debug_handler/pass/set_state.rs

@@ -0,0 +1,27 @@
+use axum_macros::debug_handler;
+use axum::extract::{FromRef, FromRequest, RequestParts};
+use axum::async_trait;
+
+#[debug_handler(state = AppState)]
+async fn handler(_: A) {}
+
+#[derive(Clone)]
+struct AppState;
+
+struct A;
+
+#[async_trait]
+impl<S, B> FromRequest<S, B> for A
+where
+    B: Send,
+    S: Send + Sync,
+    AppState: FromRef<S>,
+{
+    type Rejection = ();
+
+    async fn from_request(_req: &mut RequestParts<S, B>) -> Result<Self, Self::Rejection> {
+        unimplemented!()
+    }
+}
+
+fn main() {}

axum-macros/tests/debug_handler/pass/state_and_body.rs

@@ -0,0 +1,10 @@
+use axum_macros::debug_handler;
+use axum::{body::BoxBody, extract::State, http::Request};
+
+#[debug_handler(state = AppState, body = BoxBody)]
+async fn handler(_: State<AppState>, _: Request<BoxBody>) {}
+
+#[derive(Clone)]
+struct AppState;
+
+fn main() {}