webdriver-downloader-cli 0.5.13

Cli for webdriver download.
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

use webdriver_downloader::driver_impls::chromedriver_info::ChromedriverInfo;

use super::build_arg::*;
use super::check_arg::*;
use super::*;

pub async fn run() -> anyhow::Result<String> {
    let args = get_args()?;

    check_driver_install_path(&args)?;
    check_browser(&args)?;
    check_tries(&args)?;

    let install_result = match args.driver_type {
        DriverType::Chrome => {
            let driver_info = ChromedriverInfo::new(args.driver_install_path, args.browser_path);

            install(
                &driver_info,
                args.reinstall,
                args.skip_verification,
                args.num_tries,
            )
            .await
        }
        DriverType::Gecko => {
            let driver_info = GeckodriverInfo::new(args.driver_install_path, args.browser_path);

            install(
                &driver_info,
                args.reinstall,
                args.skip_verification,
                args.num_tries,
            )
            .await
        }
    };

    install_result.map_err(|e| e.into())
}

async fn install(
    driver_info: &impl WebdriverDownloadInfo,
    reinstall: bool,
    skip_verification: bool,
    num_tries: usize,
) -> Result<String, WebdriverDownloadError> {
    if !reinstall && driver_info.is_installed().await {
        Ok("Driver already installed.".to_string())
    } else {
        if skip_verification {
            driver_info.download_install().await?;
        } else {
            driver_info.download_verify_install(num_tries).await?;
        }

        Ok("Driver installed successfully.".to_string())
    }
}