From 83284cd99bc77232e934c0eb505b7def2410c8d2 Mon Sep 17 00:00:00 2001
From: Fuwn <contact@fuwn.me>
Date: Fri, 12 Jan 2024 17:23:28 -0800
Subject: feat(web): use any open port

---
 src/web.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

(limited to 'src/web.rs')

diff --git a/src/web.rs b/src/web.rs
index eb10ff2..1fa8a27 100644
--- a/src/web.rs
+++ b/src/web.rs
@@ -1,20 +1,21 @@
 use {thirtyfour::WebDriver, tokio::process::Command};
 
-pub fn geckodriver() -> Result<tokio::process::Child, std::io::Error> {
+pub fn geckodriver(port: u16) -> Result<tokio::process::Child, std::io::Error> {
   Command::new("geckodriver")
-    .arg("--port=9515")
+    .arg(format!("--port={port}"))
     .stdout(std::process::Stdio::piped())
     .stderr(std::process::Stdio::piped())
     .spawn()
 }
 
-pub async fn webdriver() -> Result<WebDriver, thirtyfour::error::WebDriverError>
-{
+pub async fn webdriver(
+  port: u16,
+) -> Result<WebDriver, thirtyfour::error::WebDriverError> {
   let mut caps = thirtyfour::DesiredCapabilities::firefox();
 
   caps.set_headless().expect("failed to set headless");
 
-  let driver = WebDriver::new("http://localhost:9515", caps)
+  let driver = WebDriver::new(&format!("http://localhost:{port}"), caps)
     .await
     .expect("failed to connect to webdriver");
 
-- 
cgit v1.2.3