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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
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 {
/// Input HTML file.
#[arg(short, long)]
input_path: String,
/// Output path for png image.
#[arg(short, long)]
output_path: String,
/// CSS locator to generate screenshot for.
#[arg(default_value_t = String::from("#app"))]
element_locator: 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 = ":5"; // env::var("DISPLAY")?;
let _xvfb = Command::new("Xvfb")
.arg(display).spawn()?;
info!("Current directory: {}", env::current_dir()?.display());
fs::create_dir(PROFILE_ROOT).ok();
let _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(&args.input_path)?;
info!("Opening {}", path.display());
c.goto(&format!("file://{}", path.display())).await?;
let e = c.find(Locator::Css(&args.element_locator)).await?;
let img = e.screenshot().await?;
let output_path = Path::new(&args.output_path);
let tmp_path = output_path.with_extension("png.tmp");
fs::write(&tmp_path, img)?;
Command::new("mogrify")
.arg("-colorspace")
.arg("gray")
.arg(tmp_path.as_os_str())
.spawn()?.wait()?;
fs::rename(&tmp_path, output_path)?;
time::sleep(Duration::from_secs(15)).await;
info!("Wrote {}", output_path.display());
}
// c.close().await?;
// geckodriver.kill()?;
// Ok(())
}
|
