90 lines
2.8 KiB
Rust
90 lines
2.8 KiB
Rust
//! Login to the NZZ archive and print the authentication cookie for further usage.
|
|
//!
|
|
//! Leveraging webdriver + geckodriver.
|
|
|
|
use std::time::Duration;
|
|
|
|
use anyhow::Result;
|
|
use cli::Config;
|
|
use fantoccini::{elements::Element, Client, ClientBuilder, Locator};
|
|
use serde_json::json;
|
|
use tokio::{sync::oneshot, time::sleep};
|
|
|
|
use crate::geckodriver::GECKODRIVER_HOST;
|
|
|
|
pub mod cli;
|
|
pub mod geckodriver;
|
|
|
|
const LOGIN_URL: &str = "https://zeitungsarchiv.nzz.ch/";
|
|
|
|
/// Entrypoint to login to the NZZ archive.
|
|
pub async fn run(args: Config, pw: &str) -> Result<()> {
|
|
let (stop_tx, stop_rx) = oneshot::channel();
|
|
let driver_handle = geckodriver::run(stop_rx).await?;
|
|
|
|
let driver_args = json!({ "moz:firefoxOptions": {"args": ["-headless"]} });
|
|
|
|
let client = ClientBuilder::native()
|
|
.capabilities(driver_args.as_object().unwrap().clone())
|
|
.connect(&format!("http://{GECKODRIVER_HOST}"))
|
|
.await?;
|
|
|
|
client.goto(LOGIN_URL).await?;
|
|
|
|
sleep(Duration::from_millis(500)).await;
|
|
let fu_open_button: Element = element_from_css(&client, ".cmpboxbtncustom").await?;
|
|
fu_open_button.click().await?;
|
|
|
|
let fu_button: Element = element_from_css(&client, ".cmpboxbtn.cmpboxbtnyes.cmpboxbtnyescustomchoices.cmptxt_btn_save").await?;
|
|
sleep(Duration::from_millis(500)).await;
|
|
fu_button.click().await?;
|
|
|
|
|
|
let login_button: Element = element_from_css(&client, ".fup-menu-login-container").await?;
|
|
sleep(Duration::from_millis(500)).await;
|
|
login_button.click().await?;
|
|
|
|
let login_iframe = element_from_css(&client, r#"iframe[id^="piano""#).await?;
|
|
login_iframe.enter_frame().await?;
|
|
|
|
let email_input = element_from_css(&client, r#"input[name="email"]"#).await?;
|
|
email_input.send_keys(&args.username).await?;
|
|
|
|
let pw_input: Element = element_from_css(&client, r#"input[type="password"]"#).await?;
|
|
pw_input.send_keys(pw).await?;
|
|
|
|
let submit = element_from_css(&client, r#"button[class="btn prime"]"#).await?;
|
|
submit.click().await?;
|
|
|
|
let main_frame = client.window().await?;
|
|
client.switch_to_window(main_frame).await?;
|
|
|
|
element_from_css(&client, ".fup-login-open.fup-button.fup-s-menu-login-open").await?;
|
|
|
|
let cookies = client.get_all_cookies().await?;
|
|
let cobbled_cookies = cookies
|
|
.into_iter()
|
|
.map(|cookie| format!("{}={}", cookie.name(), cookie.value()))
|
|
.fold(String::new(), |mut acc, word| {
|
|
if !acc.is_empty() {
|
|
acc.push(';');
|
|
}
|
|
acc.push_str(&word);
|
|
acc
|
|
});
|
|
println!("{cobbled_cookies}");
|
|
|
|
client.close().await?;
|
|
|
|
if let Some(driver_handle) = driver_handle {
|
|
let _ = stop_tx.send(());
|
|
driver_handle.abort();
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn element_from_css(client: &Client, selector: &str) -> Result<Element> {
|
|
Ok(client.wait().for_element(Locator::Css(selector)).await?)
|
|
}
|