webdriver-downloader 0.17.0

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

use std::io;

use async_trait::async_trait;
use tempfile::TempDir;

use crate::os_specific::DefaultPathError;
use crate::traits::installation_info::{InstallationError, WebdriverInstallationInfo};
use crate::traits::url_info::{UrlError, WebdriverUrlInfo};
use crate::traits::verification_info::{VerificationError, WebdriverVerificationInfo};

/// Information required to download, verify, install driver.
///
/// This trait is implemented for all types that implement [`WebdriverUrlInfo`], [`WebdriverInstallationInfo`] and [`WebdriverVerificationInfo`].
#[async_trait]
pub trait WebdriverDownloadInfo:
    WebdriverUrlInfo + WebdriverInstallationInfo + WebdriverVerificationInfo + Sync
{
    /// Check if the driver is installed.
    async fn is_installed(&self) -> bool;

    /// Download, install driver. Skips verification process.
    async fn download_install(&self) -> Result<(), WebdriverDownloadError>;

    /// Download, verify, install driver.
    async fn download_verify_install(&self, max_tries: usize)
        -> Result<(), WebdriverDownloadError>;
}

/// Error that can occur when installing and verifying driver.
#[derive(thiserror::Error, Debug)]
pub enum WebdriverDownloadError {
    #[error(transparent)]
    Url(#[from] UrlError),
    #[error(transparent)]
    Install(#[from] InstallationError),
    #[error(transparent)]
    Verify(#[from] VerificationError),
    #[error(transparent)]
    DefaultPath(#[from] DefaultPathError),
    #[error("Failed to move driver to driver_path: {0}")]
    Move(#[from] io::Error),
    #[error("Tried {0} possible versions, but no version passed verification.")]
    NoVersionPassedVerification(usize),
    #[error(transparent)]
    Other(#[from] anyhow::Error),
}

#[async_trait]
impl<T> WebdriverDownloadInfo for T
where
    T: WebdriverUrlInfo + WebdriverInstallationInfo + WebdriverVerificationInfo + Sync,
{
    async fn is_installed(&self) -> bool {
        let driver_path = self.driver_install_path();
        self.verify_driver(&driver_path).await.is_ok()
    }

    async fn download_install(&self) -> Result<(), WebdriverDownloadError> {
        let mut version_urls = self.version_urls(1).await?;
        let version_url = version_urls
            .pop()
            .ok_or(WebdriverDownloadError::NoVersionPassedVerification(0))?;

        println!(
            "Trying url for version {}: {}.",
            version_url.webdriver_version, version_url.url
        );

        let tempdir = TempDir::new()?;

        let temp_driver_path = self.download_in_tempdir(version_url.url, &tempdir).await?;

        self.install_driver(&temp_driver_path)?;

        Ok(())
    }

    async fn download_verify_install(
        &self,
        max_tries: usize,
    ) -> Result<(), WebdriverDownloadError> {
        let version_urls = self.version_urls(max_tries).await?;
        let url_count = version_urls.len();

        for version_url in version_urls {
            println!(
                "Trying url for version {}: {}.",
                version_url.webdriver_version, version_url.url
            );
            let tempdir = TempDir::new()?;

            let temp_driver_path = self.download_in_tempdir(version_url.url, &tempdir).await?;

            match self.verify_driver(&temp_driver_path).await {
                Ok(_) => {
                    self.install_driver(&temp_driver_path)?;
                    return Ok(());
                }
                Err(e) => {
                    println!("Verification failed: {}.", e)
                }
            }
        }

        Err(WebdriverDownloadError::NoVersionPassedVerification(
            url_count,
        ))
    }
}