Fix missing docs

crates/wassette/src/lib.rs

@@ -1,6 +1,10 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
+//! A security-oriented runtime that runs WebAssembly Components via MCP
+
+#![warn(missing_docs)]
+
 use std::collections::HashMap;
 use std::fs;
 use std::path::{Path, PathBuf};
@@ -40,21 +44,28 @@ const DOWNLOADS_DIR: &str = "downloads";
 /// Granular permission rule types
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub enum PermissionRule {
+    /// Network access permission
     #[serde(rename = "network")]
     Network(NetworkPermission),
+    /// File system storage permission
     #[serde(rename = "storage")]
     Storage(StoragePermission),
+    /// Environment variable access permission
     #[serde(rename = "environment")]
     Environment(EnvironmentPermission),
+    /// Custom permission with arbitrary data
     #[serde(rename = "custom")]
     Custom(String, serde_json::Value),
 }
 
 /// Permission grant request structure
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct PermissionGrantRequest {
+    /// The ID of the component requesting permission
     pub component_id: String,
+    /// The type of permission being requested
     pub permission_type: String,
+    /// Additional details specific to the permission type
     pub details: serde_json::Value,
 }
 
@@ -206,17 +217,25 @@ impl ComponentRegistry {
     }
 }
 
+/// Registry for storing policy templates associated with components
 #[derive(Default)]
 struct PolicyRegistry {
+    /// Maps component IDs to their associated policy templates
     component_policies: HashMap<String, Arc<WasiStateTemplate>>,
 }
 
+/// Information about a policy attached to a component
 #[derive(Debug, Clone)]
 pub struct PolicyInfo {
+    /// Unique identifier for the policy
     pub policy_id: String,
+    /// The original URI where the policy was loaded from
     pub source_uri: String,
+    /// Local filesystem path where the policy is stored
     pub local_path: PathBuf,
+    /// ID of the component this policy is attached to
     pub component_id: String,
+    /// Timestamp when the policy was created/attached
     pub created_at: std::time::SystemTime,
 }
 
@@ -761,6 +780,7 @@ impl LifecycleManager {
         self.components.read().await.get(component_id).cloned()
     }
 
+    /// Lists all loaded components by their IDs
     #[instrument(skip(self))]
     pub async fn list_components(&self) -> Vec<String> {
         self.components.read().await.keys().cloned().collect()

crates/wassette/src/wasistate.rs

@@ -28,13 +28,20 @@ pub struct NetworkPermissions {
 /// this includes the wasmtime_wasi, wasmtime_wasi_config and wasmtime_wasi_http states
 #[derive(Clone)]
 pub struct WasiStateTemplate {
+    /// Whether to allow stdout access
     pub allow_stdout: bool,
+    /// Whether to allow stderr access
     pub allow_stderr: bool,
+    /// Whether to allow command line arguments access
     pub allow_args: bool,
+    /// Network permissions configuration
     pub network_perms: NetworkPermissions,
-    pub config_vars: HashMap<String, String>, // wamstime_wasi_config specific state
+    /// Configuration variables for wasmtime_wasi_config
+    pub config_vars: HashMap<String, String>,
+    /// Preopened directories for filesystem access
     pub preopened_dirs: Vec<PreopenedDir>,
-    pub allowed_hosts: HashSet<String>, // allowed network hosts for HTTP requests
+    /// Allowed network hosts for HTTP requests
+    pub allowed_hosts: HashSet<String>,
 }
 
 impl Default for WasiStateTemplate {

src/main.rs

@@ -1,6 +1,10 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT license.
 
+//! The main `wassette(1)` command.
+
+#![warn(missing_docs)]
+
 use std::env;
 use std::future::Future;
 use std::path::PathBuf;
@@ -33,6 +37,7 @@ struct Cli {
 
 #[derive(Subcommand, Debug)]
 enum Commands {
+    /// Begin handling requests over the specified protocol.
     Serve {
         #[arg(long, default_value_t = get_component_dir().into_os_string().into_string().unwrap())]
         plugin_dir: String,
@@ -73,12 +78,17 @@ fn get_component_dir() -> PathBuf {
     }
 }
 
+/// A security-oriented runtime that runs WebAssembly Components via MCP.
 #[derive(Clone)]
 pub struct McpServer {
     lifecycle_manager: LifecycleManager,
 }
 
 impl McpServer {
+    /// Creates a new MCP server instance with the given lifecycle manager.
+    ///
+    /// # Arguments
+    /// * `lifecycle_manager` - The lifecycle manager for handling component operations
     pub fn new(lifecycle_manager: LifecycleManager) -> Self {
         Self { lifecycle_manager }
     }