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

use crate::DriverFetcher;
use eyre::{eyre, Result};
use url::Url;

pub struct Geckodriver;

impl DriverFetcher for Geckodriver {
    const BASE_URL: &'static str = "https://github.com/mozilla/geckodriver/releases";

    /// Returns the latest version of the driver
    fn latest_version(&self) -> Result<String> {
        let latest_release_url = format!("{}/latest", Self::BASE_URL);
        let resp = reqwest::blocking::get(&latest_release_url)?;
        let url = resp.url();
        Ok(url.path_segments().unwrap().last().unwrap().to_string())
    }

    /// Returns the download url for the driver executable
    fn direct_download_url(&self, version: &str) -> Result<Url> {
        Ok(Url::parse(&format!(
            "{}/download/{version}/geckodriver-{version}-{platform}",
            Self::BASE_URL,
            version = version,
            platform = Self::platform()?
        ))?)
    }
}

impl Geckodriver {
    pub fn new() -> Self {
        Self {}
    }

    fn platform() -> Result<String> {
        match sys_info::os_type()?.as_str() {
            "Linux" => Ok(format!("linux{}.tar.gz", Self::pointer_width())),
            "Darwin" => Ok(String::from("macos.tar.gz")),
            "Windows" => Ok(format!("win{}.zip", Self::pointer_width())),
            other => Err(eyre!(
                "webdriver-install doesn't support '{}' currently",
                other
            )),
        }
    }

    const fn pointer_width() -> usize {
        #[cfg(target_pointer_width = "32")]
        {
            32
        }
        #[cfg(target_pointer_width = "64")]
        {
            64
        }
    }
}

#[test]
fn direct_download_url_test() {
    #[cfg(target_os = "linux")]
    assert_eq!(
        "https://github.com/mozilla/geckodriver/releases/download/v1/geckodriver-v1-linux64.tar.gz",
        Geckodriver::new()
            .direct_download_url("v1")
            .unwrap()
            .to_string()
    );
    #[cfg(target_os = "macos")]
    assert_eq!(
        "https://github.com/mozilla/geckodriver/releases/download/v1/geckodriver-v1-macos.tar.gz",
        Geckodriver::new()
            .direct_download_url("v1")
            .unwrap()
            .to_string()
    );
    #[cfg(target_os = "windows")]
    assert_eq!(
        "https://github.com/mozilla/geckodriver/releases/download/v1/geckodriver-v1-win64.zip",
        Geckodriver::new()
            .direct_download_url("v1")
            .unwrap()
            .to_string()
    );
}