diff options
Diffstat (limited to 'crates/atuin-daemon/src/client.rs')
| -rw-r--r-- | crates/atuin-daemon/src/client.rs | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/crates/atuin-daemon/src/client.rs b/crates/atuin-daemon/src/client.rs index a832f9a9..f3fecdbc 100644 --- a/crates/atuin-daemon/src/client.rs +++ b/crates/atuin-daemon/src/client.rs @@ -1,8 +1,12 @@ use eyre::{eyre, Result}; -use tokio::net::UnixStream; +#[cfg(windows)] +use tokio::net::TcpStream; use tonic::transport::{Channel, Endpoint, Uri}; use tower::service_fn; +#[cfg(unix)] +use tokio::net::UnixStream; + use atuin_client::history::History; use crate::history::{ @@ -15,6 +19,7 @@ pub struct HistoryClient { // Wrap the grpc client impl HistoryClient { + #[cfg(unix)] pub async fn new(path: String) -> Result<Self> { let channel = Endpoint::try_from("http://atuin_local_daemon:0")? .connect_with_connector(service_fn(move |_: Uri| { @@ -30,6 +35,21 @@ impl HistoryClient { Ok(HistoryClient { client }) } + #[cfg(not(unix))] + pub async fn new(port: u64) -> Result<Self> { + let channel = Endpoint::try_from("http://atuin_local_daemon:0")? + .connect_with_connector(service_fn(move |_: Uri| { + let url = format!("127.0.0.1:{}", port); + TcpStream::connect(url) + })) + .await + .map_err(|_| eyre!("failed to connect to local atuin daemon. Is it running?"))?; + + let client = HistoryServiceClient::new(channel); + + Ok(HistoryClient { client }) + } + pub async fn start_history(&mut self, h: History) -> Result<String> { let req = StartHistoryRequest { command: h.command, |