feat: browser add user-agent

src/cli/cmd.rs

@@ -45,6 +45,11 @@ pub fn cli() -> Result<(), Box<dyn std::error::Error>> {
         ])
         .get_matches();
 
+    let buf = std::env::current_dir()
+        .unwrap()
+        .join("files/user_agent.toml");
+    std::env::set_var("user_agent", buf);
+
     if app.subcommand().is_none() {
         return crawler::browse_wikipedia(LaunchOptions::default());
     }

src/handler/crawler.rs

@@ -1,10 +1,19 @@
 use headless_chrome::{Browser, LaunchOptions};
 
+use crate::common;
+
 pub fn browse_wikipedia(launch_options: LaunchOptions) -> Result<(), Box<dyn std::error::Error>> {
     let browser = Browser::new(launch_options)?;
     let tab = browser.new_tab()?;
+    let random_ug = common::user_agent::random_user_agent();
+    tab.set_user_agent(random_ug.as_str(), None, None).unwrap();
     tab.navigate_to("https://example.com")?;
     let h1 = tab.wait_for_xpath("/html/body/div/h1")?;
     assert_eq!(h1.get_inner_text().unwrap().as_str(), "Example Domain");
+    let ug = tab
+        .evaluate("window.navigator.userAgent", false)?
+        .value
+        .unwrap();
+    assert_eq!(random_ug, ug);
     Ok(())
 }