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
|
use crate::Config;
use std::collections::HashMap;
use std::{thread, time};
use thirtyfour::common::capabilities::firefox::FirefoxPreferences;
use thirtyfour::error::{WebDriverError, WebDriverErrorInfo, WebDriverErrorValue};
use thirtyfour::prelude::{By, WebDriverResult};
use thirtyfour::{FirefoxCapabilities, WebDriver, WebDriverCommands};
/// This url is used to make the login
const LOGIN_URL: &str = "https://studenti.smartedu.unict.it/WorkFlow2011/Logon/Logon.aspx?ReturnUrl=%2fStudenti%2fDefault.aspx";
/// This url is used to go to the page where a student can book a room for study
const ROOMS_URL: &str = "https://studenti.smartedu.unict.it/StudentSpaceReserv?Type=unaTantum";
/// Browser struct
pub struct Browser {
/// The driver for Firefox, it could be `None`
driver: Option<WebDriver>,
}
impl Browser {
/// Create a new `Browser` with a Firefox driver with a personalized User-Agent
pub async fn new(driver_url: &String) -> Self {
let user_agent =
"Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:90.0) Gecko/20100101 Firefox/90.0";
let mut prefs = FirefoxPreferences::new();
let _ = prefs.set_user_agent(user_agent.to_string());
let mut caps = FirefoxCapabilities::new();
let _ = caps.set_preferences(prefs);
Self {
driver: Some(WebDriver::new(driver_url, caps).await.unwrap()),
}
}
// TODO: Close the driver fixing the borrowing error
// async fn _close(self) -> Result<(), Box<dyn Error>> {
// Ok(())
// }
/// Login on `LOGIN_URL`
pub async unsafe fn _login(&self, credentials: &Config) -> WebDriverResult<()> {
if let Some(_d) = &self.driver {
_d.get(LOGIN_URL).await?;
let cf_input = _d.find_element(By::Name("ctl01$contents$UserName")).await?;
cf_input.send_keys(&credentials.cf).await?;
thread::sleep(time::Duration::from_millis(3000));
let psw_input = _d
.find_element(By::Name("ctl01$contents$UserPassword"))
.await?;
psw_input.send_keys(&credentials.password).await?;
thread::sleep(time::Duration::from_millis(3000));
_d.find_element(By::Name("ctl01$contents$LogonButton"))
.await?
.click()
.await?;
thread::sleep(time::Duration::from_millis(2000));
// If the current url is the same as `LOGIN_URL` it means the login didn't work, so
// returns a "login error"
if _d.current_url().await? == LOGIN_URL {
return Err(WebDriverError::SessionNotCreated(WebDriverErrorInfo {
status: 400,
error: "SessionNotCreated".to_string(),
value: WebDriverErrorValue {
message: "SessionNotCreated".to_string(),
error: None,
stacktrace: None,
data: None,
},
}));
}
}
Ok(())
}
/// Get all faculties for booking and return an `HashMap<key, value>` when `key` is the
/// key for that faculty inside the `select` tag and `value` is just the text of the option.
pub async fn faculties(&self) -> WebDriverResult<Option<HashMap<String, String>>> {
if let Some(_d) = &self.driver {
_d.get(ROOMS_URL).await?;
thread::sleep(time::Duration::from_millis(1000));
_d.find_element(By::Css(
"span[aria-labelledby='select2-dipartimentoSelector-container']",
))
.await?
.click()
.await?;
let list_elements = _d
.find_elements(By::Css("#select2-dipartimentoSelector-results li"))
.await?;
let mut faculties_ids = HashMap::<String, String>::new();
for i in list_elements {
faculties_ids.insert(
i.get_attribute("data-select2-id").await.unwrap().unwrap(),
i.text().await.unwrap(),
);
}
return Ok(Some(faculties_ids));
}
Ok(None)
}
}
/// The static unsafe variable used to open a web browser
pub static mut WEB_BROWSER: Option<Browser> = None;
|