paul/autarco-scraper
paul
/
autarco-scraper
Archived
1
0
Fork 0
Code Issues 2 Pull Requests Releases Activity

Pass the gecko driver port as spawn argument

This commit is contained in:
Paul van Tilburg 2020-10-09 17:44:03 +02:00
parent 00aeed1803
commit 99d8a4c133
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main.rs
View File

@ -3,6 +3,7 @@ use lazy_static::lazy_static;
use rocket::{get, launch, routes, Rocket};
use rocket_contrib::json::Json;
use serde::Serialize;
use std::process::{Child, Command, Stdio};
use std::sync::Mutex;
use std::thread;
use std::time::{Duration, SystemTime};
@ -15,18 +16,16 @@ const POLL_INTERVAL: u64 = 300;
const GECKO_DRIVER_PORT: u16 = 18019;
use std::process::{Child, Command, Stdio};
struct GeckoDriver(Child);
impl GeckoDriver {
pub fn spawn() -> Result<Self> {
pub fn spawn(port: u16) -> Result<Self> {
// This is taken from the webdriver-client crate.
let child = Command::new("geckodriver")
.arg("-b")
.arg("firefox")
.arg("--port")
.arg(format!("{}", GECKO_DRIVER_PORT))
.arg(format!("{}", port))
.stdin(Stdio::null())
.stderr(Stdio::null())
.stdout(Stdio::null())
@ -78,7 +77,7 @@ lazy_static! {
async fn update_loop() -> Result<()> {
color_eyre::install()?;
let _gecko_driver = GeckoDriver::spawn()?;
let _gecko_driver = GeckoDriver::spawn(GECKO_DRIVER_PORT)?;
let mut caps = DesiredCapabilities::firefox();
caps.set_headless()?;
let driver = WebDriver::new(&format!("http://localhost:{}", GECKO_DRIVER_PORT), &caps).await?;