feat: get timetable of a space

3 files changed, 94 insertions, 1 deletions

diff --git a/src/browser/mod.rs b/src/browser/mod.rs
index 82c9c60..4c7ef8e 100644
--- a/src/browser/mod.rs
+++ b/src/browser/mod.rs
@@ -61,3 +61,14 @@ pub async unsafe fn select_option(
 
     Ok(false)
 }
+
+/// Get the timetable of available rooms
+pub async unsafe fn get_timetable() -> WebDriverResult<Option<HashMap<String, String>>> {
+    if let Some(driver) = &WEB_BROWSER {
+        if let Some(timetable) = driver.get_timetable().await? {
+            return Ok(Some(timetable));
+        }
+    }
+
+    Ok(None)
+}
diff --git a/src/browser/web_browser.rs b/src/browser/web_browser.rs
index 34e78d1..5a855d5 100644
--- a/src/browser/web_browser.rs
+++ b/src/browser/web_browser.rs
@@ -4,7 +4,7 @@ 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};
+use thirtyfour::{FirefoxCapabilities, WebDriver, WebDriverCommands, WebElement};
 
 /// 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";
@@ -148,6 +148,46 @@ impl Browser {
 
         Ok(false)
     }
+
+    /// Get the timetable of available hours from the driver and returns an hashmap with id->text
+    pub async fn get_timetable(&self) -> WebDriverResult<Option<HashMap<String, String>>> {
+        if let Some(_d) = &self.driver {
+            thread::sleep(time::Duration::from_millis(2000));
+
+            let table_trs = _d
+                .find_elements(By::Css("div[data-select2-id='studyPlan'] table tbody tr"))
+                .await?;
+
+            let mut timetable = HashMap::<String, String>::new();
+
+            for i in table_trs.iter() {
+                let cols: Vec<WebElement> = i.find_elements(By::Css("td")).await?;
+                if cols.len() < 6 {
+                    continue;
+                }
+
+                let col_id = i.find_element(By::Css("th")).await?.text().await.unwrap();
+                let mut text_formatted = cols[0].text().await.unwrap();
+                text_formatted.push_str(
+                    &format!(
+                        ", {} - {}.\n",
+                        cols[1].text().await.unwrap(),
+                        cols[2].text().await.unwrap()
+                    )
+                    .to_owned()[..],
+                );
+
+                text_formatted
+                    .push_str(&format!("Posti: {}", cols[6].text().await.unwrap()).to_owned()[..]);
+
+                timetable.insert(col_id, text_formatted);
+            }
+
+            return Ok(Some(timetable));
+        }
+
+        Ok(None)
+    }
 }
 
 /// The static unsafe variable used to open a web browser
diff --git a/src/callbacks.rs b/src/callbacks.rs
index a82d5e9..53ae07c 100644
--- a/src/callbacks.rs
+++ b/src/callbacks.rs
@@ -49,6 +49,48 @@ pub async fn handler(
                 }
             }
         },
+        "space" => unsafe {
+            // Select the sapce
+            match browser::select_option("select2-results__option", "data-select2-id", call[2])
+                .await
+            {
+                Ok(result) => {
+                    if result {
+                        let timetable = browser::get_timetable().await.unwrap();
+                        let keyboard = make_inline_keyboard(
+                            &timetable,
+                            "timetable",
+                            chat_id.parse::<i64>().unwrap(),
+                        )
+                        .await;
+
+                        // Edit the previous spaces message with timetable' buttons
+                        cx.requester
+                            .edit_message_text(
+                                chat_id.to_string(),
+                                cx.update.message.clone().unwrap().id,
+                                "When?",
+                            )
+                            .await?;
+
+                        cx.requester
+                            .edit_message_reply_markup(
+                                chat_id.to_string(),
+                                cx.update.message.clone().unwrap().id,
+                            )
+                            .reply_markup(keyboard)
+                            .await?;
+
+                        return Ok(true);
+                    } else {
+                        return Ok(false);
+                    }
+                }
+                Err(_) => {
+                    return Ok(false);
+                }
+            }
+        },
         _ => Ok(false),
     }
 }