eoka-cli 0.10.0

Browser automation CLI for AI agents and pentesting
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use crate::cli::Cli;
use crate::launch_spec::{self, LaunchSpec};

pub(crate) fn resolve_launch_spec(cli: &Cli) -> Result<LaunchSpec, String> {
    if cli.auto_solve_cloudflare && (cli.cdp.is_some() || cli.auto_connect) {
        return Err(
            "--auto-solve-cloudflare needs a browser eoka launches, not --cdp/--auto-connect"
                .into(),
        );
    }
    if let Some(spec) = &cli.cdp {
        let ws_url = launch_spec::resolve_cdp_spec(spec)?;
        return Ok(LaunchSpec::Connect { ws_url });
    }
    if cli.auto_connect {
        let (_port, ws_url) = launch_spec::auto_connect()?;
        return Ok(LaunchSpec::Connect { ws_url });
    }
    if !cli.daemon
        && !cli.headed
        && !cli.headless
        && cli.from_profile.is_none()
        && cli.clone_state_from.is_none()
        && !cli.no_stealth
        && cli.proxy.is_none()
        && cli.proxy_file.is_none()
        && !cli.no_js
        && cli.js_allow.is_empty()
        && cli.js_block.is_empty()
        && !cli.persist
        && !cli.no_geo_align
        && !cli.no_sandbox
        && !cli.auto_solve_cloudflare
    {
        return Ok(LaunchSpec::Reuse);
    }
    Ok(LaunchSpec::Launch {
        headless: !cli.headed,
        from_profile: cli
            .from_profile
            .as_deref()
            .map(resolve_profile_spec)
            .transpose()?,
        clone_state_from: cli.clone_state_from.clone(),
        no_stealth: cli.no_stealth,
        proxy: if cli.proxy_stdin {
            Some(read_proxy_from_stdin()?)
        } else {
            launch_spec::resolve_proxy_spec(cli.proxy.as_deref(), cli.proxy_file.as_deref())?
        },
        no_js: cli.no_js,
        js_allow: cli.js_allow.clone(),
        js_block: cli.js_block.clone(),
        persist: cli.persist,
        geo_align: !cli.no_geo_align,
        auto_solve_cloudflare: cli.auto_solve_cloudflare,
        sandbox: !cli.no_sandbox,
    })
}

fn read_proxy_from_stdin() -> Result<String, String> {
    let mut line = String::new();
    std::io::stdin()
        .read_line(&mut line)
        .map_err(|error| format!("Failed to read proxy from stdin: {error}"))?;
    let proxy = line.trim();
    if proxy.is_empty() {
        return Err("--proxy-stdin received no proxy".into());
    }
    Ok(proxy.to_owned())
}

fn resolve_profile_spec(spec: &str) -> Result<std::path::PathBuf, String> {
    if spec == "auto" {
        return crate::handler::profile::default_profile_dir()
            .ok_or_else(|| "Could not autodetect Chrome profile dir".to_string());
    }
    let p = std::path::PathBuf::from(spec);
    if !p.exists() {
        return Err(format!("Profile path does not exist: {}", p.display()));
    }
    Ok(p)
}

pub(crate) fn effective_session(cli: &Cli, _spec: &LaunchSpec) -> String {
    cli.session.clone()
}