helix-mirror/helix-dap/examples/dap-dlv.rs

use helix_dap::{Client, Event, OutputEventBody, Result, SourceBreakpoint, StoppedEventBody};
use serde::{Deserialize, Serialize};
use serde_json::from_value;
use tokio::sync::mpsc::Receiver;

#[derive(Debug, PartialEq, Clone, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
struct LaunchArguments {
    mode: String,
    program: String,
}

async fn output(mut output_event: Receiver<Event>) {
    loop {
        let body: OutputEventBody =
            from_value(output_event.recv().await.unwrap().body.unwrap()).unwrap();
        println!(
            "> [{}] {}",
            body.category.unwrap_or("unknown".to_owned()),
            body.output
        );
    }
}

#[tokio::main]
pub async fn main() -> Result<()> {
    let base_config = fern::Dispatch::new().level(log::LevelFilter::Info);

    let stderr_config = fern::Dispatch::new()
        .format(|out, message, record| out.finish(format_args!("[{}] {}", record.level(), message)))
        .chain(std::io::stderr());

    base_config
        .chain(stderr_config)
        .apply()
        .expect("Failed to set up logging");

    let client = Client::tcp_process("dlv", vec!["dap"], "-l 127.0.0.1:{}", 0).await;
    println!("create: {:?}", client);
    let mut client = client?;

    let output_event = client.listen_for_event("output".to_owned()).await;
    tokio::spawn(output(output_event));

    println!("init: {:?}", client.initialize("go".to_owned()).await);
    println!("caps: {:?}", client.capabilities());

    let args = LaunchArguments {
        mode: "exec".to_owned(),
        program: "/tmp/godebug/main".to_owned(),
    };

    println!("launch: {:?}", client.launch(args).await);

    println!(
        "breakpoints: {:#?}",
        client
            .set_breakpoints(
                "/tmp/godebug/main.go".to_owned(),
                vec![SourceBreakpoint {
                    line: 8,
                    column: Some(2),
                    condition: None,
                    hit_condition: None,
                    log_message: None,
                }]
            )
            .await
    );

    let mut _in = String::new();
    std::io::stdin()
        .read_line(&mut _in)
        .expect("Failed to read line");

    let mut stopped_event = client.listen_for_event("stopped".to_owned()).await;

    println!("configurationDone: {:?}", client.configuration_done().await);

    let stop: StoppedEventBody =
        from_value(stopped_event.recv().await.unwrap().body.unwrap()).unwrap();
    println!("stopped: {:?}", stop);

    let threads = client.threads().await?;
    println!("threads: {:#?}", threads);
    let bt = client
        .stack_trace(threads[0].id)
        .await
        .expect("expected stack trace");
    println!("stack trace: {:#?}", bt);
    let scopes = client
        .scopes(bt.0[0].id)
        .await
        .expect("expected scopes for thread");
    println!("scopes: {:#?}", scopes);
    println!(
        "vars: {:#?}",
        client.variables(scopes[1].variables_reference).await
    );

    let mut _in = String::new();
    std::io::stdin()
        .read_line(&mut _in)
        .expect("Failed to read line");

    println!("continued: {:?}", client.continue_thread(0).await);

    let mut _in = String::new();
    std::io::stdin()
        .read_line(&mut _in)
        .expect("Failed to read line");

    println!("disconnect: {:?}", client.disconnect().await);

    Ok(())
}