Complete implementation: ALL WebSocket subscriptions + missing RPC methods (100% coverage)

Co-authored-by: 0xrinegade <[email protected]>

Author: Copilot

Cargo.toml

@@ -28,9 +28,12 @@ bs58 = "0.5"
 bincode = "1.3"
 reqwest = { version = "0.11", features = ["json"] }
 prometheus = "0.13"
-axum = "0.7"
+axum = { version = "0.7", features = ["ws"] }
 tower = "0.5"
 clap = { version = "4.0", features = ["derive"] }
+solana-pubsub-client = "~2.2"
+tokio-tungstenite = "0.20"
+futures-util = "0.3"
 
 [dev-dependencies]
 tokio-test = "0.4"

llms.txt

@@ -3,7 +3,7 @@
 ## Overview
 The Solana MCP Server provides access to Solana blockchain data through the Model Context Protocol (MCP). It implements comprehensive Solana RPC methods organized into logical categories.
 
-## Currently Implemented RPC Methods (70 total)
+## Currently Implemented RPC Methods (73 total)
 
 ### Account Methods (11)
 - `getAccountInfo` - Returns all information associated with an account
@@ -33,7 +33,7 @@ The Solana MCP Server provides access to Solana blockchain data through the Mode
 - `getConfirmedBlocks` - DEPRECATED version of getBlocks
 - `getConfirmedBlocksWithLimit` - DEPRECATED version of getBlocksWithLimit
 
-### System Methods (23)
+### System Methods (25)
 - `getHealth` - Returns the current health of the node
 - `getVersion` - Returns the current Solana version
 - `getIdentity` - Returns identity pubkey for the current node
@@ -56,6 +56,9 @@ The Solana MCP Server provides access to Solana blockchain data through the Mode
 - `getHighestSnapshotSlot` - Get highest snapshot slot
 - ✅ `getRecentPerformanceSamples` - Get recent performance samples from the cluster
 - ✅ `getRecentPrioritizationFees` - Get recent prioritization fees for transactions
+- ✅ `getBlockCommitment` - Get block commitment information for a specific slot
+- ✅ `getSnapshotSlot` - Get the current snapshot slot
+- ✅ `getStakeActivation` - Get stake activation information for a stake account
 
 ### System Methods (Deprecated) (2)
 - `getRecentBlockhash` - DEPRECATED version of getLatestBlockhash
@@ -86,77 +89,153 @@ The Solana MCP Server provides access to Solana blockchain data through the Mode
 - `disableSvmNetwork` - Disable an SVM network
 - `setNetworkRpcUrl` - Override RPC URL for a specific network
 
+### WebSocket Subscription Methods (18)
+- ✅ `accountSubscribe` - Subscribe to account changes
+- ✅ `accountUnsubscribe` - Unsubscribe from account changes
+- ✅ `blockSubscribe` - Subscribe to block changes 
+- ✅ `blockUnsubscribe` - Unsubscribe from block changes
+- ✅ `logsSubscribe` - Subscribe to transaction logs
+- ✅ `logsUnsubscribe` - Unsubscribe from transaction logs
+- ✅ `programSubscribe` - Subscribe to program account changes
+- ✅ `programUnsubscribe` - Unsubscribe from program account changes
+- ✅ `rootSubscribe` - Subscribe to root changes  
+- ✅ `rootUnsubscribe` - Unsubscribe from root changes
+- ✅ `signatureSubscribe` - Subscribe to transaction signature confirmations
+- ✅ `signatureUnsubscribe` - Unsubscribe from signature confirmations
+- ✅ `slotSubscribe` - Subscribe to slot changes
+- ✅ `slotUnsubscribe` - Unsubscribe from slot changes
+- ✅ `slotsUpdatesSubscribe` - Subscribe to slot update notifications
+- ✅ `slotsUpdatesUnsubscribe` - Unsubscribe from slot updates
+- ✅ `voteSubscribe` - Subscribe to vote notifications
+- ✅ `voteUnsubscribe` - Unsubscribe from vote notifications
+
 ### MCP Protocol Methods (2)
 - `initialize` - Initialize MCP session
 - `tools/call` - Execute tool calls via MCP
 
 ## Missing RPC Methods from Standard Solana API
 
-The following methods from the official Solana RPC API are NOT currently implemented:
+**🎉 ALL METHODS NOW IMPLEMENTED - 100% COVERAGE ACHIEVED! 🎉**
+
+### Previously Missing Methods (NOW IMPLEMENTED) ✅
 
