diff options
author | Dmitry Sharshakov | 2021-08-13 17:13:27 +0000 |
---|---|---|
committer | Blaž Hrastnik | 2021-08-20 04:43:54 +0000 |
commit | fabee039835e18ec464164f60182689f7489cfe7 (patch) | |
tree | e2674941bffe886884f05f3a0f163bb4334ddc91 /helix-dap/src | |
parent | 26a55dcefd9774235cd8c40eb2724daa0808e376 (diff) |
dap: support TCP clients
Diffstat (limited to 'helix-dap/src')
-rw-r--r-- | helix-dap/src/client.rs | 49 |
1 files changed, 35 insertions, 14 deletions
diff --git a/helix-dap/src/client.rs b/helix-dap/src/client.rs index 08f7704e..cbb24d15 100644 --- a/helix-dap/src/client.rs +++ b/helix-dap/src/client.rs @@ -7,7 +7,8 @@ use serde_json::{from_value, to_value, Value}; use std::process::Stdio; use std::sync::atomic::{AtomicU64, Ordering}; use tokio::{ - io::{BufReader, BufWriter}, + io::{AsyncBufRead, AsyncWrite, BufReader, BufWriter}, + net::TcpStream, process::{Child, Command}, sync::mpsc::{channel, UnboundedReceiver, UnboundedSender}, }; @@ -257,7 +258,7 @@ struct VariablesResponseBody { #[derive(Debug)] pub struct Client { id: usize, - _process: Child, + _process: Option<Child>, server_tx: UnboundedSender<Request>, server_rx: UnboundedReceiver<Payload>, request_counter: AtomicU64, @@ -265,7 +266,33 @@ pub struct Client { } impl Client { - pub fn start(cmd: &str, args: Vec<&str>, id: usize) -> Result<Self> { + pub fn streams( + rx: Box<dyn AsyncBufRead + Unpin + Send>, + tx: Box<dyn AsyncWrite + Unpin + Send>, + id: usize, + process: Option<Child>, + ) -> Result<Self> { + let (server_rx, server_tx) = Transport::start(rx, tx, id); + + let client = Self { + id, + _process: process, + server_tx, + server_rx, + request_counter: AtomicU64::new(0), + capabilities: None, + }; + + Ok(client) + } + + pub async fn tcp(addr: std::net::SocketAddr, id: usize) -> Result<Self> { + let stream = TcpStream::connect(addr).await?; + let (rx, tx) = stream.into_split(); + Self::streams(Box::new(BufReader::new(rx)), Box::new(tx), id, None) + } + + pub fn stdio(cmd: &str, args: Vec<&str>, id: usize) -> Result<Self> { let process = Command::new(cmd) .args(args) .stdin(Stdio::piped()) @@ -280,18 +307,12 @@ impl Client { let writer = BufWriter::new(process.stdin.take().expect("Failed to open stdin")); let reader = BufReader::new(process.stdout.take().expect("Failed to open stdout")); - let (server_rx, server_tx) = Transport::start(Box::new(reader), Box::new(writer), id); - - let client = Self { + Self::streams( + Box::new(BufReader::new(reader)), + Box::new(writer), id, - _process: process, - server_tx, - server_rx, - request_counter: AtomicU64::new(0), - capabilities: None, - }; - - Ok(client) + Some(process), + ) } pub fn id(&self) -> usize { |