webdriverbidi 0.2.2

WebDriver BiDi client implementation in Rust.
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

use log::{debug, error};
use reqwest::Client;
use serde::{Deserialize, Serialize};

use crate::error::SessionError;
use crate::webdriver::capabilities::CapabilitiesRequest;

/// Represents the information returned when starting a WebDriver session.
#[derive(Debug, Serialize, Deserialize)]
pub struct SessionResponse {
    pub session_id: String,
    pub capabilities: CapabilitiesRequest,
    pub websocket_url: String,
}

/// Start a WebDriver session through HTTP.
pub async fn start_session(
    base_url: &str,
    capabilities: &CapabilitiesRequest,
) -> Result<SessionResponse, SessionError> {
    let url = format!("{}/session", base_url);
    let payload = capabilities.build();
    let client = create_http_client();

    let response = client
        .post(&url)
        .json(&payload)
        .send()
        .await
        .map_err(|e| {
            error!("Failed to send HTTP request: {}", e);
            SessionError::HttpRequestError(e)
        })?
        .json::<serde_json::Value>()
        .await
        .map_err(|e| {
            error!("Failed to parse JSON response: {}", e);
            SessionError::HttpRequestError(e)
        })?;

    // Extract sessionId and WebSocket URL
    let session_id = response["value"]["sessionId"].as_str().ok_or_else(|| {
        let msg = format!("JSON doesn't contain a sessionId field: {:?}", response);
        error!("{}", msg);
        SessionError::SessionResponseError(msg)
    })?;
    let websocket_url = response["value"]["capabilities"]["webSocketUrl"]
        .as_str()
        .ok_or_else(|| {
            let msg = format!("JSON doesn't contain a webSocketUrl field: {:?}", response);
            error!("{}", msg);
            SessionError::SessionResponseError(msg)
        })?;

    let session_response = SessionResponse {
        session_id: session_id.to_string(),
        capabilities: serde_json::from_value(response["value"]["capabilities"].clone()).map_err(
            |e| {
                let msg = format!("Failed to deserialize capabilities: {}", e);
                error!("{}", msg);
                SessionError::SessionResponseError(msg)
            },
        )?,
        websocket_url: websocket_url.to_string(),
    };

    debug!("Session started successfully: {:?}", session_response);
    Ok(session_response)
}

/// Close a WebDriver session through HTTP.
pub async fn close_session(base_url: &str, session_id: &str) -> Result<(), SessionError> {
    let url = format!("{}/session/{}", base_url, session_id);
    let client = create_http_client();

    client.delete(&url).send().await.map_err(|e| {
        error!("Failed to send HTTP request: {}", e);
        SessionError::HttpRequestError(e)
    })?;

    debug!("Session {} closed successfully", session_id);
    Ok(())
}

/// Create a new reqwest HTTP client.
fn create_http_client() -> Client {
    Client::new()
}