ferridriver-mcp 0.4.0

ferridriver MCP server library -- browser automation via Model Context Protocol
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

//! MCP server transport wiring.
//!
//! Provides ready-made functions to serve an `McpServer` over stdio or HTTP.

use crate::server::McpServer;
use ferridriver::backend::BackendKind;
use ferridriver::state::ConnectMode;
use rmcp::ServiceExt;
use std::sync::Arc;

/// Serve a default `McpServer` over stdio (for Claude Code, CLI clients).
///
/// # Errors
///
/// Returns an error if the MCP transport fails to initialize or the server
/// encounters a fatal communication error.
pub async fn serve_stdio(mode: ConnectMode, backend: BackendKind, headless: bool) -> anyhow::Result<()> {
  let svc = Box::pin(McpServer::new_headless(mode, backend, headless).serve(rmcp::transport::io::stdio())).await?;
  svc.waiting().await?;
  Ok(())
}

/// Serve a default `McpServer` over HTTP (for remote clients, web UIs).
///
/// # Errors
///
/// Returns an error if the TCP listener cannot bind to the requested port,
/// or if the HTTP server encounters a fatal error.
pub async fn serve_http(mode: ConnectMode, backend: BackendKind, port: u16, headless: bool) -> anyhow::Result<()> {
  use rmcp::transport::streamable_http_server::{
    StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
  };

  let ct = tokio_util::sync::CancellationToken::new();
  let config = StreamableHttpServerConfig::default()
    .with_cancellation_token(ct.child_token())
    .with_stateful_mode(true);

  let svc = StreamableHttpService::new(
    move || Ok(McpServer::new_headless(mode.clone(), backend, headless)),
    Arc::new(LocalSessionManager::default()),
    config,
  );

  let listener = tokio::net::TcpListener::bind(("0.0.0.0", port)).await?;
  eprintln!("ferridriver listening on http://0.0.0.0:{port}/mcp");

  axum::serve(listener, axum::Router::new().nest_service("/mcp", svc))
    .with_graceful_shutdown(async move { ct.cancelled_owned().await })
    .await?;

  Ok(())
}

/// Serve a custom `McpServer` (with config/extensions) over stdio.
///
/// # Errors
///
/// Returns an error if the MCP transport fails to initialize or the server
/// encounters a fatal communication error.
pub async fn serve_stdio_with(server: McpServer) -> anyhow::Result<()> {
  let svc = Box::pin(server.serve(rmcp::transport::io::stdio())).await?;
  svc.waiting().await?;
  Ok(())
}

/// Serve a custom `McpServer` (with config/extensions) over HTTP.
///
/// # Errors
///
/// Returns an error if the TCP listener cannot bind to the requested port,
/// or if the HTTP server encounters a fatal error.
pub async fn serve_http_with(server: McpServer, port: u16) -> anyhow::Result<()> {
  use rmcp::transport::streamable_http_server::{
    StreamableHttpServerConfig, StreamableHttpService, session::local::LocalSessionManager,
  };

  let ct = tokio_util::sync::CancellationToken::new();
  let config = StreamableHttpServerConfig::default()
    .with_cancellation_token(ct.child_token())
    .with_stateful_mode(true);

  let svc = StreamableHttpService::new(
    move || Ok(server.clone()),
    Arc::new(LocalSessionManager::default()),
    config,
  );

  let listener = tokio::net::TcpListener::bind(("0.0.0.0", port)).await?;
  eprintln!("ferridriver listening on http://0.0.0.0:{port}/mcp");

  axum::serve(listener, axum::Router::new().nest_service("/mcp", svc))
    .with_graceful_shutdown(async move { ct.cancelled_owned().await })
    .await?;

  Ok(())
}