tauri-plugin-wdio-webdriver 1.0.0

Embedded WebDriver server for WDIO Tauri testing
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

use std::sync::Arc;

use axum::extract::{Path, State};
use axum::Json;
use serde::Deserialize;
use tauri::Runtime;

use crate::platform::PrintOptions;
use crate::server::response::{WebDriverResponse, WebDriverResult};
use crate::server::AppState;

#[derive(Debug, Default, Deserialize)]
pub struct PrintRequest {
    #[serde(default)]
    pub orientation: Option<String>,
    #[serde(default)]
    pub scale: Option<f64>,
    #[serde(default)]
    pub background: Option<bool>,
    #[serde(default, rename = "pageWidth")]
    pub page_width: Option<f64>,
    #[serde(default, rename = "pageHeight")]
    pub page_height: Option<f64>,
    #[serde(default, rename = "marginTop")]
    pub margin_top: Option<f64>,
    #[serde(default, rename = "marginBottom")]
    pub margin_bottom: Option<f64>,
    #[serde(default, rename = "marginLeft")]
    pub margin_left: Option<f64>,
    #[serde(default, rename = "marginRight")]
    pub margin_right: Option<f64>,
    #[serde(default, rename = "shrinkToFit")]
    pub shrink_to_fit: Option<bool>,
    #[serde(default, rename = "pageRanges")]
    pub page_ranges: Option<Vec<String>>,
}

impl From<PrintRequest> for PrintOptions {
    fn from(req: PrintRequest) -> Self {
        PrintOptions {
            orientation: req.orientation,
            scale: req.scale,
            background: req.background,
            page_width: req.page_width,
            page_height: req.page_height,
            margin_top: req.margin_top,
            margin_bottom: req.margin_bottom,
            margin_left: req.margin_left,
            margin_right: req.margin_right,
            shrink_to_fit: req.shrink_to_fit,
            page_ranges: req.page_ranges,
        }
    }
}

/// POST `/session/{session_id}/print` - Print page to PDF
pub async fn print<R: Runtime + 'static>(
    State(state): State<Arc<AppState<R>>>,
    Path(session_id): Path<String>,
    Json(request): Json<PrintRequest>,
) -> WebDriverResult {
    let sessions = state.sessions.read().await;
    let session = sessions.get(&session_id)?;
    let current_window = session.current_window.clone();
    let timeouts = session.timeouts.clone();
    let frame_context = session.frame_context.clone();
    drop(sessions);

    let executor = state.get_executor_for_window(&current_window, timeouts, frame_context)?;
    let pdf_base64 = executor.print_page(request.into()).await?;

    Ok(WebDriverResponse::success(pdf_base64))
}