ajj 0.7.1

use crate::{
    routes::HandlerArgs, types::Response, HandlerCtx, IntoErrorPayload, ResponsePayload, Route,
    RpcRecv, RpcSend,
};
use serde_json::value::RawValue;
use std::{convert::Infallible, future::Future, marker::PhantomData, pin::Pin, task};
use tracing::{trace, Instrument};

/// Hint type for differentiating certain handler impls. See the [`Handler`]
/// trait "Handler argument type inference" section for more information.
#[derive(Debug)]
#[repr(transparent)]
pub struct State<S>(pub S);

/// Hint type for differentiating certain handler impls. See the [`Handler`]
/// trait "Handler argument type inference" section for more information.
#[derive(Debug)]
#[repr(transparent)]
pub struct Params<T>(pub T);

impl<T> From<T> for Params<T> {
    fn from(value: T) -> Self {
        Self(value)
    }
}

/// Marker type used for differentiating certain handler impls.
#[allow(missing_debug_implementations, unreachable_pub)]
pub struct PhantomState<S>(PhantomData<S>);

/// Marker type used for differentiating certain handler impls.
#[allow(missing_debug_implementations, unreachable_pub)]
pub struct PhantomParams<T>(PhantomData<T>);

/// A trait describing handlers for JSON-RPC methods.
///
/// Handlers map some input type `T` to a future that resolve to a
/// [`ResponsePayload`]. The handler may also require some state `S` to operate.
///
/// ### Returning error messages
///
/// When a [`Handler`] returns a `Result`, the error type must implement
/// [`IntoErrorPayload`]. This trait controls the JSON-RPC error code,
/// message, and optional structured data in the response.
///
/// Several common types already implement [`IntoErrorPayload`]:
///
/// - **`String` / `&str`** -- produces code `-32603` with the string as
///   the `message` field.
/// - **`()`** -- produces code `-32603` with the default `"Internal error"`
///   message.
/// - **[`InternalError<T>`]** -- produces code `-32603` with `T` as the
///   `data` field.
/// - **[`ErrorPayload<E>`]** -- passed through as-is.
///
/// ```
/// # use ajj::{IntoErrorPayload, InternalError, ErrorPayload, ResponsePayload};
/// # use std::borrow::Cow;
/// // String errors: the string appears in the `message` field.
/// let handler_a = || async { Err::<(), _>("something went wrong") };
///
/// // Custom IntoErrorPayload impl for full control.
/// # #[derive(Debug, Clone)]
/// # struct MyError;
/// # impl IntoErrorPayload for MyError {
/// #     type ErrData = ();
/// #     fn error_code(&self) -> i64 { 42 }
/// #     fn error_message(&self) -> Cow<'static, str> { "my error".into() }
/// # }
/// let handler_b = || async { Err::<(), MyError>(MyError) };
///
/// // InternalError<T> wraps any RpcSend value as `-32603` error data.
/// let handler_c = || async { Err::<(), InternalError<u32>>(InternalError(42)) };
///
/// // Or return a ResponsePayload directly for complete control.
/// let handler_d = || async {
///     ResponsePayload::<(), ()>::internal_error_message("appears in \"message\"".into())
/// };
/// ```
///
/// [`IntoErrorPayload`]: crate::IntoErrorPayload
/// [`InternalError<T>`]: crate::InternalError
/// [`ErrorPayload<E>`]: crate::ErrorPayload
///
/// ### The [`HandlerCtx`]: tasks and notifications.
///
/// Any handler may accept the [`HandlerCtx`] as its first argument. This
/// context object is used to context associated with the client connection. It
/// can be used to spawn tasks and (when `pubsub` is enabled) send notifications
/// to the client.
///
/// Handlers **SHOULD NOT** use [`tokio::spawn`] or
/// [`tokio::task::spawn_blocking`] directly. Instead, they should use the
/// [`HandlerCtx::spawn`] or [`HandlerCtx::spawn_blocking`] methods. These
/// methods ensure that tasks are associated with the client connection, and
/// are cleaned up promptly when the connection is closed, and that the server's
/// runtime configuration is respected.
///
/// When the task itself requires a context, [`HandlerCtx::spawn_with_ctx`]
/// and [`HandlerCtx::spawn_blocking_with_ctx`] can be used to provide a context
/// to the spawned task. This is a thin wrapper around cloning the context and
/// moving it into the spawned future.
///
/// ```
/// use ajj::{Router, HandlerCtx};
///
/// # fn test_fn() -> Router<()> {
/// Router::new()
///     .route("good citizenship", |ctx: HandlerCtx| async move {
///         // Properly implemented task management. This task will
///         // automatically be cleaned up when the connection is closed.
///         ctx.spawn(async {
///            // do something
///         });
///         Ok::<_, ()>(())
///     })
///     .route("bad citizenship", |ctx: HandlerCtx| async move {
///         // Incorrect task management. Will result in the task running for
///         // some amount of time after the connection is closed.
///         tokio::spawn(async {
///            // do something
///         });
///         Ok::<_, ()>(())
///     })
/// # }
/// ```
///
/// When running on pubsub, handlers can send notifications to the client. This
/// is done by calling [`HandlerCtx::notify`]. Notifications are sent as JSON
/// objects, and are queued for sending to the client. If the client is not
/// reading from the connection, the notification will be queued in a buffer.
/// If the buffer is full, the handler will be backpressured until the buffer
/// has room.
///
/// We recommend that handler tasks `await` on the result of
/// [`HandlerCtx::notify`], to ensure that they are backpressured when the
/// notification buffer is full. If many tasks are attempting to notify the
/// same client, the buffer may fill up, and backpressure the `RouteTask` from
/// reading requests from the connection. This can lead to delays in request
/// processing.
///
///
/// ### Handler argument type inference
///
/// When the following conditions are true, the compiler may fail to infer
/// whether a handler argument is `params` or `state`:
///
/// 1. The handler takes EITHER `params` or `state`.
/// 2. The `S` type of the router would also be a valid `Params` type (i.e. it
///    impls [`DeserializeOwned`] and satisfies the other
///    requirements of [`RpcRecv`]).
/// 3. The argument to the handler matches the `S` type of the router.
///
/// ```compile_fail
/// use ajj::Router;
///
/// async fn ok() -> Result<(), ()> { Ok(()) }
///
/// # fn test_fn() -> Router<()> {
/// // These will fail to infer the `Handler` impl as the argument could be
/// // either `params` or `state`.
/// //
/// // The error will look like this:
/// // cannot infer type for type parameter `T` declared on the method
/// // `route`
/// Router::<u8>::new()
///     // "foo" and "bar" are ambiguous, as the `S` type is `u8`
///     .route("foo", |params: u8| ok())
///     .route("bar", |state: u8| ok())
///     // "baz" is unambiguous, as it has both `params` and `state` arguments
///     .route("baz", |params: u8, state: u8| ok())
///     // this is unambiguous, but a bit ugly.
///     .route("qux", |params: u8, _: u8| ok())
///     .with_state(3u8)
/// # }
/// ```
///
/// In these cases the compiler will find multiple valid impls of the `Handler`
/// trait.
///
/// The easiest way to resolve this is to avoid using the `S` type of the
/// router as the `params` argument to the handler. This will ensure that the
/// compiler can always infer the correct impl of the `Handler` trait. In cases
/// where `Params` and `S` must be the same type, you can use the [`Params`] and
/// [`State`] wrapper structs in your handler arguments to disambiguate the
/// argument type.
///
/// The usual way to fix this is to reorganize route invocations to avoid the
/// ambiguity:
///
/// ```
/// # use ajj::{Router, Params, State};
/// # async fn ok() -> Result<(), ()> { Ok(()) }
/// # fn test_fn() -> Router<()> {
/// // When "foo" is routed, the argument is unambiguous, as the `S` type
/// // is no longer `u8`.
/// Router::<u8>::new()
///     // This was already unambiguous, as having both `params` and `state` is
///     // never ambiguous
///     .route("baz", |params: u8, state: u8| ok())
///
///     // "bar" presents a problem. We'll come back to this in the next
///     // example.
///     // .route("bar", |state: u8| ok())
///
///     .with_state::<()>(3u8)
///
///     // `S` is now `()`, so the argument is unambiguous.
///     .route("foo", |params: u8| ok())
/// # }
/// ```
///
/// However this still leaves the problem of "bar". There is no way to express
/// "bar" unambiguously by reordering method invocations. In this case, you can
/// use the [`Params`] and [`State`] wrapper structs to disambiguate the
/// argument type. This should seem familiar to users of [`axum`].
///
/// ```
/// # use ajj::{Router, Params, State};
/// # async fn ok() -> Result<(), ()> { Ok(()) }
/// # fn test_fn() -> Router<()> {
/// Router::<u8>::new()
///     // This is now unambiguous, as the `Params` wrapper indicates that the
///     // argument is `params`
///     .route("foo", |Params(params): Params<u8>| ok())
///
///     // This is now umabiguous, as the `State` wrapper indicates that the
///     // argument is `state`
///     .route("bar", |State(state): State<u8>| ok())
///
///     // This was already unambiguous, as having both `params` and `state` is
///     // never ambiguous
///     .route("baz", |params: u8, state: u8| ok())
///     .with_state::<()>(3u8)
/// # }
/// ```
///
/// These wrapper structs are available only when necessary, and may not be
/// used when the `Handler` impl is unambiguous. Ambiguous handlers will always
/// result in a compilation error.
///
/// ### Handler return type inference
///
/// Handlers that always succeed or always fail may have trouble with type
/// inference, as they contain an unknown type parameter, which could be
/// anything. Here's an example of code with failed type inference:
///
/// ```compile_fail
/// # use ajj::ResponsePayload;
/// // cannot infer type of the type parameter `T` declared on the enum `Result`
/// let cant_infer_ok = || async { Err(1) };
///
/// // cannot infer type of the type parameter `E` declared on the enum `Result`
/// let cant_infer_err = || async { Ok(2) };
///
/// // cannot infer type of the type parameter `ErrData` declared on the enum `ResponsePayload`
/// let cant_infer_failure = || async { ResponsePayload(Ok(3)) };
///
/// // cannot infer type of the type parameter `ErrData` declared on the enum `ResponsePayload`
/// let cant_infer_success = || async { ResponsePayload::internal_error_with_obj(4) };
/// ```
///
/// If you encounter these sorts of inference errors, you can add turbofish to
/// your handlers' return values like so:
///
/// ```
/// # use ajj::ResponsePayload;
/// // specify the Err on your Ok
/// let handler_a = || async { Ok::<_, ()>(1) };
///
/// // specify the Ok on your Err
/// let handler_b = || async { Err::<(), _>(2) };
///
/// // specify the ErrData on your Success
/// let handler_c = || async { ResponsePayload::<_, ()>(Ok(3)) };
///
/// // specify the Payload on your Failure
/// let handler_d = || async { ResponsePayload::<(), _>::internal_error_with_obj(4) };
/// ```
#[cfg_attr(
    feature = "pubsub",
    doc = "see the [`Listener`] documetnation for more information."
)]
///
/// ## Note on `S`
///
/// The `S` type parameter is "missing" state. It represents the state that the
/// handler needs to operate. This state is passed to the handler when calling
/// [`Handler::call_with_state`].
///
/// ## Blanket Implementations
///
/// This trait is blanket implemented for the following function and closure
/// types, where `Fut` is a [`Future`] returning either
/// `ResponsePayload<T: RpcSend, E: RpcSend>` or
/// `Result<T: RpcSend, E: IntoErrorPayload>`:
///
/// - `async fn()`
/// - `async fn(HandlerCtx) -> Fut`
/// - `async fn(HandlerCtx, Params) -> Fut`
/// - `async fn(HandlerCtx, S) -> Fut`
/// - `async fn(HandlerCtx, Params, S) -> Fut`
/// - `async fn(Params) -> Fut`
/// - `async fn(Params, S) -> Fut`
///
/// ### Implementer's note:
///
/// The generics on the implementation of the `Handler` trait are actually very
/// straightforward, even though they look intimidating.
///
/// The `T` type parameter is a **marker** and never needs be constructed.
/// It exists to differentiate the output impls, so that the trait can be
/// blanket implemented for many different function types. If it were simply
/// `Handler<S>`, there could only be 1 blanket impl.
///
/// However the `T` type must constrain the relevant input types. When
/// implementing `Handler<T, S>` for your own type, it is recommended to do the
/// following:
///
/// ```
/// use ajj::{Handler, HandlerArgs, ResponsePayload, RawValue};
/// use std::{pin::Pin, future::Future};
///
/// /// A marker type to differentiate this handler from other blanket impls.
/// pub struct MyMarker {
///   _sealed: (),
/// }
///
/// #[derive(Clone)]
/// pub struct MyHandler;
///
/// // the `T` type parameter should be a tuple, containing your marker type,
/// // and the components that your handler actually uses. This ensures that
/// // the implementations are always unambiguous.
/// //
/// // e.g.
/// // - (MyMarker, )
/// // - (MyMarker, HandlerArgs)
/// // - (MyMarker, HandlerArgs, Params)
/// // etc.
///
/// impl<S> Handler<(MyMarker, ), S> for MyHandler {
///   type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;
///
///   fn call_with_state(self, _args: HandlerArgs, _state: S) -> Self::Future {
///     todo!("use nothing but your struct")
///   }
/// }
///
/// impl<S> Handler<(MyMarker, HandlerArgs), S> for MyHandler {
///   type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;
///
///   fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
///     todo!("use the args")
///   }
/// }
/// ```
///
/// [`DeserializeOwned`]: serde::de::DeserializeOwned
#[cfg_attr(feature = "pubsub", doc = " [`Listener`]: crate::pubsub::Listener")]
pub trait Handler<T, S>: Clone + Send + Sync + Sized + 'static {
    /// The future returned by the handler.
    type Future: Future<Output = Option<Box<RawValue>>> + Send + 'static;

    /// Call the handler with the given request and state.
    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future;
}

