From 429489349688ce452f124a691c7542d2ca2177d1 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Thu, 5 Aug 2021 19:00:37 +0200 Subject: feat: handle the callback of the faculty --- src/browser/mod.rs | 16 ++++++++++++++++ src/browser/web_browser.rs | 21 +++++++++++++++++++++ src/callbacks.rs | 18 ++++++++++++++++++ src/main.rs | 5 +++-- 4 files changed, 58 insertions(+), 2 deletions(-) create mode 100644 src/callbacks.rs (limited to 'src') diff --git a/src/browser/mod.rs b/src/browser/mod.rs index 7205f27..1675595 100644 --- a/src/browser/mod.rs +++ b/src/browser/mod.rs @@ -34,3 +34,19 @@ pub async unsafe fn get_faculties() -> WebDriverResult WebDriverResult { + if let Some(driver) = &WEB_BROWSER { + let result = driver + .select_option_from_list(klass, property_name, property_value) + .await?; + + return Ok(result); + } + + Ok(false) +} diff --git a/src/browser/web_browser.rs b/src/browser/web_browser.rs index c7497cf..ffbdc13 100644 --- a/src/browser/web_browser.rs +++ b/src/browser/web_browser.rs @@ -114,6 +114,27 @@ impl Browser { Ok(None) } + + /// Select an option from a list of select elements + pub async fn select_option_from_list( + &self, + klass: &str, + property_name: &str, + property_value: &str, + ) -> WebDriverResult { + if let Some(_d) = &self.driver { + _d.find_element(By::Css( + &format!("li.{}[{}='{}']", klass, property_name, property_value).to_owned()[..], + )) + .await? + .click() + .await?; + + return Ok(true); + } + + Ok(false) + } } /// The static unsafe variable used to open a web browser diff --git a/src/callbacks.rs b/src/callbacks.rs new file mode 100644 index 0000000..c670884 --- /dev/null +++ b/src/callbacks.rs @@ -0,0 +1,18 @@ +use crate::browser; + +/// Handle the string of callback data +pub async fn handler(text: &String) { + let call: Vec<&str> = text.split("_").collect(); + + // First part of `_` string is the type of callback we want to handle + let type_ = call[0]; + + match type_ { + "faculty" => unsafe { + // Select the faculty + let _ = + browser::select_option("select2-results__option", "data-select2-id", call[1]).await; + }, + _ => {} + }; +} diff --git a/src/main.rs b/src/main.rs index c038a9b..79b6e6e 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,6 +1,7 @@ use std::error::Error; use teloxide::prelude::*; mod browser; +mod callbacks; mod commands; mod config; @@ -41,8 +42,8 @@ async fn main() -> Result<(), Box> { .callback_queries_handler(|rx: DispatcherHandlerRx, CallbackQuery>| { UnboundedReceiverStream::new(rx).for_each_concurrent(None, |cx| async move { let data = &cx.update.data; - if let Some(d) = data { - println!("{}", d); + if let Some(text) = data { + callbacks::handler(text).await; } }) }) -- cgit v1.2.3-18-g5258