Merge pull request #6 from KUACC-VALAR-HPC-KOC-UNIVERSITY/extra-pages-handling

Extra pages handling

Author: ealtun21

Cargo.lock

@@ -1,17 +1,24 @@
 [package]
 name = "kuvpn"
-version = "2.0.0"
+version = "2.0.1"
 edition = "2021"
 description = "A convenient tool for connecting to Koç University's VPN using OpenConnect."
 repository = "https://github.com/KUACC-VALAR-HPC-KOC-UNIVERSITY/kuvpn"
 license = "MIT"
 
 [dependencies]
 anyhow = "1.0.99"
-clap = { version = "4.5.46", features = ["derive"] }
+clap = { version = "4.5.47", features = ["derive"] }
 colored = "3.0.0"
 env_logger = "0.11.8"
-headless_chrome = { version = "1.0.17", features = ["fetch", "rustls"] }
+headless_chrome = { version = "1.0.18", features = ["fetch", "rustls"] }
 log = "0.4.27"
 rpassword = "7.4.0"
 which = "8.0.0"
+
+[profile.release]
+opt-level = "z"           # Minimize size (use "z" or "s" for small codegen)
+lto = true                # Enable link-time optimization
+panic = "abort"           # Abort on panic (smaller than "unwind")
+strip = true              # Strip debug symbols from build (requires Rust 1.70+)
+codegen-units = 1         # Single codegen unit for better LTO of small binaries

Cargo.toml

@@ -1,4 +1,4 @@
-# KUVPN v2.0.0
+# KUVPN v2.0.1
 
 KUVPN is a simple CLI tool written in Rust that retrieves the DSID cookie and executes the OpenConnect command to connect to the VPN for Koç University.
 
@@ -23,6 +23,7 @@ KUVPN is a simple CLI tool written in Rust that retrieves the DSID cookie and ex
 - ✅ **Retrieve DSID cookie**
 - ✅ **Execute OpenConnect command**
 - ✅ **Headless Execution**
+- ✅ **Automatic or Manuel Login Navigation**
 - ✅ **Session management**
 - ✅ **CLI Based**
 - ✅ **Openconnect Wrapper**
@@ -149,6 +150,9 @@ Options:
 
           [default: openconnect]
 
+      --no-auto-login
+          Disable automatic login handlers and only poll for DSID in a headful browser
+
   -h, --help
           Print help (see a summary with '-h')
 
@@ -170,6 +174,16 @@ You can clear your session like this:
 kuvpn -c
 ```
 
+### Troubleshooting: Unable to login?
+
+If you aren't able to login via the cli'a automation, you can disable the auto login so that you can do it via the browser's gui.
+To do this run the following command:
+```
+kuvpn --no-auto-login --disable-headless
+```
+
+Later, once you login, it will remember your session so you can login without the need for this, on the next logins.
+
 ## License
 
 This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for details.

README.md

@@ -52,4 +52,8 @@ pub struct Args {
     /// Can be a relative or absolute path.
     #[arg(long, default_value = "openconnect")]
     pub openconnect_path: String,
+
+    /// Disable automatic login handlers and only poll for DSID in a headful browser
+    #[arg(long, default_value_t = false)]
+    pub no_auto_login: bool,
 }

src/args.rs

@@ -0,0 +1,57 @@
+use crate::utils::get_user_data_dir;
+use headless_chrome::browser::default_executable;
+use headless_chrome::{Browser, LaunchOptions};
+use std::error::Error;
+use std::ffi::OsString;
+use std::time::Duration;
+
+/// Creates a browser instance configured with a blank page, a custom user agent,
+/// and a dedicated user data directory.
+///
+/// The browser is launched with the following settings:
+/// - **Headless or non-headless mode** based on the `headless` parameter.
+/// - **A blank app window** containing a minimal HTML page.
+/// - **Custom user agent** as specified by the `agent` parameter.
+/// - **Custom user data directory** for isolated session data.
+/// - **Custom window size** of 800x800 pixels.
+/// - **Sandbox disabled** and an infinite idle timeout.
+///
+/// # Arguments
+///
+/// * `agent` - A string slice representing the desired user agent.
+/// * `headless` - Whether to run the browser in headless mode.
+///
+/// # Returns
+///
+/// A `Result` containing the configured `Browser` instance on success, or an error on failure.
+pub fn create_browser(agent: &str, headless: bool) -> Result<Browser, Box<dyn Error>> {
+    // Retrieve the user data directory.
+    let user_data_dir = get_user_data_dir()?;
+
+    // Build the command-line arguments for launching the browser.
+    let user_agent = OsString::from(format!("--user-agent={agent}"));
+    let body = OsString::from("--app=data:text/html,");
+    let window = OsString::from("--new-window");
+
+    // Configure the browser launch options.
+    let mut options = LaunchOptions::default_builder();
+    let mut launch_options = options
+        .headless(headless)
+        .sandbox(false)
+        .idle_browser_timeout(Duration::MAX)
+        .window_size(Some((800, 800)))
+        .args(vec![
+            body.as_os_str(),
+            window.as_os_str(),
+            user_agent.as_os_str(),
+        ])
+        .user_data_dir(Some(user_data_dir));
+
+    // If the default Chrome executable is available, specify its path.
+    if let Ok(executable_path) = default_executable() {
+        launch_options = launch_options.path(Some(executable_path));
+    }
+
+    // Build and return the Browser instance.
+    Ok(Browser::new(launch_options.build()?)?)
+}