/// Extension trait for [`Handler`]s.
pub(crate) trait HandlerInternal<T, S>: Handler<T, S> {
    /// Create a new handler that wraps this handler and has some state.
    fn with_state(self, state: S) -> HandlerService<Self, T, S> {
        HandlerService::new(self, state)
    }

    /// Convert the handler into a [`Route`], ready for internal registration
    /// on a [`Router`].
    ///
    /// [`Router`]: crate::Router
    #[allow(private_interfaces)]
    fn into_route(self, state: S) -> Route
    where
        T: Send + 'static,
        S: Clone + Send + Sync + 'static,
    {
        Route::new(self.with_state(state))
    }
}

impl<T, S, H> HandlerInternal<T, S> for H where H: Handler<T, S> {}

/// A [`Handler`] with some state `S` and params type `T`.
#[derive(Debug)]
pub(crate) struct HandlerService<H, T, S> {
    handler: H,
    state: S,
    _marker: std::marker::PhantomData<fn() -> T>,
}

impl<H, T, S> Clone for HandlerService<H, T, S>
where
    H: Clone,
    S: Clone,
{
    fn clone(&self) -> Self {
        Self {
            handler: self.handler.clone(),
            state: self.state.clone(),
            _marker: PhantomData,
        }
    }
}

impl<H, T, S> HandlerService<H, T, S> {
    /// Create a new handler service.
    pub(crate) const fn new(handler: H, state: S) -> Self {
        Self {
            handler,
            state,
            _marker: PhantomData,
        }
    }
}

