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

use serde::Serialize;
use serde_json::Value;

use crate::error::WebDriverResult;

/// Helper struct for constructing arguments for `WebDriver::execute_script()`
/// and `WebDriver::execute_async_script()`.
///
/// See the examples for those methods for more info.
#[derive(Debug, Default, Clone)]
pub struct ScriptArgs {
    values: Vec<Value>,
}

impl ScriptArgs {
    /// Create a new, empty ScriptArgs struct.
    pub fn new() -> Self {
        ScriptArgs::default()
    }

    /// Push a JSON value onto the vec.
    pub fn push_value(&mut self, value: Value) -> &mut Self {
        self.values.push(value);
        self
    }

    /// Push any Serialize-able object onto the vec.
    /// This includes WebElement.
    pub fn push<T>(&mut self, value: T) -> WebDriverResult<&mut Self>
    where
        T: Serialize,
    {
        Ok(self.push_value(serde_json::to_value(value)?))
    }

    /// Get the args vec. This is used internally.
    pub fn get_args(&self) -> Vec<Value> {
        self.values.clone()
    }
}