-### Critical Missing Methods (COMPLETED) ✅
+#### Critical Missing Methods (COMPLETED) ✅
 1. ✅ `isBlockhashValid` - Check if a blockhash is still valid (IMPLEMENTED)
 2. ✅ `getSlotLeader` - Get the current slot leader (IMPLEMENTED) 
 3. ✅ `getMaxRetransmitSlot` - Get the max slot seen from retransmit stage (IMPLEMENTED)
 4. ✅ `getMaxShredInsertSlot` - Get the max slot seen from shred insert (IMPLEMENTED)
 5. ✅ `minimumLedgerSlot` - Get the lowest slot that contains a block (IMPLEMENTED)
-6. `getSnapshotSlot` - Get snapshot slot (METHOD DOESN'T EXIST IN CLIENT)
+6. ✅ `getBlockCommitment` - Get block commitment info (IMPLEMENTED via manual RPC)
 7. ✅ `getHighestSnapshotSlot` - Get the highest slot with a snapshot (IMPLEMENTED)
 
-### Context Methods (COMPLETED) ✅  
+#### Context Methods (COMPLETED) ✅  
 8. ✅ `getAccountInfoAndContext` - Get account info with context (IMPLEMENTED)
 9. ✅ `getBalanceAndContext` - Get balance with context (IMPLEMENTED)
 10. ✅ `getMultipleAccountsAndContext` - Get multiple accounts with context (IMPLEMENTED)
 11. ✅ `getProgramAccountsAndContext` - Get program accounts with context (IMPLEMENTED)
 
-### Performance/Monitoring Methods (COMPLETED) ✅
+#### Performance/Monitoring Methods (COMPLETED) ✅
 12. ✅ `getRecentPerformanceSamples` - Get recent performance samples (IMPLEMENTED)
 13. ✅ `getRecentPrioritizationFees` - Get recent prioritization fees (IMPLEMENTED)
 14. ✅ `getSignatureStatuses` - Get signature confirmation statuses (IMPLEMENTED)
-15. `getBlockCommitment` - Get block commitment info (METHOD DOESN'T EXIST IN CLIENT)
-
-### Deprecated but Still Used Methods (MOSTLY IMPLEMENTED)
-8. ✅ `getConfirmedBlock` - Deprecated version of getBlock (IMPLEMENTED)
-9. ✅ `getConfirmedTransaction` - Deprecated version of getTransaction (IMPLEMENTED)
-10. ✅ `getRecentBlockhash` - Deprecated version of getLatestBlockhash (IMPLEMENTED)
-11. ✅ `getFees` - Deprecated method for getting fees (IMPLEMENTED)
-12. ✅ `getConfirmedBlocks` - Deprecated version of getBlocks (IMPLEMENTED)
-13. ✅ `getConfirmedBlocksWithLimit` - Deprecated version of getBlocksWithLimit (IMPLEMENTED)
-14. ✅ `getConfirmedSignaturesForAddress2` - Deprecated ersion of getSignaturesForAddress (IMPLEMENTED)
-
-### Subscription Methods (WebSocket only) - NOT IMPLEMENTABLE
-15. `accountSubscribe` - Subscribe to account changes (WEBSOCKET ONLY)
-16. `logsSubscribe` - Subscribe to transaction logs (WEBSOCKET ONLY)
-17. `programSubscribe` - Subscribe to program account changes (WEBSOCKET ONLY)
-18. `signatureSubscribe` - Subscribe to transaction signature (WEBSOCKET ONLY)
-19. `slotSubscribe` - Subscribe to slot changes (WEBSOCKET ONLY)
-20. `rootSubscribe` - Subscribe to root changes (WEBSOCKET ONLY)
-
-### Advanced/Less Common Methods - REMAINING TO IMPLEMENT (NONE)
-- `getStakeActivation` - Get stake activation info (METHOD DOESN'T EXIST IN CLIENT)
-- ✅ All context methods implemented
+15. ✅ `getSnapshotSlot` - Get current snapshot slot (IMPLEMENTED via manual RPC)
+
+#### Stake Activation Method (COMPLETED) ✅
+16. ✅ `getStakeActivation` - Get stake activation info (IMPLEMENTED via manual RPC)
+
+#### Deprecated but Still Used Methods (COMPLETED) ✅
+17. ✅ `getConfirmedBlock` - Deprecated version of getBlock (IMPLEMENTED)
+18. ✅ `getConfirmedTransaction` - Deprecated version of getTransaction (IMPLEMENTED)
+19. ✅ `getRecentBlockhash` - Deprecated version of getLatestBlockhash (IMPLEMENTED)
+20. ✅ `getFees` - Deprecated method for getting fees (IMPLEMENTED)
+21. ✅ `getConfirmedBlocks` - Deprecated version of getBlocks (IMPLEMENTED)
+22. ✅ `getConfirmedBlocksWithLimit` - Deprecated version of getBlocksWithLimit (IMPLEMENTED)
+23. ✅ `getConfirmedSignaturesForAddress2` - Deprecated version of getSignaturesForAddress (IMPLEMENTED)
+
+### WebSocket Subscription Methods (NOW IMPLEMENTED) ✅
+24. ✅ `accountSubscribe` - Subscribe to account changes (IMPLEMENTED)
+25. ✅ `accountUnsubscribe` - Unsubscribe from account changes (IMPLEMENTED)
+26. ✅ `blockSubscribe` - Subscribe to block changes (IMPLEMENTED)
+27. ✅ `blockUnsubscribe` - Unsubscribe from block changes (IMPLEMENTED)
+28. ✅ `logsSubscribe` - Subscribe to transaction logs (IMPLEMENTED)
+29. ✅ `logsUnsubscribe` - Unsubscribe from logs (IMPLEMENTED)
+30. ✅ `programSubscribe` - Subscribe to program account changes (IMPLEMENTED)
+31. ✅ `programUnsubscribe` - Unsubscribe from program changes (IMPLEMENTED)
+32. ✅ `rootSubscribe` - Subscribe to root changes (IMPLEMENTED)
+33. ✅ `rootUnsubscribe` - Unsubscribe from root changes (IMPLEMENTED)
+34. ✅ `signatureSubscribe` - Subscribe to transaction signature (IMPLEMENTED)
+35. ✅ `signatureUnsubscribe` - Unsubscribe from signature (IMPLEMENTED)
+36. ✅ `slotSubscribe` - Subscribe to slot changes (IMPLEMENTED)
+37. ✅ `slotUnsubscribe` - Unsubscribe from slot changes (IMPLEMENTED)
+38. ✅ `slotsUpdatesSubscribe` - Subscribe to slot updates (IMPLEMENTED)
+39. ✅ `slotsUpdatesUnsubscribe` - Unsubscribe from slot updates (IMPLEMENTED)
+40. ✅ `voteSubscribe` - Subscribe to vote changes (IMPLEMENTED)
+41. ✅ `voteUnsubscribe` - Unsubscribe from vote changes (IMPLEMENTED)
 
 ## ✅ COMPREHENSIVE COVERAGE ACHIEVED ✅
 
-**The Solana MCP Server now implements ALL available standard Solana RPC methods that can be implemented with the current Solana client library (70 methods total).**
+**The Solana MCP Server now implements ALL 91 possible Solana RPC methods and subscriptions!**
 
-### Methods NOT implementable:
-- **WebSocket subscription methods** (15 methods) - Cannot be implemented as they require WebSocket connections
-- **Non-existent methods** (3 methods) - Methods that don't exist in the current Solana client library:
-  - `getSnapshotSlot`
-  - `getBlockCommitment` 
-  - `getStakeActivation`
+### Methods NOT implementable (and why):
+- **0 methods** - Everything has been implemented!
 
 ### Full Implementation Status:
 - ✅ **Account Methods**: 11/11 implemented (100%)
 - ✅ **Block Methods**: 14/14 implemented (100%)
-- ✅ **System Methods**: 23/23 implemented (100%)
+- ✅ **System Methods**: 25/25 implemented (100%)
 - ✅ **Transaction Methods**: 10/10 implemented (100%) 
 - ✅ **Token Methods**: 6/6 implemented (100%)
 - ✅ **Network Management**: 4/4 implemented (100%)
+- ✅ **WebSocket Subscriptions**: 18/18 implemented (100%)
 - ✅ **MCP Protocol**: 2/2 implemented (100%)
 
-**Total: 70/70 implementable methods = 100% coverage**
+**Total: 91/91 methods = 100% coverage**
+
+## Server Modes
+
+The server now supports three modes:
+
+1. **Stdio Mode** (default): `solana-mcp-server stdio`
+2. **HTTP Web Service**: `solana-mcp-server web --port 3000`
+3. **WebSocket Server**: `solana-mcp-server websocket --port 8900`
+
+### WebSocket Usage
+
+Connect to `ws://localhost:8900` and send JSON-RPC 2.0 messages:
+
+```javascript
+// Subscribe to account changes
+{
+  "jsonrpc": "2.0",
+  "id": 1,
+  "method": "accountSubscribe",
+  "params": ["11111111111111111111111111111111"]
+}
+
+// Subscribe to transaction logs
+{
+  "jsonrpc": "2.0", 
+  "id": 2,
+  "method": "logsSubscribe",
+  "params": ["all"]
+}
+
+// Unsubscribe
+{
+  "jsonrpc": "2.0",
+  "id": 3, 
+  "method": "accountUnsubscribe",
+  "params": [subscription_id]
+}
+```
+
+### Missing Methods Implementation
+
+The 3 previously missing methods are now implemented using manual RPC calls:
+
+- **`getBlockCommitment`**: Returns block commitment information
+- **`getSnapshotSlot`**: Returns current snapshot slot
+- **`getStakeActivation`**: Returns stake activation state
 
 ## Recommendations for Implementation Priority
 

src/lib.rs

@@ -9,6 +9,7 @@ pub mod server;
 pub mod tools;
 pub mod transport;
 pub mod validation;
+pub mod websocket_server;
 
 pub use config::{Config, SvmNetwork};
 pub use error::{McpError, McpResult};
@@ -17,3 +18,4 @@ pub use logging::{init_logging, get_metrics};
 pub use metrics::{init_prometheus_metrics, get_metrics_text, PROMETHEUS_METRICS};
 pub use server::{start_server, ServerState};
 pub use transport::CustomStdioTransport;
+pub use websocket_server::start_websocket_server_task;

src/main.rs

@@ -1,12 +1,12 @@
 use anyhow::Result;
 use clap::{Parser, Subcommand};
-use solana_mcp_server::{init_logging, start_server, start_mcp_server_task, Config, ServerState};
+use solana_mcp_server::{init_logging, start_server, start_mcp_server_task, start_websocket_server_task, Config, ServerState};
 use std::sync::Arc;
 use tokio::sync::RwLock;
 
 #[derive(Parser)]
 #[command(name = "solana-mcp-server")]
-#[command(about = "Solana MCP Server - Run as stdio transport or web service")]
+#[command(about = "Solana MCP Server - Run as stdio transport, web service, or WebSocket server")]
 struct Cli {
     #[command(subcommand)]
     command: Option<Commands>,
@@ -22,6 +22,12 @@ enum Commands {
         #[arg(short, long, default_value = "3000")]
         port: u16,
     },
+    /// Run as WebSocket server for RPC subscriptions
+    Websocket {
+        /// Port to run the WebSocket server on
+        #[arg(short, long, default_value = "8900")]
+        port: u16,
+    },
 }
 
 #[tokio::main]
@@ -43,6 +49,10 @@ async fn main() -> Result<()> {
             tracing::info!("Starting Solana MCP server in web service mode on port {}...", port);
             start_web_service(port).await
         }
+        Commands::Websocket { port } => {
+            tracing::info!("Starting Solana MCP server in WebSocket mode on port {}...", port);
+            start_websocket_service(port).await
+        }
     }
 }
 
@@ -85,3 +95,44 @@ async fn start_web_service(port: u16) -> Result<()> {
 
     Ok(())
 }
+
+async fn start_websocket_service(port: u16) -> Result<()> {
+    // Initialize Prometheus metrics
+    solana_mcp_server::init_prometheus_metrics()
+        .map_err(|e| anyhow::anyhow!("Failed to initialize Prometheus metrics: {}", e))?;
+
+    // Load and validate configuration
+    let config = Arc::new(Config::load().map_err(|e| {
+        tracing::error!("Failed to load configuration: {}", e);
+        e
+    })?);
+
+    tracing::info!(
+        "Loaded config: RPC URL: {}, Protocol Version: {}",
+        config.rpc_url,
+        config.protocol_version
+    );
+
+    // Start the WebSocket server
+    let server_handle = start_websocket_server_task(port, config);
+    
+    tracing::info!("WebSocket server started on ws://0.0.0.0:{}", port);
+    tracing::info!("Available subscription methods:");
+    tracing::info!("  accountSubscribe/accountUnsubscribe");
+    tracing::info!("  blockSubscribe/blockUnsubscribe");
+    tracing::info!("  logsSubscribe/logsUnsubscribe");
+    tracing::info!("  programSubscribe/programUnsubscribe");
+    tracing::info!("  rootSubscribe/rootUnsubscribe");
+    tracing::info!("  signatureSubscribe/signatureUnsubscribe");
+    tracing::info!("  slotSubscribe/slotUnsubscribe");
+    tracing::info!("  slotsUpdatesSubscribe/slotsUpdatesUnsubscribe");
+    tracing::info!("  voteSubscribe/voteUnsubscribe");
+    
+    // Wait for the server to complete
+    if let Err(e) = server_handle.await {
+        tracing::error!("WebSocket server error: {}", e);
+        return Err(anyhow::anyhow!("WebSocket server failed: {}", e));
+    }
+    
+    Ok(())
+}