chore: add #[non_exhaustive] to remaining public types

Author: DaleSeo

crates/rmcp/src/handler/server/prompt.rs

@@ -118,6 +118,7 @@ impl<T: IntoGetPromptResult> IntoGetPromptResult for Result<T, crate::ErrorData>
 // Future wrapper that automatically handles IntoGetPromptResult conversion
 pin_project_lite::pin_project! {
     #[project = IntoGetPromptResultFutProj]
+    #[non_exhaustive]
     pub enum IntoGetPromptResultFut<F, R> {
         Pending {
             #[pin]

crates/rmcp/src/handler/server/router.rs

@@ -13,6 +13,7 @@ use crate::{
 pub mod prompt;
 pub mod tool;
 
+#[non_exhaustive]
 pub struct Router<S> {
     pub tool_router: tool::ToolRouter<S>,
     pub prompt_router: prompt::PromptRouter<S>,

crates/rmcp/src/handler/server/router/prompt.rs

@@ -6,6 +6,7 @@ use crate::{
     service::{MaybeBoxFuture, MaybeSend},
 };
 
+#[non_exhaustive]
 pub struct PromptRoute<S> {
     #[allow(clippy::type_complexity)]
     pub get: Arc<DynGetPromptHandler<S>>,
@@ -103,6 +104,7 @@ where
 }
 
 #[derive(Debug)]
+#[non_exhaustive]
 pub struct PromptRouter<S> {
     #[allow(clippy::type_complexity)]
     pub map: std::collections::HashMap<Cow<'static, str>, PromptRoute<S>>,

crates/rmcp/src/handler/server/router/tool.rs

@@ -136,6 +136,7 @@ use crate::{
     service::{MaybeBoxFuture, MaybeSend},
 };
 
+#[non_exhaustive]
 pub struct ToolRoute<S> {
     #[allow(clippy::type_complexity)]
     pub call: Arc<DynCallToolHandler<S>>,
@@ -251,6 +252,7 @@ where
     }
 }
 
+#[non_exhaustive]
 pub struct WithToolAttr<C, S, A>
 where
     C: CallToolHandler<S, A> + MaybeSend + Clone + 'static,
@@ -292,6 +294,7 @@ where
     }
 }
 #[derive(Debug)]
+#[non_exhaustive]
 pub struct ToolRouter<S> {
     #[allow(clippy::type_complexity)]
     pub map: std::collections::HashMap<Cow<'static, str>, ToolRoute<S>>,

crates/rmcp/src/handler/server/tool.rs

@@ -105,6 +105,7 @@ impl<T: IntoCallToolResult> IntoCallToolResult for Result<T, crate::ErrorData> {
 
 pin_project_lite::pin_project! {
     #[project = IntoCallToolResultFutProj]
+    #[non_exhaustive]
     pub enum IntoCallToolResultFut<F, R> {
         Pending {
             #[pin]

crates/rmcp/src/service/server.rs

@@ -571,6 +571,7 @@ macro_rules! elicit_safe {
 }
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
+#[non_exhaustive]
 pub enum ElicitationMode {
     Form,
     Url,

crates/rmcp/src/task_manager.rs

@@ -19,6 +19,7 @@ pub type OperationFuture =
 
 /// Describes metadata associated with an enqueued task.
 #[derive(Debug, Clone)]
+#[non_exhaustive]
 pub struct OperationDescriptor {
     pub operation_id: String,
     pub name: String,
@@ -55,6 +56,7 @@ impl OperationDescriptor {
 }
 
 /// Operation message describing a unit of asynchronous work.
+#[non_exhaustive]
 pub struct OperationMessage {
     pub descriptor: OperationDescriptor,
     pub future: OperationFuture,
@@ -91,6 +93,7 @@ struct RunningTask {
     descriptor: OperationDescriptor,
 }
 
+#[non_exhaustive]
 pub struct TaskResult {
     pub descriptor: OperationDescriptor,
     pub result: Result<Box<dyn OperationResultTransport>, Error>,

crates/rmcp/src/transport.rs

@@ -151,6 +151,7 @@ where
     fn into_transport(self) -> impl Transport<R, Error = E> + 'static;
 }
 
+#[non_exhaustive]
 pub enum TransportAdapterIdentity {}
 impl<R, T, E> IntoTransport<R, E, TransportAdapterIdentity> for T
 where
@@ -231,6 +232,7 @@ where
 
 #[derive(Debug, thiserror::Error)]
 #[error("Transport [{transport_name}] error: {error}")]
+#[non_exhaustive]
 pub struct DynamicTransportError {
     pub transport_name: Cow<'static, str>,
     pub transport_type_id: std::any::TypeId,

crates/rmcp/src/transport/async_rw.rs

@@ -16,6 +16,7 @@ use tokio_util::{
 use super::{IntoTransport, Transport};
 use crate::service::{RxJsonRpcMessage, ServiceRole, TxJsonRpcMessage};
 
+#[non_exhaustive]
 pub enum TransportAdapterAsyncRW {}
 
 impl<Role, R, W> IntoTransport<Role, std::io::Error, TransportAdapterAsyncRW> for (R, W)
@@ -29,6 +30,7 @@ where
     }
 }
 
+#[non_exhaustive]
 pub enum TransportAdapterAsyncCombinedRW {}
 impl<Role, S> IntoTransport<Role, std::io::Error, TransportAdapterAsyncCombinedRW> for S
 where
@@ -277,6 +279,7 @@ fn try_parse_with_compatibility<T: serde::de::DeserializeOwned>(
 }
 
 #[derive(Debug, Error)]
+#[non_exhaustive]
 pub enum JsonRpcMessageCodecError {
     #[error("max line length exceeded")]
     MaxLineLengthExceeded,

crates/rmcp/src/transport/auth.rs

@@ -174,6 +174,7 @@ impl std::fmt::Debug for StoredAuthorizationState {
 /// }
 /// ```
 #[derive(Debug, Clone, Default, Serialize, Deserialize)]
+#[non_exhaustive]
 pub struct VendorExtraTokenFields(pub HashMap<String, Value>);
 
 impl ExtraTokenFields for VendorExtraTokenFields {}
@@ -467,6 +468,7 @@ pub const EXTENSION_OAUTH_CLIENT_CREDENTIALS: &str =
 /// JWT signing algorithm for private_key_jwt authentication (SEP-1046)
 #[cfg(feature = "auth-client-credentials-jwt")]
 #[derive(Debug, Clone, Copy)]
+#[non_exhaustive]
 pub enum JwtSigningAlgorithm {
     RS256,
     RS384,
@@ -504,6 +506,7 @@ impl JwtSigningAlgorithm {
 /// - `ClientSecret`: credentials sent in the request body
 /// - `PrivateKeyJwt`: RFC 7523 signed JWT assertion (requires `auth-client-credentials-jwt` feature)
 #[derive(Debug, Clone)]
+#[non_exhaustive]
 pub enum ClientCredentialsConfig {
     /// Client secret authentication (credentials in request body)
     ClientSecret {
@@ -607,6 +610,7 @@ pub(crate) struct ClientRegistrationRequest {
 }
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
+#[non_exhaustive]
 pub struct ClientRegistrationResponse {
     pub client_id: String,
     pub client_secret: Option<String>,
@@ -2043,6 +2047,7 @@ impl AuthorizationManager {
 }
 
 /// oauth2 authorization session, for guiding user to complete the authorization process
+#[non_exhaustive]
 pub struct AuthorizationSession {
     pub auth_manager: AuthorizationManager,
     pub auth_url: String,
@@ -2195,6 +2200,7 @@ impl AuthorizedHttpClient {
 /// OAuth state machine
 /// Use the OAuthState to manage the OAuth client is more recommend
 /// But also you can use the AuthorizationManager,AuthorizationSession,AuthorizedHttpClient directly
+#[non_exhaustive]
 pub enum OAuthState {
     /// the AuthorizationManager
     Unauthorized(AuthorizationManager),