ff-rdp-cli 0.1.0

CLI for Firefox Remote Debugging Protocol
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

use ff_rdp_core::LongStringActor;
use serde_json::json;

use crate::cli::args::Cli;
use crate::error::AppError;
use crate::hints::{HintContext, HintSource};
use crate::output;
use crate::output_pipeline::OutputPipeline;

use super::connect_tab::connect_and_get_target;
use super::js_helpers::eval_or_bail;

pub fn run(cli: &Cli) -> Result<(), AppError> {
    let mut ctx = connect_and_get_target(cli)?;
    let console_actor = ctx.target.console_actor.clone();

    let eval_result = eval_or_bail(
        &mut ctx,
        &console_actor,
        "document.body.innerText",
        "failed to extract page text",
    )?;

    let text = resolve_string_result(&mut ctx, &eval_result.result)?;

    let meta = json!({"host": cli.host, "port": cli.port});
    let envelope = output::envelope(&json!(text), 1, &meta);

    let hint_ctx = HintContext::new(HintSource::PageText);
    OutputPipeline::from_cli(cli)?
        .finalize_with_hints(&envelope, Some(&hint_ctx))
        .map_err(AppError::from)
}

/// Resolve a Grip to a string, fetching the full content if it's a LongString.
fn resolve_string_result(
    ctx: &mut super::connect_tab::ConnectedTab,
    grip: &ff_rdp_core::Grip,
) -> Result<String, AppError> {
    match grip {
        ff_rdp_core::Grip::Value(serde_json::Value::String(s)) => Ok(s.clone()),
        ff_rdp_core::Grip::LongString {
            actor,
            length,
            initial: _,
        } => LongStringActor::full_string(ctx.transport_mut(), actor.as_ref(), *length)
            .map_err(AppError::from),
        ff_rdp_core::Grip::Null | ff_rdp_core::Grip::Undefined => Ok(String::new()),
        other => Ok(other.to_json().to_string()),
    }
}