fix: add a few more with_ mutator methods

Author: jamadeo

crates/rmcp/src/model.rs

@@ -793,6 +793,11 @@ impl InitializeRequestParams {
             client_info,
         }
     }
+
+    pub fn with_protocol_version(mut self, protocol_version: ProtocolVersion) -> Self {
+        self.protocol_version = protocol_version;
+        self
+    }
 }
 
 impl RequestParamsMeta for InitializeRequestParams {
@@ -978,6 +983,30 @@ impl Implementation {
             website_url: None,
         }
     }
+
+    /// Set the human-readable title.
+    pub fn with_title(mut self, title: impl Into<String>) -> Self {
+        self.title = Some(title.into());
+        self
+    }
+
+    /// Set the description.
+    pub fn with_description(mut self, description: impl Into<String>) -> Self {
+        self.description = Some(description.into());
+        self
+    }
+
+    /// Set the icons.
+    pub fn with_icons(mut self, icons: Vec<Icon>) -> Self {
+        self.icons = Some(icons);
+        self
+    }
+
+    /// Set the website URL.
+    pub fn with_website_url(mut self, website_url: impl Into<String>) -> Self {
+        self.website_url = Some(website_url.into());
+        self
+    }
 }
 
 #[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Default)]
@@ -992,6 +1021,13 @@ pub struct PaginatedRequestParams {
     pub cursor: Option<String>,
 }
 
+impl PaginatedRequestParams {
+    pub fn with_cursor(mut self, cursor: Option<String>) -> Self {
+        self.cursor = cursor;
+        self
+    }
+}
+
 impl RequestParamsMeta for PaginatedRequestParams {
     fn meta(&self) -> Option<&Meta> {
         self.meta.as_ref()
@@ -1037,6 +1073,18 @@ impl ProgressNotificationParam {
             message: None,
         }
     }
+
+    /// Set the total number of items to process.
+    pub fn with_total(mut self, total: f64) -> Self {
+        self.total = Some(total);
+        self
+    }
+
+    /// Set a message describing the current progress.
+    pub fn with_message(mut self, message: impl Into<String>) -> Self {
+        self.message = Some(message.into());
+        self
+    }
 }
 
 pub type ProgressNotification = Notification<ProgressNotificationMethod, ProgressNotificationParam>;
@@ -1116,6 +1164,12 @@ impl ReadResourceRequestParams {
             uri: uri.into(),
         }
     }
+
+    /// Set the metadata for this request.
+    pub fn with_meta(mut self, meta: Meta) -> Self {
+        self.meta = Some(meta);
+        self
+    }
 }
 
 impl RequestParamsMeta for ReadResourceRequestParams {
@@ -1289,6 +1343,12 @@ impl GetPromptRequestParams {
         self.arguments = Some(arguments);
         self
     }
+
+    /// Set the metadata for this request.
+    pub fn with_meta(mut self, meta: Meta) -> Self {
+        self.meta = Some(meta);
+        self
+    }
 }
 
 impl RequestParamsMeta for GetPromptRequestParams {
@@ -2835,6 +2895,12 @@ impl CreateMessageResult {
         self
     }
 
+    /// Set the model identifier.
+    pub fn with_model(mut self, model: impl Into<String>) -> Self {
+        self.model = model.into();
+        self
+    }
+
     /// Validate the result per SEP-1577: role must be "assistant".
     pub fn validate(&self) -> Result<(), String> {
         if self.message.role != Role::Assistant {

crates/rmcp/src/model/resource.rs

@@ -80,6 +80,7 @@ pub enum ResourceContents {
 }
 
 impl ResourceContents {
+    /// Create text resource contents.
     pub fn text(text: impl Into<String>, uri: impl Into<String>) -> Self {
         Self::TextResourceContents {
             uri: uri.into(),
@@ -88,6 +89,34 @@ impl ResourceContents {
             meta: None,
         }
     }
+
+    /// Create blob resource contents.
+    pub fn blob(blob: impl Into<String>, uri: impl Into<String>) -> Self {
+        Self::BlobResourceContents {
+            uri: uri.into(),
+            mime_type: None,
+            blob: blob.into(),
+            meta: None,
+        }
+    }
+
+    /// Set the MIME type on this resource contents.
+    pub fn with_mime_type(mut self, mime_type: impl Into<String>) -> Self {
+        match &mut self {
+            Self::TextResourceContents { mime_type: mt, .. } => *mt = Some(mime_type.into()),
+            Self::BlobResourceContents { mime_type: mt, .. } => *mt = Some(mime_type.into()),
+        }
+        self
+    }
+
+    /// Set the metadata on this resource contents.
+    pub fn with_meta(mut self, meta: Meta) -> Self {
+        match &mut self {
+            Self::TextResourceContents { meta: m, .. } => *m = Some(meta),
+            Self::BlobResourceContents { meta: m, .. } => *m = Some(meta),
+        }
+        self
+    }
 }
 
 impl RawResource {
@@ -104,6 +133,80 @@ impl RawResource {
             meta: None,
         }
     }
+
+    /// Set the human-readable title.
+    pub fn with_title(mut self, title: impl Into<String>) -> Self {
+        self.title = Some(title.into());
+        self
+    }
+
+    /// Set the description.
+    pub fn with_description(mut self, description: impl Into<String>) -> Self {
+        self.description = Some(description.into());
+        self
+    }
+
+    /// Set the MIME type.
+    pub fn with_mime_type(mut self, mime_type: impl Into<String>) -> Self {
+        self.mime_type = Some(mime_type.into());
+        self
+    }
+
+    /// Set the size in bytes.
+    pub fn with_size(mut self, size: u32) -> Self {
+        self.size = Some(size);
+        self
+    }
+
+    /// Set the icons.
+    pub fn with_icons(mut self, icons: Vec<Icon>) -> Self {
+        self.icons = Some(icons);
+        self
+    }
+
+    /// Set the metadata.
+    pub fn with_meta(mut self, meta: Meta) -> Self {
+        self.meta = Some(meta);
+        self
+    }
+}
+
+impl RawResourceTemplate {
+    /// Creates a new RawResourceTemplate with a URI template and name.
+    pub fn new(uri_template: impl Into<String>, name: impl Into<String>) -> Self {
+        Self {
+            uri_template: uri_template.into(),
+            name: name.into(),
+            title: None,
+            description: None,
+            mime_type: None,
+            icons: None,
+        }
+    }
+
+    /// Set the human-readable title.
+    pub fn with_title(mut self, title: impl Into<String>) -> Self {
+        self.title = Some(title.into());
+        self
+    }
+
+    /// Set the description.
+    pub fn with_description(mut self, description: impl Into<String>) -> Self {
+        self.description = Some(description.into());
+        self
+    }
+
+    /// Set the MIME type.
+    pub fn with_mime_type(mut self, mime_type: impl Into<String>) -> Self {
+        self.mime_type = Some(mime_type.into());
+        self
+    }
+
+    /// Set the icons.
+    pub fn with_icons(mut self, icons: Vec<Icon>) -> Self {
+        self.icons = Some(icons);
+        self
+    }
 }
 
 #[cfg(test)]