From dce9360485622d64e43a44c920095bd0ea7965da Mon Sep 17 00:00:00 2001
From: j-mendez <jeff@a11ywatch.com>
Date: Fri, 19 Jan 2024 22:29:26 -0500
Subject: [PATCH] feat(chrome): add with_wait_for_idle_network builder config

---
 Cargo.toml     |  6 +++---
 src/website.rs | 22 ++++++++++++++++++++++
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 7026eda..b7fb6e2 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 edition = "2021"
 name = "spider_rs"
-version = "0.0.16"
+version = "0.0.17"
 description = "The fastest web crawler written in Rust ported to nodejs."
 repository = "https://github.com/spider-rs/spider-nodejs"
 
@@ -11,8 +11,8 @@ crate-type = ["cdylib"]
 [dependencies]
 indexmap = "2.1.0"
 num_cpus = "1.16.0"
-spider = { version = "1.80.68", features = ["budget", "cron", "regex", "cookies", "socks", "chrome", "control", "smart", "chrome_intercept", "cache" ] }
-pyo3 = { version = "0.20.0", features = ["extension-module"] }
+spider = { version = "1.80.75", features = ["budget", "cron", "regex", "cookies", "socks", "chrome", "control", "smart", "chrome_intercept", "cache" ] }
+pyo3 = { version = "0.20.2", features = ["extension-module"] }
 pyo3-asyncio = { version = "0.20", features = ["attributes", "tokio-runtime"] }
 
 [target.x86_64-unknown-linux-gnu.dependencies]
diff --git a/src/website.rs b/src/website.rs
index e2b98bf..c2a6720 100644
--- a/src/website.rs
+++ b/src/website.rs
@@ -2,6 +2,7 @@ use crate::{new_page, NPage, BUFFER};
 use indexmap::IndexMap;
 use pyo3::prelude::*;
 use spider::compact_str::CompactString;
+use spider::configuration::WaitForIdleNetwork;
 use spider::tokio::select;
 use spider::tokio::task::JoinHandle;
 use spider::utils::shutdown;
@@ -781,6 +782,27 @@ impl Website {
     slf
   }
 
+  /// Wait for idle network request. This method does nothing if the `chrome` feature is not enabled.
+  /// Set the timeout to 0 to disable the timeout.
+  pub fn with_wait_for_idle_network(
+    mut slf: PyRefMut<'_, Self>,
+    wait_for_idle_network: bool,
+    timeout: u64,
+  ) -> PyRefMut<'_, Self> {
+    slf
+      .inner
+      .with_wait_for_idle_network(if wait_for_idle_network {
+        Some(WaitForIdleNetwork::new(if timeout == 0 {
+          None
+        } else {
+          Some(core::time::Duration::from_millis(timeout))
+        }))
+      } else {
+        None
+      });
+    slf
+  }
+
   /// Setup cron jobs to run
   pub fn with_cron(
     mut slf: PyRefMut<'_, Self>,