impl<H, T, S> tower::Service<HandlerArgs> for HandlerService<H, T, S>
where
    Self: Clone,
    H: Handler<T, S>,
    T: Send + 'static,
    S: Clone + Send + Sync + 'static,
{
    type Response = Option<Box<RawValue>>;
    type Error = Infallible;
    type Future = Pin<Box<dyn Future<Output = Result<Option<Box<RawValue>>, Infallible>> + Send>>;

    fn poll_ready(&mut self, _cx: &mut task::Context<'_>) -> task::Poll<Result<(), Self::Error>> {
        task::Poll::Ready(Ok(()))
    }

    fn call(&mut self, args: HandlerArgs) -> Self::Future {
        let this = self.clone();
        Box::pin(async move {
            // This span captures standard OpenTelemetry attributes for
            // JSON-RPC according to OTEL semantic conventions.
            let span = args.span().clone();

            Ok(this
                .handler
                .call_with_state(args, this.state.clone())
                .instrument(span)
                .await
                .inspect(|res| {
                    trace!(?res, "handler response");
                }))
        })
    }
}

/// A marker type for handlers that return a [`Result`].
///
/// This type should never be constructed, and importing it is almost certainly
/// a mistake.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct OutputResult {
    _sealed: (),
}

/// A marker type for handlers that return a [`ResponsePayload`].
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct OutputResponsePayload {
    _sealed: (),
}

