rustenium 1.1.9

A modern, robust, high-performance WebDriver BiDi automation library for 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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

use std::sync::Arc;
use rustenium_bidi_definitions::browsing_context::types::BrowsingContext;
use rustenium_core::transport::ConnectionTransport;
use rand::Rng;

use crate::error::bidi::InputError;
use super::bidi::touchscreen::Touchscreen;
use super::mouse::Point;
use super::touch::{Touch, SwipeOptions, ScrollOptions, Viewport};
use super::trajectory::{
    generate_trajectory, generate_durations, random_curve_params, weighted_pick,
};

// ── 9-zone grid ─────────────────────────────────────────────────────────────
//
//  +--------+--------+--------+
//  |  TL    |  TC    |  TR    |   top
//  +--------+--------+--------+
//  |  ML    |  MC    |  MR    |   middle
//  +--------+--------+--------+
//  |  BL    |  BC    |  BR    |   bottom    ← thumb rests here
//  +--------+--------+--------+

const ZONE_WEIGHTS: &[(usize, f64)] = &[
    (0, 0.02), // TL
    (1, 0.05), // TC
    (2, 0.02), // TR
    (3, 0.05), // ML
    (4, 0.15), // MC
    (5, 0.05), // MR
    (6, 0.07), // BL
    (7, 0.45), // BC
    (8, 0.07), // BR
];

fn zone_origin(rng: &mut impl Rng, viewport: &Viewport) -> Point {
    let zone = weighted_pick(rng, ZONE_WEIGHTS);
    let col = zone % 3;
    let row = zone / 3;
    let w3 = viewport.width / 3.0;
    let h3 = viewport.height / 3.0;
    let x_lo = col as f64 * w3;
    let y_lo = row as f64 * h3;
    Point {
        x: rng.random_range(x_lo..(x_lo + w3)),
        y: rng.random_range(y_lo..(y_lo + h3)),
    }
}

// ── HumanTouchscreen ────────────────────────────────────────────────────────

pub struct HumanTouchscreen<OT: ConnectionTransport> {
    touchscreen: Arc<Touchscreen<OT>>,
}

impl<OT: ConnectionTransport> HumanTouchscreen<OT> {
    pub fn new(touchscreen: Arc<Touchscreen<OT>>) -> Self {
        Self { touchscreen }
    }

    async fn swipe_internal(
        &self,
        from: Point,
        to: Point,
        duration_ms: u64,
        context: &BrowsingContext,
    ) -> Result<(), InputError> {
        let params = random_curve_params(from, to);
        let trajectory = generate_trajectory(from, to, &params);

        if trajectory.points.is_empty() {
            return Ok(());
        }

        let durations = generate_durations(
            trajectory.points.len(),
            duration_ms as f64 / 1000.0,
            (0.004, 0.025),
        );

        let first = trajectory.points[0];
        let handle = self.touchscreen
            .touch_start(first.x.max(0.0), first.y.max(0.0), context, None)
            .await?;

        for (i, pt) in trajectory.points.iter().enumerate().skip(1) {
            tokio::time::sleep(tokio::time::Duration::from_secs_f64(durations[i])).await;
            handle.move_to(pt.x.max(0.0), pt.y.max(0.0), context).await?;
        }

        let lift_delay = { let mut rng = rand::rng(); 10 + rng.random_range(0..30_u64) };
        tokio::time::sleep(tokio::time::Duration::from_millis(lift_delay)).await;

        handle.end(context).await?;
        Ok(())
    }
}

impl<OT: ConnectionTransport> Touch for HumanTouchscreen<OT> {
    async fn tap(
        &self,
        point: Point,
        context: &BrowsingContext,
    ) -> Result<(), InputError> {
        let handle = self.touchscreen.touch_start(point.x, point.y, context, None).await?;

        let hold = { let mut rng = rand::rng(); 50 + rng.random_range(0..60_u64) };
        tokio::time::sleep(tokio::time::Duration::from_millis(hold)).await;

        handle.end(context).await?;
        Ok(())
    }

    async fn swipe(
        &self,
        from: Point,
        to: Point,
        context: &BrowsingContext,
        options: SwipeOptions,
    ) -> Result<(), InputError> {
        let duration_ms = options.duration_ms.unwrap_or(600);
        self.swipe_internal(from, to, duration_ms, context).await
    }

    async fn scroll_to(
        &self,
        point: Point,
        viewport: &Viewport,
        context: &BrowsingContext,
        options: ScrollOptions,
    ) -> Result<(), InputError> {
        let duration_ms = options.duration_ms.unwrap_or(600);
        let origin = { let mut rng = rand::rng(); zone_origin(&mut rng, viewport) };
        self.swipe_internal(origin, point, duration_ms, context).await
    }

    async fn long_press(
        &self,
        point: Point,
        hold_ms: u64,
        context: &BrowsingContext,
    ) -> Result<(), InputError> {
        let handle = self.touchscreen.touch_start(point.x, point.y, context, None).await?;

        let jitter = { let mut rng = rand::rng(); rng.random_range(0..50_u64) };
        tokio::time::sleep(tokio::time::Duration::from_millis(hold_ms + jitter)).await;

        handle.end(context).await?;
        Ok(())
    }
}