feat: get faculties for the /room command

2 files changed, 51 insertions, 0 deletions

diff --git a/src/browser/mod.rs b/src/browser/mod.rs
index 8359c51..2765c6d 100644
--- a/src/browser/mod.rs
+++ b/src/browser/mod.rs
@@ -1,3 +1,5 @@
+use std::collections::HashMap;
+
 use self::web_browser::{Browser, WEB_BROWSER};
 use crate::Config;
 use thirtyfour::prelude::WebDriverResult;
@@ -17,3 +19,18 @@ pub async unsafe fn login(credentials: &Config) -> WebDriverResult<()> {
 
     Ok(())
 }
+
+pub async unsafe fn get_faculties() -> WebDriverResult<Option<HashMap<String, String>>> {
+    if let Some(driver) = &WEB_BROWSER {
+        match driver.faculties().await? {
+            Some(faculties) => {
+                return Ok(Some(faculties));
+            }
+            None => {
+                return Ok(Some(HashMap::<String, String>::new()));
+            }
+        };
+    }
+
+    Ok(Some(HashMap::<String, String>::new()))
+}
diff --git a/src/browser/web_browser.rs b/src/browser/web_browser.rs
index 804c82f..d2ba6c4 100644
--- a/src/browser/web_browser.rs
+++ b/src/browser/web_browser.rs
@@ -1,10 +1,13 @@
 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};
 
 const LOGIN_URL: &str = "https://studenti.smartedu.unict.it/WorkFlow2011/Logon/Logon.aspx?ReturnUrl=%2fStudenti%2fDefault.aspx";
+const ROOMS_URL: &str = "https://studenti.smartedu.unict.it/StudentSpaceReserv?Type=unaTantum";
 
 pub struct Browser {
     driver: Option<WebDriver>,
@@ -70,6 +73,37 @@ impl Browser {
 
         Ok(())
     }
+
+    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)
+    }
 }
 
 pub static mut WEB_BROWSER: Option<Browser> = None;