// Takes nothing, returns ResponsePayload
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResponsePayload,), S> for F
where
    F: FnOnce() -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: RpcSend,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self())
    }
}

// Takes Ctx, returns ResponsePayload
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResponsePayload, HandlerCtx), S> for F
where
    F: FnOnce(HandlerCtx) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: RpcSend,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx))
    }
}

// Takes Params, returns ResponsePayload
impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResponsePayload, PhantomParams<Input>), S>
    for F
where
    F: FnOnce(Input) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: RpcSend,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(params: Input))
    }
}

// Takes Params<Params>, returns ResponsePayload
impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResponsePayload, Params<Input>), S> for F
where
    F: FnOnce(Params<Input>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: RpcSend,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(params: Input))
    }
}

// Takes State, returns ResponsePayload
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResponsePayload, PhantomState<S>), S> for F
where
    F: FnOnce(S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: RpcSend,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(state))
    }
}

// Takes State<State>, returns ResponsePayload
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResponsePayload, State<S>), S> for F
where
    F: FnOnce(State<S>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: RpcSend,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(State(state)))
    }
}

// Takes Ctx, Params, returns ResponsePayload
impl<F, Fut, Input, Payload, ErrData, S>
    Handler<(OutputResponsePayload, HandlerCtx, PhantomParams<Input>), S> for F
