kak-tree-sitter 3.2.1

Server between Kakoune and tree-sitter
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

//! Client implementation.
//!
//! This module exports the [`Client`] type that is used to send requests to the
//! server.

use std::{
  io::{self, Write},
  os::unix::net::UnixStream,
};

use crate::{error::OhNo, protocol::request::Request, server::resources::Paths};

/// Connected client (UNIX socket).
#[derive(Debug)]
pub struct Client {
  stream: UnixStream,
}

impl Client {
  pub fn connect(paths: &Paths) -> Result<Self, OhNo> {
    let stream = UnixStream::connect(paths.socket_path())
      .map_err(|err| OhNo::CannotConnectToServer { err })?;
    log::debug!("connected to KTS");

    Ok(Self { stream })
  }

  /// Convenient method to connect to the server and initiate a session.
  pub fn init_session(paths: &Paths, session: impl Into<String>) -> Result<(), OhNo> {
    let mut client = Self::connect(paths)?;
    client.send(&Request::init_session(session))
  }

  /// Asynchronously send a request.
  pub fn send(&mut self, req: &Request) -> Result<(), OhNo> {
    let bytes = serde_json::to_string(req).map_err(|err| OhNo::CannotSendRequest {
      err: err.to_string(),
    })?;

    loop {
      let r = self.stream.write_all(bytes.as_bytes());
      match r {
        Err(err) => match err.kind() {
          io::ErrorKind::Interrupted => continue,
          _ => {
            return Err(OhNo::CannotSendRequest {
              err: err.to_string(),
            });
          }
        },

        _ => return Ok(()),
      }
    }
  }
}