feat(page): add wait_for_network_idle

src/page.rs

@@ -303,6 +303,65 @@ impl Page {
         Ok(self)
     }
 
+    /// Wait for the network to be idle for 500ms
+    #[cfg(feature = "tokio-runtime")]
+    pub async fn wait_for_network_idle(&self) -> Result<&Self> {
+        let mut events = self.event_listener::<chromiumoxide_cdp::cdp::browser_protocol::network::EventLoadingFinished>().await?;
+
+        loop {
+            let sleep = tokio::time::sleep(tokio::time::Duration::from_millis(500));
+            tokio::pin!(sleep);
+            tokio::select! {
+                _ = &mut sleep => break,
+                _ = events.next() => (),
+                else => break,
+            }
+        }
+
+        Ok(self)
+    }
+
+    /// Wait for the network to be idle for 500ms
+    #[cfg(feature = "async-std-runtime")]
+    pub async fn wait_for_network_idle(&self) -> Result<&Self> {
+        use futures::{future::FutureExt, pin_mut, select};
+        let mut events = self.event_listener::<chromiumoxide_cdp::cdp::browser_protocol::network::EventLoadingFinished>().await?;
+
+        loop {
+            let t1 = async_std::task::sleep(std::time::Duration::from_millis(500)).fuse();
+            let t2 = events.next().fuse();
+
+            pin_mut!(t1, t2);
+
+            select! {
+                () = t1 => break,
+                _ = t2 => (),
+            }
+        }
+
+        Ok(self)
+    }
+
+    /// Wait for the network to be idle for 500ms with a duration timeout
+    #[cfg(feature = "tokio-runtime")]
+    pub async fn wait_for_network_idle_with_timeout(
+        &self,
+        timeout: core::time::Duration,
+    ) -> Result<&Self> {
+        if let Err(_) = tokio::time::timeout(timeout, self.wait_for_network_idle()).await {}
+        Ok(self)
+    }
+
+    /// Wait for the network to be idle for 500ms with a duration timeout
+    #[cfg(feature = "async-std-runtime")]
+    pub async fn wait_for_network_idle_with_timeout(
+        &self,
+        timeout: core::time::Duration,
+    ) -> Result<&Self> {
+        let _ = (async_std::future::timeout(timeout, self.wait_for_network_idle()).await).is_err();
+        Ok(self)
+    }
+
     /// Navigate directly to the given URL.
     ///
     /// This resolves directly after the requested URL is fully loaded.