where
    F: FnOnce(HandlerCtx, Input) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: RpcSend,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, params: Input))
    }
}

// Takes Ctx, Params<Params>, returns ResponsePayload
impl<F, Fut, Input, Payload, ErrData, S>
    Handler<(OutputResponsePayload, HandlerCtx, Params<Input>), S> for F
where
    F: FnOnce(HandlerCtx, Params<Input>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: RpcSend,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, params: Input))
    }
}

// Takes Params, State, returns ResponsePayload
impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResponsePayload, Input, S), S> for F
where
    F: FnOnce(Input, S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: RpcSend,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(params: Input, state))
    }
}

// Takes Ctx, State, returns ResponsePayload
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResponsePayload, HandlerCtx, PhantomState<S>), S>
    for F
where
    F: FnOnce(HandlerCtx, S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: RpcSend,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, state))
    }
}

// Takes Ctx, State<State>, returns ResponsePayload
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResponsePayload, HandlerCtx, State<S>), S> for F
where
    F: FnOnce(HandlerCtx, State<S>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: RpcSend,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, State(state)))
    }
}

// Takes Ctx, Params, State, returns ResponsePayload
impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResponsePayload, HandlerCtx, Input, S), S>
    for F
where
    F: FnOnce(HandlerCtx, Input, S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = ResponsePayload<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: RpcSend,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, params: Input, state))
    }
}

// Takes nothing, returns Result
impl<F, Fut, Payload, ErrData, S> Handler<(OutputResult,), S> for F
where
    F: FnOnce() -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self())
    }
}

impl<F, Fut, Payload, ErrData, S> Handler<(OutputResult, HandlerCtx), S> for F
where
    F: FnOnce(HandlerCtx) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx))
    }
}

impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResult, PhantomParams<Input>), S> for F
where
    F: FnOnce(Input) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(params: Input))
    }
}

impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResult, Params<Input>), S> for F
where
    F: FnOnce(Params<Input>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(params: Input))
    }
}

impl<F, Fut, Payload, ErrData, S> Handler<(OutputResult, PhantomState<S>), S> for F
where
    F: FnOnce(S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(state))
    }
}

impl<F, Fut, Payload, ErrData, S> Handler<(OutputResult, State<S>), S> for F
where
    F: FnOnce(State<S>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(State(state)))
    }
}

