1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
use std::{process::Command, error::Error, fs, env, time::Duration, path::Path};
use fantoccini::{ClientBuilder, Locator};
use log::info;
use tokio::time;
use clap::Parser;
static PROFILE_ROOT: &str = "/tmp/geckodriver-dashboard";
/// Simple program to greet a person
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
/// Output path for png image.
#[arg(short, long)]
path: String,
}
// let's set up the sequence of steps we want the browser to take
#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
pretty_env_logger::init();
let args = Args::parse();
let display = env::var("DISPLAY")?;
info!("Current directory: {}", env::current_dir()?.display());
fs::create_dir(PROFILE_ROOT).ok();
let mut geckodriver = Command::new("geckodriver")
.env("MOZ_REMOTE_SETTINGS_DEVTOOLS", "1")
.env("DISPLAY", display)
.arg("--profile-root").arg("/tmp/geckodriver-dashboard").spawn()?;
let c = ClientBuilder::native().connect("http://localhost:4444").await.expect("failed to connect to WebDriver");
loop {
let path = fs::canonicalize("index.html")?;
info!("Opening {}", path.display());
c.goto(&format!("file://{}", path.display())).await?;
let e = c.find(Locator::Css("#app")).await?;
let img = e.screenshot().await?;
let output_path = Path::new(&args.path);
let tmp_path = output_path.join(".tmp");
fs::write(&tmp_path, img)?;
fs::rename(&tmp_path, output_path)?;
time::sleep(Duration::from_secs(15)).await;
info!("Wrote {}", output_path.display());
}
// c.close().await?;
// geckodriver.kill()?;
// Ok(())
}
|
