Handle EU consent popup

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "gaboja"
-version = "0.4.0"
+version = "0.4.1"
 edition = "2021"
 license = "MIT"
 description = "Gaboja: CLI helper for solving BOJ problems"
@@ -23,6 +23,6 @@ regex = "1.10.4"
 serde = "1.0.203"
 serde_json = "1.0.117"
 similar = { version = "2.5.0", features = ["inline"] }
-thirtyfour = "0.32.0"
+thirtyfour = "0.33.0"
 tokio = { version = "1.37.0", features = ["time", "process", "rt"] }
 toml = "0.8.13"

src/command.rs

@@ -60,6 +60,9 @@ pub(crate) enum Command {
     Help,
     Exit,
     Shell(String),
+    DebugCache,
+    DebugScreenshot,
+    DebugSource,
 }
 
 #[derive(Debug, Clone)]

src/command/executor.rs

@@ -167,6 +167,16 @@ impl GlobalState {
             Command::Shell(shell_cmd) => {
                 run_interactive(shell_cmd)?;
             }
+            Command::DebugCache => {
+                println!("{:?}", self.problem_cache);
+            }
+            Command::DebugScreenshot => {
+                self.browser.screenshot()?;
+            }
+            Command::DebugSource => {
+                let source = self.browser.source()?;
+                std::fs::write("./source", &source)?;
+            }
         }
         Ok(())
     }

src/command/parser.rs

@@ -317,6 +317,9 @@ impl std::str::FromStr for Command {
                 Ok(Self::Exit)
             }
             "help" => Ok(Self::Help),
+            "debugcache" => Ok(Self::DebugCache),
+            "debugscreenshot" => Ok(Self::DebugScreenshot),
+            "debugsource" => Ok(Self::DebugSource),
             _ => Err(CommandParseError {
                 msg: format!("Unknown command `{}`", main_cmd),
             }),

src/infra/browser.rs

@@ -2,6 +2,7 @@ use crate::data::{ExampleIO, Problem, ProblemId, ProblemKind};
 use crate::infra::console::Spinner;
 use crate::infra::subprocess::{spawn_cmd_background, run_silent};
 use std::future::Future;
+use std::path::Path;
 use std::process::Stdio;
 use thirtyfour::common::cookie::SameSite;
 use thirtyfour::prelude::*;
@@ -103,6 +104,10 @@ impl Browser {
             let driver = &self.webdriver;
             let problem_page = problem_id.problem_url();
             driver.get(problem_page).await?;
+            let fc_cta_consent = driver.find_all(By::ClassName("fc-cta-consent")).await?;
+            if let Some(cta_el) = fc_cta_consent.first() {
+                cta_el.click().await?;
+            }
             let title = driver.find(By::Id("problem_title")).await?.text().await?;
             let label_elems = driver.find_all(By::ClassName("problem-label")).await?;
             let mut kind = vec![];
@@ -216,6 +221,22 @@ impl Browser {
         })
     }
 
+    pub(crate) fn screenshot(&self) -> anyhow::Result<()> {
+        with_async_runtime(async {
+            let driver = &self.webdriver;
+            driver.screenshot(Path::new("./screenshot.png")).await?;
+            Ok(())
+        })
+    }
+
+    pub(crate) fn source(&self) -> anyhow::Result<String> {
+        with_async_runtime(async {
+            let driver = &self.webdriver;
+            let source = driver.source().await?;
+            Ok(source)
+        })
+    }
+
     /// Gracefully terminate the browser. Should be called even on error.
     pub(crate) fn quit(self) -> anyhow::Result<()> {
         with_async_runtime(async {