impl<F, Fut, Input, Payload, ErrData, S>
    Handler<(OutputResult, HandlerCtx, PhantomParams<Input>), S> for F
where
    F: FnOnce(HandlerCtx, Input) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, params: Input))
    }
}

impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResult, HandlerCtx, Params<Input>), S> for F
where
    F: FnOnce(HandlerCtx, Params<Input>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, _state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, params: Input))
    }
}

impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResult, Input, S), S> for F
where
    F: FnOnce(Input, S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(params: Input, state))
    }
}

impl<F, Fut, Payload, ErrData, S> Handler<(OutputResult, HandlerCtx, PhantomState<S>), S> for F
where
    F: FnOnce(HandlerCtx, S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, state))
    }
}

impl<F, Fut, Payload, ErrData, S> Handler<(OutputResult, HandlerCtx, State<S>), S> for F
where
    F: FnOnce(HandlerCtx, State<S>) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, State(state)))
    }
}

impl<F, Fut, Input, Payload, ErrData, S> Handler<(OutputResult, HandlerCtx, Input, S), S> for F
where
    F: FnOnce(HandlerCtx, Input, S) -> Fut + Clone + Send + Sync + 'static,
    Fut: Future<Output = Result<Payload, ErrData>> + Send + 'static,
    Input: RpcRecv,
    Payload: RpcSend,
    ErrData: IntoErrorPayload,
    S: Send + Sync + 'static,
{
    type Future = Pin<Box<dyn Future<Output = Option<Box<RawValue>>> + Send>>;

    fn call_with_state(self, args: HandlerArgs, state: S) -> Self::Future {
        impl_handler_call!(args, self(ctx, params: Input, state))
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::{HandlerCtx, ResponsePayload};

    #[derive(Clone)]
    struct NewType;

    async fn resp_ok() -> ResponsePayload<(), ()> {
        ResponsePayload(Ok(()))
    }

    async fn ok() -> Result<(), ()> {
        Ok(())
    }

    #[test]
    #[ignore = "compilation_only"]
    #[allow(unused_variables)]
    fn combination_inference() {
        let router: crate::Router<()> = crate::Router::<NewType>::new()
            //responses
            .route("respnse", ok)
            .route("respnse, ctx", |_: HandlerCtx| resp_ok())
            .route("respnse, params", |_: u16| resp_ok())
            .route("respnse, ctx, params", |_: HandlerCtx, _: u16| resp_ok())
            .route("respnse, params, state", |_: u8, _: NewType| resp_ok())
            .route("respnse, ctx, state", |_: HandlerCtx, _: NewType| resp_ok())
            .route(
                "respnse, ctx, params, state",
                |_: HandlerCtx, _: u8, _: NewType| resp_ok(),
            )
            // results
            .route("result", ok)
            .route("result, ctx", |_: HandlerCtx| ok())
            .route("result, params", |_: u16| ok())
            .route("result, ctx, params", |_: HandlerCtx, _: u16| ok())
            .route("result, params, state", |_: u8, _: NewType| ok())
            .route("result, ctx, state", |_: HandlerCtx, _: NewType| ok())
            .route(
                "result, ctx, params, state",
                |_: HandlerCtx, _: u8, _: NewType| ok(),
            )
            .with_state::<u8>(NewType)
            .route::<_, (OutputResult, State<u8>)>("no inference error", |State(state)| ok())
            .with_state(1u8);
    }
}

// Some code is this file is reproduced under the terms of the MIT license. It
// originates from the `axum` crate. The original source code can be found at
// the following URL, and the original license is included below.
//
// https://github.com/tokio-rs/axum/
//
// The MIT License (MIT)
//
// Copyright (c) 2019 Axum Contributors
//
// Permission is hereby granted, free of charge, to any
// person obtaining a copy of this software and associated
// documentation files (the "Software"), to deal in the
// Software without restriction, including without
// limitation the rights to use, copy, modify, merge,
// publish, distribute, sublicense, and/or sell copies of
// the Software, and to permit persons to whom the Software
// is furnished to do so, subject to the following
// conditions:
//
// The above copyright notice and this permission notice
// shall be included in all copies or substantial portions
// of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF
// ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED
// TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
// PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
// SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
// IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.